DDDigi/scripts/TestEdm4hepOutput.py

0001 # ==========================================================================
0002 #  AIDA Detector description implementation
0003 # --------------------------------------------------------------------------
0004 # Copyright (C) Organisation europeenne pour la Recherche nucleaire (CERN)
0005 # All rights reserved.
0006 #
0007 # For the licensing terms see $DD4hepINSTALL/LICENSE.
0008 # For the list of contributors see $DD4hepINSTALL/doc/CREDITS.
0009 #
0010 # ==========================================================================
0011
0012
0013 # ---------------------------------------------------------------------------
0014 def run():
0015   import DigiTest
0016   digi = DigiTest.Test(geometry=None)
0017   read = digi.input_action('DigiDDG4ROOT/SignalReader', mask=0x0, input=[digi.next_input()])
0018   dump = digi.event_action('DigiStoreDump/StoreDump', parallel=False)
0019   writ = digi.output_action('DigiEdm4hepOutput/Writer',
0020                             parallel=True,
0021                             input_mask=0x0,
0022                             input_segment='input',
0023                             output='MiniTel_DDDigi_edm4hep_data.root')
0024   proc = digi.create_action('DigiEdm4hepOutputProcessor/edm4hep')
0025   hit_type = 'TrackerHits'
0026   if digi.hit_type:
0027     hit_type = digi.hit_type
0028   cont = [c + '/' + hit_type for c in digi.containers()]
0029   writ.adopt_container_processor(proc, cont)
0030   writ.adopt_container_processor(proc, 'MCParticles/MCParticles')
0031   digi.check_creation([read, dump])
0032   digi.run_checked(num_events=10, num_threads=10, parallel=3)
0033
0034
0035 # ---------------------------------------------------------------------------
0036 if __name__ == '__main__':
0037   run()