Blame view

stim/parser/filename.h 7.05 KB
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
18
19
20
      #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>

  #include <vector>

  #include <stack>

  #include <algorithm>

f92397d2   David Mayerich   fixed a Win32 bug...
21
  #include <iostream>

8157c392   David Mayerich   added parser and ...
22
23
  

  #include "../parser/parser.h"

8157c392   David Mayerich   added parser and ...
24
25
26
27
28
29
30
31
32
  namespace stim{

  

  //filename class designed to work with both Windows and Unix

  

  class filename{

  

  private:

  	void init(){

  

8b7be670   David Mayerich   implemented savin...
33
  		absolute = false;

8157c392   David Mayerich   added parser and ...
34
35
36
37
38
39
40
41
42
43
  

  	}

  

  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...
44
45
  	bool absolute;					//filename is an absolute path

  

8157c392   David Mayerich   added parser and ...
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
  	//replace any incorrect dividers with the appropriate version for the OS

  	std::string parse_div(std::string s) {

  		#ifdef _WIN32

  			std::replace( s.begin(), s.end(), '/', '\\');

  		#else

  			std::replace( s.begin(), s.end(), '\\', '/');

  		#endif

  

  		return s;

  	}

  

  	//parse the file name (prefix and extension)

  	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){

  

  		//determine the drive (if Windows)

  		drive = "";		

  		#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...
78
  				absolute = true;			//this is an absolute path

8157c392   David Mayerich   added parser and ...
79
  				drive = loc[0];				//copy the drive letter

8b7be670   David Mayerich   implemented savin...
80
81
82
83
84
85
  				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 ...
86
87
88
  			}

  		#endif

  

8157c392   David Mayerich   added parser and ...
89
90
91
92
93
94
95
96
97
98
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
  		//determine the file name

  		std::string fname = loc.substr( loc.find_last_of(STIM_FILENAME_DIV) + 1 );	//find the file name (including extension)

  

  		//parse the file name

  		parse_name(fname);

  

  		//find the directory hierarchy

  		std::string dir = loc.substr(0, loc.find_last_of(STIM_FILENAME_DIV) + 1 );

  		path = stim::parser::split(dir, STIM_FILENAME_DIV);

  	}

  

  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...
131
132
133
134
135
136
137
138
  		//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 ...
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
  

  		//output the directory

  		for(unsigned int d = 0; d < path.size(); d++)

  			ss<<path[d]<<STIM_FILENAME_DIV;

  

  		return ss.str();

  

  	}

  

  	std::string str(){

  

  		std::stringstream ss;

  

  		ss<<dir()<<get_name();

  

  		return ss.str();

  	}

4191c034   Pavel Govyadinov   minor:bug fixes. ...
156
  

945ee13c   Laila Saadatifard   the get_list func...
157
158
159
160
  	//*****************************************************************************************************************

  	// output is the directory and the prefix and the extension of the files, which are looking for in that directory. 

  	std::string dir_fname(){

  		std::stringstream ss;

8157c392   David Mayerich   added parser and ...
161
  

945ee13c   Laila Saadatifard   the get_list func...
162
163
164
165
166
167
168
169
  		//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 ...
170
  

945ee13c   Laila Saadatifard   the get_list func...
171
172
173
174
  		

  		for(unsigned int d = 0; d < path.size(); d++)

  			ss<<path[d]<<STIM_FILENAME_DIV;

  		ss<<get_name();

8157c392   David Mayerich   added parser and ...
175
  

945ee13c   Laila Saadatifard   the get_list func...
176
  		return ss.str();

8157c392   David Mayerich   added parser and ...
177
  

945ee13c   Laila Saadatifard   the get_list func...
178
179
180
181
182
  	}

  	

  	// 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 ...
183
  

945ee13c   Laila Saadatifard   the get_list func...
184
185
186
187
188
189
190
191
  		//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 ...
192
  

945ee13c   Laila Saadatifard   the get_list func...
193
194
195
  		//output the directory

  		for(unsigned int d = 0; d < path.size(); d++)

  			ss<<path[d]<<STIM_FILENAME_DIV;

8157c392   David Mayerich   added parser and ...
196
  

945ee13c   Laila Saadatifard   the get_list func...
197
198
199
200
  		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 ...
201
  

945ee13c   Laila Saadatifard   the get_list func...
202
  		return ss.str();

8157c392   David Mayerich   added parser and ...
203
  

945ee13c   Laila Saadatifard   the get_list func...
204
  	}

8157c392   David Mayerich   added parser and ...
205
  

945ee13c   Laila Saadatifard   the get_list func...
206
207
  	

  #ifdef _WIN32

945ee13c   Laila Saadatifard   the get_list func...
208
209
  	//get a list of files matching the current template

  	std::vector<stim::filename> get_list(){

8157c392   David Mayerich   added parser and ...
210
  

945ee13c   Laila Saadatifard   the get_list func...
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
  		//stim::filename file_path;

  		stim::filename filepath(dir_fname());

  		HANDLE            hFind = INVALID_HANDLE_VALUE;

  		WIN32_FIND_DATAA   FindFileData;

  		std::vector<stim::filename> file_list;

  		

  		hFind = FindFirstFileA((filepath.str().c_str()), &FindFileData);

  

  		if (hFind == INVALID_HANDLE_VALUE) {

  		printf ("Invalid file handle. Error is %u.\n", GetLastError());

  		} 

  		else {

  			std::string file_name = FindFileData.cFileName;

  			stim::filename current_file(f_name(file_name));

  			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 ...
232
  			}

945ee13c   Laila Saadatifard   the get_list func...
233
  			FindClose(hFind);

f92397d2   David Mayerich   fixed a Win32 bug...
234
  

8157c392   David Mayerich   added parser and ...
235
236
  		return file_list;

  	}

945ee13c   Laila Saadatifard   the get_list func...
237
238
  	

  	}

788862df   Tianshu Cheng   added a BOOST_PRE...
239
  #endif

945ee13c   Laila Saadatifard   the get_list func...
240
241
  	//**************************************************************************************************

  

8157c392   David Mayerich   added parser and ...
242
243
244
245
246
247
248
249
250
251
252
  	//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...
253
  		 std::cout<<cCurrentPath<<std::endl;

8157c392   David Mayerich   added parser and ...
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
  		 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...
288
  		stim::filename result = *this;

8157c392   David Mayerich   added parser and ...
289
290
291
292
293
294
  		result.path = result_path;

  		result.set_name(rel_path.back());

  

  		return result;

  	}

  

8b7be670   David Mayerich   implemented savin...
295
296
297
298
299
300
301
302
303
  	bool is_relative(){

  		return !absolute;

  	}

  

  	bool is_absolute(){

  		return absolute;

  	}

  

  

8157c392   David Mayerich   added parser and ...
304
305
306
307
308
309
310
311
312
313
314
315
  

  

  

  

  

  

  

  };

  

  

  }	//end namespace stim

  

f92397d2   David Mayerich   fixed a Win32 bug...
316
  

8157c392   David Mayerich   added parser and ...
317
  #endif

f92397d2   David Mayerich   fixed a Win32 bug...
318