Example #1
0
def getMatchingFractionTool(jetcollname, matchType, seq=AlgSequence()):
    """ ATTENTION this tools returns (tool, alg) where alg is a dependency for  Matching.
    It *has* to be scheduled before (caller's reponsability).
    It return None if it can not guess what truth alg to request.
    """
    from JetMomentTools.GhostAssociation import interpretJetName
    from JetMomentTools.JetMomentToolsConf import JetMatchingFractionTool
    recognizedJet = interpretJetName(jetcollname)
    if recognizedJet is None:
        return None

    finder, input, mainParam = recognizedJet
    algtype = finder+str(int(mainParam*10))        
    from JetRec.JetGetters import make_StandardJetGetter
    if matchType == "Truth":
        from JetRec.JetRec_defaults import standardJetAlgorithm
        alg = standardJetAlgorithm(algtype+"TruthJets")
        if alg is None: # possible if truth collection already exist
            # !!!! WARNNIG minPt cut is ambiguous !!!!
            alg=make_StandardJetGetter(finder,  mainParam,"Truth",  minPt=4*GeV, jetMoments=[], outputCollectionName=algtype+"TruthNewJets", seq=seq).jetAlgorithmHandle()

        # Then the mftool can be scheduled
        mftool = JetMatchingFractionTool("TruthMF",MFCollection = alg.JetCollectionName, AssociatedConstitKey="TruthAssoc",
                                         MomentTag="Truth",UseHighestShareFrac=True)
    elif matchType == "Track":
        alg=make_StandardJetGetter(finder,  mainParam,"TrackZ",   addJetQualityMoments=False,  seq=seq).jetAlgorithmHandle()        
        # Then the mftool can be scheduled
        mftool = JetMatchingFractionTool("TrackMF",MFCollection = alg.JetCollectionName, AssociatedConstitKey="TrackAssoc",
                                         MomentTag="Track",UseHighestShareFrac=True)
    else:
        return None

    return (mftool, alg)
Example #2
0
def myjets():
    from JetRec.JetRecFlags import jetFlags
    from JetRec.JetGetters import make_StandardJetGetter
    
    # Algorithms to test: AntiKt R = {0.4, 0.6} scheme = {EM, LC}
    #jetAlg4EM  = make_StandardJetGetter('AntiKt',0.4,'Topo',calibName='EM:OFFSET_ORIGIN_ETAJES',outputCollectionName='AntiKt4TopoJetsTest').jetAlgorithmHandle()
    jetAlg4EM  = make_StandardJetGetter('AntiKt',0.4,'Topo',doCalib=False,minPt=5000,outputCollectionName='AntiKt4TopoEMJetsTest').jetAlgorithmHandle()
    jetAlg4LC  = make_StandardJetGetter('AntiKt',0.4,'LCTopo',outputCollectionName='AntiKt4LCTopoJetsTest').jetAlgorithmHandle()
    
    #jetAlg6EM  = make_StandardJetGetter('AntiKt',0.6,'Topo',calibName='EM:OFFSET_ORIGIN_ETAJES',outputCollectionName='AntiKt6TopoJetsTest').jetAlgorithmHandle()
    jetAlg6EM  = make_StandardJetGetter('AntiKt',0.6,'Topo',doCalib=False,minPt=5000,outputCollectionName='AntiKt6TopoEMJetsTest').jetAlgorithmHandle()
    jetAlg6LC  = make_StandardJetGetter('AntiKt',0.6,'LCTopo',outputCollectionName='AntiKt6LCTopoJetsTest').jetAlgorithmHandle()
   
    # Prepare to dump relevant jet collections and associated moments
    from JetValidation.JetTestConfig import scheduleJetTester
    moments = {'moments':['nTrk', 'nTrk_allpv','JVF','WIDTH','trackWIDTH','trackWIDTH_allpv','ActiveArea','ActiveAreaPx','ActiveAreaPy','ActiveAreaPz','ActiveAreaE','VoronoiArea','VoronoiAreaPx','VoronoiAreaPy','VoronoiAreaPz','VoronoiAreaE']}
    
    # Schedule the jet testers for the existing collections
    scheduleJetTester("CaloCalTopoCluster",JetMoments=False, CollectionType="CaloClusterContainer", JetSigState=False, JetAuthor=False, JetNConstituents=False, JetConstituents=False)
    scheduleJetTester("AntiKt4TopoEMJets",JetMoments_opts = moments)
    scheduleJetTester("AntiKt4LCTopoJets",JetMoments_opts = moments)
    scheduleJetTester("AntiKt6TopoEMJets",JetMoments_opts = moments)
    scheduleJetTester("AntiKt6LCTopoJets",JetMoments_opts = moments)

    # Schedule the jet testers for the recalculated collections
    scheduleJetTester("AntiKt4TopoEMJetsTest",JetMoments_opts = moments)
    scheduleJetTester("AntiKt4LCTopoJetsTest",JetMoments_opts = moments)
    scheduleJetTester("AntiKt6TopoEMJetsTest",JetMoments_opts = moments)
    scheduleJetTester("AntiKt6LCTopoJetsTest",JetMoments_opts = moments)
