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
bool	hasAbilityToProduceInLumis () const final
bool	hasAbilityToProduceInRuns () const final

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

Private Attributes
double	alpha_
double	beta_
std::vector< std::string >	cuts_
std::string	ecftype_
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_

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

Detailed Description

Definition at line 16 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, gen::n, Njets_, routine_, selectors_, 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;
  }

    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_ == "" ) {
      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(), DEFINE_FWK_MODULE, 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;");
  descriptions.add("ECFAdder", iDesc);
}

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

Definition at line 99 of file ECFAdder.cc.

References edm::Ptr< T >::get(), edm::Ptr< T >::isAvailable(), edm::Ptr< T >::isNonnull(), join(), gen::k, checklumidiff::l, Njets_, and routine_.

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 ) {
      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);
          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 65 of file ECFAdder.cc.

References edm::Event::getByToken(), getECF(), mps_fire::i, fwrapper::jets, eostools::move(), gen::n, Njets_, edm::Event::put(), selectors_, src_token_, protons_cff::t, and variables_.

                                                                      {
  // read input collection
  edm::Handle<edm::View<reco::Jet> > jets;
  iEvent.getByToken(src_token_, jets);
  
  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 32 of file ECFAdder.h.

Referenced by ECFAdder().

double ECFAdder::beta_

private

Definition at line 33 of file ECFAdder.h.

Referenced by ECFAdder().

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

private

Definition at line 29 of file ECFAdder.h.

Referenced by ECFAdder().

std::string ECFAdder::ecftype_

private

Definition at line 30 of file ECFAdder.h.

Referenced by ECFAdder().

std::vector<unsigned> ECFAdder::Njets_

private

Definition at line 28 of file ECFAdder.h.

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

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

private

Definition at line 35 of file ECFAdder.h.

Referenced by ECFAdder(), and getECF().

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

private

Definition at line 36 of file ECFAdder.h.

Referenced by ECFAdder(), and produce().

edm::InputTag ECFAdder::src_

private

Definition at line 26 of file ECFAdder.h.

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

private

Definition at line 27 of file ECFAdder.h.

Referenced by produce().

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

private

Definition at line 31 of file ECFAdder.h.

Referenced by ECFAdder(), and produce().