# This file is part of MAUS: http://micewww.pp.rl.ac.uk/projects/maus
#
# MAUS is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# MAUS is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with MAUS.  If not, see <http://www.gnu.org/licenses/>.

"""
Does some global data structure set up
"""

import json

class MapPyMCReconSetup:
    """
    Creates a recon_events branch containing a list of branches one for each
    detector
    """
    def __init__(self):
        """
        Does nothing
        """
        pass

    def birth(self, json_configuration): #pylint: disable=W0613, R0201
        """
        Just returns true
        """
        return True

    def process(self, json_document): #pylint: disable=R0201
        """
        Set up recon_events branch

        Loop over the mc_events and for each one make a new recon_event
        """
        try:
            spill = json.loads(json_document)
        except ValueError:
            spill = {"errors": {"bad_json_document":
                                "unable to do json.loads on input"} }
            return json.dumps(spill)

        # we make spill a dict initially to ensure a one-to-one mapping of
        # part_event_numbers to recon_events
        spill["recon_events"] = []

        # check data integrity
        if "mc_events" not in spill or type(spill["mc_events"]) != type([]):
            return json.dumps(spill)

        # loop over all triggers and fill the recon event
        for i, event in enumerate(spill["mc_events"]): #pylint: disable=W0612
            spill["recon_events"].append({"part_event_number":i,
                                          "trigger_event":{},
                                          "tof_event":{},
                                          "sci_fi_event":{},
                                          "ckov_event":{},
                                          "emr_event":{},
                                          "kl_event":{},
                                          "global_event":{}})
        return json.dumps(spill)

    def death(self): #pylint: disable=R0201
        """
        Just returns true
        """
        return True