BKLMTrackFitter.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/bklm/geometry/GeometryPar.h>
#include <klm/dataobjects/KLMHit2d.h>
 
/* CLHEP headers. */
#include <CLHEP/Matrix/SymMatrix.h>
#include <CLHEP/Matrix/Vector.h>
 
/* C++ headers. */
#include <list>
 
namespace Belle2 {

  class BKLMTrackFitter {
 
  public:

    BKLMTrackFitter();

    ~BKLMTrackFitter();

    double fit(std::list<KLMHit2d*>& listTrackPoint);

    double distanceToHit(const KLMHit2d* hit,
                         double& error,
                         double& sigma);

    double globalDistanceToHit(const KLMHit2d* hit,
                               double& error,
                               double& sigma);

    double fit1dSectorTrack(std::list< KLMHit2d* > hitList,
                            CLHEP::HepVector&  eta,
                            CLHEP::HepSymMatrix&  error,
                            int depDir,    int indDir);

    double fit1dTrack(const std::list< KLMHit2d* >& hitList,
                      CLHEP::HepVector&  eta,
                      CLHEP::HepSymMatrix&  error,
                      int depDir,    int indDir);

    CLHEP::HepVector    getTrackParam()
    {
      return m_GlobalPar;
    }

    CLHEP::HepSymMatrix getTrackParamErr()
    {
      return m_GlobalErr;
    }

    CLHEP::HepVector    getTrackParamSector()
    {
      return m_SectorPar;
    }

    CLHEP::HepSymMatrix getTrackParamSectorErr()
    {
      return m_SectorErr;
    }

    bool isValid()
    {
      return m_Valid;
    }

    bool  isGood()
    {
      return m_Good;
    }

    float getChi2()
    {
      return m_Chi2;
    }

    int getNumHit()
    {
      return m_NumHit;
    }

    void  inValidate()
    {
      m_Valid = false;
    }

    void  setGlobalFit(bool localOrGlobal)
    {
      m_globalFit = localOrGlobal;
    }
 
  private:

    bool m_Valid;

    bool m_Good;

    float m_Chi2;

    int m_NumHit;

    bool m_globalFit;

    CLHEP::HepVector     m_SectorPar;

    CLHEP::HepSymMatrix  m_SectorErr;

    CLHEP::HepVector     m_GlobalPar;

    CLHEP::HepSymMatrix  m_GlobalErr;

    bklm::GeometryPar*   m_GeoPar;
 
  };

} // end of namespace Belle2