提取文件名（非查找文件）

曾经有这么一个需要，就想自己实现一个。虽然我知道java内置文件操作的库而C++没有，但还是不想换语言（文人相轻，程序猿更相轻）。于是就上网找了一段C版本的代码，就是这个：

#include <iostream>
#include <vector>
#include <string>
#include <cstring>
#include <io.h>
using namespace std;
/**
*获取文件夹中所有文件名
    input
        path:   文件夹路径
        exd:    文件后缀，如为空则全选
    output
        files:  容器
*/
void getFileName(string path, string exd, vector<string>& files)
{
    /// 文件句柄
    long hFile = 0;
    /// 文件信息
    struct _finddata_t fileinfo;
    string pathName(path), exdName = "\*."+exd;
    if(hFile = _findfirst((pathName+exdName).c_str(), &fileinfo) != -1)
    {
        do
        {///如果有文件夹，迭代之
            if(fileinfo.attrib&_A_SUBDIR)
            {
                if(strcmp(fileinfo.name,".") && strcmp(fileinfo.name,".."))
                    getFileName(pathName+"\"+fileinfo.name, exd, files);
            }
            else
            {
                if(strcmp(fileinfo.name,".") && strcmp(fileinfo.name,".."))
                    files.push_back(fileinfo.name);
            }
        }while(_findnext(hFile, &fileinfo)==0);
        /**这段代码跑出来实际上只能找到一个文件就结束了，找原因还很麻烦
        */
        _findclose(hFile);
    }
}
int main()
{
    string path("E:");
    vector<string> files;
    getFileName(path,"",files);
    for(int i = 0; i < files.size(); i++)
        cout<<files[i]<<endl;
    return 0;
}

结果调了一上午还没调过，真是坑爹。

后来发现用dos也能做，而且简单到让我想撞墙，就像这样

@echo off
dir /s/b *.txt > 文件名.txt
exit

就这么一行，dir就不用说了，/s表示递归遍历，/b表示一个文件一行，*.txt表示查找txt文件，然后写入到一个文件中。

把它保存为bat，放入当前文件夹就可了。不过它输出的是绝对路径，还得自己再加工。

问题是解决了，可是也太快了，我还没感觉呢。

后来，我试着用NB的python解决，在实现的过程中，我感觉我就是在搞翻译。比如我知道要“获取当前文件夹”，然后python里是哪个函数有这个功能呢？找。我知道要“递归遍历”，但是python里是哪个函数有这个功能呢？找。大部分时间就是在看文档，学习语法，模块，并没有什么乐趣，也许是因为我对这个语言不熟吧。最后实现了递归与非递归两个版本，如下所示：

import os
import fnmatch
f=open("图片.txt","w+")
for file in os.listdir(os.getcwd()):
    if fnmatch.fnmatch(file, '*.jpg'):
        f.write(file+"
")
    #    f.write(os.path.realpath(file)+"
")
f.close()

import os
import fnmatch
f=open("图片.txt","w+")
for tup in os.walk(os.getcwd()):
    lst = tup[2]
    dirpath = tup[0]
    for file in lst:
        if fnmatch.fnmatch(file, '*.jpg'):
            f.write(os.path.join(dirpath, file)+"
")
       #     f.write(file+"
")
f.close()

其中，getcwd()就是获取当前目录，listdir()是获取目录下所有子目录（非递归），walk()是递归获取，返回一个list of tuple，tuple包含三个元素，分别是当前目录，list of当前目录下的文件夹，list of当前目录下的文件。fnmatch()就是一个匹配的函数；realpath()返回绝对路径，join()连接路径。不得不说，脚本语言还是很方便的，源代码可以当exe直接运行。

后来，我遇到了吊炸天的Boost，终于又可以用C++实现了。Boost的filesystem库实现了文件操作，而且它的可读性非常好，几乎不需要解读。

#include <iostream>
#include <fstream>
#include <string>
#include <boost/filesystem/operations.hpp>
using namespace std;
using namespace boost;
namespace fs = boost::filesystem;
using rd_iter = fs::recursive_directory_iterator;
using d_iter = fs::directory_iterator;
int main()
{
    string ext;
    cout<<"请输入文件格式（含.）"<<endl;
    cin>>ext;
    ofstream fout("output.txt");
    rd_iter pos("./"), End;
    for(; pos != End; ++pos)
    {
        if(!is_directory(*pos) &&
           pos->path().extension() == ext)
        {   //仅文件
        //    fout << pos->path().filename().string() << endl;
            //绝对路径
        //    fout << system_complete(pos->path()).string() << endl;
            //相对路径
            fout << pos->path().relative_path().string() << endl;
        }
    }
}

用"./"表示当前目录，多么简单直观。d_iter是非递归迭代器，rd_iter是递归迭代器，这里用迭代器来遍历路径。Boost把迭代器玩出魔性来了，好多用法第一次看的时候是跪着看的。更多的内容可以去看Chrono的书，这里就不说了。