def addJetPruner(self, output, rcut, zcut, input, modifiersin ="groomed",
isTrigger =False, useTriggerStore =False, doArea =True):
from JetRec.JetRecConf import JetPruner
from JetRec.JetRecConf import JetRecTool
groomer = JetPruner(output + "Groomer")
groomer.RCut = rcut
groomer.ZCut = zcut
if doArea:
groomer.JetBuilder = self.jetBuilderWithArea
else:
groomer.JetBuilder = self.jetBuilderWithoutArea
self += groomer
jetrec = JetRecTool(output)
jetrec.JetGroomer = groomer
jetrec.InputContainer = input
jetrec.OutputContainer = output
jetrec.JetModifiers = self.getModifiers(modifiersin)
jetrec.Trigger = isTrigger or useTriggerStore
jetrec.Timer = jetFlags.timeJetRecTool()
self += jetrec
if isTrigger:
self.trigjetrecs += [jetrec]
else:
self.jetrecs += [jetrec]
self.jetcons += [output]
return jetrec
def addJetReclusterer(self, output, alg, radius, input, modifiersin =None,
consumers =None, ivtx =None,
ghostArea =0.0, ptmin =0.0, ptminFilter =0.0, rndseed =1,
isTrigger =False, useTriggerStore =False,
variableRMinRadius =-1.0, variableRMassScale =-1.0,
calibOpt ="", jetPseudojetCopier =""):
self.msg(2, "Adding reclusterer")
from JetRec.JetRecConf import JetRecTool
from JetRec.JetRecConf import JetReclusterer
# Retrieve/build the jet finder.
lofinder,hifinder = self.addJetFinderTool(output+"Finder", alg, radius, ivtx, ghostArea, ptmin, rndseed,
variableRMinRadius, variableRMassScale)
reclname = output + "Reclusterer"
groomer = JetReclusterer(
reclname,
JetConstituentsRetriever = self.tools["jconretriever"],
JetFinder = hifinder
)
self += groomer
jetrec = JetRecTool(output)
jetrec.InputContainer = input
jetrec.OutputContainer = output
jetrec.JetGroomer = groomer
jetrec.JetModifiers = self.getModifiers(modifiersin)
if consumers != None:
jetrec.JetConsumers = consumers
jetrec.Trigger = isTrigger or useTriggerStore
jetrec.Timer = jetFlags.timeJetRecTool()
self += jetrec
if isTrigger:
self.trigjetrecs += [jetrec]
else:
self.jetrecs += [jetrec]
self.jetcons += [output]
return jetrec
def addJetSplitter(self, output, mumax, ymin, input, modifiersin ="groomed",
isTrigger =False, useTriggerStore =False, doArea =True):
from JetRec.JetRecConf import JetSplitter
from JetRec.JetRecConf import JetRecTool
groomer = JetSplitter(output + "Groomer")
groomer.MuMax = mumax
groomer.YMin = ymin
groomer.BDRS = False
groomer.NSubjetMax = 3
if doArea:
groomer.JetBuilder = self.jetBuilderWithArea
else:
groomer.JetBuilder = self.jetBuilderWithoutArea
self += groomer
jetrec = JetRecTool(output)
jetrec.JetGroomer = groomer
jetrec.InputContainer = input
jetrec.OutputContainer = output
jetrec.JetModifiers = self.getModifiers(modifiersin)
jetrec.Trigger = isTrigger or useTriggerStore
jetrec.Timer = jetFlags.timeJetRecTool()
self += jetrec
if isTrigger:
self.trigjetrecs += [jetrec]
else:
self.jetrecs += [jetrec]
self.jetcons += [output]
return jetrec
def addJetTrimmer(self, output, rclus, ptfrac, input, modifiersin ="groomed",
pseudojetRetriever ="jpjretriever",
isTrigger =False, useTriggerStore =False, doArea =True):
from JetRec.JetRecConf import JetTrimmer
from JetRec.JetRecConf import JetRecTool
groomer = JetTrimmer(output + "Groomer")
groomer.RClus = rclus
groomer.PtFrac = ptfrac
if doArea:
groomer.JetBuilder = self.jetBuilderWithArea
else:
groomer.JetBuilder = self.jetBuilderWithoutArea
self += groomer
jetrec = JetRecTool(output)
jetrec.JetGroomer = groomer
jetrec.InputContainer = input
jetrec.OutputContainer = output
jetrec.JetModifiers = self.getModifiers(modifiersin)
jetrec.Trigger = isTrigger or useTriggerStore
jetrec.Timer = jetFlags.timeJetRecTool()
if pseudojetRetriever in self.tools:
jetrec.JetPseudojetRetriever = self.tools[pseudojetRetriever]
else:
print "Requested jet pseudojet retriever is not a registered tool: " \
+ pseudojetRetriever
raise KeyError
self += jetrec
if isTrigger:
self.trigjetrecs += [jetrec]
else:
self.jetrecs += [jetrec]
self.jetcons += [output]
return jetrec
def addJetSplitter(self,
output,
mumax,
ymin,
input,
modifiersin="groomed",
isTrigger=False,
useTriggerStore=False,
doArea=True):
from JetRec.JetRecConf import JetSplitter
from JetRec.JetRecConf import JetRecTool
groomer = JetSplitter(output + "Groomer")
groomer.MuMax = mumax
groomer.YMin = ymin
groomer.BDRS = False
groomer.NSubjetMax = 3
if doArea:
groomer.JetBuilder = self.jetBuilderWithArea
else:
groomer.JetBuilder = self.jetBuilderWithoutArea
self += groomer
jetrec = JetRecTool(output)
jetrec.JetGroomer = groomer
jetrec.InputContainer = input
jetrec.OutputContainer = output
jetrec.JetModifiers = self.getModifiers(modifiersin)
jetrec.Trigger = isTrigger or useTriggerStore
jetrec.Timer = jetFlags.timeJetRecTool()
self += jetrec
if isTrigger:
self.trigjetrecs += [jetrec]
else:
self.jetrecs += [jetrec]
self.jetcons += [output]
return jetrec
def addJetPruner(self,
output,
rcut,
zcut,
input,
modifiersin="groomed",
isTrigger=False,
useTriggerStore=False,
doArea=True):
from JetRec.JetRecConf import JetPruner
from JetRec.JetRecConf import JetRecTool
groomer = JetPruner(output + "Groomer")
groomer.RCut = rcut
groomer.ZCut = zcut
if doArea:
groomer.JetBuilder = self.jetBuilderWithArea
else:
groomer.JetBuilder = self.jetBuilderWithoutArea
self += groomer
jetrec = JetRecTool(output)
jetrec.JetGroomer = groomer
jetrec.InputContainer = input
jetrec.OutputContainer = output
jetrec.JetModifiers = self.getModifiers(modifiersin)
jetrec.Trigger = isTrigger or useTriggerStore
jetrec.Timer = jetFlags.timeJetRecTool()
self += jetrec
if isTrigger:
self.trigjetrecs += [jetrec]
else:
self.jetrecs += [jetrec]
self.jetcons += [output]
return jetrec
def addJetReclusterer(self, output, alg, radius, input, modifiersin =None,
consumers =None, ivtx =None,
ghostArea =0.0, ptmin =0.0, ptminFilter =0.0, rndseed =1,
isTrigger =False, useTriggerStore =False,
variableRMinRadius =-1.0, variableRMassScale =-1.0,
calibOpt ="", jetPseudojetCopier =""):
self.msg(2, "Adding reclusterer")
from JetRec.JetRecConf import JetRecTool
from JetRec.JetRecConf import JetReclusterer
# Retrieve/build the jet finder.
lofinder,hifinder = self.addJetFinderTool(output+"Finder", alg, radius, ivtx, ghostArea, ptmin, rndseed,
variableRMinRadius, variableRMassScale)
reclname = output + "Reclusterer"
groomer = JetReclusterer(
reclname,
JetConstituentsRetriever = self.tools["jconretriever"],
JetFinder = hifinder
)
self += groomer
jetrec = JetRecTool(output)
jetrec.InputContainer = input
jetrec.OutputContainer = output
jetrec.JetGroomer = groomer
jetrec.JetModifiers = self.getModifiers(modifiersin)
if consumers != None:
jetrec.JetConsumers = consumers
jetrec.Trigger = isTrigger or useTriggerStore
jetrec.Timer = jetFlags.timeJetRecTool()
self += jetrec
if isTrigger:
self.trigjetrecs += [jetrec]
else:
self.jetrecs += [jetrec]
self.jetcons += [output]
return jetrec
def addJetTrimmer(self, output, rclus, ptfrac, input, modifiersin ="groomed",
pseudojetRetriever ="jpjretriever",
isTrigger =False, useTriggerStore =False, doArea =True):
from JetRec.JetRecConf import JetTrimmer
from JetRec.JetRecConf import JetRecTool
groomer = JetTrimmer(output + "Groomer")
groomer.RClus = rclus
groomer.PtFrac = ptfrac
if doArea:
groomer.JetBuilder = self.jetBuilderWithArea
else:
groomer.JetBuilder = self.jetBuilderWithoutArea
self += groomer
jetrec = JetRecTool(output)
jetrec.JetGroomer = groomer
jetrec.InputContainer = input
jetrec.OutputContainer = output
jetrec.JetModifiers = self.getModifiers(modifiersin)
jetrec.Trigger = isTrigger or useTriggerStore
jetrec.Timer = jetFlags.timeJetRecTool()
if pseudojetRetriever in self.tools:
jetrec.JetPseudojetRetriever = self.tools[pseudojetRetriever]
else:
jetlog.info( "Requested jet pseudojet retriever is not a registered tool: " \
+ pseudojetRetriever )
raise KeyError
self += jetrec
if isTrigger:
self.trigjetrecs += [jetrec]
else:
self.jetrecs += [jetrec]
self.jetcons += [output]
return jetrec
def addJetPruner(self, output, rcut, zcut, input, modifiersin ="groomed",
isTrigger =False, useTriggerStore =False):
if self.m_jetBuilder == None:
self.msg(0, "Jet builder must be specified")
from JetRec.JetRecConf import JetPruner
from JetRec.JetRecConf import JetRecTool
groomer = JetPruner(output + "Groomer")
groomer.RCut = rcut
groomer.ZCut = zcut
groomer.JetBuilder = self.m_jetBuilder
self += groomer
jetrec = JetRecTool(output)
jetrec.JetGroomer = groomer
jetrec.InputContainer = input
jetrec.OutputContainer = output
jetrec.JetModifiers = self.getModifiers(modifiersin)
jetrec.Trigger = isTrigger or useTriggerStore
jetrec.Timer = self.timer
self += jetrec
if isTrigger:
self.trigjetrecs += [jetrec]
else:
self.jetrecs += [jetrec]
self.jetcons += [output]
return jetrec
def addJetSplitter(self, output, mumax, ymin, input, modifiersin ="groomed",
isTrigger =False, useTriggerStore =False):
if self.m_jetBuilder == None:
self.msg(0, "Jet builder must be specified")
from JetRec.JetRecConf import JetSplitter
from JetRec.JetRecConf import JetRecTool
groomer = JetSplitter(output + "Groomer")
groomer.MuMax = mumax
groomer.YMin = ymin
groomer.BDRS = False
groomer.NSubjetMax = 3
groomer.JetBuilder = self.m_jetBuilder
self += groomer
jetrec = JetRecTool(output)
jetrec.JetGroomer = groomer
jetrec.InputContainer = input
jetrec.OutputContainer = output
jetrec.JetModifiers = self.getModifiers(modifiersin)
jetrec.Trigger = isTrigger or useTriggerStore
jetrec.Timer = self.timer
self += jetrec
if isTrigger:
self.trigjetrecs += [jetrec]
else:
self.jetrecs += [jetrec]
self.jetcons += [output]
return jetrec
def addJetCopier(self,
output,
input,
modifiersin,
ptminFilter=0.0,
radius=0.0,
alg="",
inp="",
isTrigger=False,
useTriggerStore=False,
calibOpt="",
shallow=True,
constmods=[]):
from JetRec.JetRecConf import JetRecTool
jetrec = JetRecTool(output)
jetrec.InputContainer = input
jetrec.OutputContainer = output
ptminSave = self.ptminFilter
if ptminFilter > 0.0: self.ptminFilter = ptminFilter
class finder:
JetRadius = radius
JetAlgorithm = alg
class get:
Label = inp
getters = [get]
jetrec.JetModifiers = self.buildModifiers(modifiersin,
finder,
getters,
None,
output,
calibOpt,
constmods=constmods)
self.autoconfigureModifiers(jetrec.JetModifiers, output)
self.ptminFilter = ptminSave
jetrec.Trigger = isTrigger or useTriggerStore
jetrec.Timer = jetFlags.timeJetRecTool()
### jetrec.ShallowCopy = shallow
self += jetrec
self.jetrecs += [jetrec]
self.jetcons += [output]
return jetrec
def addJetRecursiveSoftDrop(self,
output,
beta,
zcut,
N,
r0,
input,
modifiersin="groomed",
isTrigger=False,
useTriggerStore=False,
doArea=True):
from JetRec.JetRecConf import JetRecursiveSoftDrop
from JetRec.JetRecConf import JetRecTool
groomer = JetRecursiveSoftDrop(output + "Groomer")
groomer.ZCut = zcut
groomer.Beta = beta
groomer.N = N
groomer.R0 = r0
if doArea:
groomer.JetBuilder = self.jetBuilderWithArea
else:
groomer.JetBuilder = self.jetBuilderWithoutArea
self += groomer
jetrec = JetRecTool(output)
jetrec.JetGroomer = groomer
jetrec.InputContainer = input
jetrec.OutputContainer = output
jetrec.JetModifiers = self.getModifiers(modifiersin)
self.autoconfigureModifiers(jetrec.JetModifiers, output)
jetrec.Trigger = isTrigger or useTriggerStore
jetrec.Timer = jetFlags.timeJetRecTool()
self += jetrec
if isTrigger:
self.trigjetrecs += [jetrec]
else:
self.jetrecs += [jetrec]
self.jetcons += [output]
return jetrec
def addJetCopier(self, output, input, modifiersin, ptminFilter =0.0, radius =0.0, alg ="", inp ="",
isTrigger=False, useTriggerStore=False, calibOpt =""):
from JetRec.JetRecConf import JetRecTool
jetrec = JetRecTool(output)
jetrec.InputContainer = input
jetrec.OutputContainer = output
ptminSave = self.ptminFilter
if ptminFilter > 0.0: self.ptminFilter = ptminFilter
class finder:
JetRadius = radius
JetAlgorithm = alg
class get:
Label = inp
getters = [get]
jetrec.JetModifiers = self.buildModifiers(modifiersin, finder, getters, None, output, calibOpt)
self.ptminFilter = ptminSave
jetrec.Trigger = isTrigger or useTriggerStore
jetrec.Timer = self.timer
self += jetrec
self.jetrecs += [jetrec]
self.jetcons += [output]
return jetrec
#jetrec.JetModifiers += [energycorrelator]
#jetrec.JetModifiers += [energycorrelatorratios]
# jetrec.JetModifiers += [pull]
# jetrec.JetModifiers += [charge]
# jetrec.JetModifiers += [subjetmaker]
# jetrec.JetModifiers += [subjetfinder]
ToolSvc += jetrec
jetPseudojetRetriever = JetPseudojetRetriever(
"JetPseudojetRetriever_" + name)
ToolSvc += jetPseudojetRetriever
# JetRec tool for finding.
jetrec_trimm = JetRecTool("JetRecTool_" + name + "Trimm")
jetrec_trimm.JetGroomer = groomer
jetrec_trimm.InputContainer = "AntiKt10" + name + "Jets"
jetrec_trimm.OutputContainer = "AntiKt10" + name + "TrimmedJets"
jetrec_trimm.JetPseudojetRetriever = jetPseudojetRetriever
#jetrec_trimm.JetModifiers += [nsubjettiness]
#jetrec_trimm.JetModifiers += [nsubjettinessratios]
# jetrec_trimm.JetModifiers += [ktsplittingscale]
# jetrec_trimm.JetModifiers += [dipolarity]
# jetrec_trimm.JetModifiers += [angularity]
# jetrec_trimm.JetModifiers += [ktdr]
# jetrec_trimm.JetModifiers += [ktmassdrop]
# jetrec_trimm.JetModifiers += [planarflow]
# jetrec_trimm.JetModifiers += [centerofmassshapes]
#jetrec_trimm.JetModifiers += [energycorrelator]
#jetrec_trimm.JetModifiers += [energycorrelatorratios]
# jetrec_trimm.JetModifiers += [pull]
# jetrec_trimm.JetModifiers += [charge]