Example #3
0
def getMatchingFractionTool(jetcollname, matchType, seq=AlgSequence()):
    """ ATTENTION this tools returns (tool, alg) where alg is a dependency for  Matching.
    It *has* to be scheduled before (caller's reponsability).
    It return None if it can not guess what truth alg to request.
    """
    from JetMomentTools.GhostAssociation import interpretJetName
    from JetMomentTools.JetMomentToolsConf import JetMatchingFractionTool
    recognizedJet = interpretJetName(jetcollname)
    if recognizedJet is None:
        return None

    finder, input, mainParam = recognizedJet
    algtype = finder + str(int(mainParam * 10))
    from JetRec.JetGetters import make_StandardJetGetter
    if matchType == "Truth":
        from JetRec.JetRec_defaults import standardJetAlgorithm
        alg = standardJetAlgorithm(algtype + "TruthJets")
        if alg is None:  # possible if truth collection already exist
            # !!!! WARNNIG minPt cut is ambiguous !!!!
            alg = make_StandardJetGetter(finder,
                                         mainParam,
                                         "Truth",
                                         minPt=4 * GeV,
                                         jetMoments=[],
                                         outputCollectionName=algtype +
                                         "TruthNewJets",
                                         seq=seq).jetAlgorithmHandle()

        # Then the mftool can be scheduled
        mftool = JetMatchingFractionTool("TruthMF",
                                         MFCollection=alg.JetCollectionName,
                                         AssociatedConstitKey="TruthAssoc",
                                         MomentTag="Truth",
                                         UseHighestShareFrac=True)
    elif matchType == "Track":
        alg = make_StandardJetGetter(finder,
                                     mainParam,
                                     "TrackZ",
                                     addJetQualityMoments=False,
                                     seq=seq).jetAlgorithmHandle()
        # Then the mftool can be scheduled
        mftool = JetMatchingFractionTool("TrackMF",
                                         MFCollection=alg.JetCollectionName,
                                         AssociatedConstitKey="TrackAssoc",
                                         MomentTag="Track",
                                         UseHighestShareFrac=True)
    else:
        return None

    return (mftool, alg)
