def run(): """ Run the macro """ # This input generates empty spills, to be filled by the beam maker later on input_file = open('maus_output.json', 'r') # my_input = MAUS.InputPyJSON(input_file) #my_input = MAUS.InputPySpillGenerator() # Create an empty array of mappers, then populate it # with the functionality you want to use. my_map = MAUS.MapPyGroup() #my_map.append(MAUS.MapPyBeamMaker()) # beam construction #my_map.append(MAUS.MapCppSimulation()) # geant4 simulation my_map.append( MAUS.MapCppTrackerMCDigitization()) # SciFi electronics model my_map.append(MAUS.MapCppTrackerRecon()) # SciFi recon # can specify datacards here or by using appropriate command line calls datacards = io.StringIO(u"") #reducer = MAUS.ReduceCppTracker() reducer = MAUS.ReducePyDoNothing() output_file = open("recon_mc", 'w') # Uncompressed # 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 # (default simulation.out) for output MAUS.Go(my_input, my_map, reducer, my_output, datacards)
def run(): """ Run the macro """ # Use the G4BL JSON chunks as an input to the simulation my_input = MAUS.InputPyJSON() # Create an empty array of mappers, then populate it # with the functionality you want to use. my_map = MAUS.MapPyGroup() # No need for the beam maker, as we use G4BL chunks # my_map.append(MAUS.MapPyBeamMaker()) # beam construction # Run the GEANT4 simulation my_map.append(MAUS.MapCppSimulation()) # geant4 simulation # Pre detector set up my_map.append(MAUS.MapCppMCReconSetup()) # geant4 simulation # TOF my_map.append(MAUS.MapCppTOFMCDigitizer()) # TOF MC Digitizer my_map.append(MAUS.MapCppTOFSlabHits()) # TOF MC Slab Hits my_map.append(MAUS.MapCppTOFSpacePoints()) # TOF Space Points # KL my_map.append(MAUS.MapCppKLMCDigitizer()) # KL MC Digitizer my_map.append(MAUS.MapCppKLCellHits()) # KL CellHit Reco # SciFi my_map.append( MAUS.MapCppTrackerMCDigitization()) # SciFi electronics model my_map.append(MAUS.MapCppTrackerClusterRecon()) # SciFi channel clustering my_map.append( MAUS.MapCppTrackerSpacePointRecon()) # SciFi spacepoint recon my_map.append( MAUS.MapCppTrackerPatternRecognition()) # SciFi track finding my_map.append(MAUS.MapCppTrackerPRSeed()) # Set the Seed from PR my_map.append(MAUS.MapCppTrackerTrackFit()) # SciFi track fit # EMR my_map.append(MAUS.MapCppEMRMCDigitization()) # EMR MC Digitization my_map.append(MAUS.MapCppEMRSpacePoints()) # EMR MC Digitization my_map.append(MAUS.MapCppEMRRecon()) # EMR Recon # Ckov my_map.append(MAUS.MapCppCkovMCDigitizer()) # Global Digits - post detector digitisation # Then construct a MAUS output component - filename comes from datacards my_output = MAUS.OutputCppRoot() # can specify datacards here or by using appropriate command line calls datacards = io.StringIO(u"") # The Go() drives all the components you pass in, then check the file # (default simulation.out) for output MAUS.Go(my_input, my_map, MAUS.ReducePyDoNothing(), my_output, datacards)
def run(input_beam, configuration, out_file): """run""" document_file = io.StringIO(unicode(input_beam)) my_input = MAUS.InputPyJSON(document_file) my_map = MAUS.MapCppSimulation() my_output = MAUS.OutputPyJSON(open(out_file, 'w')) MAUS.Go(my_input, my_map, MAUS.ReducePyDoNothing(), my_output, io.StringIO(unicode(configuration)))
def run(): """Run the macro""" # This generates events (usually spills) from a json-formatted ascii file my_input = MAUS.InputPyJSON() # This outputs events (usually spills) to a root-formatted binary file my_output = MAUS.OutputCppRoot() # Execute inputter and outputter # Mapper and Reducer does nothing MAUS.Go(my_input, MAUS.MapPyDoNothing(), MAUS.ReducePyDoNothing(), \ my_output, io.StringIO(u""))
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" : 13,"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.MapCppTOFMCDigitizer()) # TOF electronics model my_map.append(MAUS.MapCppTrackerMCDigitization()) # SCiFi electronics datacards = io.StringIO(u"keep_steps = True") # 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(os.environ["MAUS_ROOT_DIR"] + "/tmp/simulation.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)
def run(number_of_spills): #pylint: disable =W0621 """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. """ # 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: # # documentFile = open('myFileName.txt', 'r') # # where the file format has a JSON document per line. I just toss the file # in here for simplicity. document_file = io.StringIO( number_of_spills * u"""{"mc": [{"position": { "x": 0.0, "y": -0.0, "z": -5000 },"particle_id" : 13,"energy" : 210, "random_seed" : 10, "unit_momentum" : { "x":0, "y":0, "z":1 }}]}\n""" ) #pylint: disable =C0301 my_input = MAUS.InputPyJSON(document_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.MapCppTOFMCDigitizer()) # TOF detector digitization my_map.append(MAUS.MapCppTrackerMCDigitization()) # tracker digitization datacards = io.StringIO(u"keep_tracks = False\n"\ "simulation_geometry_filename = \"Stage5.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') # 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(), MAUS.OutputPyJSON(), datacards)