#include <PuppiProducer.h>

Inheritance diagram for PuppiProducer:

Public Types
typedef edm::View< reco::Candidate >	CandidateView

typedef math::XYZTLorentzVector	LorentzVector

typedef std::vector< LorentzVector >	LorentzVectorCollection

typedef std::vector< pat::PackedCandidate >	PackedOutputCollection

typedef std::vector< reco::PFCandidate >	PFInputCollection

typedef std::vector< reco::PFCandidate >	PFOutputCollection

typedef edm::View< reco::PFCandidate >	PFView

typedef reco::VertexCollection	VertexCollection

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

Public Member Functions
	PuppiProducer (const edm::ParameterSet &)

	~PuppiProducer () 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
virtual void	beginJob ()

virtual void	endJob ()

void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
bool	fClonePackedCands

float	fDZCut

std::unique_ptr< PackedOutputCollection >	fPackedPuppiCandidates

std::string	fPFName

float	fPtMax

std::unique_ptr< PFOutputCollection >	fPuppiCandidates

std::unique_ptr< PuppiContainer >	fPuppiContainer

bool	fPuppiDiagnostics

bool	fPuppiForLeptons

std::string	fPuppiName

std::string	fPVName

std::vector< RecoObj >	fRecoObjCollection

bool	fUseDZ

bool	fUseExistingWeights

bool	fUseWeightsNoLep

int	fVtxNdofCut

double	fVtxZCut

edm::EDGetTokenT< CandidateView >	tokenPFCandidates_

edm::EDGetTokenT< VertexCollection >	tokenVertices_

Detailed Description

Definition at line 22 of file PuppiProducer.h.

Member Typedef Documentation

typedef edm::View<reco::Candidate> PuppiProducer::CandidateView

Definition at line 32 of file PuppiProducer.h.

typedef math::XYZTLorentzVector PuppiProducer::LorentzVector

Definition at line 29 of file PuppiProducer.h.

typedef std::vector<LorentzVector> PuppiProducer::LorentzVectorCollection

Definition at line 30 of file PuppiProducer.h.

typedef std::vector< pat::PackedCandidate > PuppiProducer::PackedOutputCollection

Definition at line 35 of file PuppiProducer.h.

typedef std::vector< reco::PFCandidate > PuppiProducer::PFInputCollection

Definition at line 33 of file PuppiProducer.h.

typedef std::vector< reco::PFCandidate > PuppiProducer::PFOutputCollection

Definition at line 34 of file PuppiProducer.h.

typedef edm::View<reco::PFCandidate> PuppiProducer::PFView

Definition at line 36 of file PuppiProducer.h.

typedef reco::VertexCollection PuppiProducer::VertexCollection

Definition at line 31 of file PuppiProducer.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 25 of file PuppiProducer.cc.

References fClonePackedCands, fDZCut, fPtMax, fPuppiContainer, fPuppiDiagnostics, fPuppiForLeptons, fUseDZ, fUseExistingWeights, fUseWeightsNoLep, fVtxNdofCut, fVtxZCut, edm::ParameterSet::getParameter(), tokenPFCandidates_, and tokenVertices_.

                                                            {
   fPuppiDiagnostics = iConfig.getParameter<bool>("puppiDiagnostics");
   fPuppiForLeptons = iConfig.getParameter<bool>("puppiForLeptons");
   fUseDZ     = iConfig.getParameter<bool>("UseDeltaZCut");
   fDZCut     = iConfig.getParameter<double>("DeltaZCut");
   fPtMax     = iConfig.getParameter<double>("PtMaxNeutrals");
   fUseExistingWeights     = iConfig.getParameter<bool>("useExistingWeights");
   fUseWeightsNoLep        = iConfig.getParameter<bool>("useWeightsNoLep");
   fClonePackedCands       = iConfig.getParameter<bool>("clonePackedCands");
   fVtxNdofCut = iConfig.getParameter<int>("vtxNdofCut");
   fVtxZCut = iConfig.getParameter<double>("vtxZCut");
   fPuppiContainer = std::unique_ptr<PuppiContainer> ( new PuppiContainer(iConfig) );
 
   tokenPFCandidates_
     = consumes<CandidateView>(iConfig.getParameter<edm::InputTag>("candName"));
   tokenVertices_
     = consumes<VertexCollection>(iConfig.getParameter<edm::InputTag>("vertexName"));
  
 
   produces<edm::ValueMap<float> > ();
   produces<edm::ValueMap<LorentzVector> > ();
   produces< edm::ValueMap<reco::CandidatePtr> >(); 
   
   if (fUseExistingWeights || fClonePackedCands)
     produces<pat::PackedCandidateCollection>();
   else
     produces<reco::PFCandidateCollection>();
 
   if (fPuppiDiagnostics){
     produces<double> ("PuppiNAlgos");
     produces<std::vector<double>> ("PuppiRawAlphas");
     produces<std::vector<double>> ("PuppiAlphas");
     produces<std::vector<double>> ("PuppiAlphasMed");
     produces<std::vector<double>> ("PuppiAlphasRms");
   }
 }

