CMS 3D CMS Logo

List of all members | Public Member Functions | Public Attributes
TrackCollectionCloner::Producer Struct Reference

#include <TrackCollectionCloner.h>

Public Member Functions

void operator() (Tokens const &tokens, std::vector< unsigned int > const &selected)
 process one event More...
 
 Producer (edm::Event &ievt, TrackCollectionCloner const &cloner)
 
 ~Producer ()
 

Public Attributes

bool copyExtras_
 copy only the tracks, not extras and rechits (for AOD) More...
 
bool copyTrajectories_
 copy also trajectories and trajectory->track associations More...
 
edm::Eventevt
 the event More...
 
std::unique_ptr< TrackingRecHitCollectionselHits_
 
std::unique_ptr< reco::TrackExtraCollectionselTrackExtras_
 
std::unique_ptr< reco::TrackCollectionselTracks_
 
std::unique_ptr< std::vector< Trajectory > > selTrajs_
 
std::unique_ptr< TrajTrackAssociationCollectionselTTAss_
 

Detailed Description

Definition at line 58 of file TrackCollectionCloner.h.

Constructor & Destructor Documentation

TrackCollectionCloner::Producer::Producer ( edm::Event ievt,
TrackCollectionCloner const &  cloner 
)

Definition at line 9 of file TrackCollectionCloner.cc.

References copyExtras_, copyTrajectories_, evt, edm::Event::getRefBeforePut(), selHits_, selTrackExtras_, selTracks_, selTrajs_, and selTTAss_.

9  :
10  copyExtras_(cloner.copyExtras_), copyTrajectories_(cloner.copyTrajectories_), evt(ievt) {
11  selTracks_ = std::make_unique<reco::TrackCollection>();
12  if (copyExtras_) {
13  selTrackExtras_ = std::make_unique<reco::TrackExtraCollection>();
14  selHits_ = std::make_unique<TrackingRecHitCollection>();
15  }
16  if ( copyTrajectories_ ) {
17  selTrajs_ = std::make_unique< std::vector<Trajectory> >();
18  selTTAss_ = std::make_unique< TrajTrackAssociationCollection >(evt.getRefBeforePut< std::vector<Trajectory> >(), evt.getRefBeforePut<reco::TrackCollection>());
19  }
20 
21 }
std::unique_ptr< std::vector< Trajectory > > selTrajs_
std::vector< Track > TrackCollection
collection of Tracks
Definition: TrackFwd.h:14
std::unique_ptr< reco::TrackExtraCollection > selTrackExtras_
std::unique_ptr< reco::TrackCollection > selTracks_
std::unique_ptr< TrackingRecHitCollection > selHits_
bool copyTrajectories_
copy also trajectories and trajectory->track associations
RefProd< PROD > getRefBeforePut()
Definition: Event.h:147
bool copyExtras_
copy only the tracks, not extras and rechits (for AOD)
std::unique_ptr< TrajTrackAssociationCollection > selTTAss_
TrackCollectionCloner::Producer::~Producer ( )

Definition at line 82 of file TrackCollectionCloner.cc.

References copyExtras_, copyTrajectories_, evt, eostools::move(), edm::Event::put(), selHits_, selTrackExtras_, selTracks_, selTrajs_, and selTTAss_.

82  {
83  selTracks_->shrink_to_fit();
84  auto tsize = selTracks_->size();
86  if (copyExtras_) {
87  selTrackExtras_->shrink_to_fit();
88  assert(selTrackExtras_->size()==tsize);
89  selHits_->shrink_to_fit();
92  }
93  if ( copyTrajectories_ ) {
94  selTrajs_->shrink_to_fit();
95  assert(selTrajs_->size()==tsize);
96  assert(selTTAss_->size()==tsize);
99 
100  }
101 }
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:127
std::unique_ptr< std::vector< Trajectory > > selTrajs_
std::unique_ptr< reco::TrackExtraCollection > selTrackExtras_
std::unique_ptr< reco::TrackCollection > selTracks_
std::unique_ptr< TrackingRecHitCollection > selHits_
bool copyTrajectories_
copy also trajectories and trajectory->track associations
bool copyExtras_
copy only the tracks, not extras and rechits (for AOD)
def move(src, dest)
Definition: eostools.py:510
std::unique_ptr< TrajTrackAssociationCollection > selTTAss_

Member Function Documentation

void TrackCollectionCloner::Producer::operator() ( Tokens const &  tokens,
std::vector< unsigned int > const &  selected 
)

process one event

Definition at line 25 of file TrackCollectionCloner.cc.

References copyExtras_, copyTrajectories_, evt, edm::Event::getRefBeforePut(), gen::k, selHits_, selTrackExtras_, selTracks_, selTrajs_, selTTAss_, TrackCollectionTokens::tracks(), and TrackCollectionTokens::trajectories().

