ECFAdder Class Reference

#include <ECFAdder.h>

Inheritance diagram for ECFAdder:

Public Member Functions
	ECFAdder (const edm::ParameterSet &iConfig)
float	getECF (unsigned index, const edm::Ptr< reco::Jet > &object) const
void	produce (edm::Event &iEvent, const edm::EventSetup &iSetup) 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
double	alpha_
double	beta_
std::vector< std::string >	cuts_
std::string	ecftype_
edm::EDGetTokenT < edm::ValueMap< float > >	input_weights_token_
std::vector< unsigned >	Njets_
std::vector< std::shared_ptr < fastjet::FunctionOfPseudoJet < double > > >	routine_
std::vector < StringCutObjectSelector < reco::Jet > >	selectors_
edm::InputTag	src_
edm::EDGetTokenT< edm::View < reco::Jet > >	src_token_
std::vector< std::string >	variables_
edm::ValueMap< float > const *	weightsHandle_

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

Definition at line 15 of file ECFAdder.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 9 of file ECFAdder.cc.

References alpha_, beta_, cuts_, ecftype_, Exception, edm::ParameterSet::getParameter(), input_weights_token_, edm::InputTag::label(), dqmiodumpmetadata::n, Njets_, routine_, selectors_, HLT_FULL_cff::srcWeights, and variables_.

    : src_(iConfig.getParameter<edm::InputTag>("src")),
      src_token_(consumes<edm::View<reco::Jet>>(src_)),
      Njets_(iConfig.getParameter<std::vector<unsigned>>("Njets")),
      cuts_(iConfig.getParameter<std::vector<std::string>>("cuts")),
      ecftype_(iConfig.getParameter<std::string>("ecftype")),
      alpha_(iConfig.getParameter<double>("alpha")),
      beta_(iConfig.getParameter<double>("beta")) {
  if (cuts_.size() != Njets_.size()) {
    throw cms::Exception("ConfigurationError") << "cuts and Njets must be the same size in ECFAdder" << std::endl;
  }

  edm::InputTag srcWeights = iConfig.getParameter<edm::InputTag>("srcWeights");
  if (!srcWeights.label().empty())
    input_weights_token_ = consumes<edm::ValueMap<float>>(srcWeights);

  for (std::vector<unsigned>::const_iterator n = Njets_.begin(); n != Njets_.end(); ++n) {
    std::ostringstream ecfN_str;
    std::shared_ptr<fastjet::FunctionOfPseudoJet<double>> pfunc;

    if (ecftype_ == "ECF" || ecftype_.empty()) {
      ecfN_str << "ecf" << *n;
      pfunc.reset(new fastjet::contrib::EnergyCorrelator(*n, beta_, fastjet::contrib::EnergyCorrelator::pt_R));
    } else if (ecftype_ == "C") {
      ecfN_str << "ecfC" << *n;
      pfunc.reset(new fastjet::contrib::EnergyCorrelatorCseries(*n, beta_, fastjet::contrib::EnergyCorrelator::pt_R));
    } else if (ecftype_ == "D") {
      ecfN_str << "ecfD" << *n;
      pfunc.reset(
          new fastjet::contrib::EnergyCorrelatorGeneralizedD2(alpha_, beta_, fastjet::contrib::EnergyCorrelator::pt_R));
    } else if (ecftype_ == "N") {
      ecfN_str << "ecfN" << *n;
      pfunc.reset(new fastjet::contrib::EnergyCorrelatorNseries(*n, beta_, fastjet::contrib::EnergyCorrelator::pt_R));
    } else if (ecftype_ == "M") {
      ecfN_str << "ecfM" << *n;
      pfunc.reset(new fastjet::contrib::EnergyCorrelatorMseries(*n, beta_, fastjet::contrib::EnergyCorrelator::pt_R));
    } else if (ecftype_ == "U") {
      ecfN_str << "ecfU" << *n;
      pfunc.reset(new fastjet::contrib::EnergyCorrelatorUseries(*n, beta_, fastjet::contrib::EnergyCorrelator::pt_R));
    }
    variables_.push_back(ecfN_str.str());
    produces<edm::ValueMap<float>>(ecfN_str.str());
    routine_.push_back(pfunc);

    selectors_.push_back(StringCutObjectSelector<reco::Jet>(cuts_[n - Njets_.begin()]));
  }
}

