Fitter3D.h

//-----------------------------------------------------------------------------
// $Id$
//-----------------------------------------------------------------------------
// Filename : Fitter3D.h
// Section  : TRG CDC
// Owner    : KyungTae KIM (K.U.)
// Email    : ktkim@hep.korea.ac.kr
//-----------------------------------------------------------------------------
// Description : A class to fit tracks in 3D
//-----------------------------------------------------------------------------
// $Log$
//-----------------------------------------------------------------------------
 
#ifndef TRGCDCFitter3D_FLAG_
#define TRGCDCFitter3D_FLAG_
 
#include <string>
#include <vector>
#include <TFile.h>
#include <TTree.h>
#include <TVectorD.h>
 
#ifdef TRGCDC_SHORT_NAMES
#define TCFitter3D TRGCDCFitter3D
#endif
 
namespace Belle2 {
  class TRGCDC;
  class TRGCDCTrack;
  class TRGCDCSegmentHit;
  class TRGCDCJSignal;
  class TRGCDCJLUT;
  class TRGCDCJSignalData;
 
  class TRGCDCFitter3D {
 
  public:
 
    TRGCDCFitter3D(const std::string& name,
                   const std::string& rootFitter3DFile,
                   const TRGCDC&,
                   const std::map<std::string, bool>& flags);
    virtual ~TRGCDCFitter3D();
 
  public:
 
    void initialize();
    void terminate();
 
    int doit(std::vector<TRGCDCTrack*>& trackList);
 
    int doitComplex(std::vector<TRGCDCTrack*>& trackList);
 
    static double calPhi(TRGCDCSegmentHit const* segmentHit, double eventTime);
    static void getMCValues(const TRGCDC& m_cdc_in, TRGCDCTrack* aTrack, std::map<std::string, double>& m_mConstD_in,
                            std::map<std::string, double>& m_mDouble_in, std::map<std::string, std::vector<double> >& m_mVector_in);
 
    bool isAxialTrackFull(TRGCDCTrack& aTrack);
 
    bool isStereoTrackFull(TRGCDCTrack& aTrack);
 
    static void findHitAxialSuperlayers(TRGCDCTrack& aTrack, std::vector<double>& useAxSL , bool printError);
 
    static void findHitStereoSuperlayers(TRGCDCTrack& aTrack, std::vector<double>& useStSL, bool printError);
 
    void removeImpossibleStereoSuperlayers(std::vector<double>& useStSL);
 
    static void selectAxialTSs(TRGCDCTrack& aTrack, std::vector<int>& bestTSIndex);
 
    static int do2DFit(TRGCDCTrack& aTrack, std::map<std::string, bool>& m_mBool_in, std::map<std::string, double>& m_mConstD_in,
                       std::map<std::string, std::vector<double> >& m_mConstV_in, std::map<std::string, double>& m_mDouble_in,
                       std::map<std::string, std::vector<double> >& m_mVector_in);
 
    void print3DInformation(int iTrack);
 
    void saveVhdlAndCoe();
    void saveAllSignals();
    void saveIoSignals();
 
    std::string name(void) const;
    std::string version(void) const;
 
    static void getStereoGeometry(std::map<std::string, std::vector<double> >& stGeometry);
 
    static void getStereoXt(std::vector<double> const& stPriorityLayer, std::vector<std::vector<double> >& stXts, bool isSimple = 0);
 
    static void getConstants(std::map<std::string, double>& mConstD, std::map<std::string, std::vector<double> >& mConstV,
                             bool isXtSimple = 0);
 
  private:
    const std::string m_name;
 
    const TRGCDC& m_cdc;
 
    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, bool> m_mBool;
 
    std::map<std::string, TRGCDCJSignal> m_mSignalStorage;
    std::map<std::string, TRGCDCJLUT*> m_mLutStorage;
    TRGCDCJSignalData* m_commonData;
 
    std::string m_rootFitter3DFileName;
 
    TFile* m_fileFitter3D;
 
    TTree* m_treeTrackFitter3D;
 
    TTree* m_treeConstantsFitter3D;
 
    std::map<std::string, TVectorD*> m_mTVectorD;
    std::map<std::string, TClonesArray*> m_mTClonesArray;
    std::map<std::string, std::vector<signed long long> > m_mSavedSignals;
    std::map<std::string, std::vector<signed long long> > m_mSavedIoSignals;
 
  };
 
} 
 
 
#endif