ttFullHadEvtBuilder_cff.py

Go to the documentation of this file.

import FWCore.ParameterSet.Config as cms

#
# produce ttFullHadEvent structure with all necessary ingredients,
# needs ttGenEvent as input
#


from TopQuarkAnalysis.TopEventProducers.sequences.ttFullHadEvtHypotheses_cff import *


from TopQuarkAnalysis.TopEventProducers.producers.TtFullHadEvtBuilder_cfi import *


makeTtFullHadEventTask = cms.Task(
    makeTtFullHadHypothesesTask,
    ttFullHadEvent
)
#makeTtFullHadEvent = cms.Sequence(makeTtFullHadEventTask)




def addTtFullHadHypotheses(process,
                           names
                           ):

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

    



def removeTtFullHadHypGenMatch(process):
    #process.makeTtFullHadHypotheses.remove(process.makeHypothesis_genMatch)
    process.ttFullHadEvent.hypotheses.remove("ttFullHadHypGenMatch")



def setForAllTtFullHadHypotheses(process, attribute, value):
    modules = ["ttFullHadJetPartonMatch",
               "ttFullHadHypGenMatch",
               "kinFitTtFullHadEventHypothesis",
               "ttFullHadHypKinFit"]
    for obj in range(len(modules)):
        object = getattr(process, modules[obj])
        if hasattr(object, attribute):
            setattr(object, attribute, value)