jaco
11 years ago
parent
6f1b6eb469
commit
dae0dc9c8c
17 changed files with 548 additions and 346 deletions
@ -0,0 +1,129 @@ |
||||
#include "kyheader.h" |
||||
#include "CmFile.h" |
||||
|
||||
|
||||
// Get image names from a wildcard. Eg: GetNames("D:\\*.jpg", imgNames);
|
||||
int CmFile::GetNames(CStr &_nameW, vecS &_names, string _dir) |
||||
{ |
||||
_dir = GetFolder(_nameW); |
||||
_names.clear(); |
||||
|
||||
DIR *dir; |
||||
struct dirent *ent; |
||||
if((dir = opendir(_dir.c_str()))!=NULL){ |
||||
//print all the files and directories within directory
|
||||
while((ent = readdir(dir))!=NULL){ |
||||
if(ent->d_name[0] == '.') |
||||
continue; |
||||
if(ent->d_type ==4) |
||||
continue; |
||||
_names.push_back(ent->d_name); |
||||
} |
||||
closedir(dir); |
||||
} else { |
||||
perror(""); |
||||
return EXIT_FAILURE; |
||||
} |
||||
return (int)_names.size(); |
||||
} |
||||
int CmFile::GetSubFolders(CStr &folder, vecS &subFolders) |
||||
{ |
||||
subFolders.clear(); |
||||
string nameWC = GetFolder(folder);//folder + "/*";
|
||||
|
||||
DIR *dir; |
||||
struct dirent *ent; |
||||
if((dir = opendir(nameWC.c_str()))!=NULL){ |
||||
while((ent = readdir(dir))!=NULL){ |
||||
if(ent->d_name[0] == '.') |
||||
continue; |
||||
if(ent->d_type == 4){ |
||||
subFolders.push_back(ent->d_name); |
||||
} |
||||
} |
||||
closedir(dir); |
||||
} else { |
||||
perror(""); |
||||
return EXIT_FAILURE; |
||||
} |
||||
return (int)subFolders.size(); |
||||
} |
||||
int CmFile::GetNames(CStr& rootFolder, CStr &fileW, vecS &names) |
||||
{ |
||||
GetNames(rootFolder + fileW, names); |
||||
vecS subFolders, tmpNames; |
||||
int subNum = CmFile::GetSubFolders(rootFolder, subFolders);//
|
||||
for (int i = 0; i < subNum; i++){ |
||||
subFolders[i] += "/"; |
||||
int subNum = GetNames(rootFolder + subFolders[i], fileW, tmpNames); |
||||
for (int j = 0; j < subNum; j++) |
||||
names.push_back(subFolders[i] + tmpNames[j]); |
||||
} |
||||
return (int)names.size(); |
||||
} |
||||
int CmFile::GetNamesNE(CStr& nameWC, vecS &names, string dir, string ext) |
||||
{ |
||||
int fNum = GetNames(nameWC, names, dir); |
||||
ext = GetExtention(nameWC); |
||||
for (int i = 0; i < fNum; i++) |
||||
names[i] = GetNameNE(names[i]); |
||||
return fNum; |
||||
} |
||||
int CmFile::GetNamesNE(CStr& rootFolder, CStr &fileW, vecS &names) |
||||
{ |
||||
int fNum = GetNames(rootFolder, fileW, names); |
||||
int extS = GetExtention(fileW).size(); |
||||
for (int i = 0; i < fNum; i++) |
||||
names[i].resize(names[i].size() - extS); |
||||
return fNum; |
||||
} |
||||
bool CmFile::MkDir(CStr &_path) |
||||
{ |
||||
if(_path.size() == 0) |
||||
return false; |
||||
static char buffer[1024]; |
||||
strcpy(buffer, _S(_path)); |
||||
#ifdef _WIN32 |
||||
for (int i = 0; buffer[i] != 0; i ++) { |
||||
if (buffer[i] == '\\' || buffer[i] == '/') { |
||||
buffer[i] = '\0'; |
||||
CreateDirectoryA(buffer, 0); |
||||
buffer[i] = '/'; |
||||
} |
||||
} |
||||
return CreateDirectoryA(_S(_path), 0); |
||||
#else |
||||
for (int i = 0; buffer[i] != 0; i ++) { |
||||
if (buffer[i] == '\\' || buffer[i] == '/') { |
||||
buffer[i] = '\0'; |
||||
mkdir(buffer, 0); |
||||
buffer[i] = '/'; |
||||
} |
||||
} |
||||
return mkdir(_S(_path), 0); |
||||
#endif |
||||
} |
||||
void CmFile::loadStrList(CStr &fName, vecS & strs, bool flag) |
||||
{ |
||||
ifstream fIn(fName); |
||||
string line; |
||||
//vecS strs;
|
||||
while(getline(fIn, line)){ |
||||
unsigned sz = line.size(); |
||||
if(flag) |
||||
line.resize(sz - 1); |
||||
printf("%s\n",_S(line)); |
||||
strs.push_back(line); |
||||
} |
||||
//return strs;
|
||||
} |
||||
bool CmFile::writeStrList(CStr &fName, const vecS &strs) |
||||
{ |
||||
FILE *f = fopen(_S(fName), "w"); |
||||
if (f == NULL) |
||||
return false; |
||||
for (size_t i = 0; i < strs.size(); i++) |
||||
fprintf(f, "%s\n", _S(strs[i])); |
||||
fclose(f); |
||||
return true; |
||||
} |
||||
@ -0,0 +1,78 @@ |
||||
#pragma once |
||||
|
||||
#ifdef _WIN32 |
||||
#include <windows.h> |
||||
#else |
||||
#include <iostream> |
||||
#include <stdlib.h> |
||||
#include <sys/stat.h> |
||||
#include <dirent.h> |
||||
#endif |
||||
|
||||
struct CmFile |
||||
{ |
||||
|
||||
static inline string GetFolder(CStr& path); |
||||
static inline string GetName(CStr& path); |
||||
static inline string GetNameNE(CStr& path); |
||||
static inline string GetPathNE(CStr& path); |
||||
|
||||
// Get file names from a wildcard. Eg: GetNames("D:\\*.jpg", imgNames);
|
||||
static int GetNames(CStr &nameW, vecS &names, std::string _dir = std::string()); |
||||
static int GetNames(CStr& rootFolder, CStr &fileW, vecS &names); |
||||
static int GetNamesNE(CStr& nameWC, vecS &names, string dir = string(), string ext = string()); |
||||
static int GetNamesNE(CStr& rootFolder, CStr &fileW, vecS &names); |
||||
static inline string GetExtention(CStr name); |
||||
|
||||
static int GetSubFolders(CStr& folder, vecS& subFolders); |
||||
|
||||
static inline string GetWkDir(); |
||||
|
||||
static bool MkDir(CStr& path); |
||||
static void loadStrList(CStr &fName, vecS &strs, bool flag=false); |
||||
static bool writeStrList(CStr &fName, const vecS &strs); |
||||
}; |
||||
|
||||
/************************************************************************/ |
||||
/* Implementation of inline functions */ |
||||
/************************************************************************/ |
||||
string CmFile::GetFolder(CStr& path) |
||||
{ |
||||
return path.substr(0, path.find_last_of("\\/")+1); |
||||
} |
||||
|
||||
string CmFile::GetName(CStr& path) |
||||
{ |
||||
int start = path.find_last_of("\\/")+1; |
||||
int end = path.find_last_not_of(' ')+1; |
||||
return path.substr(start, end - start); |
||||
} |
||||
|
||||
string CmFile::GetNameNE(CStr& path) |
||||
{ |
||||
int start = path.find_last_of("\\/")+1; |
||||
int end = path.find_last_of('.'); |
||||
if (end >= 0) |
||||
return path.substr(start, end - start); |
||||
else |
||||
return path.substr(start, path.find_last_not_of(' ')+1 - start); |
||||
} |
||||
|
||||
string CmFile::GetPathNE(CStr& path) |
||||
{ |
||||
int end = path.find_last_of('.'); |
||||
if (end >= 0) |
||||
return path.substr(0, end); |
||||
else |
||||
return path.substr(0, path.find_last_not_of(' ') + 1); |
||||
} |
||||
|
||||
string CmFile::GetExtention(CStr name) |
||||
{ |
||||
return name.substr(name.find_last_of('.')); |
||||
} |
||||
|
||||
/************************************************************************/ |
||||
/* Implementations */ |
||||
/************************************************************************/ |
||||
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Loading…
Reference in new issue