TTTrack.h

Go to the documentation of this file.

#ifndef L1_TRACK_TRIGGER_TRACK_FORMAT_H
#define L1_TRACK_TRIGGER_TRACK_FORMAT_H

#include "DataFormats/L1TrackTrigger/interface/TTStub.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "DataFormats/Common/interface/DetSetVectorNew.h"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"
#include "DataFormats/Phase2TrackerDigi/interface/Phase2TrackerDigi.h"

template< typename T >
class TTTrack
{
  private:
    std::vector< edm::Ref< edmNew::DetSetVector< TTStub< T > >, TTStub< T > > >
                 theStubRefs;
    GlobalVector theMomentum4Par;
    GlobalVector theMomentum5Par;
    GlobalPoint  thePOCA4Par;
    GlobalPoint  thePOCA5Par;
    double       theRInv4Par;
    double       theRInv5Par;
    unsigned int theSector;
    unsigned int theWedge;
    double       theStubPtConsistency4Par;
    double       theStubPtConsistency5Par;
    double       theChi24Par;
    double       theChi25Par;
    bool         valid4ParFit;
    bool         valid5ParFit;

  public:
    TTTrack();
    TTTrack( std::vector< edm::Ref< edmNew::DetSetVector< TTStub< T > >, TTStub< T > > > aStubs );

    ~TTTrack();

    std::vector< edm::Ref< edmNew::DetSetVector< TTStub< T > >, TTStub< T > > > getStubRefs() const        { return theStubRefs; }
    void addStubRef( edm::Ref< edmNew::DetSetVector< TTStub< T > >, TTStub< T > > aStub )                  { theStubRefs.push_back( aStub ); }
    void setStubRefs( std::vector< edm::Ref< edmNew::DetSetVector< TTStub< T > >, TTStub< T > > > aStubs ) { theStubRefs = aStubs; }

    GlobalVector getMomentum(unsigned int nPar=4) const;
    void         setMomentum( GlobalVector aMomentum, unsigned int nPar=5);

    double getRInv(unsigned int nPar=4) const;
    void   setRInv( double aRInv, unsigned int nPar=5 );

    GlobalPoint getPOCA(unsigned int nPar=4) const;
    void        setPOCA( GlobalPoint aPOCA, unsigned int nPar=5 );

    unsigned int getSector() const                 { return theSector; }
    void         setSector( unsigned int aSector ) { theSector = aSector; }
    unsigned int getWedge() const                  { return theWedge; }
    void         setWedge( unsigned int aWedge )   { theWedge = aWedge; }

    double       getChi2(unsigned int nPar=4) const;
    double       getChi2Red(unsigned int nPar=4) const;
    void         setChi2( double aChi2, unsigned int nPar=5 );

    double       getStubPtConsistency(unsigned int nPar=4) const;
    void         setStubPtConsistency( double aPtConsistency, unsigned int nPar=5 );

    void         setFitParNo( unsigned int aFitParNo ) { return; }

/*
    uint32_t getSuperStrip() const { return 0; }
*/
    bool isTheSameAs( TTTrack< T > aTrack ) const;

    bool hasStubInBarrel( unsigned int aLayer ) const;

    std::string print( unsigned int i = 0 ) const;

 private:

    bool checkValidArgs(unsigned int nPar) const;
    bool checkValidArgsForSet(unsigned int nPar) const;
  

}; 

template< typename T >
TTTrack< T >::TTTrack()
{
  theStubRefs.clear();
  theMomentum4Par = GlobalVector(0.0,0.0,0.0);
  theMomentum5Par = GlobalVector(0.0,0.0,0.0);
  theRInv4Par     = 0.0;
  theRInv5Par     = 0.0;
  thePOCA4Par     = GlobalPoint(0.0,0.0,0.0);
  thePOCA5Par     = GlobalPoint(0.0,0.0,0.0);
  theSector       = 0;
  theWedge        = 0;
  theChi24Par     = 0.0;
  theChi25Par     = 0.0;
  theStubPtConsistency4Par = 0.0;
  theStubPtConsistency5Par = 0.0;
  valid4ParFit    = false;
  valid5ParFit    = false;
}