Member Function Documentation

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

static

Definition at line 137 of file ECFAdder.cc.

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), HLT_FULL_cff::InputTag, edm::ParameterSetDescription::setComment(), edm::ParameterDescriptionNode::setComment(), and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                          {
  edm::ParameterSetDescription iDesc;
  iDesc.setComment("Energy Correlation Functions adder");

  iDesc.add<edm::InputTag>("src", edm::InputTag("no default"))->setComment("input collection");
  iDesc.add<std::vector<unsigned>>("Njets", {1, 2, 3})->setComment("Number of jets to emulate");
  iDesc.add<std::vector<std::string>>("cuts", {"", "", ""})
      ->setComment("Jet selections for each N value. Size must match Njets.");
  iDesc.add<double>("alpha", 1.0)->setComment("alpha factor, only valid for N2");
  iDesc.add<double>("beta", 1.0)->setComment("angularity factor");
  iDesc.add<std::string>("ecftype", "")->setComment("ECF type: ECF or empty; C; D; N; M; U;");
  iDesc.add<edm::InputTag>("srcWeights", edm::InputTag("puppi"));
  descriptions.add("ECFAdder", iDesc);
}

float ECFAdder::getECF	(	unsigned	index,
		const edm::Ptr< reco::Jet > &	object
	)		const

Definition at line 92 of file ECFAdder.cc.

References runTheMatrix::const, edm::Ptr< T >::get(), input_weights_token_, edm::Ptr< T >::isAvailable(), edm::Ptr< T >::isNonnull(), edm::EDGetTokenT< T >::isUninitialized(), join(), isotrackApplyRegressor::k, cmsLHEtoEOSManager::l, Njets_, routine_, and w().

Referenced by produce().

                                                                          {
  std::vector<fastjet::PseudoJet> fjParticles;
  for (unsigned k = 0; k < object->numberOfDaughters(); ++k) {
    const reco::CandidatePtr& dp = object->daughterPtr(k);
    if (dp.isNonnull() && dp.isAvailable()) {
      // Here, the daughters are the "end" node, so this is a PFJet
      if (dp->numberOfDaughters() == 0) {
        if (object->isWeighted()) {
          if (input_weights_token_.isUninitialized())
            throw cms::Exception("MissingConstituentWeight")
                << "ECFAdder: No weights (e.g. PUPPI) given for weighted jet collection" << std::endl;
          float w = (*weightsHandle_)[dp];
          fjParticles.push_back(fastjet::PseudoJet(dp->px() * w, dp->py() * w, dp->pz() * w, dp->energy() * w));
        } else
          fjParticles.push_back(fastjet::PseudoJet(dp->px(), dp->py(), dp->pz(), dp->energy()));
      } else {  // Otherwise, this is a BasicJet, so you need to descend further.
        auto subjet = dynamic_cast<reco::Jet const*>(dp.get());
        for (unsigned l = 0; l < subjet->numberOfDaughters(); ++l) {
          if (subjet != nullptr) {
            const reco::CandidatePtr& ddp = subjet->daughterPtr(l);
            if (subjet->isWeighted()) {
              if (input_weights_token_.isUninitialized())
                throw cms::Exception("MissingConstituentWeight")
                    << "ECFAdder: No weights (e.g. PUPPI) given for weighted jet collection" << std::endl;
              float w = (*weightsHandle_)[ddp];
              fjParticles.push_back(fastjet::PseudoJet(ddp->px() * w, ddp->py() * w, ddp->pz() * w, ddp->energy() * w));
            } else
              fjParticles.push_back(fastjet::PseudoJet(ddp->px(), ddp->py(), ddp->pz(), ddp->energy()));
          } else {
            edm::LogWarning("MissingJetConstituent") << "BasicJet constituent required for ECF computation is missing!";
          }
        }
      }  // end if basic jet
    }    // end if daughter pointer is nonnull and available
    else
      edm::LogWarning("MissingJetConstituent") << "Jet constituent required for ECF computation is missing!";
  }
  if (fjParticles.size() > Njets_[index]) {
    return routine_[index]->result(join(fjParticles));
  } else {
    return -1.0;
  }
}

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

