TTStubAssociationMap.h

Go to the documentation of this file.

#ifndef L1_TRACK_TRIGGER_STUB_ASSOCIATION_FORMAT_H
#define L1_TRACK_TRIGGER_STUB_ASSOCIATION_FORMAT_H

#include "DataFormats/Common/interface/Ref.h"
#include "DataFormats/Common/interface/RefProd.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/SiPixelDigi/interface/PixelDigi.h"
#include "DataFormats/GeometryCommonDetAlgo/interface/MeasurementPoint.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"  
#include "DataFormats/L1TrackTrigger/interface/TTStub.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/TTClusterAssociationMap.h"

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

  ~TTStubAssociationMap();


  std::map<edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> >, edm::Ptr<TrackingParticle> >
  getTTStubToTrackingParticleMap() const {
    return stubToTrackingParticleMap;
  }
  std::map<edm::Ptr<TrackingParticle>, std::vector<edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> > > >
  getTrackingParticleToTTStubsMap() const {
    return trackingParticleToStubVectorMap;
  }

  void setTTStubToTrackingParticleMap(
      std::map<edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> >, edm::Ptr<TrackingParticle> > aMap) {
    stubToTrackingParticleMap = aMap;
  }
  void setTrackingParticleToTTStubsMap(
      std::map<edm::Ptr<TrackingParticle>, std::vector<edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> > > > aMap) {
    trackingParticleToStubVectorMap = aMap;
  }
  void setTTClusterAssociationMap(edm::RefProd<TTClusterAssociationMap<T> > aCluAssoMap) {
    theClusterAssociationMap = aCluAssoMap;
  }

  edm::Ptr<TrackingParticle> findTrackingParticlePtr(edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> > aStub) const;
  std::vector<edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> > > findTTStubRefs(
      edm::Ptr<TrackingParticle> aTrackingParticle) const;

  bool isGenuine(edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> > aStub) const;
  bool isCombinatoric(edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> > aStub) const;
  bool isUnknown(edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> > aStub) const;

private:
  std::map<edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> >, edm::Ptr<TrackingParticle> >
      stubToTrackingParticleMap;
  std::map<edm::Ptr<TrackingParticle>, std::vector<edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> > > >
      trackingParticleToStubVectorMap;
  edm::RefProd<TTClusterAssociationMap<T> > theClusterAssociationMap;

};  

template <typename T>
TTStubAssociationMap<T>::TTStubAssociationMap() {
  stubToTrackingParticleMap.clear();
  trackingParticleToStubVectorMap.clear();
  edm::RefProd<TTClusterAssociationMap<T> >* aRefProd = new edm::RefProd<TTClusterAssociationMap<T> >();
  theClusterAssociationMap = *aRefProd;
}

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

template <typename T>
edm::Ptr<TrackingParticle> TTStubAssociationMap<T>::findTrackingParticlePtr(
    edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> > aStub) const {
  if (stubToTrackingParticleMap.find(aStub) != stubToTrackingParticleMap.end()) {
    return stubToTrackingParticleMap.find(aStub)->second;
  }

  //edm::Ptr< TrackingParticle >* temp = new edm::Ptr< TrackingParticle >();
  return edm::Ptr<TrackingParticle>();
}

template <typename T>
std::vector<edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> > > TTStubAssociationMap<T>::findTTStubRefs(
    edm::Ptr<TrackingParticle> aTrackingParticle) const {
  if (trackingParticleToStubVectorMap.find(aTrackingParticle) != trackingParticleToStubVectorMap.end()) {
    return trackingParticleToStubVectorMap.find(aTrackingParticle)->second;
  }

  std::vector<edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> > > tempVector;
  tempVector.clear();
  return tempVector;
}

template <typename T>
bool TTStubAssociationMap<T>::isGenuine(edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> > aStub) const {
  if ((this->findTrackingParticlePtr(aStub)).isNull())
    return false;

  return true;
}

template <typename T>
bool TTStubAssociationMap<T>::isCombinatoric(edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> > aStub) const {
  if (this->isGenuine(aStub))
    return false;

  if (this->isUnknown(aStub))
    return false;

  return true;
}

template <typename T>
bool TTStubAssociationMap<T>::isUnknown(edm::Ref<edmNew::DetSetVector<TTStub<T> >, TTStub<T> > aStub) const {
  //std::vector< edm::Ref< edmNew::DetSetVector< TTCluster< T > >, TTCluster< T > > > theseClusters = aStub->getClusterRefs();

  if (theClusterAssociationMap.isNull()) {
    return true;
  }

  if (theClusterAssociationMap->isUnknown(aStub->clusterRef(0)) &&
      theClusterAssociationMap->isUnknown(aStub->clusterRef(1)))
    return true;

  return false;
}

#endif