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
 
void convertCurrentProcessAlias (std::string const &processName)
 Convert "@currentProcess" in InputTag process names to the actual current process name. More...
 
 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 & itemsToGetFrom (BranchType iType) 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 edm::EDGetTokenT< edm::ValueMap< bool > > ChargedHadronIsolation_
 
const std::vector< int > covariancePackingSchemas_
 
const int covarianceVersion_
 
const double minPtForChargedHadronProperties_
 
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_
 
const bool storeChargedHadronIsolation_
 
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 103 of file PATPackedCandidateProducer.cc.

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

103  :
104  usePuppi_(!iConfig.getParameter<edm::InputTag>("PuppiSrc").encode().empty() ||
105  !iConfig.getParameter<edm::InputTag>("PuppiNoLepSrc").encode().empty()),
106  Cands_(consumes<reco::PFCandidateCollection>(iConfig.getParameter<edm::InputTag>("inputCollection"))),
107  PVs_(consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("inputVertices"))),
109  PVAssoQuality_(consumes<edm::ValueMap<int> >(iConfig.getParameter<edm::InputTag>("vertexAssociator"))),
110  PVOrigs_(consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("originalVertices"))),
111  TKOrigs_(consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("originalTracks"))),
112  PuppiWeight_(usePuppi_ ? consumes<edm::ValueMap<float> >(iConfig.getParameter<edm::InputTag>("PuppiSrc")) : edm::EDGetTokenT< edm::ValueMap<float> >()),
113  PuppiWeightNoLep_(usePuppi_ ? consumes<edm::ValueMap<float> >(iConfig.getParameter<edm::InputTag>("PuppiNoLepSrc")) : edm::EDGetTokenT< edm::ValueMap<float> >()),
114  PuppiCandsMap_(usePuppi_ ? consumes<edm::ValueMap<reco::CandidatePtr> >(iConfig.getParameter<edm::InputTag>("PuppiSrc")) : edm::EDGetTokenT<edm::ValueMap<reco::CandidatePtr> >() ),
115  PuppiCands_(usePuppi_ ? consumes<std::vector< reco::PFCandidate > >(iConfig.getParameter<edm::InputTag>("PuppiSrc")) : edm::EDGetTokenT<std::vector< reco::PFCandidate > >() ),
116  PuppiCandsNoLep_(usePuppi_ ? consumes<std::vector< reco::PFCandidate > >(iConfig.getParameter<edm::InputTag>("PuppiNoLepSrc")) : edm::EDGetTokenT<std::vector< reco::PFCandidate > >()),
117  storeChargedHadronIsolation_(!iConfig.getParameter<edm::InputTag>("chargedHadronIsolation").encode().empty()),
118  ChargedHadronIsolation_(consumes<edm::ValueMap<bool> >(iConfig.getParameter<edm::InputTag>("chargedHadronIsolation"))),
119  minPtForChargedHadronProperties_(iConfig.getParameter<double>("minPtForChargedHadronProperties")),
120  minPtForTrackProperties_(iConfig.getParameter<double>("minPtForTrackProperties")),
121  covarianceVersion_(iConfig.getParameter<int >("covarianceVersion")),
122  covariancePackingSchemas_(iConfig.getParameter<std::vector<int> >("covariancePackingSchemas"))
123 {
124  std::vector<edm::InputTag> sv_tags = iConfig.getParameter<std::vector<edm::InputTag> >("secondaryVerticesForWhiteList");
125  for(auto itag : sv_tags){
126  SVWhiteLists_.push_back(
128  );
129  }
130 
131  produces< std::vector<pat::PackedCandidate> > ();
132  produces< edm::Association<pat::PackedCandidateCollection> > ();
133  produces< edm::Association<reco::PFCandidateCollection> > ();
134 }
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:166
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_
const edm::EDGetTokenT< edm::ValueMap< bool > > ChargedHadronIsolation_
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 136 of file PATPackedCandidateProducer.cc.

136 {}

Member Function Documentation

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

Definition at line 94 of file PATPackedCandidateProducer.cc.

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

94  {
95  return - dx * std::sin(phi) + dy * std::cos(phi);
96  }
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 97 of file PATPackedCandidateProducer.cc.

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

