#include <RecoTrackAccumulator.h>

Inheritance diagram for RecoTrackAccumulator:

Public Member Functions
virtual void	accumulate (edm::Event const &e, edm::EventSetup const &c)

virtual void	accumulate (PileUpEventPrincipal const &e, edm::EventSetup const &c, edm::StreamID const &) override

virtual void	finalizeEvent (edm::Event &e, edm::EventSetup const &c)

virtual void	initializeEvent (edm::Event const &e, edm::EventSetup const &c)

	RecoTrackAccumulator (const edm::ParameterSet &conf, edm::one::EDProducerBase &mixMod, edm::ConsumesCollector &iC)

virtual	~RecoTrackAccumulator ()

Public Member Functions inherited from DigiAccumulatorMixMod
virtual void	beginLuminosityBlock (edm::LuminosityBlock const &lumi, edm::EventSetup const &setup)

virtual void	beginRun (edm::Run const &run, edm::EventSetup const &setup)

	DigiAccumulatorMixMod ()

virtual void	endLuminosityBlock (edm::LuminosityBlock const &lumi, edm::EventSetup const &setup)

virtual void	endRun (edm::Run const &run, edm::EventSetup const &setup)

virtual void	finalizeBunchCrossing (edm::Event &event, edm::EventSetup const &setup, int bunchCrossing)

virtual PileupMixingContent *	getEventPileupInfo ()

virtual void	initializeBunchCrossing (edm::Event const &event, edm::EventSetup const &setup, int bunchCrossing)

virtual void	StorePileupInformation (std::vector< int > &numInteractionList, std::vector< int > &bunchCrossingList, std::vector< float > &TrueInteractionList, std::vector< edm::EventID > &eventList, int bunchSpace)

virtual	~DigiAccumulatorMixMod ()

Private Member Functions
template<class T >
void	accumulateEvent (const T &e, edm::EventSetup const &c, const edm::InputTag &label, const edm::InputTag &MVALabel)

Private Attributes
std::string	MVAOutputLabel

std::auto_ptr < TrackingRecHitCollection >	newHits_

std::vector< float >	newMVAVals_

std::auto_ptr < reco::TrackExtraCollection >	newTrackExtras_

std::auto_ptr < reco::TrackCollection >	newTracks_

std::string	outputLabel

edm::InputTag	pileUpMVAValuesTag

edm::InputTag	pileUpTracksTag

TrackingRecHitRefProd	rNewHits

reco::TrackExtraRefProd	rNewTrackExtras

reco::TrackRefProd	rNewTracks

edm::InputTag	signalMVAValuesTag

edm::InputTag	signalTracksTag

Detailed Description

RecoTrackAccumulator accumulates generalTracks from the hard and the pileup events

Author: Andrea Giammanco

Version: Mar 11 2013

Definition at line 36 of file RecoTrackAccumulator.h.

Constructor & Destructor Documentation

RecoTrackAccumulator::RecoTrackAccumulator	(	const edm::ParameterSet &	conf,
		edm::one::EDProducerBase &	mixMod,
		edm::ConsumesCollector &	iC
	)

explicit

Definition at line 9 of file RecoTrackAccumulator.cc.

References edm::ConsumesCollector::consumes(), MVAOutputLabel, outputLabel, signalMVAValuesTag, and signalTracksTag.

                                                                                                                               :
   signalTracksTag(conf.getParameter<edm::InputTag>("signalTracks")),
   signalMVAValuesTag(conf.getParameter<edm::InputTag>("signalMVAValues")),
   pileUpTracksTag(conf.getParameter<edm::InputTag>("pileUpTracks")),
   pileUpMVAValuesTag(conf.getParameter<edm::InputTag>("pileUpMVAValues")),
   outputLabel(conf.getParameter<std::string>("outputLabel")),
   MVAOutputLabel(conf.getParameter<std::string>("MVAOutputLabel"))
 {
   mixMod.produces<reco::TrackCollection>(outputLabel);
   mixMod.produces<TrackingRecHitCollection>(outputLabel);
   mixMod.produces<reco::TrackExtraCollection>(outputLabel); 
   mixMod.produces<edm::ValueMap<float> >(MVAOutputLabel);
   
   iC.consumes<reco::TrackCollection>(signalTracksTag);
   iC.consumes<TrackingRecHitCollection>(signalTracksTag);
   iC.consumes<reco::TrackExtraCollection>(signalTracksTag);
   iC.consumes<edm::ValueMap<float> >(signalMVAValuesTag);
 }

