Belle II Software  release-05-01-25
NoKickCuts_evaluation.py
1 
20 
21 from basf2 import *
22 from simulation import add_simulation
23 from reconstruction import add_reconstruction, add_mdst_output
24 from ROOT import Belle2
25 
26 import argparse
27 parser = argparse.ArgumentParser(description="NoKickCuts evaluation module")
28 parser.add_argument(
29  '--useValidation',
30  dest='use_validation',
31  action='store_const',
32  const=True,
33  default=False,
34  help='print validation plots')
35 parser.add_argument(
36  '--useFitMethod',
37  dest='use_fitMethod',
38  action='store_const',
39  const=True,
40  default=False,
41  help='definition of the cuts with the double-gaussian fit')
42 
43 args = parser.parse_args()
44 use_Validation = args.use_validation
45 use_fit = args.use_fitMethod
46 
47 
48 main_path = create_path()
49 
50 mctrackfinder = register_module('TrackFinderMCTruthRecoTracks')
51 mctrackfinder.param('WhichParticles', ['SVD'])
52 mctrackfinder.param('EnergyCut', 0)
53 
54 gearbox = register_module('Gearbox')
55 
56 geometry = register_module('Geometry')
57 
58 rootinput = register_module('RootInput')
59 # param("inputFileNames", "/home/belle2/vberta/storage/release1_validation/training_sample/simulated*.root")
60 
61 progressbar = register_module('ProgressBar')
62 
63 mctrackmatcher = register_module('MCRecoTracksMatcher')
64 mctrackmatcher.param('UseCDCHits', False)
65 mctrackmatcher.param('UseSVDHits', True)
66 mctrackmatcher.param('UsePXDHits', True)
67 mctrackmatcher.param('mcRecoTracksStoreArrayName', 'MCRecoTracks')
68 
69 NoKickCuts = register_module('NoKickCutsEval')
70 NoKickCuts.param('useValidation', use_Validation)
71 NoKickCuts.param('useFitMethod', use_fit)
72 
73 
74 main_path.add_module(rootinput)
75 main_path.add_module(gearbox)
76 main_path.add_module(geometry)
77 main_path.add_module(mctrackfinder)
78 main_path.add_module(NoKickCuts)
79 main_path.add_module(progressbar)
80 
81 process(main_path)
82 print(statistics)
tracking.validation.run.TrackingValidationRun.create_path
def create_path(self)
Definition: run.py:114