HiFJRhoProducer Class Reference

#include <HiJetBackground/HiFJRhoProducer/plugins/HiFJRhoProducer.cc>

Inheritance diagram for HiFJRhoProducer:

Public Member Functions
	HiFJRhoProducer (const edm::ParameterSet &)
	~HiFJRhoProducer () 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 Member Functions
void	beginStream (edm::StreamID) override
double	calcMd (const reco::Jet *jet)
double	calcMedian (std::vector< double > &v)
void	endStream () override
bool	isPackedCandidate (const reco::Candidate *candidate)
void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
bool	checkJetCand
double	etaMaxExcl2_
double	etaMaxExcl_
std::vector< double >	etaRanges
edm::EDGetTokenT< edm::View< reco::Jet > >	jetsToken_
unsigned int	nExcl2_
unsigned int	nExcl_
double	ptMinExcl2_
double	ptMinExcl_
edm::InputTag	src_
bool	usingPackedCand

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

Description: [one line class summary]

Implementation: [Notes on implementation]

Definition at line 26 of file HiFJRhoProducer.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 51 of file HiFJRhoProducer.cc.

References etaRanges, edm::ParameterSet::getParameter(), jetsToken_, and src_.

                                                               :
  src_(iConfig.getParameter<edm::InputTag>("jetSource")),
  nExcl_(iConfig.getParameter<int>("nExcl")),
  etaMaxExcl_(iConfig.getParameter<double>("etaMaxExcl")),
  ptMinExcl_(iConfig.getParameter<double>("ptMinExcl")),
  nExcl2_(iConfig.getParameter<int>("nExcl2")),
  etaMaxExcl2_(iConfig.getParameter<double>("etaMaxExcl2")),
  ptMinExcl2_(iConfig.getParameter<double>("ptMinExcl2")),
  checkJetCand(true),
  usingPackedCand(false)
{
  jetsToken_ = consumes<edm::View<reco::Jet> >(src_);

  //register your products
  produces<std::vector<double > >("mapEtaEdges");
  produces<std::vector<double > >("mapToRho");
  produces<std::vector<double > >("mapToRhoM");
  produces<std::vector<double > >("ptJets");
  produces<std::vector<double > >("areaJets");
  produces<std::vector<double > >("etaJets");
  etaRanges = iConfig.getParameter<std::vector<double> >("etaRanges");

}

HiFJRhoProducer::~HiFJRhoProducer ( )

override

Definition at line 76 of file HiFJRhoProducer.cc.

{
 
   // do anything here that needs to be done at desctruction time
   // (e.g. close files, deallocate resources etc.)

}

Member Function Documentation

void HiFJRhoProducer::beginStream ( edm::StreamID  )

overrideprivate

Definition at line 186 of file HiFJRhoProducer.cc.

{
}

double HiFJRhoProducer::calcMd ( const reco::Jet *  jet )

private

Definition at line 195 of file HiFJRhoProducer.cc.

References reco::Jet::getJetConstituentsQuick(), isPackedCandidate(), and mathSSE::sqrt().

Referenced by produce().

                                                 {
  //
  //get md as defined in http://arxiv.org/pdf/1211.2811.pdf
  //

  //Loop over the jet constituents
  double sum = 0.;
  for(auto daughter : jet->getJetConstituentsQuick()){
    if(isPackedCandidate(daughter)){     //packed candidate situation
      auto part = static_cast<const pat::PackedCandidate*>(daughter);
      sum += sqrt(part->mass()*part->mass() + part->pt()*part->pt()) - part->pt();
    } else {
      auto part = static_cast<const reco::PFCandidate*>(daughter);
      sum += sqrt(part->mass()*part->mass() + part->pt()*part->pt()) - part->pt();
    }
  }

  return sum;
}

double HiFJRhoProducer::calcMedian ( std::vector< double > &  v )

private

Definition at line 240 of file HiFJRhoProducer.cc.

References DEFINE_FWK_MODULE, and gen::n.

Referenced by produce().

