pat::PATJetUpdater Class Reference

Produces pat::Jet's. More...

Inheritance diagram for pat::PATJetUpdater:

Public Member Functions
	PATJetUpdater (const edm::ParameterSet &iConfig)
void	produce (edm::Event &iEvent, const edm::EventSetup &iSetup) override
	~PATJetUpdater () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Private Attributes
bool	addBTagInfo_
bool	addDiscriminators_
bool	addJetCorrFactors_
bool	addTagInfos_
std::vector< std::string >	discriminatorLabels_
std::vector< edm::InputTag >	discriminatorTags_
std::vector< edm::EDGetTokenT< reco::JetFloatAssociation::Container > >	discriminatorTokens_
std::vector< edm::EDGetTokenT< edm::ValueMap< JetCorrFactors > > >	jetCorrFactorsTokens_
edm::EDGetTokenT< edm::View< reco::Jet > >	jetsToken_
bool	printWarning_
GreaterByPt< Jet >	pTComparator_
bool	sort_
std::vector< std::string >	tagInfoLabels_
std::vector< edm::InputTag >	tagInfoTags_
std::vector< edm::EDGetTokenT< edm::View< reco::BaseTagInfo > > >	tagInfoTokens_
pat::PATUserDataHelper< pat::Jet >	userDataHelper_
bool	useUserData_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
using	CacheTypes = CacheContexts< T... >
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T... >
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Produces pat::Jet's.

The PATJetUpdater produces analysis-level pat::Jet's starting from a collection of pat::Jet's and updates information.

Author

Andreas Hinzmann

Version

Id

PATJetUpdater.h,v 1.00 2014/03/11 18:13:54 srappocc Exp

Definition at line 37 of file PATJetUpdater.cc.

Constructor & Destructor Documentation

PATJetUpdater()

PATJetUpdater::PATJetUpdater ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 75 of file PATJetUpdater.cc.

    : useUserData_(iConfig.exists("userData")), printWarning_(iConfig.getParameter<bool>("printWarning")) {
  // initialize configurables
  jetsToken_ = consumes<edm::View<reco::Jet>>(iConfig.getParameter<edm::InputTag>("jetSource"));
  sort_ = iConfig.getParameter<bool>("sort");
  addJetCorrFactors_ = iConfig.getParameter<bool>("addJetCorrFactors");
  if (addJetCorrFactors_) {
    jetCorrFactorsTokens_ = edm::vector_transform(
        iConfig.getParameter<std::vector<edm::InputTag>>("jetCorrFactorsSource"),
        [this](edm::InputTag const& tag) { return mayConsume<edm::ValueMap<JetCorrFactors>>(tag); });
  }
  addBTagInfo_ = iConfig.getParameter<bool>("addBTagInfo");
  addDiscriminators_ = iConfig.getParameter<bool>("addDiscriminators");
  discriminatorTags_ = iConfig.getParameter<std::vector<edm::InputTag>>("discriminatorSources");
  discriminatorTokens_ = edm::vector_transform(discriminatorTags_, [this](edm::InputTag const& tag) {
    return mayConsume<reco::JetFloatAssociation::Container>(tag);
  });
  addTagInfos_ = iConfig.getParameter<bool>("addTagInfos");
  tagInfoTags_ = iConfig.getParameter<std::vector<edm::InputTag>>("tagInfoSources");
  tagInfoTokens_ = edm::vector_transform(
      tagInfoTags_, [this](edm::InputTag const& tag) { return mayConsume<edm::View<reco::BaseTagInfo>>(tag); });
  if (discriminatorTags_.empty()) {
    addDiscriminators_ = false;
  } else {
    for (std::vector<edm::InputTag>::const_iterator it = discriminatorTags_.begin(), ed = discriminatorTags_.end();
         it != ed;
         ++it) {
      std::string label = it->label();
      std::string::size_type pos = label.find("JetTags");
      if ((pos != std::string::npos) && (pos != label.length() - 7)) {
        label.erase(pos + 7);  // trim a tail after "JetTags"
      }
      if (!it->instance().empty()) {
        label = (label + std::string(":") + it->instance());
      }
      discriminatorLabels_.push_back(label);
    }
  }
  if (tagInfoTags_.empty()) {
    addTagInfos_ = false;
  } else {
    for (std::vector<edm::InputTag>::const_iterator it = tagInfoTags_.begin(), ed = tagInfoTags_.end(); it != ed;
         ++it) {
      std::string label = it->label();
      std::string::size_type pos = label.find("TagInfos");
      if ((pos != std::string::npos) && (pos != label.length() - 8)) {
        label.erase(pos + 8);  // trim a tail after "TagInfos"
      }
      tagInfoLabels_.push_back(label);
    }
  }
  if (!addBTagInfo_) {
    addDiscriminators_ = false;
    addTagInfos_ = false;
  }
  // Check to see if the user wants to add user data
  if (useUserData_) {
    userDataHelper_ = PATUserDataHelper<Jet>(iConfig.getParameter<edm::ParameterSet>("userData"), consumesCollector());
  }
  // produces vector of jets
  produces<std::vector<Jet>>();
  produces<edm::OwnVector<reco::BaseTagInfo>>("tagInfos");
}

