//去除路径最后多余的斜杠和反斜杠

std::string TrimPath(std::string path)

{

    //string test3("内容");   使用引用字符数组作为参数传给构造函数

    std::string illegal(" \t\f\v\n\r\\/");//string 赋值

    size_t pos = path.find_last_not_of(illegal);//从后往前查找与illegal不匹配的第一个位置,返回的是下标不是位数

    return path.substr(, pos + );

}

//获取当前路径下的所有的图片文件，输入绝对路径，输出完整路径

int FindImages(std::string path, std::vector<std::string>& output)

{

    const size_t FILE_TYPE = ;

    const char* support_format[FILE_TYPE] = { ".bmp", ".dib",             //Windows bitmaps

        ".jpeg", ".jpg",           //JPEG files

        ".png",                    //Portable Network Graphics

        ".pbm", ".pgm", ".ppm",    //Portable image format

        ".tiff", ".tif" }; //TIFF files

    std::string trim_path = TrimPath(path);

    const char* lpPath = trim_path.c_str();

    char szFind[MAX_PATH];

    int n = ;

    WIN32_FIND_DATAA FindFileData;

    strcpy_s(szFind, lpPath);

    strcat_s(szFind, "/*.*");

    HANDLE hFind = ::FindFirstFileA(szFind, &FindFileData);//创建搜索句柄

    if (INVALID_HANDLE_VALUE == hFind)

    {

        return ;

    }

    int i = ;

    do

    {

        if (!(FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))//判断查找的是不是文件夹，通过位于运算，dwFileAttributes有很多属性

        {

            bool is_img = false;

            for (size_t i = ; i != FILE_TYPE; i++)

            {

                std::string lower_name(FindFileData.cFileName);

                transform(lower_name.begin(), lower_name.end(),

                    lower_name.begin(), tolower); //tolower是大写转换成小写的操作，将lower_name全部小写，再存入lower_name

                const char* lower_char = lower_name.c_str();//转换成c语言风格

                //lower_char中含有图片格式，则is_img=true

                if (is_img = bool(strstr(lower_char, support_format[i])))

                    //strstr函数是看第二个参数是不是第一个参数的子串，如果是返回str2在str1中首次出现的地址，否则返回NULL

                {

                    break;

                }

            }

            if (is_img)

            {

                n++;

                std::string img_full_path = lpPath;

                img_full_path += "/";

                img_full_path += FindFileData.cFileName; //这里输出完整路径

                output.push_back(img_full_path);//output是引用类型

            }

        }

        ++i;

    } while (FindNextFileA(hFind, &FindFileData));//当下一个文件目录不存在的时候退出循环

    FindClose(hFind);//关闭搜索句柄

    return n;

}