#!/usr/bin/env python

#################################################################

###!!! YOU ARE NOT ALLOWED TO MODIFY THIS FILE DIRECTLY    !!!###

###!!! PLEASE MAKE A COPY OF THIS FILE WITH THE CP COMMAND !!!###

#################################################################

"""Simulate the MICE experiment

This will simulate 'number_of_spills' MICE events through the entirity

of MICE using Geant4.  At present, TOF and Tracker hits will be digitized.

"""

import io   #  generic python library for I/O

import gzip #  For compressed output # pylint: disable=W0611

import MAUS

def run(number_of_spills):

    """ Run the macro

    """

    # Here we create a pseudo-file with an event in it.  If you were to copy

    # and paste this to a file, then you could also do:

    #

    #   input_file = open('myFileName.txt', 'r')

    #

    # where the file format has a JSON document per line.  I just toss the file

    # in here for simplicity.

    input_file = io.StringIO(number_of_spills*u"""{"mc": [{"primary":{"position": { "x": 0.0, "y": -0.0, "z": -5000.0 },"particle_id" : 11,"energy" : 210.0, "random_seed" : 10, "momentum" : { "x":0.0, "y":0.0, "z":1.0 }, "time" : 0.0}}]}\n""") # pylint: disable=C0301  

    my_input = MAUS.InputPyJSON(input_file)

    # Create an empty array of mappers, then populate it

    # with the functionality you want to use.

    my_map = MAUS.MapPyGroup()

    my_map.append(MAUS.MapCppSimulation())  #  geant4 simulation

    my_map.append(MAUS.MapCppTOFDigitization())  #  TOF electronics model

    my_map.append(MAUS.MapCppTrackerDigitization())  #   SCiFi electronics model

#    datacards = io.StringIO(u"keep_steps = True")

    datacards = io.StringIO(u"keep_tracks = True\nsimulation_geometry_filename=\"Stage4-stefania.dat\"")

    #  You may specify datacards if you wish.  To do so you create a file object

    #  which can either be a StringIO object or a native python file.  If you

    #  want to store your datacards in a file 'datacards.dat' then uncomment:

    # datacards = open('datacards.dat', 'r')

    #  Choose from either a compressed or uncompressed output file

    #

    output_file = open("simulation_electron.out", 'w')  #  Uncompressed

    #output_file = gzip.GzipFile("mausput.gz", 'wb')  #  Compressed

    #

    # Then construct a MAUS output component

    my_output = MAUS.OutputPyJSON(output_file)

    #  The Go() drives all the components you pass in, then check the file

    #  'mausput' for the output

    MAUS.Go(my_input, my_map, MAUS.ReducePyDoNothing(), my_output, datacards)

if __name__ == '__main__':

    NUMBER_OF_SPILLS = 1

    run(NUMBER_OF_SPILLS)