CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
TrackAndVertexUnpacker.cc
Go to the documentation of this file.
1 
13 
14 //#include "DataFormats/Common/interface/ValueMap.h"
15 //#include "DataFormats/Common/interface/View.h"
16 //#include "DataFormats/PatCandidates/interface/Vertexing.h"
17 
22 
23 //#include "PhysicsTools/PatAlgos/interface/VertexingHelper.h"
24 
25 
26 namespace pat {
27 
29 
30 
31  public:
32 
33  explicit PATTrackAndVertexUnpacker(const edm::ParameterSet & iConfig);
35 
36  virtual void produce(edm::StreamID, edm::Event & iEvent, const edm::EventSetup& iSetup) const override;
37 
38  private:
39  typedef std::vector<edm::InputTag> VInputTag;
40  // configurables
46 
47  };
48 
49 }
50 
52 
53 PATTrackAndVertexUnpacker::PATTrackAndVertexUnpacker(const edm::ParameterSet& iConfig) :
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 }
63 
64 
66 }
67 
68 
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  std::auto_ptr< std::vector<reco::Track> > outTks( new 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.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  outTks->push_back((*addTracks)[i].pseudoTrack());
100  for(size_t ipv=0;ipv< pvs->size(); ++ipv) {
101  if((*addTracks)[i].fromPV(ipv)==pat::PackedCandidate::PVUsedInFit)
102  asso[ipv].push_back(j);
103  }
104  j++;
105  }
106 
107  edm::OrphanHandle< std::vector<reco::Track> > oh = iEvent.put( outTks );
108 
109  std::auto_ptr< std::vector<reco::Vertex> > outPv( new std::vector<reco::Vertex> );
110 
111  for(size_t ipv=0;ipv< pvs->size(); ++ipv) {
112  reco::Vertex pv = (*pvs)[ipv];
113  for(unsigned int i=0;i<asso[ipv].size();i++)
114  {
115  TrackRef r(oh,asso[ipv][i]);
116  TrackBaseRef rr(r);
117  pv.add(rr);
118  }
119  outPv->push_back(pv);
120  }
121  iEvent.put(outPv);
122 
123  std::auto_ptr< std::vector<reco::Vertex> > outSv( new std::vector<reco::Vertex> );
124  for(size_t i=0;i< svs->size(); i++) {
125  const reco::VertexCompositePtrCandidate &sv = (*svs)[i];
126  outSv->push_back(reco::Vertex(sv.vertex(),sv.vertexCovariance(),sv.vertexChi2(),sv.vertexNdof(),0));
127  for(size_t j=0;j<sv.numberOfDaughters();j++){
128  TrackRef r;
129  if(sv.daughterPtr(j).id() == cands.id()) {
130  r= TrackRef(oh,trackKeys[sv.daughterPtr(j).key()]); // use trackKeys because cand->track has gaps from neutral
131  } else {
132 // std::cout << "vertex " << i << " using lost Track " << sv.daughterPtr(j).key() << " " << offsetAdd+sv.daughterPtr(j).key() << std::endl;
133  r=TrackRef(oh,offsetAdd+sv.daughterPtr(j).key()); // use directly the key because addTracks is only charged
134  }
135  TrackBaseRef rr(r);
136  outSv->back().add(rr);
137 
138  }
139  }
140 
141  iEvent.put(outSv,"secondary");
142 
143 }
144 
145 
147 
const edm::EDGetTokenT< reco::VertexCollection > PVs_
int i
Definition: DBlmapReader.cc:9
virtual void produce(edm::StreamID, edm::Event &iEvent, const edm::EventSetup &iSetup) const override
virtual double vertexChi2() const
chi-squares
int numberOfHits() const
key_type key() const
Definition: Ptr.h:186
CandidatePtr daughterPtr(size_type i) const
reference to daughter at given position
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:462
virtual const Point & vertex() const
vertex position (overwritten by PF...)
#define DEFINE_FWK_MODULE(type)
Definition: MakerMacros.h:17
const edm::EDGetTokenT< std::vector< pat::PackedCandidate > > Cands_
std::vector< pat::PackedCandidate > PackedCandidateCollection
std::vector< Vertex > VertexCollection
collection of Vertex objects
Definition: VertexFwd.h:9
std::vector< VertexCompositePtrCandidate > VertexCompositePtrCandidateCollection
collection of Candidate objects
virtual double vertexCovariance(int i, int j) const
(i, j)-th element of error matrix, i, j = 0, ... 2
int iEvent
Definition: GenABIO.cc:230
virtual size_t numberOfDaughters() const
number of daughters
OrphanHandle< PROD > put(std::auto_ptr< PROD > product)
Put a new product.
Definition: Event.h:121
const PVAssoc fromPV(size_t ipv=0) const
int j
Definition: DBlmapReader.cc:9
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_
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
std::vector< edm::InputTag > VInputTag
PATTrackAndVertexUnpacker(const edm::ParameterSet &iConfig)
ProductIndex id() const
Definition: ProductID.h:38