CMS 3D CMS Logo

List of all members | Public Member Functions | Private Member Functions | Private Attributes
pat::PATPackedCandidateProducer Class Reference
Inheritance diagram for pat::PATPackedCandidateProducer:
edm::global::EDProducer<> edm::global::EDProducerBase edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

bool candsOrdering (pat::PackedCandidate i, pat::PackedCandidate j) const
 
 PATPackedCandidateProducer (const edm::ParameterSet &)
 
virtual void produce (edm::StreamID, edm::Event &, const edm::EventSetup &) const override
 
template<typename T >
std::vector< size_t > sort_indexes (const std::vector< T > &v) const
 
 ~PATPackedCandidateProducer ()
 
- Public Member Functions inherited from edm::global::EDProducer<>
 EDProducer ()=default
 
- Public Member Functions inherited from edm::global::EDProducerBase
 EDProducerBase ()
 
ModuleDescription const & moduleDescription () const
 
virtual ~EDProducerBase ()
 
- Public Member Functions inherited from edm::ProducerBase
void callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
 
std::vector< edm::ProductResolverIndex > const & indiciesForPutProducts (BranchType iBranchType) const
 
 ProducerBase ()
 
void registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
 
std::function< void(BranchDescription const &)> registrationCallback () const
 used by the fwk to register list of products More...
 
void resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)
 
virtual ~ProducerBase () noexcept(false)
 
- Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfoconsumesInfo () const
 
 EDConsumerBase ()
 
 EDConsumerBase (EDConsumerBase const &)=delete
 
 EDConsumerBase (EDConsumerBase &&)=default
 
ProductResolverIndexAndSkipBit indexFrom (EDGetToken, BranchType, TypeID const &) const
 
void itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
void itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
std::vector< ProductResolverIndexAndSkipBit > const & itemsToGetFromEvent () const
 
void labelsForToken (EDGetToken iToken, Labels &oLabels) const
 
void modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
 
EDConsumerBase const & operator= (EDConsumerBase const &)=delete
 
EDConsumerBaseoperator= (EDConsumerBase &&)=default
 
bool registeredToConsume (ProductResolverIndex, bool, BranchType) const
 
bool registeredToConsumeMany (TypeID const &, BranchType) const
 
void updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
 
virtual ~EDConsumerBase () noexcept(false)
 

Private Member Functions

float calcDxy (float dx, float dy, float phi) const
 
float calcDz (reco::Candidate::Point p, reco::Candidate::Point v, const reco::Candidate &c) const
 

Private Attributes

const edm::EDGetTokenT< reco::PFCandidateCollectionCands_
 
const std::vector< int > covariancePackingSchemas_
 
const int covarianceVersion_
 
const double minPtForTrackProperties_
 
const edm::EDGetTokenT< std::vector< reco::PFCandidate > > PuppiCands_
 
const edm::EDGetTokenT< edm::ValueMap< reco::CandidatePtr > > PuppiCandsMap_
 
const edm::EDGetTokenT< std::vector< reco::PFCandidate > > PuppiCandsNoLep_
 
const edm::EDGetTokenT< edm::ValueMap< float > > PuppiWeight_
 
const edm::EDGetTokenT< edm::ValueMap< float > > PuppiWeightNoLep_
 
const edm::EDGetTokenT< edm::Association< reco::VertexCollection > > PVAsso_
 
const edm::EDGetTokenT< edm::ValueMap< int > > PVAssoQuality_
 
const edm::EDGetTokenT< reco::VertexCollectionPVOrigs_
 
const edm::EDGetTokenT< reco::VertexCollectionPVs_
 
std::vector< edm::EDGetTokenT< edm::View< reco::Candidate > > > SVWhiteLists_
 
const edm::EDGetTokenT< reco::TrackCollectionTKOrigs_
 
const bool usePuppi_
 

Additional Inherited Members

- Public Types inherited from edm::global::EDProducerBase
typedef EDProducerBase ModuleType
 
- Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList TypeLabelList
 
- Public Types inherited from edm::EDConsumerBase
typedef ProductLabels Labels
 
- Static Public Member Functions inherited from edm::global::EDProducerBase
static const std::string & baseType ()
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &descriptions)
 
- Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > consumes (edm::InputTag const &tag)
 
