TrigVertex.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRIGINDETEVENT_TRIGVERTEX_H
#define TRIGINDETEVENT_TRIGVERTEX_H
 
#include <list>
 
#include "CLHEP/Geometry/Point3D.h"
#include "TrigInDetEvent/TrigInDetTrack.h"
#include <math.h>
#include <map>
#include <ostream>
#include "GaudiKernel/MsgStream.h"
 
typedef std::list< const TrigInDetTrack* >   TrackInVertexList;
 
class TrigVertex{
 
 public:
  enum AlgoId{NULLID=0, SITRACKID=1, IDSCANID=2, HISTOPRMVTXSITRACKID=3, VKALSECVTXID=4, BPHYSSECID=5,
          BSFULLSITRACKID=6, BSFULLIDSCANID=7, BSSPLITSITRACKID=8, BSSPLITIDSCANID=9, HISTOPRMVTXIDSCANID=10,
          HISTOPRMVTXEFID=11, HISTOPRMVTXSITRACKIDTAU=12, HISTOPRMVTXEFIDTAU=13, BJETSECVTXID=14, 
              BSFULL_STRATEGY_B_ID=15, BSFULL_STRATEGY_A_ID=16, BSSPLIT_STRATEGY_B_ID=17, BSSPLIT_STRATEGY_A_ID=18,
          HISTOPRMVTXFTKID=19, HISTOPRMVTXFTKIDTAU=20, BSFULL_STRATEGY_F_ID=21, BSSPLIT_STRATEGY_F_ID = 22,
          HISTOPRMVTXFTKID_REFIT = 23};
 
  // Constructors
  TrigVertex() :  m_x(0.0), m_y(0.0), m_z(0.), m_position(0,0,0), 
    m_chiSquared(0.0), m_nDOF(0), 
    m_tracks(NULL), m_algId(NULLID),
    m_ownTracks(false)
    {
      for(int i=0;i<6;i++) m_cov[i]=0.0;
      m_mass=0.0;m_massVar=0.0;m_P=NULL;m_energyFraction=0.0;
      m_nTwoTracksSecVtx=0;
      m_decayLength = m_decayLengthSignificance = 0;
    }
 
  TrigVertex( double zPosition ) : m_x(0.0), m_y(0.0), m_z(zPosition),m_position(0,0,zPosition),
    m_chiSquared(0.0), m_nDOF(0), 
    m_tracks(NULL), m_algId(NULLID),
    m_ownTracks(false)
    {
      for(int i=0;i<6;i++) m_cov[i]=0.0;
      m_mass=0.0;m_massVar=0.0;m_P=NULL;m_energyFraction=0.0;
      m_nTwoTracksSecVtx=0;
      m_decayLength = m_decayLengthSignificance = 0;
    }
 
  TrigVertex( double zPosition, AlgoId id ) : m_x(0.0), m_y(0.0), m_z(zPosition), m_position(0,0,zPosition),
    m_chiSquared(0.0), m_nDOF(0),
    m_tracks(NULL), m_algId(id),
    m_ownTracks(false)
    {
      for(int i=0;i<6;i++) m_cov[i]=0.0;
      m_mass=0.0;m_massVar=0.0;m_P=NULL;m_energyFraction=0.0;
      m_nTwoTracksSecVtx=0;
      m_decayLength = m_decayLengthSignificance = 0;
    }
 
  TrigVertex( double zPosition, double cv, AlgoId id ) : m_x(0.0), m_y(0.0), m_z(zPosition), m_position(0,0,zPosition),
    m_chiSquared(0.0), m_nDOF(0),
    m_tracks(NULL), m_algId(id),
    m_ownTracks(false)
    {
      for(int i=0;i<5;i++) m_cov[i]=0.0;
      m_cov[5]=cv;
      m_mass=0.0;m_massVar=0.0;m_P=NULL;m_energyFraction=0.0;
      m_nTwoTracksSecVtx=0;
      m_decayLength = m_decayLengthSignificance = 0;
    }
 
  TrigVertex(double x, double y, double z, double cv[6], double chi2, int ndf, TrackInVertexList* tracks) :
    m_x(x), m_y(y), m_z(z), m_position(x,y,z), m_chiSquared(chi2), 
    m_nDOF(ndf), m_tracks(tracks), m_algId(NULLID), 
    m_ownTracks(false)
    {
      for(int i=0;i<6;i++) m_cov[i]=cv[i];
      m_mass=0.0;m_massVar=0.0;m_P=NULL;m_energyFraction=0.0;
      m_nTwoTracksSecVtx=0;
      m_decayLength = m_decayLengthSignificance = 0;
    }
    
