c++遍历当前windows目录

原创已于 2023-12-20 13:55:30 修改 · 564 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

c++学习

于 2023-08-09 10:45:30 首次发布

文章介绍了在C++17环境下，如何避免VS中的字符集兼容问题，以及使用`<filesystem>`库简化目录遍历与文件操作。对比了C++11和C++17在处理文件系统的差异。

文章目录

前言
代码
如果使用c++17，就简单很多了

前言

设置vs的高级属性为使用多字节字符集，不然会报char类型的实参与LPCWSTR类型的形参类型不兼容的错误

代码

#include <iostream>
#include <cstring>
#include <windows.h>

void listFiles(const char* dir);

int main()
{
    using namespace std;
    char dir[100] = "D:/prostate_run/";
    /*cout << "Enter a directory (ends with \'\\\'): ";
    cin.getline(dir, 100);*/
    strcat_s(dir, "*.*");    // 需要在目录后面加上*.*表示所有文件/目录
    listFiles(dir);
    return 0;
}

void listFiles(const char* dir)
{
    using namespace std;
    HANDLE hFind;
    WIN32_FIND_DATA findData;
    LARGE_INTEGER size;
    hFind = FindFirstFile(dir, &findData);
    if (hFind == INVALID_HANDLE_VALUE)
    {
        cout << "Failed to find first file!\n";
        return;
    }
    do
    {
        // 忽略"."和".."两个结果 
        if (strcmp(findData.cFileName, ".") == 0 || strcmp(findData.cFileName, "..") == 0)
            continue;
        if (findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)    // 是否是目录 
        {
            cout << findData.cFileName << "\t<dir>\n";
        }
        else
        {
            size.LowPart = findData.nFileSizeLow;
            size.HighPart = findData.nFileSizeHigh;
            cout << findData.cFileName << "\t" << size.QuadPart << " bytes\n";
        }
    } while (FindNextFile(hFind, &findData));
    cout << "Done!\n";
}

在这里插入图片描述

如果使用c++17，就简单很多了

#include"ICommonFuncs.h"
#include<filesystem>
#include<iostream>

namespace fs = std::filesystem;

int main()
{
	std::string dataDir("F:\\dicom2nii_data\\dicom-data");
	fs::path currentPath(dataDir);

	ICommonFuncs ifc;
	ImagePointerType inputImage;

	for (const auto& entry : fs::directory_iterator(currentPath))
	{
		if (fs::is_directory(entry.status()))
		{
			/*std::cout << entry.path() << std::endl;
			std::cout << entry.path().filename() << std::endl;*/

			ifc.readDicomSeries(entry.path().string(), inputImage);
			std::string saveNiiPath(dataDir+"/" +entry.path().filename().string() + ".nii.gz");
			ifc.writeData<ImageType, ImagePointerType>(inputImage, saveNiiPath);
		}
	}

	return 0;
}