PuppiProducer::~PuppiProducer ( )

override

Definition at line 62 of file PuppiProducer.cc.

62 {

63 }

Member Function Documentation

void PuppiProducer::beginJob ( void )

privatevirtual

Definition at line 325 of file PuppiProducer.cc.

325 {

326 }

void PuppiProducer::endJob ( void )

privatevirtual

Definition at line 328 of file PuppiProducer.cc.

Referenced by o2olib.O2ORunMgr::executeJob().

328 {

329 }

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

static

Definition at line 331 of file PuppiProducer.cc.

References edm::ConfigurationDescriptions::addDefault(), DEFINE_FWK_MODULE, and edm::ParameterSetDescription::setUnknown().

                                                                                {
   //The following says we do not know what parameters are allowed so do no validation
   // Please change this to state exactly what you do use, even if it is no parameters
   edm::ParameterSetDescription desc;
   desc.setUnknown();
   descriptions.addDefault(desc);
 }

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

overrideprivate

Definition at line 65 of file PuppiProducer.cc.

Referenced by JSONExport.JsonExport::export(), HTMLExport.HTMLExport::export(), and HTMLExport.HTMLExportStatic::export().

                                                                          {
 
   // Get PFCandidate Collection
   edm::Handle<CandidateView> hPFProduct;
   iEvent.getByToken(tokenPFCandidates_,hPFProduct);
   const CandidateView *pfCol = hPFProduct.product();
 
   // Get vertex collection w/PV as the first entry?
   edm::Handle<reco::VertexCollection> hVertexProduct;
   iEvent.getByToken(tokenVertices_,hVertexProduct);
   const reco::VertexCollection *pvCol = hVertexProduct.product();
 
    int npv = 0;
    const reco::VertexCollection::const_iterator vtxEnd = pvCol->end();
    for (reco::VertexCollection::const_iterator vtxIter = pvCol->begin(); vtxEnd != vtxIter; ++vtxIter) {
       if (!vtxIter->isFake() && vtxIter->ndof()>=fVtxNdofCut && std::abs(vtxIter->z())<=fVtxZCut)
          npv++;
    }
 
   //Fill the reco objects
   fRecoObjCollection.clear();
   for(CandidateView::const_iterator itPF = pfCol->begin(); itPF!=pfCol->end(); itPF++) {
     // std::cout << "itPF->pdgId() = " << itPF->pdgId() << std::endl;
     RecoObj pReco;
     pReco.pt  = itPF->pt();
     pReco.eta = itPF->eta();
     pReco.phi = itPF->phi();
     pReco.m   = itPF->mass();
     pReco.rapidity = itPF->rapidity();
     pReco.charge = itPF->charge(); 
     const reco::Vertex *closestVtx = nullptr;
     double pDZ    = -9999; 
     double pD0    = -9999; 
     int    pVtxId = -9999; 
     bool lFirst = true;
     const pat::PackedCandidate *lPack = dynamic_cast<const pat::PackedCandidate*>(&(*itPF));
     if(lPack == nullptr ) {
 
       const reco::PFCandidate *pPF = dynamic_cast<const reco::PFCandidate*>(&(*itPF));
       double curdz = 9999;
       int closestVtxForUnassociateds = -9999;
       for(reco::VertexCollection::const_iterator iV = pvCol->begin(); iV!=pvCol->end(); ++iV) {
         if(lFirst) {
           if      ( pPF->trackRef().isNonnull()    ) pDZ = pPF->trackRef()   ->dz(iV->position());
           else if ( pPF->gsfTrackRef().isNonnull() ) pDZ = pPF->gsfTrackRef()->dz(iV->position());
           if      ( pPF->trackRef().isNonnull()    ) pD0 = pPF->trackRef()   ->d0();
           else if ( pPF->gsfTrackRef().isNonnull() ) pD0 = pPF->gsfTrackRef()->d0();
           lFirst = false;
           if(pDZ > -9999) pVtxId = 0;
         }
         if(iV->trackWeight(pPF->trackRef())>0) {
             closestVtx  = &(*iV);
             break;
           }        
         // in case it's unassocciated, keep more info
         double tmpdz = 99999;
         if      ( pPF->trackRef().isNonnull()    ) tmpdz = pPF->trackRef()   ->dz(iV->position());
         else if ( pPF->gsfTrackRef().isNonnull() ) tmpdz = pPF->gsfTrackRef()->dz(iV->position());
         if (std::abs(tmpdz) < curdz){
           curdz = std::abs(tmpdz);
           closestVtxForUnassociateds = pVtxId;
         }
         pVtxId++;
 
       }
       int tmpFromPV = 0;  
       // mocking the miniAOD definitions
       if (closestVtx != nullptr && std::abs(pReco.charge) > 0 && pVtxId > 0) tmpFromPV = 0;
       if (closestVtx != nullptr && std::abs(pReco.charge) > 0 && pVtxId == 0) tmpFromPV = 3;
       if (closestVtx == nullptr && std::abs(pReco.charge) > 0 && closestVtxForUnassociateds == 0) tmpFromPV = 2;
       if (closestVtx == nullptr && std::abs(pReco.charge) > 0 && closestVtxForUnassociateds != 0) tmpFromPV = 1;
       pReco.dZ      = pDZ;
       pReco.d0      = pD0;
       pReco.id = 0; 
       if (std::abs(pReco.charge) == 0){ pReco.id = 0; }
       else{
         if (tmpFromPV == 0){ pReco.id = 2; } // 0 is associated to PU vertex
         if (tmpFromPV == 3){ pReco.id = 1; }
         if (tmpFromPV == 1 || tmpFromPV == 2){ 
           pReco.id = 0;
           if (!fPuppiForLeptons && fUseDZ && (std::abs(pDZ) < fDZCut)) pReco.id = 1;
           if (!fPuppiForLeptons && fUseDZ && (std::abs(pDZ) > fDZCut)) pReco.id = 2;
           if (fPuppiForLeptons && tmpFromPV == 1) pReco.id = 2;
           if (fPuppiForLeptons && tmpFromPV == 2) pReco.id = 1;
         }
       }
     } 
     else if(lPack->vertexRef().isNonnull() )  {
       pDZ        = lPack->dz();
       pD0        = lPack->dxy();
       pReco.dZ      = pDZ;
       pReco.d0      = pD0;
   
       pReco.id = 0; 
       if (std::abs(pReco.charge) == 0){ pReco.id = 0; }
       if (std::abs(pReco.charge) > 0){
         if (lPack->fromPV() == 0){ pReco.id = 2; } // 0 is associated to PU vertex
         if (lPack->fromPV() == (pat::PackedCandidate::PVUsedInFit)){ pReco.id = 1; }
         if (lPack->fromPV() == (pat::PackedCandidate::PVTight) || lPack->fromPV() == (pat::PackedCandidate::PVLoose)){ 
           pReco.id = 0;
           if (!fPuppiForLeptons && fUseDZ && (std::abs(pDZ) < fDZCut)) pReco.id = 1;
           if (!fPuppiForLeptons && fUseDZ && (std::abs(pDZ) > fDZCut)) pReco.id = 2;
           if (fPuppiForLeptons && lPack->fromPV() == (pat::PackedCandidate::PVLoose)) pReco.id = 2;
           if (fPuppiForLeptons && lPack->fromPV() == (pat::PackedCandidate::PVTight)) pReco.id = 1;
         }
       }
     }
 
     fRecoObjCollection.push_back(pReco);
       
   }
 
   fPuppiContainer->initialize(fRecoObjCollection);
   fPuppiContainer->setNPV( npv );
 
   std::vector<double> lWeights;
   std::vector<PuppiCandidate> lCandidates;
   if (!fUseExistingWeights){
     //Compute the weights and get the particles
     lWeights = fPuppiContainer->puppiWeights();
     lCandidates = fPuppiContainer->puppiParticles();
   }
   else{
     //Use the existing weights
     int lPackCtr = 0;
     for(CandidateView::const_iterator itPF = pfCol->begin(); itPF!=pfCol->end(); itPF++) {  
       const pat::PackedCandidate *lPack = dynamic_cast<const pat::PackedCandidate*>(&(*itPF));
       float curpupweight = -1.;
       if(lPack == nullptr ) { 
         // throw error
         throw edm::Exception(edm::errors::LogicError,"PuppiProducer: cannot get weights since inputs are not PackedCandidates");
       }
       else{
         if (fUseWeightsNoLep){ curpupweight = lPack->puppiWeightNoLep(); }
         else{ curpupweight = lPack->puppiWeight();  }
       }
       lWeights.push_back(curpupweight);
       PuppiCandidate curjet( curpupweight*lPack->px(), curpupweight*lPack->py(), curpupweight*lPack->pz(), curpupweight*lPack->energy());
       curjet.set_user_index(lPackCtr);
       lCandidates.push_back(curjet);
       lPackCtr++;
     }
   }
 
   //Fill it into the event
   std::unique_ptr<edm::ValueMap<float> > lPupOut(new edm::ValueMap<float>());
   edm::ValueMap<float>::Filler  lPupFiller(*lPupOut);
   lPupFiller.insert(hPFProduct,lWeights.begin(),lWeights.end());
   lPupFiller.fill();
 
   // This is a dummy to access the "translate" method which is a
   // non-static member function even though it doesn't need to be. 
   // Will fix in the future. 
   static const reco::PFCandidate dummySinceTranslateIsNotStatic;
 
   // Fill a new PF/Packed Candidate Collection and write out the ValueMap of the new p4s.
   // Since the size of the ValueMap must be equal to the input collection, we need
   // to search the "puppi" particles to find a match for each input. If none is found,
   // the input is set to have a four-vector of 0,0,0,0
   fPuppiCandidates.reset( new PFOutputCollection );
   fPackedPuppiCandidates.reset( new PackedOutputCollection );
   std::unique_ptr<edm::ValueMap<LorentzVector> > p4PupOut(new edm::ValueMap<LorentzVector>());
   LorentzVectorCollection puppiP4s;
   std::vector<reco::CandidatePtr> values(hPFProduct->size());
 
   for ( auto i0 = hPFProduct->begin(),
       i0begin = hPFProduct->begin(),
       i0end = hPFProduct->end(); i0 != i0end; ++i0 ) {
     std::unique_ptr<pat::PackedCandidate> pCand;
     std::unique_ptr<reco::PFCandidate>    pfCand;
     if (fUseExistingWeights || fClonePackedCands) {
       const pat::PackedCandidate *cand = dynamic_cast<const pat::PackedCandidate*>(&(*i0));
       if(!cand)
         throw edm::Exception(edm::errors::LogicError,"PuppiProducer: inputs are not PackedCandidates");
       pCand.reset( new pat::PackedCandidate(*cand) );
     } else {
       auto id = dummySinceTranslateIsNotStatic.translatePdgIdToType(i0->pdgId());
       const reco::PFCandidate *cand = dynamic_cast<const reco::PFCandidate*>(&(*i0));
       pfCand.reset( new reco::PFCandidate( cand ? *cand : reco::PFCandidate(i0->charge(), i0->p4(), id) ) );
     }
     LorentzVector pVec = i0->p4();
     int val = i0 - i0begin;
 
     // Find the Puppi particle matched to the input collection using the "user_index" of the object. 
     auto puppiMatched = find_if( lCandidates.begin(), lCandidates.end(), [&val]( PuppiCandidate const & i ){ return i.user_index() == val; } );
     if ( puppiMatched != lCandidates.end() ) {
       pVec.SetPxPyPzE(puppiMatched->px(),puppiMatched->py(),puppiMatched->pz(),puppiMatched->E());
       if(fClonePackedCands && (!fUseExistingWeights)) {
         if(fPuppiForLeptons)
           pCand->setPuppiWeight(pCand->puppiWeight(),lWeights[puppiMatched-lCandidates.begin()]);
         else
           pCand->setPuppiWeight(lWeights[puppiMatched-lCandidates.begin()],pCand->puppiWeightNoLep());
       }
     } else {
       pVec.SetPxPyPzE( 0, 0, 0, 0);
       if(fClonePackedCands && (!fUseExistingWeights)) {
         pCand->setPuppiWeight(0,0);
       }
     }
     puppiP4s.push_back( pVec );
 
     if (fUseExistingWeights || fClonePackedCands) {
       pCand->setP4(pVec);
       pCand->setSourceCandidatePtr( i0->sourceCandidatePtr(0) );
       fPackedPuppiCandidates->push_back(*pCand);
     } else {
       pfCand->setP4(pVec);
       pfCand->setSourceCandidatePtr( i0->sourceCandidatePtr(0) );
       fPuppiCandidates->push_back(*pfCand);
     }
   }
 
   //Compute the modified p4s
   edm::ValueMap<LorentzVector>::Filler  p4PupFiller(*p4PupOut);
   p4PupFiller.insert(hPFProduct,puppiP4s.begin(), puppiP4s.end() );
   p4PupFiller.fill();
   
   iEvent.put(std::move(lPupOut));
   iEvent.put(std::move(p4PupOut));
   if (fUseExistingWeights || fClonePackedCands) {
     edm::OrphanHandle<pat::PackedCandidateCollection> oh = iEvent.put(std::move(fPackedPuppiCandidates));
     for(unsigned int ic=0, nc = oh->size(); ic < nc; ++ic) {
         reco::CandidatePtr pkref( oh, ic );
         values[ic] = pkref;
     }
   } else {
     edm::OrphanHandle<reco::PFCandidateCollection> oh = iEvent.put(std::move(fPuppiCandidates));
     for(unsigned int ic=0, nc = oh->size(); ic < nc; ++ic) {
         reco::CandidatePtr pkref( oh, ic );
         values[ic] = pkref;
     }
   }
   std::unique_ptr<edm::ValueMap<reco::CandidatePtr> > pfMap_p(new edm::ValueMap<reco::CandidatePtr>());
   edm::ValueMap<reco::CandidatePtr>::Filler filler(*pfMap_p);
   filler.insert(hPFProduct, values.begin(), values.end());
   filler.fill();
   iEvent.put(std::move(pfMap_p));
 
 
   if (fPuppiDiagnostics && !fUseExistingWeights){
 
     // all the different alphas per particle
     // THE alpha per particle
     std::unique_ptr<std::vector<double> > theAlphas(new std::vector<double>(fPuppiContainer->puppiAlphas()));
     std::unique_ptr<std::vector<double> > theAlphasMed(new std::vector<double>(fPuppiContainer->puppiAlphasMed()));
     std::unique_ptr<std::vector<double> > theAlphasRms(new std::vector<double>(fPuppiContainer->puppiAlphasRMS()));
     std::unique_ptr<std::vector<double> > alphas(new std::vector<double>(fPuppiContainer->puppiRawAlphas()));
     std::unique_ptr<double> nalgos(new double(fPuppiContainer->puppiNAlgos()));
     
     iEvent.put(std::move(alphas),"PuppiRawAlphas");
     iEvent.put(std::move(nalgos),"PuppiNAlgos");
     iEvent.put(std::move(theAlphas),"PuppiAlphas");
     iEvent.put(std::move(theAlphasMed),"PuppiAlphasMed");
     iEvent.put(std::move(theAlphasRms),"PuppiAlphasRms");
   }
   
 }