#include <PuppiContainer.h>

Classes
class	PuppiUserInfo

Public Member Functions
void	initialize (const std::vector< RecoObj > &iRecoObjects)

std::vector < fastjet::PseudoJet > const &	pfParticles () const

	PuppiContainer (const edm::ParameterSet &iConfig)

std::vector < fastjet::PseudoJet > const &	puppiParticles () const

std::vector< double > const &	puppiWeights ()

std::vector < fastjet::PseudoJet > const &	pvParticles () const

	~PuppiContainer ()

Protected Member Functions
double	getChi2FromdZ (double iDZ)

int	getPuppiId (float iPt, float iEta)

void	getRMSAvg (int iOpt, std::vector< fastjet::PseudoJet > const &iConstits, std::vector< fastjet::PseudoJet > const &iParticles, std::vector< fastjet::PseudoJet > const &iChargeParticles)

double	goodVar (fastjet::PseudoJet const &iPart, std::vector< fastjet::PseudoJet > const &iParts, int iOpt, double iRCone)

double	var_within_R (int iId, const std::vector< fastjet::PseudoJet > &particles, const fastjet::PseudoJet &centre, double R)

Protected Attributes
bool	fApplyCHS

std::vector< fastjet::PseudoJet >	fChargedPV

int	fNAlgos

double	fNeutralMinPt

double	fNeutralSlope

int	fNPV

std::vector< fastjet::PseudoJet >	fPFParticles

std::vector< fastjet::PseudoJet >	fPupParticles

std::vector< PuppiAlgo >	fPuppiAlgo

double	fPuppiWeightCut

double	fPVFrac

std::vector< RecoObj >	fRecoParticles

bool	fUseExp

std::vector< double >	fVals

std::vector< double >	fWeights

Detailed Description

Definition at line 13 of file PuppiContainer.h.

Constructor & Destructor Documentation

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

Definition at line 14 of file PuppiContainer.cc.

References edm::ParameterSet::getParameter().

                                                              {
   fApplyCHS        = iConfig.getParameter<bool>("applyCHS"); 
   fUseExp          = iConfig.getParameter<bool>("useExp");
   fPuppiWeightCut  = iConfig.getParameter<double>("MinPuppiWeight");
   std::vector<edm::ParameterSet> lAlgos = iConfig.getParameter<std::vector<edm::ParameterSet> >("algos"); 
   fNAlgos = lAlgos.size();
   for(unsigned int i0 = 0; i0 < lAlgos.size(); i0++) { 
     PuppiAlgo pPuppiConfig(lAlgos[i0]);
     fPuppiAlgo.push_back(pPuppiConfig);
   }
 }

PuppiContainer::~PuppiContainer ( )

Definition at line 60 of file PuppiContainer.cc.

60 {}

Member Function Documentation

double PuppiContainer::getChi2FromdZ ( double iDZ )

protected

Definition at line 131 of file PuppiContainer.cc.

References funct::abs(), and alignCSCRings::e.

                                                { 
   //We need to obtain prob of PU + (1-Prob of LV)
   // Prob(LV) = Gaus(dZ,sigma) where sigma = 1.5mm  (its really more like 1mm)
   //double lProbLV = ROOT::Math::normal_cdf_c(std::abs(iDZ),0.2)*2.; //*2 is to do it double sided
   //Take iDZ to be corrected by sigma already
   double lProbLV = ROOT::Math::normal_cdf_c(std::abs(iDZ),1.)*2.; //*2 is to do it double sided
   double lProbPU = 1-lProbLV;
   if(lProbPU <= 0) lProbPU = 1e-16;   //Quick Trick to through out infs
   if(lProbPU >= 0) lProbPU = 1-1e-16; //Ditto
   double lChi2PU = TMath::ChisquareQuantile(lProbPU,1);
   lChi2PU*=lChi2PU;
   return lChi2PU;
 }

int PuppiContainer::getPuppiId	(	float	iPt,
		float	iEta
	)

protected

Definition at line 119 of file PuppiContainer.cc.

References funct::abs(), and PtMinSelector_cfg::ptMin.

                                                         { 
   int lId = -1; 
   for(int i0 = 0; i0 < fNAlgos; i0++) { 
     if(std::abs(iEta) < fPuppiAlgo[i0].etaMin()) continue;
     if(std::abs(iEta) > fPuppiAlgo[i0].etaMax()) continue;
     if(iPt        < fPuppiAlgo[i0].ptMin())  continue;
     lId = i0; 
     break;
   }
   //if(lId == -1) std::cerr << "Error : Full fiducial range is not defined " << std::endl;
   return lId;
 }

