帮我用C++读文件夹path内的文件,用结构体fk存数据,文件名 包含jpg时候 存入fk.pic ,文件名包含 xlx时候 存入fk.xs,文件名包含 NC时候 存入vector<string> vF,遇到下一个文件包含 jpg之前,fk存入 vFF , 返回vector<fk> vFF,写出完整代码
时间: 2024-03-03 19:51:18 浏览: 73
c++读取文件中的文件名
好的,以下是实现上述功能的C++代码:
```c++
#include <iostream>
#include <fstream>
#include <string>
#include <vector>
#include <dirent.h>
using namespace std;
struct fk {
string pic;
string xs;
vector<string> vF;
};
vector<fk> readFiles(string path) {
vector<fk> vFF;
fk f;
bool isJpg = false;
bool isXlx = false;
bool isNC = false;
string fileName;
DIR* dir;
struct dirent* ptr;
dir = opendir(path.c_str()); // 打开目录
while ((ptr = readdir(dir)) != NULL) { // 循环读取目录下的文件
fileName = ptr->d_name;
if (fileName.find(".jpg") != string::npos) {
f.pic = fileName;
isJpg = true;
} else if (fileName.find(".xlx") != string::npos) {
f.xs = fileName;
isXlx = true;
} else if (fileName.find("NC") != string::npos) {
f.vF.push_back(fileName);
isNC = true;
}
if (isJpg && (isXlx || isNC)) {
vFF.push_back(f);
f = fk();
isJpg = false;
isXlx = false;
isNC = false;
}
}
closedir(dir);
return vFF;
}
int main() {
vector<fk> vFF = readFiles("path/to/directory");
for (int i = 0; i < vFF.size(); i++) {
cout << "File " << i << ":" << endl;
cout << "pic: " << vFF[i].pic << endl;
cout << "xs: " << vFF[i].xs << endl;
cout << "vF: ";
for (int j = 0; j < vFF[i].vF.size(); j++) {
cout << vFF[i].vF[j] << " ";
}
cout << endl;
}
return 0;
}
```
其中,`readFiles`函数接收一个字符串类型的参数`path`,返回一个`vector<fk>`类型的向量,表示读取到的文件信息。
在函数中,首先定义了一个结构体`fk`,用于存储每个文件的信息。然后定义一个`fk`类型的变量`f`,用于存储当前正在处理的文件信息。接着定义了三个布尔类型的变量`isJpg`、`isXlx`和`isNC`,用于判断当前文件名是否包含指定的字符串。定义了一个字符串类型的变量`fileName`,用于存储当前正在处理的文件名。
接下来,使用`opendir`函数打开目录,循环使用`readdir`函数读取目录下的文件。对于每个文件,如果文件名包含`.jpg`,则将其存入`fk`类型变量`f`的`pic`成员中,同时将`isJpg`设置为`true`;如果文件名包含`.xlx`,则将其存入`f`的`xs`成员中,同时将`isXlx`设置为`true`;如果文件名包含`NC`,则将其存入`f`的`vF`向量中,同时将`isNC`设置为`true`。如果当前文件名包含`.jpg`,且之前的文件名包含`.xlx`或`NC`,则将当前`fk`类型变量`f`存入`vector<fk>`类型的向量`vFF`中,同时重新初始化`f`、`isJpg`、`isXlx`和`isNC`。
最后,使用`closedir`函数关闭目录,返回`vFF`向量。
在`main`函数中,首先调用`readFiles`函数读取目录下的文件,并输出每个文件的信息,以验证程序的正确性。
阅读全文