CMS 3D CMS Logo

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

Public Member Functions

 PATTrackAndVertexUnpacker (const edm::ParameterSet &iConfig)
 
virtual void produce (edm::StreamID, edm::Event &iEvent, const edm::EventSetup &iSetup) const override
 
 ~PATTrackAndVertexUnpacker ()
 
- 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 Types

typedef std::vector< edm::InputTagVInputTag
 

Private Attributes

const edm::EDGetTokenT< std::vector< pat::PackedCandidate > > AdditionalTracks_
 
const edm::EDGetTokenT< std::vector< pat::PackedCandidate > > Cands_
 
const edm::EDGetTokenT< reco::VertexCollectionPVs_
 
const edm::EDGetTokenT< reco::VertexCompositePtrCandidateCollectionSVs_
 

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 28 of file TrackAndVertexUnpacker.cc.

Member Typedef Documentation

Definition at line 39 of file TrackAndVertexUnpacker.cc.

Constructor & Destructor Documentation

PATTrackAndVertexUnpacker::PATTrackAndVertexUnpacker ( const edm::ParameterSet iConfig)
explicit

Definition at line 53 of file TrackAndVertexUnpacker.cc.

53  :
54  Cands_(consumes< std::vector<pat::PackedCandidate> >(iConfig.getParameter<edm::InputTag>("packedCandidates"))),
55  PVs_(consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("slimmedVertices"))),
56  SVs_(consumes<reco::VertexCompositePtrCandidateCollection>(iConfig.getParameter<edm::InputTag>("slimmedSecondaryVertices"))),
57  AdditionalTracks_(consumes<pat::PackedCandidateCollection>(iConfig.getParameter<edm::InputTag>("additionalTracks")))
58 {
59  produces<reco::TrackCollection>();
60  produces<reco::VertexCollection>();
61  produces<reco::VertexCollection>("secondary");
62 }
const edm::EDGetTokenT< reco::VertexCollection > PVs_
T getParameter(std::string const &) const
const edm::EDGetTokenT< std::vector< pat::PackedCandidate > > Cands_
EDGetTokenT< ProductType > consumes(edm::InputTag const &tag)
const edm::EDGetTokenT< std::vector< pat::PackedCandidate > > AdditionalTracks_
const edm::EDGetTokenT< reco::VertexCompositePtrCandidateCollection > SVs_
PATTrackAndVertexUnpacker::~PATTrackAndVertexUnpacker ( )

Definition at line 65 of file TrackAndVertexUnpacker.cc.

65  {
66 }

Member Function Documentation

void PATTrackAndVertexUnpacker::produce ( edm::StreamID  ,
edm::Event iEvent,
const edm::EventSetup iSetup 
) const
overridevirtual

Definition at line 69 of file TrackAndVertexUnpacker.cc.

References reco::Vertex::add(), AdditionalTracks_, EnergyCorrector::c, Cands_, pat::PackedCandidate::charge(), reco::CompositePtrCandidate::daughterPtr(), DEFINE_FWK_MODULE, pat::PackedCandidate::fromPV(), edm::Event::getByToken(), pat::PackedCandidate::hasTrackDetails(), mps_fire::i, edm::HandleBase::id(), edm::Ptr< T >::id(), edm::Ptr< T >::key(), eostools::move(), reco::CompositePtrCandidate::numberOfDaughters(), pat::PackedCandidate::numberOfHits(), pat::PackedCandidate::pseudoTrack(), edm::Event::put(), MetAnalyzer::pv(), PVs_, pat::PackedCandidate::PVUsedInFit, alignCSCRings::r, findQualityFiles::rr, SVs_, reco::LeafCandidate::vertex(), reco::VertexCompositePtrCandidate::vertexChi2(), reco::VertexCompositePtrCandidate::vertexCovariance(), and reco::VertexCompositePtrCandidate::vertexNdof().

