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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
}
public:
filename(){
init();
}
filename(std::string loc){
init();
parse(loc);
}
std::string get_name(){
if(prefix == "" && ext == "")
return "";
else
return prefix + "." + ext;
}
std::string get_extension(){
return ext;
}
std::string get_prefix(){
return prefix;
}
std::string dir(){
std::stringstream ss;
|
8b7be670
David Mayerich
implemented savin...
|
132
133
134
135
136
137
138
139
|
//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 ...
|
140
141
142
143
144
145
146
147
148
|
//output the directory
for(unsigned int d = 0; d < path.size(); d++)
ss<<path[d]<<STIM_FILENAME_DIV;
return ss.str();
}
|
6422f96a
Pavel Govyadinov
minor bug fixes a...
|
149
150
151
152
153
154
155
156
|
void clear()
{
drive = ""; //drive letter (for Windows)
path.resize(0);
prefix= ""; //file prefix (without extension)
ext = ""; //file extension
}
|
8157c392
David Mayerich
added parser and ...
|
157
158
159
160
161
162
163
164
|
std::string str(){
std::stringstream ss;
ss<<dir()<<get_name();
return ss.str();
}
|
4191c034
Pavel Govyadinov
minor:bug fixes. ...
|
165
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
166
|
//*****************************************************************************************************************
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
167
|
// output is the directory and the prefix and the extension of the files, which are looking for in that directory.
|
945ee13c
Laila Saadatifard
the get_list func...
|
168
169
|
std::string dir_fname(){
std::stringstream ss;
|
8157c392
David Mayerich
added parser and ...
|
170
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
171
172
173
174
175
176
177
178
|
//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 ...
|
179
|
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
180
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
181
182
183
|
for(unsigned int d = 0; d < path.size(); d++)
ss<<path[d]<<STIM_FILENAME_DIV;
ss<<get_name();
|
8157c392
David Mayerich
added parser and ...
|
184
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
185
|
return ss.str();
|
8157c392
David Mayerich
added parser and ...
|
186
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
187
|
}
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
188
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
189
190
191
|
// 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 ...
|
192
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
193
194
195
196
197
198
199
200
|
//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 ...
|
201
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
202
203
204
|
//output the directory
for(unsigned int d = 0; d < path.size(); d++)
ss<<path[d]<<STIM_FILENAME_DIV;
|
8157c392
David Mayerich
added parser and ...
|
205
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
206
207
208
209
|
stim::filename fn = file_name;
std::string fn_prefix = fn.prefix;
std::string fn_ext = fn.ext;
ss<<fn_prefix + '.' + fn_ext;
|
8157c392
David Mayerich
added parser and ...
|
210
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
211
|
return ss.str();
|
8157c392
David Mayerich
added parser and ...
|
212
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
213
|
}
|
8157c392
David Mayerich
added parser and ...
|
214
|
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
215
|
std::vector<stim::filename> get_list(){
|
945ee13c
Laila Saadatifard
the get_list func...
|
216
|
#ifdef _WIN32
|
945ee13c
Laila Saadatifard
the get_list func...
|
217
|
//get a list of files matching the current template
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
218
|
|
8157c392
David Mayerich
added parser and ...
|
219
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
220
221
|
//stim::filename file_path;
stim::filename filepath(dir_fname());
|
f28202b1
David Mayerich
fixed OS interop ...
|
222
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
223
224
225
|
HANDLE hFind = INVALID_HANDLE_VALUE;
WIN32_FIND_DATAA FindFileData;
std::vector<stim::filename> file_list;
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
226
|
|
945ee13c
Laila Saadatifard
the get_list func...
|
227
228
229
|
hFind = FindFirstFileA((filepath.str().c_str()), &FindFileData);
if (hFind == INVALID_HANDLE_VALUE) {
|
43dec788
David Mayerich
added code for si...
|
230
|
printf ("Invalid file handle. Error is %u.\n", GetLastError());
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
231
|
}
|
945ee13c
Laila Saadatifard
the get_list func...
|
232
|
else {
|
f28202b1
David Mayerich
fixed OS interop ...
|
233
234
235
|
std::string file_name = FindFileData.cFileName; //get the file name
std::string file_path = dir();
stim::filename current_file(file_path + file_name);
|
945ee13c
Laila Saadatifard
the get_list func...
|
236
237
238
239
240
241
242
|
file_list.push_back(current_file);
// List all the other files in the directory.
while (FindNextFileA(hFind, &FindFileData) != 0){
file_name = FindFileData.cFileName;
current_file = (f_name(file_name));
file_list.push_back(current_file);
|
8157c392
David Mayerich
added parser and ...
|
243
|
}
|
945ee13c
Laila Saadatifard
the get_list func...
|
244
|
FindClose(hFind);
|
43dec788
David Mayerich
added code for si...
|
245
|
}
|
84bae1ab
David Mayerich
fixed an error pr...
|
246
|
return file_list;
|
4e2809fa
David Mayerich
fixed an error pr...
|
247
248
|
#elif BOOST_PRECOMPILED
|
9b766f1f
Pavel Govyadinov
completed merge f...
|
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
|
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...
|
277
278
|
}
return file_list;
|
4e2809fa
David Mayerich
fixed an error pr...
|
279
280
281
|
#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...
|
282
|
#endif
|
84bae1ab
David Mayerich
fixed an error pr...
|
283
|
|
eb5dfb2b
David Mayerich
fixed linux compa...
|
284
|
}
|
945ee13c
Laila Saadatifard
the get_list func...
|
285
286
|
//**************************************************************************************************
|
8157c392
David Mayerich
added parser and ...
|
287
288
289
290
291
292
293
294
295
296
297
|
//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...
|
298
|
std::cout<<cCurrentPath<<std::endl;
|
8157c392
David Mayerich
added parser and ...
|
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
329
330
331
332
|
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...
|
333
|
stim::filename result = *this;
|
8157c392
David Mayerich
added parser and ...
|
334
335
336
337
338
339
|
result.path = result_path;
result.set_name(rel_path.back());
return result;
}
|
3d0b6243
David Mayerich
fixed hsiproc bug...
|
340
341
342
343
344
345
|
/// 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...
|
346
347
348
|
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...
|
349
350
351
|
return result; //return the result
}
|
9d3ba0b1
David Mayerich
added stim::hsi a...
|
352
|
stim::filename insert(size_t i, size_t n = 2){
|
3d0b6243
David Mayerich
fixed hsiproc bug...
|
353
354
355
356
357
358
|
std::stringstream ss;
ss << std::setw(n) << std::setfill('0') << i;
return insert(ss.str());
}
|
8b7be670
David Mayerich
implemented savin...
|
359
360
361
362
363
364
365
366
367
|
bool is_relative(){
return !absolute;
}
bool is_absolute(){
return absolute;
}
|
8157c392
David Mayerich
added parser and ...
|
368
369
370
371
372
373
374
375
376
377
378
379
|
};
} //end namespace stim
|
f92397d2
David Mayerich
fixed a Win32 bug...
|
380
|
|
8157c392
David Mayerich
added parser and ...
|
381
|
#endif
|
f92397d2
David Mayerich
fixed a Win32 bug...
|
382
|
|