beamtime/cern-oct12/go4/RICH/CbmRichRingFinderHoughImpl.h (r4864/r3568)

Go to the documentation of this file.

#ifndef CBM_RICH_RING_FINDER_HOUGH_IMPL
#define CBM_RICH_RING_FINDER_HOUGH_IMPL

#include "CbmRichRingLight.h"

#include <vector>
#include <map>
#include <functional>

using std::vector;

class CbmRichRingFitterCOP;

class CbmRichHoughHit {
public:
   CbmRichHoughHit():
      fHit(),
      fX2plusY2(0.f),
      fId(0),
      fIsUsed(false)
   { }

   virtual ~CbmRichHoughHit(){}

        CbmRichHitLight fHit;
        float fX2plusY2;
   unsigned short fId;
        bool fIsUsed;
};

class CbmRichHoughHitCmpUp:
       public std::binary_function<
                  const CbmRichHoughHit,
                  const CbmRichHoughHit,
                  bool>
{
public:
        bool operator()(
              const CbmRichHoughHit &m1,
              const CbmRichHoughHit &m2) const
        {
                return m1.fHit.fX < m2.fHit.fX;
        }
};


class CbmRichRingComparatorMore:
       public std::binary_function<
                  const CbmRichRingLight*,
                  const CbmRichRingLight*,
                  bool>
{
public:
        bool operator()(
              const CbmRichRingLight* ring1,
              const CbmRichRingLight* ring2) const
        {
                return ring1->GetSelectionNN() > ring2->GetSelectionNN();
        }
};


class CbmRichRingFinderHoughImpl
{

protected:
        static const unsigned short MAX_NOF_HITS = 65000; // maximum number of hits in RICH detector

   // parameters of the Hough Transform algorithm
        unsigned short fNofParts; // number of groups of hits for HT

        float fMaxDistance; // maximum distance between two hits
        float fMinDistance; // minimum distance between two hits
        float fMinDistanceSq; // = fMinDistance*fMinDistance
        float fMaxDistanceSq; // = fMaxDistance*fMaxDistance

        float fMinRadius; // minimum radius of the ring
        float fMaxRadius; // maximum radius of the ring

        float fDx; // x bin width of the ring center histogram
        float fDy; // y bin width of the ring center histogram
        float fDr; // width of the ring radius histogram
        unsigned short fNofBinsX; // number of bins in X direction
        unsigned short fNofBinsY; // number of bins in Y direction
        unsigned short fNofBinsXY; // fNofBinsX*fNofBinsY

        unsigned short fHTCut; // cut number of entries in maximum bin of XY histogram

        unsigned short fNofBinsR;// number of bins in radius histogram
        unsigned short fHTCutR; // cut number of entries in maximum bin of Radius histogram

        unsigned short fMinNofHitsInArea; // minimum number of hits in the local area

        float fRmsCoeffCOP;
        float fMaxCutCOP;

        float fCurMinX; // current minimum X position of the local area
        float fCurMinY; // current minimum Y position of the local area

        vector<CbmRichHoughHit> fData; // Rich hits
        vector<unsigned short> fHist; // XY histogram
        vector<unsigned short> fHistR; // Radius histogram
        vector< vector<unsigned short> > fHitInd; // store hit indexes for different group of hits
        vector<CbmRichRingLight*> fFoundRings; // collect found rings
        CbmRichRingFitterCOP* fFitCOP; // COP ring fitter

public:
        CbmRichRingFinderHoughImpl ();

        virtual ~CbmRichRingFinderHoughImpl();

        void SetParameters();

        virtual void HoughTransformReconstruction();

        virtual void DefineLocalAreaAndHits(
              float x0,
              float y0,
              int *indmin,
              int *indmax);

        virtual void HoughTransform(
              unsigned short indmin,
                        unsigned short indmax);

        virtual void HoughTransformGroup(
              unsigned short indmin,
                        unsigned short indmax,
                        int iPart);

        void FindPeak(
              int indmin,
              int indmax);

        void DoFind(
                        const vector<CbmRichHitLight>& hits);

        vector<CbmRichRingLight*>& GetFoundRings()
        {
                return fFoundRings;
        }

private:
   CbmRichRingFinderHoughImpl(const CbmRichRingFinderHoughImpl&);

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