  TrigVertex(double x, double y, double z, double cv[6], double chi2, int ndf, TrackInVertexList* tracks,
             double mass, double energyFraction, int n2trkvtx,
             double decayLength, double decayLengthSignificance,
             AlgoId algo_id) :
    m_x(x), m_y(y), m_z(z), m_mass(mass), m_massVar(0),
    m_energyFraction(energyFraction),
    m_nTwoTracksSecVtx(n2trkvtx),
    m_decayLength(decayLength),
    m_decayLengthSignificance(decayLengthSignificance),
    m_position(x,y,z), m_chiSquared(chi2), 
    m_nDOF(ndf), m_tracks(tracks), m_algId(algo_id), 
    m_ownTracks(false)
  {
    for(int i=0;i<6;i++) m_cov[i]=cv[i];
    m_P=NULL;
  }
 
  // Destructor
  ~TrigVertex() 
    {
      if (m_ownTracks) {
        for (TrackInVertexList::iterator i = m_tracks->begin();
             i != m_tracks->end();
             ++i)
          delete *i;
      }
      if(m_tracks!=NULL) delete m_tracks;
      if(m_P!=NULL) delete m_P;
    }
 
  // Methods to retrieve data members 
 
  void algorithmId(const AlgoId  id) { m_algId = id;}
  AlgoId algorithmId() const { return m_algId;}
 
  const HepGeom::Point3D<double>& position() const { return m_position; } 
  double chi2()          const                     { return m_chiSquared; } 
  int    ndof()          const                     { return m_nDOF; } 
  const double* cov()    const                     { return &m_cov[0];} 
 
  TrackInVertexList* tracks() { return m_tracks; }
  const TrackInVertexList* tracks() const  { return m_tracks; }
  double x() const { return m_x; } 
  double y() const { return m_y; } 
  double z() const { return m_z; } 
 
  double mass() const   { return m_mass; }
  double massVariance() const { return m_massVar; }
  const TrigInDetTrackFitPar* getMotherTrack() const { return m_P;} 
 
  double energyFraction() const { return m_energyFraction; } 
  int nTwoTracksSecVtx() const { return m_nTwoTracksSecVtx; }
  double decayLength() const { return m_decayLength; } 
  double decayLengthSignificance() const { return m_decayLengthSignificance;} 
 
  // Methods to set data members 
 
  void setMass (double m)          { m_mass = m; }
  void setMassVariance (double m)  { m_massVar = m; }
  void setMotherTrack(const TrigInDetTrackFitPar* P) { m_P=P;}
 
  void setEnergyFraction (double e) { m_energyFraction = e; }
  void setNTwoTrackSecVtx (int n)   { m_nTwoTracksSecVtx = n; }
 
  void setDecayLength (double v) {m_decayLength = v;}
  void setDecayLengthSignificance (double v) {m_decayLengthSignificance = v;}
 
 private:
  friend class TrigVertexCnv_p1;
  friend class TrigVertexCnv_p2;
  
 
  double m_x,m_y,m_z,m_mass,m_massVar;
  double m_cov[6];
 
  double m_energyFraction;
 
  int m_nTwoTracksSecVtx;
 
  double m_decayLength;
  double m_decayLengthSignificance; 
  
  HepGeom::Point3D<double> m_position;
  double     m_chiSquared;
  int        m_nDOF;
  TrackInVertexList*  m_tracks;
  AlgoId m_algId;
  const TrigInDetTrackFitPar* m_P;
  bool m_ownTracks;
 
};
 
std::string str( const TrigVertex& v );                      //<! printing helper
MsgStream& operator<< ( MsgStream& m, const TrigVertex& v ); //<! printing helper (wraps above)
bool operator== ( const TrigVertex& a, const TrigVertex& b ); 
inline bool operator!= ( const TrigVertex& a, const TrigVertex& b ) { return !(a==b); }
 
void diff( const TrigVertex& a, const TrigVertex& b, std::map<std::string, double>& variableChange ); 
 
 
CLASS_DEF( TrigVertex , 224516498 , 1 )
 
#endif // TRIGINDETEVENT_TRIGVERTEX_H