Hough3DUtility.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.                  *
 **************************************************************************/
#ifndef HOUGH3DUTILITY_H
#include <vector>
#include <TVectorD.h>
#define HOUGH3DUTILITY_H
 
namespace Belle2 {
  class TRGCDCJSignal;
  class TRGCDCJLUT;
  class TRGCDCJSignalData;
}
 
class Hough3DFinder {
public:
  Hough3DFinder(void);
  ~Hough3DFinder(void);
  Hough3DFinder(Hough3DFinder&) = delete;
  Hough3DFinder& operator=(Hough3DFinder&) = delete;
  void setMode(int mode);
  int getMode(void);
  void initialize(const TVectorD& geometryVariables, std::vector<float >& initVariables);
  void destruct(void);
  void runFinder(const std::vector<double>& trackVariables, std::vector<std::vector<double> >& stTSs,
                 const std::vector<std::vector<int> >& stTSDrift);
  void initVersion1(const std::vector<float >& initVariables);
  void initVersion2(std::vector<float >& initVariables);
  void initVersion3(std::vector<float >& initVariables);
  void setInputFileName(const std::string& inputFileName);
  void destVersion1(void);
  void destVersion2(void);
  void destVersion3(void);
  void runFinderVersion1(const std::vector<double>& trackVariables, const std::vector<std::vector<double> >& stTSs,
                         const std::vector<double>& tsArcS,
                         const std::vector<std::vector<double> >& tsZ);
  void runFinderVersion2(const std::vector<double>& trackVariables, std::vector<std::vector<double> >& stTSs,
                         const std::vector<std::vector<int> >& stTSDrift);
  void runFinderVersion3(const std::vector<double>& trackVariables, std::vector<std::vector<double> >& stTSs,
                         const std::vector<std::vector<int> >& stTSDrift);
  void getValues(const std::string& input, std::vector<double>& result);
  void getHoughMeshLayer(bool***& houghMeshLayer);
  int m_mode;
  int m_nWires[4];
  double m_rr[4];
  double m_ztostraw[4];
  double m_anglest[4];
  double m_cotStart;
  double m_cotEnd;
  double m_z0Start;
  double m_z0End;
  int m_nCotSteps;
  int m_nZ0Steps;
  double m_cotStepSize;
  double m_z0StepSize;
  float*** m_houghMeshLayerDiff;
  bool*** m_houghMeshLayer;
  int** m_houghMesh;
  float** m_houghMeshDiff;
  bool** m_hitMap;
  int** m_driftMap;
  std::vector< std::vector< int> >* m_geoCandidatesIndex;
  std::vector< std::vector< double> >* m_geoCandidatesPhi;
  std::vector< std::vector< double> >* m_geoCandidatesDiffStWires;
  double m_stAxPhi[4];
  double m_bestCot;
  double m_bestZ0;
  double m_houghMax;
  double m_minDiffHough;
  double m_foundZ[4];
  double m_foundPhiSt[4];
  int m_bestTSIndex[4];
  double m_bestTS[4];
  std::string m_inputFileName;
  std::vector<double> m_FPGAInput;
  std::vector<double> m_FPGAOutput;
  double m_findRhoMax;
  double m_findRhoMin;
  double m_findRhoIntMax;
  double m_findRhoIntMin;
  double m_findPhi0Max;
  double m_findPhi0Min;
  double m_findPhi0IntMax;
  double m_findPhi0IntMin;
  double m_findArcCosMax;
  double  m_findArcCosMin;
  double m_findArcCosIntMax;
  double m_findArcCosIntMin;
  double m_findPhiZMax;
  double  m_findPhiZMin;
  double m_findPhiZIntMax;
  double m_findPhiZIntMin;
  double m_rhoMax;
  double m_rhoMin;
  int m_rhoBit;
  double m_phi0Max;
  double m_phi0Min;
  int m_phi0Bit;
  int m_stAxWireFactor;
  bool m_LUT;
  int** m_arcCosLUT;
  int** m_wireConvertLUT;
 
  std::map<std::string, Belle2::TRGCDCJSignal> m_mSignalStorage;
  std::map<std::string, Belle2::TRGCDCJLUT*> m_mLutStorage;
  Belle2::TRGCDCJSignalData* m_commonData;
  std::map<std::string, std::vector<signed long long> > m_mSavedSignals;
  std::map<std::string, std::vector<signed long long> > m_mSavedIoSignals;
  std::map<std::string, bool> m_mBool;
  std::string m_outputVhdlDirname;
  std::string m_outputLutDirname;
 
 
};
 
#ifdef __CINT__
#include "../src/Hough3DUtility.cc"
#endif
 
#endif