customiseTrackingNtuple Namespace Reference

Functions
def	customiseTrackingNtuple

Function Documentation

def customiseTrackingNtuple.customiseTrackingNtuple

(

process

)

Definition at line 3 of file customiseTrackingNtuple.py.

def customiseTrackingNtuple(process):
    process.load("Validation.RecoTrack.trackingNtuple_cff")
    process.TFileService = cms.Service("TFileService",
        fileName = cms.string('trackingNtuple.root')
    )

    if process.trackingNtuple.includeSeeds.value() and not hasattr(process, "reconstruction_step"):
        raise Exception("TrackingNtuple includeSeeds=True needs reconstruction which is missing")

    # Replace validation_step with ntuplePath
    if not hasattr(process, "validation_step"):
        raise Exception("TrackingNtuple customise assumes process.validation_step exists")


    # Should replay mixing for pileup simhits?
    usePileupSimHits = hasattr(process, "mix") and hasattr(process.mix, "input") and len(process.mix.input.fileNames) > 0
#    process.eda = cms.EDAnalyzer("EventContentAnalyzer")

    ntuplePath = cms.EndPath(process.trackingNtupleSequence)
    if process.trackingNtuple.includeAllHits and usePileupSimHits:
        ntuplePath.insert(0, cms.SequencePlaceholder("mix"))

        process.load("Validation.RecoTrack.crossingFramePSimHitToPSimHits_cfi")
        instanceLabels = [tag.getModuleLabel()+tag.getProductInstanceLabel() for tag in process.simHitTPAssocProducer.simHitSrc]
        process.crossingFramePSimHitToPSimHits.src = ["mix:"+l for l in instanceLabels]
        process.simHitTPAssocProducer.simHitSrc = ["crossingFramePSimHitToPSimHits:"+l for l in instanceLabels]
        process.trackingNtupleSequence.insert(0, process.crossingFramePSimHitToPSimHits)

    # Bit of a hack but works
    modifier = cms.Modifier()
    modifier._setChosen()
    modifier.toReplaceWith(process.validation_step, ntuplePath)

    if hasattr(process, "prevalidation_step"):
        modifier.toReplaceWith(process.prevalidation_step, cms.Path())

    # Remove all output modules
    for outputModule in process.outputModules_().itervalues():
        for path in process.paths_().itervalues():
            path.remove(outputModule)
        for path in process.endpaths_().itervalues():
            path.remove(outputModule)
        

    return process