BKLMTrack.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Yinghui GUAN                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
/* Belle 2 headers. */
#include <framework/datastore/RelationsObject.h>
 
/* ROOT headers. */
#include <TVectorD.h>
#include <TVector3.h>
#include <TMatrixDSym.h>
 
/* CLHEP headers. */
#include <CLHEP/Matrix/Vector.h>
#include <CLHEP/Matrix/SymMatrix.h>
 
namespace CLHEP {
  class HepVector;
  class HepSymMatrix;
}
 
namespace Belle2 {
  class BKLMTrack : public RelationsObject {
 
  public:
 
    BKLMTrack();
 
    explicit BKLMTrack(const BKLMTrack&);
 
    BKLMTrack& operator=(const BKLMTrack&);
 
    virtual ~BKLMTrack()
    {
    }
 
    TVectorD  getTrackParam();
 
    TMatrixDSym  getTrackParamErr();
 
    TVectorD   getLocalTrackParam();
 
    TMatrixDSym  getLocalTrackParamErr();
 
    TVector3 getLocalIntercept(double x);
 
    TMatrixD getLocalInterceptVariance(double x);
 
    float getTrackChi2()
    {
      return m_Chi2;
    }
 
    int getNumHitOnTrack()
    {
      return m_NumHit;
    }
 
    bool isValid()
    {
      return m_Valid;
    }
 
    bool  isGood()
    {
      return m_Good;
    }
 
    void setTrackParam(const CLHEP::HepVector& trkPar);
 
    void setTrackParamErr(const CLHEP::HepSymMatrix& trkParErr);
 
    void setLocalTrackParam(const CLHEP::HepVector& trkPar);
 
    void setLocalTrackParamErr(const CLHEP::HepSymMatrix& trkParErr);
 
    void setTrackChi2(const float chi2)
    {
      m_Chi2 = chi2;
    }
 
    void setNumHitOnTrack(const int NumHit)
    {
      m_NumHit = NumHit;
    }
 
    void setIsValid(const bool valid)
    {
      m_Valid = valid;
    }
 
    void setIsGood(const bool good)
    {
      m_Good = good;
    }
 
  private:
 
    bool m_Valid;
 
    bool m_Good;
 
    float m_Chi2;
 
    int m_NumHit;
 
    float m_TrackParam[4];
 
    float m_TrackParamErr[4][4];
 
    float m_LocalTrackParam[4];
 
    float m_LocalTrackParamErr[4][4];
 
    ClassDef(BKLMTrack, 1)
 
  };
 
} // end of namespace Belle2