{
  //post-condition: After returning, the elements in v may be reordered and the resulting order is implementation defined.
  //works for even and odd collections
  if(v.empty()) {
    return 0.0;
  }
  auto n = v.size() / 2;
  std::nth_element(v.begin(), v.begin()+n, v.end());
  auto med = v[n];
  if(!(v.size() & 1)) { //If the set size is even
    auto max_it = std::max_element(v.begin(), v.begin()+n);
    med = (*max_it + med) / 2.0;
  }
  return med;    
}

void HiFJRhoProducer::endStream ( )

overrideprivate

Definition at line 192 of file HiFJRhoProducer.cc.

                           {
}

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

static

Definition at line 225 of file HiFJRhoProducer.cc.

References edm::ConfigurationDescriptions::add(), and edm::ParameterSetDescription::add().

                                                                                 {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("jetSource",edm::InputTag("kt4PFJets"));
  desc.add<int>("nExcl", 2);
  desc.add<double>("etaMaxExcl",2.);
  desc.add<double>("ptMinExcl",20.);
  desc.add<int>("nExcl2", 2);
  desc.add<double>("etaMaxExcl2",2.);
  desc.add<double>("ptMinExcl2",20.);
  desc.add<std::vector<double> >("etaRanges");
  descriptions.add("hiFJRhoProducer",desc);
}

bool HiFJRhoProducer::isPackedCandidate ( const reco::Candidate *  candidate )

private

Definition at line 215 of file HiFJRhoProducer.cc.

References checkJetCand, Exception, and usingPackedCand.

Referenced by calcMd().

                                                                     {
  if(checkJetCand) {
    if(typeid(pat::PackedCandidate)==typeid(*candidate)) usingPackedCand = true;
    else if(typeid(reco::PFCandidate)==typeid(*candidate)) usingPackedCand = false;
    else throw cms::Exception("WrongJetCollection", "Jet constituents are not particle flow candidates");
    checkJetCand = false;
  }
  return usingPackedCand;
}

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

overrideprivate

Definition at line 85 of file HiFJRhoProducer.cc.

References jets_cff::area, calcMd(), calcMedian(), MillePedeFileConverter_cfg::e, PVValHelper::eta, ALCARECOTkAlBeamHalo_cff::etaMax, etaMaxExcl2_, etaMaxExcl_, ALCARECOTkAlBeamHalo_cff::etaMin, etaRanges, edm::Event::getByToken(), mps_fire::i, createfilelist::int, metsig::jet, fwrapper::jets, jetsToken_, eostools::move(), nExcl2_, nExcl_, EnergyCorrector::pt, ptMinExcl2_, ptMinExcl_, edm::Event::put(), and TCMET_cfi::radius.

