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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
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
envi(){
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
|
}
|
180d7f3a
David Mayerich
added binary file...
|
71
72
73
74
75
76
77
78
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
|
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
|
108
|
{
|
180d7f3a
David Mayerich
added binary file...
|
109
|
std::cout<<"ENVI Header error - data not found for token: "<<get_token(line)<<std::endl;
|
a47a23a9
dmayerich
added ENVI functions
|
110
111
|
exit(1);
}
|
180d7f3a
David Mayerich
added binary file...
|
112
|
return result;
|
a47a23a9
dmayerich
added ENVI functions
|
113
114
|
}
|
180d7f3a
David Mayerich
added binary file...
|
115
|
std::string get_brace_str(std::string token, std::string line, std::ifstream &file)
|
a47a23a9
dmayerich
added ENVI functions
|
116
|
{
|
180d7f3a
David Mayerich
added binary file...
|
117
118
|
//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
|
119
|
|
180d7f3a
David Mayerich
added binary file...
|
120
121
122
123
124
125
126
127
128
129
130
131
132
|
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
|
133
|
{
|
180d7f3a
David Mayerich
added binary file...
|
134
135
|
std::cout<<"ENVI Header error - string token declared without being defined: "<<token<<std::endl;
exit(1);
|
a47a23a9
dmayerich
added ENVI functions
|
136
|
}
|
180d7f3a
David Mayerich
added binary file...
|
137
|
line = line.substr(i+1);
|
a47a23a9
dmayerich
added ENVI functions
|
138
|
|
180d7f3a
David Mayerich
added binary file...
|
139
140
|
//copy character data into the result string until we find a closing brace
while(file)
|
a47a23a9
dmayerich
added ENVI functions
|
141
|
{
|
180d7f3a
David Mayerich
added binary file...
|
142
143
144
145
|
i = line.find_first_of('}');
if(i != std::string::npos)
|
a47a23a9
dmayerich
added ENVI functions
|
146
|
{
|
180d7f3a
David Mayerich
added binary file...
|
147
148
|
result += line.substr(0, i);
break;
|
a47a23a9
dmayerich
added ENVI functions
|
149
|
}
|
180d7f3a
David Mayerich
added binary file...
|
150
151
152
153
|
else
result += line;
getline(file, line);
|
a47a23a9
dmayerich
added ENVI functions
|
154
155
|
}
|
180d7f3a
David Mayerich
added binary file...
|
156
|
if(i == std::string::npos)
|
a47a23a9
dmayerich
added ENVI functions
|
157
|
{
|
180d7f3a
David Mayerich
added binary file...
|
158
159
160
|
std::cout<<"ENVI Header error - string token declared without a terminating '}': "<<token<<std::endl;
exit(1);
}
|
a47a23a9
dmayerich
added ENVI functions
|
161
|
|
180d7f3a
David Mayerich
added binary file...
|
162
|
return trim(result);
|
a47a23a9
dmayerich
added ENVI functions
|
163
164
|
}
|
180d7f3a
David Mayerich
added binary file...
|
165
|
std::vector<std::string> get_string_seq(std::string token, std::string sequence)
|
a47a23a9
dmayerich
added ENVI functions
|
166
|
{
|
180d7f3a
David Mayerich
added binary file...
|
167
168
169
170
171
172
173
174
175
176
177
178
179
180
|
//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
|
181
182
|
}
|
180d7f3a
David Mayerich
added binary file...
|
183
|
std::vector<double> get_double_seq(std::string token, std::string sequence)
|
a47a23a9
dmayerich
added ENVI functions
|
184
|
{
|
180d7f3a
David Mayerich
added binary file...
|
185
186
187
188
189
190
191
192
193
194
195
196
197
198
|
//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
|
199
200
|
}
|
180d7f3a
David Mayerich
added binary file...
|
201
|
bool load(std::string filename)
|
a47a23a9
dmayerich
added ENVI functions
|
202
|
{
|
180d7f3a
David Mayerich
added binary file...
|
203
204
|
//open the header file
std::ifstream file(filename.c_str());
|
a47a23a9
dmayerich
added ENVI functions
|
205
|
|
180d7f3a
David Mayerich
added binary file...
|
206
207
208
|
if(!file)
{
return false;
|
a47a23a9
dmayerich
added ENVI functions
|
209
|
}
|
180d7f3a
David Mayerich
added binary file...
|
210
211
212
213
214
|
//the first line should just be "ENVI"
std::string line;
file>>line;
if(line != "ENVI")
|
6e257ab3
dmayerich
ENVI and colormap...
|
215
|
{
|
180d7f3a
David Mayerich
added binary file...
|
216
|
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...
|
217
218
|
exit(1);
}
|
a47a23a9
dmayerich
added ENVI functions
|
219
|
|
180d7f3a
David Mayerich
added binary file...
|
220
221
|
//for each line in the file, get the token
std::string token;
|
a47a23a9
dmayerich
added ENVI functions
|
222
|
|
180d7f3a
David Mayerich
added binary file...
|
223
224
225
226
|
//get a line
getline(file, line);
while(file)
{
|
a47a23a9
dmayerich
added ENVI functions
|
227
|
|
a47a23a9
dmayerich
added ENVI functions
|
228
|
|
a47a23a9
dmayerich
added ENVI functions
|
229
|
|
180d7f3a
David Mayerich
added binary file...
|
230
231
|
//get the token
token = get_token(line);
|
a47a23a9
dmayerich
added ENVI functions
|
232
|
|
180d7f3a
David Mayerich
added binary file...
|
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
|
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
|
259
|
|
180d7f3a
David Mayerich
added binary file...
|
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
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
|
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
|
312
313
|
}
|
180d7f3a
David Mayerich
added binary file...
|
314
|
void save(std::string filename)
|
a47a23a9
dmayerich
added ENVI functions
|
315
|
{
|
180d7f3a
David Mayerich
added binary file...
|
316
317
318
319
320
321
322
323
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
|
//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
|
365
366
|
}
|
180d7f3a
David Mayerich
added binary file...
|
367
|
void save()
|
a47a23a9
dmayerich
added ENVI functions
|
368
|
{
|
180d7f3a
David Mayerich
added binary file...
|
369
370
|
//std::cout<<"ENVI Header Name: "<<name<<std::endl;
save(name);
|
a47a23a9
dmayerich
added ENVI functions
|
371
372
|
}
|
180d7f3a
David Mayerich
added binary file...
|
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
|
//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
|
397
|
|
180d7f3a
David Mayerich
added binary file...
|
398
399
400
401
|
}
}; //end EnviHeader
}
#endif
|