C++ 使用指定目录的图片设置壁纸，并输出图片名称
写这个是因为目录中的壁纸图片太多了，有些我不喜欢，有些我喜欢，但是我也不想专门去一张一张的筛选，就写了这么个小东西，我看到不喜欢的壁纸了，就看看当前是哪张壁纸，找到后删除掉。
也试过其他办法，比如找到Windows的壁纸目录，但是壁纸文件到了那边之后就被修改名字了，我压根不知道这张壁纸在原来的目录的哪个位置。
#include <locale> #include <Windows.h> //设置壁纸 #include <iostream> #include <string> #include <io.h> //获取目录下的文件列表 #include <random> enum class WallpaperStyle { Tile = 0, //平铺 Center = 1, //居中 Stretch = 2, //拉伸 Fit = 3, //适应 Fill = 4 //填充 }; HRESULT SetDesktopEallpaper(wchar_t* pszFile, WallpaperStyle style) { HRESULT hr = S_OK; //以可读写的方式打开HKCU\Control Panel\Desktop注册表项 HKEY hkey = NULL; hr = HRESULT_FROM_WIN32(RegOpenKeyEx(HKEY_CURRENT_USER, L"Control Panel\\Desktop", 0, KEY_READ | KEY_WRITE, &hkey)); if (SUCCEEDED(hr)) { PWSTR pszWallpaperStyle{}, pszTileWallpaper{}; WCHAR WallpaperStyle[][3] = { L"0" ,L"1",L"2", L"6",L"10" }; //std::wstring WallpaperStyle[]{ L"0" ,L"1",L"2", L"6",L"10" }; switch (style) { case WallpaperStyle::Tile: pszWallpaperStyle = WallpaperStyle[0]; pszTileWallpaper = WallpaperStyle[1]; break; case WallpaperStyle::Center: pszWallpaperStyle = WallpaperStyle[0]; pszTileWallpaper = WallpaperStyle[0]; break; case WallpaperStyle::Stretch: pszWallpaperStyle = WallpaperStyle[2]; pszTileWallpaper = WallpaperStyle[0]; break; case WallpaperStyle::Fit: pszWallpaperStyle = WallpaperStyle[3]; pszTileWallpaper = WallpaperStyle[0]; break; case WallpaperStyle::Fill: pszWallpaperStyle = WallpaperStyle[4]; pszTileWallpaper = WallpaperStyle[0]; break; default: pszWallpaperStyle = WallpaperStyle[4]; pszTileWallpaper = WallpaperStyle[0]; break; } //设置pszWallpaperStyle和pszTileWallpaper到注册表项中 DWORD cdData = lstrlen(pszWallpaperStyle) * sizeof(*pszWallpaperStyle); hr = HRESULT_FROM_WIN32(RegSetValueEx(hkey, L"WallpaperStyle", 0, REG_SZ, reinterpret_cast<const BYTE*>(pszWallpaperStyle), cdData)); if (SUCCEEDED(hr)) { cdData = lstrlen(pszTileWallpaper) * sizeof(*pszTileWallpaper); hr = HRESULT_FROM_WIN32(RegSetValueEx(hkey, L"pszTileWallpaper", 0, REG_SZ, reinterpret_cast<const BYTE*>(pszTileWallpaper), cdData)); } RegCloseKey(hkey); } else { std::cout << "打开注册表失败，错误码：" << GetLastError() << std::endl; return hr; } //通过调用Windows 32api 函数SystemParametersInfo 设置桌面壁纸 //SPI_SETDESKWALLPAPER使得壁纸改变保存并持续可见； if (SUCCEEDED(hr)) { if (!SystemParametersInfo(SPI_SETDESKWALLPAPER, 0, static_cast<PVOID>(pszFile), SPIF_UPDATEINIFILE | SPIF_SENDWININICHANGE)) { //实际上，根据微软自己的文档说明，当SystemParametersInfo函数的第一个参数指定为SPI_SETDESKWALLPAPER时，函数总会返回True； //若是文件不存在，权限不够等情况，会使得桌面纯黑色，这样的情况也不会使得函数返回false； hr = HRESULT_FROM_WIN32(GetLastError()); } } return hr; } //获取指定目录下的文件数量 unsigned int getFileNumber(std::wstring& path) { WIN32_FIND_DATAW stFD{}; //存放文件信息的结构体 HANDLE h_handle{}; std::wstring temp = path + L"\\*.jpg"; //查找jpg文件; unsigned int num{ 1 }; h_handle = FindFirstFileW(temp.c_str(), &stFD); if (INVALID_HANDLE_VALUE == h_handle) { //查找失败 std::cout << "查找失败,错误代码：" << GetLastError() << std::endl; return EXIT_FAILURE; } while (FindNextFileW(h_handle, &stFD)) { //继续查找下一个符合条件的文件 //因为后缀的原因，找到的文件不可能是个目录，所以没有写处理目录的情况的代码 num++; } return num; } //获取指定目录下的文件名，使用new分配内存 std::wstring* getFileNames(std::wstring& path, unsigned int num) { if (num == 0) { return nullptr; } std::wstring* ret = new std::wstring[num]{}; WIN32_FIND_DATAW stFD{}; //存放文件信息的结构体 HANDLE h_handle{}; std::wstring temp = path + L"\\*.jpg"; //查找jpg文件; unsigned int index{}; h_handle = FindFirstFileW(temp.c_str(), &stFD); if (INVALID_HANDLE_VALUE == h_handle) { //查找失败 std::cout << "查找失败,错误代码：" << GetLastError() << std::endl; delete[]ret; return nullptr; } else { ret[index] = stFD.cFileName; index += 2; //强迫症犯了，后面的ret[index-1] = stFD.cFileName;，下标必须－1，不然就有警告说可能会越界~ } while (FindNextFileW(h_handle, &stFD)) { //继续查找下一个符合条件的文件 //因为后缀的原因，找到的文件不可能是个目录，所以没有写处理目录的情况的代码 ret[index-1] = stFD.cFileName; index++; } return ret; } //检查文件是否存在 bool FindFirstFileExists(std::wstring& path) { WIN32_FIND_DATAW stFD{}; //存放文件信息的结构体 HANDLE hFind = FindFirstFile(path.c_str(), &stFD); if (INVALID_HANDLE_VALUE == hFind) { //查找失败，文件不存在 //std::cout << "文件查找失败,错误代码：" << GetLastError() << std::endl; return false; } else { return true; } } int main(void) { std::locale::global(std::locale("")); std::wstring* pathList{}; std::wstring path = LR"(C:\Users\gyj\Pictures\美女)"; unsigned int fileNum{}; WallpaperStyle style{ WallpaperStyle::Fill }; PWSTR filePath{}; std::wstring picPath = path + L"\\"; std::random_device rd; // 非确定性生成器 std::mt19937 gen(rd()); // 将随机数种子传递过去 while (true) { fileNum = getFileNumber(path); std::wcout << L"文件数量：" << fileNum << std::endl; pathList = getFileNames(path, fileNum); if (pathList != nullptr) { for (unsigned int i{}; i < fileNum; i++) { std::uniform_int_distribution<> dist(0, fileNum - 1); //设置随机数范围 unsigned int index = dist(gen); //生成随机数 picPath += pathList[index]; if (FindFirstFileExists(picPath)) { Sleep(30*1000); //30秒换一次壁纸 filePath = new WCHAR[picPath.size() + 1]{}; memcpy(filePath, picPath.c_str(), sizeof(wchar_t) * picPath.size()); unsigned long ret = SetDesktopEallpaper(filePath, style); //std::wcout << L"filePath：" << filePath << L'\n'; std::wcout << pathList[index] << std::endl; std::wcout << L"第：" << i << L"张。" << std::endl; delete[] filePath; } else { std::wcout << L"不存在：" << picPath << std::endl; } picPath = path + L"\\"; } } else { std::wcout << L"pathList=nullptr\n" << L"path=" << path << L"\nfileNum：" << fileNum << std::endl; } delete[]pathList; } return EXIT_SUCCESS; }
