DDDigi/scripts/TestWriteDigi.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 from __future__ import absolute_import
0012
0013
0014 # ---------------------------------------------------------------------------
0015 def run():
0016   import DigiTest
0017   digi = DigiTest.Test(geometry=None)
0018   read = digi.input_action('DigiDDG4ROOT/SignalReader', mask=0x0, input=[digi.next_input()])
0019   dump = digi.event_action('DigiStoreDump/StoreDump', parallel=False)
0020   writ = digi.output_action('Digi2ROOTWriter/EventWriter',
0021                             parallel=True,
0022                             input_mask=0x0,
0023                             input_segment='input',
0024                             output='dddigi_write_digi.root')
0025   proc = digi.create_action('Digi2ROOTProcessor/Writer')
0026   hit_type = 'TrackerHits'
0027   if digi.hit_type:
0028     hit_type = digi.hit_type
0029   cont = [c + '/' + hit_type for c in digi.containers()]
0030   writ.adopt_container_processor(proc, cont)
0031   writ.adopt_container_processor(proc, 'MCParticles/MCParticles')
0032   digi.check_creation([read, dump])
0033   digi.run_checked(num_events=5, num_threads=10, parallel=5)
0034
0035
0036 # ---------------------------------------------------------------------------
0037 if __name__ == '__main__':
0038   run()