JetCalibToolsCfg Namespace Reference

Functions
	defineJetCalibTool (jetdef, modspec)
	getJetCalibToolPrereqs (jetdef, modspec)

Variables
	jetcaliblog = Logging.logging.getLogger('JetCalibToolsConfig')
list	all = ['getJetCalibTool']
dict	calibdic

Function Documentation

defineJetCalibTool()

JetCalibToolsCfg.defineJetCalibTool	(		jetdef,
			modspec )

Definition at line 17 of file JetCalibToolsCfg.py.

def defineJetCalibTool(jetdef, modspec):
    from JetCalibTools.JetCalibStepsConfig import calibToolFromConfigFile
 
    jetcollection = jetdef.basename
    cfg = calibdic[f"{jetdef.basename}:{modspec}"]
    path_configFile = PathResolver.FindCalibFile(cfg) 
    toolname = "jetcalib_new_{0}_{1}".format(jetcollection,modspec)
    jct = calibToolFromConfigFile(jetdef._cflags, path_configFile, toolname)
 
    return jct
 
# This method extends the basic config getter to specify the requisite jet
# moments or other inputs

getJetCalibToolPrereqs()

JetCalibToolsCfg.getJetCalibToolPrereqs	(		jetdef,
			modspec )

Definition at line 30 of file JetCalibToolsCfg.py.

def getJetCalibToolPrereqs(jetdef, modspec):
    from JetCalibTools.JetCalibStepsConfig import load_yaml_cfg
 
    cfg = calibdic[f"{jetdef.basename}:{modspec}"]
    configDic = load_yaml_cfg(cfg)
 
    prereqs = ["mod:ConstitFourMom"]
    pvname = "PrimaryVertices" # this can be set dinamically in future
 
    for step, step_config in configDic.items():
        # JetArea
        if step_config.get("DoJetArea", False):
            if modspec.startswith("Trig"):
                prereqs.append("input:HLT_EventDensity")
            elif pvname == "PrimaryVertices_initial":
                prereqs.append("input:EventDensityCustomVtxGNN")
            elif pvname != "PrimaryVertices":
                prereqs.append("input:EventDensityCustomVtx")
            else:
                prereqs.append(inputsFromContext("EventDensity")(jetdef))
 
        # read prereqs from context or default config block
        prereq_block = step_config.get("prereqs", {})
        step_prereqs = prereq_block.get(modspec, prereq_block.get("default", []))
        prereqs.extend(step_prereqs)
 
    # remove duplication and keep order
    seen = set()
    prereqs_unique = []
    for p in prereqs:
        if p not in seen:
            prereqs_unique.append(p)
            seen.add(p)
 
    return prereqs_unique
 

Variable Documentation

all

list JetCalibToolsCfg.all = ['getJetCalibTool']

Definition at line 9 of file JetCalibToolsCfg.py.

calibdic

dict JetCalibToolsCfg.calibdic

Initial value:

=  {
    "AntiKt4EMPFlow:T0" : "JetCalibTools/calibConfigExample.yaml",
    "AntiKt4EMPFlow:TrigRun2" : "JetCalibTools/calibConfigExample.yaml", # this is for testing
}

Definition at line 11 of file JetCalibToolsCfg.py.

jetcaliblog

JetCalibToolsCfg.jetcaliblog = Logging.logging.getLogger('JetCalibToolsConfig')

Definition at line 7 of file JetCalibToolsCfg.py.