CbmRichRingFinderHough.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CbmRichRingFinderHough.h

#ifndef CBM_RICH_RING_FINDER_HOUGH
#define CBM_RICH_RING_FINDER_HOUGH

#include "CbmRichRingFinder.h"
#include <vector>

#include <TClonesArray.h>

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);

private:
  void AddRingsToOutputArray(
        TClonesArray *rRingArray,
        const vector<CbmRichRingLight*>& rings);

  CbmRichRingFinderHough(const CbmRichRingFinderHough&);

  CbmRichRingFinderHough& operator=(const CbmRichRingFinderHough&);
};

#endif