97  {
98  return p.Z()-v.Z() - ((p.X()-v.X()) * c.px() + (p.Y()-v.Y())*c.py()) * c.pz()/(c.pt()*c.pt());
99  }
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:265
virtual double eta() const
momentum pseudorapidity
const reco::VertexRef vertexRef() const
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
virtual int charge() const
electric charge
bool hasTrackDetails() const
Return true if a bestTrack can be extracted from this Candidate.
void pat::PATPackedCandidateProducer::produce ( edm::StreamID  ,
edm::Event iEvent,
const edm::EventSetup iSetup 
) const
overridevirtual

Definition at line 140 of file PATPackedCandidateProducer.cc.

References funct::abs(), begin, EnergyCorrector::c, Cands_, packedPFCandidates_cfi::chargedHadronIsolation, ChargedHadronIsolation_, covariancePackingSchemas_, covarianceVersion_, DEFINE_FWK_MODULE, PVValHelper::dz, reco::TrackBase::dz(), reco::PFCandidate::ecalEnergy(), edm::View< T >::end(), reco::LeafCandidate::energy(), reco::LeafCandidate::eta(), reco::TrackBase::eta(), edm::helper::Filler< Map >::fill(), edm::Event::getByToken(), reco::HitPattern::getHitPattern(), 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, minPtForChargedHadronProperties_, 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::PFCandidate::rawEcalEnergy(), reco::PFCandidate::rawHcalEnergy(), reco::TrackBase::referencePoint(), edm::View< T >::size(), sort_indexes(), storeChargedHadronIsolation_, SVWhiteLists_, TKOrigs_, reco::RecoChargedCandidate::track(), reco::HitPattern::TRACK_HITS, reco::PFCandidate::trackRef(), pat::PackedCandidate::UsedInFitTight, usePuppi_, pat::PackedCandidate::validHitInFirstPixelBarrelLayer, reco::PFCandidate::vertex(), and badGlobalMuonTaggersAOD_cff::vtx.