EDGetToken consumes (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken consumes (TypeToGet const &id, edm::InputTag const &tag)
 
ConsumesCollector consumesCollector ()
 Use a ConsumesCollector to gather consumes information from helper functions. More...
 
template<typename ProductType , BranchType B = InEvent>
void consumesMany ()
 
void consumesMany (const TypeToGet &id)
 
template<BranchType B>
void consumesMany (const TypeToGet &id)
 
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > mayConsume (edm::InputTag const &tag)
 
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 

Detailed Description

Definition at line 36 of file PATPackedCandidateProducer.cc.

Constructor & Destructor Documentation

pat::PATPackedCandidateProducer::PATPackedCandidateProducer ( const edm::ParameterSet iConfig)
explicit

Definition at line 100 of file PATPackedCandidateProducer.cc.

References edm::EDConsumerBase::consumes(), edm::ParameterSet::getParameter(), and SVWhiteLists_.

100  :
101  usePuppi_(!iConfig.getParameter<edm::InputTag>("PuppiSrc").encode().empty() ||
102  !iConfig.getParameter<edm::InputTag>("PuppiNoLepSrc").encode().empty()),
103  Cands_(consumes<reco::PFCandidateCollection>(iConfig.getParameter<edm::InputTag>("inputCollection"))),
104  PVs_(consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("inputVertices"))),
106  PVAssoQuality_(consumes<edm::ValueMap<int> >(iConfig.getParameter<edm::InputTag>("vertexAssociator"))),
107  PVOrigs_(consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("originalVertices"))),
108  TKOrigs_(consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("originalTracks"))),
109  PuppiWeight_(usePuppi_ ? consumes<edm::ValueMap<float> >(iConfig.getParameter<edm::InputTag>("PuppiSrc")) : edm::EDGetTokenT< edm::ValueMap<float> >()),
110  PuppiWeightNoLep_(usePuppi_ ? consumes<edm::ValueMap<float> >(iConfig.getParameter<edm::InputTag>("PuppiNoLepSrc")) : edm::EDGetTokenT< edm::ValueMap<float> >()),
111  PuppiCandsMap_(usePuppi_ ? consumes<edm::ValueMap<reco::CandidatePtr> >(iConfig.getParameter<edm::InputTag>("PuppiSrc")) : edm::EDGetTokenT<edm::ValueMap<reco::CandidatePtr> >() ),
112  PuppiCands_(usePuppi_ ? consumes<std::vector< reco::PFCandidate > >(iConfig.getParameter<edm::InputTag>("PuppiSrc")) : edm::EDGetTokenT<std::vector< reco::PFCandidate > >() ),
113  PuppiCandsNoLep_(usePuppi_ ? consumes<std::vector< reco::PFCandidate > >(iConfig.getParameter<edm::InputTag>("PuppiNoLepSrc")) : edm::EDGetTokenT<std::vector< reco::PFCandidate > >()),
114  minPtForTrackProperties_(iConfig.getParameter<double>("minPtForTrackProperties")),
115  covarianceVersion_(iConfig.getParameter<int >("covarianceVersion")),
116  covariancePackingSchemas_(iConfig.getParameter<std::vector<int> >("covariancePackingSchemas"))
117 
118 {
119  std::vector<edm::InputTag> sv_tags = iConfig.getParameter<std::vector<edm::InputTag> >("secondaryVerticesForWhiteList");
120  for(auto itag : sv_tags){
121  SVWhiteLists_.push_back(
123  );
124  }
125 
126  produces< std::vector<pat::PackedCandidate> > ();
127  produces< edm::Association<pat::PackedCandidateCollection> > ();
128  produces< edm::Association<reco::PFCandidateCollection> > ();
129 }
T getParameter(std::string const &) const
const edm::EDGetTokenT< edm::ValueMap< float > > PuppiWeight_
const edm::EDGetTokenT< reco::PFCandidateCollection > Cands_
std::string encode() const
Definition: InputTag.cc:165
const edm::EDGetTokenT< edm::ValueMap< int > > PVAssoQuality_
EDGetTokenT< ProductType > consumes(edm::InputTag const &tag)
const edm::EDGetTokenT< reco::TrackCollection > TKOrigs_
const edm::EDGetTokenT< std::vector< reco::PFCandidate > > PuppiCands_
const edm::EDGetTokenT< edm::ValueMap< reco::CandidatePtr > > PuppiCandsMap_
const edm::EDGetTokenT< edm::ValueMap< float > > PuppiWeightNoLep_
edm::Ptr< Candidate > CandidatePtr
persistent reference to an object in a collection of Candidate objects
Definition: CandidateFwd.h:25
const std::vector< int > covariancePackingSchemas_
const edm::EDGetTokenT< std::vector< reco::PFCandidate > > PuppiCandsNoLep_
def encode(args, files)
const edm::EDGetTokenT< reco::VertexCollection > PVOrigs_
fixed size matrix
HLT enums.
std::vector< edm::EDGetTokenT< edm::View< reco::Candidate > > > SVWhiteLists_
const edm::EDGetTokenT< edm::Association< reco::VertexCollection > > PVAsso_
const edm::EDGetTokenT< reco::VertexCollection > PVs_
pat::PATPackedCandidateProducer::~PATPackedCandidateProducer ( )

