Skip to content
Snippets Groups Projects

Draft: Tuple-making for Z xsec + jet using collision23 spruced data

Closed Davide Zuliani requested to merge dzuliani/my-analysis into master
4 unresolved threads
Files
2
ew_jets/dv.py 0 → 100644
+ 79
0
import Functors as F
from FunTuple import FunctorCollection as FC
from FunTuple import FunTuple_Particles as Funtuple
from PyConf.reading import get_particles, get_odin
from DaVinci.algorithms import create_lines_filter
from DaVinci import Options, make_config
from FunTuple.functorcollections import Kinematics, HltTisTos
def main(options: Options):
line_name = "SpruceQEE_ZJet"
line_data = get_particles(f"/Event/Spruce/{line_name}/Particles")
fields = {'H_10': 'H_10 -> Z0 CELLjet',
'Z0': 'H_10 -> ^Z0 CELLjet',
'jet': 'H_10 -> Z0 ^CELLjet',}
Hlt1_decisions = [
'Hlt1SingleHighPtMuonDecision','Hlt1DiMuonHighMassDecision',
]
Kin = Kinematics()
composite_variables = FC(
{
"ID": F.PARTICLE_ID,
"KEY": F.OBJECT_KEY,
"END_VCHI2": F.CHI2 @ F.ENDVERTEX,
"END_VCHI2DOF": F.CHI2DOF @ F.ENDVERTEX,
"TRCHI2DOF": F.CHI2DOF,
"CHI2": F.CHI2,
"NDF": F.NDOF @ F.ENDVERTEX,
"DOCA": F.DOCA(1, 2),
}
)
daughter_variables = FC(
{
"ID": F.PARTICLE_ID,
"KEY": F.OBJECT_KEY,
'ISMUON': F.ISMUON,
"ProbNNmu": F.PROBNN_MU,
"PIDmu": F.PID_MU,
}
)
Tistos_var = HltTisTos(
selection_type="Hlt1", trigger_lines=Hlt1_decisions, data=line_data)
variables = {
"H_10": Kin,
"Z0": Kin,
"jet": Kin,
}
# get odin which hold the event information
odin = get_odin()
evt_variables = FC(
{"RUNNUMBER": F.RUNNUMBER(odin), "EVENTNUMBER": F.EVENTNUMBER(odin)}
)
my_filter = create_lines_filter(name="PreFilter_for_ZMuMuJet", lines=[line_name])
# define FunTuple instance
my_tuple = Funtuple(
name="Tuple",
tuple_name="DecayTree",
fields=fields,
variables=variables,
event_variables=evt_variables,
inputs=line_data,
)
user_algorithms = {
"Alg1": [my_filter, my_tuple],
}
return make_config(options, user_algorithms)
\ No newline at end of file
Loading