HoughFinder.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.                  *
 **************************************************************************/
 
//-----------------------------------------------------------------------------
// Description : A class to find tracks usning Hough algorithm
//-----------------------------------------------------------------------------
 
#ifndef TRGCDCHoughFinder_FLAG_
#define TRGCDCHoughFinder_FLAG_
 
#include <string>
#include "trg/cdc/HoughPlaneMulti2.h"
#include "trg/cdc/HoughTransformationCircle.h"
#include "trg/cdc/PeakFinder.h"
#include <map>
#ifdef TRGCDC_SHORT_NAMES
#define TCHFinder TRGCDCHoughFinder
#endif
 
namespace Belle2 {
  class TRGCDC;
  class TRGCDCPeakFinder;
  class TRGCDCTrack;
  class TRGCDCLink;
  class TRGCDCJSignal;
  class TRGCDCJLUT;
  class TRGCDCJSignalData;
 
  class TRGCDCHoughFinder {
 
  public:
 
    TRGCDCHoughFinder(const std::string& name,
                      const TRGCDC&,
                      unsigned peakMin,
                      const std::string& mappingFilePlus,
                      const std::string& mappingFileMinus,
                      unsigned doit);
 
    virtual ~TRGCDCHoughFinder();
 
  public:
 
    std::string name(void) const;
 
    std::string version(void) const;
 
    void terminate();
 
    int FindAndFit(std::vector<TRGCDCTrack*>& trackList2D,
                   std::vector<TRGCDCTrack*>& trackList2DFitted);
 
  public:
 
    bool perfect(bool);
 
  private:
 
    int doFindingAndFittingTrasan(std::vector<TRGCDCTrack*>& trackList2D,
                                  std::vector<TRGCDCTrack*>& trackList2DFitted);
 
    int doFindingAndFitting(std::vector<TRGCDCTrack*>& trackList2D,
                            std::vector<TRGCDCTrack*>& trackList2DFitted);
 
    std::vector<TRGCDCLink*> selectBestHits(
      const std::vector<TRGCDCLink*>& links) const;
 
    void mappingByFile(const std::string& mappingFilePlus,
                       const std::string& mappingFileMinus);
 
    int doFindingTrasan(std::vector<unsigned> peaks[],
                        std::vector<TRGCDCTrack*>& trackList2D) const;
 
    int doFinding(std::vector<std::vector<unsigned>> peaks[],
                  std::vector<TRGCDCTrack*>& trackList2D);
 
    int doFittingTrasan(std::vector<unsigned> peaks[],
                        std::vector<TRGCDCTrack*>& trackList2DFitted) const;
 
    int doFitting(std::vector<TRGCDCTrack*>& trackList2D,
                  std::vector<TRGCDCTrack*>& trackList2DFitted);
 
    TRGCDCTrack* makeTrack(const unsigned serialID, const unsigned pm) const;
 
    static double calPhi(TRGCDCSegmentHit const* segmentHit,
                         double eventTime);
 
    static void calCosPhi(std::map<std::string, Belle2::TRGCDCJSignal>
                          & mSignalStorage,
                          std::map<std::string, Belle2::TRGCDCJLUT* >
                          & mLutStorage);
 
    static void calSinPhi(std::map<std::string, Belle2::TRGCDCJSignal>
                          & mSignalStorage,
                          std::map<std::string, Belle2::TRGCDCJLUT* >
                          & mLutStorage);
 
    static void rPhi(std::map<std::string, Belle2::TRGCDCJSignal>
                     & mSignalStorage,
                     std::map<std::string, Belle2::TRGCDCJLUT* >
                     & mLutStorage);
  private:
 
    const std::string _name;
 
    const TRGCDC& _cdc;
 
    TRGCDCHoughPlaneMulti2* _plane[2];
 
    TRGCDCHoughTransformationCircle _circleH;
 
    const unsigned _doit;
 
    TRGCDCPeakFinder _peakFinder;
 
    const unsigned _peakMin;
 
    std::map<std::string, double> m_mDouble;
 
    std::map<std::string, std::vector<double> > m_mVector;
 
    std::map<std::string, double> m_mConstants;
 
    std::map<std::string, double> m_mConstD;
 
    std::map<std::string, std::vector<double> > m_mConstV;
 
    std::map<std::string, bool> m_mBool;
 
    std::map<std::string, TRGCDCJSignal> m_mSignalStorage;
 
    std::map<std::string, TRGCDCJLUT*> m_mLutStorage;
 
    TRGCDCJSignalData* _commonData;
  };
 
//-----------------------------------------------------------------------------
 
  inline
  std::string
  TRGCDCHoughFinder::name(void) const
  {
    return _name;
  }
 
 
} // namespace Belle2
 
#endif