Definition at line 131 of file PATPackedCandidateProducer.cc.

131 {}

Member Function Documentation

float pat::PATPackedCandidateProducer::calcDxy ( float  dx,
float  dy,
float  phi 
) const
inlineprivate

Definition at line 91 of file PATPackedCandidateProducer.cc.

References funct::cos(), and funct::sin().

91  {
92  return - dx * std::sin(phi) + dy * std::cos(phi);
93  }
Sin< T >::type sin(const T &t)
Definition: Sin.h:22
Cos< T >::type cos(const T &t)
Definition: Cos.h:22
float pat::PATPackedCandidateProducer::calcDz ( reco::Candidate::Point  p,
reco::Candidate::Point  v,
const reco::Candidate c 
) const
inlineprivate

Definition at line 94 of file PATPackedCandidateProducer.cc.

References reco::Candidate::pt(), reco::Candidate::px(), reco::Candidate::py(), and reco::Candidate::pz().

94  {
95  return p.Z()-v.Z() - ((p.X()-v.X()) * c.px() + (p.Y()-v.Y())*c.py()) * c.pz()/(c.pt()*c.pt());
96  }
virtual double pz() const =0
z coordinate of momentum vector
virtual double py() const =0
y coordinate of momentum vector
virtual double pt() const =0
transverse momentum
virtual double px() const =0
x coordinate of momentum vector
bool pat::PATPackedCandidateProducer::candsOrdering ( pat::PackedCandidate  i,
pat::PackedCandidate  j 
) const
inline

Definition at line 44 of file PATPackedCandidateProducer.cc.

References funct::abs(), pat::PackedCandidate::charge(), pat::PackedCandidate::covarianceSchema(), pat::PackedCandidate::eta(), pat::PackedCandidate::hasTrackDetails(), edm::Ref< C, T, F >::key(), and pat::PackedCandidate::vertexRef().

Referenced by sort_indexes().

44  {
45  if (std::abs(i.charge()) == std::abs(j.charge())) {
46  if(i.charge()!=0){
47  if(i.hasTrackDetails() and ! j.hasTrackDetails() ) return true;
48  if(! i.hasTrackDetails() and j.hasTrackDetails() ) return false;
49  if(i.covarianceSchema() > j.covarianceSchema() ) return true;
50  if(i.covarianceSchema() < j.covarianceSchema() ) return false;
51 
52  }
53  if(i.vertexRef() == j.vertexRef())
54  return i.eta() > j.eta();
55  else
56  return i.vertexRef().key() < j.vertexRef().key();
57  }
58  return std::abs(i.charge()) > std::abs(j.charge());
59  }
int covarianceSchema() const
key_type key() const
Accessor for product key.
Definition: Ref.h:264
virtual double eta() const
momentum pseudorapidity
const reco::VertexRef vertexRef() const
bool hasTrackDetails()
Return true if a bestTrack can be extracted from this Candidate.
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
virtual int charge() const
electric charge
void pat::PATPackedCandidateProducer::produce ( edm::StreamID  ,
edm::Event iEvent,
const edm::EventSetup iSetup 
) const
overridevirtual

Definition at line 135 of file PATPackedCandidateProducer.cc.

