DDDigi/scripts/TestMultiInteractions.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 def run():
0014   import DigiTest
0015   digi = DigiTest.Test(geometry=None)
0016
0017   input_action = digi.input_action('DigiParallelActionSequence/READER')
0018   # ========================================================================================================
0019   digi.info('Created SIGNAL input')
0020   signal = input_action.adopt_action('DigiDDG4ROOT/SignalReader', mask=0x0, input=[digi.next_input()])
0021   digi.check_creation([signal])
0022   # ========================================================================================================
0023   digi.info('Creating collision overlays....')
0024   # ========================================================================================================
0025   overlay = input_action.adopt_action('DigiSequentialActionSequence/Overlay-1')
0026   evtreader = overlay.adopt_action('DigiDDG4ROOT/Reader-1', mask=0x1, input=[digi.next_input()])
0027   hist_drop = overlay.adopt_action('DigiHitHistoryDrop/Drop-1', masks=[evtreader.mask])
0028   digi.check_creation([overlay, evtreader, hist_drop])
0029   digi.info('Created input.overlay25')
0030   # ========================================================================================================
0031   overlay = input_action.adopt_action('DigiSequentialActionSequence/Overlay-2')
0032   evtreader = overlay.adopt_action('DigiDDG4ROOT/Reader-2', mask=0x2, input=[digi.next_input()])
0033   hist_drop = overlay.adopt_action('DigiHitHistoryDrop/Drop-2', masks=[evtreader.mask])
0034   digi.check_creation([overlay, evtreader, hist_drop])
0035   digi.info('Created input.overlay50')
0036   # ========================================================================================================
0037   overlay = input_action.adopt_action('DigiSequentialActionSequence/Overlay-3')
0038   evtreader = overlay.adopt_action('DigiDDG4ROOT/Reader-3', mask=0x3, input=[digi.next_input()])
0039   hist_drop = overlay.adopt_action('DigiHitHistoryDrop/Drop-3', masks=[evtreader.mask])
0040   digi.check_creation([overlay, evtreader, hist_drop])
0041   digi.info('Created input.overlay75')
0042   # ========================================================================================================
0043   event = digi.event_action('DigiSequentialActionSequence/EventAction')
0044   combine = event.adopt_action('DigiContainerCombine/Combine',
0045                                parallel=True,
0046                                input_masks=[0x0, 0x1, 0x2, 0x3],
0047                                output_mask=0xFEED,
0048                                output_segment='deposits',
0049                                erase_combined=False)
0050   combine.erase_combined = True  # Not thread-safe! only do in SequentialActionSequence
0051   dump = event.adopt_action('DigiStoreDump/StoreDump')
0052   digi.check_creation([combine, dump])
0053   digi.info('Created event.dump')
0054
0055   # ========================================================================================================
0056   digi.run_checked(num_events=5, num_threads=10, parallel=3)
0057
0058
0059 if __name__ == '__main__':
0060   run()