TTTrackAssociationMap.h

Go to the documentation of this file.

#ifndef L1_TRACK_TRIGGER_TRACK_ASSOCIATION_FORMAT_H
#define L1_TRACK_TRIGGER_TRACK_ASSOCIATION_FORMAT_H

#include "DataFormats/Common/interface/Ref.h"
#include "DataFormats/Common/interface/Ptr.h"
#include "DataFormats/Common/interface/DetSet.h"
#include "DataFormats/Common/interface/DetSetVector.h"
#include "DataFormats/Common/interface/DetSetVectorNew.h"
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/Phase2TrackerDigi/interface/Phase2TrackerDigi.h"
#include "DataFormats/GeometryCommonDetAlgo/interface/MeasurementPoint.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h" 
#include "DataFormats/L1TrackTrigger/interface/TTTypes.h"
#include "SimDataFormats/TrackerDigiSimLink/interface/PixelDigiSimLink.h"
#include "SimDataFormats/Track/interface/SimTrack.h"
#include "SimDataFormats/Track/interface/SimTrackContainer.h"
#include "SimDataFormats/EncodedEventId/interface/EncodedEventId.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"
#include "SimTracker/TrackTriggerAssociation/interface/TTStubAssociationMap.h"

template< typename T >
class TTTrackAssociationMap
{
  public:
    TTTrackAssociationMap();

    ~TTTrackAssociationMap();


    std::map< edm::Ptr< TTTrack< T > >, edm::Ptr< TrackingParticle > >                getTTTrackToTrackingParticleMap() const
      { return trackToTrackingParticleMap; }
    std::map< edm::Ptr< TrackingParticle >, std::vector< edm::Ptr< TTTrack< T > > > > getTrackingParticleToTTTracksMap() const
      { return trackingParticleToTrackVectorMap; }

    void setTTTrackToTrackingParticleMap( std::map< edm::Ptr< TTTrack< T > >, edm::Ptr< TrackingParticle > > aMap )
      { trackToTrackingParticleMap = aMap; }
    void setTrackingParticleToTTTracksMap( std::map< edm::Ptr< TrackingParticle >, std::vector< edm::Ptr< TTTrack< T > > > > aMap )
      { trackingParticleToTrackVectorMap = aMap; }
    void setTTStubAssociationMap( edm::RefProd< TTStubAssociationMap< T > > aStubAssoMap )
      { theStubAssociationMap = aStubAssoMap; }

    edm::Ptr< TrackingParticle >            findTrackingParticlePtr( edm::Ptr< TTTrack< T > > aTrack ) const;
    std::vector< edm::Ptr< TTTrack< T > > > findTTTrackPtrs( edm::Ptr< TrackingParticle > aTrackingParticle ) const; 

    bool isGenuine( edm::Ptr< TTTrack< T > > aTrack ) const;
    bool isLooselyGenuine( edm::Ptr< TTTrack< T > > aTrack ) const;
    bool isCombinatoric( edm::Ptr< TTTrack< T > > aTrack ) const;
    bool isUnknown( edm::Ptr< TTTrack< T > > aTrack ) const;

  private:
    std::map< edm::Ptr< TTTrack< T > >, edm::Ptr< TrackingParticle > >                trackToTrackingParticleMap;
    std::map< edm::Ptr< TrackingParticle >, std::vector< edm::Ptr< TTTrack< T > > > > trackingParticleToTrackVectorMap;
    edm::RefProd< TTStubAssociationMap< T > >                                         theStubAssociationMap;

}; 

template< typename T >
TTTrackAssociationMap< T >::TTTrackAssociationMap()
{
  trackToTrackingParticleMap.clear();
  trackingParticleToTrackVectorMap.clear();
}

template< typename T >
TTTrackAssociationMap< T >::~TTTrackAssociationMap(){}

template< >
edm::Ptr< TrackingParticle > TTTrackAssociationMap< Ref_Phase2TrackerDigi_ >::findTrackingParticlePtr( edm::Ptr< TTTrack< Ref_Phase2TrackerDigi_ > > aTrack ) const;

template< >
std::vector< edm::Ptr< TTTrack< Ref_Phase2TrackerDigi_ > > > TTTrackAssociationMap< Ref_Phase2TrackerDigi_ >::findTTTrackPtrs( edm::Ptr< TrackingParticle > aTrackingParticle ) const;

template< >
bool TTTrackAssociationMap< Ref_Phase2TrackerDigi_ >::isLooselyGenuine( edm::Ptr< TTTrack< Ref_Phase2TrackerDigi_ > > aTrack ) const;

template< >
bool TTTrackAssociationMap< Ref_Phase2TrackerDigi_ >::isGenuine( edm::Ptr< TTTrack< Ref_Phase2TrackerDigi_ > > aTrack ) const;


template< >
bool TTTrackAssociationMap< Ref_Phase2TrackerDigi_ >::isCombinatoric( edm::Ptr< TTTrack< Ref_Phase2TrackerDigi_ > > aTrack ) const;

template< >
bool TTTrackAssociationMap< Ref_Phase2TrackerDigi_ >::isUnknown( edm::Ptr< TTTrack< Ref_Phase2TrackerDigi_ > > aTrack ) const;

#endif