8157c392
David Mayerich
added parser and ...
|
1
2
3
4
|
#ifndef STIM_FILENAME_H
#define STIM_FILENAME_H
#include <stdio.h> /* defines FILENAME_MAX */
|
5cda84ab
David Mayerich
putting pranathi ...
|
5
|
#ifdef _WIN32
|
f92397d2
David Mayerich
fixed a Win32 bug...
|
6
|
#include <windows.h>
|
8157c392
David Mayerich
added parser and ...
|
7
8
9
10
11
12
13
14
15
16
17
|
#include <direct.h>
#define GetCurrentDir _getcwd
#define STIM_FILENAME_DIV '\\'
#else
#include <unistd.h>
#define GetCurrentDir getcwd
#define STIM_FILENAME_DIV '/'
#endif
#include <string>
#include <sstream>
|
3d0b6243
David Mayerich
fixed hsiproc bug...
|
18
|
#include <iomanip>
|
8157c392
David Mayerich
added parser and ...
|
19
20
21
|
#include <vector>
#include <stack>
#include <algorithm>
|
f92397d2
David Mayerich
fixed a Win32 bug...
|
22
|
#include <iostream>
|
8157c392
David Mayerich
added parser and ...
|
23
24
|
#include "../parser/parser.h"
|
9b766f1f
Pavel Govyadinov
completed merge f...
|
25
26
27
|
#ifdef BOOST_PRECOMPILED
#include <boost/filesystem.hpp>
#endif
|
8157c392
David Mayerich
added parser and ...
|
28
29
30
31
32
33
34
35
36
|
namespace stim{
//filename class designed to work with both Windows and Unix
class filename{
private:
void init(){
|
8b7be670
David Mayerich
implemented savin...
|
37
|
absolute = false;
|
8157c392
David Mayerich
added parser and ...
|
38
39
40
41
42
43
44
45
46
47
|
}
protected:
std::string drive; //drive letter (for Windows)
std::vector<std::string> path; //directory hierarchy
std::string prefix; //file prefix (without extension)
std::string ext; //file extension
|
8b7be670
David Mayerich
implemented savin...
|
48
49
|
bool absolute; //filename is an absolute path
|
f28202b1
David Mayerich
fixed OS interop ...
|
50
51
52
|
//replaces win32 dividers with the Linux standard
std::string unix_div(std::string s) {
std::replace( s.begin(), s.end(), '\\', '/');
|
8157c392
David Mayerich
added parser and ...
|
53
54
55
|
return s;
}
|
f28202b1
David Mayerich
fixed OS interop ...
|
56
|
//break up a filename into a prefix and extension
|
8157c392
David Mayerich
added parser and ...
|
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
void parse_name(std::string fname){
//find the extension
size_t xpos = fname.find_last_of('.'); //find the position of the extension period (if there is one)
if(xpos != std::string::npos){ //split the prefix and extension
prefix = fname.substr(0, xpos);
ext = fname.substr(xpos + 1);
}
else
prefix = fname;
}
//parse a file locator string
void parse(std::string loc){
|
f28202b1
David Mayerich
fixed OS interop ...
|
72
73
|
loc = unix_div(loc);
|
8157c392
David Mayerich
added parser and ...
|
74
|
//determine the drive (if Windows)
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
75
|
drive = "";
|
8157c392
David Mayerich
added parser and ...
|
76
77
78
|
#ifdef _WIN32
//if the second character is a colon, the character right before it is the drive
if(loc[1] == ':'){
|
8b7be670
David Mayerich
implemented savin...
|
79
|
absolute = true; //this is an absolute path
|
8157c392
David Mayerich
added parser and ...
|
80
|
drive = loc[0]; //copy the drive letter
|
8b7be670
David Mayerich
implemented savin...
|
81
82
83
84
85
86
|
loc = loc.substr(3); //remove the drive information from the locator string
}
#else
if(loc[0] == STIM_FILENAME_DIV){
absolute = true;
loc = loc.substr(1);
|
8157c392
David Mayerich
added parser and ...
|
87
88
89
|
}
#endif
|
8157c392
David Mayerich
added parser and ...
|
90
|
//determine the file name
|
f28202b1
David Mayerich
fixed OS interop ...
|
91
|
std::string fname = loc.substr( loc.find_last_of('/') + 1 ); //find the file name (including extension)
|
8157c392
David Mayerich
added parser and ...
|
92
93
94
95
96
|
//parse the file name
parse_name(fname);
//find the directory hierarchy
|
f28202b1
David Mayerich
fixed OS interop ...
|
97
98
|
std::string dir = loc.substr(0, loc.find_last_of('/') + 1 );
path = stim::parser::split(dir, '/');
|
8157c392
David Mayerich
added parser and ...
|
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
}
public:
filename(){
init();
}
filename(std::string loc){
init();
parse(loc);
}
|
4f007f77
David Mayerich
added better comm...
|
113
|
/// Outputs the file name (including prefix and extension)
|
8157c392
David Mayerich
added parser and ...
|
114
115
116
117
118
119
120
|
std::string get_name(){
if(prefix == "" && ext == "")
return "";
else
return prefix + "." + ext;
}
|
4f007f77
David Mayerich
added better comm...
|
121
|
/// Output the file extension only (usually three characters after a '.')
|
8157c392
David Mayerich
added parser and ...
|
122
123
124
125
|
std::string get_extension(){
return ext;
}
|
4f007f77
David Mayerich
added better comm...
|
126
|
/// Output the file prefix only (name before the extension)
|
8157c392
David Mayerich
added parser and ...
|
127
128
129
130
|
std::string get_prefix(){
return prefix;
}
|
4f007f77
David Mayerich
added better comm...
|
131
132
|
/// Output the entire path (not including the filename)
/// ex. "c:\this\is\the\directory\"
|
8157c392
David Mayerich
added parser and ...
|
133
134
135
|
std::string dir(){
std::stringstream ss;
|
8b7be670
David Mayerich
implemented savin...
|
136
137
138
139
140
141
142
143
|
//if the path is absolute
if(absolute){
//output the drive letter if in Windows
#ifdef _WIN32
ss<<drive<<":";
#endif
ss<<STIM_FILENAME_DIV;
}
|
8157c392
David Mayerich
added parser and ...
|
144
145
146
147
148
149
150
151
152
|
//output the directory
for(unsigned int d = 0; d < path.size(); d++)
ss<<path[d]<<STIM_FILENAME_DIV;
return ss.str();
}
|
4f007f77
David Mayerich
added better comm...
|
153
154
|
/// Output the entire string, including the path and filename
/// ex. "c:\this\is\a\directory\file.txt"
|
8157c392
David Mayerich
added parser and ...
|
155
|
std::string str(){
|
4f007f77
David Mayerich
added better comm...
|
156
157
158
|
std::stringstream ss; //create a string stream
ss<<dir()<<get_name(); //push the directory and filename to that stream
return ss.str(); //convert the stream to a string and return it
|
8157c392
David Mayerich
added parser and ...
|
159
|
}
|
4191c034
Pavel Govyadinov
minor:bug fixes. ...
|
160
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
161
|
//*****************************************************************************************************************
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
162
|
// output is the directory and the prefix and the extension of the files, which are looking for in that directory.
|
4f007f77
David Mayerich
added better comm...
|
163
|
/// David: I have no idea what this is doing. It looks identical to dir()
|
945ee13c
Laila Saadatifard
the get_list func...
|
164
165
|
std::string dir_fname(){
std::stringstream ss;
|
8157c392
David Mayerich
added parser and ...
|
166
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
167
168
169
170
171
172
173
174
|
//if the path is absolute
if(absolute){
//output the drive letter if in Windows
#ifdef _WIN32
ss<<drive<<":";
#endif
ss<<STIM_FILENAME_DIV;
}
|
8157c392
David Mayerich
added parser and ...
|
175
|
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
176
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
177
178
179
|
for(unsigned int d = 0; d < path.size(); d++)
ss<<path[d]<<STIM_FILENAME_DIV;
ss<<get_name();
|
8157c392
David Mayerich
added parser and ...
|
180
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
181
|
return ss.str();
|
8157c392
David Mayerich
added parser and ...
|
182
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
183
|
}
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
184
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
185
186
187
|
// output is the directory and the name of the file which are found in that given directory
std::string f_name(std::string file_name){
std::stringstream ss;
|
8157c392
David Mayerich
added parser and ...
|
188
|
|
4f007f77
David Mayerich
added better comm...
|
189
|
if(absolute){ //if the path is absolute
|
945ee13c
Laila Saadatifard
the get_list func...
|
190
|
#ifdef _WIN32
|
4f007f77
David Mayerich
added better comm...
|
191
|
ss<<drive<<":"; //output the drive letter if in Windows
|
945ee13c
Laila Saadatifard
the get_list func...
|
192
|
#endif
|
4f007f77
David Mayerich
added better comm...
|
193
|
ss<<STIM_FILENAME_DIV; //output a path divider
|
945ee13c
Laila Saadatifard
the get_list func...
|
194
|
}
|
8157c392
David Mayerich
added parser and ...
|
195
|
|
4f007f77
David Mayerich
added better comm...
|
196
197
|
for(unsigned int d = 0; d < path.size(); d++) //for each directory in the current path
ss<<path[d]<<STIM_FILENAME_DIV; //add that directory, interspersing path dividers
|
8157c392
David Mayerich
added parser and ...
|
198
|
|
4f007f77
David Mayerich
added better comm...
|
199
|
stim::filename fn = file_name;
|
945ee13c
Laila Saadatifard
the get_list func...
|
200
201
202
|
std::string fn_prefix = fn.prefix;
std::string fn_ext = fn.ext;
ss<<fn_prefix + '.' + fn_ext;
|
8157c392
David Mayerich
added parser and ...
|
203
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
204
|
return ss.str();
|
8157c392
David Mayerich
added parser and ...
|
205
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
206
|
}
|
8157c392
David Mayerich
added parser and ...
|
207
|
|
4f007f77
David Mayerich
added better comm...
|
208
209
210
211
|
/// Returns a list of files using the current filename as a template.
/// For example:
/// C:\this\is\a\path\file*.txt
/// can be used as a template to find a series of files file001.txt, file002.txt, file003.txt, etc.
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
212
|
std::vector<stim::filename> get_list(){
|
4f007f77
David Mayerich
added better comm...
|
213
214
|
//this is OS dependent, so we're starting with Windows
//the Unix version requires Boost
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
215
|
|
6d242237
David Mayerich
added comments an...
|
216
|
#ifdef _WIN32
|
4f007f77
David Mayerich
added better comm...
|
217
|
stim::filename filepath(dir_fname()); //initialize filepath with the mask
|
8157c392
David Mayerich
added parser and ...
|
218
|
|
4f007f77
David Mayerich
added better comm...
|
219
|
HANDLE hFind = INVALID_HANDLE_VALUE; //allocate data structures for looping through files
|
945ee13c
Laila Saadatifard
the get_list func...
|
220
|
WIN32_FIND_DATAA FindFileData;
|
4f007f77
David Mayerich
added better comm...
|
221
|
std::vector<stim::filename> file_list; //initialize a list to hold all matching filenames
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
222
|
|
4f007f77
David Mayerich
added better comm...
|
223
|
hFind = FindFirstFileA((filepath.str().c_str()), &FindFileData); //find the first file that matches the specified file path
|
945ee13c
Laila Saadatifard
the get_list func...
|
224
|
|
4f007f77
David Mayerich
added better comm...
|
225
226
|
if (hFind == INVALID_HANDLE_VALUE) { //if there are no matching files
printf ("Invalid file handle. Error is %u.\n", GetLastError()); //print an error
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
227
|
}
|
945ee13c
Laila Saadatifard
the get_list func...
|
228
|
else {
|
4f007f77
David Mayerich
added better comm...
|
229
230
231
232
|
std::string file_name = FindFileData.cFileName; //save the file name
std::string file_path = dir(); //the file is in the specified directory, so save it
stim::filename current_file(file_path + file_name); //create a stim::filename structure representing this file
file_list.push_back(current_file); //push the new stim::filename to the file list
|
945ee13c
Laila Saadatifard
the get_list func...
|
233
234
|
// List all the other files in the directory.
|
4f007f77
David Mayerich
added better comm...
|
235
236
237
238
|
while (FindNextFileA(hFind, &FindFileData) != 0){ //iterate until there are no more matching files
file_name = FindFileData.cFileName; //save the next file
current_file = (f_name(file_name)); //append the directory
file_list.push_back(current_file); //push it to the list
|
8157c392
David Mayerich
added parser and ...
|
239
|
}
|
4f007f77
David Mayerich
added better comm...
|
240
|
FindClose(hFind); //close the file data structure
|
43dec788
David Mayerich
added code for si...
|
241
|
}
|
4f007f77
David Mayerich
added better comm...
|
242
|
return file_list; //return the list of files
|
4e2809fa
David Mayerich
fixed an error pr...
|
243
244
|
#elif BOOST_PRECOMPILED
|
9b766f1f
Pavel Govyadinov
completed merge f...
|
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
|
boost::filesystem::path p(dir()); //create a path from the current filename
std::vector<stim::filename> file_list;
if(boost::filesystem::exists(p)){
if(boost::filesystem::is_directory(p)){
typedef std::vector<boost::filesystem::path> vec; // store paths,
vec v; // so we can sort them later
std::copy(boost::filesystem::directory_iterator(p), boost::filesystem::directory_iterator(), back_inserter(v));
std::sort(v.begin(), v.end()); // sort, since directory iteration
// is not ordered on some file systems
//compare file names to the current template (look for wild cards)
for (vec::const_iterator it(v.begin()), it_end(v.end()); it != it_end; ++it)
{
//if the filename is a wild card *or* it matches the read file name
if( prefix == "*" || prefix == (*it).filename().stem().string()){
//if the extension is a wild card *or* it matches the read file extension
if( ext == "*" || "." + ext == (*it).filename().extension().string()){
file_list.push_back((*it).string()); //include it in the final list
}
}
}
}
|
84bae1ab
David Mayerich
fixed an error pr...
|
273
274
|
}
return file_list;
|
4e2809fa
David Mayerich
fixed an error pr...
|
275
276
277
|
#else
std::cout<<"ERROR: UNIX systems don't support file lists without the Boost precompiled libraries."<<std::endl;
exit(1);
|
788862df
Tianshu Cheng
added a BOOST_PRE...
|
278
|
#endif
|
84bae1ab
David Mayerich
fixed an error pr...
|
279
|
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
280
|
}
|
945ee13c
Laila Saadatifard
the get_list func...
|
281
282
|
//**************************************************************************************************
|
8157c392
David Mayerich
added parser and ...
|
283
284
285
286
287
288
289
290
291
292
293
|
//gets the current working directory
static stim::filename cwd(){
char cCurrentPath[FILENAME_MAX];
if (!GetCurrentDir(cCurrentPath, sizeof(cCurrentPath))){
std::cout<<"ERROR getting current working directory."<<std::endl;
exit(1);
}
|
8b7be670
David Mayerich
implemented savin...
|
294
|
std::cout<<cCurrentPath<<std::endl;
|
8157c392
David Mayerich
added parser and ...
|
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
|
return stim::filename(std::string(cCurrentPath) + STIM_FILENAME_DIV);
}
void set_name(std::string fname){
parse_name(fname);
}
//get a path relative to the current one
stim::filename get_relative(std::string rel){
std::vector<std::string> rel_path = stim::parser::split(rel, STIM_FILENAME_DIV);
//if the relative path doesn't contain a file, add a blank string to be used as the filename
if(rel[rel.size()-1] == STIM_FILENAME_DIV)
rel_path.push_back("");
//create a stack representing the current absolute path
std::stack<std::string, std::vector<std::string> > s(path);
//for each token in the relative path
for(int d=0; d<rel_path.size() - 1; d++){
//if the token is a double-dot, pop a directory off of the stack
if(rel_path[d] == "..")
s.pop();
else
s.push(rel_path[d]);
}
std::string* end = &s.top() + 1;
std::string* begin = end - s.size();
std::vector<std::string> result_path(begin, end);
//create a new path to be returned
|
8b7be670
David Mayerich
implemented savin...
|
329
|
stim::filename result = *this;
|
8157c392
David Mayerich
added parser and ...
|
330
331
332
333
334
335
|
result.path = result_path;
result.set_name(rel_path.back());
return result;
}
|
3d0b6243
David Mayerich
fixed hsiproc bug...
|
336
337
338
339
340
341
|
/// This function replaces a wildcard in the prefix with the specified string
stim::filename insert(std::string str){
stim::filename result = *this; //initialize the result filename to the current filename
size_t loc = result.prefix.find('*'); //find a wild card in the string
if(loc == std::string::npos) //if a wildcard isn't found
|
a2bf1d08
David Mayerich
general bug fixes...
|
342
343
344
|
result.prefix += str; //append the value to the prefix
else
result.prefix.replace(loc, 1, str); //replace the wildcard with the string
|
3d0b6243
David Mayerich
fixed hsiproc bug...
|
345
346
347
|
return result; //return the result
}
|
9d3ba0b1
David Mayerich
added stim::hsi a...
|
348
|
stim::filename insert(size_t i, size_t n = 2){
|
3d0b6243
David Mayerich
fixed hsiproc bug...
|
349
350
351
352
353
354
|
std::stringstream ss;
ss << std::setw(n) << std::setfill('0') << i;
return insert(ss.str());
}
|
8b7be670
David Mayerich
implemented savin...
|
355
356
357
358
359
360
361
362
363
|
bool is_relative(){
return !absolute;
}
bool is_absolute(){
return absolute;
}
|
8157c392
David Mayerich
added parser and ...
|
364
365
366
367
368
369
370
371
372
373
374
375
|
};
} //end namespace stim
|
f92397d2
David Mayerich
fixed a Win32 bug...
|
376
|
|
8157c392
David Mayerich
added parser and ...
|
377
|
#endif
|
f92397d2
David Mayerich
fixed a Win32 bug...
|
378
|
|