ttSemiLepEvtBuilder_cff.py

Go to the documentation of this file.

import FWCore.ParameterSet.Config as cms





from TopQuarkAnalysis.TopEventProducers.sequences.ttSemiLepEvtHypotheses_cff import *


from TopQuarkAnalysis.TopEventProducers.producers.TtSemiLepEvtBuilder_cfi import *


makeTtSemiLepEventTask = cms.Task(
    makeTtSemiLepHypothesesTask,
    ttSemiLepEvent
)
#makeTtSemiLepEvent = cms.Sequence(makeTtSemiLepEventTask)




def addTtSemiLepHypotheses(process,
                           names
                           ):

    
    labels =  getattr(process.ttSemiLepEvent, "hypotheses")
    for obj in range(len(names)):
        
        label = "ttSemiLepHyp" + names[obj][1:]
        
        labels.append(label)
    process.ttSemiLepEvent.hypotheses = labels

    



def removeTtSemiLepHypGenMatch(process):
    #process.makeTtSemiLepHypotheses.remove(process.makeHypothesis_genMatch)
    process.ttSemiLepEvent.hypotheses.remove("ttSemiLepHypGenMatch")
    process.ttSemiLepEvent.genEvent = ''



def setForAllTtSemiLepHypotheses(process, attribute, value):
    modules = ["findTtSemiLepJetCombGeom",
               "findTtSemiLepJetCombMaxSumPtWMass",
               "findTtSemiLepJetCombMVA",
               "findTtSemiLepJetCombWMassDeltaTopMass",
               "findTtSemiLepJetCombWMassMaxSumPt",
               "hitFitTtSemiLepEventHypothesis",
               "kinFitTtSemiLepEventHypothesis",
               "ttSemiLepJetPartonMatch",
               "ttSemiLepHypGenMatch",
               "ttSemiLepHypGeom",
               "ttSemiLepHypHitFit",
               "ttSemiLepHypKinFit",
               "ttSemiLepHypMaxSumPtWMass",
               "ttSemiLepHypMVADisc",
               "ttSemiLepHypWMassDeltaTopMass",
               "ttSemiLepHypWMassMaxSumPt"
               ]
    for obj in range(len(modules)):
        object = getattr(process, modules[obj])
        if hasattr(object, attribute):
            setattr(object, attribute, value)


def useElectronsForAllTtSemiLepHypotheses(process, elecLabel = "selectedPatElectrons"):
    
    import TopQuarkAnalysis.TopKinFitter.TtSemiLepKinFitProducer_Electrons_cfi
    process.kinFitTtSemiLepEventHypothesis = TopQuarkAnalysis.TopKinFitter.TtSemiLepKinFitProducer_Electrons_cfi.kinFitTtSemiLepEvent.clone()
    import TopQuarkAnalysis.TopHitFit.TtSemiLepHitFitProducer_Electrons_cfi
    process.hitFitTtSemiLepEventHypothesis = TopQuarkAnalysis.TopHitFit.TtSemiLepHitFitProducer_Electrons_cfi.hitFitTtSemiLepEvent.clone()
    
    setForAllTtSemiLepHypotheses(process, "leps", elecLabel)