mirror of http://192.168.1.51:8099/lmh188/twain3.0
108 lines
3.2 KiB
C++
108 lines
3.2 KiB
C++
#include "ImageApplyConcatenation.h"
|
|
|
|
CImageApplyConcatenation::CImageApplyConcatenation()
|
|
: m_direction(autoDirection)
|
|
, m_BG_color(255, 255, 255)
|
|
{
|
|
}
|
|
|
|
CImageApplyConcatenation::CImageApplyConcatenation(ConcatMode dir, const cv::Scalar& background)
|
|
: m_direction(dir)
|
|
, m_BG_color(background)
|
|
{
|
|
}
|
|
|
|
CImageApplyConcatenation::~CImageApplyConcatenation(void)
|
|
{
|
|
}
|
|
|
|
void CImageApplyConcatenation::apply(cv::Mat& pDib, int side)
|
|
{
|
|
(void)pDib;
|
|
(void)side;
|
|
}
|
|
|
|
void CImageApplyConcatenation::apply(std::vector<cv::Mat>& mats, bool isTwoSide)
|
|
{
|
|
(void)isTwoSide;
|
|
if (mats.size() < 2)
|
|
{
|
|
mats.clear();
|
|
return;
|
|
}
|
|
|
|
ConcatMode direction;
|
|
if (m_direction == ConcatMode::autoDirection)
|
|
direction = (mats[0].cols > mats[0].rows) ? ConcatMode::vertical : ConcatMode::horizontal;
|
|
else
|
|
direction = m_direction;
|
|
cv::Mat dst = concat(mats[0], mats[1], direction);
|
|
mats.clear();
|
|
mats.push_back(dst);
|
|
}
|
|
|
|
cv::Mat CImageApplyConcatenation::concat(cv::Mat &front, cv::Mat &back, ConcatMode direction)
|
|
{
|
|
cv::Mat dst;
|
|
if (direction == horizontal)
|
|
{
|
|
#if 0
|
|
int top, bottom;
|
|
if (front.rows > back.rows)
|
|
{
|
|
top = (front.rows - back.rows) / 2;
|
|
bottom = front.rows - back.rows - top;
|
|
cv::copyMakeBorder(back, back, top, bottom, 0, 0, cv::BORDER_CONSTANT);
|
|
}
|
|
else if (front.rows < back.rows)
|
|
{
|
|
top = (back.rows - front.rows) / 2;
|
|
bottom = back.rows - front.rows - top;
|
|
cv::copyMakeBorder(front, front, top, bottom, 0, 0, cv::BORDER_CONSTANT);
|
|
}
|
|
cv::hconcat(front, back, dst);
|
|
#else
|
|
int width = cv::max(front.cols, back.cols) * 2;
|
|
int height = cv::max(front.rows, back.rows);
|
|
if (front.channels() != back.channels())
|
|
cv::cvtColor(front.channels() > back.channels() ? back : front, front.channels() > back.channels() ? back : front, cv::COLOR_GRAY2BGR);
|
|
dst = cv::Mat(height, width, front.type(), m_BG_color);
|
|
front.copyTo(dst(cv::Rect(0, 0, front.cols, front.rows)));
|
|
int offset = front.cols;
|
|
front.release();
|
|
back.copyTo(dst(cv::Rect(offset, 0, back.cols, back.rows)));
|
|
back.release();
|
|
#endif
|
|
}
|
|
else if (direction == vertical)
|
|
{
|
|
#if 0
|
|
int left, right;
|
|
if (front.cols > back.cols)
|
|
{
|
|
left = (front.cols - back.cols) / 2;
|
|
right = front.cols - back.cols - left;
|
|
cv::copyMakeBorder(back, back, 0, 0, left, right, cv::BORDER_CONSTANT);
|
|
}
|
|
else if (front.cols < back.cols)
|
|
{
|
|
left = (back.cols - front.cols) / 2;
|
|
right = back.cols - front.cols - left;
|
|
cv::copyMakeBorder(front, front, 0, 0, left, right, cv::BORDER_CONSTANT);
|
|
}
|
|
cv::vconcat(front, back, dst);
|
|
#else
|
|
int width = cv::max(front.cols, back.cols);
|
|
int height = cv::max(front.rows, back.rows) * 2;
|
|
|
|
dst = cv::Mat(height, width, front.type(), m_BG_color);
|
|
front.copyTo(dst(cv::Rect(0, 0, front.cols, front.rows)));
|
|
int offset = front.rows;
|
|
front.release();
|
|
back.copyTo(dst(cv::Rect(0, offset, back.cols, back.rows)));
|
|
back.release();
|
|
#endif
|
|
}
|
|
return dst;
|
|
}
|