References addBTagInfo_, addDiscriminators_, addJetCorrFactors_, addTagInfos_, discriminatorLabels_, discriminatorTags_, discriminatorTokens_, edm::ParameterSet::getParameter(), jetCorrFactorsTokens_, jetsToken_, label, sort_, AlCaHLTBitMon_QueryRunRegistry::string, makeGlobalPositionRcd_cfg::tag, tagInfoLabels_, tagInfoTags_, tagInfoTokens_, userDataHelper_, useUserData_, and edm::vector_transform().

~PATJetUpdater()

PATJetUpdater::~PATJetUpdater ( )

override

Definition at line 139 of file PATJetUpdater.cc.

{}

Member Function Documentation

fillDescriptions()

void PATJetUpdater::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 281 of file PATJetUpdater.cc.

                                                                               {
  edm::ParameterSetDescription iDesc;
  iDesc.setComment("PAT jet producer module");
 
  // input source
  iDesc.add<edm::InputTag>("jetSource", edm::InputTag("no default"))->setComment("input collection");
 
  // sort inputs (by pt)
  iDesc.add<bool>("sort", true);
 
  // tag info
  iDesc.add<bool>("addTagInfos", true);
  std::vector<edm::InputTag> emptyVInputTags;
  iDesc.add<std::vector<edm::InputTag>>("tagInfoSources", emptyVInputTags);
 
  // jet energy corrections
  iDesc.add<bool>("addJetCorrFactors", true);
  iDesc.add<std::vector<edm::InputTag>>("jetCorrFactorsSource", emptyVInputTags);
 
  // btag discriminator tags
  iDesc.add<bool>("addBTagInfo", true);
  iDesc.add<bool>("addDiscriminators", true);
  iDesc.add<std::vector<edm::InputTag>>("discriminatorSources", emptyVInputTags);
 
  // silent warning if false
  iDesc.add<bool>("printWarning", true);
 
  // Check to see if the user wants to add user data
  edm::ParameterSetDescription userDataPSet;
  PATUserDataHelper<Jet>::fillDescription(userDataPSet);
  iDesc.addOptional("userData", userDataPSet);
 
  descriptions.add("PATJetUpdater", iDesc);
}

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), edm::ParameterSetDescription::addOptional(), pat::PATUserDataHelper< ObjectType >::fillDescription(), HLT_FULL_cff::InputTag, and edm::ParameterSetDescription::setComment().

produce()

void PATJetUpdater::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

override