69  {
70  using namespace edm; using namespace std; using namespace reco;
72  iEvent.getByToken(Cands_, cands);
74  iEvent.getByToken(PVs_, pvs);
76  iEvent.getByToken(SVs_, svs);
78  iEvent.getByToken(AdditionalTracks_, addTracks);
79 
80  auto outTks = std::make_unique<std::vector<reco::Track>>();
81  std::map<unsigned int, std::vector<unsigned int> > asso;
82  std::map<unsigned int, unsigned int> trackKeys;
83  unsigned int j=0;
84  for(unsigned int i=0;i<cands->size();i++) {
85  const pat::PackedCandidate & c = (*cands)[i];
86  if(c.hasTrackDetails() && c.charge() != 0 && c.numberOfHits()> 0){
87  outTks->push_back(c.pseudoTrack());
88  for(size_t ipv=0;ipv< pvs->size(); ++ipv) {
90  asso[ipv].push_back(j);
91  }
92  trackKeys[i]=j;
93  j++;
94  }
95  }
96 
97  int offsetAdd=j;
98  for(unsigned int i = 0; i < addTracks->size(); i++) {
99  if( (*addTracks)[i].hasTrackDetails() ){
100  outTks->push_back((*addTracks)[i].pseudoTrack());
101  for(size_t ipv=0;ipv< pvs->size(); ++ipv) {
102  if((*addTracks)[i].fromPV(ipv)==pat::PackedCandidate::PVUsedInFit)
103  asso[ipv].push_back(j);
104  }
105  j++;
106  }
107  }
108 
110 
111  auto outPv = std::make_unique<std::vector<reco::Vertex>>();
112 
113  for(size_t ipv=0;ipv< pvs->size(); ++ipv) {
114  reco::Vertex pv = (*pvs)[ipv];
115  for(unsigned int i=0;i<asso[ipv].size();i++)
116  {
117  TrackRef r(oh,asso[ipv][i]);
118  TrackBaseRef rr(r);
119  pv.add(rr);
120  }
121  outPv->push_back(pv);
122  }
123  iEvent.put(std::move(outPv));
124 
125  auto outSv = std::make_unique<std::vector<reco::Vertex>>();
126  for(size_t i=0;i< svs->size(); i++) {
127  const reco::VertexCompositePtrCandidate &sv = (*svs)[i];
128  outSv->push_back(reco::Vertex(sv.vertex(),sv.vertexCovariance(),sv.vertexChi2(),sv.vertexNdof(),0));
129  for(size_t j=0;j<sv.numberOfDaughters();j++){
130  TrackRef r;
131  if(sv.daughterPtr(j).id() == cands.id()) {
132  r= TrackRef(oh,trackKeys[sv.daughterPtr(j).key()]); // use trackKeys because cand->track has gaps from neutral
133  } else {
134 // std::cout << "vertex " << i << " using lost Track " << sv.daughterPtr(j).key() << " " << offsetAdd+sv.daughterPtr(j).key() << std::endl;
135  r=TrackRef(oh,offsetAdd+sv.daughterPtr(j).key()); // use directly the key because addTracks is only charged
136  }
137  TrackBaseRef rr(r);
138  outSv->back().add(rr);
139 
140  }
141  }
142 
143  iEvent.put(std::move(outSv),"secondary");
144 
145 }
const edm::EDGetTokenT< reco::VertexCollection > PVs_
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:122
virtual double vertexChi2() const
chi-squares
int numberOfHits() const
key_type key() const
Definition: Ptr.h:186
ProductID id() const
Definition: HandleBase.cc:15
CandidatePtr daughterPtr(size_type i) const
reference to daughter at given position
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:460
virtual const Point & vertex() const
vertex position (overwritten by PF...)
const edm::EDGetTokenT< std::vector< pat::PackedCandidate > > Cands_
virtual double vertexCovariance(int i, int j) const
(i, j)-th element of error matrix, i, j = 0, ... 2
virtual size_t numberOfDaughters() const
number of daughters
const PVAssoc fromPV(size_t ipv=0) const
def pv(vc)
Definition: MetAnalyzer.py:6
virtual int charge() const
electric charge
virtual const reco::Track & pseudoTrack() const
Return reference to a pseudo track made with candidate kinematics, parameterized error for eta...
const edm::EDGetTokenT< std::vector< pat::PackedCandidate > > AdditionalTracks_
bool hasTrackDetails() const
Return true if a bestTrack can be extracted from this Candidate.
void add(const TrackBaseRef &r, float w=1.0)
add a reference to a Track
const edm::EDGetTokenT< reco::VertexCompositePtrCandidateCollection > SVs_
edm::Ref< TrackCollection > TrackRef
persistent reference to a Track
Definition: TrackFwd.h:20
ProductID id() const
Accessor for product ID.
Definition: Ptr.h:181
fixed size matrix
HLT enums.
def move(src, dest)
Definition: eostools.py:510

Member Data Documentation

const edm::EDGetTokenT<std::vector<pat::PackedCandidate> > pat::PATTrackAndVertexUnpacker::AdditionalTracks_
private

Definition at line 44 of file TrackAndVertexUnpacker.cc.

Referenced by produce().

const edm::EDGetTokenT<std::vector<pat::PackedCandidate> > pat::PATTrackAndVertexUnpacker::Cands_
private

Definition at line 41 of file TrackAndVertexUnpacker.cc.

Referenced by produce().

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

Definition at line 42 of file TrackAndVertexUnpacker.cc.

Referenced by produce().

const edm::EDGetTokenT<reco::VertexCompositePtrCandidateCollection> pat::PATTrackAndVertexUnpacker::SVs_
private

Definition at line 43 of file TrackAndVertexUnpacker.cc.

Referenced by produce().