mirror of
https://github.com/opencv/opencv.git
synced 2024-11-29 13:47:32 +08:00
b4b35cff15
Megre together with https://github.com/opencv/opencv_contrib/pull/3325 1. Move aruco_detector, aruco_board, aruco_dictionary, aruco_utils to objdetect 1.1 add virtual Board::draw(), virtual ~Board() 1.2 move `testCharucoCornersCollinear` to Board classes (and rename to `checkCharucoCornersCollinear`) 1.3 add wrappers to keep the old api working 3. Reduce inludes 4. Fix java tests (add objdetect import) 5. Refactoring ### Pull Request Readiness Checklist See details at https://github.com/opencv/opencv/wiki/How_to_contribute#making-a-good-pull-request - [x] I agree to contribute to the project under Apache 2 License. - [x] To the best of my knowledge, the proposed patch is not based on a code under GPL or another license that is incompatible with OpenCV - [x] The PR is proposed to the proper branch - [x] There is a reference to the original bug report and related work - [x] There is accuracy test, performance test and test data in opencv_extra repository, if applicable Patch to opencv_extra has the same branch name. - [x] The feature is well documented and sample code can be built with the project CMake ``` **WIP** force_builders=linux,win64,docs,Linux x64 Debug,Custom Xbuild_contrib:Docs=OFF build_image:Custom=ubuntu:22.04 build_worker:Custom=linux-1 ```
108 lines
3.0 KiB
C++
108 lines
3.0 KiB
C++
// 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 FRAME_PROCESSOR_HPP
|
|
#define FRAME_PROCESSOR_HPP
|
|
|
|
#include <opencv2/core.hpp>
|
|
#include <opencv2/calib3d.hpp>
|
|
#ifdef HAVE_OPENCV_ARUCO
|
|
#include <opencv2/aruco/charuco.hpp>
|
|
#endif
|
|
|
|
#include "calibCommon.hpp"
|
|
#include "calibController.hpp"
|
|
|
|
namespace calib
|
|
{
|
|
class FrameProcessor
|
|
{
|
|
protected:
|
|
|
|
public:
|
|
virtual ~FrameProcessor();
|
|
virtual cv::Mat processFrame(const cv::Mat& frame) = 0;
|
|
virtual bool isProcessed() const = 0;
|
|
virtual void resetState() = 0;
|
|
};
|
|
|
|
class CalibProcessor : public FrameProcessor
|
|
{
|
|
protected:
|
|
cv::Ptr<calibrationData> mCalibData;
|
|
TemplateType mBoardType;
|
|
cv::Size mBoardSize;
|
|
std::vector<cv::Point2f> mTemplateLocations;
|
|
std::vector<cv::Point2f> mCurrentImagePoints;
|
|
cv::Mat mCurrentCharucoCorners;
|
|
cv::Mat mCurrentCharucoIds;
|
|
|
|
cv::Ptr<cv::SimpleBlobDetector> mBlobDetectorPtr;
|
|
#ifdef HAVE_OPENCV_ARUCO
|
|
cv::aruco::Dictionary mArucoDictionary;
|
|
cv::Ptr<cv::aruco::CharucoBoard> mCharucoBoard;
|
|
#endif
|
|
|
|
int mNeededFramesNum;
|
|
unsigned mDelayBetweenCaptures;
|
|
int mCapuredFrames;
|
|
double mMaxTemplateOffset;
|
|
float mSquareSize;
|
|
float mTemplDist;
|
|
bool mSaveFrames;
|
|
float mZoom;
|
|
|
|
bool detectAndParseChessboard(const cv::Mat& frame);
|
|
bool detectAndParseChAruco(const cv::Mat& frame);
|
|
bool detectAndParseCircles(const cv::Mat& frame);
|
|
bool detectAndParseACircles(const cv::Mat& frame);
|
|
bool detectAndParseDualACircles(const cv::Mat& frame);
|
|
void saveFrameData();
|
|
void showCaptureMessage(const cv::Mat &frame, const std::string& message);
|
|
bool checkLastFrame();
|
|
|
|
public:
|
|
CalibProcessor(cv::Ptr<calibrationData> data, captureParameters& capParams);
|
|
virtual cv::Mat processFrame(const cv::Mat& frame) CV_OVERRIDE;
|
|
virtual bool isProcessed() const CV_OVERRIDE;
|
|
virtual void resetState() CV_OVERRIDE;
|
|
~CalibProcessor() CV_OVERRIDE;
|
|
};
|
|
|
|
enum visualisationMode {Grid, Window};
|
|
|
|
class ShowProcessor : public FrameProcessor
|
|
{
|
|
protected:
|
|
cv::Ptr<calibrationData> mCalibdata;
|
|
cv::Ptr<calibController> mController;
|
|
TemplateType mBoardType;
|
|
visualisationMode mVisMode;
|
|
bool mNeedUndistort;
|
|
double mGridViewScale;
|
|
double mTextSize;
|
|
|
|
void drawBoard(cv::Mat& img, cv::InputArray points);
|
|
void drawGridPoints(const cv::Mat& frame);
|
|
public:
|
|
ShowProcessor(cv::Ptr<calibrationData> data, cv::Ptr<calibController> controller, TemplateType board);
|
|
virtual cv::Mat processFrame(const cv::Mat& frame) CV_OVERRIDE;
|
|
virtual bool isProcessed() const CV_OVERRIDE;
|
|
virtual void resetState() CV_OVERRIDE;
|
|
|
|
void setVisualizationMode(visualisationMode mode);
|
|
void switchVisualizationMode();
|
|
void clearBoardsView();
|
|
void updateBoardsView();
|
|
|
|
void switchUndistort();
|
|
void setUndistort(bool isEnabled);
|
|
~ShowProcessor() CV_OVERRIDE;
|
|
};
|
|
|
|
}
|
|
|
|
|
|
#endif
|