Definition at line 141 of file PATJetUpdater.cc.

                                                                         {
  // Get the vector of jets
  edm::Handle<edm::View<reco::Jet>> jets;
  iEvent.getByToken(jetsToken_, jets);
 
  // read in the jet correction factors ValueMap
  std::vector<edm::ValueMap<JetCorrFactors>> jetCorrs;
  if (addJetCorrFactors_) {
    for (size_t i = 0; i < jetCorrFactorsTokens_.size(); ++i) {
      edm::Handle<edm::ValueMap<JetCorrFactors>> jetCorr;
      iEvent.getByToken(jetCorrFactorsTokens_[i], jetCorr);
      jetCorrs.push_back(*jetCorr);
    }
  }
 
  // Get the vector of jet tags with b-tagging info
  std::vector<edm::Handle<reco::JetFloatAssociation::Container>> jetDiscriminators;
  if (addBTagInfo_ && addDiscriminators_) {
    jetDiscriminators.resize(discriminatorTokens_.size());
    for (size_t i = 0; i < discriminatorTokens_.size(); ++i) {
      iEvent.getByToken(discriminatorTokens_[i], jetDiscriminators[i]);
    }
  }
  std::vector<edm::Handle<edm::View<reco::BaseTagInfo>>> jetTagInfos;
  if (addBTagInfo_ && addTagInfos_) {
    jetTagInfos.resize(tagInfoTokens_.size());
    for (size_t i = 0; i < tagInfoTokens_.size(); ++i) {
      iEvent.getByToken(tagInfoTokens_[i], jetTagInfos[i]);
    }
  }
 
  // loop over jets
  auto patJets = std::make_unique<std::vector<Jet>>();
 
  auto tagInfosOut = std::make_unique<edm::OwnVector<reco::BaseTagInfo>>();
 
  edm::RefProd<edm::OwnVector<reco::BaseTagInfo>> h_tagInfosOut =
      iEvent.getRefBeforePut<edm::OwnVector<reco::BaseTagInfo>>("tagInfos");
 
  for (edm::View<reco::Jet>::const_iterator itJet = jets->begin(); itJet != jets->end(); itJet++) {
    // construct the Jet from the ref -> save ref to original object
    unsigned int idx = itJet - jets->begin();
    const edm::RefToBase<reco::Jet> jetRef = jets->refAt(idx);
    const edm::RefToBase<Jet> patJetRef(jetRef.castTo<JetRef>());
    Jet ajet(patJetRef);
 
    if (addJetCorrFactors_) {
      // undo previous jet energy corrections
      ajet.setP4(ajet.correctedP4(0));
      // clear previous JetCorrFactors
      ajet.jec_.clear();
      // add additional JetCorrs to the jet
      for (unsigned int i = 0; i < jetCorrFactorsTokens_.size(); ++i) {
        const JetCorrFactors& jcf = jetCorrs[i][jetRef];
        // uncomment for debugging
        // jcf.print();
        ajet.addJECFactors(jcf);
      }
      std::vector<std::string> levels = jetCorrs[0][jetRef].correctionLabels();
      if (std::find(levels.begin(), levels.end(), "L2L3Residual") != levels.end()) {
        ajet.initializeJEC(jetCorrs[0][jetRef].jecLevel("L2L3Residual"));
      } else if (std::find(levels.begin(), levels.end(), "L3Absolute") != levels.end()) {
        ajet.initializeJEC(jetCorrs[0][jetRef].jecLevel("L3Absolute"));
      } else {
        ajet.initializeJEC(jetCorrs[0][jetRef].jecLevel("Uncorrected"));
        if (printWarning_) {
          edm::LogWarning("L3Absolute not found")
              << "L2L3Residual and L3Absolute are not part of the jetCorrFactors\n"
              << "of module " << jetCorrs[0][jetRef].jecSet() << ". Jets will remain"
              << " uncorrected.";
          printWarning_ = false;
        }
      }
    }
 
    // add b-tag info if available & required
    if (addBTagInfo_) {
      if (addDiscriminators_) {
        for (size_t k = 0; k < jetDiscriminators.size(); ++k) {
          float value = (*jetDiscriminators[k])[jetRef];
          ajet.addBDiscriminatorPair(std::make_pair(discriminatorLabels_[k], value));
        }
      }
      if (addTagInfos_) {
        for (size_t k = 0; k < jetTagInfos.size(); ++k) {
          const edm::View<reco::BaseTagInfo>& taginfos = *jetTagInfos[k];
          // This is not associative, so we have to search the jet
          edm::Ptr<reco::BaseTagInfo> match;
          // Try first by 'same index'
          if ((idx < taginfos.size()) && (taginfos[idx].jet() == jetRef)) {
            match = taginfos.ptrAt(idx);
          } else {
            // otherwise fail back to a simple search
            for (edm::View<reco::BaseTagInfo>::const_iterator itTI = taginfos.begin(), edTI = taginfos.end();
                 itTI != edTI;
                 ++itTI) {
              if (itTI->jet() == jetRef) {
                match = taginfos.ptrAt(itTI - taginfos.begin());
                break;
              }
            }
          }
          if (match.isNonnull()) {
            tagInfosOut->push_back(match->clone());
            // set the "forward" ptr to the thinned collection
            edm::Ptr<reco::BaseTagInfo> tagInfoForwardPtr(
                h_tagInfosOut.id(), &tagInfosOut->back(), tagInfosOut->size() - 1);
            // set the "backward" ptr to the original collection for association
            const edm::Ptr<reco::BaseTagInfo>& tagInfoBackPtr(match);
            // make FwdPtr
            TagInfoFwdPtrCollection::value_type tagInfoFwdPtr(tagInfoForwardPtr, tagInfoBackPtr);
            ajet.addTagInfo(tagInfoLabels_[k], tagInfoFwdPtr);
          }
        }
      }
    }
 
    if (useUserData_) {
      userDataHelper_.add(ajet, iEvent, iSetup);
    }
 
    // reassign the original object reference to preserve reference to the original jet the input PAT jet was derived from
    // (this needs to be done at the end since cloning the input PAT jet would interfere with adding UserData)
    ajet.refToOrig_ = patJetRef->originalObjectRef();
 
    patJets->push_back(ajet);
  }
 
  // sort jets in pt
  if (sort_) {
    std::sort(patJets->begin(), patJets->end(), pTComparator_);
  }
 
  // put genEvt  in Event
  iEvent.put(std::move(patJets));
 
  iEvent.put(std::move(tagInfosOut), "tagInfos");
}

