25 from background
import add_output
39 sad =
'phase2.1.4_collimators_1'
48 sad =
'phase2.1.4_collimators_1'
57 sad =
'phase2.1.4_collimators_1'
79 print(
'basf2', argvs[0],
80 '(Touschek,Coulomb,Brems) (HER,LER) equivTime_us num [(study,usual,ECL,PXD) phase outdir]')
84 inputSAD =
"/group/belle2/BGcampaigns/SAD/forG4/"
85 bgType = bgType +
'_' + accRing
86 sadFile = inputSAD + bgType +
'_' + sad +
'.root'
87 realTime = float(equivTime) * 1000
88 fname = bgType +
'_' + sampleType +
'-phase' + str(phase) +
'-' + num
89 outputFile = outdir +
'/' + fname +
'.root'
93 if not os.path.exists(sadFile):
94 B2ERROR(
'SAD file ' + sadFile +
' not found')
99 if not os.path.exists(outdir):
103 B2RESULT(
'Output file: ' + outputFile)
104 B2RESULT(
'Corresponds to ' + equivTime +
' us of running phase ' + str(phase))
107 set_log_level(LogLevel.RESULT)
114 eventinfosetter = register_module(
'EventInfoSetter')
115 eventinfosetter.param(
'evtNumList', [10000000])
116 main.add_module(eventinfosetter)
119 gearbox = register_module(
'Gearbox')
121 gearbox.param(
'fileName',
'geometry/Beast2_phase2.xml')
123 gearbox.param(
'fileName',
'geometry/Belle2_earlyPhase3.xml')
124 if sampleType ==
'study':
125 gearbox.param(
'override', [
126 (
"/DetectorComponent[@name='PXD']//ActiveChips",
'true',
''),
127 (
"/DetectorComponent[@name='PXD']//SeeNeutrons",
'true',
''),
128 (
"/DetectorComponent[@name='SVD']//ActiveChips",
'true',
''),
129 (
"/DetectorComponent[@name='SVD']//SeeNeutrons",
'true',
''),
130 (
"/DetectorComponent[@name='TOP']//BeamBackgroundStudy",
'1',
''),
131 (
"/DetectorComponent[@name='ARICH']//BeamBackgroundStudy",
'1',
''),
132 (
"/DetectorComponent[@name='ECL']//BeamBackgroundStudy",
'1',
''),
133 (
"/DetectorComponent[@name='KLM']//BeamBackgroundStudy",
'1',
''),
135 main.add_module(gearbox)
138 generator = register_module(
'BeamBkgGenerator')
139 generator.param(
'fileName', sadFile)
140 generator.param(
'ringName', accRing)
141 generator.param(
'realTime', realTime)
142 main.add_module(generator)
145 geometry = register_module(
'Geometry')
146 geometry.param(
'useDB',
False)
148 geometry.param(
'additionalComponents', [
'BEAMABORT',
'MICROTPC',
'CLAWS',
'HE3TUBE'])
149 main.add_module(geometry)
152 fullsim = register_module(
'FullSim')
153 if sampleType ==
'study':
154 fullsim.param(
'PhysicsList',
'FTFP_BERT_HP')
155 fullsim.param(
'UICommandsAtIdle', [
'/process/inactivate nKiller'])
156 fullsim.param(
'StoreAllSecondaries',
True)
157 fullsim.param(
'SecondariesEnergyCut', 0.000001)
158 main.add_module(fullsim)
161 progress = register_module(
'Progress')
162 main.add_module(progress)
169 excludeBranch = [
'MCParticles']
170 add_output(main, bgType, realTime, sampleType, phase, fileName=outputFile, excludeBranches=excludeBranch)