fourier_descriptors.hpp

// This file is part of OpenCV project.
// It is subject to the license terms in the LICENSE file found in the top-level directory
// of this distribution and at http://opencv.org/license.html.
 
#ifndef __OPENCV_FOURIERDESCRIPTORS_HPP__
#define __OPENCV_FOURIERDESCRIPTORS_HPP__
 
#include <opencv2/core.hpp>
 
namespace cv {
namespace ximgproc {
 
 
    class CV_EXPORTS_W ContourFitting : public Algorithm
    {
        int ctrSize;
        int fdSize;
        std::vector<std::complex<double> > b;
        std::vector<std::complex<double> > a;
        std::vector<double> frequence;
        std::vector<double> rho, psi;
        void frequencyInit();
        void fAlpha(double x, double &fn, double &df);
        double distance(std::complex<double> r, double alpha);
        double  newtonRaphson(double x1, double x2);
    public:
        ContourFitting(int ctr=1024,int fd=16):ctrSize(ctr),fdSize(fd){};
        void estimateTransformation(InputArray src, InputArray dst, OutputArray alphaPhiST, double *dist = 0, bool fdContour = false);
        CV_WRAP void estimateTransformation(InputArray src, InputArray dst, OutputArray alphaPhiST, CV_OUT double &dist , bool fdContour = false);
        CV_WRAP void setCtrSize(int n);
        CV_WRAP void setFDSize(int n);
        CV_WRAP int getCtrSize() { return ctrSize; };
        CV_WRAP int getFDSize() { return fdSize; };
    };
    CV_EXPORTS_W void fourierDescriptor(InputArray src, OutputArray dst, int nbElt=-1,int nbFD=-1);
    CV_EXPORTS_W void transformFD(InputArray src, InputArray t,OutputArray dst, bool fdContour=true);
    CV_EXPORTS_W void contourSampling(InputArray src, OutputArray out, int nbElt);
 
    CV_EXPORTS_W Ptr<ContourFitting> createContourFitting(int ctr = 1024, int fd = 16);
 
}
}
#endif