140  {
141 
143  iEvent.getByToken( Cands_, cands );
144  std::vector<reco::Candidate>::const_iterator cand;
145 
146  edm::Handle<edm::ValueMap<float> > puppiWeight;
149  edm::Handle<edm::ValueMap<float> > puppiWeightNoLep;
151  std::vector<reco::CandidatePtr> puppiCandsNoLepPtrs;
152  if(usePuppi_){
153  iEvent.getByToken( PuppiWeight_, puppiWeight );
154  iEvent.getByToken( PuppiCandsMap_, puppiCandsMap );
155  iEvent.getByToken( PuppiCands_, puppiCands );
156  iEvent.getByToken( PuppiWeightNoLep_, puppiWeightNoLep );
157  iEvent.getByToken( PuppiCandsNoLep_, puppiCandsNoLep );
158  for (auto pup : *puppiCandsNoLep){
159  puppiCandsNoLepPtrs.push_back(pup.sourceCandidatePtr(0));
160  }
161  }
162  std::vector<int> mappingPuppi(usePuppi_ ? puppiCands->size() : 0);
163 
165  iEvent.getByToken( PVOrigs_, PVOrigs );
166 
168  iEvent.getByToken(PVAsso_,assoHandle);
169  edm::Handle<edm::ValueMap<int> > assoQualityHandle;
170  iEvent.getByToken(PVAssoQuality_,assoQualityHandle);
171  const edm::Association<reco::VertexCollection> & associatedPV=*(assoHandle.product());
172  const edm::ValueMap<int> & associationQuality=*(assoQualityHandle.product());
173 
174  edm::Handle<edm::ValueMap<bool> > chargedHadronIsolationHandle;
176  iEvent.getByToken(ChargedHadronIsolation_,chargedHadronIsolationHandle);
177 
178  std::set<unsigned int> whiteList;
179  std::set<reco::TrackRef> whiteListTk;
180  for(auto itoken : SVWhiteLists_) {
181  edm::Handle<edm::View<reco::Candidate > > svWhiteListHandle;
182  iEvent.getByToken(itoken, svWhiteListHandle);
183  const edm::View<reco::Candidate > & svWhiteList=*(svWhiteListHandle.product());
184  for(unsigned int i=0; i<svWhiteList.size();i++) {
185  //Whitelist via Ptrs
186  for(unsigned int j=0; j< svWhiteList[i].numberOfSourceCandidatePtrs(); j++) {
187  const edm::Ptr<reco::Candidate> & c = svWhiteList[i].sourceCandidatePtr(j);
188  if(c.id() == cands.id()) whiteList.insert(c.key());
189 
190  }
191  //Whitelist via RecoCharged
192  for(auto dau = svWhiteList[i].begin(); dau != svWhiteList[i].end() ; dau++){
193  const reco::RecoChargedCandidate * chCand=dynamic_cast<const reco::RecoChargedCandidate *>(&(*dau));
194  if(chCand!=nullptr) {
195  whiteListTk.insert(chCand->track());
196  }
197  }
198  }
199  }
200 
201 
203  iEvent.getByToken( PVs_, PVs );
204  reco::VertexRef PV(PVs.id());
205  reco::VertexRefProd PVRefProd(PVs);
206  math::XYZPoint PVpos;
207 
208 
210  iEvent.getByToken( TKOrigs_, TKOrigs );
211  auto outPtrP = std::make_unique<std::vector<pat::PackedCandidate>>();
212  std::vector<int> mapping(cands->size());
213  std::vector<int> mappingReverse(cands->size());
214  std::vector<int> mappingTk(TKOrigs->size(), -1);
215 
216  for(unsigned int ic=0, nc = cands->size(); ic < nc; ++ic) {
217  const reco::PFCandidate &cand=(*cands)[ic];
218  const reco::Track *ctrack = 0;
219  if ((abs(cand.pdgId()) == 11 || cand.pdgId() == 22) && cand.gsfTrackRef().isNonnull()) {
220  ctrack = &*cand.gsfTrackRef();
221  } else if (cand.trackRef().isNonnull()) {
222  ctrack = &*cand.trackRef();
223  }
224  if (ctrack) {
225  float dist=1e99;
226  int pvi=-1;
227  for(size_t ii=0;ii<PVs->size();ii++){
228  float dz=std::abs(ctrack->dz( ((*PVs)[ii]).position()));
229  if(dz<dist) {pvi=ii;dist=dz; }
230  }
231  PV = reco::VertexRef(PVs, pvi);
232  math::XYZPoint vtx = cand.vertex();
234  const reco::VertexRef & PVOrig = associatedPV[reco::CandidatePtr(cands,ic)];
235  if(PVOrig.isNonnull()) PV = reco::VertexRef(PVs, PVOrig.key()); // WARNING: assume the PV slimmer is keeping same order
236  int quality=associationQuality[reco::CandidatePtr(cands,ic)];
237 // if ((size_t)pvi!=PVOrig.key()) std::cout << "not closest in Z" << pvi << " " << PVOrig.key() << " " << cand.pt() << " " << quality << std::endl;
238  // TrajectoryStateOnSurface tsos = extrapolator.extrapolate(trajectoryStateTransform::initialFreeState(*ctrack,&*magneticField), RecoVertex::convertPos(PV->position()));
239  // vtx = tsos.globalPosition();
240  // phiAtVtx = tsos.globalDirection().phi();
241  vtx = ctrack->referencePoint();
242  float ptTrk = ctrack->pt();
243  float etaAtVtx = ctrack->eta();
244  float phiAtVtx = ctrack->phi();
245 
247  if (nlost == 0) {
250  }
251  } else {
253  }
254 
255 
256  outPtrP->push_back( pat::PackedCandidate(cand.polarP4(), vtx, ptTrk, etaAtVtx, phiAtVtx, cand.pdgId(), PVRefProd, PV.key()));
257  outPtrP->back().setAssociationQuality(pat::PackedCandidate::PVAssociationQuality(qualityMap[quality]));
258  outPtrP->back().setCovarianceVersion(covarianceVersion_);
259  if(cand.trackRef().isNonnull() && PVOrig->trackWeight(cand.trackRef()) > 0.5 && quality == 7) {
260  outPtrP->back().setAssociationQuality(pat::PackedCandidate::UsedInFitTight);
261  }
262  // properties of the best track
263  outPtrP->back().setLostInnerHits( lostHits );
264  if(outPtrP->back().pt() > minPtForTrackProperties_ ||
265  outPtrP->back().ptTrk() > minPtForTrackProperties_ ||
266  whiteList.find(ic)!=whiteList.end() ||
267  (cand.trackRef().isNonnull() && whiteListTk.find(cand.trackRef())!=whiteListTk.end())
268  ) {
269  outPtrP->back().setFirstHit(ctrack->hitPattern().getHitPattern(reco::HitPattern::TRACK_HITS, 0));
270  if(abs(outPtrP->back().pdgId())==22) {
271  outPtrP->back().setTrackProperties(*ctrack,covariancePackingSchemas_[4],covarianceVersion_);
272  } else {
273  if( ctrack->hitPattern().numberOfValidPixelHits() >0) {
274  outPtrP->back().setTrackProperties(*ctrack,covariancePackingSchemas_[0],covarianceVersion_); //high quality
275  } else {
276  outPtrP->back().setTrackProperties(*ctrack,covariancePackingSchemas_[1],covarianceVersion_);
277  }
278  }
279  //outPtrP->back().setTrackProperties(*ctrack,tsos.curvilinearError());
280  } else {
281  if(outPtrP->back().pt() > 0.5 ){
282  if(ctrack->hitPattern().numberOfValidPixelHits() >0) outPtrP->back().setTrackProperties(*ctrack,covariancePackingSchemas_[2],covarianceVersion_); //low quality, with pixels
283  else outPtrP->back().setTrackProperties(*ctrack,covariancePackingSchemas_[3],covarianceVersion_); //low quality, without pixels
284  }
285  }
286 
287  // these things are always for the CKF track
288  outPtrP->back().setTrackHighPurity( cand.trackRef().isNonnull() && cand.trackRef()->quality(reco::Track::highPurity) );
289  if (cand.muonRef().isNonnull()) {
290  outPtrP->back().setMuonID(cand.muonRef()->isStandAloneMuon(), cand.muonRef()->isGlobalMuon());
291  }
292  } else {
293 
294  if (!PVs->empty()) {
295  PV = reco::VertexRef(PVs, 0);
296  PVpos = PV->position();
297  }
298 
299  outPtrP->push_back( pat::PackedCandidate(cand.polarP4(), PVpos, cand.pt(), cand.eta(), cand.phi(), cand.pdgId(), PVRefProd, PV.key()));
301  }
302 
303  // neutrals and isolated charged hadrons
304 
305  bool isIsolatedChargedHadron = false;
307  const edm::ValueMap<bool> & chargedHadronIsolation=*(chargedHadronIsolationHandle.product());
308  isIsolatedChargedHadron=((cand.pt()>minPtForChargedHadronProperties_)&&(chargedHadronIsolation[reco::PFCandidateRef(cands,ic)]));
309  outPtrP->back().setIsIsolatedChargedHadron(isIsolatedChargedHadron);
310  }
311 
312  if(abs(cand.pdgId()) == 1 || abs(cand.pdgId()) == 130) {
313  outPtrP->back().setHcalFraction(cand.hcalEnergy()/(cand.ecalEnergy()+cand.hcalEnergy()));
314  } else if(isIsolatedChargedHadron) {
315  outPtrP->back().setRawCaloFraction((cand.rawEcalEnergy()+cand.rawHcalEnergy())/cand.energy());
316  outPtrP->back().setHcalFraction(cand.rawHcalEnergy()/(cand.rawEcalEnergy()+cand.rawHcalEnergy()));
317  } else {
318  outPtrP->back().setHcalFraction(0);
319  }
320 
321 
322  if (usePuppi_){
323  reco::PFCandidateRef pkref( cands, ic );
324  // outPtrP->back().setPuppiWeight( (*puppiWeight)[pkref]);
325 
326  float puppiWeightVal = (*puppiWeight)[pkref];
327  float puppiWeightNoLepVal = 0.0;
328  // Check the "no lepton" puppi weights.
329  // If present, then it is not a lepton, use stored weight
330  // If absent, it is a lepton, so set the weight to 1.0
331  if ( puppiWeightNoLep.isValid() ) {
332  // Look for the pointer inside the "no lepton" candidate collection.
333  auto pkrefPtr = pkref->sourceCandidatePtr(0);
334 
335  bool foundNoLep = false;
336  for ( size_t ipcnl = 0; ipcnl < puppiCandsNoLepPtrs.size(); ipcnl++){
337  if (puppiCandsNoLepPtrs[ipcnl] == pkrefPtr){
338  foundNoLep = true;
339  puppiWeightNoLepVal = puppiCandsNoLep->at(ipcnl).pt()/cand.pt(); // a hack for now, should use the value map
340  break;
341  }
342  }
343  if ( !foundNoLep || puppiWeightNoLepVal > 1 ) {
344  puppiWeightNoLepVal = 1.0;
345  }
346  }
347  outPtrP->back().setPuppiWeight( puppiWeightVal, puppiWeightNoLepVal );
348 
349  mappingPuppi[((*puppiCandsMap)[pkref]).key()]=ic;
350  }
351 
352  mapping[ic] = ic; // trivial at the moment!
353  if (cand.trackRef().isNonnull() && cand.trackRef().id() == TKOrigs.id()) {
354  mappingTk[cand.trackRef().key()] = ic;
355  }
356 
357  }
358 
359  auto outPtrPSorted = std::make_unique<std::vector<pat::PackedCandidate>>();
360  std::vector<size_t> order=sort_indexes(*outPtrP);
361  std::vector<size_t> reverseOrder(order.size());
362  for(size_t i=0,nc=cands->size();i<nc;i++) {
363  outPtrPSorted->push_back((*outPtrP)[order[i]]);
364  reverseOrder[order[i]] = i;
365  mappingReverse[order[i]]=i;
366  }
367 
368  // Fix track association for sorted candidates
369  for(size_t i=0,ntk=mappingTk.size();i<ntk;i++){
370  if(mappingTk[i] >= 0)
371  mappingTk[i]=reverseOrder[mappingTk[i]];
372  }
373 
374  for(size_t i=0,ntk=mappingPuppi.size();i<ntk;i++){
375  mappingPuppi[i]=reverseOrder[mappingPuppi[i]];
376  }
377 
379 
380  // now build the two maps
381  auto pf2pc = std::make_unique<edm::Association<pat::PackedCandidateCollection>>(oh);
382  auto pc2pf = std::make_unique<edm::Association<reco::PFCandidateCollection>>(cands);
385  pf2pcFiller.insert(cands, mappingReverse.begin(), mappingReverse.end());
386  pc2pfFiller.insert(oh , order.begin(), order.end());
387  // include also the mapping track -> packed PFCand
388  pf2pcFiller.insert(TKOrigs, mappingTk.begin(), mappingTk.end());
389  if(usePuppi_) pf2pcFiller.insert(puppiCands, mappingPuppi.begin(), mappingPuppi.end());
390 
391  pf2pcFiller.fill();
392  pc2pfFiller.fill();
393  iEvent.put(std::move(pf2pc));
394  iEvent.put(std::move(pc2pf));
395 
396 }
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:681
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:253
double rawEcalEnergy() const
return corrected Ecal energy
Definition: PFCandidate.h:224
bool hasValidHitInPixelLayer(enum PixelSubdetector::SubDetector, uint16_t layer) const
Definition: HitPattern.cc:321
key_type key() const
Definition: Ptr.h:185
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:460
size_type size() const
double phi() const
azimuthal angle of momentum vector
Definition: TrackBase.h:645
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:265
ProductID id() const
Accessor for product ID.
Definition: Ref.h:259
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:651
double pt() const
track transverse momentum
Definition: TrackBase.h:621
virtual int pdgId() const final
PDG identifier.
virtual double energy() const final
energy
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:609
LostInnerHits
Enumerator specifying the.
const edm::EDGetTokenT< edm::ValueMap< bool > > ChargedHadronIsolation_
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:446
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:180
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_
uint16_t getHitPattern(HitCategory category, int position) const
Definition: HitPattern.h:515
def move(src, dest)
Definition: eostools.py:510
double rawHcalEnergy() const
return raw Hcal energy
Definition: PFCandidate.h:234
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 edm::EDGetTokenT<edm::ValueMap<bool> > pat::PATPackedCandidateProducer::ChargedHadronIsolation_
private

Definition at line 87 of file PATPackedCandidateProducer.cc.

Referenced by produce().

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

Definition at line 92 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const int pat::PATPackedCandidateProducer::covarianceVersion_
private

Definition at line 91 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const double pat::PATPackedCandidateProducer::minPtForChargedHadronProperties_
private

Definition at line 89 of file PATPackedCandidateProducer.cc.

Referenced by produce().

const double pat::PATPackedCandidateProducer::minPtForTrackProperties_
private

Definition at line 90 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().

const bool pat::PATPackedCandidateProducer::storeChargedHadronIsolation_
private

Definition at line 86 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().