void PuppiContainer::getRMSAvg	(	int	iOpt,
		std::vector< fastjet::PseudoJet > const &	iConstits,
		std::vector< fastjet::PseudoJet > const &	iParticles,
		std::vector< fastjet::PseudoJet > const &	iChargeParticles
	)

protected

Definition at line 95 of file PuppiContainer.cc.

References eta(), edm::isFinite(), LogDebug, and EnergyCorrector::pt.

                                                                                                                                                                                        { 
   for(unsigned int i0 = 0; i0 < iConstits.size(); i0++ ) { 
     double pVal = -1;
     //Calculate the Puppi Algo to use
     int  pPupId   = getPuppiId(iConstits[i0].pt(),iConstits[i0].eta());
     if(fPuppiAlgo[pPupId].numAlgos() <= iOpt) pPupId = -1;
     if(pPupId == -1) {fVals.push_back(-1); continue;}
     //Get the Puppi Sub Algo (given iteration)
     int  pAlgo    = fPuppiAlgo[pPupId].algoId   (iOpt); 
     bool pCharged = fPuppiAlgo[pPupId].isCharged(iOpt);
     double pCone  = fPuppiAlgo[pPupId].coneSize (iOpt);
     //Compute the Puppi Metric 
     if(!pCharged) pVal = goodVar(iConstits[i0],iParticles       ,pAlgo,pCone);
     if( pCharged) pVal = goodVar(iConstits[i0],iChargedParticles,pAlgo,pCone);
     fVals.push_back(pVal);
     //if(std::isnan(pVal) || std::isinf(pVal)) cerr << "====> Value is Nan " << pVal << " == " << iConstits[i0].pt() << " -- " << iConstits[i0].eta() << endl;
     if( ! edm::isFinite(pVal)) {
       LogDebug( "NotFound" )  << "====> Value is Nan " << pVal << " == " << iConstits[i0].pt() << " -- " << iConstits[i0].eta() << endl;
       continue;
     }
     fPuppiAlgo[pPupId].add(iConstits[i0],pVal,iOpt);
   }
   for(int i0 = 0; i0 < fNAlgos; i0++) fPuppiAlgo[i0].computeMedRMS(iOpt,fPVFrac);
 }

double PuppiContainer::goodVar	(	fastjet::PseudoJet const &	iPart,
		std::vector< fastjet::PseudoJet > const &	iParts,
		int	iOpt,
		double	iRCone
	)

protected

Definition at line 62 of file PuppiContainer.cc.

                                                                                                                 {
   double lPup = 0;
   lPup = var_within_R(iOpt,iParts,iPart,iRCone);
   return lPup;
 }

void PuppiContainer::initialize ( const std::vector< RecoObj > & iRecoObjects )

Definition at line 26 of file PuppiContainer.cc.

References funct::abs(), i, and or.

                                                                       { 
     //Clear everything
   fRecoParticles.resize(0);
   fPFParticles  .resize(0);
   fChargedPV    .resize(0);
   fPupParticles .resize(0);
   fWeights      .resize(0);
   fVals.resize(0);
     //fChargedNoPV.resize(0);
     //Link to the RecoObjects
   fPVFrac = 0.; 
   fNPV    = 1.;
   fRecoParticles = iRecoObjects;
   for (unsigned int i = 0; i < fRecoParticles.size(); i++){
     fastjet::PseudoJet curPseudoJet;
     auto fRecoParticle = fRecoParticles[i];
     curPseudoJet.reset_PtYPhiM(fRecoParticle.pt,fRecoParticle.eta,fRecoParticle.phi,fRecoParticle.m);
     int puppi_register = 0;
     if(fRecoParticle.id == 0 or fRecoParticle.charge == 0)  puppi_register = 0; // zero is neutral hadron                                                         
     if(fRecoParticle.id == 1 and fRecoParticle.charge != 0) puppi_register = fRecoParticle.charge; // from PV use the                             
     if(fRecoParticle.id == 2 and fRecoParticle.charge != 0) puppi_register = fRecoParticle.charge+5; // from NPV use the charge as key +5 as key
     curPseudoJet.set_user_info( new PuppiUserInfo( puppi_register ) );
     // fill vector of pseudojets for internal references
     fPFParticles.push_back(curPseudoJet);
     //Take Charged particles associated to PV
     if(std::abs(fRecoParticle.id) == 1) fChargedPV.push_back(curPseudoJet);
     if(std::abs(fRecoParticle.id) >= 1 ) fPVFrac+=1.;
     //if((fRecoParticle.id == 0) && (inParticles[i].id == 2))  _genParticles.push_back( curPseudoJet);
     //if(fRecoParticle.id <= 2 && !(inParticles[i].pt < fNeutralMinE && fRecoParticle.id < 2)) _pfchsParticles.push_back(curPseudoJet); 
     //if(fRecoParticle.id == 3) _chargedNoPV.push_back(curPseudoJet);
     if(fNPV < fRecoParticle.vtxId) fNPV = fRecoParticle.vtxId;
   }
   fPVFrac = double(fChargedPV.size())/fPVFrac;
 }

