C++实现批量图片拼接

/**函数功能：不同图片拼接
 *     参数：
 *          vector<string> pic_list      : 图片名称列表
 *          int pic_cols_rows           ： horizontal==true,pic_cols_rows为生成图片的行数
                                           horizontal==false,pic_cols_rows为生成图片的列数
 *          bool horizontal             ： true-先横向后纵向合成图片  false-先纵向后横向合成图片
 *          bool draw_rect              :  true-在图片边缘画矩形框 false-不在图片边缘画矩形框
 *
 */
void mergeDiffPic(vector<string> pic_list, int pic_cols_rows, string output_file, bool horizontal=true, bool draw_rect=false)
{
    int pic_cols = 0;
    int pic_rows = 0;
    int max_cols=0;
    int max_rows=0;
    int size_cols=0;
    int size_rows=0;
    vector<int> tmp_cols;
    vector<int> tmp_rows;
    //获取图片数量
    int pic_num = pic_list.size();
    vector<Mat>input(pic_num);
    Mat merge;
    for(int i=0; i<pic_num; i++){
        input[i] = imread(pic_list[i]);
        //draw_rect为true,画矩形
        if(draw_rect){
            Rect rect = Rect(0,0,input[i].cols,input[i].rows);
            rectangle(input[i],rect,Scalar(0, 0, 255));
        }
    }
    //按水平方向合成
    if (horizontal){
        pic_cols = pic_cols_rows;
        pic_rows = pic_num/pic_cols;
        //生成的图片行数
        if (pic_num%pic_cols != 0) pic_rows += 1;
        int i = 0;
        int j = 0;
        for (i=0;i<pic_rows;i++){
            max_cols = 0;
            //保存每行图片的最大高度，方便后面确定图片的摆放位置
            tmp_rows.push_back(size_rows);
            max_rows = 0;
            for (j=0;j<pic_cols;j++){
                if ((i*pic_cols+j) >= pic_num) break;
                //保存每行图片的最大宽度，用于确定合成图的宽度
                max_cols += input[i*pic_cols+j].cols;
                max_rows = (max_rows>input[i*pic_cols+j].rows?max_rows:input[i*pic_cols+j].rows);
            }
            //合成图的宽度和高度
            size_cols = (max_cols>size_cols?max_cols:size_cols);
            size_rows += max_rows;
            if ((i*pic_cols+j) >= pic_num) break;
        }
        //创建size_cols×size_rows大小的空白图片，用于摆放小图
        Size mergesize(size_cols,size_rows);
        merge.create(mergesize, CV_MAKETYPE(input[0].depth(), 3));//rgb 3通道
        merge = Scalar::all(0);
        
        vector<Mat>temp(pic_num);
        //摆放图片
        for (i=0;i<pic_rows;i++){
            int sum_cols = 0;
            for (j=0;j<pic_cols;j++){
                if ((i*pic_cols+j) >= pic_num) break;
                //确定第（i*pic_cols+j）张图在merge上的位置
                temp[i*pic_cols+j] = merge(Rect(sum_cols, tmp_rows[i], input[i*pic_cols+j].cols, input[i*pic_cols+j].rows));
                //下一张图的起始位置（x坐标）
                sum_cols += input[i*pic_cols+j].cols;
                input[i*pic_cols+j].copyTo(temp[i*pic_cols+j]);
            }
            if ((i*pic_cols+j) >= pic_num) break;
        }
    }else{
        pic_rows = pic_cols_rows;
        pic_cols = pic_num/pic_rows;
        if (pic_num%pic_rows != 0) pic_cols += 1;
        int i = 0;
        int j = 0;
        for (i=0;i<pic_cols;i++){
            max_rows = 0;
            tmp_cols.push_back(size_cols);
            max_cols = 0;
            for (j=0;j<pic_rows;j++){
                if ((i*pic_rows+j) >= pic_num) break;
                max_rows += input[i*pic_rows+j].rows;
                max_cols = (max_cols>input[i*pic_rows+j].cols?max_cols:input[i*pic_rows+j].cols);
            }
            size_rows = (max_rows>size_rows?max_rows:size_rows);
            size_cols += max_cols;
            if ((i*pic_rows+j) >= pic_num) break;
        }
        //std::cout<<size_cols<<std::endl;
        //std::cout<<size_rows<<std::endl;
        Size mergesize(size_cols,size_rows);
        vector<Mat>temp(pic_num);
        merge.create(mergesize, CV_MAKETYPE(input[0].depth(), 3));//rgb 3通道
        merge = Scalar::all(0);
        for (i=0;i<pic_cols;i++){
            int sum_rows = 0;
            for (j=0;j<pic_rows;j++){
                if ((i*pic_rows+j) >= pic_num) break;
                temp[i*pic_rows+j] = merge(Rect(tmp_cols[i], sum_rows, input[i*pic_rows+j].cols, input[i*pic_rows+j].rows));
                sum_rows += input[i*pic_rows+j].rows;
                input[i*pic_rows+j].copyTo(temp[i*pic_rows+j]);
            }
            if ((i*pic_rows+j) >= pic_num) break;
        }
    }
    //显示图片
    //imshow("merge", merge);
    //保存图片
    imwrite(output_file.c_str(), merge);

    //waitKey(0);
    
}

//调用
#include<iostream>
#include<string>
#include<vector>
#include<opencv2/opencv.hpp>

using namespace std;
using namespace cv;

int main(){
 vector<string> pic_list;
 pic_list.push_back("1.jpg");
 pic_list.push_back("2.jpg");
 pic_list.push_back("3.jpg");
 mergeDiffPic(pic_list, 2, "merge1.jpg");
 mergeDiffPic(pic_list, 1, "merge2.jpg",false);
 mergeDiffPic(pic_list, 3, "merge3.jpg",false,true);
 return 0;
}

//编译
g++ merge.cpp `pkg-config --cflags --libs opencv`