25 from background
import add_output
66 print(
'basf2', argvs[0],
67 '(Touschek,Coulomb,Brems) (HER,LER) equivTime_us num [(study,usual,ECL,PXD) phase outdir]')
73 subdir =
'phase3-15th/'
75 subdir =
'phase3-early/'
79 print(
'phase ', phase,
'not supported')
82 bgType = bgType +
'_' + accRing
83 sadFile =
'input/' + subdir + bgType +
'.root'
84 realTime = float(equivTime) * 1000
85 fname = bgType +
'_' + sampleType +
'-phase' + str(phase) +
'-' + num
86 outputFile = outdir +
'/' + fname +
'.root'
90 if not os.path.exists(sadFile):
91 B2ERROR(
'SAD file ' + sadFile +
' not found')
96 if not os.path.exists(outdir):
100 B2RESULT(
'Output file: ' + outputFile)
101 B2RESULT(
'Corresponds to ' + equivTime +
' us of running phase ' + str(phase))
104 set_log_level(LogLevel.RESULT)
111 eventinfosetter = register_module(
'EventInfoSetter')
112 eventinfosetter.param(
'evtNumList', [10000000])
113 main.add_module(eventinfosetter)
116 gearbox = register_module(
'Gearbox')
118 gearbox.param(
'fileName',
'geometry/Beast2_phase2.xml')
120 gearbox.param(
'fileName',
'geometry/Belle2_earlyPhase3.xml')
121 if sampleType ==
'study':
122 gearbox.param(
'override', [
123 (
"/DetectorComponent[@name='PXD']//ActiveChips",
'true',
''),
124 (
"/DetectorComponent[@name='PXD']//SeeNeutrons",
'true',
''),
125 (
"/DetectorComponent[@name='SVD']//ActiveChips",
'true',
''),
126 (
"/DetectorComponent[@name='SVD']//SeeNeutrons",
'true',
''),
127 (
"/DetectorComponent[@name='TOP']//BeamBackgroundStudy",
'1',
''),
128 (
"/DetectorComponent[@name='ARICH']//BeamBackgroundStudy",
'1',
''),
129 (
"/DetectorComponent[@name='ECL']//BeamBackgroundStudy",
'1',
''),
130 (
"/DetectorComponent[@name='KLM']//BeamBackgroundStudy",
'1',
''),
132 main.add_module(gearbox)
135 generator = register_module(
'BeamBkgGenerator')
136 generator.param(
'fileName', sadFile)
137 generator.param(
'ringName', accRing)
138 generator.param(
'realTime', realTime)
139 main.add_module(generator)
142 geometry = register_module(
'Geometry')
143 geometry.param(
'useDB',
False)
145 if phase == 31
and sampleType ==
'study':
146 geometry.param(
'additionalComponents', [
"BEAMABORT",
"MICROTPC",
"CLAWS",
"HE3TUBE"])
147 main.add_module(geometry)
150 fullsim = register_module(
'FullSim')
151 if sampleType ==
'study':
152 fullsim.param(
'PhysicsList',
'FTFP_BERT_HP')
153 fullsim.param(
'UICommandsAtIdle', [
'/process/inactivate nKiller'])
154 fullsim.param(
'StoreAllSecondaries',
True)
155 fullsim.param(
'SecondariesEnergyCut', 0.000001)
156 main.add_module(fullsim)
159 progress = register_module(
'Progress')
160 main.add_module(progress)
165 add_output(main, bgType, realTime, sampleType, phase, fileName=outputFile)