template< typename T >
TTTrack< T >::TTTrack( std::vector< edm::Ref< edmNew::DetSetVector< TTStub< T > >, TTStub< T > > > aStubs )
{
  theStubRefs     = aStubs;
  theMomentum4Par = GlobalVector(0.0,0.0,0.0);
  theMomentum5Par = GlobalVector(0.0,0.0,0.0);
  theRInv4Par     = 0.0;
  theRInv5Par     = 0.0;
  thePOCA4Par     = GlobalPoint(0.0,0.0,0.0);
  thePOCA5Par     = GlobalPoint(0.0,0.0,0.0);
  theSector       = 0;
  theWedge        = 0;
  theChi24Par     = 0.0;
  theChi25Par     = 0.0;
  theStubPtConsistency4Par = 0.0;
  theStubPtConsistency5Par = 0.0;
  valid4ParFit    = false;
  valid5ParFit    = false;
}

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

template< typename T >
void TTTrack< T >::setMomentum( GlobalVector aMomentum, unsigned int nPar ) {

  if (!checkValidArgsForSet(nPar)) return;

  if (nPar==4) {
    valid4ParFit = true;
    theMomentum4Par=aMomentum;
  }

  if (nPar==5) {
    valid5ParFit = true;
    theMomentum5Par=aMomentum;
  }

  return;


} 


template< typename T >
GlobalVector TTTrack< T >::getMomentum(unsigned int nPar) const{

  if (!checkValidArgs(nPar))
    return GlobalVector(0.0,0.0,0.0);

  if (nPar==4) {
    return theMomentum4Par;
  }

  if (nPar==5) {
    return theMomentum5Par;
  }

  return GlobalVector(0.0,0.0,0.0);


} 


template< typename T >
void TTTrack< T >::setRInv(double aRInv, unsigned int nPar) {

  if (!checkValidArgsForSet(nPar)) return;

  if (nPar==4) {
    valid4ParFit = true;
    theRInv4Par=aRInv;
  }

  if (nPar==5) {
    valid5ParFit = true;
    theRInv5Par=aRInv;
  }

  return;

}


template< typename T >
double TTTrack< T >::getRInv(unsigned int nPar) const {

  if (!checkValidArgs(nPar)) return 0.0;

  if (nPar==4) {
    return theRInv4Par;
  }

  if (nPar==5) {
    return theRInv5Par;
  }

  return 0.0;

}


template< typename T >
void TTTrack< T >::setPOCA(GlobalPoint aPOCA, unsigned int nPar){

  if (!checkValidArgsForSet(nPar)) return;

  if (nPar==4) {
    valid4ParFit = true;
    thePOCA4Par=aPOCA;
  }

  if (nPar==5) {
    valid5ParFit = true;
    thePOCA5Par=aPOCA;
  }

  return;

}

template< typename T >
GlobalPoint TTTrack< T >::getPOCA(unsigned int nPar) const
{

  if (!checkValidArgs(nPar)) return GlobalPoint(0.0,0.0,0.0);
 
  if (nPar==4) {
    return thePOCA4Par;
  }

  if (nPar==5) {
    return thePOCA5Par;
  }

  return GlobalPoint(0.0,0.0,0.0);

}

template< typename T >
void TTTrack< T >::setChi2(double aChi2, unsigned int nPar) {

  if (!checkValidArgsForSet(nPar)) return;

  if (nPar==4) {
    valid4ParFit = true;
    theChi24Par=aChi2;
  }

  if (nPar==5) {
    valid5ParFit = true;
    theChi25Par=aChi2;
  }

  return;

}



template< typename T >
double TTTrack< T >::getChi2(unsigned int nPar) const
{

  if (!checkValidArgs(nPar)) return 0.0;

  if (nPar==4) {
    return theChi24Par;
  }

  if (nPar==5) {
    return theChi25Par;
  }

  return 0.0;

}



template< typename T >
double TTTrack< T >::getChi2Red(unsigned int nPar) const
{

  if (!checkValidArgs(nPar)) return 0.0;

  if (nPar==4) {
    return theChi24Par/( 2*theStubRefs.size() - 4 );
  }

  if (nPar==5) {
    return theChi25Par/( 2*theStubRefs.size() - 5 );
  }

  return 0.0;

}


