MarkDetector.h

#pragma once

class CircleDetector
    {
    private:
        double      dCfgCirclesGaussSigmaStepSize, dCfgCirclesGaussSigmaMin, dCfgCirclesGaussSigmaMax;
        double      accuResolution;
        double      accuThreshold;
        double      cannyThreshold;
        cv::Mat     mat;
        bool            detect                  (cv::Mat &mat, int radiusMin, int radiusMax, MarkImageData &mark);
    public:
                        CircleDetector          ();
        bool            detectCircle            (cv::Mat &mRoi, int radiusMax, int radiusMin, int maxResult, int minResult, MarkImageData &mark);
    };

class SubPixelLocator
    {
    private:
        double      dCfgSubPixelGaussSigma;
        cv::Size    zeroZone;
        cv::Mat     mat;
    public:
                        SubPixelLocator         ();
        bool            locateCornerSubPixel    (cv::Mat &mRoi, int winSize, MarkImageData &mark);
    };

class MarkDetector
    {
    public:
        enum Result {
            RES_ERROR = -1,
            RES_SUCCESS,
            RES_ADJUSTED,
            RES_CALCULATED
        };
    private:
        int                     result;
        MarkImageData           markData;
        CircleDetector          circleDetector;
        SubPixelLocator         subPixler;
        MarkOrientationDetector markOrientationDetector;
        void            adjustRoi               (RoiRect &rRoi, MarkImageData &mark);
        int             handleDetection         (RoiRect &rRoi, HeadModel *lpHeadModel, MarkDetectDesc &detectDesc, MarkImageData &markData);
        int             adjustMarkDetection     (HeadModel *lpHeadModel, MarkDetectDesc &detectDesc);
    public:
                        MarkDetector            ();
                        ~MarkDetector           ();
        int             getResult               () const                { return result; }
        void            processRoi              (RoiRect &rRoi, HeadModel *lpHeadModel, MarkDetectDesc &detectDesc);
        bool            getMarkData             (MarkImageData &markData);
    };