ellipse_refinement.h

Go to the documentation of this file.

#ifndef  TUW_ELLIPSE_REFINEMENT_H
#define  TUW_ELLIPSE_REFINEMENT_H

#include <vector>
#include <opencv2/core/core.hpp>

namespace tuw
{

class EllipseRefinement
{
public:


    class Parameter
    {
    public:
        int border;
        double min_gradient;
        Parameter(int _border=3, double _min_gradient=30)
            : border(_border), min_gradient(_min_gradient) {}
    };


    class Ellipse
    {
    private:

        double a2, b2, a4, b4;  // a^2 etc., stored for Distance()
        double x2,y2;
    public:
        int id;
        // ellipse position      [px]
        double x, y;
        // ellipse long/short axis length [px]
        double a, b;
        // ellipse rotation in radiants [rad]
        double phi;
        // mean fit error
        double fit_error;
        // ellipse support
        double support;

        Ellipse();
        Ellipse(const Ellipse &e);
        
        void setEllipse(const double &_x, const double &_y, const double &_a, const double &_b, const double &_phi);

        void setEllipse(const cv::RotatedRect& rect);

        void get(cv::RotatedRect &r);

        double ellipseCircumference(double a, double b);
        
        bool insideEllipse(double _a, double _b, double _x0, double _y0, double _phi, double _x, double _y) const;

        unsigned ellipseSupport(const std::vector<cv::Point2f> &points, double inl_dist, std::vector<bool> &inl_idx);
        
        bool computeAndSetGeomFromConic(double A, double B, double C, double D, double E, double F);

    };
private:

    std::vector<cv::Point2d> pointsToUse;

public:
    Parameter param;

    EllipseRefinement(const Parameter &_param = Parameter());
    ~EllipseRefinement();

    bool refine(const cv::Mat_<short> &im_dx, const cv::Mat_<short> &im_dy, const std::vector<cv::Point2f> &points, Ellipse &ellipse);
};

}

#endif //TUW_ELLIPSE_REFINEMENT_H