TrackQuality.h

Go to the documentation of this file.

/*
 *  TrackQuality.h
 *
 *  Created by Christophe Saout on 9/25/08.
 *  2007 __MyCompanyName__. 
 *
 */

#ifndef TrackQuality_h
#define TrackQuality_h

#include <vector>
#include <memory>

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/Common/interface/DetSetVector.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"

#include "SimTracker/TrackerHitAssociation/interface/TrackerHitAssociator.h"

class TrackerTopology;

class TrackQuality
{
public:
    typedef std::vector<TrackingParticleRef> SimParticleTrail;

    struct Layer
    {
        enum SubDet
        {
            Invalid = 0,
            PixelBarrel, PixelForward,
            StripTIB, StripTID, StripTOB, StripTEC,
            MuonDT, MuonCSC, MuonRPCBarrel, MuonRPCEndcap
        };

        enum State
        {
            Unknown = 0,
            Good,
            Missed,
            Noise,
            Bad,
            Dead,
            Shared,
            Misassoc
        };

        struct Hit
        {
            short int recHitId;
            State state;
        };

        SubDet subDet;
        short int layer;
        std::vector<Hit> hits;
    };

public:
    /* Creates a TrackQuality object from a pset.

       /param[in] pset with the configuration values
    */
    TrackQuality(const edm::ParameterSet &, edm::ConsumesCollector& iC);

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

    void evaluate(SimParticleTrail const &, reco::TrackBaseRef const &, const TrackerTopology *tTopo);

    unsigned int numberOfLayers() const
    {
        return layers_.size();
    }

    const Layer &layer(unsigned int index) const
    {
        return layers_[index];
    }

private:
    TrackerHitAssociator::Config trackerHitAssociatorConfig_;
    std::unique_ptr<TrackerHitAssociator> associator_;

    std::vector<Layer> layers_;
};

#endif