hpstr
The Heavy Photon Search Toolkit for Reconstruction (hpstr) provides an interface to physics data from the HPS experiment saved in the LCIO format and converts it into an ROOT based format.
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Pages
anaVtxTuple_cfg.py
Go to the documentation of this file.
1import HpstrConf
2import sys
3import os
4import baseConfig as base
5
6base.parser.add_argument("-f", "--makeFlatTuple", type=int, dest="makeFlatTuple", help="Make True to make vertex ana flat tuple", metavar="makeFlatTuple", default=0)
7
8options = base.parser.parse_args()
9
10
11# Use the input file to set the output file name
12infile = options.inFilename
13outfile = options.outFilename
14
15print('Input file: %s' % infile)
16print('Output file: %s' % outfile)
17
19
20p.run_mode = 1
21p.skip_events = options.skip_events
22p.max_events = options.nevents
23
24#p.max_events = 1000
25
26# Library containing processors
27p.add_library("libprocessors")
28
29
32
33vtxana = HpstrConf.Processor('vtxana', 'VertexAnaProcessor')
34
35
39vtxana.parameters["debug"] = 0
40vtxana.parameters["anaName"] = "vtxana"
41vtxana.parameters["tsColl"] = "TSBank"
42#vtxana.parameters["trkColl"] = "GBLTracks"
43#vtxana.parameters["hitColl"] = "RotatedHelicalOnTrackHits"
44#vtxana.parameters["vtxColl"] = "UnconstrainedV0Vertices"
45vtxana.parameters["trkColl"] = "KalmanFullTracks"
46vtxana.parameters["hitColl"] = "SiClustersOnTrack"
47vtxana.parameters["vtxColl"] = "UnconstrainedV0Vertices_KF"
48vtxana.parameters["mcColl"] = "MCParticle"
49vtxana.parameters["analysis"] = "vertex"
50vtxana.parameters["vtxSelectionjson"] = os.environ['HPSTR_BASE']+'/analysis/selections/vertexSelection_2019.json'
51vtxana.parameters["mcHistoCfg"] = os.environ['HPSTR_BASE']+'/analysis/plotconfigs/mc/basicMC.json'
52vtxana.parameters["histoCfg"] = os.environ['HPSTR_BASE']+"/analysis/plotconfigs/tracking/vtxAnalysis_2019.json"
53vtxana.parameters["beamE"] = base.beamE[str(options.year)]
54vtxana.parameters["isData"] = options.isData
55vtxana.parameters["isRadPDG"] = 622
56vtxana.parameters["makeFlatTuple"] = options.makeFlatTuple
57
58CalTimeOffset = -999
59
60if (options.isData == 1):
61 CalTimeOffset = 56.
62 print("Running on data file: Setting CalTimeOffset %d" % CalTimeOffset)
63
64elif (options.isData == 0):
65 CalTimeOffset = 43.
66 print("Running on MC file: Setting CalTimeOffset %d" % CalTimeOffset)
67else:
68 print("Specify which type of ntuple you are running on: -t 1 [for Data] / -t 0 [for MC]")
69
70
71vtxana.parameters["CalTimeOffset"] = CalTimeOffset
72
73#Region definitions
74
75RegionPath = os.environ['HPSTR_BASE']+"/analysis/selections/"
76if (options.year == 2019):
77 vtxana.parameters["regionDefinitions"] = [RegionPath+'Tight_2019.json', RegionPath+'Tight_pTop_2019.json', RegionPath+'Tight_pBot_2019.json']
78if (options.year == 2021):
79 vtxana.parameters["regionDefinitions"] = [RegionPath+'Tight_2021.json', RegionPath+'Tight_pTop_2021.json', RegionPath+'Tight_pBot_2021.json']
80
81# Sequence which the processors will run.
82p.sequence = [vtxana]
83
84p.input_files = infile
85p.output_files = [outfile]
86
87p.printProcess()
Process python class.
Definition HpstrConf.py:20
Processor python class.
Definition HpstrConf.py:4