RecoTrackAccumulator::~RecoTrackAccumulator ( )

virtual

Definition at line 28 of file RecoTrackAccumulator.cc.

                                             {
     
 }

Member Function Documentation

void RecoTrackAccumulator::accumulate	(	edm::Event const &	e,
		edm::EventSetup const &	c
	)

virtual

Implements DigiAccumulatorMixMod.

Definition at line 45 of file RecoTrackAccumulator.cc.

References accumulateEvent(), signalMVAValuesTag, and signalTracksTag.

                                                                                     {
   accumulateEvent( e, iSetup,signalTracksTag,signalMVAValuesTag);
 }

void RecoTrackAccumulator::accumulate	(	PileUpEventPrincipal const &	e,
		edm::EventSetup const &	c,
		edm::StreamID const &
	)

overridevirtual

Implements DigiAccumulatorMixMod.

Definition at line 49 of file RecoTrackAccumulator.cc.

References accumulateEvent(), PileUpEventPrincipal::bunchCrossing(), pileUpMVAValuesTag, and pileUpTracksTag.

                                                                                                                     {
   if (e.bunchCrossing()==0) {
     accumulateEvent( e, iSetup,pileUpTracksTag,pileUpMVAValuesTag);
   }
 }

template<class T >

void RecoTrackAccumulator::accumulateEvent	(	const T &	e,
		edm::EventSetup const &	c,
		const edm::InputTag &	label,
		const edm::InputTag &	MVALabel
	)

private

Definition at line 69 of file RecoTrackAccumulator.cc.

References cmsRelvalreport::exit, reco::Track::extra(), i, edm::HandleBase::id(), edm::Ref< C, T, F >::key(), diffTwoXMLs::label, newHits_, newMVAVals_, newTrackExtras_, newTracks_, rNewHits, rNewTrackExtras, tree::t, and testEve_cfg::tracks.

Referenced by accumulate().

                                                                                                                                                            {
 
   edm::Handle<reco::TrackCollection> tracks;
   edm::Handle<TrackingRecHitCollection> hits;
   edm::Handle<reco::TrackExtraCollection> trackExtras;
   edm::Handle<edm::ValueMap<float> > mvaVals;
   if(!(e.getByLabel(label, tracks) and e.getByLabel(label, hits) and e.getByLabel(label, trackExtras))){
     edm::LogError ("RecoTrackAccumulator") << "Failed to find track, hit or trackExtra collections with inputTag " << label;
     exit(1);
   }
   else if(!e.getByLabel(MVALabel,mvaVals)){
     edm::LogError ("RecoTrackAccumulator") << "Failed to find mva values with inputTag" << MVALabel;
     exit(1);
   }
 
   if(tracks->size() != mvaVals->size()){
     edm::LogError("RecoTrackAccumulator") << "RecoTrackAccumulator expects the track collection and the MVA values to have the same number of entries" << std::endl;
   }
   if (tracks->size()==0)
     return;
 
   // crude way to get the track product id
   // (for PU events, the usual reco::TrackRef(tracks,t) does not result in a proper product id in the TrackRef)
   edm::ProductID tracksProdId = mvaVals->begin().id();
   
   for (size_t t = 0; t < tracks->size();++t){
     const reco::Track & track = tracks->at(t);
     newTracks_->push_back(track);
     // track extras:
     auto const& extra = trackExtras->at(track.extra().key());
     newTrackExtras_->emplace_back(extra.outerPosition(), extra.outerMomentum(), extra.outerOk(),
                   extra.innerPosition(),extra.innerMomentum(), extra.innerOk(),
                   extra.outerStateCovariance(), extra.outerDetId(),
                   extra.innerStateCovariance(), extra.innerDetId(),
                   extra.seedDirection(),
                   //If TrajectorySeeds are needed, then their list must be gotten from the
                   // secondary event directly and looked up similarly to TrackExtras.
                   //We can't use a default constructed RefToBase due to a bug in RefToBase
                   // which causes an seg fault when calling isAvailable on a default constructed one.
                   edm::RefToBase<TrajectorySeed>{edm::Ref<std::vector<TrajectorySeed>>{}});
     newTracks_->back().setExtra( reco::TrackExtraRef( rNewTrackExtras, newTracks_->size() - 1) );
     // rechits:
     // note: extra.recHit(i) does not work for pileup events
     // probably the Ref does not know its product id applies on a pileup event
     auto & newExtra = newTrackExtras_->back();
     auto const firstTrackIndex = newHits_->size();
     for(unsigned int i = 0;i<extra.recHitsSize();i++){
       newHits_->push_back( (*hits)[extra.recHit(i).key()] );
     }
     newExtra.setHits( rNewHits, firstTrackIndex, newHits_->size() - firstTrackIndex);
     // crude way to get the mva value that belongs to the track
     // the usual way valueMap[trackref] does not work in PU events, 
     // see earlier comment about track id
     float _val = mvaVals->get(tracksProdId,t);
     newMVAVals_.push_back(_val);
   }
 }

