6 import simulation
as sim
7 import reconstruction
as reco
8 import modularAnalysis
as ana
9 import beamparameters
as beam
10 import generators
as gen
13 main = basf2.create_path()
20 main.add_module(
"EventInfoSetter")
21 main.add_module(
"Gearbox")
22 main.add_module(
"Geometry")
23 main.add_module(
"HistoManager", histoFileName=
"CollectorOutput.root")
24 beam.add_beamparameters(main,
'Y4S',
None, vertex=[0., 0., 0.])
26 gen.add_evtgen_generator(main, finalstate=
'signal', signaldecfile=
'/home/tadeas/belle2/clean_master/decfiles/dec/1111440100.dec')
28 sim.add_simulation(main)
29 reco.add_reconstruction(main, pruneTracks=
False, add_muid_hits=
True)
30 main.add_module(
'DAFRecoFitter')
32 ana.fillParticleList(
'mu+:Jpsi',
'muonID > 0.1', writeOut=
True, path=main)
33 ana.fillParticleList(
'pi+:K_S0',
'pionID > 0.1', writeOut=
True, path=main)
34 ana.reconstructDecay(
'J/psi:mumu -> mu-:Jpsi mu+:Jpsi',
'InvM > 2.9969 and InvM < 3.1969', writeOut=
True, path=main)
35 ana.reconstructDecay(
'K_S0:pipi -> pi-:K_S0 pi+:K_S0',
'InvM > 0.44761 and InvM < 0.54761', writeOut=
True, path=main)
37 ana.vertexRaveDaughtersUpdate(
'J/psi:mumu', 0.0, path=main, silence_warning=
True)
38 ana.vertexRaveDaughtersUpdate(
'K_S0:pipi', 0.0, path=main, silence_warning=
True)
40 ana.printVariableValues(
'J/psi:mumu', [
'InvM'], path=main)
41 ana.printVariableValues(
'K_S0:pipi', [
'InvM'], path=main)
43 m1 = main.add_module(
'VariablesToEventBasedTree', fileName=
'Jpsi.root',
44 particleList=
'J/psi:mumu',
45 variables=[
'InvM',
'Ecms',
46 'daughter(0, p)',
'daughter(1, p)'],
47 event_variables=[
'nTracks',
'expNum',
'runNum',
'evtNum'])
48 main.add_module(
'VariablesToEventBasedTree', fileName=
'K_S0.root',
49 particleList=
'K_S0:pipi',
50 variables=[
'InvM',
'Ecms',
51 'daughter(0, p)',
'daughter(1, p)'],
52 event_variables=[
'nTracks',
'expNum',
'runNum',
'evtNum'])
56 twoBodyDecays=[
'K_S0:pipi',
'J/psi:mumu'], tracks=[], primaryVertices=[],
60 main.add_module(
'RootOutput')
61 main.add_module(
"Progress")
63 basf2.print_path(main)
65 print(basf2.statistics)