Hough3DFinder.h

//-----------------------------------------------------------------------------
// $Id$
//-----------------------------------------------------------------------------
// Filename : Hough3DFinder.h
// Section  : TRG CDC
// Owner    : Jaebak Kim
// Email    : jbkim@hep.korea.ac.kr
//-----------------------------------------------------------------------------
// Description : A class to find 3D tracks using Hough algorithm
//-----------------------------------------------------------------------------
// $Log$
//-----------------------------------------------------------------------------
 
#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();
    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