HGCalValidator_cfi.py

Go to the documentation of this file.

import FWCore.ParameterSet.Config as cms

from Validation.HGCalValidation.CaloParticleSelectionForEfficiency_cfi import *
from Validation.HGCalValidation.HGVHistoProducerAlgoBlock_cfi import *

from SimCalorimetry.HGCalAssociatorProducers.LCToCPAssociation_cfi import layerClusterCaloParticleAssociation
from SimCalorimetry.HGCalAssociatorProducers.LCToSCAssociation_cfi import layerClusterSimClusterAssociation

from DQMServices.Core.DQMEDAnalyzer import DQMEDAnalyzer

from RecoHGCal.TICL.iterativeTICL_cff import ticlIterLabels, ticlIterLabelsMerge

labelTst = [cms.InputTag("ticlTracksters"+iteration) for iteration in ticlIterLabelsMerge]
labelTst.extend([cms.InputTag("ticlSimTracksters", "fromCPs"), cms.InputTag("ticlSimTracksters")])
lcInputMask = [cms.InputTag("ticlTracksters"+iteration) for iteration in ticlIterLabels]
lcInputMask.extend([cms.InputTag("ticlSimTracksters", "fromCPs"), cms.InputTag("ticlSimTracksters")])

hgcalValidator = DQMEDAnalyzer(
    "HGCalValidator",

    
    CaloParticleSelectionForEfficiency,

    hits = cms.VInputTag([cms.InputTag("HGCalRecHit", "HGCEERecHits"), cms.InputTag("HGCalRecHit", "HGCHEFRecHits"), cms.InputTag("HGCalRecHit", "HGCHEBRecHits")]),
    
    label_lcl = layerClusterCaloParticleAssociation.label_lc,
    label_tst = cms.VInputTag(labelTst),
    label_simTS = cms.InputTag("ticlSimTracksters"),
    label_simTSFromCP = cms.InputTag("ticlSimTracksters", "fromCPs"),

    associator = cms.untracked.InputTag("layerClusterCaloParticleAssociationProducer"),

    associatorSim = cms.untracked.InputTag("layerClusterSimClusterAssociationProducer"),

    #General info on layers etc.
    SaveGeneralInfo = cms.untracked.bool(True),
    #CaloParticle related plots
    doCaloParticlePlots = cms.untracked.bool(True),
    #Select caloParticles for efficiency or pass through
    doCaloParticleSelection = cms.untracked.bool(True),
    #SimCluster related plots
    doSimClustersPlots = cms.untracked.bool(True),
    label_SimClusters = cms.InputTag("SimClusters"),
    label_SimClustersLevel = cms.InputTag("ClusterLevel"),
    #Layer Cluster related plots
    doLayerClustersPlots = cms.untracked.bool(True),
    label_layerClusterPlots = cms.InputTag("hgcalMergeLayerClusters"),
    label_LCToCPLinking = cms.InputTag("LCToCP_association"),
    #Trackster related plots
    doTrackstersPlots = cms.untracked.bool(True),
    label_TS = cms.string("Morphology"),
    label_TSToCPLinking = cms.string("TSToCP_linking"),
    label_TSToSTSPR = cms.string("TSToSTS_patternRecognition"),
    #candidates plots
    doCandidatesPlots = cms.untracked.bool(True),
    ticlCandidates = cms.string("ticlCandidates"),

    ticlTrackstersMerge = cms.InputTag("ticlTrackstersMerge"),
    simTiclCandidates = cms.InputTag("ticlSimTracksters"),
    recoTracks = cms.InputTag("generalTracks"),
    mergeRecoToSimAssociator = cms.InputTag("tracksterSimTracksterAssociationLinking", "recoToSim"),
    mergeSimToRecoAssociator = cms.InputTag("tracksterSimTracksterAssociationLinking", "simToReco"),
    mergeRecoToSimAssociatorPU = cms.InputTag("tracksterSimTracksterAssociationLinkingPU", "recoToSim"),

    #The cumulative material budget in front of each layer. To be more specific, it
    #is the material budget just in front of the active material (not including it).
    #This file is created using the official material budget code.
    cummatbudinxo = cms.FileInPath('Validation/HGCalValidation/data/D41.cumulative.xo'),

    
    label_cp_effic = layerClusterCaloParticleAssociation.label_cp,
    label_cp_fake = cms.InputTag("mix","MergedCaloTruth"),
    #simClusters
    label_scl = layerClusterSimClusterAssociation.label_scl,

    simVertices = cms.InputTag("g4SimHits"),

    LayerClustersInputMask = cms.VInputTag(lcInputMask),

    #Total number of layers of HGCal that we want to monitor
    #Could get this also from HGCalImagingAlgo::maxlayer but better to get it from here
    totallayers_to_monitor = cms.int32(52),
    #Thicknesses we want to monitor. -1 is for scintillator
    thicknesses_to_monitor = cms.vint32(120,200,300,-1),

    # HistoProducerAlgo. Defines the set of plots to be booked and filled
    histoProducerAlgoBlock = HGVHistoProducerAlgoBlock,

    
    dirName = cms.string('HGCAL/HGCalValidator/'),

    isticlv5 = cms.untracked.bool(False)
)

