【C++项目实战】批量读写文件

#include <iostream>
#include <sys/types.h>
#include <dirent.h>
#include <vector>
#include <string.h>
#include <fstream>
#include <regex>
#include <string>

 
using namespace std;
  
void GetFileNames(string path,vector<string>& filenames)
{
    DIR *pDir;
    struct dirent* ptr;
    if(!(pDir = opendir(path.c_str())))
        return;
    while((ptr = readdir(pDir))!=0) {
        if (strcmp(ptr->d_name, ".") != 0 && strcmp(ptr->d_name, "..") != 0)
            //filenames.push_back(path + "/" + ptr->d_name);
            filenames.push_back(ptr->d_name);
    }
    closedir(pDir);
}
string readFile(string path, vector<string> file_name, int i){
    ifstream infile(path+"/"+file_name[i]);    
    std::ostringstream tmp;
    tmp << infile.rdbuf();
    std::string some_str = tmp.str();
    infile.close();
    return some_str;
}
 
int main(int argc, char *argv[]) {
    vector<string> file_name;
    string data;
    string path = argv[1];
    ofstream outfile;
    outfile.open("Enc_psnr_sum.txt"); // outfile
    outfile << "Sequence,Bitrate,Y_PSNR,U_PSNR,V_PSNR,YUV_PSNR,CO_time1,CO_time2" <<endl;
    GetFileNames(path, file_name); 
    for(int i = 0; i <file_name.size(); i++)
    {    
        data = file_name[i] + ",";
        string target = readFile(path, file_name, i); 
        regex r1("(Total)(\\n|.)*");// regular expression - pattern1
        smatch temp; 
        regex_search(target, temp, r1); // regular expression - search1
        regex r("[0-9]+\\.[0-9]+");// regular expression - pattern2
        smatch ret;
        string target2 = temp.str();    
        while(regex_search(target2, ret, r)){ // regular expression - search2
            data += ret.str() + ", ";
            target2 = ret.suffix().str();
        }
        outfile << data << endl;
    }
    outfile.close();
    return 0;
}