TrackHistory Class Reference

This class traces the simulated and generated history of a given track. More...

#include <TrackHistory.h>

Inheritance diagram for TrackHistory:

Public Member Functions
bool	evaluate (TrackingParticleRef tpr)
	Evaluate track history using a TrackingParticleRef. More...
bool	evaluate (reco::TrackBaseRef)
	Evaluate reco::Track history using a given association. More...
const std::pair < TrackingParticleRef, double >	getMatchedTrackingParticle () const
void	newEvent (const edm::Event &, const edm::EventSetup &)
	Pre-process event information (for accessing reconstruction information) More...
double	quality () const
const reco::TrackBaseRef &	recoTrack () const
	Return a reference to the reconstructed track. More...
	TrackHistory (const edm::ParameterSet &, edm::ConsumesCollector &&)
	Constructor by pset. More...
Public Member Functions inherited from HistoryBase
void	depth (int d)
	Set the depth of the history. More...
const HepMC::GenParticle *	genParticle () const
	Returns a pointer to most primitive status 1 or 2 particle in the genParticleTrail_. More...
GenParticleTrail const &	genParticleTrail () const
	Return all generated particle (HepMC::GenParticle) in the history. More...
GenVertexTrail const &	genVertexTrail () const
	Return all generated vertex in the history. More...
	HistoryBase ()
const reco::GenParticle *	recoGenParticle () const
	Returns a pointer to most primitive status 1 or 2 particle in the recoGenParticleTrail_. More...
RecoGenParticleTrail const &	recoGenParticleTrail () const
	Return all reco::GenParticle in the history. More...
const TrackingParticleRef &	simParticle () const
	Return the initial tracking particle from the history. More...
SimParticleTrail const &	simParticleTrail () const
	Return all the simulated particle in the history. More...
const TrackingVertexRef &	simVertex () const
	Return the initial tracking vertex from the history. More...
SimVertexTrail const &	simVertexTrail () const
	Return all the simulated vertices in the history. More...

Private Attributes
bool	bestMatchByMaxValue_
bool	enableRecoToSim_
bool	enableSimToReco_
bool	newEvent_
double	quality_
reco::RecoToSimCollection	recoToSim_
reco::TrackBaseRef	recotrack_
reco::SimToRecoCollection	simToReco_
edm::InputTag	trackAssociator_
TrackingParticleRef	trackingParticle_
edm::InputTag	trackingTruth_
edm::InputTag	trackProducer_

Additional Inherited Members
Public Types inherited from HistoryBase
typedef std::vector< const HepMC::GenParticle * >	GenParticleTrail
	HepMC::GenParticle trail type. More...
typedef std::vector< const HepMC::GenVertex * >	GenVertexTrail
	GenVertex trail type. More...
typedef std::set< const HepMC::GenVertex * >	GenVertexTrailHelper
	GenVertex trail helper type. More...
typedef std::vector< const reco::GenParticle * >	RecoGenParticleTrail
	reco::GenParticle trail type. More...
typedef std::set< const reco::GenParticle * >	RecoGenParticleTrailHelper
	reco::GenParticle trail helper type. More...
typedef std::vector < TrackingParticleRef >	SimParticleTrail
	SimParticle trail type. More...
typedef std::vector < TrackingVertexRef >	SimVertexTrail
	SimVertex trail type. More...
Protected Member Functions inherited from HistoryBase
bool	evaluate (TrackingParticleRef tpr)
	Evaluate track history using a TrackingParticleRef. More...
bool	evaluate (TrackingVertexRef tvr)
	Evaluate track history using a TrackingParticleRef. More...
Protected Attributes inherited from HistoryBase
GenParticleTrail	genParticleTrail_
GenVertexTrail	genVertexTrail_
GenVertexTrailHelper	genVertexTrailHelper_
RecoGenParticleTrail	recoGenParticleTrail_
RecoGenParticleTrailHelper	recoGenParticleTrailHelper_
SimParticleTrail	simParticleTrail_
SimVertexTrail	simVertexTrail_

Detailed Description

This class traces the simulated and generated history of a given track.

Definition at line 17 of file TrackHistory.h.

Constructor & Destructor Documentation

TrackHistory::TrackHistory	(	const edm::ParameterSet &	config,
		edm::ConsumesCollector &&	collector
	)

Constructor by pset.

Definition at line 6 of file TrackHistory.cc.

References bestMatchByMaxValue_, enableRecoToSim_, enableSimToReco_, edm::ParameterSet::getUntrackedParameter(), or, quality_, trackAssociator_, trackingTruth_, and trackProducer_.

  : HistoryBase()
{
    // Name of the track collection
    trackProducer_ = config.getUntrackedParameter<edm::InputTag> ( "trackProducer" );
    collector.consumes<edm::View<reco::Track> >(trackProducer_);

    // Name of the traking pariticle collection
    trackingTruth_ = config.getUntrackedParameter<edm::InputTag> ( "trackingTruth" );
    collector.consumes<TrackingParticleCollection>(trackingTruth_);

    // Track association record
    trackAssociator_ = config.getUntrackedParameter<edm::InputTag> ( "trackAssociator" );

    // Association by max. value
    bestMatchByMaxValue_ = config.getUntrackedParameter<bool> ( "bestMatchByMaxValue" );

    // Enable RecoToSim association
    enableRecoToSim_ = config.getUntrackedParameter<bool> ( "enableRecoToSim" );

    // Enable SimToReco association
    enableSimToReco_ = config.getUntrackedParameter<bool> ( "enableSimToReco" );

    if(enableRecoToSim_ or enableSimToReco_) {
      collector.consumes<reco::TrackToTrackingParticleAssociator>(trackAssociator_);
    }

    quality_ = 0.;
}

Member Function Documentation

