RLECompressor/FileSystem.cpp at master · ArfanR/RLECompressor · GitHub

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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#include "FileSystem.h"
#include <string>

#ifdef _WIN32
// Windows
#include <windows.h>
#else
// Mac
#include <sys/types.h>
#include <sys/stat.h>
#include <dirent.h>
#endif

namespace FileSystem
{

#ifdef _WIN32
// Windows
void getFilesInDirectory(const std::string& directory, std::list<std::string>& files)
{
	files.clear();

	// Find the first file in the directory.
	WIN32_FIND_DATA ffd;
	std::string path = directory + "\\*.*";
	HANDLE hFind = FindFirstFile(path.c_str(), &ffd);
	if (INVALID_HANDLE_VALUE == hFind)
	{
		// No files found, display error message and return!
		return;
	}

	// Add all files in this directory to a list of strings
	do
	{
		std::string temp = ffd.cFileName;
		if (temp != "." && temp != ".." &&
			!(ffd.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN))
		{
			files.push_back(temp);
		}
	} while (FindNextFile(hFind, &ffd) != 0);
}

void createDirectory(const std::string& directory)
{
	CreateDirectory(directory.c_str(), NULL);
}
#else
// Mac
void getFilesInDirectory(const std::string& directory, std::list<std::string>& files)
{
	// Thanks http://www.lemoda.net/c/list-directory/
	DIR* d;
	const char* dir_name = directory.c_str();

	// Open
	d = opendir(dir_name);

	if (d)
	{
		while (1)
		{
			dirent* entry;

			entry = readdir(d);
			if (!entry)
			{
				break;
			}

			std::string fileName(entry->d_name);

			// Ignore hidden stuff
			if (fileName[0] != '.')
			{
				files.push_back(fileName);
			}
		}

		// Close
		closedir(d);
	}
}

void createDirectory(const std::string& directory)
{
	mkdir(directory.c_str(), 0700);
}
#endif

} // namespace