Belle II Software  release-05-01-25
collectKinematicConstrainedMuMu.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 
11 main = basf2.create_path()
12 
13 # basf2.reset_database()
14 # basf2.use_local_database('centraldb/dbcache.txt')
15 # basf2.use_local_database('localdb/database.txt')
16 
17 
18 main.add_module("RootInput")
19 # main.add_module("EventInfoSetter")
20 main.add_module("Gearbox")
21 main.add_module("Geometry")
22 main.add_module("HistoManager", histoFileName="CollectorOutput.root")
23 beam.add_beamparameters(main, 'Y4S', 10.5796, vertex=[0., 0., 0.])
24 
25 # main.add_module('PairGen', pdgCode=13)
26 sim.add_simulation(main)
27 reco.add_reconstruction(main, pruneTracks=False)
28 
29 # Pre-fit with beam+vertex constraint decays for alignment
30 ana.fillParticleList('mu+:qed', 'muonID > 0.1 and useCMSFrame(p) > 2.', writeOut=True, path=main)
31 ana.reconstructDecay('Z0:mumu -> mu-:qed mu+:qed', 'InvM > 10.5296 and InvM < 10.6296', writeOut=True, path=main)
32 
33 ana.vertexRaveDaughtersUpdate('Z0:mumu', 0.0, path=main, silence_warning=True)
34 
35 ana.matchMCTruth('mu+:qed', main)
36 ana.matchMCTruth('Z0:mumu', main)
37 
38 ana.printVariableValues('Z0:mumu', ['E', 'deltaE', 'M', 'InvM', 'mcPDG', 'p', 'Ecms'], path=main)
39 ana.printVariableValues('mu+:qed', ['E', 'deltaE', 'M', 'mcPDG', 'p', 'muonID'], path=main)
40 
41 main.add_module('VariablesToEventBasedTree',
42  particleList='Z0:mumu',
43  variables=['InvM', 'Ecms',
44  'daughter(0, p)', 'daughter(1, p)'],
45  event_variables=['nTracks', 'expNum', 'runNum', 'evtNum'])
46 main.add_module(
47  'MillepedeCollector',
48  components=['beamparameters'],
49  primaryTwoBodyDecays=['Z0:mumu'], tracks=[],
50  calibrateVertex=True,
51  useGblTree=False)
52 
53 main.add_module('RootOutput')
54 main.add_module("Progress")
55 
56 basf2.print_path(main)
57 basf2.process(main)
58 print(basf2.statistics)
basf2.process
def process(path, max_event=0)
Definition: __init__.py:25