bool TrackHistory::evaluate ( TrackingParticleRef  tpr )

inline

Evaluate track history using a TrackingParticleRef.

Definition at line 41 of file TrackHistory.h.

References bestMatchByMaxValue_, enableSimToReco_, HistoryBase::evaluate(), match(), quality_, recotrack_, mps_fire::result, and simToReco_.

Referenced by TrackClassifier::evaluate(), and GenTrackMatcher::produce().

    {
        if ( enableSimToReco_ )
        {
            std::pair<reco::TrackBaseRef, double> result =  match(tpr, simToReco_, bestMatchByMaxValue_);
            recotrack_ = result.first;
            quality_ =  result.second;
        }
        return HistoryBase::evaluate(tpr);
    }

bool TrackHistory::evaluate

(

reco::TrackBaseRef

tr

)

Evaluate reco::Track history using a given association.

Definition at line 66 of file TrackHistory.cc.

References bestMatchByMaxValue_, enableRecoToSim_, HistoryBase::evaluate(), edm::Ref< C, T, F >::isNull(), match(), quality_, recoToSim_, recotrack_, mps_fire::result, and trackingParticle_.

{
    if ( !enableRecoToSim_ ) return false;

    std::pair<TrackingParticleRef, double> result =  match(tr, recoToSim_, bestMatchByMaxValue_);

    TrackingParticleRef tpr( result.first );
    quality_ = result.second;
    trackingParticle_ = tpr;

    if ( !tpr.isNull() )
    {
        HistoryBase::evaluate(tpr);

        recotrack_ = tr;

        return true;
    }

    return false;
}

const std::pair<TrackingParticleRef, double> TrackHistory::getMatchedTrackingParticle ( ) const

inline

Definition at line 68 of file TrackHistory.h.

References quality_, mps_fire::result, and trackingParticle_.

    {
        std::pair<TrackingParticleRef, double> result;
        result.first = trackingParticle_;
        result.second = quality_;
        
        return result;
    }

void TrackHistory::newEvent	(	const edm::Event &	event,
		const edm::EventSetup &	setup
	)

Pre-process event information (for accessing reconstruction information)

Definition at line 39 of file TrackHistory.cc.

References enableRecoToSim_, enableSimToReco_, recoToSim_, simToReco_, ecalTPGAnalyzer_cfg::TPCollection, trackAssociator_, HLT_FULL_cff::trackCollection, trackingTruth_, and trackProducer_.

Referenced by TrackClassifier::newEvent(), and GenTrackMatcher::produce().

{
    if ( enableRecoToSim_ || enableSimToReco_ )
    {
        // Track collection
        edm::Handle<edm::View<reco::Track> > trackCollection;
        event.getByLabel(trackProducer_, trackCollection);

        // Tracking particle information
        edm::Handle<TrackingParticleCollection>  TPCollection;
        event.getByLabel(trackingTruth_, TPCollection);

        // Get the track associator
        edm::Handle<reco::TrackToTrackingParticleAssociator> associator;
        event.getByLabel(trackAssociator_, associator);

        // Calculate the map between recotracks -> tp
        if ( enableRecoToSim_ ) recoToSim_ = associator->associateRecoToSim(trackCollection, TPCollection);

        // Calculate the map between recotracks <- tp
        if ( enableSimToReco_ ) simToReco_ = associator->associateSimToReco(trackCollection, TPCollection );
    }
}

double TrackHistory::quality ( void  ) const

inline

Definition at line 77 of file TrackHistory.h.

References quality_.

    {
        return quality_;
    }

const reco::TrackBaseRef& TrackHistory::recoTrack ( ) const

inline

Return a reference to the reconstructed track.

Definition at line 62 of file TrackHistory.h.

References recotrack_.

Referenced by TrackClassifier::evaluate().

    {
        return recotrack_;
    }

Member Data Documentation

bool TrackHistory::bestMatchByMaxValue_

private

Definition at line 86 of file TrackHistory.h.

Referenced by evaluate(), and TrackHistory().

bool TrackHistory::enableRecoToSim_

private

Definition at line 88 of file TrackHistory.h.

Referenced by evaluate(), newEvent(), and TrackHistory().

bool TrackHistory::enableSimToReco_

private

Definition at line 88 of file TrackHistory.h.

Referenced by evaluate(), newEvent(), and TrackHistory().

bool TrackHistory::newEvent_

private

Definition at line 84 of file TrackHistory.h.

double TrackHistory::quality_

private

Definition at line 90 of file TrackHistory.h.

Referenced by evaluate(), getMatchedTrackingParticle(), quality(), and TrackHistory().

reco::RecoToSimCollection TrackHistory::recoToSim_

private

Definition at line 102 of file TrackHistory.h.

Referenced by evaluate(), and newEvent().

reco::TrackBaseRef TrackHistory::recotrack_

private

Definition at line 98 of file TrackHistory.h.

Referenced by evaluate(), and recoTrack().

reco::SimToRecoCollection TrackHistory::simToReco_

private

Definition at line 104 of file TrackHistory.h.

Referenced by evaluate(), and newEvent().

edm::InputTag TrackHistory::trackAssociator_

private

Definition at line 96 of file TrackHistory.h.

Referenced by newEvent(), and TrackHistory().

TrackingParticleRef TrackHistory::trackingParticle_

private

Definition at line 100 of file TrackHistory.h.

Referenced by evaluate(), and getMatchedTrackingParticle().

edm::InputTag TrackHistory::trackingTruth_

private

Definition at line 94 of file TrackHistory.h.

Referenced by newEvent(), and TrackHistory().

edm::InputTag TrackHistory::trackProducer_

private

Definition at line 92 of file TrackHistory.h.

Referenced by newEvent(), and TrackHistory().