#include <VoronoiSubtractor.h>

Inheritance diagram for VoronoiSubtractor:

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

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

bool	match (fastjet::PseudoJet, fastjet::PseudoJet)

virtual void	offsetCorrectJets ()

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

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

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

	~VoronoiSubtractor ()

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

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

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

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

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

virtual double	getPileUpEnergy (int ijet) const

virtual double	getSigmaAtTower (const reco::CandidatePtr &in) 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 ()

Private Attributes
bool	addNegative_

bool	addNegativesFromCone_

edm::Handle< reco::VoronoiMap >	backgrounds_

edm::Handle< reco::CandidateView >	candidates_

std::vector< int >	droppedCandidates_

bool	dropZeroTowers_

double	infinitesimalPt_

double	rParam_

edm::EDGetTokenT< reco::CandidateView >	srcCand_

edm::EDGetTokenT< edm::ValueMap< reco::VoronoiBackground > >	srcVor_

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

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

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

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

Protected Attributes inherited from PileUpSubtractor
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_

RangeDefPtr	fjRangeDef_

CaloGeometry const *	geo_

std::map< int, int >	geomtowers_

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 8 of file VoronoiSubtractor.h.

Constructor & Destructor Documentation

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

Definition at line 13 of file VoronoiSubtractor.cc.

                                                                                                :
   PileUpSubtractor(iConfig, std::move(iC)),
   srcCand_(iC.consumes<reco::CandidateView>(iConfig.getParameter<edm::InputTag>("src"))),
   srcVor_(iC.consumes<edm::ValueMap<reco::VoronoiBackground> >(iConfig.getParameter<edm::InputTag>("bkg"))), 
   dropZeroTowers_(iConfig.getParameter<bool>("dropZeros")),
   addNegative_(iConfig.getParameter<bool>("addNegative")),
   addNegativesFromCone_(iConfig.getParameter<bool>("addNegativesFromCone")),
   infinitesimalPt_(iConfig.getParameter<double>("infinitesimalPt")),
   rParam_(iConfig.getParameter<double>("rParam"))
 {
 
 }

VoronoiSubtractor::~VoronoiSubtractor ( )

inline

Definition at line 19 of file VoronoiSubtractor.h.

19 {;}

Member Function Documentation

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

virtual

Reimplemented from PileUpSubtractor.

Definition at line 184 of file VoronoiSubtractor.cc.

References LogDebug.

 {
    LogDebug("VoronoiSubtractor")<<"do nothing...\n";
 }

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

virtual

Reimplemented from PileUpSubtractor.

Definition at line 178 of file VoronoiSubtractor.cc.

References LogDebug.

 {
    LogDebug("VoronoiSubtractor")<<"do nothing...\n";
 }

bool VoronoiSubtractor::match	(	fastjet::PseudoJet	cand1,
		fastjet::PseudoJet	cand2
	)

Definition at line 8 of file VoronoiSubtractor.cc.

Referenced by offsetCorrectJets().

                                                                                        {
    return (cand1.delta_R(cand2) < rParam_);
 }

void VoronoiSubtractor::offsetCorrectJets ( )

virtual

Reimplemented from PileUpSubtractor.

Definition at line 41 of file VoronoiSubtractor.cc.

