#include <fstream>
#include <boost/lexical_cast.hpp>
#include <boost/tokenizer.hpp>
#include <opencv2/core/core.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/highgui/highgui.hpp>
// 貼り付ける画像情報
struct ImageInfo {
std::string path; // ファイルの絶対パス
int param[6]; // パラメータ
};
// src_imgをdst_imgのrect領域にコピー
void MatCopyPaste(const cv::Rect& rect, cv::Mat& dst_img, cv::Mat& src_img) {
cv::Mat d = dst_img(rect); // 貼り付け先のMatを取得
src_img.copyTo(d); // 貼り付け画像をコピーする
}
// 1画像の処理
void ImageProc(cv::Mat& dst_img, const ImageInfo& info) {
// 貼り付ける画像を読み込む
cv::Mat src_img = cv::imread(info.path, 1);
assert(!src_img.empty());
// 貼り付け先の画像のサイズ
int imgWidth = info.param[3];
int imgHeight = info.param[4];
// 貼り付け位置
int imgX = info.param[1];
int imgY = info.param[2];
// 貼り付け画像
cv::Mat tmp_img(cv::Size(imgWidth, imgHeight), src_img.type());
if(info.param[5] == 1) {
// 画像を回転
int src_w = src_img.size().width;
int src_h = src_img.size().height;
cv::Mat rot_img(cv::Size(src_h, src_w), src_img.type(), cv::Scalar(0, 0, 0));
cv::transpose(src_img, rot_img); // 転置 左回り 反時計回りに90度回転
cv::flip(rot_img, rot_img, 1); // 左右反転 時計回りに90度回転
cv::resize(rot_img, tmp_img, tmp_img.size(), cv::INTER_CUBIC);
} else {
// サイズ変更
cv::resize(src_img, tmp_img, tmp_img.size(), cv::INTER_CUBIC);
}
// 指定した矩形に画像を貼り付ける
cv::Rect rect(imgX, imgY, imgWidth, imgHeight);
MatCopyPaste(rect, dst_img, tmp_img);
}
int main( int argc, char **argv ) {
int count = 0;
std::string str;
// 貼り付ける画像を記述したテキストファイルを読み込む
std::ifstream ifs( "./data/ImgList.txt" );
typedef boost::char_separator char_sep;
typedef boost::tokenizer tokenizer;
char_sep sep(" "); // スペース区切り
int imgNum, imgWidth, imgHeight;
std::vector imgDataArray;
ImageInfo imgData;
while(std::getline(ifs, str)) {
std::vector tokenArray;
// 読み込んだ文字列をトークンに分割して配列に格納
tokenizer tok(str, sep);
for (tokenizer::iterator tok_iter = tok.begin(); tok_iter != tok.end(); ++tok_iter) {
tokenArray.push_back( *tok_iter );
}
switch(count) {
case 0: // 画像数
// int にキャスト
imgNum = boost::lexical_cast(tokenArray[0]);
break;
case 1: // 画像の幅と高さ
imgWidth = boost::lexical_cast(tokenArray[0]);
imgHeight = boost::lexical_cast(tokenArray[1]);
break;
default: // 画像のパラメータ
if(0 == count % 2) {
// ファイルパス
imgData.path = tokenArray[0];
} else {
// パラメータ
for(int i = 0; i < 6; i++) {
imgData.param[i] = boost::lexical_cast(tokenArray[i]);
}
imgDataArray.push_back(imgData);
}
break;
}
count++;
}
// 貼り付け先の画像
cv::Mat dstImg(cv::Size(imgWidth, imgHeight), CV_8UC3, cv::Scalar(0, 0, 0));
// 画像処理
for(size_t i = 0; i < imgDataArray.size(); i++) {
ImageProc(dstImg, imgDataArray[i]);
}
// 結果を保存
// cv::imwrite("./data/result.png", dstImg);
// 結果を表示
cv::namedWindow("Result", CV_WINDOW_AUTOSIZE|CV_WINDOW_FREERATIO);
cv::imshow("Result", dstImg);
cv::waitKey(0);
return 0;
}
|
