CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
TrackFullCloneSelectorBase.h
Go to the documentation of this file.
1 #ifndef RecoAlgos_TrackFullCloneSelectorBase_h
2 #define RecoAlgos_TrackFullCloneSelectorBase_h
3 
15 #include <utility>
16 #include <vector>
17 #include <memory>
18 #include <algorithm>
19 #include <map>
25 
31 
32 
33 namespace reco { namespace modules {
34 
35 template<typename Selector>
37 public:
40  hSrcTrackToken_( consumes<reco::TrackCollection>( cfg.template getParameter<edm::InputTag>( "src" ) ) ),
41  hTrajToken_( mayConsume< std::vector<Trajectory> >( cfg.template getParameter<edm::InputTag>( "src" ) ) ),
42  hTTAssToken_( mayConsume< TrajTrackAssociationCollection >( cfg.template getParameter<edm::InputTag>( "src" ) ) ),
43  copyExtras_(cfg.template getUntrackedParameter<bool>("copyExtras", false)),
44  copyTrajectories_(cfg.template getUntrackedParameter<bool>("copyTrajectories", false)),
45  selector_( cfg, consumesCollector() ) {
46  std::string alias( cfg.getParameter<std::string>( "@module_label" ) );
47  produces<reco::TrackCollection>().setBranchAlias( alias + "Tracks" );
48  if (copyExtras_) {
49  produces<reco::TrackExtraCollection>().setBranchAlias( alias + "TrackExtras" );
50  produces<TrackingRecHitCollection>().setBranchAlias( alias + "RecHits" );
51  if (copyTrajectories_) {
52  produces< std::vector<Trajectory> >().setBranchAlias( alias + "Trajectories" );
53  produces< TrajTrackAssociationCollection >().setBranchAlias( alias + "TrajectoryTrackAssociations" );
54  }
55  }
56  }
59 
60 private:
62  void produce( edm::Event& evt, const edm::EventSetup& es) override {
64  evt.getByToken( hSrcTrackToken_, hSrcTrack );
65 
66  selTracks_ = std::auto_ptr<reco::TrackCollection>(new reco::TrackCollection());
67  if (copyExtras_) {
68  selTrackExtras_ = std::auto_ptr<reco::TrackExtraCollection>(new reco::TrackExtraCollection());
69  selHits_ = std::auto_ptr<TrackingRecHitCollection>(new TrackingRecHitCollection());
70  }
71 
72  TrackRefProd rTracks = evt.template getRefBeforePut<TrackCollection>();
73 
75  TrackExtraRefProd rTrackExtras;
76  if (copyExtras_) {
77  rHits = evt.template getRefBeforePut<TrackingRecHitCollection>();
78  rTrackExtras = evt.template getRefBeforePut<TrackExtraCollection>();
79  }
80 
81  typedef reco::TrackRef::key_type TrackRefKey;
82  std::map<TrackRefKey, reco::TrackRef > goodTracks;
83  TrackRefKey current = 0;
84 
85  for (reco::TrackCollection::const_iterator it = hSrcTrack->begin(), ed = hSrcTrack->end(); it != ed; ++it, ++current) {
86  const reco::Track & trk = * it;
87  if (!selector_(trk, evt)) continue;
88 
89  selTracks_->push_back( Track( trk ) ); // clone and store
90  if (!copyExtras_) continue;
91 
92  // TrackExtras
93  selTrackExtras_->push_back( TrackExtra( trk.outerPosition(), trk.outerMomentum(), trk.outerOk(),
94  trk.innerPosition(), trk.innerMomentum(), trk.innerOk(),
95  trk.outerStateCovariance(), trk.outerDetId(),
96  trk.innerStateCovariance(), trk.innerDetId(),
97  trk.seedDirection() ) );
98  selTracks_->back().setExtra( TrackExtraRef( rTrackExtras, selTrackExtras_->size() - 1) );
99  TrackExtra & tx = selTrackExtras_->back();
100  // TrackingRecHits
101  for( trackingRecHit_iterator hit = trk.recHitsBegin(); hit != trk.recHitsEnd(); ++ hit ) {
102  selHits_->push_back( (*hit)->clone() );
103  tx.add( TrackingRecHitRef( rHits, selHits_->size() - 1) );
104  }
105  if (copyTrajectories_) {
106  goodTracks[current] = reco::TrackRef(rTracks, selTracks_->size() - 1);
107  }
108  }
109  if ( copyTrajectories_ ) {
112  evt.getByToken(hTTAssToken_, hTTAss);
113  evt.getByToken(hTrajToken_, hTraj);
114  edm::RefProd< std::vector<Trajectory> > TrajRefProd = evt.template getRefBeforePut< std::vector<Trajectory> >();
115  selTrajs_ = std::auto_ptr< std::vector<Trajectory> >(new std::vector<Trajectory>());
116  selTTAss_ = std::auto_ptr< TrajTrackAssociationCollection >(new TrajTrackAssociationCollection());
117  for (size_t i = 0, n = hTraj->size(); i < n; ++i) {
118  edm::Ref< std::vector<Trajectory> > trajRef(hTraj, i);
120  if (match != hTTAss->end()) {
121  const edm::Ref<reco::TrackCollection> &trkRef = match->val;
122  TrackRefKey oldKey = trkRef.key();
123  std::map<TrackRefKey, reco::TrackRef>::iterator getref = goodTracks.find(oldKey);
124  if (getref != goodTracks.end()) {
125  // do the clone
126  selTrajs_->push_back( Trajectory(*trajRef) );
127  selTTAss_->insert ( edm::Ref< std::vector<Trajectory> >(TrajRefProd, selTrajs_->size() - 1),
128  getref->second );
129  }
130  }
131  }
132  }
133 
134  evt.put(selTracks_);
135  if (copyExtras_) {
136  evt.put(selTrackExtras_);
137  evt.put(selHits_);
138  if ( copyTrajectories_ ) {
139  evt.put(selTrajs_);
140  evt.put(selTTAss_);
141  }
142  }
143  }
154  // some space
155  std::auto_ptr<reco::TrackCollection> selTracks_;
156  std::auto_ptr<reco::TrackExtraCollection> selTrackExtras_;
157  std::auto_ptr<TrackingRecHitCollection> selHits_;
158  std::auto_ptr< std::vector<Trajectory> > selTrajs_;
159  std::auto_ptr< TrajTrackAssociationCollection > selTTAss_;
160 };
161 
162 } }
163 #endif
T getParameter(std::string const &) const
int i
Definition: DBlmapReader.cc:9
edm::EDGetTokenT< std::vector< Trajectory > > hTrajToken_
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:446
void produce(edm::Event &evt, const edm::EventSetup &es) override
process one event
TrackFullCloneSelectorBase(const edm::ParameterSet &cfg)
constructor
std::vector< Track > TrackCollection
collection of Tracks
Definition: TrackFwd.h:13
bool innerOk() const
return true if the innermost hit is valid
Definition: Track.h:50
edm::Ref< TrackExtraCollection > TrackExtraRef
persistent reference to a TrackExtra
Definition: TrackExtraFwd.h:17
key_type key() const
Accessor for product key.
Definition: Ref.h:266
const math::XYZPoint & outerPosition() const
position of the outermost hit
Definition: Track.h:65
std::auto_ptr< reco::TrackExtraCollection > selTrackExtras_
EDGetTokenT< ProductType > consumes(edm::InputTag const &tag)
const math::XYZPoint & innerPosition() const
position of the innermost hit
Definition: Track.h:55
void add(const TrackingRecHitRef &ref)
add a reference to a RecHit
OrphanHandle< PROD > put(std::auto_ptr< PROD > product)
Put a new product.
Definition: Event.h:113
edm::Ref< TrackingRecHitCollection > TrackingRecHitRef
persistent reference to a TrackingRecHit
ConsumesCollector consumesCollector()
Use a ConsumesCollector to gather consumes information from helper functions.
CovarianceMatrix outerStateCovariance() const
outermost trajectory state curvilinear errors
Definition: Track.h:75
std::auto_ptr< TrajTrackAssociationCollection > selTTAss_
unsigned int outerDetId() const
DetId of the detector on which surface the outermost state is located.
Definition: Track.h:94
bool copyExtras_
copy only the tracks, not extras and rechits (for AOD)
trackingRecHit_iterator recHitsBegin() const
Iterator to first hit on the track.
Definition: Track.h:104
edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::TrackCollection, unsigned short > > TrajTrackAssociationCollection
std::vector< TrackExtra > TrackExtraCollection
collection of TrackExtra objects
Definition: TrackExtraFwd.h:11
edm::OwnVector< TrackingRecHit > TrackingRecHitCollection
collection of TrackingRecHits
const math::XYZVector & outerMomentum() const
momentum vector at the outermost hit position
Definition: Track.h:70
bool outerOk() const
return true if the outermost hit is valid
Definition: Track.h:45
edm::Ref< TrackCollection > TrackRef
persistent reference to a Track
Definition: TrackFwd.h:19
CovarianceMatrix innerStateCovariance() const
innermost trajectory state curvilinear errors
Definition: Track.h:80
edm::EDGetTokenT< reco::TrackCollection > hSrcTrackToken_
source collection label
std::auto_ptr< std::vector< Trajectory > > selTrajs_
bool copyTrajectories_
copy also trajectories and trajectory-&gt;track associations
const math::XYZVector & innerMomentum() const
momentum vector at the innermost hit position
Definition: Track.h:60
std::auto_ptr< reco::TrackCollection > selTracks_
edm::EDGetTokenT< TrajTrackAssociationCollection > hTTAssToken_
PropagationDirection seedDirection() const
direction of how the hits were sorted in the original seed
Definition: Track.h:204
std::pair< typename Association::data_type::first_type, double > match(Reference key, Association association, bool bestMatchByMaxValue)
Generic matching function.
Definition: Utils.h:6
std::auto_ptr< TrackingRecHitCollection > selHits_
volatile std::atomic< bool > shutdown_flag false
EDGetTokenT< ProductType > mayConsume(edm::InputTag const &tag)
unsigned int innerDetId() const
DetId of the detector on which surface the innermost state is located.
Definition: Track.h:99
def template
Definition: svgfig.py:520
boost::remove_cv< typename boost::remove_reference< argument_type >::type >::type key_type
Definition: Ref.h:170
TrackingRecHitCollection::base::const_iterator trackingRecHit_iterator
iterator over a vector of reference to TrackingRecHit in the same collection
trackingRecHit_iterator recHitsEnd() const
Iterator to last hit on the track.
Definition: Track.h:109