DDDigi/scripts/TestMultiContainerParallel.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 import dddigi
0013
0014
0015 def run():
0016   import DigiTest
0017   digi = DigiTest.Test(geometry=None)
0018   # digi.load_geo()
0019   input_action = digi.input_action('DigiParallelActionSequence/READER')
0020   # ========================================================================
0021   digi.info('Created SIGNAL input')
0022   signal = input_action.adopt_action('DigiDDG4ROOT/SignalReader',
0023                                      mask=0xFEED,
0024                                      input=[digi.next_input()])
0025   digi.check_creation([signal])
0026   # ========================================================================
0027   event = digi.event_action('DigiSequentialActionSequence/EventAction')
0028   proc = event.adopt_action('DigiMultiContainerProcessor/ContainerProc',
0029                             parallel=True, input_masks=[0x0, 0x1, 0x2, 0x3, 0xFEED])
0030   conts = digi.containers(2)
0031   for i in range(len(conts)):
0032     merge = dddigi.Action(digi.kernel(), 'DigiContainerProcessor/SegmentPrint_%03d' % (i,))
0033     proc.adopt_processor(merge, conts[i])
0034   # ========================================================================
0035   digi.run_checked(num_events=5, num_threads=15, parallel=3)
0036
0037
0038 if __name__ == '__main__':
0039   run()