C++ Code Sample: Get Total Items in Folder
C++ Code Sample: Get Total Items in a Folder
Liệt kê các file có trong một folder sắn có, chức năng này thường được sử dụng nhiều trong các phần mềm cơ bản.
Hãy cùng xem mẫu code cho chức năng này.
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
#include <windows.h> #include <string> #include <vector> wchar_t* StringToWchar(std::string str) { int idx = 0; int count = str.size(); wchar_t *ws_str = new wchar_t[count + 1]; while (idx < str.size()) { ws_str[idx] = (wchar_t)str[idx]; idx++; } ws_str[idx] = 0; return ws_str; } std::string TBasic::WcharToString(wchar_t* wchar_str) { std::string str(""); int idx = 0; while (wchar_str[idx] != 0) { str += (char)wchar_str[idx]; ++idx; } return str; } |
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
std::vector<std::string> GetItemFolder(std::string path) { std::vector<std::string> file_list; std::string search_path = path + "/*.*"; WIN32_FIND_DATA fd; wchar_t* file_path = StringToWchar(search_path); HANDLE hFind = ::FindFirstFile(file_path, &fd); if (hFind != INVALID_HANDLE_VALUE) { do { if (!(fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)) { std::string str = WcharToString(fd.cFileName); file_list.push_back(str); } } while (::FindNextFile(hFind, &fd)); ::FindClose(hFind); } return file_list; } |
|
1 2 3 4 5 |