rapid.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_RAPID_HPP_
#define OPENCV_RAPID_HPP_
 
#include <opencv2/core.hpp>
#include <opencv2/imgproc.hpp>
 
namespace cv
{
namespace rapid
{
 
CV_EXPORTS_W void drawCorrespondencies(InputOutputArray bundle, InputArray cols,
                                       InputArray colors = noArray());
 
CV_EXPORTS_W void drawSearchLines(InputOutputArray img, InputArray locations, const Scalar& color);
 
CV_EXPORTS_W void drawWireframe(InputOutputArray img, InputArray pts2d, InputArray tris,
                                const Scalar& color, int type = LINE_8, bool cullBackface = false);
CV_EXPORTS_W void extractControlPoints(int num, int len, InputArray pts3d, InputArray rvec, InputArray tvec,
                                       InputArray K, const Size& imsize, InputArray tris, OutputArray ctl2d,
                                       OutputArray ctl3d);
CV_EXPORTS_W void extractLineBundle(int len, InputArray ctl2d, InputArray img, OutputArray bundle,
                                    OutputArray srcLocations);
 
CV_EXPORTS_W void findCorrespondencies(InputArray bundle, OutputArray cols,
                                       OutputArray response = noArray());
 
CV_EXPORTS_W void convertCorrespondencies(InputArray cols, InputArray srcLocations, OutputArray pts2d,
                                          InputOutputArray pts3d = noArray(), InputArray mask = noArray());
 
CV_EXPORTS_W float rapid(InputArray img, int num, int len, InputArray pts3d, InputArray tris, InputArray K,
                         InputOutputArray rvec, InputOutputArray tvec, CV_OUT double* rmsd = 0);
 
class CV_EXPORTS_W Tracker : public Algorithm
{
public:
    virtual ~Tracker();
    CV_WRAP virtual float
    compute(InputArray img, int num, int len, InputArray K, InputOutputArray rvec, InputOutputArray tvec,
            const TermCriteria& termcrit = TermCriteria(TermCriteria::MAX_ITER | TermCriteria::EPS, 5, 1.5)) = 0;
    CV_WRAP virtual void clearState() = 0;
};
 
class CV_EXPORTS_W Rapid : public Tracker
{
public:
    CV_WRAP static Ptr<Rapid> create(InputArray pts3d, InputArray tris);
};
 
class CV_EXPORTS_W OLSTracker : public Tracker
{
public:
    CV_WRAP static Ptr<OLSTracker> create(InputArray pts3d, InputArray tris, int histBins = 8, uchar sobelThesh = 10);
};
 
class CV_EXPORTS_W GOSTracker : public Tracker
{
public:
    CV_WRAP static Ptr<OLSTracker> create(InputArray pts3d, InputArray tris, int histBins = 4, uchar sobelThesh = 10);
};
} /* namespace rapid */
} /* namespace cv */
 
#endif /* OPENCV_RAPID_HPP_ */