180d7f3a
David Mayerich
added binary file...
|
1
2
|
#ifndef ENVI_HEADER_H
#define ENVI_HEADER_H
|
a47a23a9
dmayerich
added ENVI functions
|
3
4
|
#include <string>
|
180d7f3a
David Mayerich
added binary file...
|
5
|
#include <iostream>
|
a47a23a9
dmayerich
added ENVI functions
|
6
|
#include <fstream>
|
180d7f3a
David Mayerich
added binary file...
|
7
8
9
10
11
12
13
14
15
16
|
#include <sstream>
#include <vector>
#include <stdlib.h>
//information from an ENVI header file
//A good resource can be found here: http://www.exelisvis.com/docs/enviheaderfiles.html
namespace rts{
struct envi
|
a47a23a9
dmayerich
added ENVI functions
|
17
|
{
|
180d7f3a
David Mayerich
added binary file...
|
18
19
20
|
enum dataType {dummy0, int8, int16, int32, float32, float64, complex32, dummy7, dummy8, complex64, dummy10, dummy11, uint16, uint32, int64, uint64};
enum interleaveType {BIP, BIL, BSQ}; //bip = Z,X,Y; bil = X,Z,Y; bsq = X,Y,Z
enum endianType {endianLittle, endianBig};
|
a47a23a9
dmayerich
added ENVI functions
|
21
|
|
180d7f3a
David Mayerich
added binary file...
|
22
|
std::string name;
|
a47a23a9
dmayerich
added ENVI functions
|
23
|
|
180d7f3a
David Mayerich
added binary file...
|
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
|
std::string description;
unsigned int samples; //x-axis
unsigned int lines; //y-axis
unsigned int bands; //spectral axis
unsigned int header_offset; //header offset for binary file (in bytes)
std::string file_type; //should be "ENVI Standard"
envi::dataType data_type; //data representation; common value is 4 (32-bit float)
envi::interleaveType interleave;
std::string sensor_type; //not really used
envi::endianType byte_order; //little = least significant bit first (most systems)
double x_start, y_start; //coordinates of the upper-left corner of the image
std::string wavelength_units; //stored wavelength units
std::string z_plot_titles[2];
double pixel_size[2]; //pixel size along X and Y
std::vector<std::string> band_names; //name for each band in the image
std::vector<double> wavelength; //wavelength for each band
|
bc14c9de
David Mayerich
removed old rts f...
|
50
|
void init(){
|
180d7f3a
David Mayerich
added binary file...
|
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
name = "";
//specify default values for a new or empty ENVI file
samples = 0;
lines = 0;
bands = 0;
header_offset = 0;
data_type = float32;
interleave = BSQ;
byte_order = endianLittle;
x_start = y_start = 0;
pixel_size[0] = pixel_size[1] = 1;
//strings
file_type = "ENVI Standard";
sensor_type = "Unknown";
wavelength_units = "Unknown";
z_plot_titles[0] = z_plot_titles[1] = "Unknown";
|
a47a23a9
dmayerich
added ENVI functions
|
69
70
|
}
|
bc14c9de
David Mayerich
removed old rts f...
|
71
72
73
74
75
76
77
78
|
envi(){
init();
}
envi(std::string name){
init();
load(name);
}
|
180d7f3a
David Mayerich
added binary file...
|
79
80
81
82
83
84
85
86
87
88
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
|
std::string trim(std::string line){
//trims whitespace from the beginning and end of line
unsigned int start_i, end_i;
for(start_i=0; start_i < line.length(); start_i++)
if(line[start_i] != 32)
{
break;
}
for(end_i = line.length()-1; end_i >= start_i; end_i--)
if(line[end_i] != ' ')
{
break;
}
return line.substr(start_i, end_i - start_i+1);
}
std::string get_token(std::string line){
//returns a variable name; in this case we look for the '=' sign
size_t i = line.find_first_of('=');
std::string result;
if(i != std::string::npos)
result = trim(line.substr(0, i-1));
return result;
}
std::string get_data_str(std::string line){
size_t i = line.find_first_of('=');
std::string result;
if(i != std::string::npos)
result = trim(line.substr(i+1));
else
|
a47a23a9
dmayerich
added ENVI functions
|
116
|
{
|
180d7f3a
David Mayerich
added binary file...
|
117
|
std::cout<<"ENVI Header error - data not found for token: "<<get_token(line)<<std::endl;
|
a47a23a9
dmayerich
added ENVI functions
|
118
119
|
exit(1);
}
|
180d7f3a
David Mayerich
added binary file...
|
120
|
return result;
|
a47a23a9
dmayerich
added ENVI functions
|
121
122
|
}
|
180d7f3a
David Mayerich
added binary file...
|
123
|
std::string get_brace_str(std::string token, std::string line, std::ifstream &file)
|
a47a23a9
dmayerich
added ENVI functions
|
124
|
{
|
180d7f3a
David Mayerich
added binary file...
|
125
126
|
//this function assembles all of the characters between curly braces
//this is how strings are defined within an ENVI file
|
a47a23a9
dmayerich
added ENVI functions
|
127
|
|
180d7f3a
David Mayerich
added binary file...
|
128
129
130
131
132
133
134
135
136
137
138
139
140
|
std::string result;
//first, find the starting brace
size_t i;
do
{
i = line.find_first_of('{');
if(i != std::string::npos)
break;
}while(file);
//if i is still npos, we have reached the end of the file without a brace...something is wrong
if(i == std::string::npos)
|
a47a23a9
dmayerich
added ENVI functions
|
141
|
{
|
180d7f3a
David Mayerich
added binary file...
|
142
143
|
std::cout<<"ENVI Header error - string token declared without being defined: "<<token<<std::endl;
exit(1);
|
a47a23a9
dmayerich
added ENVI functions
|
144
|
}
|
180d7f3a
David Mayerich
added binary file...
|
145
|
line = line.substr(i+1);
|
a47a23a9
dmayerich
added ENVI functions
|
146
|
|
180d7f3a
David Mayerich
added binary file...
|
147
148
|
//copy character data into the result string until we find a closing brace
while(file)
|
a47a23a9
dmayerich
added ENVI functions
|
149
|
{
|
180d7f3a
David Mayerich
added binary file...
|
150
151
152
153
|
i = line.find_first_of('}');
if(i != std::string::npos)
|
a47a23a9
dmayerich
added ENVI functions
|
154
|
{
|
180d7f3a
David Mayerich
added binary file...
|
155
156
|
result += line.substr(0, i);
break;
|
a47a23a9
dmayerich
added ENVI functions
|
157
|
}
|
180d7f3a
David Mayerich
added binary file...
|
158
159
160
161
|
else
result += line;
getline(file, line);
|
a47a23a9
dmayerich
added ENVI functions
|
162
163
|
}
|
180d7f3a
David Mayerich
added binary file...
|
164
|
if(i == std::string::npos)
|
a47a23a9
dmayerich
added ENVI functions
|
165
|
{
|
180d7f3a
David Mayerich
added binary file...
|
166
167
168
|
std::cout<<"ENVI Header error - string token declared without a terminating '}': "<<token<<std::endl;
exit(1);
}
|
a47a23a9
dmayerich
added ENVI functions
|
169
|
|
180d7f3a
David Mayerich
added binary file...
|
170
|
return trim(result);
|
a47a23a9
dmayerich
added ENVI functions
|
171
172
|
}
|
180d7f3a
David Mayerich
added binary file...
|
173
|
std::vector<std::string> get_string_seq(std::string token, std::string sequence)
|
a47a23a9
dmayerich
added ENVI functions
|
174
|
{
|
180d7f3a
David Mayerich
added binary file...
|
175
176
177
178
179
180
181
182
183
184
185
186
187
188
|
//this function returns a sequence of comma-delimited strings
std::vector<std::string> result;
std::string entry;
size_t i;
do
{
i = sequence.find_first_of(',');
entry = sequence.substr(0, i);
sequence = sequence.substr(i+1);
result.push_back(trim(entry));
}while(i != std::string::npos);
return result;
|
a47a23a9
dmayerich
added ENVI functions
|
189
190
|
}
|
180d7f3a
David Mayerich
added binary file...
|
191
|
std::vector<double> get_double_seq(std::string token, std::string sequence)
|
a47a23a9
dmayerich
added ENVI functions
|
192
|
{
|
180d7f3a
David Mayerich
added binary file...
|
193
194
195
196
197
198
199
200
201
202
203
204
205
206
|
//this function returns a sequence of comma-delimited strings
std::vector<double> result;
std::string entry;
size_t i;
do
{
i = sequence.find_first_of(',');
entry = sequence.substr(0, i);
sequence = sequence.substr(i+1);
result.push_back(atof(entry.c_str()));
//std::cout<<entry<<" ";
}while(i != std::string::npos);
return result;
|
a47a23a9
dmayerich
added ENVI functions
|
207
208
|
}
|
180d7f3a
David Mayerich
added binary file...
|
209
|
bool load(std::string filename)
|
a47a23a9
dmayerich
added ENVI functions
|
210
|
{
|
180d7f3a
David Mayerich
added binary file...
|
211
212
|
//open the header file
std::ifstream file(filename.c_str());
|
a47a23a9
dmayerich
added ENVI functions
|
213
|
|
180d7f3a
David Mayerich
added binary file...
|
214
215
216
|
if(!file)
{
return false;
|
a47a23a9
dmayerich
added ENVI functions
|
217
|
}
|
180d7f3a
David Mayerich
added binary file...
|
218
219
220
221
222
|
//the first line should just be "ENVI"
std::string line;
file>>line;
if(line != "ENVI")
|
6e257ab3
dmayerich
ENVI and colormap...
|
223
|
{
|
180d7f3a
David Mayerich
added binary file...
|
224
|
std::cout<<"ENVI Header Error: The header doesn't appear to be an ENVI file. The first line should be 'ENVI'."<<std::endl;
|
6e257ab3
dmayerich
ENVI and colormap...
|
225
226
|
exit(1);
}
|
a47a23a9
dmayerich
added ENVI functions
|
227
|
|
180d7f3a
David Mayerich
added binary file...
|
228
229
|
//for each line in the file, get the token
std::string token;
|
a47a23a9
dmayerich
added ENVI functions
|
230
|
|
180d7f3a
David Mayerich
added binary file...
|
231
232
233
234
|
//get a line
getline(file, line);
while(file)
{
|
a47a23a9
dmayerich
added ENVI functions
|
235
|
|
a47a23a9
dmayerich
added ENVI functions
|
236
|
|
a47a23a9
dmayerich
added ENVI functions
|
237
|
|
180d7f3a
David Mayerich
added binary file...
|
238
239
|
//get the token
token = get_token(line);
|
a47a23a9
dmayerich
added ENVI functions
|
240
|
|
180d7f3a
David Mayerich
added binary file...
|
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
|
if(token == "description")
description = get_brace_str(token, line, file);
else if(token == "band names")
{
std::string string_sequence = get_brace_str(token, line, file);
band_names = get_string_seq(token, string_sequence);
}
else if(token == "wavelength")
{
std::string string_sequence = get_brace_str(token, line, file);
wavelength = get_double_seq(token, string_sequence);
}
else if(token == "pixel size")
{
std::string string_sequence = get_brace_str(token, line, file);
std::vector<double> pxsize = get_double_seq(token, string_sequence);
pixel_size[0] = pxsize[0];
pixel_size[1] = pxsize[1];
}
else if(token == "z plot titles")
{
std::string string_sequence = get_brace_str(token, line, file);
std::vector<std::string> titles = get_string_seq(token, string_sequence);
z_plot_titles[0] = titles[0];
z_plot_titles[1] = titles[1];
}
|
a47a23a9
dmayerich
added ENVI functions
|
267
|
|
180d7f3a
David Mayerich
added binary file...
|
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
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
|
else if(token == "samples")
samples = atoi(get_data_str(line).c_str());
else if(token == "lines")
lines = atoi(get_data_str(line).c_str());
else if(token == "bands")
bands = atoi(get_data_str(line).c_str());
else if(token == "header offset")
header_offset = atoi(get_data_str(line).c_str());
else if(token == "file type")
file_type = get_data_str(line);
else if(token == "data type")
data_type = (dataType)atoi(get_data_str(line).c_str());
else if(token == "interleave")
{
std::string interleave_str = get_data_str(line);
if(interleave_str == "bip")
interleave = BIP;
else if(interleave_str == "bil")
interleave = BIL;
else if(interleave_str == "bsq")
interleave = BSQ;
}
else if(token == "sensor type")
sensor_type = get_data_str(line);
else if(token == "byte order")
byte_order = (endianType)atoi(get_data_str(line).c_str());
else if(token == "x start")
x_start = atof(get_data_str(line).c_str());
else if(token == "y start")
y_start = atof(get_data_str(line).c_str());
else if(token == "wavelength units")
wavelength_units = get_data_str(line);
//get the next line
getline(file, line);
}
//make sure the number of bands matches the number of wavelengths
unsigned int wavelengths = wavelength.size();
if(bands != wavelengths)
{
std::cout<<"ENVI Header Error -- Number of wavelengths doesn't match the number of bands. Bands = "<<bands<<", Wavelengths = "<<wavelength.size()<<std::endl;
exit(1);
}
//close the file
file.close();
//set the file name
name = filename;
return true;
|
a47a23a9
dmayerich
added ENVI functions
|
320
321
|
}
|
180d7f3a
David Mayerich
added binary file...
|
322
|
void save(std::string filename)
|
a47a23a9
dmayerich
added ENVI functions
|
323
|
{
|
180d7f3a
David Mayerich
added binary file...
|
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
|
//open a file
std::ofstream outfile(filename.c_str());
//write the ENVI type identifier
outfile<<"ENVI"<<std::endl;
//output all of the data
outfile<<"description = {"<<std::endl;
outfile<<" "<<description<<"}"<<std::endl;
outfile<<"samples = "<<samples<<std::endl;
outfile<<"lines = "<<lines<<std::endl;
outfile<<"bands = "<<bands<<std::endl;
outfile<<"header offset = "<<header_offset<<std::endl;
outfile<<"file type = "<<file_type<<std::endl;
outfile<<"data type = "<<data_type<<std::endl;
outfile<<"interleave = ";
if(interleave == BIP)
outfile<<"bip";
if(interleave == BIL)
outfile<<"bil";
if(interleave == BSQ)
outfile<<"bsq";
outfile<<std::endl;
outfile<<"sensor type = "<<sensor_type<<std::endl;
outfile<<"byte order = "<<byte_order<<std::endl;
outfile<<"x start = "<<x_start<<std::endl;
outfile<<"y start = "<<y_start<<std::endl;
outfile<<"wavelength units = "<<wavelength_units<<std::endl;
outfile<<"z plot titles = {";
outfile<<z_plot_titles[0]<<", "<<z_plot_titles[1]<<"}"<<std::endl;
outfile<<"pixel size = {"<<pixel_size[0]<<", "<<pixel_size[1]<<", units=Meters}"<<std::endl;
if(band_names.size() > 0)
{
outfile<<"band names = {"<<std::endl;
for(unsigned int i=0; i<band_names.size(); i++)
{
outfile<<band_names[i];
if(i < band_names.size() - 1)
outfile<<", ";
}
outfile<<"}"<<std::endl;
}
outfile<<"wavelength = {"<<std::endl;
for(unsigned int i=0; i<wavelength.size()-1; i++)
outfile<<wavelength[i]<<", ";
outfile<<wavelength.back()<<"}"<<std::endl;
outfile.close();
|
a47a23a9
dmayerich
added ENVI functions
|
373
374
|
}
|
180d7f3a
David Mayerich
added binary file...
|
375
|
void save()
|
a47a23a9
dmayerich
added ENVI functions
|
376
|
{
|
180d7f3a
David Mayerich
added binary file...
|
377
378
|
//std::cout<<"ENVI Header Name: "<<name<<std::endl;
save(name);
|
a47a23a9
dmayerich
added ENVI functions
|
379
380
|
}
|
180d7f3a
David Mayerich
added binary file...
|
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
|
//returns the size of a single value (in bytes)
unsigned int valsize()
{
switch(data_type)
{
case int8: //1 = 8-bit byte
return 1;
case int16: //16-bit signed integer
case uint16: //16-bit unsigned integer
return 2;
case int32: //32-bit signed long integer
case float32: //32-bit floating point
case uint32: //32-bit unsigned long integer
return 4;
case float64: //64-bit double-precision floating point
case complex32: //32-bit complex value
case int64: //64-bit signed long integer
case uint64: //64-bit unsigned long integer
return 8;
case complex64: //64-bit complex value
return 16;
default:
return 0;
}
|
a47a23a9
dmayerich
added ENVI functions
|
405
|
|
180d7f3a
David Mayerich
added binary file...
|
406
407
408
409
|
}
}; //end EnviHeader
}
#endif
|