#include <ParametrizedSubtractor.h>

Inheritance diagram for ParametrizedSubtractor:

Public Member Functions
void	calculateOrphanInput (std::vector< fastjet::PseudoJet > &orphanInput) override

void	calculatePedestal (std::vector< fastjet::PseudoJet > const &coll) override

double	getEt (const reco::CandidatePtr &in) const

double	getEta (const reco::CandidatePtr &in) const

double	getMeanAtTower (const reco::CandidatePtr &in) const override

double	getPileUpAtTower (const reco::CandidatePtr &in) const override

double	getPU (int ieta, bool addMean, bool addSigma) const

double	getSigmaAtTower (const reco::CandidatePtr &in) const override

void	offsetCorrectJets () override

	ParametrizedSubtractor (const edm::ParameterSet &iConfig, edm::ConsumesCollector &&iC)

void	rescaleRMS (double s)

void	setupGeometryMap (edm::Event &iEvent, const edm::EventSetup &iSetup) override

void	subtractPedestal (std::vector< fastjet::PseudoJet > &coll) override

	~ParametrizedSubtractor () override

Public Member Functions inherited from PileUpSubtractor
virtual double	getCone (double cone, double eta, double phi, double &et, double &pu)

int	getN (const reco::CandidatePtr &in) const

int	getNwithJets (const reco::CandidatePtr &in) const

virtual double	getPileUpEnergy (int ijet) const

int	ieta (const reco::CandidatePtr &in) const

int	iphi (const reco::CandidatePtr &in) const

	PileUpSubtractor (const edm::ParameterSet &iConfig, edm::ConsumesCollector &&iC)

virtual void	reset (std::vector< edm::Ptr< reco::Candidate > > &input, std::vector< fastjet::PseudoJet > &towers, std::vector< fastjet::PseudoJet > &output)

virtual void	setDefinition (JetDefPtr const &jetDef)

virtual	~PileUpSubtractor ()

Public Attributes
int	bin_

const CentralityBins *	cbins_

double	centrality_

edm::EDGetTokenT < reco::Centrality >	centTag_

bool	dropZeroTowers_

TF1 *	fMean

TF1 *	fPU

TF1 *	fRMS

TH1D *	hC

std::vector< TH1D * >	hEta

std::vector< TH1D * >	hEtaMean

std::vector< TH1D * >	hEtaRMS

bool	interpolate_

bool	sumRecHits_

Additional Inherited Members
Public Types inherited from PileUpSubtractor
typedef std::shared_ptr < fastjet::GhostedAreaSpec >	ActiveAreaSpecPtr

typedef std::shared_ptr < fastjet::ClusterSequence >	ClusterSequencePtr

typedef std::shared_ptr < fastjet::JetDefinition >	JetDefPtr

typedef std::shared_ptr < fastjet::RangeDefinition >	RangeDefPtr

Protected Attributes inherited from PileUpSubtractor
int	activeAreaRepeats

std::vector< HcalDetId >	allgeomid_

bool	doAreaFastjet_

bool	doRhoFastjet_

std::map< int, double >	emean_

std::map< int, double >	esigma_

ActiveAreaSpecPtr	fjActiveArea_

ClusterSequencePtr	fjClusterSeq_

std::vector< fastjet::PseudoJet > *	fjInputs_

JetDefPtr	fjJetDefinition_

std::vector< fastjet::PseudoJet > *	fjJets_

std::vector< fastjet::PseudoJet >	fjOriginalInputs_

CaloGeometry const *	geo_

std::map< int, int >	geomtowers_

edm::ESGetToken< CaloGeometry, CaloGeometryRecord >	geoToken_

double	ghostArea

double	ghostEtaMax

int	ietamax_

int	ietamin_

std::vector< edm::Ptr < reco::Candidate > > *	inputs_

std::vector< double >	jetOffset_

double	jetPtMin_

double	nSigmaPU_

std::map< int, int >	ntowersWithJets_

double	puPtMin_

double	radiusPU_

bool	reRunAlgo_

Detailed Description

Definition at line 14 of file ParametrizedSubtractor.h.

Constructor & Destructor Documentation

ParametrizedSubtractor::ParametrizedSubtractor	(	const edm::ParameterSet &	iConfig,
		edm::ConsumesCollector &&	iC
	)

Definition at line 23 of file ParametrizedSubtractor.cc.