def myjets():


  from AthenaCommon.AlgSequence import AlgSequence
  topSequence = AlgSequence()

  
  from EventShapeRec.EventShapeRecConf import EventEtDensityTester
  
  # Check if an alg already exists 
  if hasattr( topSequence, "EventEtDensityTester" ):
    alg  = topSequence.EventEtDensityTester
  elif hasattr( AlgSequence(), "EventEtDensityTester" ):
    alg  = seq.EventEtDensityTester
  else:
      alg = EventEtDensityTester("EventEtDensityTester", FillHistos=False, FillEventStore=True)
      topSequence += alg        
      
  alg.EventEtDensityTools += [ ConfLC3Tool, ConfLC4Tool, ConfEM3Tool, ConfEM4Tool ] 

  from EventShapeRec.EventEtDensityConfig import calculateSimpleEventDensity
  calculateSimpleEventDensity(Radius=0.4, SignalState="UNCALIBRATED")
  calculateSimpleEventDensity(Radius=0.4, SignalState="CALIBRATED")
  calculateSimpleEventDensity(Radius=0.6, SignalState="UNCALIBRATED")
  calculateSimpleEventDensity(Radius=0.6, SignalState="CALIBRATED")

  from JetRec.JetGetters  import make_StandardJetGetter
  from JetMomentTools.JetMomentsConfigHelpers import *
  
  specialJet('AntiKt',0.4,'LCTopo', doEMCut=False, minPt=4*GeV, doCalib=True, calibName='LC:ApplyAreaOffsetJES',
             addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt4LCTopoNewJets')
  specialJet('AntiKt',0.6,'LCTopo', doEMCut=False, minPt=4*GeV, doCalib=True, calibName='LC:ApplyAreaOffset',
             addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt6LCTopoNewJets')
  specialJet('AntiKt',0.4,'Topo',   doEMCut=False, minPt=4*GeV, doCalib=True, calibName='EM:ApplyAreaOffsetJES',
             addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt4TopoEMNewJets')
  specialJet('AntiKt',0.6,'Topo',   doEMCut=False, minPt=4*GeV, doCalib=True, calibName='EM:ApplyAreaOffset',
               addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt6TopoEMNewJets')

  
#   from JetRec.JetGetters  import make_StandardJetGetter

#   make_StandardJetGetter('AntiKt',0.4,'LCTopo', doEMCut=False, minPt=4*GeV, doCalib=True, calibName='LC:ApplyAreaOffsetJES',
#                          addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt4LCTopoNewJets')
#   make_StandardJetGetter('AntiKt',0.6,'LCTopo', doEMCut=False, minPt=4*GeV, doCalib=True, calibName='LC:ApplyAreaOffset',
#                          addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt6LCTopoNewJets')
#   make_StandardJetGetter('AntiKt',0.4,'Topo',   doEMCut=False, minPt=4*GeV, doCalib=True, calibName='EM:ApplyAreaOffsetJES',
#                          addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt4TopoEMNewJets')
#   make_StandardJetGetter('AntiKt',0.6,'Topo',   doEMCut=False, minPt=4*GeV, doCalib=True, calibName='EM:ApplyAreaOffset',
#                          addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt6TopoEMNewJets')


#   from JetMomentTools.JetMomentsConfigHelpers import *

  recommendedAreaAndJVFMoments("AntiKt4TopoEMNewJets")
  specialMoment("AntiKt4TopoEMNewJets")
  recommendedAreaAndJVFMoments("AntiKt6TopoEMNewJets")
  specialMoment("AntiKt6TopoEMNewJets")
  recommendedAreaAndJVFMoments("AntiKt4LCTopoNewJets")
  specialMoment("AntiKt4LCTopoNewJets")
  recommendedAreaAndJVFMoments("AntiKt6LCTopoNewJets")
  specialMoment("AntiKt6LCTopoNewJets")

#   specialMoment("AntiKt4TopoEMNewJets", ["origin"])
#   specialMoment("AntiKt6TopoEMNewJets", ["origin"])
#   specialMoment("AntiKt4LCTopoNewJets", ["origin"])
#   specialMoment("AntiKt6LCTopoNewJets", ["origin"])

  
  from JetMomentTools.JetCopyMomentsAlg import JetCopyMomentsAlg
  topSequence+= JetCopyMomentsAlg('AntiKt4LCTopoJetsCopyMomAlg',
                                  JetCollectionSrc  = 'AntiKt4LCTopoJets',
                                  JetCollectionDest = 'AntiKt4LCTopoNewJets',
                                  Moments=  [ 'BCH_CORR_CELL', 'BCH_CORR_JET', 'BCH_CORR_DOTX',
                                              'LArQuality','HECQuality',
                                              'isBadLooseMinus','isBadLoose','isBadMedium','isBadTight',
                                              'EtaOrigin', 'PhiOrigin', 'MOrigin',
                                              'EtaOriginEM', 'PhiOriginEM', 'MOriginEM',
                                              'N_BAD_CELLS', 'ENG_BAD_CELLS', 'n90', 'NumTowers', 'ootFracCells5', 'ootFracCells10'] )
  topSequence+= JetCopyMomentsAlg('AntiKt6LCTopoJetsCopyMomAlg',
                                  JetCollectionSrc  = 'AntiKt6LCTopoJets',
                                  JetCollectionDest = 'AntiKt6LCTopoNewJets',
                                  Moments= [ 'BCH_CORR_CELL', 'BCH_CORR_JET', 'BCH_CORR_DOTX',
                                             'LArQuality','HECQuality',
                                             'isBadLooseMinus','isBadLoose','isBadMedium','isBadTight',
                                              'EtaOrigin', 'PhiOrigin', 'MOrigin',
                                              'EtaOriginEM', 'PhiOriginEM', 'MOriginEM',
                                             'N_BAD_CELLS', 'ENG_BAD_CELLS', 'n90', 'NumTowers', 'ootFracCells5', 'ootFracCells10'] )
  topSequence+= JetCopyMomentsAlg('AntiKt4TopoEMJetsCopyMomAlg',
                                  JetCollectionSrc  = 'AntiKt4TopoEMJets',
                                  JetCollectionDest = 'AntiKt4TopoEMNewJets',
                                  Moments=  [ 'BCH_CORR_CELL', 'BCH_CORR_JET', 'BCH_CORR_DOTX',
                                              'LArQuality','HECQuality',
                                              'isBadLooseMinus','isBadLoose','isBadMedium','isBadTight',
                                              'EtaOrigin', 'PhiOrigin', 'MOrigin',
                                              'EtaOriginEM', 'PhiOriginEM', 'MOriginEM',
                                              'N_BAD_CELLS', 'ENG_BAD_CELLS', 'n90', 'NumTowers', 'ootFracCells5', 'ootFracCells10'] )
  topSequence+= JetCopyMomentsAlg('AntiKt6TopoEMJetsCopyMomAlg',
                                  JetCollectionSrc  = 'AntiKt6TopoEMJets',
                                  JetCollectionDest = 'AntiKt6TopoEMNewJets',
                                  Moments= [ 'BCH_CORR_CELL', 'BCH_CORR_JET', 'BCH_CORR_DOTX',
                                             'LArQuality','HECQuality',
                                             'isBadLooseMinus','isBadLoose','isBadMedium','isBadTight',
                                              'EtaOrigin', 'PhiOrigin', 'MOrigin',
                                              'EtaOriginEM', 'PhiOriginEM', 'MOriginEM',
                                             'N_BAD_CELLS', 'ENG_BAD_CELLS', 'n90', 'NumTowers', 'ootFracCells5', 'ootFracCells10'] )

  from JetD3PDAnalysis.JetMomentCalcConfig import JetMomentCalcConfig
  from JetRec.JetRecFlags import jetFlags
  jetFlags.jetPerformanceJob = True

#   JetMomentCalcConfig ("AntiKt4LCTopoNewJets", 'all', topSequence)
#   JetMomentCalcConfig ("AntiKt6LCTopoNewJets", 'all', topSequence)
#   JetMomentCalcConfig ("AntiKt4TopoEMNewJets", 'all', topSequence)
#   JetMomentCalcConfig ("AntiKt6TopoEMNewJets", 'all', topSequence)


  # truth WZ jets
  if ZeeDisMC :
      make_StandardJetGetter('AntiKt',0.6,'Truth')
      
      from TruthD3PDMaker.TruthJetFilterConfig import TruthJetFilterConfig
      TruthJetFilterConfig (topSequence, writePartons = False, writeHadrons = True, excludeWZdecays = True, photonCone=0.1)
      from ParticleBuilderOptions.AODFlags import AODFlags
      AODFlags.MissingEtTruth = True
      AODFlags.TruthParticleJet = True
      AODFlags.McEventKey="GEN_EVENT"
      antikt4truthAlgWZ = make_StandardJetGetter('AntiKt',0.4,'Truth',disable=False,
                                                 inputCollectionNames=['FilteredD3PDTruth'],
                                                 outputCollectionName='AntiKt4TruthJets_WZ',
                                                 useInteractingOnly=False,
                                                 includeMuons=True
                                                 ).jetAlgorithmHandle()
      antikt6truthAlgWZ = make_StandardJetGetter('AntiKt',0.6,'Truth',disable=False,
                                                 inputCollectionNames=['FilteredD3PDTruth'],
                                                 outputCollectionName='AntiKt6TruthJets_WZ',
                                                 useInteractingOnly=False,
                                                 includeMuons=True
                                                 ).jetAlgorithmHandle()
Example #5
0
   'AntiKt6TopoEMNew->AntiKt6TopoEM,AntiKt6H1Topo,AntiKt6H1Tower',
   'AntiKt4TowerNew->AntiKt4Tower,AntiKt4H1Tower',
   'AntiKt6TowerNew->AntiKt6Tower,AntiKt6H1Tower',
   'AntiKt4TowerEMNew->AntiKt4Tower,AntiKt4H1Tower',
   'AntiKt6TowerEMNew->AntiKt6Tower,AntiKt6H1Tower',
   'AntiKt4LCTopoNew->AntiKt4LCTopo',
   'AntiKt6LCTopoNew->AntiKt6LCTopo'
   ]

  #BTaggingFlags.JetFitterNN = False
  include("BTagging/BTagging_LoadTools.py")

  #
  if QcdD3PDMakerFlags.doTopoJet() and not QcdD3PDMakerFlags.doRecJetOnlyAdditional():
    #TopoJets
    make_StandardJetGetter('AntiKt',0.4,'Topo', doEMCut=False, minPt=7*GeV, doCalib=True, calibName='EM:GCW_ORIGIN_ETAJES', GCWScheme='Snowmass', addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt4TopoNewJets')
    make_StandardJetGetter('AntiKt',0.6,'Topo', doEMCut=False, minPt=7*GeV, doCalib=True, calibName='EM:GCW_ORIGIN_ETAJES', GCWScheme='Snowmass', addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt6TopoNewJets')
  if QcdD3PDMakerFlags.doTopoEMJet() and not QcdD3PDMakerFlags.doRecJetOnlyAdditional():
    #TopoJets (EM+JES)
    make_StandardJetGetter('AntiKt',0.4,'Topo', doEMCut=False, minPt=7*GeV, doCalib=True, calibName='EM:ORIGIN_ETAJES', addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt4TopoEMNewJets')
    make_StandardJetGetter('AntiKt',0.6,'Topo', doEMCut=False, minPt=7*GeV, doCalib=True, calibName='EM:ORIGIN_ETAJES', addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt6TopoEMNewJets')

  if not rec.readAOD():
    if QcdD3PDMakerFlags.doTowerJet() and not QcdD3PDMakerFlags.doRecJetOnlyAdditional():
      #TopoTowerJets (GCW+JES)
      make_StandardJetGetter('AntiKt',0.4,'TopoTower', doEMCut=False, minPt=7*GeV, doCalib=True, calibName='EM:GCW_ORIGIN_ETAJES', GCWScheme='Snowmass', addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt4TowerNewJets')
      make_StandardJetGetter('AntiKt',0.6,'TopoTower', doEMCut=False, minPt=7*GeV, doCalib=True, calibName='EM:GCW_ORIGIN_ETAJES', GCWScheme='Snowmass', addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt6TowerNewJets')
    if QcdD3PDMakerFlags.doTowerJet():   # not default
      #TopoTowerJets (EM+JES)
      make_StandardJetGetter('AntiKt',0.4,'TopoTower', doEMCut=False, minPt=7*GeV, doCalib=True, calibName='EM:ORIGIN_ETAJES', addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt4TowerEMNewJets')
      make_StandardJetGetter('AntiKt',0.6,'TopoTower', doEMCut=False, minPt=7*GeV, doCalib=True, calibName='EM:ORIGIN_ETAJES', addJetQualityMoments=True, writeMoments=True, outputCollectionName='AntiKt6TowerEMNewJets')
Example #6
0
#
# JetTrackZClusterTool jobOptions file
#
# updated May 2011, with recommendations from:
#  TopInputsD3PDMaker/share/ PrepareTrackJets.py
#  https://twiki.cern.ch/twiki/bin/view/AtlasProtected/TrackJetsInAthena
#  https://svnweb.cern.ch/trac/atlasgrp/browser/Institutes/SLAC/JetCaloValidation/trunk/share/Configs/Configure7TeVTrackJets.py
#==============================================================

from JetRec.JetGetters import make_StandardJetGetter
try:
    from JetRec.TrackSelectionForJets import getStandardTrackJetFinderTool
    TrackJetAlg1 = make_StandardJetGetter(
        'AntiKt',
        0.4,
        'TrackZ',
        outputCollectionName='AntiKt4TrackJets',
        doCalib=False,
        jetMoments=[],
        minPt=5 * GeV).jetAlgorithmHandle()
    #TrackJetAlg2=make_StandardJetGetter('AntiKt', 0.6, 'TrackZ', outputCollectionName='AntiKt6TrackJets', doCalib=False, jetMoments=[] , minPt=5*GeV).jetAlgorithmHandle()
    #TrackJetAlg3=make_StandardJetGetter('AntiKt', 0.5, 'TrackZ', outputCollectionName='AntiKt5TrackJets', doCalib=False, jetMoments=[] , minPt=5*GeV).jetAlgorithmHandle()
    #TrackJetAlg4=make_StandardJetGetter('AntiKt', 0.7, 'TrackZ', outputCollectionName='AntiKt7TrackJets', doCalib=False, jetMoments=[] , minPt=5*GeV).jetAlgorithmHandle()
    TrackJetAlg5 = make_StandardJetGetter(
        'AntiKt',
        0.2,
        'TrackZ',
        outputCollectionName='AntiKt2TrackJets',
        doCalib=False,
        jetMoments=[],
        minPt=5 * GeV).jetAlgorithmHandle()
    TrackJetAlg6 = make_StandardJetGetter(
Example #7
0
def myjets():
    from JetRec.JetRecFlags import jetFlags
    from JetRec.JetGetters import make_StandardJetGetter

    # Algorithms to test: AntiKt R = {0.4, 0.6} scheme = {EM, LC}
    #jetAlg4EM  = make_StandardJetGetter('AntiKt',0.4,'Topo',calibName='EM:OFFSET_ORIGIN_ETAJES',outputCollectionName='AntiKt4TopoJetsTest').jetAlgorithmHandle()
    #jetAlg4EM  = make_StandardJetGetter('AntiKt',0.4,'Topo',doCalib=False,minPt=5000,jetMoments=[],outputCollectionName='AntiKt4TopoJetsTest').jetAlgorithmHandle()
    #jetAlg4LC  = make_StandardJetGetter('AntiKt',0.4,'LCTopo',jetMoments=[],outputCollectionName='AntiKt4LCTopoJetsTest').jetAlgorithmHandle()
    jetAlg4EM = make_StandardJetGetter(
        'AntiKt',
        0.4,
        'Topo',
        doCalib=False,
        minPt=5000,
        outputCollectionName='AntiKt4TopoJetsTest').jetAlgorithmHandle()
    jetAlg4LC = make_StandardJetGetter(
        'AntiKt', 0.4, 'LCTopo',
        outputCollectionName='AntiKt4LCTopoJetsTest').jetAlgorithmHandle()
    jetAlg4Tr = make_StandardJetGetter(
        'AntiKt', 0.4, 'Truth',
        outputCollectionName='AntiKt4TruthJetsTest').jetAlgorithmHandle()

    #jetAlg6EM  = make_StandardJetGetter('AntiKt',0.6,'Topo',calibName='EM:OFFSET_ORIGIN_ETAJES',outputCollectionName='AntiKt6TopoJetsTest').jetAlgorithmHandle()
    #jetAlg6EM  = make_StandardJetGetter('AntiKt',0.6,'Topo',doCalib=False,minPt=5000,jetMoments=[],outputCollectionName='AntiKt6TopoJetsTest').jetAlgorithmHandle()
    #jetAlg6LC  = make_StandardJetGetter('AntiKt',0.6,'LCTopo',jetMoments=[],outputCollectionName='AntiKt6LCTopoJetsTest').jetAlgorithmHandle()
    jetAlg6EM = make_StandardJetGetter(
        'AntiKt',
        0.6,
        'Topo',
        doCalib=False,
        minPt=5000,
        outputCollectionName='AntiKt6TopoJetsTest').jetAlgorithmHandle()
    jetAlg6LC = make_StandardJetGetter(
        'AntiKt', 0.6, 'LCTopo',
        outputCollectionName='AntiKt6LCTopoJetsTest').jetAlgorithmHandle()
    jetAlg6Tr = make_StandardJetGetter(
        'AntiKt', 0.6, 'Truth',
        outputCollectionName='AntiKt6TruthJetsTest').jetAlgorithmHandle()

    # In devval, areas are set by default
    # However, we need to set second areas
    g = topSequence.AntiKt4TopoJetsTest.AlgTools[0]
    f = g.JetFindingSequence[-1]
    f.CalculateSecondArea = True

    g = topSequence.AntiKt4LCTopoJetsTest.AlgTools[0]
    f = g.JetFindingSequence[-1]
    f.CalculateSecondArea = True

    #g = topSequence.AntiKt4TruthJetsTest.AlgTools[0]
    #f = g.JetFindingSequence[-1]
    g = topSequence.AntiKt4TruthJetsTest.AlgTools[2]
    g.CalculateSecondArea = True

    g = topSequence.AntiKt6TopoJetsTest.AlgTools[0]
    f = g.JetFindingSequence[-1]
    f.CalculateSecondArea = True

    g = topSequence.AntiKt6LCTopoJetsTest.AlgTools[0]
    f = g.JetFindingSequence[-1]
    f.CalculateSecondArea = True

    #g = topSequence.AntiKt6TruthJetsTest.AlgTools[0]
    #f = g.JetFindingSequence[-1]
    g = topSequence.AntiKt6TruthJetsTest.AlgTools[2]
    g.CalculateSecondArea = True

    from JetValidation.JetTestConfig import scheduleJetTester
    # Now schedule the testers
    scheduleJetTester("CaloCalTopoCluster",
                      JetMoments=False,
                      CollectionType="CaloClusterContainer",
                      JetSigState=False,
                      JetAuthor=False,
                      JetNConstituents=False,
                      JetConstituents=False)

    scheduleJetTester("AntiKt4TopoJetsTest",
                      JetMoments_opts={
                          'moments': [
                              'nTrk', 'nTrk_allpv', 'JVF', 'WIDTH',
                              'trackWIDTH', 'trackWIDTH_allpv', 'ActiveArea',
                              'ActiveAreaPx', 'ActiveAreaPy', 'ActiveAreaPz',
                              'ActiveAreaE', 'VoronoiArea', 'VoronoiAreaPx',
                              'VoronoiAreaPy', 'VoronoiAreaPz', 'VoronoiAreaE'
                          ]
                      })
    scheduleJetTester("AntiKt4LCTopoJetsTest",
                      JetMoments_opts={
                          'moments': [
                              'nTrk', 'nTrk_allpv', 'JVF', 'WIDTH',
                              'trackWIDTH', 'trackWIDTH_allpv', 'ActiveArea',
                              'ActiveAreaPx', 'ActiveAreaPy', 'ActiveAreaPz',
                              'ActiveAreaE', 'VoronoiArea', 'VoronoiAreaPx',
                              'VoronoiAreaPy', 'VoronoiAreaPz', 'VoronoiAreaE'
                          ]
                      })
    scheduleJetTester("AntiKt4TruthJetsTest",
                      JetMoments_opts={
                          'moments': [
                              'nTrk', 'nTrk_allpv', 'JVF', 'WIDTH',
                              'trackWIDTH', 'trackWIDTH_allpv', 'ActiveArea',
                              'ActiveAreaPx', 'ActiveAreaPy', 'ActiveAreaPz',
                              'ActiveAreaE', 'VoronoiArea', 'VoronoiAreaPx',
                              'VoronoiAreaPy', 'VoronoiAreaPz', 'VoronoiAreaE'
                          ]
                      })

    scheduleJetTester("AntiKt6TopoJetsTest",
                      JetMoments_opts={
                          'moments': [
                              'nTrk', 'nTrk_allpv', 'JVF', 'WIDTH',
                              'trackWIDTH', 'trackWIDTH_allpv', 'ActiveArea',
                              'ActiveAreaPx', 'ActiveAreaPy', 'ActiveAreaPz',
                              'ActiveAreaE', 'VoronoiArea', 'VoronoiAreaPx',
                              'VoronoiAreaPy', 'VoronoiAreaPz', 'VoronoiAreaE'
                          ]
                      })
    scheduleJetTester("AntiKt6LCTopoJetsTest",
                      JetMoments_opts={
                          'moments': [
                              'nTrk', 'nTrk_allpv', 'JVF', 'WIDTH',
                              'trackWIDTH', 'trackWIDTH_allpv', 'ActiveArea',
                              'ActiveAreaPx', 'ActiveAreaPy', 'ActiveAreaPz',
                              'ActiveAreaE', 'VoronoiArea', 'VoronoiAreaPx',
                              'VoronoiAreaPy', 'VoronoiAreaPz', 'VoronoiAreaE'
                          ]
                      })
    scheduleJetTester("AntiKt6TruthJetsTest",
                      JetMoments_opts={
                          'moments': [
                              'nTrk', 'nTrk_allpv', 'JVF', 'WIDTH',
                              'trackWIDTH', 'trackWIDTH_allpv', 'ActiveArea',
                              'ActiveAreaPx', 'ActiveAreaPy', 'ActiveAreaPz',
                              'ActiveAreaE', 'VoronoiArea', 'VoronoiAreaPx',
                              'VoronoiAreaPy', 'VoronoiAreaPz', 'VoronoiAreaE'
                          ]
                      })
Example #8
0
    BTaggingFlags.TrackCountingFlip = JetTagD3PD_calibTaggers
    BTaggingFlags.MV1Flip = JetTagD3PD_calibTaggers
    BTaggingFlags.MV2Flip = JetTagD3PD_calibTaggers
    BTaggingFlags.JetProb = False

    include("BTagging/BTagging_LoadTools.py")

    ### stop btag and create the new jet collection
    ###(in the case of new taggers with new code not available in the jet code)
    jetFlags.doBTagging = False
    if JetTagD3PD_AntiKt4MuonNonInteractingTruth:
        from JetRec.JetGetters import make_StandardJetGetter
        make_StandardJetGetter(
            'AntiKt',
            0.4,
            'Truth',
            disable=False,
            includeMuons=True,
            useInteractingOnly=False,
            outputCollectionName='AntiKt4MuonNonInteractingTruthJets')
        if JetTagD3PD_runJVF:
            myjvatoolakt4tmn = getJetVertexAssociationTool(
                'AntiKt', 0.4, 'Topo')
            make_JetMomentGetter('AntiKt4MuonNonInteractingTruthJets',
                                 [myjvatoolakt4tmn])

    if JetTagD3PD_AntiKt6MuonNonInteractingTruth:
        from JetRec.JetGetters import make_StandardJetGetter
        make_StandardJetGetter(
            'AntiKt',
            0.6,
            'Truth',