DDDigi/scripts/TestHitProcessing.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   input_action.adopt_action('DigiDDG4ROOT/SignalReader', mask=0x0, input=[digi.next_input()])
0021   # ========================================================================================================
0022   digi.info('Creating collision overlays....')
0023   # ========================================================================================================
0024   overlay = input_action.adopt_action('DigiSequentialActionSequence/Overlay-1')
0025   overlay.adopt_action('DigiDDG4ROOT/Read-1', mask=0x1, input=[digi.next_input()])
0026   digi.info('Created input.overlay-1')
0027   # ========================================================================================================
0028   overlay = input_action.adopt_action('DigiSequentialActionSequence/Overlay-2')
0029   overlay.adopt_action('DigiDDG4ROOT/Read-2', mask=0x2, input=[digi.next_input()])
0030   digi.info('Created input.overlay-2')
0031   # ========================================================================================================
0032   event = digi.event_action('DigiSequentialActionSequence/EventAction')
0033   event.adopt_action('DigiStoreDump/StoreDump')
0034   combine = event.adopt_action('DigiContainerCombine/Combine',
0035                                parallel=True,
0036                                input_masks=[0x0, 0x1, 0x2],
0037                                output_mask=0xAAA0,
0038                                output_segment='inputs')
0039   combine.erase_combined = True
0040   proc = event.adopt_action('DigiContainerSequenceAction/HitP2',
0041                             parallel=True,
0042                             input_mask=0xAAA0,
0043                             input_segment='inputs',
0044                             output_mask=0xAAA1,
0045                             output_segment='inputs')
0046   count = digi.create_action('DigiDepositWeightedPosition/DepoCombine')
0047   proc.adopt_container_processor(count, digi.containers())
0048   event.adopt_action('DigiStoreDump/DumpWeighted')
0049   # ========================================================================================================
0050   digi.info('Starting digitization core')
0051   digi.run_checked(num_events=1, num_threads=25, parallel=5)
0052
0053
0054 if __name__ == '__main__':
0055   run()