override

Definition at line 57 of file ECFAdder.cc.

References edm::Event::get(), edm::Event::getByToken(), getECF(), mps_fire::i, input_weights_token_, edm::EDGetTokenT< T >::isUninitialized(), fwrapper::jets, eostools::move(), dqmiodumpmetadata::n, Njets_, edm::Event::put(), selectors_, src_token_, submitPVValidationJobs::t, variables_, and weightsHandle_.

                                                                    {
  // read input collection
  edm::Handle<edm::View<reco::Jet>> jets;
  iEvent.getByToken(src_token_, jets);

  // Get Weights Collection
  if (!input_weights_token_.isUninitialized())
    weightsHandle_ = &iEvent.get(input_weights_token_);

  unsigned i = 0;
  for (std::vector<unsigned>::const_iterator n = Njets_.begin(); n != Njets_.end(); ++n) {
    // prepare room for output
    std::vector<float> ecfN;
    ecfN.reserve(jets->size());

    for (typename edm::View<reco::Jet>::const_iterator jetIt = jets->begin(); jetIt != jets->end(); ++jetIt) {
      edm::Ptr<reco::Jet> jetPtr = jets->ptrAt(jetIt - jets->begin());

      float t = -1.0;
      if (selectors_[n - Njets_.begin()](*jetIt))
        t = getECF(i, jetPtr);

      ecfN.push_back(t);
    }

    auto outT = std::make_unique<edm::ValueMap<float>>();
    edm::ValueMap<float>::Filler fillerT(*outT);
    fillerT.insert(jets, ecfN.begin(), ecfN.end());
    fillerT.fill();

    iEvent.put(std::move(outT), variables_[i]);
    ++i;
  }
}

Member Data Documentation

double ECFAdder::alpha_

private

Definition at line 31 of file ECFAdder.h.

Referenced by ECFAdder().

double ECFAdder::beta_

private

Definition at line 32 of file ECFAdder.h.

Referenced by ECFAdder().

std::vector<std::string> ECFAdder::cuts_

private

Definition at line 28 of file ECFAdder.h.

Referenced by ECFAdder().

std::string ECFAdder::ecftype_

private

Definition at line 29 of file ECFAdder.h.

Referenced by ECFAdder().

edm::EDGetTokenT<edm::ValueMap<float> > ECFAdder::input_weights_token_

private

Definition at line 36 of file ECFAdder.h.

Referenced by ECFAdder(), getECF(), and produce().

std::vector<unsigned> ECFAdder::Njets_

private

Definition at line 27 of file ECFAdder.h.

Referenced by ECFAdder(), getECF(), and produce().

std::vector<std::shared_ptr<fastjet::FunctionOfPseudoJet<double> > > ECFAdder::routine_

private

Definition at line 34 of file ECFAdder.h.

Referenced by ECFAdder(), and getECF().

std::vector<StringCutObjectSelector<reco::Jet> > ECFAdder::selectors_

private

Definition at line 35 of file ECFAdder.h.

Referenced by ECFAdder(), and produce().

edm::InputTag ECFAdder::src_

private

Definition at line 25 of file ECFAdder.h.

edm::EDGetTokenT<edm::View<reco::Jet> > ECFAdder::src_token_

private

Definition at line 26 of file ECFAdder.h.

Referenced by produce().

std::vector<std::string> ECFAdder::variables_

private

Definition at line 30 of file ECFAdder.h.

Referenced by ECFAdder(), and produce().

edm::ValueMap<float> const* ECFAdder::weightsHandle_

private

Definition at line 37 of file ECFAdder.h.

Referenced by produce().