CbmRichRingFinderHough.h

Go to the documentation of this file.

/* Copyright (C) 2006-2021 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Semen Lebedev [committer] */

 
#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 {
 
 public:
  CbmRichRingFinderHough();

  virtual ~CbmRichRingFinderHough();

  virtual void Init();

  virtual Int_t DoFind(CbmEvent* event, TClonesArray* rHitArray, TClonesArray* rProjArray, TClonesArray* rRingArray);
 
  void SetUseAnnSelect(bool use) { fUseAnnSelect = use; }
  void SetUseSubdivide(bool use) { fUseSubdivide = use; }
 
 private:
  Int_t fEventNum      = 0;
  Bool_t fUseAnnSelect = true;
  Bool_t fUseSubdivide = true;
 
// choose between serial and SIMD implementation of the ring finder
#ifdef HOUGH_SERIAL
  CbmRichRingFinderHoughImpl* fHTImpl = nullptr;
#endif
 
#ifdef HOUGH_SIMD
  CbmRichRingFinderHoughSimd* fHTImpl = nullptr;
#endif

  void AddRingsToOutputArray(CbmEvent* event, TClonesArray* rRingArray, TClonesArray* rHitArray,
                             const vector<CbmRichRingLight*>& rings);

  CbmRichRingFinderHough(const CbmRichRingFinderHough&);

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