References centTag_, data, fMean, fPU, fRMS, contentValuesFiles::fullPath, edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), hC, hEta, hEtaMean, hEtaRMS, mps_fire::i, EcalCondDB::inf, HLT_FULL_cff::InputTag, interpolate_, AlCaHLTBitMon_QueryRunRegistry::string, and sumRecHits_.

     : PileUpSubtractor(iConfig, std::move(iC)),
       dropZeroTowers_(iConfig.getUntrackedParameter<bool>("dropZeroTowers", true)),
       cbins_(nullptr) {
   centTag_ = iC.consumes<reco::Centrality>(
       iConfig.getUntrackedParameter<edm::InputTag>("centTag", edm::InputTag("hiCentrality", "", "RECO")));
 
   interpolate_ = iConfig.getParameter<bool>("interpolate");
   sumRecHits_ = iConfig.getParameter<bool>("sumRecHits");
 
   std::string ifname = "RecoHI/HiJetAlgos/data/PU_DATA.root";
   TFile* inf = new TFile(edm::FileInPath(ifname).fullPath().data());
   fPU = (TF1*)inf->Get("fPU");
   fMean = (TF1*)inf->Get("fMean");
   fRMS = (TF1*)inf->Get("fRMS");
   hC = (TH1D*)inf->Get("hC");
 
   for (int i = 0; i < 40; ++i) {
     hEta.push_back((TH1D*)inf->Get(Form("hEta_%d", i)));
     hEtaMean.push_back((TH1D*)inf->Get(Form("hEtaMean_%d", i)));
     hEtaRMS.push_back((TH1D*)inf->Get(Form("hEtaRMS_%d", i)));
   }
 }

ParametrizedSubtractor::~ParametrizedSubtractor ( )

inlineoverride

Definition at line 30 of file ParametrizedSubtractor.h.

30 { ; }

Member Function Documentation

void ParametrizedSubtractor::calculateOrphanInput ( std::vector< fastjet::PseudoJet > & orphanInput )

overridevirtual

Reimplemented from PileUpSubtractor.

Definition at line 117 of file ParametrizedSubtractor.cc.

References PileUpSubtractor::fjInputs_.

117 { orphanInput = *fjInputs_; }

PileUpSubtractor::fjInputs_

std::vector< fastjet::PseudoJet > * fjInputs_

Definition: PileUpSubtractor.h:60

void ParametrizedSubtractor::calculatePedestal ( std::vector< fastjet::PseudoJet > const & coll )

overridevirtual

Reimplemented from PileUpSubtractor.

Definition at line 72 of file ParametrizedSubtractor.cc.

72 { return; }

double ParametrizedSubtractor::getEt ( const reco::CandidatePtr & in ) const

Definition at line 180 of file ParametrizedSubtractor.cc.

References CaloTower::constituents(), CaloTower::emEnergy(), relval_parameters_module::energy, PileUpSubtractor::geo_, edm::Ptr< T >::get(), CaloGeometry::getPosition(), CaloTower::hadEnergy(), mps_fire::i, CaloTower::id(), funct::sin(), and PV3DBase< T, PVType, FrameType >::theta().

Referenced by offsetCorrectJets(), and subtractPedestal().

                                                                      {
   const CaloTower* ctc = dynamic_cast<const CaloTower*>(in.get());
   const GlobalPoint& pos = geo_->getPosition(ctc->id());
   double energy = ctc->emEnergy() + ctc->hadEnergy();
 
   if (false) {
     energy = 0;
     const std::vector<DetId>& hitids = ctc->constituents();
     for (unsigned int i = 0; i < hitids.size(); ++i) {
     }
   }
 
   double et = energy * sin(pos.theta());
   return et;
 }

double ParametrizedSubtractor::getEta ( const reco::CandidatePtr & in ) const

Definition at line 196 of file ParametrizedSubtractor.cc.

References PVValHelper::eta, PV3DBase< T, PVType, FrameType >::eta(), PileUpSubtractor::geo_, edm::Ptr< T >::get(), CaloGeometry::getPosition(), and CaloTower::id().

                                                                       {
   const CaloTower* ctc = dynamic_cast<const CaloTower*>(in.get());
   const GlobalPoint& pos = geo_->getPosition(ctc->id());
   double eta = pos.eta();
   return eta;
 }

double ParametrizedSubtractor::getMeanAtTower ( const reco::CandidatePtr & in ) const

overridevirtual

Reimplemented from PileUpSubtractor.

Definition at line 203 of file ParametrizedSubtractor.cc.

References getPU(), and PileUpSubtractor::ieta().

                                                                               {
   int it = ieta(in);
   return getPU(it, true, false);
 }

double ParametrizedSubtractor::getPileUpAtTower ( const reco::CandidatePtr & in ) const

overridevirtual

Reimplemented from PileUpSubtractor.

Definition at line 213 of file ParametrizedSubtractor.cc.

