QualityFilter Class Reference

Inheritance diagram for QualityFilter:

Public Member Functions
	QualityFilter (const edm::ParameterSet &)
	~QualityFilter () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
bool	hasAbilityToProduceInLumis () const final
bool	hasAbilityToProduceInRuns () const final

Private Member Functions
virtual void	endJob ()
void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
bool	copyExtras_
edm::EDGetTokenT< TrajTrackAssociationCollection >	tassTag
reco::TrackBase::TrackQuality	trackQuality_
edm::EDGetTokenT< std::vector< Trajectory > >	trajTag

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

Definition at line 11 of file QualityFilter.cc.

Constructor & Destructor Documentation

QualityFilter::QualityFilter ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 45 of file QualityFilter.cc.

References edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), and AlCaHLTBitMon_QueryRunRegistry::string.

{
  copyExtras_ = iConfig.getUntrackedParameter<bool>("copyExtras", false);

  produces<reco::TrackCollection>();
  if (copyExtras_) {
      produces<TrackingRecHitCollection>();
      produces<reco::TrackExtraCollection>();
  }
  produces<std::vector<Trajectory> >();
  produces<TrajTrackAssociationCollection>();

  trajTag = consumes<std::vector<Trajectory> >(iConfig.getParameter<edm::InputTag>("recTracks"));
  tassTag = consumes<TrajTrackAssociationCollection>(iConfig.getParameter<edm::InputTag>("recTracks"));
  trackQuality_=TrackBase::qualityByName(iConfig.getParameter<std::string>("TrackQuality"));
}

QualityFilter::~QualityFilter ( )

override

Definition at line 63 of file QualityFilter.cc.

{
 
   // do anything here that needs to be done at desctruction time
   // (e.g. close files, deallocate resources etc.)

}

Member Function Documentation

void QualityFilter::endJob ( void  )

privatevirtual

Definition at line 189 of file QualityFilter.cc.

References DEFINE_FWK_MODULE.

                      {
}

void QualityFilter::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

overrideprivate

Definition at line 78 of file QualityFilter.cc.

References edm::AssociationMap< Tag >::begin(), edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::TrackCollection, unsigned short > >::const_iterator, edm::AssociationMap< Tag >::end(), edm::Event::getByToken(), edm::Event::getRefBeforePut(), hfClusterShapes_cfi::hits, edm::Ref< C, T, F >::key(), eostools::move(), nTracks(), edm::Event::put(), and HiIsolationCommonParameters_cff::track.

{

  
  unique_ptr<TrackCollection> selTracks(new TrackCollection);
  unique_ptr<TrackingRecHitCollection> selHits(copyExtras_ ? new TrackingRecHitCollection() : nullptr);
  unique_ptr<TrackExtraCollection> selTrackExtras(copyExtras_ ? new TrackExtraCollection() : nullptr);
  unique_ptr<vector<Trajectory> > outputTJ(new vector<Trajectory> );
  unique_ptr<TrajTrackAssociationCollection> trajTrackMap( new TrajTrackAssociationCollection() );
  
  
  TrackExtraRefProd rTrackExtras; 
  TrackingRecHitRefProd rHits;
  if (copyExtras_) {
    rTrackExtras = iEvent.getRefBeforePut<TrackExtraCollection>();
    rHits = iEvent.getRefBeforePut<TrackingRecHitCollection>();
  }

  Handle<std::vector<Trajectory> > TrajectoryCollection;
  Handle<TrajTrackAssociationCollection> assoMap;
  
  iEvent.getByToken(trajTag,TrajectoryCollection);
  iEvent.getByToken(tassTag,assoMap);



  TrajTrackAssociationCollection::const_iterator it = assoMap->begin();
  TrajTrackAssociationCollection::const_iterator lastAssoc = assoMap->end();
  for( ; it != lastAssoc; ++it ) {
    const Ref<vector<Trajectory> > traj = it->key;
    const reco::TrackRef itc = it->val;
    bool goodTk = (itc->quality(trackQuality_));
 
    
    if (goodTk){
      auto const & track =(*itc);
      //tracks and trajectories
      selTracks->push_back( track );
      outputTJ->push_back( *traj );
      if (copyExtras_) {
          //TRACKING HITS
          trackingRecHit_iterator irhit   =(*itc).recHitsBegin();
          trackingRecHit_iterator lasthit =(*itc).recHitsEnd();
          for (; irhit!=lasthit; ++irhit) {
            selHits->push_back((*irhit)->clone() );
          }
      }
          
    }

  }

  unsigned nTracks = selTracks->size();
  if (copyExtras_) {
      //PUT TRACKING HITS IN THE EVENT
    OrphanHandle<TrackingRecHitCollection> theRecoHits = iEvent.put(std::move(selHits) );
    edm::RefProd<TrackingRecHitCollection> theRecoHitsProd(theRecoHits);

      //PUT TRACK EXTRA IN THE EVENT
      selTrackExtras->reserve(nTracks);
      unsigned hits=0;

      for ( unsigned index = 0; index<nTracks; ++index ) { 
        auto const & aTrack = (*selTracks)[index];
        selTrackExtras->emplace_back(aTrack.outerPosition(),
                               aTrack.outerMomentum(),
                               aTrack.outerOk(),
                               aTrack.innerPosition(),
                               aTrack.innerMomentum(),
                               aTrack.innerOk(),
                               aTrack.outerStateCovariance(),
                               aTrack.outerDetId(),
                               aTrack.innerStateCovariance(),
                               aTrack.innerDetId(),
                               aTrack.seedDirection(),
                               aTrack.seedRef()
                               );
            
        auto & aTrackExtra = selTrackExtras->back();
        //unsigned nHits = aTrack.numberOfValidHits();
        auto nHits = aTrack.recHitsSize();
        aTrackExtra.setHits(theRecoHitsProd, hits, nHits);
        hits += nHits;
        selTrackExtras->push_back(aTrackExtra);
      }

      //CORRECT REF TO TRACK
      OrphanHandle<TrackExtraCollection> theRecoTrackExtras = iEvent.put(std::move(selTrackExtras)); 
      for ( unsigned index = 0; index<nTracks; ++index ) { 
        const reco::TrackExtraRef theTrackExtraRef(theRecoTrackExtras,index);
          (*selTracks)[index].setExtra(theTrackExtraRef);
      }
  } // END IF COPY EXTRAS

  //TRACKS AND TRAJECTORIES
  OrphanHandle<TrackCollection> theRecoTracks = iEvent.put(std::move(selTracks));
  OrphanHandle<vector<Trajectory> > theRecoTrajectories = iEvent.put(std::move(outputTJ));

  //TRACKS<->TRAJECTORIES MAP 
  nTracks = theRecoTracks->size();
  for ( unsigned index = 0; index<nTracks; ++index ) { 
    Ref<vector<Trajectory> > trajRef( theRecoTrajectories, index );
    Ref<TrackCollection>    tkRef( theRecoTracks, index );
    trajTrackMap->insert(trajRef,tkRef);
  }
  //MAP IN THE EVENT
  iEvent.put( std::move(trajTrackMap) );
}

Member Data Documentation

bool QualityFilter::copyExtras_

private

Definition at line 26 of file QualityFilter.cc.

edm::EDGetTokenT<TrajTrackAssociationCollection> QualityFilter::tassTag

private

Definition at line 24 of file QualityFilter.cc.

reco::TrackBase::TrackQuality QualityFilter::trackQuality_

private

Definition at line 25 of file QualityFilter.cc.

edm::EDGetTokenT<std::vector<Trajectory> > QualityFilter::trajTag

private

Definition at line 23 of file QualityFilter.cc.