C++ 獲取目錄下具有指定后綴名的所有文件名_linux

原博客搬移到：https://blog.csdn.net/u013171226/article/details/107680297 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

/************************************************************
 * Description:
 *     獲取dir目錄下具有制定后綴名字的所有文件名，
 * parameter:
 *     src_dir:目錄路徑，例如 "../test"
 *     ext:后綴名,例如".jpg"
 * return；
 *     vector<string>：包含文件名的數組，
 * ***********************************************************/
vector<string> GetFiles(const char *src_dir, const char *ext)
{
    vector<string> result;
    string directory(src_dir);
    string m_ext(ext);
    //printf("ext length:%d\n",m_ext.length());

    // 打開目錄, DIR是類似目錄句柄的東西
    DIR *dir = opendir(src_dir);
    if ( dir == NULL )
    {
        printf("[ERROR] %s is not a directory or not exist!", src_dir);
        return result;
    }

    // dirent會存儲文件的各種屬性
    struct dirent* d_ent = NULL;

    // linux每個目錄下面都有一個"."和".."要把這兩個都去掉
    char dot[3] = ".";
    char dotdot[6] = "..";

    // 一行一行的讀目錄下的東西,這個東西的屬性放到dirent的變量中
    while ( (d_ent = readdir(dir)) != NULL )
    {
        // 忽略 "." 和 ".."
        if ( (strcmp(d_ent->d_name, dot) != 0) && (strcmp(d_ent->d_name, dotdot) != 0) )
        {
            // d_type可以看到當前的東西的類型,DT_DIR代表當前都到的是目錄,在usr/include/dirent.h中定義的
            if ( d_ent->d_type != DT_DIR)
            {
                string d_name(d_ent->d_name);
                //printf("%s\n",d_ent->d_name);
                if (strcmp(d_name.c_str () + d_name.length () - m_ext.length(), m_ext.c_str ()) == 0)
                {
                    // 構建絕對路徑
                    //string absolutePath = directory + string("/") + string(d_ent->d_name);
                    // 如果傳入的目錄最后是/--> 例如"a/b/", 那么后面直接鏈接文件名
                    //if (directory[directory.length()-1] == '/')
                    //    absolutePath = directory + string(d_ent->d_name);
                    //result.push_back(absolutePath);
                    result.push_back(string(d_ent->d_name));
                }
            }
        }
    }

    // sort the returned files
    sort(result.begin(), result.end());

    closedir(dir);
    return result;
}