CbmRichProtRingFinderHough.h

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

#ifndef CBM_RICH_PROT_RING_FINDER_HOUGH_H_
#define CBM_RICH_PROT_RING_FINDER_HOUGH_H_

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

class CbmRichProtRingFinderHoughImpl;
class CbmRichRingFinderHoughSimd;
class CbmRichRing;
class CbmRichRingLight;

#define HOUGH_SERIAL
//#define HOUGH_SIMD

using std::vector;

class CbmRichProtRingFinderHough : 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
  CbmRichProtRingFinderHoughImpl *fHTImpl;
#endif

#ifdef HOUGH_SIMD
  CbmRichRingFinderHoughSimd *fHTImpl;
#endif

public:

    CbmRichProtRingFinderHough ();

  virtual ~CbmRichProtRingFinderHough();

  virtual void Init();

  virtual Int_t DoFind(
        TClonesArray* rHitArray,
      TClonesArray* rProjArray,
       TClonesArray* rRingArray);

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

  CbmRichProtRingFinderHough(const CbmRichProtRingFinderHough&);

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

#endif