Belle II Software  release-05-01-25
collectMassConstrainedJPsiKshort.py
1 #!/usr/bin/env python
2 # -*- coding: utf-8 -*-
3 
4 import basf2
5 
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
11 
12 
13 main = basf2.create_path()
14 
15 # basf2.reset_database()
16 # basf2.use_local_database('centraldb/dbcache.txt')
17 # basf2.use_local_database('localdb/database.txt')
18 
19 # main.add_module("RootInput")
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.])
25 
26 gen.add_evtgen_generator(main, finalstate='signal', signaldecfile='/home/tadeas/belle2/clean_master/decfiles/dec/1111440100.dec')
27 
28 sim.add_simulation(main)
29 reco.add_reconstruction(main, pruneTracks=False, add_muid_hits=True)
30 main.add_module('DAFRecoFitter')
31 # Pre-fit with beam+vertex constraint decays for alignment
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)
36 
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)
39 
40 ana.printVariableValues('J/psi:mumu', ['InvM'], path=main)
41 ana.printVariableValues('K_S0:pipi', ['InvM'], path=main)
42 
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'])
53 main.add_module(
54  'MillepedeCollector',
55  components=[],
56  twoBodyDecays=['K_S0:pipi', 'J/psi:mumu'], tracks=[], primaryVertices=[],
57  calibrateVertex=True,
58  useGblTree=False)
59 
60 main.add_module('RootOutput')
61 main.add_module("Progress")
62 
63 basf2.print_path(main)
64 basf2.process(main)
65 print(basf2.statistics)
basf2.process
def process(path, max_event=0)
Definition: __init__.py:25