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->getClusterRef(0) ) &&
       theClusterAssociationMap->isUnknown( aStub->getClusterRef(1) ) )
    return true;

  return false;
}


#endif