C++学习笔记——批量生成字符串

一个很实际的例子：我们在做图像处理的时候会检测到若干ROI，并且为了方便分析会把他们保存下来，但是在处理之前我们并不知道有多少个这样的ROI，而如果只知道路径我们势必会面临批量生成字符串的问题——把所有的ROI保存为ROI1.jpg、ROI2.jpg、ROI3.jpg的形式等等。这里我们用到的一个重要函数就是sprintf（高版本的VS使用sprintf_s）。这是C标准库中的函数之一，用于格式化字符串，代码如下：

vector<Mat> ROI_Group;

/*
//通过某种方式检测到ROI并保存到ROI_Group中
//......
*/

char fileName[50] = {'0'};
//开始批量保存ROI
for(int i = 0, size = ROI_Group.size(); i < size; ++i)
{
    sprintf(fileName, "E:\\MyFile\\Image %d.jpg", i);
    imwrite(fileName, ROI_Group[i]);
}

//后续操作

另一个问题是，我们如何批量读取一个文件夹的所有文件名。代码如下（以下代码我们假设文件夹中文件都是图片）

// LPCWSTR转string

std::string WChar2Ansi(LPCWSTR pwszSrc)
{
	int nLen = WideCharToMultiByte(CP_ACP, 0, pwszSrc, -1, NULL, 0, NULL, NULL);
 
	if (nLen <= 0) return std::string("");
 
	char* pszDst = new char[nLen];
	if (NULL == pszDst) return std::string("");
 
	WideCharToMultiByte(CP_ACP, 0, pwszSrc, -1, pszDst, nLen, NULL, NULL);
	pszDst[nLen - 1] = 0;
 
	std::string strTemp(pszDst);
	delete[] pszDst;
 
	return strTemp;
}


void readImgNamefromFile(char* fileName, vector <string> &imgNames)
{
	// vector清零 参数设置
	imgNames.clear();
	WIN32_FIND_DATA file;
	int i = 0;
	char tempFilePath[MAX_PATH + 1];
	char tempFileName[50];
	// 转换输入文件名
	sprintf_s(tempFilePath, "%s/*", fileName);
	// 多字节转换
	WCHAR   wstr[MAX_PATH] = { 0 };
	MultiByteToWideChar(CP_ACP, 0, tempFilePath, -1, wstr, sizeof(wstr));
	// 查找该文件待操作文件的相关属性读取到WIN32_FIND_DATA
	HANDLE handle = FindFirstFile(wstr, &file);
	if (handle != INVALID_HANDLE_VALUE)
	{
		FindNextFile(handle, &file);
		FindNextFile(handle, &file);
		// 循环遍历得到文件夹的所有文件名	
		do
		{
			sprintf(tempFileName, "%s", fileName);
			imgNames.push_back(WChar2Ansi(file.cFileName));
			imgNames[i].insert(0, tempFileName);
			i++;
		} while (FindNextFile(handle, &file));
	}
	FindClose(handle);
}

int main()
{
	// 设置读入图像序列文件夹的路径
	char* fileName = "G:\\FileRoot\\";
	std::vector <string>  imgNames;
	// 获取对应文件夹下所有文件名
	readImgNamefromFile(fileName, imgNames);
	// 遍历对应文件夹下所有文件名
	for (int i = 0; i < imgNames.size(); i++)
	{
		string imgName = imgNames[i];
        Mat img = imread(imgName);
        //对读取的每张图片进行后续图像处理操作
        //......
	}
	return 0;
}