{  
  // Get the vector of jets
  edm::Handle<edm::View<reco::Jet> > jets;
  iEvent.getByToken(jetsToken_, jets);

  int neta = (int)etaRanges.size();
  auto mapEtaRangesOut = std::make_unique<std::vector<double>>(neta,-999.);

  for(int ieta = 0; ieta < neta; ieta++){
   mapEtaRangesOut->at(ieta) = etaRanges[ieta];
  }
  auto mapToRhoOut = std::make_unique<std::vector<double>>(neta-1,1e-6);
  auto mapToRhoMOut = std::make_unique<std::vector<double>>(neta-1,1e-6);
  
  int njets = jets->size();
  
  auto ptJetsOut = std::make_unique<std::vector<double>>(njets,1e-6);
  auto areaJetsOut = std::make_unique<std::vector<double>>(njets,1e-6);
  auto etaJetsOut = std::make_unique<std::vector<double>>(njets,1e-6);
    
  double rhoVec[999];
  double rhomVec[999];
  double etaVec[999];

  // int neta = (int)mapEtaRangesOut->size();
  int nacc = 0;
  unsigned int njetsEx = 0;
  unsigned int njetsEx2 = 0;
  for(auto jet = jets->begin(); jet != jets->end(); ++jet) {
    if(njetsEx<nExcl_ && fabs(jet->eta())<etaMaxExcl_ && jet->pt()>ptMinExcl_) {
      njetsEx++;
      continue;
    }
    if(njetsEx2<nExcl2_ && fabs(jet->eta())<etaMaxExcl2_ && fabs(jet->eta())>etaMaxExcl_ && jet->pt()>ptMinExcl2_) {
      njetsEx2++;
      continue;
    }
    float pt = jet->pt();
    float area = jet->jetArea();
    float eta = jet->eta();
    
    if(eta<mapEtaRangesOut->at(0) || eta>mapEtaRangesOut->at(neta-1)) continue;
    if(area>0.) {
      rhoVec[nacc] = pt/area;
      rhomVec[nacc] = calcMd(&*jet)/area;
      etaVec[nacc] = eta;
      ptJetsOut->at(nacc) = pt;
      areaJetsOut->at(nacc) = area;
      etaJetsOut->at(nacc) = eta;
      ++nacc;
    }
  }

  ptJetsOut->resize(nacc);
  areaJetsOut->resize(nacc);
  etaJetsOut->resize(nacc);
  //calculate rho and rhom in eta ranges
  double radius = 0.2; //distance kt clusters needs to be from edge
  for(int ieta = 0; ieta<(neta-1); ieta++) {
    std::vector<double> rhoVecCur;
    std::vector<double> rhomVecCur;
    
    double etaMin = mapEtaRangesOut->at(ieta)+radius;
    double etaMax = mapEtaRangesOut->at(ieta+1)-radius;
     
    int    naccCur    = 0 ;
    double rhoCurSum  = 0.;
    double rhomCurSum = 0.;
    for(int i = 0; i<nacc; i++) {
      if(etaVec[i]>=etaMin && etaVec[i]<etaMax) {
        rhoVecCur.push_back(rhoVec[i]);
        rhomVecCur.push_back(rhomVec[i]);
        
        rhoCurSum += rhoVec[i];
        rhomCurSum += rhomVec[i];
        ++naccCur;
      }//eta selection
    }//accepted jet loop

    if(naccCur>0) {
      double rhoCur = calcMedian(rhoVecCur);
      double rhomCur = calcMedian(rhomVecCur);
      mapToRhoOut->at(ieta) = rhoCur;
      mapToRhoMOut->at(ieta) = rhomCur;
    }
  }//eta ranges
  
  iEvent.put(std::move(mapEtaRangesOut),"mapEtaEdges");
  iEvent.put(std::move(mapToRhoOut),"mapToRho");
  iEvent.put(std::move(mapToRhoMOut),"mapToRhoM");
  
  iEvent.put(std::move(ptJetsOut),"ptJets");
  iEvent.put(std::move(areaJetsOut),"areaJets");
  iEvent.put(std::move(etaJetsOut),"etaJets");

  return;
}

Member Data Documentation

bool HiFJRhoProducer::checkJetCand

private

Definition at line 55 of file HiFJRhoProducer.h.

Referenced by isPackedCandidate().

double HiFJRhoProducer::etaMaxExcl2_

private

Definition at line 52 of file HiFJRhoProducer.h.

Referenced by produce().

double HiFJRhoProducer::etaMaxExcl_

private

Definition at line 49 of file HiFJRhoProducer.h.

Referenced by produce().

std::vector<double> HiFJRhoProducer::etaRanges

private

Definition at line 54 of file HiFJRhoProducer.h.

Referenced by HiFJRhoProducer(), and produce().

edm::EDGetTokenT<edm::View<reco::Jet> > HiFJRhoProducer::jetsToken_

private

Definition at line 44 of file HiFJRhoProducer.h.

Referenced by HiFJRhoProducer(), and produce().

unsigned int HiFJRhoProducer::nExcl2_

private

Definition at line 51 of file HiFJRhoProducer.h.

Referenced by produce().

unsigned int HiFJRhoProducer::nExcl_

private

Definition at line 48 of file HiFJRhoProducer.h.

Referenced by produce().

double HiFJRhoProducer::ptMinExcl2_

private

Definition at line 53 of file HiFJRhoProducer.h.

Referenced by produce().

double HiFJRhoProducer::ptMinExcl_

private

Definition at line 50 of file HiFJRhoProducer.h.

Referenced by produce().

edm::InputTag HiFJRhoProducer::src_

private

Definition at line 47 of file HiFJRhoProducer.h.

Referenced by HiFJRhoProducer().

bool HiFJRhoProducer::usingPackedCand

private

Definition at line 55 of file HiFJRhoProducer.h.

Referenced by isPackedCandidate().