C++ 常用代碼片段整理

C++ 常用代碼片段整理

一丶文件 文件夾相關

1.1 遞歸刪除文件

wstring Utilstring2wstring(string str)
{
	wstring result;
	//獲取緩衝區大小，並申請空間，緩衝區大小按字符計算  
	int len = MultiByteToWideChar(CP_ACP, 0, str.c_str(), str.size(), NULL, 0);
	wchar_t* buffer = NULL;
	buffer = new wchar_t[len + 1];
	//多字節編碼轉換成寬字節編碼  
	MultiByteToWideChar(CP_ACP, 0, str.c_str(), str.size(), buffer, len);
	buffer[len] = L'\0';             //添加字符串結尾  
	//刪除緩衝區並返回值  
	result.append(buffer);
	if (buffer != NULL)
	{
		delete[] buffer;
		buffer = NULL;
	}
	return result;
}

//將wstring轉換成string  
string Utilwstring2string(wstring wstr)
{
	string result;
	//獲取緩衝區大小，並申請空間，緩衝區大小事按字節計算的  
	int len = WideCharToMultiByte(CP_ACP, 0, wstr.c_str(), wstr.size(), NULL, 0, NULL, NULL);
	char* buffer = new char[len + 1];
	//寬字節編碼轉換成多字節編碼  
	WideCharToMultiByte(CP_ACP, 0, wstr.c_str(), wstr.size(), buffer, len, NULL, NULL);
	buffer[len] = '\0';
	//刪除緩衝區並返回值  
	result.append(buffer);
	delete[] buffer;
	return result;
}

bool DeleteAllFileA(std::string FileName)
{
	string RootFileName = FileName;
	wstring wRootFileName = Utilstring2wstring(RootFileName);
	
	RootFileName = endsWith(wRootFileName, L"\\") ? RootFileName : RootFileName + "\\";
	string RmFilter = RootFileName + "*.*";
	WIN32_FIND_DATAA windata = { 0 };
	HANDLE hFileHandle = FindFirstFileA(RmFilter.c_str(), &windata);
	do
	{
		if (windata.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
		{
			string compareDirName = windata.cFileName;
			if (compareDirName.find(".") != compareDirName.npos
				|| compareDirName.find("..") != compareDirName.npos)
			{
				continue;
			}
			//遞歸刪除
			string RmFileName = RootFileName + compareDirName;
			DeleteAllFileA(RmFileName);
			RemoveDirectoryA(RmFileName.c_str()); //遞歸刪除文件夾
			continue;
		}
		//直接刪除文件
		string RmFileName = RootFileName + windata.cFileName;
		DeleteFileA(RmFileName.c_str());
		

	} while (FindNextFileA(hFileHandle, &windata));
	
	return true;
}