References addNegative_, addNegativesFromCone_, candidates_, droppedCandidates_, PileUpSubtractor::fjJets_, mps_fire::i, diffTreeTool::index, PileUpSubtractor::jetOffset_, LogDebug, match(), reco::VoronoiBackground::pt(), and reco::VoronoiBackground::pt_subtracted().

 {
 
   LogDebug("VoronoiSubtractor")<<"finalizing the output...\n";
 
   jetOffset_.reserve(fjJets_->size());
 
   for (unsigned int ijet = 0;ijet <fjJets_->size();++ijet) {
      fastjet::PseudoJet& fjJet = (*fjJets_)[ijet];
 
      LogDebug("VoronoiSubtractor")<<"fjJets_ "<<ijet<<"   pt : "<<fjJet.pt()
                   <<" --- eta : "<<fjJet.eta()<<" --- phi : "<<fjJet.phi()<<endl;
 
      fastjet::PseudoJet subtracted;
      fastjet::PseudoJet unsubtracted;
      fastjet::PseudoJet unsubtractedDropped;
      jetOffset_[ijet] = 0;
 
      // Loop over constituents to determine the momentum
      // before and after subtraction
 
      std::vector<fastjet::PseudoJet> fjConstituents = fastjet::sorted_by_pt(fjJet.constituents());
      for (unsigned int i=0;i<fjConstituents.size();++i) { 
     unsigned int index = fjConstituents[i].user_index();
 
     reco::CandidateViewRef ref(candidates_,index);
     const reco::VoronoiBackground& voronoi = (*backgrounds_)[ref];
 
     fastjet::PseudoJet candidate(ref->px(),ref->py(),ref->pz(),ref->energy());
     double orpt = candidate.perp();
     unsubtracted += candidate;
     if(addNegative_ || voronoi.pt() > 0){
       candidate.reset_PtYPhiM(addNegative_ ? voronoi.pt_subtracted() : voronoi.pt(),ref->rapidity(),ref->phi(),ref->mass());
       LogDebug("VoronoiSubtractor")<<"candidate "<<index
                        <<" --- original pt : "<<orpt
                        <<" ---  voronoi pt : "<<voronoi.pt()
                        <<" --- ref pt : "<<ref->pt()
                        <<" --- constituent "<<index
                        <<" --- pt : "<<fjConstituents[i].perp()
                        <<" --- eta : "<<fjConstituents[i].pseudorapidity()
                        <<" --- phi : "<<fjConstituents[i].phi()
                        <<" --- mass : "<<fjConstituents[i].m()<<endl;
       subtracted += candidate;
     }
      }
   
      // Loop over dropped candidates to see whether there is any of them
      // that would belong to this jet:
      if(addNegativesFromCone_){
        for(unsigned int i=0; i < droppedCandidates_.size(); ++i){
      reco::CandidateViewRef ref(candidates_,droppedCandidates_[i]);
      fastjet::PseudoJet dropcand(ref->px(),ref->py(),ref->pz(),ref->energy());
      
      if(match(fjJet,dropcand)){
        unsubtractedDropped += dropcand;
        unsubtracted += dropcand;
      }
        }
      }
 
      fjJet.reset_momentum(subtracted);
 
      LogDebug("VoronoiSubtractor")<<"fjJets_ "<<ijet<<"   unsubtracted : "<<unsubtracted.pt()<<endl;
      LogDebug("VoronoiSubtractor")<<"fjJets_ "<<ijet<<"   subtracted : "<<subtracted.pt()<<endl;
      LogDebug("VoronoiSubtractor")<<"fjJets_ "<<ijet<<"   dropped : "<<unsubtractedDropped.pt()<<endl;
      jetOffset_[ijet]  = unsubtracted.pt() - fjJet.pt();
 
   }
 
 
 
 
 
 
 
 
 
 
 }

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

virtual

Reimplemented from PileUpSubtractor.

Definition at line 28 of file VoronoiSubtractor.cc.

References backgrounds_, candidates_, droppedCandidates_, PileUpSubtractor::geo_, edm::Event::getByToken(), PileUpSubtractor::jetOffset_, LogDebug, srcCand_, and srcVor_.

 {
 
    LogDebug("VoronoiSubtractor")<<"The subtractor retrieving Voronoi background...\n";
    geo_ = 0;
    droppedCandidates_.clear();
    jetOffset_.clear();
 
    iEvent.getByToken(srcCand_,candidates_);
    iEvent.getByToken(srcVor_,backgrounds_);
 
 }

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

virtual

Reimplemented from PileUpSubtractor.

Definition at line 121 of file VoronoiSubtractor.cc.

References candidates_, droppedCandidates_, diffTreeTool::index, infinitesimalPt_, LogDebug, and reco::VoronoiBackground::pt().

 { 
 
    LogDebug("VoronoiSubtractor")<<"The subtractor subtracting pedestals...\n";
    vector<fastjet::PseudoJet> newcoll;
 
    for (vector<fastjet::PseudoJet>::iterator input_object = coll.begin (),
        fjInputsEnd = coll.end(); 
     input_object != fjInputsEnd; ++input_object) {
     
       reco::CandidateViewRef ref(candidates_,input_object->user_index());
       const reco::VoronoiBackground& voronoi = (*backgrounds_)[ref];
 
       double ptold = input_object->pt();
       double ptnew = voronoi.pt();
 
       LogDebug("VoronoiSubtractor")<<"pt old : "<<ptold<<" ;   pt new : "<<ptnew
                    <<"  E : "<<input_object->e()
                    <<" rap : "<<input_object->rapidity()
                    <<"  phi : "<<input_object->phi()
                    <<" MASS : "<<input_object->m()<<endl;
 
       // Treatment of candidates with negative pt after subtraction
       if(ptnew < infinitesimalPt_){
     if(infinitesimalPt_ > 0){
       // Low-pt candidate is assigned a very small finite pt
       // so that the jet clustering includes the candidate 
       // and can associate it to the jet.
       // The original candidate pt is restored
       // in the offsetCorrectJets() function.
       ptnew = infinitesimalPt_;
     }else{
       // Low-pt candidate is removed from the input collection,
       // so that the jet clustering algorithm can function properly.
       // However, we need to keep track of these candidates
       // in order to determine how much energy has been subtracted in total.
       droppedCandidates_.push_back(input_object->user_index());
       continue;
     }
       }
 
       int index = input_object->user_index();
 
       fastjet::PseudoJet ps(input_object->four_mom());
       ps.reset_PtYPhiM(ptnew,input_object->rapidity(),input_object->phi(),input_object->m());
       ps.set_user_index(index);
 
       LogDebug("VoronoiSubtractor")<<"New momentum : "<<ps.pt()
                    <<"   rap : "<<ps.rap()
                    <<"   phi : "<<ps.phi()
                    <<" MASS : "<<ps.m()<<endl;
 
       newcoll.push_back(ps); 
    }
    coll = newcoll;
 }