19 set_log_level(LogLevel.ERROR)
22 basf2datadir = os.path.join(os.environ.get(
'BELLE2_LOCAL_DIR',
None),
'data')
27 evtmetagen = register_module(
'EventInfoSetter')
28 evtmetainfo = register_module(
'Progress')
29 paramloader = register_module(
'Gearbox')
30 geobuilder = register_module(
'Geometry')
31 particlegun = register_module(
'ParticleGun')
32 evtgeninput = register_module(
'EvtGenInput')
33 kkgeninput = register_module(
'KKGenInput')
34 mcparticleprinter = register_module(
'PrintMCParticles')
35 g4sim = register_module(
'FullSim')
36 bkgmixer = register_module(
'BeamBkgMixer')
37 cdcdigitizer = register_module(
'CDCDigitizer')
38 cdctrg = register_module(
"TRGCDC")
39 rootOut = register_module(
'RootOutput')
40 rootIn = register_module(
'RootInput')
41 neuro = register_module(
'NeuroTrigger')
48 simulatedComponents = [
'MagneticField',
'CDC',
'PXD',
'SVD',
'BeamPipe'
56 evtmetagen.param({
'evtNumList': [1000],
'runList': [1]})
60 geobuilder.param(
'components', simulatedComponents)
65 particlegun.param(
'pdgCodes', [13, -13])
66 particlegun.param(
'nTracks', 5)
67 particlegun.param(
'momentumGeneration',
'inversePt')
70 particlegun.param(
'momentumParams', [0.2, 8.0])
75 particlegun.param(
'phiGeneration',
'uniform')
76 particlegun.param(
'phiParams', [0, 360])
79 particlegun.param(
'vertexGeneration',
'uniform')
80 particlegun.param(
'xVertexParams', [0, 0])
81 particlegun.param(
'yVertexParams', [0, 0])
82 particlegun.param(
'zVertexParams', [-20.0, 20.0])
91 kkdir = os.path.join(os.environ.get(
'BELLE2_LOCAL_DIR',
None),
'generators')
92 kkgeninput.param(
'tauinputFile', kkdir +
'kkmc/data/mu.input.dat')
93 kkgeninput.param(
'KKdefaultFile', kkdir +
'kkmc/data/KK2f_defaults.dat')
94 kkgeninput.param(
'taudecaytableFile',
'')
97 mcparticleprinter.param(
'maxLevel', -1)
109 dir =
'/sw/belle2/bkg/'
118 bkg_files = glob.glob(dir +
'/*.root')
120 bkgmixer.param(
'backgroundFiles', bkg_files)
121 bkgmixer.param(
'components', simulatedComponents)
122 bkgmixer.param(
'scaleFactors', [
123 (
'Coulomb_LER', bkgScaleFactor),
124 (
'Coulomb_HER', bkgScaleFactor),
125 (
'RBB_LER', bkgScaleFactor),
126 (
'RBB_HER', bkgScaleFactor),
127 (
'Touschek_LER', bkgScaleFactor),
128 (
'Touschek_HER', bkgScaleFactor)
144 cdctrg.param(
'ConfigFile', os.path.join(basf2datadir,
"trg/cdc/TRGCDCConfig_0_20101111.dat"))
146 cdctrg.param(
'CurlBackStop', 1)
147 cdctrg.param(
'SimulationMode', 1)
148 cdctrg.param(
'FastSimulationMode', 0)
154 cdctrg.param(
'InnerTSLUTFile', os.path.join(basf2datadir,
"trg/cdc/innerLUT_v3.0.coe"))
155 cdctrg.param(
'OuterTSLUTFile', os.path.join(basf2datadir,
"trg/cdc/outerLUT_v3.0.coe"))
160 cdctrg.param(
'HoughFinderMappingFileMinus', os.path.join(basf2datadir,
"trg/cdc/HoughMappingMinus20160223.dat"))
161 cdctrg.param(
'HoughFinderMappingFilePlus', os.path.join(basf2datadir,
"trg/cdc/HoughMappingPlus20160223.dat"))
180 neuro.param(
'filename', os.path.join(basf2datadir,
"trg/cdc/Neuro20160309Nonlin.root"))
187 rootOut.param(
'outputFileName',
'basf2.root')
190 rootIn.param(
'inputFileName',
'basf2.root')
198 fullMain = create_path()
200 fullMain.add_module(evtmetagen)
201 fullMain.add_module(evtmetainfo)
202 fullMain.add_module(paramloader)
203 fullMain.add_module(geobuilder)
204 fullMain.add_module(particlegun)
207 fullMain.add_module(mcparticleprinter)
208 fullMain.add_module(g4sim)
209 fullMain.add_module(bkgmixer)
210 fullMain.add_module(cdcdigitizer)
211 fullMain.add_module(cdctrg)
212 fullMain.add_module(neuro)
215 g4SimMain = create_path()
217 g4SimMain.add_module(evtmetagen)
218 g4SimMain.add_module(evtmetainfo)
219 g4SimMain.add_module(paramloader)
220 g4SimMain.add_module(geobuilder)
221 g4SimMain.add_module(particlegun)
224 g4SimMain.add_module(mcparticleprinter)
225 g4SimMain.add_module(g4sim)
226 g4SimMain.add_module(bkgmixer)
227 g4SimMain.add_module(cdcdigitizer)
228 g4SimMain.add_module(rootOut)
231 savedG4SimMain = create_path()
233 savedG4SimMain.add_module(rootIn)
234 savedG4SimMain.add_module(evtmetainfo)
235 savedG4SimMain.add_module(paramloader)
236 savedG4SimMain.add_module(geobuilder)
237 savedG4SimMain.add_module(cdctrg)
240 generatorMain = create_path()
242 generatorMain.add_module(evtmetagen)
243 generatorMain.add_module(evtmetainfo)
244 generatorMain.add_module(paramloader)
245 generatorMain.add_module(geobuilder)
246 generatorMain.add_module(particlegun)
249 generatorMain.add_module(mcparticleprinter)
250 generatorMain.add_module(g4sim)
251 generatorMain.add_module(rootOut)
254 savedGeneratorMain = create_path()
256 savedGeneratorMain.add_module(rootIn)
257 savedGeneratorMain.add_module(evtmetainfo)
258 savedGeneratorMain.add_module(paramloader)
259 savedGeneratorMain.add_module(geobuilder)
260 savedGeneratorMain.add_module(bkgmixer)
261 savedGeneratorMain.add_module(cdcdigitizer)
262 savedGeneratorMain.add_module(cdctrg)