References getPU(), and PileUpSubtractor::ieta().

                                                                                 {
   int it = ieta(in);
   return getPU(it, true, true);
 }

double ParametrizedSubtractor::getPU	(	int	ieta,
		bool	addMean,
		bool	addSigma
	)		const

Definition at line 218 of file ParametrizedSubtractor.cc.

References bin_, centrality_, fMean, fRMS, hC, hEtaMean, hEtaRMS, interpolate_, dqmiodumpmetadata::n, and PileUpSubtractor::nSigmaPU_.

Referenced by getMeanAtTower(), getPileUpAtTower(), getSigmaAtTower(), offsetCorrectJets(), and subtractPedestal().

                                                                                 {
   //double e = hEta[bin_]->GetBinContent(hEta[bin_]->FindBin(ieta));
   //double c = fPU->Eval(centrality_);
 
   double em = hEtaMean[bin_]->GetBinContent(hEtaMean[bin_]->FindBin(ieta));
   double cm = fMean->Eval(centrality_);
 
   double er = hEtaRMS[bin_]->GetBinContent(hEtaRMS[bin_]->FindBin(ieta));
   double cr = fRMS->Eval(centrality_);
 
   if (interpolate_) {
     double n = 0;
     int hbin = 40 - bin_;
     double centerweight = (centrality_ - hC->GetBinCenter(hbin));
     double lowerweight = (centrality_ - hC->GetBinLowEdge(hbin));
     double upperweight = (centrality_ - hC->GetBinLowEdge(hbin + 1));
 
     em *= lowerweight * upperweight;
     er *= lowerweight * upperweight;
     n += lowerweight * upperweight;
 
     if (bin_ > 0) {
       em += upperweight * centerweight * hEtaMean[bin_]->GetBinContent(hEtaMean[bin_ - 1]->FindBin(ieta));
       er += upperweight * centerweight * hEtaRMS[bin_]->GetBinContent(hEtaRMS[bin_ - 1]->FindBin(ieta));
       n += upperweight * centerweight;
     }
 
     if (bin_ < 39) {
       em += lowerweight * centerweight * hEtaMean[bin_]->GetBinContent(hEtaMean[bin_ + 1]->FindBin(ieta));
       er += lowerweight * centerweight * hEtaRMS[bin_]->GetBinContent(hEtaRMS[bin_ + 1]->FindBin(ieta));
       n += lowerweight * centerweight;
     }
     em /= n;
     er /= n;
   }
 
   //   return e*c;
   return addMean * em * cm + addSigma * nSigmaPU_ * er * cr;
 }

double ParametrizedSubtractor::getSigmaAtTower ( const reco::CandidatePtr & in ) const

overridevirtual

Reimplemented from PileUpSubtractor.

Definition at line 208 of file ParametrizedSubtractor.cc.

References getPU(), and PileUpSubtractor::ieta().

                                                                                {
   int it = ieta(in);
   return getPU(it, false, true);
 }

void ParametrizedSubtractor::offsetCorrectJets ( )

overridevirtual

Reimplemented from PileUpSubtractor.

Definition at line 119 of file ParametrizedSubtractor.cc.

                                                {
   LogDebug("PileUpSubtractor") << "The subtractor correcting jets...\n";
   jetOffset_.clear();
 
   using namespace reco;
 
   (*fjInputs_) = fjOriginalInputs_;
   rescaleRMS(nSigmaPU_);
   subtractPedestal(*fjInputs_);
 
   if (false) {
     const fastjet::JetDefinition& def = fjClusterSeq_->jet_def();
     if (!doAreaFastjet_ && !doRhoFastjet_) {
       fastjet::ClusterSequence newseq(*fjInputs_, def);
       (*fjClusterSeq_) = newseq;
     } else {
       fastjet::ClusterSequenceArea newseq(*fjInputs_, def, *fjActiveArea_);
       (*fjClusterSeq_) = newseq;
     }
 
     (*fjJets_) = fastjet::sorted_by_pt(fjClusterSeq_->inclusive_jets(jetPtMin_));
   }
 
   jetOffset_.reserve(fjJets_->size());
 
   vector<fastjet::PseudoJet>::iterator pseudojetTMP = fjJets_->begin(), jetsEnd = fjJets_->end();
   for (; pseudojetTMP != jetsEnd; ++pseudojetTMP) {
     int ijet = pseudojetTMP - fjJets_->begin();
     jetOffset_[ijet] = 0;
 
     std::vector<fastjet::PseudoJet> towers = sorted_by_pt(fjClusterSeq_->constituents(*pseudojetTMP));
 
     double newjetet = 0.;
     for (vector<fastjet::PseudoJet>::const_iterator ito = towers.begin(), towEnd = towers.end(); ito != towEnd; ++ito) {
       const reco::CandidatePtr& originalTower = (*inputs_)[ito->user_index()];
       int it = ieta(originalTower);
       double Original_Et = originalTower->et();
 
       if (sumRecHits_) {
         Original_Et = getEt(originalTower);
       }
 
       double etnew = Original_Et - getPU(it, true, true);
       if (etnew < 0.)
         etnew = 0;
       newjetet = newjetet + etnew;
       jetOffset_[ijet] += Original_Et - etnew;
     }
 
     if (sumRecHits_) {
       double mScale = newjetet / pseudojetTMP->Et();
       int cshist = pseudojetTMP->cluster_hist_index();
       pseudojetTMP->reset(pseudojetTMP->px() * mScale,
                           pseudojetTMP->py() * mScale,
                           pseudojetTMP->pz() * mScale,
                           pseudojetTMP->e() * mScale);
       pseudojetTMP->set_cluster_hist_index(cshist);
     }
   }
 }