std::vector<fastjet::PseudoJet> const& PuppiContainer::pfParticles ( ) const

inline

Definition at line 40 of file PuppiContainer.h.

References fPFParticles.

40 { return fPFParticles; }

PuppiContainer::fPFParticles

std::vector< fastjet::PseudoJet > fPFParticles

Definition: PuppiContainer.h:53

std::vector<fastjet::PseudoJet> const& PuppiContainer::puppiParticles ( ) const

inline

Definition at line 43 of file PuppiContainer.h.

References fPupParticles.

43 { return fPupParticles;}

PuppiContainer::fPupParticles

std::vector< fastjet::PseudoJet > fPupParticles

Definition: PuppiContainer.h:55

std::vector< double > const & PuppiContainer::puppiWeights ( )

Definition at line 144 of file PuppiContainer.cc.

References funct::abs(), alignCSCRings::e, eta(), edm::isFinite(), LogDebug, bookConverter::max, EnergyCorrector::pt, and reset().

                                                        {
   fPupParticles .resize(0);
   fWeights      .resize(0);
   fVals         .resize(0);
   for(int i0 = 0; i0 < fNAlgos; i0++) fPuppiAlgo[i0].reset();
 
     int lNMaxAlgo = 1;
     for(int i0 = 0; i0 < fNAlgos; i0++) lNMaxAlgo = std::max(fPuppiAlgo[i0].numAlgos(),lNMaxAlgo);
     //Run through all compute mean and RMS
     int lNParticles    = fRecoParticles.size();
   for(int i0 = 0; i0 < lNMaxAlgo; i0++) { 
     getRMSAvg(i0,fPFParticles,fPFParticles,fChargedPV);
   }
   std::vector<double> pVals;
   for(int i0 = 0; i0 < lNParticles; i0++) {
     //Refresh
     pVals.clear();
     double pWeight = 1;
     //Get the Puppi Id and if ill defined move on
     int  pPupId   = getPuppiId(fRecoParticles[i0].pt,fRecoParticles[i0].eta);
     if(pPupId == -1) {
      fWeights .push_back(pWeight);
      continue;
    }
       // fill the p-values
    double pChi2   = 0;
    if(fUseExp){ 
     //Compute an Experimental Puppi Weight with delta Z info (very simple example)
      pChi2 = getChi2FromdZ(fRecoParticles[i0].dZ);
     //Now make sure Neutrals are not set
      if(fRecoParticles[i0].pfType > 3) pChi2 = 0;
    }
       //Fill and compute the PuppiWeight
    int lNAlgos = fPuppiAlgo[pPupId].numAlgos();
    for(int i1 = 0; i1 < lNAlgos; i1++) pVals.push_back(fVals[lNParticles*i1+i0]);
     pWeight = fPuppiAlgo[pPupId].compute(pVals,pChi2);
       //Apply the CHS weights
     if(fRecoParticles[i0].id == 1 && fApplyCHS ) pWeight = 1;
     if(fRecoParticles[i0].id == 2 && fApplyCHS ) pWeight = 0;
       //Basic Weight Checks
     if( ! edm::isFinite(pWeight)) {
       pWeight = 0.0;
       LogDebug("PuppiWeightError") << "====> Weight is nan : " << pWeight << " : pt " << fRecoParticles[i0].pt << " -- eta : " << fRecoParticles[i0].eta << " -- Value" << fVals[i0] << " -- id :  " << fRecoParticles[i0].id << " --  NAlgos: " << lNAlgos << std::endl;
     }
     //Basic Cuts      
     if(pWeight                         < fPuppiWeightCut) pWeight = 0;  //==> Elminate the low Weight stuff
     if(pWeight*fPFParticles[i0].pt()   < fPuppiAlgo[pPupId].neutralPt(fNPV) && fRecoParticles[i0].id == 0 ) pWeight = 0;  //threshold cut on the neutral Pt
     fWeights .push_back(pWeight);
     //Now get rid of the thrown out weights for the particle collection
     if(std::abs(pWeight) < std::numeric_limits<double>::denorm_min() ) continue;
       //Produce
     PseudoJet curjet( pWeight*fPFParticles[i0].px(), pWeight*fPFParticles[i0].py(), pWeight*fPFParticles[i0].pz(), pWeight*fPFParticles[i0].e());
     curjet.set_user_index(i0);
     fPupParticles.push_back(curjet);
   }
   return fWeights;
 }

