BKLMTrackFitter.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
 
/* KLM headers. */
#include <klm/bklm/geometry/GeometryPar.h>
#include <klm/dataobjects/bklm/BKLMHit2d.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<BKLMHit2d*>& listTrackPoint);
 
    double distanceToHit(BKLMHit2d* hit,
                         double& error,
                         double& sigma);
 
    double globalDistanceToHit(BKLMHit2d* hit,
                               double& error,
                               double& sigma);
 
    double fit1dSectorTrack(std::list< BKLMHit2d* > hitList,
                            CLHEP::HepVector&  eta,
                            CLHEP::HepSymMatrix&  error,
                            int depDir,    int indDir);
 
    double fit1dTrack(std::list< BKLMHit2d* > 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