9 from ROOT
import gROOT, Belle2
10 gROOT.ProcessLine(
"gErrorIgnoreLevel = 4000;")
13 Compare the CDCTrigger2DFitterModule output with the old TRGCDCModule,
14 to confirm that the behaviour of the new code is correct.
25 particlegun_params = {
26 'pdgCodes': [-13, 13],
28 'momentumGeneration':
'inversePt',
29 'momentumParams': [0.3, 10.],
30 'thetaGeneration':
'uniform',
31 'thetaParams': [35, 123],
32 'phiGeneration':
'uniform',
33 'phiParams': [0, 360],
34 'vertexGeneration':
'fixed',
35 'xVertexParams': [0.],
36 'yVertexParams': [0.],
37 'zVertexParams': [0.]}
39 bkgdir =
'/sw/belle2/bkg/'
46 basf2.set_random_seed(seed)
48 basf2.set_log_level(basf2.LogLevel.ERROR)
50 main = basf2.create_path()
52 main.add_module(
'EventInfoSetter', evtNumList=evtnum)
53 main.add_module(
'Progress')
54 main.add_module(
'Gearbox')
55 main.add_module(
'Geometry', components=[
'CDC',
56 'MagneticFieldConstant4LimitedRCDC'])
57 particlegun = basf2.register_module(
'ParticleGun')
58 particlegun.param(particlegun_params)
59 main.add_module(particlegun)
60 main.add_module(
'FullSim')
62 bkgmixer = basf2.register_module(
'BeamBkgMixer')
63 bkgfiles = glob.glob(os.path.join(bkgdir,
'*[!(PXD)(ECL)]??.root'))
64 bkgmixer.param(
'backgroundFiles', bkgfiles)
65 bkgmixer.param(
'components', [
'CDC'])
66 main.add_module(bkgmixer)
67 cdcdigitizer = basf2.register_module(
'CDCDigitizer')
69 cdcdigitizer.param(
'TrigTimeJitter', 32.)
70 main.add_module(cdcdigitizer)
76 trgcdc = basf2.register_module(
'TRGCDC')
85 'FastSimulationMode': simMode,
89 trgcdc.param(trgcdc_params)
91 trgcdc.param(
'inputCollection',
'CDCHits4Trg')
92 main.add_module(trgcdc)
95 main.add_module(
'CDCTriggerETF')
104 test module to compare the output of TRGCDC and CDCTriggerETF
109 give info for both modules and warnings in the case of mismatches
113 newT0 =
Belle2.PyStoreObj(
"BinnedEventT0").obj().getBinnedEventT0(Belle2.Const.CDC)
117 basf2.B2INFO(
"T0 %d" % oldT0)
119 basf2.B2WARNING(
"old T0 %d, new T0 %d" % (oldT0, newT0))
122 main.add_module(
TestModule(), logLevel=basf2.LogLevel.INFO)
128 print(basf2.statistics)