5 from simulation
import *
6 from optparse
import OptionParser
7 from tracking
import add_tracking_reconstruction
8 from L1trigger
import add_tsim
9 from ROOT
import Belle2
10 from reconstruction
import *
13 background_files = glob.glob(
'/group/belle2/BGFile/OfficialBKG/phase2/pseudorandom/BGOverlay.physics.0003.*.root')
18 parser = OptionParser()
19 parser.add_option(
'-n',
'--nevents', dest=
'nevents', default=200000,
20 help=
'Number of events to process')
21 parser.add_option(
'-f',
'--file', dest=
'filename',
22 default=
'Bhabha_Prod5_MC.root')
23 parser.add_option(
'-o',
'--output', dest=
'cdstname',
24 default=
'Bhabha_Prod5_MC_cdst.root')
25 (options, args) = parser.parse_args()
27 home = os.environ[
'BELLE2_LOCAL_DIR']
30 set_log_level(LogLevel.ERROR)
34 use_central_database(
"data_reprocessing_prod5", LogLevel.WARNING)
41 eventinfosetter = register_module(
'EventInfoSetter')
42 eventinfosetter.param({
'evtNumList': [int(options.nevents)],
'runList': [1]})
43 main.add_module(eventinfosetter)
46 histo = register_module(
'HistoManager')
47 histo.param(
'histoFileName',
'DQMhistograms.root')
48 main.add_module(histo)
51 gearbox = register_module(
'Gearbox', fileName=
"geometry/Beast2_phase2.xml")
52 main.add_module(gearbox)
55 geometry = register_module(
'Geometry')
63 main.add_module(geometry)
66 generator = register_module(
'BabayagaNLOInput')
67 generator.param(
'FinalState',
'ee')
68 generator.param(
'MaxAcollinearity', 180.0)
69 generator.param(
'ScatteringAngleRange', [15., 165.])
70 generator.param(
'FMax', 75000)
71 generator.param(
'MinEnergy', 0.01)
72 generator.param(
'Order',
'exp')
73 generator.param(
'DebugEnergySpread', 0.01)
74 generator.param(
'Epsilon', 0.00005)
75 generatorpreselection = register_module(
'GeneratorPreselection')
76 generatorpreselection.param(
'nChargedMin', 1)
77 generatorpreselection.param(
'nChargedMax', 999)
78 generatorpreselection.param(
'MinChargedP', 4.)
80 generatorpreselection.param(
'MinChargedTheta', 17.)
81 generatorpreselection.param(
'MaxChargedTheta', 36.)
82 generatorpreselection.param(
'applyInCMS',
False)
83 main.add_module(generator)
84 main.add_module(generatorpreselection)
85 generatorpreselection.if_value(
'!=11', empty)
92 add_simulation(main,
None, bkgfiles=background_files, bkgOverlay=
True)
95 arich_digitizer = register_module(
'ARICHDigitizer')
96 arich_digitizer.param(
'MagneticFieldDistorsion', 1)
97 main.add_module(arich_digitizer)
100 add_tsim(main, shortTracks=
True, Belle2Phase=
"Phase2")
104 add_reconstruction(main)
107 arichHits = register_module(
'ARICHFillHits')
108 arichHits.param(
'MagFieldCorrection', 1)
109 main.add_module(arichHits)
113 arichreco = register_module(
'ARICHReconstructor')
115 arichreco.param(
'storePhotons', 1)
116 arichreco.param(
'useAlignment', 0)
117 main.add_module(arichreco)
121 arichNtuple = register_module(
'ARICHNtuple')
122 arichNtuple.param(
'outputFile', options.filename)
123 main.add_module(arichNtuple)
127 arichdqm = register_module(
'ARICHDQM')
128 main.add_module(arichdqm)
130 add_cdst_output(main,
True,
'/ghi/fs01/belle2/bdata/users/sachit/' + options.cdstname)
144 progress = register_module(
'Progress')
145 main.add_module(progress)