CommonTools/ParticleFlow/python/Isolation/customiseEarlyDeleteForCandIsoDeposits.py - master - githubmirror/cmssw

CommonTools/ParticleFlow/python/Isolation/customiseEarlyDeleteForCandIsoDeposits.py

17 строк681 B

David

remove unneeded import six

17 июл 2021, 11:39

17 июл 2021, 11:392b86720

import FWCore.ParameterSet.Config as cms

import collections

def customiseEarlyDeleteForCandIsoDeposits(process, products):
    # Find the producers

    def _branchName(productType, moduleLabel, instanceLabel=""):
        return "%s_%s_%s_%s" % (productType, moduleLabel, instanceLabel, process.name_())

    for name, module in process.producers_().items():
        cppType = module._TypedParameterizable__type
        if cppType == "CandIsoDepositProducer":
            if module.ExtractorPSet.ComponentName in ["CandViewExtractor", "PFCandWithSuperClusterExtractor"] :
                products[name].append(_branchName("recoIsoDepositedmValueMap", name))

    return products