References funct::abs(), begin, EnergyCorrector::c, Cands_, covariancePackingSchemas_, covarianceVersion_, DEFINE_FWK_MODULE, allConversions_cfi::dz, reco::TrackBase::dz(), reco::PFCandidate::ecalEnergy(), edm::View< T >::end(), reco::LeafCandidate::eta(), reco::TrackBase::eta(), edm::helper::Filler< Map >::fill(), edm::Event::getByToken(), reco::PFCandidate::gsfTrackRef(), reco::HitPattern::hasValidHitInPixelLayer(), reco::PFCandidate::hcalEnergy(), reco::TrackBase::highPurity, reco::TrackBase::hitPattern(), mps_fire::i, edm::HandleBase::id(), edm::Ptr< T >::id(), edm::Ref< C, T, F >::id(), cuy::ii, edm::helper::Filler< Map >::insert(), edm::Ref< C, T, F >::isNonnull(), edm::HandleBase::isValid(), crabWrapper::key, edm::Ptr< T >::key(), edm::Ref< C, T, F >::key(), cutsCKF_cfi::lostHits, HPSPFTaus_cff::mapping, minPtForTrackProperties_, reco::HitPattern::MISSING_INNER_HITS, pat::PackedCandidate::moreLostInnerHits, eostools::move(), reco::PFCandidate::muonRef(), pat::PackedCandidate::noLostInnerHits, reco::HitPattern::numberOfLostHits(), reco::HitPattern::numberOfValidPixelHits(), pat::PackedCandidate::oneLostInnerHit, reco::LeafCandidate::pdgId(), reco::LeafCandidate::phi(), reco::TrackBase::phi(), GeomDetEnumerators::PixelBarrel, reco::LeafCandidate::polarP4(), edm::Handle< T >::product(), reco::LeafCandidate::pt(), reco::TrackBase::pt(), PuppiCands_, PuppiCandsMap_, PuppiCandsNoLep_, PuppiWeight_, PuppiWeightNoLep_, edm::Event::put(), PVAsso_, PVAssoQuality_, PVOrigs_, PVs_, btvTracks_cfi::quality, pat::qualityMap, reco::TrackBase::referencePoint(), edm::View< T >::size(), sort_indexes(), SVWhiteLists_, TKOrigs_, reco::RecoChargedCandidate::track(), reco::PFCandidate::trackRef(), pat::PackedCandidate::UsedInFitTight, usePuppi_, pat::PackedCandidate::validHitInFirstPixelBarrelLayer, reco::PFCandidate::vertex(), and badGlobalMuonTaggersAOD_cff::vtx.