void RecoTrackAccumulator::finalizeEvent	(	edm::Event &	e,
		edm::EventSetup const &	c
	)

virtual

Implements DigiAccumulatorMixMod.

Definition at line 55 of file RecoTrackAccumulator.cc.

References edm::helper::Filler< Map >::fill(), edm::RefProd< T >::id(), edm::helper::Filler< Map >::insert(), MVAOutputLabel, newHits_, newMVAVals_, newTrackExtras_, newTracks_, outputLabel, edm::Event::put(), and rNewTracks.

                                                                                  {
   
   std::auto_ptr< edm::ValueMap<float> > _newMVAVals(new edm::ValueMap<float>);
   edm::ValueMap<float>::Filler filler(*_newMVAVals);
   edm::TestHandle<reco::TrackCollection> newTracksHandle(newTracks_.get(),rNewTracks.id());
   filler.insert(newTracksHandle,newMVAVals_.begin(),newMVAVals_.end());
   filler.fill();
   e.put( newTracks_, outputLabel );
   e.put( newHits_, outputLabel );
   e.put( newTrackExtras_, outputLabel );
   e.put(_newMVAVals,MVAOutputLabel);
 }

void RecoTrackAccumulator::initializeEvent	(	edm::Event const &	e,
		edm::EventSetup const &	c
	)

virtual

Implements DigiAccumulatorMixMod.

Definition at line 32 of file RecoTrackAccumulator.cc.

References alignCSCRings::e, newHits_, newMVAVals_, newTrackExtras_, newTracks_, outputLabel, rNewHits, rNewTrackExtras, and rNewTracks.

                                                                                          {
     
   newTracks_ = std::auto_ptr<reco::TrackCollection>(new reco::TrackCollection);
   newHits_ = std::auto_ptr<TrackingRecHitCollection>(new TrackingRecHitCollection);
   newTrackExtras_ = std::auto_ptr<reco::TrackExtraCollection>(new reco::TrackExtraCollection);
   newMVAVals_.clear();
   
   // this is needed to get the ProductId of the TrackExtra and TrackingRecHit and Track collections
   rNewTracks=const_cast<edm::Event&>( e ).getRefBeforePut<reco::TrackCollection>(outputLabel);
   rNewTrackExtras=const_cast<edm::Event&>( e ).getRefBeforePut<reco::TrackExtraCollection>(outputLabel);
   rNewHits=const_cast<edm::Event&>( e ).getRefBeforePut<TrackingRecHitCollection>(outputLabel);
 }