template< typename T >
void TTTrack< T >::setStubPtConsistency(double aStubPtConsistency, unsigned int nPar) {

  if (!checkValidArgsForSet(nPar)) return;

  if (nPar==4) {
    valid4ParFit = true;
    theStubPtConsistency4Par=aStubPtConsistency;
  }

  if (nPar==5) {
    valid5ParFit = true;
    theStubPtConsistency5Par=aStubPtConsistency;
  }

  return;

}



template< typename T >
double TTTrack< T >::getStubPtConsistency(unsigned int nPar) const
{

  if (!checkValidArgs(nPar)) return 0.0;

  if (nPar==4) {
    return theStubPtConsistency4Par;
  }

  if (nPar==5) {
    return theStubPtConsistency5Par;
  }

  return 0.0;

}





template< typename T>
bool TTTrack< T >::isTheSameAs( TTTrack< T > aTrack ) const
{
  std::vector< edm::Ref< edmNew::DetSetVector< TTStub< T > >, TTStub< T > > > otherStubRefs = aTrack.getStubRefs();

  unsigned int nShared = 0;
  for ( unsigned int i = 0; i < theStubRefs.size() && nShared < 2; i++)
  {
    for ( unsigned int j = 0; j < otherStubRefs.size() && nShared < 2; j++)
    {
      if ( theStubRefs.at(i) == otherStubRefs.at(j) )
      {
        nShared++;
      }
    }
  }

  return ( nShared > 1 );
}

template< typename T >
bool TTTrack< T >::hasStubInBarrel( unsigned int aLayer ) const
{
  for ( unsigned int i = 0; i < theStubRefs.size(); i++)
  {
    DetId detid =  theStubRefs.at(i)->getDetId();
    if ( detid.subdetId()==StripSubdetector::TOB )
    {
      return true;
    }
  }

  return false;
}


template< typename T >
std::string TTTrack< T >::print( unsigned int i ) const
{
  std::string padding("");
  for ( unsigned int j = 0; j != i; ++j )
  {
    padding+="\t";
  }

  std::stringstream output;
  output<<padding<<"TTTrack:\n";
  padding+='\t';
  output << '\n';
  unsigned int iStub = 0;

  typename std::vector< edm::Ref< edmNew::DetSetVector< TTStub< T > >, TTStub< T > > >::const_iterator stubIter;
  for ( stubIter = theStubRefs.begin();
        stubIter!= theStubRefs.end();
        ++stubIter )
  {
    output << padding << "stub: " << iStub++ << ", DetId: " << ((*stubIter)->getDetId()).rawId() << '\n';
  }

  return output.str();
}


template< typename T >
bool TTTrack< T >::checkValidArgs(unsigned int nPar) const {
      
  if (!(nPar==4||nPar==5)) {
    edm::LogError("TTTrack") << " A getter method was called with nPar = "<< nPar << 
      " but only authorized values are 4 and 5" << std::endl;
    return false;
  }

  if ((nPar==4)&&!valid4ParFit) {
    edm::LogError("TTTrack") << " You try to get info with nPar="<<nPar<<" but no valid 4 parameter fit info is present, use 5 instead"<<std::endl;
    return false;
  }

  if ((nPar==5)&&!valid5ParFit) {
    edm::LogError("TTTrack") << " You try to get info with nPar="<<nPar<<" but no valid 5 parameter fit info is present, use 4 instead"<<std::endl;
    return false;
  }

  return true;
  
}
template< typename T >
bool TTTrack< T >::checkValidArgsForSet(unsigned int nPar) const {
      
  if (!(nPar==4||nPar==5)) {
    edm::LogError("TTTrack") << " A setter method is called with nPar="<<nPar<< " only possible values are 4/5" << std::endl;
    return false;
  }

  return true;
  
}

template< typename T >
std::ostream& operator << ( std::ostream& os, const TTTrack< T >& aTTTrack ) { return ( os << aTTTrack.print() ); }


#endif