135  {
136 
138  iEvent.getByToken( Cands_, cands );
139  std::vector<reco::Candidate>::const_iterator cand;
140 
141  edm::Handle<edm::ValueMap<float> > puppiWeight;
144  edm::Handle<edm::ValueMap<float> > puppiWeightNoLep;
146  std::vector<reco::CandidatePtr> puppiCandsNoLepPtrs;
147  if(usePuppi_){
148  iEvent.getByToken( PuppiWeight_, puppiWeight );
149  iEvent.getByToken( PuppiCandsMap_, puppiCandsMap );
150  iEvent.getByToken( PuppiCands_, puppiCands );
151  iEvent.getByToken( PuppiWeightNoLep_, puppiWeightNoLep );
152  iEvent.getByToken( PuppiCandsNoLep_, puppiCandsNoLep );
153  for (auto pup : *puppiCandsNoLep){
154  puppiCandsNoLepPtrs.push_back(pup.sourceCandidatePtr(0));
155  }
156  }
157  std::vector<int> mappingPuppi(usePuppi_ ? puppiCands->size() : 0);
158 
160  iEvent.getByToken( PVOrigs_, PVOrigs );
161 
163  iEvent.getByToken(PVAsso_,assoHandle);
164  edm::Handle<edm::ValueMap<int> > assoQualityHandle;
165  iEvent.getByToken(PVAssoQuality_,assoQualityHandle);
166  const edm::Association<reco::VertexCollection> & associatedPV=*(assoHandle.product());
167  const edm::ValueMap<int> & associationQuality=*(assoQualityHandle.product());
168 
169 
170  std::set<unsigned int> whiteList;
171  std::set<reco::TrackRef> whiteListTk;
172  for(auto itoken : SVWhiteLists_) {
173  edm::Handle<edm::View<reco::Candidate > > svWhiteListHandle;
174  iEvent.getByToken(itoken, svWhiteListHandle);
175  const edm::View<reco::Candidate > & svWhiteList=*(svWhiteListHandle.product());
176  for(unsigned int i=0; i<svWhiteList.size();i++) {
177  //Whitelist via Ptrs
178  for(unsigned int j=0; j< svWhiteList[i].numberOfSourceCandidatePtrs(); j++) {
179  const edm::Ptr<reco::Candidate> & c = svWhiteList[i].sourceCandidatePtr(j);
180  if(c.id() == cands.id()) whiteList.insert(c.key());
181 
182  }
183  //Whitelist via RecoCharged
184  for(auto dau = svWhiteList[i].begin(); dau != svWhiteList[i].end() ; dau++){
185  const reco::RecoChargedCandidate * chCand=dynamic_cast<const reco::RecoChargedCandidate *>(&(*dau));
186  if(chCand!=nullptr) {
187  whiteListTk.insert(chCand->track());
188  }
189  }
190  }
191  }
192 
193 
195  iEvent.getByToken( PVs_, PVs );
196  reco::VertexRef PV(PVs.id());
197  reco::VertexRefProd PVRefProd(PVs);
198  math::XYZPoint PVpos;
199 
200 
202  iEvent.getByToken( TKOrigs_, TKOrigs );
203  auto outPtrP = std::make_unique<std::vector<pat::PackedCandidate>>();
204  std::vector<int> mapping(cands->size());
205  std::vector<int> mappingReverse(cands->size());
206  std::vector<int> mappingTk(TKOrigs->size(), -1);
207 
208  for(unsigned int ic=0, nc = cands->size(); ic < nc; ++ic) {
209  const reco::PFCandidate &cand=(*cands)[ic];
210  const reco::Track *ctrack = 0;
211  if ((abs(cand.pdgId()) == 11 || cand.pdgId() == 22) && cand.gsfTrackRef().isNonnull()) {
212  ctrack = &*cand.gsfTrackRef();
213  } else if (cand.trackRef().isNonnull()) {
214  ctrack = &*cand.trackRef();
215  }
216  if (ctrack) {
217  float dist=1e99;
218  int pvi=-1;
219  for(size_t ii=0;ii<PVs->size();ii++){
220  float dz=std::abs(ctrack->dz( ((*PVs)[ii]).position()));
221  if(dz<dist) {pvi=ii;dist=dz; }
222  }
223  PV = reco::VertexRef(PVs, pvi);
224  math::XYZPoint vtx = cand.vertex();
226  const reco::VertexRef & PVOrig = associatedPV[reco::CandidatePtr(cands,ic)];
227  if(PVOrig.isNonnull()) PV = reco::VertexRef(PVs, PVOrig.key()); // WARNING: assume the PV slimmer is keeping same order
228  int quality=associationQuality[reco::CandidatePtr(cands,ic)];
229 // if ((size_t)pvi!=PVOrig.key()) std::cout << "not closest in Z" << pvi << " " << PVOrig.key() << " " << cand.pt() << " " << quality << std::endl;
230  // TrajectoryStateOnSurface tsos = extrapolator.extrapolate(trajectoryStateTransform::initialFreeState(*ctrack,&*magneticField), RecoVertex::convertPos(PV->position()));
231  // vtx = tsos.globalPosition();
232  // phiAtVtx = tsos.globalDirection().phi();
233  vtx = ctrack->referencePoint();
234  float ptTrk = ctrack->pt();
235  float etaAtVtx = ctrack->eta();
236  float phiAtVtx = ctrack->phi();
237 
239  if (nlost == 0) {
242  }
243  } else {
245  }
246 
247 
248  outPtrP->push_back( pat::PackedCandidate(cand.polarP4(), vtx, ptTrk, etaAtVtx, phiAtVtx, cand.pdgId(), PVRefProd, PV.key()));
249  outPtrP->back().setAssociationQuality(pat::PackedCandidate::PVAssociationQuality(qualityMap[quality]));
250  outPtrP->back().setCovarianceVersion(covarianceVersion_);
251  if(cand.trackRef().isNonnull() && PVOrig->trackWeight(cand.trackRef()) > 0.5 && quality == 7) {
252  outPtrP->back().setAssociationQuality(pat::PackedCandidate::UsedInFitTight);
253  }
254  // properties of the best track
255  outPtrP->back().setLostInnerHits( lostHits );
256  if(outPtrP->back().pt() > minPtForTrackProperties_ ||
257  outPtrP->back().ptTrk() > minPtForTrackProperties_ ||
258  whiteList.find(ic)!=whiteList.end() ||
259  (cand.trackRef().isNonnull() && whiteListTk.find(cand.trackRef())!=whiteListTk.end())
260  ) {
261  if(abs(outPtrP->back().pdgId())==22) {outPtrP->back().setTrackProperties(*ctrack,covariancePackingSchemas_[4],covarianceVersion_);}
262  else {
263  if( ctrack->hitPattern().numberOfValidPixelHits() >0) { outPtrP->back().setTrackProperties(*ctrack,covariancePackingSchemas_[0],covarianceVersion_);} //high quality
264  else { outPtrP->back().setTrackProperties(*ctrack,covariancePackingSchemas_[1],covarianceVersion_);}
265  }
266 
267  //outPtrP->back().setTrackProperties(*ctrack,tsos.curvilinearError());
268  } else {
269  if(outPtrP->back().pt() > 0.5 ){
270  if(ctrack->hitPattern().numberOfValidPixelHits() >0) outPtrP->back().setTrackProperties(*ctrack,covariancePackingSchemas_[2],covarianceVersion_); //low quality, with pixels
271  else outPtrP->back().setTrackProperties(*ctrack,covariancePackingSchemas_[3],covarianceVersion_); //low quality, without pixels
272  }
273  }
274 
275  // these things are always for the CKF track
276  outPtrP->back().setTrackHighPurity( cand.trackRef().isNonnull() && cand.trackRef()->quality(reco::Track::highPurity) );
277  if (cand.muonRef().isNonnull()) {
278  outPtrP->back().setMuonID(cand.muonRef()->isStandAloneMuon(), cand.muonRef()->isGlobalMuon());
279  }
280  } else {
281 
282  if (!PVs->empty()) {
283  PV = reco::VertexRef(PVs, 0);
284  PVpos = PV->position();
285  }
286 
287  outPtrP->push_back( pat::PackedCandidate(cand.polarP4(), PVpos, cand.pt(), cand.eta(), cand.phi(), cand.pdgId(), PVRefProd, PV.key()));
289  }
290 
291  // neutrals
292 
293  if(abs(cand.pdgId()) == 1 || abs(cand.pdgId()) == 130) {
294  outPtrP->back().setHcalFraction(cand.hcalEnergy()/(cand.ecalEnergy()+cand.hcalEnergy()));
295  } else {
296  outPtrP->back().setHcalFraction(0);
297  }
298 
299 
300  if (usePuppi_){
301  reco::PFCandidateRef pkref( cands, ic );
302  // outPtrP->back().setPuppiWeight( (*puppiWeight)[pkref]);
303 
304  float puppiWeightVal = (*puppiWeight)[pkref];
305  float puppiWeightNoLepVal = 0.0;
306  // Check the "no lepton" puppi weights.
307  // If present, then it is not a lepton, use stored weight
308  // If absent, it is a lepton, so set the weight to 1.0
309  if ( puppiWeightNoLep.isValid() ) {
310  // Look for the pointer inside the "no lepton" candidate collection.
311  auto pkrefPtr = pkref->sourceCandidatePtr(0);
312 
313  bool foundNoLep = false;
314  for ( size_t ipcnl = 0; ipcnl < puppiCandsNoLepPtrs.size(); ipcnl++){
315  if (puppiCandsNoLepPtrs[ipcnl] == pkrefPtr){
316  foundNoLep = true;
317  puppiWeightNoLepVal = puppiCandsNoLep->at(ipcnl).pt()/cand.pt(); // a hack for now, should use the value map
318  break;
319  }
320  }
321  if ( !foundNoLep || puppiWeightNoLepVal > 1 ) {
322  puppiWeightNoLepVal = 1.0;
323  }
324  }
325  outPtrP->back().setPuppiWeight( puppiWeightVal, puppiWeightNoLepVal );
326 
327  mappingPuppi[((*puppiCandsMap)[pkref]).key()]=ic;
328  }
329 
330  mapping[ic] = ic; // trivial at the moment!
331  if (cand.trackRef().isNonnull() && cand.trackRef().id() == TKOrigs.id()) {
332  mappingTk[cand.trackRef().key()] = ic;
333  }
334 
335  }
336 
337  auto outPtrPSorted = std::make_unique<std::vector<pat::PackedCandidate>>();
338  std::vector<size_t> order=sort_indexes(*outPtrP);
339  std::vector<size_t> reverseOrder(order.size());
340  for(size_t i=0,nc=cands->size();i<nc;i++) {
341  outPtrPSorted->push_back((*outPtrP)[order[i]]);
342  reverseOrder[order[i]] = i;
343  mappingReverse[order[i]]=i;
344  }
345 
346  // Fix track association for sorted candidates
347  for(size_t i=0,ntk=mappingTk.size();i<ntk;i++){
348  if(mappingTk[i] >= 0)
349  mappingTk[i]=reverseOrder[mappingTk[i]];
350  }
351 
352  for(size_t i=0,ntk=mappingPuppi.size();i<ntk;i++){
353  mappingPuppi[i]=reverseOrder[mappingPuppi[i]];
354  }
355 
357 
358  // now build the two maps
359  auto pf2pc = std::make_unique<edm::Association<pat::PackedCandidateCollection>>(oh);
360  auto pc2pf = std::make_unique<edm::Association<reco::PFCandidateCollection>>(cands);
363  pf2pcFiller.insert(cands, mappingReverse.begin(), mappingReverse.end());
364  pc2pfFiller.insert(oh , order.begin(), order.end());
365  // include also the mapping track -> packed PFCand
366  pf2pcFiller.insert(TKOrigs, mappingTk.begin(), mappingTk.end());
367  if(usePuppi_) pf2pcFiller.insert(puppiCands, mappingPuppi.begin(), mappingPuppi.end());
368 
369  pf2pcFiller.fill();
370  pc2pfFiller.fill();
371  iEvent.put(std::move(pf2pc));
372  iEvent.put(std::move(pc2pf));
373 
374 }
virtual double pt() const final
transverse momentum
double ecalEnergy() const
return corrected Ecal energy
Definition: PFCandidate.h:221
const Point & referencePoint() const
Reference point on the track.
Definition: TrackBase.h:676
const edm::EDGetTokenT< edm::ValueMap< float > > PuppiWeight_
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:122
bool isNonnull() const
Checks for non-null.
Definition: Ref.h:252
bool hasValidHitInPixelLayer(enum PixelSubdetector::SubDetector, uint16_t layer) const
Definition: HitPattern.cc:327
key_type key() const
Definition: Ptr.h:186
ProductID id() const
Definition: HandleBase.cc:15
virtual double eta() const final
momentum pseudorapidity
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:457
size_type size() const
double phi() const
azimuthal angle of momentum vector
Definition: TrackBase.h:640
static const int qualityMap[8]
conversion map from quality flags used in PV association and miniAOD one
const edm::EDGetTokenT< reco::PFCandidateCollection > Cands_
key_type key() const
Accessor for product key.
Definition: Ref.h:264
ProductID id() const
Accessor for product ID.
Definition: Ref.h:258
const edm::EDGetTokenT< edm::ValueMap< int > > PVAssoQuality_
reco::TrackRef trackRef() const
Definition: PFCandidate.cc:438
virtual double phi() const final
momentum azimuthal angle
double eta() const
pseudorapidity of momentum vector
Definition: TrackBase.h:646
double pt() const
track transverse momentum
Definition: TrackBase.h:616
virtual int pdgId() const final
PDG identifier.
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
virtual const Point & vertex() const
vertex position (overwritten by PF...)
Definition: PFCandidate.cc:652
const edm::EDGetTokenT< reco::TrackCollection > TKOrigs_
const edm::EDGetTokenT< std::vector< reco::PFCandidate > > PuppiCands_
std::vector< size_t > sort_indexes(const std::vector< T > &v) const
const edm::EDGetTokenT< edm::ValueMap< reco::CandidatePtr > > PuppiCandsMap_
bool isValid() const
Definition: HandleBase.h:74
const edm::EDGetTokenT< edm::ValueMap< float > > PuppiWeightNoLep_
edm::Ref< VertexCollection > VertexRef
persistent reference to a Vertex
Definition: VertexFwd.h:13
reco::MuonRef muonRef() const
Definition: PFCandidate.cc:455
ii
Definition: cuy.py:588
double dz() const
dz parameter (= dsz/cos(lambda)). This is the track z0 w.r.t (0,0,0) only if the refPoint is close to...
Definition: TrackBase.h:604
LostInnerHits
Enumerator specifying the.
edm::Ptr< Candidate > CandidatePtr
persistent reference to an object in a collection of Candidate objects
Definition: CandidateFwd.h:25
T const * product() const
Definition: Handle.h:81
const HitPattern & hitPattern() const
Access the hit pattern, indicating in which Tracker layers the track has hits.
Definition: TrackBase.h:445
const std::vector< int > covariancePackingSchemas_
const edm::EDGetTokenT< std::vector< reco::PFCandidate > > PuppiCandsNoLep_
XYZPointD XYZPoint
point in space with cartesian internal representation
Definition: Point3D.h:12
virtual reco::TrackRef track() const
reference to a track
ProductID id() const
Accessor for product ID.
Definition: Ptr.h:181
const edm::EDGetTokenT< reco::VertexCollection > PVOrigs_
int numberOfLostHits(HitCategory category) const
Definition: HitPattern.h:902
Particle reconstructed by the particle flow algorithm.
Definition: PFCandidate.h:39
#define begin
Definition: vmac.h:30
reco::GsfTrackRef gsfTrackRef() const
Definition: PFCandidate.cc:476
std::vector< edm::EDGetTokenT< edm::View< reco::Candidate > > > SVWhiteLists_
virtual const PolarLorentzVector & polarP4() const final
four-momentum Lorentz vector
double hcalEnergy() const
return corrected Hcal energy
Definition: PFCandidate.h:231
int numberOfValidPixelHits() const
Definition: HitPattern.h:838
const_iterator end() const
const edm::EDGetTokenT< edm::Association< reco::VertexCollection > > PVAsso_
const edm::EDGetTokenT< reco::VertexCollection > PVs_
def move(src, dest)
Definition: eostools.py:510
template<typename T >
std::vector<size_t> pat::PATPackedCandidateProducer::sort_indexes ( const std::vector< T > &  v) const
inline

