KLMTrack.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
/* KLM headers. */
#include <klm/dataobjects/KLMElementNumbers.h>
 
/* Basf2 headers. */
#include <framework/datastore/RelationsObject.h>
 
/* ROOT headers. */
#include <TVectorD.h>
#include <Math/Vector3D.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 KLMTrack : public RelationsObject {
 
  public:
 
    KLMTrack();
 
    explicit KLMTrack(const KLMTrack&);
 
    KLMTrack& operator=(const KLMTrack&);
 
    ~KLMTrack()
    {
    }
 
    TVectorD  getTrackParam();
 
    TMatrixDSym  getTrackParamErr();
 
    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 setTrackChi2(const float chi2)
    {
      m_Chi2 = chi2;
    }
 
    void setNumHitOnTrack(const int NumHit)
    {
      m_NumHit = NumHit;
    }
 
    void setInSubdetector(int nBKLM, int nEKLM)
    {
      m_inBKLM = (nBKLM > 0) ? true : false;
      m_inEKLM = (nEKLM > 0) ? true : false;
    }
 
    void setIsValid(const bool valid)
    {
      m_Valid = valid;
    }
 
    void setIsGood(const bool good)
    {
      m_Good = good;
    }
 
    bool inBKLM() const
    {
      return (m_inBKLM);
    }
 
    bool inEKLM() const
    {
      return (m_inEKLM);
    }
 
 
 
 
  private:
 
    bool m_Valid;
 
    bool m_Good;
 
    float m_Chi2;
 
    int m_NumHit;
 
    float m_TrackParam[4];
 
    float m_TrackParamErr[4][4];
 
    bool m_inBKLM;
 
    bool m_inEKLM;
 
    ClassDef(KLMTrack, 1)
 
  };
 
} // end of namespace Belle2