SecVtxInfo.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
/***************************************************************************
                          SecVtxInfo.h  -  Description
                             -------------------
    begin   : 04-06-2004
    authors : Andreas Wildauer (CERN PH-ATC), Fredrik Akesson (CERN PH-ATC)
    email   : andreas.wildauer@cern.ch, fredrik.akesson@cern.ch
    changes :
 
 ***************************************************************************/
 
#ifndef JETTAGINFO_SECVTXINFO_H
#define JETTAGINFO_SECVTXINFO_H
 
#include "JetTagInfo/BaseTagInfo.h"
#include "Particle/TrackParticle.h"
#include <vector>
#include <iostream>
 
namespace Trk
{
  class RecVertex;
}
 
namespace Analysis
{
typedef std::vector<double> FloatVec;
typedef std::vector<std::string> StringVector;
typedef std::vector<std::string>::iterator StringIterator;
typedef std::vector<const Rec::TrackParticle*> TrackVec;

 
class SecVtxInfo : public BaseTagInfo
{
public:
    enum FitType { TearDown=0, BuildUp=1 , NoFit=2 };
    SecVtxInfo();

    SecVtxInfo(const TagInfoType& x);

    SecVtxInfo(const SecVtxInfo& rhs);

    SecVtxInfo &operator= (const SecVtxInfo& rhs);

    virtual ~SecVtxInfo();
 
    virtual SecVtxInfo* clone() const;

    int numSelTracksForFit() const;

    double distance() const;
    double rphidistance() const;

    double probability() const;

    double mass() const;

    double energyFraction() const;

    int mult()const;

    FitType fitType() const; // Type of fit

    const Trk::RecVertex& secVertexPos() const;
    
    int NumberOfG2TrackVertices() const;

    void setFitType(FitType type);
    void setSecVtx(const Trk::RecVertex& secVtx,
                   double fitProb,
                   const TrackVec& fittedTrks);
    void setNumSelTracksForFit(int numSelTracksForFit);
    void setMult(int mult);
    void setMass(double mass);
    void setEnergyFraction(double energyFraction);
    void setDist(double dist);
    void setRPhiDist(double dist);
    void setNumberOfG2TrackVertices(int NSVPair);
    
private:
    int m_numSelTracksForFit; 
    double m_dist;
    double m_rphidist;
    double m_prob; // CumulativeChiSquare of vtx chi2
    double m_mass;
    double m_energyFraction;
    TrackVec m_fittedTrks;
    int m_mult;
    FitType m_fitType;
    Trk::RecVertex m_secVtxPos;
    int m_NGood2TrackVertices;
}; // End class
 
inline int SecVtxInfo::numSelTracksForFit() const
{
    return m_numSelTracksForFit;
}
inline double SecVtxInfo::distance() const
{
    return m_dist;
}
inline double SecVtxInfo::rphidistance() const
{
    return m_rphidist;
}
inline double SecVtxInfo::probability() const
{
    return m_prob;
}
inline double SecVtxInfo::mass() const
{
    return m_mass;
}
inline double SecVtxInfo::energyFraction() const
{
    return m_energyFraction;
}
inline int SecVtxInfo::mult() const
{
    return m_mult;
}
inline SecVtxInfo::FitType SecVtxInfo::fitType() const
{
    return m_fitType;
}
inline const Trk::RecVertex& SecVtxInfo::secVertexPos() const
{
    return m_secVtxPos;
}
inline void SecVtxInfo::setSecVtx(const Trk::RecVertex& secVtx,
                                  double fitProb, 
                                  const TrackVec& fittedTrks)
{
    m_secVtxPos=secVtx;
    m_prob=fitProb;
    m_fittedTrks=fittedTrks;
}
inline void SecVtxInfo::setNumSelTracksForFit(int numSelTracksForFit)
{
    m_numSelTracksForFit=numSelTracksForFit;
}
inline void SecVtxInfo::setMult(int mult)
{
    m_mult=mult;
}
inline void SecVtxInfo::setMass(double mass)
{
    m_mass=mass;
}
inline void SecVtxInfo::setEnergyFraction(double energyFraction)
{
    m_energyFraction = energyFraction;
}
inline void SecVtxInfo::setFitType(FitType type)
{
    m_fitType=type;
}
inline void SecVtxInfo::setDist(double dist)
{
    m_dist=dist;
}
inline void SecVtxInfo::setRPhiDist(double dist)
{
    m_rphidist=dist;
}
inline void SecVtxInfo::setNumberOfG2TrackVertices(int NGood2TrackVertices)
{
    m_NGood2TrackVertices=NGood2TrackVertices;
}
inline int SecVtxInfo::NumberOfG2TrackVertices() const
{
    return m_NGood2TrackVertices;
}
inline Analysis::SecVtxInfo* Analysis::SecVtxInfo::clone() const
{
    return new SecVtxInfo(*this);
}
 
} // End namespace
#endif