25  {
26 
27  auto rTracks = evt.getRefBeforePut<reco::TrackCollection>();
28 
30  reco::TrackExtraRefProd rTrackExtras;
31  if (copyExtras_) {
34  }
35 
37  if ( copyTrajectories_ ) {
38  trajRefProd = evt.getRefBeforePut< std::vector<Trajectory> >();
39  }
40 
41  std::vector<Trajectory> dummy;
42 
43  auto const & trajIn = copyTrajectories_ ? tokens.trajectories(evt) : dummy;
44 
45  auto const & tracksIn = tokens.tracks(evt);
46  for (auto k : selected) {
47  auto const & trk = tracksIn[k];
48  selTracks_->emplace_back( trk ); // clone and store
49  if (copyTrajectories_) {
50  // we assume tracks and trajectories are one-to-one and the assocMap is useless
51  selTrajs_->emplace_back(trajIn[k]);
52  assert(selTrajs_->back().measurements().size()==trk.recHitsSize());
53  selTTAss_->insert ( edm::Ref< std::vector<Trajectory> >(trajRefProd, selTrajs_->size() - 1),
54  reco::TrackRef(rTracks, selTracks_->size() - 1)
55  );
56  }
57 
58  if (!copyExtras_) continue;
59 
60  // TrackExtras
61  selTrackExtras_->emplace_back( trk.outerPosition(), trk.outerMomentum(), trk.outerOk(),
62  trk.innerPosition(), trk.innerMomentum(), trk.innerOk(),
63  trk.outerStateCovariance(), trk.outerDetId(),
64  trk.innerStateCovariance(), trk.innerDetId(),
65  trk.seedDirection(), trk.seedRef()
66  );
67  selTracks_->back().setExtra( reco::TrackExtraRef( rTrackExtras, selTrackExtras_->size() - 1) );
68  auto & tx = selTrackExtras_->back();
69  tx.setResiduals(trk.residuals());
70  auto nh1=trk.recHitsSize();
71  tx.setHits(rHits,selHits_->size(),nh1);
72  tx.setTrajParams(trk.extra()->trajParams(),trk.extra()->chi2sX5());
73  assert(tx.trajParams().size()==tx.recHitsSize());
74  // TrackingRecHits
75  for( auto hit = trk.recHitsBegin(); hit != trk.recHitsEnd(); ++ hit ) {
76  selHits_->push_back( (*hit)->clone() );
77  }
78  }
79 
80 }
std::unique_ptr< std::vector< Trajectory > > selTrajs_
std::vector< Track > TrackCollection
collection of Tracks
Definition: TrackFwd.h:14
std::unique_ptr< reco::TrackExtraCollection > selTrackExtras_
std::unique_ptr< reco::TrackCollection > selTracks_
std::unique_ptr< TrackingRecHitCollection > selHits_
bool copyTrajectories_
copy also trajectories and trajectory->track associations
RefProd< PROD > getRefBeforePut()
Definition: Event.h:147
int k[5][pyjets_maxn]
std::vector< TrackExtra > TrackExtraCollection
collection of TrackExtra objects
Definition: TrackExtraFwd.h:11
bool copyExtras_
copy only the tracks, not extras and rechits (for AOD)
std::unique_ptr< TrajTrackAssociationCollection > selTTAss_

Member Data Documentation

bool TrackCollectionCloner::Producer::copyExtras_

copy only the tracks, not extras and rechits (for AOD)

Definition at line 66 of file TrackCollectionCloner.h.

Referenced by operator()(), Producer(), and ~Producer().

bool TrackCollectionCloner::Producer::copyTrajectories_

copy also trajectories and trajectory->track associations

Definition at line 68 of file TrackCollectionCloner.h.

Referenced by operator()(), Producer(), and ~Producer().

edm::Event& TrackCollectionCloner::Producer::evt

the event

Definition at line 71 of file TrackCollectionCloner.h.

Referenced by operator()(), Producer(), and ~Producer().

std::unique_ptr<TrackingRecHitCollection> TrackCollectionCloner::Producer::selHits_

Definition at line 75 of file TrackCollectionCloner.h.

Referenced by operator()(), Producer(), and ~Producer().

std::unique_ptr<reco::TrackExtraCollection> TrackCollectionCloner::Producer::selTrackExtras_

Definition at line 74 of file TrackCollectionCloner.h.

Referenced by operator()(), Producer(), and ~Producer().

std::unique_ptr<reco::TrackCollection> TrackCollectionCloner::Producer::selTracks_

Definition at line 73 of file TrackCollectionCloner.h.

Referenced by operator()(), Producer(), and ~Producer().

std::unique_ptr< std::vector<Trajectory> > TrackCollectionCloner::Producer::selTrajs_

Definition at line 76 of file TrackCollectionCloner.h.

Referenced by operator()(), Producer(), and ~Producer().

std::unique_ptr< TrajTrackAssociationCollection > TrackCollectionCloner::Producer::selTTAss_

Definition at line 77 of file TrackCollectionCloner.h.

Referenced by operator()(), Producer(), and ~Producer().