from Configuration.ProcessModifiers.premix_stage2_cff import premix_stage2
premix_stage2.toModify(hgcalValidator,
    label_cp_fake = "mixData:MergedCaloTruth"
)

from Configuration.Eras.Modifier_phase2_hgcalV10_cff import phase2_hgcalV10
phase2_hgcalV10.toModify(hgcalValidator, totallayers_to_monitor = cms.int32(50))

from Configuration.Eras.Modifier_phase2_hgcalV16_cff import phase2_hgcalV16
phase2_hgcalV16.toModify(hgcalValidator, totallayers_to_monitor = cms.int32(47))

from Configuration.ProcessModifiers.ticl_v5_cff import ticl_v5
from Configuration.ProcessModifiers.ticl_superclustering_dnn_cff import ticl_superclustering_dnn
from Configuration.ProcessModifiers.ticl_superclustering_mustache_ticl_cff import ticl_superclustering_mustache_ticl
# labelTst_v5 = ["ticlTrackstersCLUE3DEM", "ticlTrackstersCLUE3DHAD", "ticlTracksterLinks"] # for separate CLUE3D iterations
labelTst_v5 = ["ticlTrackstersCLUE3DHigh", "ticlTracksterLinks"]
labelTst_v5.extend([cms.InputTag("ticlSimTracksters", "fromCPs"), cms.InputTag("ticlSimTracksters")])
# lcInputMask_v5  = ["ticlTrackstersCLUE3DEM", "ticlTrackstersCLUE3DHAD", "ticlTracksterLinks"] # for separate CLUE3D iterations
lcInputMask_v5  = ["ticlTrackstersCLUE3DHigh", "ticlTracksterLinks"]
lcInputMask_v5.extend([cms.InputTag("ticlSimTracksters", "fromCPs"), cms.InputTag("ticlSimTracksters")])

ticl_v5.toModify(hgcalValidator,
    label_tst = cms.VInputTag(labelTst_v5),
    LayerClustersInputMask = cms.VInputTag(lcInputMask_v5),
    ticlTrackstersMerge = cms.InputTag("ticlCandidate"),
    isticlv5 = cms.untracked.bool(True)
)
(ticl_v5 & ticl_superclustering_mustache_ticl).toModify(
    hgcalValidator, label_tst = cms.VInputTag(labelTst_v5 + ["ticlTracksterLinksSuperclusteringMustache"])
)
(ticl_v5 & ticl_superclustering_dnn).toModify(
    hgcalValidator, label_tst = cms.VInputTag(labelTst_v5 + ["ticlTracksterLinksSuperclusteringDNN"])
)