std::vector<fastjet::PseudoJet> const& PuppiContainer::pvParticles ( ) const

inline

Definition at line 41 of file PuppiContainer.h.

References fChargedPV.

41 { return fChargedPV; }

PuppiContainer::fChargedPV

std::vector< fastjet::PseudoJet > fChargedPV

Definition: PuppiContainer.h:54

double PuppiContainer::var_within_R	(	int	iId,
		const std::vector< fastjet::PseudoJet > &	particles,
		const fastjet::PseudoJet &	centre,
		double	R
	)

protected

Definition at line 67 of file PuppiContainer.cc.

References funct::abs(), i, fff_deleter::log, M_PI, phi, EgammaValidation_Wenu_cff::sel, and MetTreeProducer::var().

                                                                                                                   {
   if(iId == -1) return 1;
   fastjet::Selector sel = fastjet::SelectorCircle(R);
   sel.set_reference(centre);
   vector<PseudoJet> near_particles = sel(particles);
   double var = 0;
   //double lSumPt = 0;
   //if(iId == 1) for(unsigned int i=0; i<near_particles.size(); i++) lSumPt += near_particles[i].pt();
   for(unsigned int i=0; i<near_particles.size(); i++){
     double pDEta = near_particles[i].eta()-centre.eta();
     double pDPhi = std::abs(near_particles[i].phi()-centre.phi());
     if(pDPhi > 2.*M_PI-pDPhi) pDPhi =  2.*M_PI-pDPhi;
     double pDR2 = pDEta*pDEta+pDPhi*pDPhi;
     if(std::abs(pDR2)  <  0.0001) continue;
     if(iId == 0) var += (near_particles[i].pt()/pDR2);
     if(iId == 1) var += near_particles[i].pt();
     if(iId == 2) var += (1./pDR2);
     if(iId == 3) var += (1./pDR2);
     if(iId == 4) var += near_particles[i].pt();  
     if(iId == 5) var += (near_particles[i].pt() * near_particles[i].pt()/pDR2);
   }
   if(iId == 1) var += centre.pt(); //Sum in a cone
   if(iId == 0 && var != 0) var = log(var);
   if(iId == 3 && var != 0) var = log(var);
   if(iId == 5 && var != 0) var = log(var);
   return var;
 }

Member Data Documentation

bool PuppiContainer::fApplyCHS

protected

Definition at line 58 of file PuppiContainer.h.

std::vector<fastjet::PseudoJet> PuppiContainer::fChargedPV

protected

Definition at line 54 of file PuppiContainer.h.

Referenced by pvParticles().

int PuppiContainer::fNAlgos

protected

Definition at line 63 of file PuppiContainer.h.

double PuppiContainer::fNeutralMinPt

protected

Definition at line 60 of file PuppiContainer.h.

double PuppiContainer::fNeutralSlope

protected

Definition at line 61 of file PuppiContainer.h.

int PuppiContainer::fNPV

protected

Definition at line 64 of file PuppiContainer.h.

std::vector<fastjet::PseudoJet> PuppiContainer::fPFParticles

protected

Definition at line 53 of file PuppiContainer.h.

Referenced by pfParticles().

std::vector<fastjet::PseudoJet> PuppiContainer::fPupParticles

protected

Definition at line 55 of file PuppiContainer.h.

Referenced by puppiParticles().

std::vector<PuppiAlgo> PuppiContainer::fPuppiAlgo

protected

Definition at line 66 of file PuppiContainer.h.

double PuppiContainer::fPuppiWeightCut

protected

Definition at line 62 of file PuppiContainer.h.

double PuppiContainer::fPVFrac

protected

Definition at line 65 of file PuppiContainer.h.

std::vector<RecoObj> PuppiContainer::fRecoParticles

protected

Definition at line 52 of file PuppiContainer.h.

bool PuppiContainer::fUseExp

protected

Definition at line 59 of file PuppiContainer.h.

std::vector<double> PuppiContainer::fVals

protected

Definition at line 57 of file PuppiContainer.h.

std::vector<double> PuppiContainer::fWeights

protected

Definition at line 56 of file PuppiContainer.h.

Classes

Public Member Functions

Protected Member Functions

Protected Attributes

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation