CbmRichRingFinderHough.h

Go to the documentation of this file.

 
#ifndef CBM_RICH_RING_FINDER_HOUGH
#define CBM_RICH_RING_FINDER_HOUGH
 
#include "CbmRichRingFinder.h"
#include <vector>
 
class CbmRichRingFinderHoughImpl;
class CbmRichRingFinderHoughSimd;
class CbmRichRing;
class CbmRichRingLight;
 
#define HOUGH_SERIAL
//#define HOUGH_SIMD
 
using std::vector;
 
class CbmRichRingFinderHough : public CbmRichRingFinder {
protected:
  Int_t fNEvent;     // event number
  Int_t fRingCount;  // number of found rings
 
// choose between serial and SIMD implementation of the ring finder
#ifdef HOUGH_SERIAL
  CbmRichRingFinderHoughImpl* fHTImpl;
#endif
 
#ifdef HOUGH_SIMD
  CbmRichRingFinderHoughSimd* fHTImpl;
#endif
 
public:
  CbmRichRingFinderHough();
 
  virtual ~CbmRichRingFinderHough();
 
  virtual void Init();
 
  virtual Int_t DoFind(TClonesArray* rHitArray,
                       TClonesArray* rProjArray,
                       TClonesArray* rRingArray);
 
  void SetUseAnnSelect(bool use) { fUseAnnSelect = use; }
 
private:
  bool fUseAnnSelect;
  void AddRingsToOutputArray(TClonesArray* rRingArray,
                             TClonesArray* rHitArray,
                             const vector<CbmRichRingLight*>& rings);
 
  CbmRichRingFinderHough(const CbmRichRingFinderHough&);
 
  CbmRichRingFinderHough& operator=(const CbmRichRingFinderHough&);
};
 
#endif