Definition at line 61 of file PATPackedCandidateProducer.cc.

References candsOrdering(), mps_fire::i, training_settings::idx, and findQualityFiles::v.

Referenced by produce().

61  {
62  std::vector<size_t> idx(v.size());
63  for (size_t i = 0; i != idx.size(); ++i) idx[i] = i;
64  std::sort(idx.begin(), idx.end(),[&v,this](size_t i1, size_t i2) { return candsOrdering(v[i1],v[i2]);});
65  return idx;
66  }
bool candsOrdering(pat::PackedCandidate i, pat::PackedCandidate j) const

Member Data Documentation

const edm::EDGetTokenT<reco::PFCandidateCollection> pat::PATPackedCandidateProducer::Cands_
private

Definition at line 74 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const std::vector<int> pat::PATPackedCandidateProducer::covariancePackingSchemas_
private

Definition at line 89 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const int pat::PATPackedCandidateProducer::covarianceVersion_
private

Definition at line 88 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const double pat::PATPackedCandidateProducer::minPtForTrackProperties_
private

Definition at line 87 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<std::vector< reco::PFCandidate > > pat::PATPackedCandidateProducer::PuppiCands_
private

Definition at line 83 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<edm::ValueMap<reco::CandidatePtr> > pat::PATPackedCandidateProducer::PuppiCandsMap_
private

