TTTrackAssociationMap.h

Go to the documentation of this file.

#ifndef SimDataFormats_Associations_TTTrackAssociationMap_h
#define SimDataFormats_Associations_TTTrackAssociationMap_h

#include "DataFormats/Common/interface/Ref.h"
#include "DataFormats/Common/interface/Ptr.h"
#include "DataFormats/L1TrackTrigger/interface/TTTypes.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticleFwd.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 "SimDataFormats/Track/interface/SimTrack.h"
#include "SimDataFormats/Track/interface/SimTrackContainer.h"
#include "SimDataFormats/EncodedEventId/interface/EncodedEventId.h"
#include "SimDataFormats/TrackingAnalysis/interface/TrackingParticle.h"
#include "SimDataFormats/Associations/interface/TTStubAssociationMap.h"

// Templated aliases
template <typename T>
using MapL1TrackToTP = std::map<TTTrackPtrT<T>, TrackingParticlePtr>;
template <typename T>
using MapTPToVecL1Track = std::map<TrackingParticlePtr, std::vector<TTTrackPtrT<T>>>;

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

  ~TTTrackAssociationMap();


  const MapL1TrackToTP<T>& getTTTrackToTrackingParticleMap() const { return trackToTrackingParticleMap_; }
  const MapTPToVecL1Track<T>& getTrackingParticleToTTTracksMap() const { return trackingParticleToTrackVectorMap_; }

  void setTTTrackToTrackingParticleMap(const MapL1TrackToTP<T>& aMap) { trackToTrackingParticleMap_ = aMap; }
  void setTrackingParticleToTTTracksMap(const MapTPToVecL1Track<T>& aMap) { trackingParticleToTrackVectorMap_ = aMap; }

  void setTTStubAssociationMap(edm::RefProd<TTStubAssociationMap<T>> aStubAssoMap) {
    theStubAssociationMap_ = aStubAssoMap;
  }

  const TrackingParticlePtr& findTrackingParticlePtr(TTTrackPtrT<T> aTrack) const;

  const std::vector<TTTrackPtrT<T>>& findTTTrackPtrs(TrackingParticlePtr aTrackingParticle) const;


  bool isGenuine(TTTrackPtrT<T> aTrack) const;
  bool isLooselyGenuine(TTTrackPtrT<T> aTrack) const;
  bool isUnknown(TTTrackPtrT<T> aTrack) const;
  bool isCombinatoric(TTTrackPtrT<T> aTrack) const;

  // Cfg param allowing one incorrect 2S stub in "genuine" tracks.
  void setAllowOneFalse2SStub(bool allowFalse2SStub);
  bool getAllowOneFalse2SStub();

private:
  MapL1TrackToTP<T> trackToTrackingParticleMap_;
  MapTPToVecL1Track<T> trackingParticleToTrackVectorMap_;
  edm::RefProd<TTStubAssociationMap<T>> theStubAssociationMap_;

  bool AllowOneFalse2SStub;

  // Allow functions to return reference to null.
  static const TrackingParticlePtr nullTrackingParticlePtr_;
  static const std::vector<TTTrackPtr> nullVecTTTrackPtr_;

};  

// Static constant data members.
template <typename T>
const TrackingParticlePtr TTTrackAssociationMap<T>::nullTrackingParticlePtr_;
template <typename T>
const std::vector<TTTrackPtr> TTTrackAssociationMap<T>::nullVecTTTrackPtr_;

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

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

template <>
const TrackingParticlePtr& TTTrackAssociationMap<Ref_Phase2TrackerDigi_>::findTrackingParticlePtr(
    TTTrackPtr aTrack) const;

template <>
const std::vector<TTTrackPtr>& TTTrackAssociationMap<Ref_Phase2TrackerDigi_>::findTTTrackPtrs(
    TrackingParticlePtr aTrackingParticle) const;

template <>
bool TTTrackAssociationMap<Ref_Phase2TrackerDigi_>::isLooselyGenuine(TTTrackPtr aTrack) const;

template <>
bool TTTrackAssociationMap<Ref_Phase2TrackerDigi_>::isGenuine(TTTrackPtr aTrack) const;

template <>
bool TTTrackAssociationMap<Ref_Phase2TrackerDigi_>::isCombinatoric(TTTrackPtr aTrack) const;

template <>
bool TTTrackAssociationMap<Ref_Phase2TrackerDigi_>::isUnknown(TTTrackPtr aTrack) const;

#endif