References pat::PATUserDataHelper< ObjectType >::add(), pat::Jet::addBDiscriminatorPair(), addBTagInfo_, addDiscriminators_, pat::Jet::addJECFactors(), addJetCorrFactors_, pat::Jet::addTagInfo(), addTagInfos_, edm::View< T >::begin(), edm::RefToBase< T >::castTo(), pat::Jet::correctedP4(), discriminatorLabels_, discriminatorTokens_, edm::View< T >::end(), spr::find(), mps_fire::i, heavyIonCSV_trainingSettings::idx, iEvent, pat::Jet::initializeJEC(), pat::Jet::jec_, metsig::jet, jetCorrFactorsTokens_, singleTopDQM_cfi::jets, jetsToken_, dqmdumpme::k, jets_cff::levels, match(), eostools::move(), hemisphereProducer_cfi::patJets, printWarning_, pTComparator_, edm::View< T >::ptrAt(), pat::PATObject< ObjectType >::refToOrig_, reco::LeafCandidate::setP4(), edm::View< T >::size(), jetUpdater_cfi::sort, sort_, tagInfoLabels_, tagInfoTokens_, userDataHelper_, and useUserData_.

Member Data Documentation

addBTagInfo_

bool pat::PATJetUpdater::addBTagInfo_

private

Definition at line 53 of file PATJetUpdater.cc.

Referenced by PATJetUpdater(), and produce().

addDiscriminators_

bool pat::PATJetUpdater::addDiscriminators_

private

Definition at line 54 of file PATJetUpdater.cc.

Referenced by PATJetUpdater(), and produce().

addJetCorrFactors_

bool pat::PATJetUpdater::addJetCorrFactors_

private

Definition at line 50 of file PATJetUpdater.cc.

Referenced by PATJetUpdater(), and produce().

addTagInfos_

bool pat::PATJetUpdater::addTagInfos_

private

Definition at line 58 of file PATJetUpdater.cc.

Referenced by PATJetUpdater(), and produce().

discriminatorLabels_

std::vector<std::string> pat::PATJetUpdater::discriminatorLabels_

private

Definition at line 57 of file PATJetUpdater.cc.

Referenced by PATJetUpdater(), and produce().

discriminatorTags_

std::vector<edm::InputTag> pat::PATJetUpdater::discriminatorTags_

private

Definition at line 55 of file PATJetUpdater.cc.

Referenced by PATJetUpdater().

discriminatorTokens_

std::vector<edm::EDGetTokenT<reco::JetFloatAssociation::Container> > pat::PATJetUpdater::discriminatorTokens_

private

Definition at line 56 of file PATJetUpdater.cc.

Referenced by PATJetUpdater(), and produce().

jetCorrFactorsTokens_

std::vector<edm::EDGetTokenT<edm::ValueMap<JetCorrFactors> > > pat::PATJetUpdater::jetCorrFactorsTokens_

private

Definition at line 51 of file PATJetUpdater.cc.

Referenced by PATJetUpdater(), and produce().

jetsToken_

edm::EDGetTokenT<edm::View<reco::Jet> > pat::PATJetUpdater::jetsToken_

private

Definition at line 48 of file PATJetUpdater.cc.

Referenced by PATJetUpdater(), and produce().

printWarning_

bool pat::PATJetUpdater::printWarning_

private

Definition at line 68 of file PATJetUpdater.cc.

Referenced by produce().

pTComparator_

GreaterByPt<Jet> pat::PATJetUpdater::pTComparator_

private

Definition at line 63 of file PATJetUpdater.cc.

Referenced by produce().

sort_

bool pat::PATJetUpdater::sort_

private

Definition at line 49 of file PATJetUpdater.cc.

Referenced by PATJetUpdater(), and produce().

tagInfoLabels_

std::vector<std::string> pat::PATJetUpdater::tagInfoLabels_

private

Definition at line 61 of file PATJetUpdater.cc.

Referenced by PATJetUpdater(), and produce().

tagInfoTags_

std::vector<edm::InputTag> pat::PATJetUpdater::tagInfoTags_

private

Definition at line 59 of file PATJetUpdater.cc.

Referenced by PATJetUpdater().

tagInfoTokens_

std::vector<edm::EDGetTokenT<edm::View<reco::BaseTagInfo> > > pat::PATJetUpdater::tagInfoTokens_

private

Definition at line 60 of file PATJetUpdater.cc.

Referenced by PATJetUpdater(), and produce().

userDataHelper_

pat::PATUserDataHelper<pat::Jet> pat::PATJetUpdater::userDataHelper_

private

Definition at line 66 of file PATJetUpdater.cc.

Referenced by PATJetUpdater(), and produce().

useUserData_

bool pat::PATJetUpdater::useUserData_

private

Definition at line 65 of file PATJetUpdater.cc.

Referenced by PATJetUpdater(), and produce().