Definition at line 82 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<std::vector< reco::PFCandidate > > pat::PATPackedCandidateProducer::PuppiCandsNoLep_
private

Definition at line 84 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const edm::EDGetTokenT< edm::ValueMap<float> > pat::PATPackedCandidateProducer::PuppiWeight_
private

Definition at line 80 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const edm::EDGetTokenT< edm::ValueMap<float> > pat::PATPackedCandidateProducer::PuppiWeightNoLep_
private

Definition at line 81 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<edm::Association<reco::VertexCollection> > pat::PATPackedCandidateProducer::PVAsso_
private

Definition at line 76 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<edm::ValueMap<int> > pat::PATPackedCandidateProducer::PVAssoQuality_
private

Definition at line 77 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<reco::VertexCollection> pat::PATPackedCandidateProducer::PVOrigs_
private

Definition at line 78 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const edm::EDGetTokenT<reco::VertexCollection> pat::PATPackedCandidateProducer::PVs_
private

Definition at line 75 of file PATPackedCandidateProducer.cc.

Referenced by produce().

std::vector< edm::EDGetTokenT<edm::View<reco::Candidate> > > pat::PATPackedCandidateProducer::SVWhiteLists_
private

Definition at line 85 of file PATPackedCandidateProducer.cc.

Referenced by PATPackedCandidateProducer(), and produce().

const edm::EDGetTokenT<reco::TrackCollection> pat::PATPackedCandidateProducer::TKOrigs_
private

Definition at line 79 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const bool pat::PATPackedCandidateProducer::usePuppi_
private

Definition at line 72 of file PATPackedCandidateProducer.cc.

Referenced by produce().