mirror of
https://github.com/opencv/opencv.git
synced 2024-12-21 05:28:01 +08:00
75 lines
2.5 KiB
C++
75 lines
2.5 KiB
C++
#ifndef __OPENCV_BLENDERS_HPP__
|
|
#define __OPENCV_BLENDERS_HPP__
|
|
|
|
#include <vector>
|
|
#include <opencv2/core/core.hpp>
|
|
|
|
// Simple blender which puts one image over another
|
|
class Blender
|
|
{
|
|
public:
|
|
enum { NO, FEATHER, MULTI_BAND };
|
|
|
|
static cv::Ptr<Blender> createDefault(int type);
|
|
|
|
cv::Point operator ()(const std::vector<cv::Mat> &src, const std::vector<cv::Point> &corners, const std::vector<cv::Mat> &masks,
|
|
cv::Mat& dst);
|
|
cv::Point operator ()(const std::vector<cv::Mat> &src, const std::vector<cv::Point> &corners, const std::vector<cv::Mat> &masks,
|
|
cv::Mat& dst, cv::Mat& dst_mask);
|
|
|
|
protected:
|
|
virtual cv::Point blend(const std::vector<cv::Mat> &src, const std::vector<cv::Point> &corners, const std::vector<cv::Mat> &masks,
|
|
cv::Mat& dst, cv::Mat& dst_mask);
|
|
};
|
|
|
|
|
|
class FeatherBlender : public Blender
|
|
{
|
|
public:
|
|
FeatherBlender(float sharpness = 0.02f) : sharpness_(sharpness) {}
|
|
|
|
private:
|
|
cv::Point blend(const std::vector<cv::Mat> &src, const std::vector<cv::Point> &corners, const std::vector<cv::Mat> &masks,
|
|
cv::Mat &dst, cv::Mat &dst_mask);
|
|
|
|
float sharpness_;
|
|
};
|
|
|
|
|
|
class MultiBandBlender : public Blender
|
|
{
|
|
public:
|
|
MultiBandBlender(int num_bands = 7) : num_bands_(num_bands) {}
|
|
|
|
int numBands() const { return num_bands_; }
|
|
void setNumBands(int val) { num_bands_ = val; }
|
|
|
|
private:
|
|
cv::Point blend(const std::vector<cv::Mat> &src, const std::vector<cv::Point> &corners, const std::vector<cv::Mat> &masks,
|
|
cv::Mat& dst, cv::Mat& dst_mask);
|
|
|
|
int num_bands_;
|
|
};
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
// Auxiliary functions
|
|
|
|
cv::Rect resultRoi(const std::vector<cv::Mat> &src, const std::vector<cv::Point> &corners);
|
|
|
|
cv::Point computeResultMask(const std::vector<cv::Mat> &masks, const std::vector<cv::Point> &corners, cv::Mat &mask);
|
|
|
|
cv::Point blendLinear(const std::vector<cv::Mat> &src, const std::vector<cv::Point> &corners, const std::vector<cv::Mat> &weights,
|
|
cv::Mat& dst, cv::Mat& dst_weight);
|
|
|
|
void normalize(const cv::Mat& weight, cv::Mat& src);
|
|
|
|
void createWeightMap(const cv::Mat& mask, float sharpness, cv::Mat& weight);
|
|
|
|
void createLaplacePyr(const std::vector<cv::Mat>& pyr_gauss, std::vector<cv::Mat>& pyr_laplace);
|
|
|
|
// Restores source image in-place. Result will be stored in pyr[0].
|
|
void restoreImageFromLaplacePyr(std::vector<cv::Mat>& pyr);
|
|
|
|
#endif // __OPENCV_BLENDERS_HPP__
|