Hough3DFinder.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 3D tracks using Hough algorithm
//-----------------------------------------------------------------------------
 
#ifndef TRGCDCHough3DFinder_FLAG_
#define TRGCDCHough3DFinder_FLAG_
 
#include <string>
#include "trg/cdc/Hough3DUtility.h"
 
#ifdef TRGCDC_SHORT_NAMES
#define TCH3DFinder TRGCDCHough3DFinder
#endif
 
namespace Belle2 {
  class TRGCDC;
  class TRGCDCTrack;
  class TRGCDCSegmentHit;
  class TRGCDCLink;
 
  class TRGCDCHough3DFinder {
  public:
    TRGCDCHough3DFinder(const TRGCDC&, bool makeRootFile, int finderMode);
    ~TRGCDCHough3DFinder();
    TRGCDCHough3DFinder(TRGCDCHough3DFinder&) = delete;
    TRGCDCHough3DFinder& operator=(TRGCDCHough3DFinder&) = delete;
    void doit(std::vector<TRGCDCTrack*> const& trackList2D, std::vector<TRGCDCTrack*>& trackList3D);
    void doit(std::vector<TRGCDCTrack*>& trackList);
    void perfectFinder(std::vector<TRGCDCTrack*>& trackList, unsigned j, std::vector<const TRGCDCSegmentHit*>& mcTSList);
    void doitPerfectly(std::vector<TRGCDCTrack*>& trackList);
    void doitFind(std::vector<TRGCDCTrack*>& trackList);
    void doitNewFind(std::vector<TRGCDCTrack*>& trackList);
    void findNumberOfHitSuperlayersForMcParticles(std::vector<std::vector<const TRGCDCSegmentHit*> >& p_stTSs,
                                                  std::map<unsigned, unsigned>& numberTSsForParticle);
    void terminate(void);
 
    const TRGCDC& _cdc;
    bool m_makeRootFile;
    int m_finderMode;
    Hough3DFinder* m_Hough3DFinder;
    std::string m_rootHough3DFilename;
    TFile* m_fileFinder3D;
    TTree* m_treeTrackFinder3D;
    TTree* m_treeConstantsFinder3D;
    std::map<std::string, double> m_mDouble;
    std::map<std::string, std::vector<double> > m_mVector;
    std::map<std::string, double> m_mConstD;
    std::map<std::string, std::vector<double> > m_mConstV;
    std::map<std::string, double> m_mEventD;
    std::map<std::string, std::vector<double> > m_mEventV;
    std::map<std::string, bool> m_mBool;
    std::map<std::string, TVectorD*> m_mRunTVectorD;
    std::map<std::string, TVectorD*> m_mEventTVectorD;
    std::map<std::string, TClonesArray*> m_mTClonesArray;
 
  };
 
} // namespace Belle2
 
#endif