void ParametrizedSubtractor::rescaleRMS ( double s )

Definition at line 17 of file ParametrizedSubtractor.cc.

Referenced by offsetCorrectJets().

                                                 {
   for (std::map<int, double>::iterator iter = esigma_.begin(); iter != esigma_.end(); ++iter) {
     iter->second = s * (iter->second);
   }
 }

void ParametrizedSubtractor::setupGeometryMap	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overridevirtual

Reimplemented from PileUpSubtractor.

Definition at line 47 of file ParametrizedSubtractor.cc.

References bin_, centrality_, centTag_, PileUpSubtractor::emean_, PileUpSubtractor::esigma_, edm::Event::getByToken(), hC, mps_fire::i, PileUpSubtractor::ietamax_, PileUpSubtractor::ietamin_, LogDebug, and PileUpSubtractor::setupGeometryMap().

                                                                                            {
   LogDebug("PileUpSubtractor") << "The subtractor setting up geometry...\n";
 
   // The function below that is commented out was deleted from
   // DataFormats/HeavyIonEvent/src/Centrality.cc
   // in June 2015. See comments associated with that commit.
   //   if(!cbins_) getCentralityBinsFromDB(iSetup);
 
   edm::Handle<reco::Centrality> cent;
   iEvent.getByToken(centTag_, cent);
 
   centrality_ = cent->EtHFhitSum();
   bin_ = 40 - hC->FindBin(centrality_);
   if (bin_ > 39)
     bin_ = 39;
   if (bin_ < 0)
     bin_ = 0;
 
   PileUpSubtractor::setupGeometryMap(iEvent, iSetup);
   for (int i = ietamin_; i < ietamax_ + 1; i++) {
     emean_[i] = 0.;
     esigma_[i] = 0.;
   }
 }

void ParametrizedSubtractor::subtractPedestal ( std::vector< fastjet::PseudoJet > & coll )

overridevirtual

Reimplemented from PileUpSubtractor.

Definition at line 74 of file ParametrizedSubtractor.cc.

References dropZeroTowers_, getEt(), getPU(), PileUpSubtractor::ieta(), PileUpSubtractor::iphi(), LogDebug, and sumRecHits_.

Referenced by offsetCorrectJets().

                                                                             {
   if (false) {
     return;
   } else {
     LogDebug("PileUpSubtractor") << "The subtractor subtracting pedestals...\n";
 
     int it = -100;
     vector<fastjet::PseudoJet> newcoll;
 
     for (vector<fastjet::PseudoJet>::iterator input_object = coll.begin(), fjInputsEnd = coll.end();
          input_object != fjInputsEnd;
          ++input_object) {
       reco::CandidatePtr const& itow = (*inputs_)[input_object->user_index()];
 
       it = ieta(itow);
       iphi(itow);
 
       double Original_Et = itow->et();
       if (sumRecHits_) {
         Original_Et = getEt(itow);
       }
 
       double etnew = Original_Et - getPU(it, true, true);
       float mScale = etnew / input_object->Et();
       if (etnew < 0.)
         mScale = 0.;
 
       math::XYZTLorentzVectorD towP4(input_object->px() * mScale,
                                      input_object->py() * mScale,
                                      input_object->pz() * mScale,
                                      input_object->e() * mScale);
 
       int index = input_object->user_index();
       input_object->reset(towP4.px(), towP4.py(), towP4.pz(), towP4.energy());
       input_object->set_user_index(index);
       if (etnew > 0. && dropZeroTowers_)
         newcoll.push_back(*input_object);
     }
     if (dropZeroTowers_)
       coll = newcoll;
   }
 }