---

TrackHistory.h

Go to the documentation of this file.

/*
 *  TrackHistory.h
 *
 *  Created by Victor Eduardo Bazterra on 7/13/07.
 *  Copyright 2007 __MyCompanyName__. All rights reserved.
 *
 */

#ifndef TrackHistory_h
#define TrackHistory_h

#include "DataFormats/TrackReco/interface/TrackFwd.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticleFwd.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingVertex.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingVertexContainer.h"

#include "SimTracker/TrackAssociation/interface/TrackAssociatorBase.h"

class TrackHistory
{

public:

    typedef std::vector<const HepMC::GenParticle *> GenParticleTrail;

    typedef std::vector<const HepMC::GenVertex *> GenVertexTrail;

    typedef std::vector<TrackingParticleRef> SimParticleTrail;

    typedef std::vector<TrackingVertexRef> SimVertexTrail;

    /* Creates a TrackHistory with association given by pset.

       /param[in] pset with the configuration values
    */
    TrackHistory(const edm::ParameterSet &);

    void newEvent(const edm::Event &, const edm::EventSetup &);

    /* Set TrackHistory to given depth. Positive values
       constrain the number of TrackingVertex visit in the history.
       Negatives values set the limit of the iteration over generated
       information i.e. (-1 -> status 1 or -2 -> status 2 particles).

       /param[in] depth the history
    */
    void depth(int d)
    {
        depth_ = d;
    }

    /* Return false when the history cannot be determined upto a given depth.
       If not depth is pass to the function no restriction are apply to it.

       /param[in] TrackingParticleRef of a simulated track
       /param[in] depth of the track history
       /param[out] boolean that is true when history can be determined
    */
    bool evaluate(TrackingParticleRef tpr)
    {
        resetTrails(tpr);
        return traceSimHistory(tpr, depth_);
    }

    /* Return false when the track association is not possible (fake track).

       /param[in] TrackRef to a reco::track
       /param[out] boolean that is false when a fake track is detected
    */
    bool evaluate (reco::TrackBaseRef);

    const TrackingParticleRef & simParticle() const
    {
        return simParticleTrail_[0];
    }

    const HepMC::GenParticle * genParticle() const
    {
        if ( genParticleTrail_.empty() ) return 0;
        return genParticleTrail_[genParticleTrail_.size()-1];
    }

    const SimVertexTrail & simVertexTrail() const
    {
        return simVertexTrail_;
    }

    const SimParticleTrail & simParticleTrail() const
    {
        return simParticleTrail_;
    }

    const GenVertexTrail & genVertexTrail() const
    {
        return genVertexTrail_;
    }

    const GenParticleTrail & genParticleTrail() const
    {
        return genParticleTrail_;
    }

protected:

    int depth_;

    GenVertexTrail genVertexTrail_;
    GenParticleTrail genParticleTrail_;
    SimVertexTrail simVertexTrail_;
    SimParticleTrail simParticleTrail_;

private:

    bool newEvent_;

    void resetTrails(TrackingParticleRef tpr)
    {
        // save the initial particle in the trail
        simParticleTrail_.clear();
        simParticleTrail_.push_back(tpr);

        // clear the remaining trails
        simVertexTrail_.clear();
        genVertexTrail_.clear();
        genParticleTrail_.clear();
    }

    void traceGenHistory (const HepMC::GenParticle *);

    bool traceSimHistory (TrackingParticleRef, int);

    bool bestMatchByMaxValue_;

    edm::InputTag trackProducer_;

    edm::InputTag trackingTruth_;

    std::string trackAssociator_;

    reco::RecoToSimCollection association_;

    TrackingParticleRef match ( reco::TrackBaseRef );
};

#endif

---