e8eb202f
David Mayerich
added a new ENVI ...
|
1
2
3
|
#ifndef ENVI_HEADER_H
#define ENVI_HEADER_H
|
6e257ab3
dmayerich
ENVI and colormap...
|
4
|
#include <string>
|
e8eb202f
David Mayerich
added a new ENVI ...
|
5
|
#include <iostream>
|
6e257ab3
dmayerich
ENVI and colormap...
|
6
|
#include <fstream>
|
e8eb202f
David Mayerich
added a new ENVI ...
|
7
8
|
#include <sstream>
#include <vector>
|
dc8eb8aa
David Mayerich
added band trimmi...
|
9
|
#include <algorithm>
|
e8eb202f
David Mayerich
added a new ENVI ...
|
10
11
|
#include <stdlib.h>
|
6e257ab3
dmayerich
ENVI and colormap...
|
12
|
//information from an ENVI header file
|
e8eb202f
David Mayerich
added a new ENVI ...
|
13
14
|
//A good resource can be found here: http://www.exelisvis.com/docs/enviheaderfiles.html
|
8a86bd56
David Mayerich
changed rts names...
|
15
|
namespace stim{
|
e8eb202f
David Mayerich
added a new ENVI ...
|
16
17
|
struct envi_header
|
6e257ab3
dmayerich
ENVI and colormap...
|
18
|
{
|
e8eb202f
David Mayerich
added a new ENVI ...
|
19
20
21
22
|
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};
|
6e257ab3
dmayerich
ENVI and colormap...
|
23
24
25
26
|
std::string name;
std::string description;
|
d0ef98e3
David Mayerich
fixed a Linux com...
|
27
28
29
30
|
size_t samples; //x-axis
size_t lines; //y-axis
size_t bands; //spectral axis
size_t header_offset; //header offset for binary file (in bytes)
|
e8eb202f
David Mayerich
added a new ENVI ...
|
31
32
33
|
std::string file_type; //should be "ENVI Standard"
envi_header::dataType data_type; //data representation; common value is 4 (32-bit float)
|
6e257ab3
dmayerich
ENVI and colormap...
|
34
35
|
|
e8eb202f
David Mayerich
added a new ENVI ...
|
36
|
envi_header::interleaveType interleave;
|
6e257ab3
dmayerich
ENVI and colormap...
|
37
38
39
|
std::string sensor_type; //not really used
|
e8eb202f
David Mayerich
added a new ENVI ...
|
40
|
envi_header::endianType byte_order; //little = least significant bit first (most systems)
|
6e257ab3
dmayerich
ENVI and colormap...
|
41
42
43
44
45
46
47
48
49
50
|
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
|
e8eb202f
David Mayerich
added a new ENVI ...
|
51
52
|
void init(){
name = "";
|
6e257ab3
dmayerich
ENVI and colormap...
|
53
54
55
56
57
58
|
//specify default values for a new or empty ENVI file
samples = 0;
lines = 0;
bands = 0;
header_offset = 0;
|
e8eb202f
David Mayerich
added a new ENVI ...
|
59
60
61
|
data_type = float32;
interleave = BSQ;
byte_order = endianLittle;
|
6e257ab3
dmayerich
ENVI and colormap...
|
62
63
64
65
66
67
68
69
70
71
|
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";
}
|
e8eb202f
David Mayerich
added a new ENVI ...
|
72
73
74
75
76
77
78
79
80
|
envi_header(){
init();
}
envi_header(std::string name){
init();
load(name);
}
std::string trim(std::string line){
|
3d0b6243
David Mayerich
fixed hsiproc bug...
|
81
82
83
|
if(line.length() == 0)
return line;
|
6e257ab3
dmayerich
ENVI and colormap...
|
84
|
//trims whitespace from the beginning and end of line
|
9d3ba0b1
David Mayerich
added stim::hsi a...
|
85
|
size_t start_i, end_i;
|
6e257ab3
dmayerich
ENVI and colormap...
|
86
87
88
89
90
91
92
|
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--)
|
f6aa45e7
David Mayerich
fixed Windows EOL...
|
93
|
if(line[end_i] != ' ' && line[end_i] != '\r')
|
6e257ab3
dmayerich
ENVI and colormap...
|
94
95
96
97
98
99
100
101
|
{
break;
}
return line.substr(start_i, end_i - start_i+1);
}
|
e8eb202f
David Mayerich
added a new ENVI ...
|
102
|
std::string get_token(std::string line){
|
6e257ab3
dmayerich
ENVI and colormap...
|
103
104
105
106
107
108
109
110
111
112
|
//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;
}
|
e8eb202f
David Mayerich
added a new ENVI ...
|
113
|
std::string get_data_str(std::string line){
|
6e257ab3
dmayerich
ENVI and colormap...
|
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
|
size_t i = line.find_first_of('=');
std::string result;
if(i != std::string::npos)
result = trim(line.substr(i+1));
else
{
std::cout<<"ENVI Header error - data not found for token: "<<get_token(line)<<std::endl;
exit(1);
}
return result;
}
std::string get_brace_str(std::string token, std::string line, std::ifstream &file)
{
//this function assembles all of the characters between curly braces
//this is how strings are defined within an ENVI file
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)
{
std::cout<<"ENVI Header error - string token declared without being defined: "<<token<<std::endl;
exit(1);
}
line = line.substr(i+1);
//copy character data into the result string until we find a closing brace
while(file)
{
i = line.find_first_of('}');
if(i != std::string::npos)
{
result += line.substr(0, i);
break;
}
else
result += line;
getline(file, line);
}
if(i == std::string::npos)
{
std::cout<<"ENVI Header error - string token declared without a terminating '}': "<<token<<std::endl;
exit(1);
}
return trim(result);
}
std::vector<std::string> get_string_seq(std::string token, std::string sequence)
{
//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;
}
std::vector<double> get_double_seq(std::string token, std::string sequence)
{
//this function returns a sequence of comma-delimited strings
|
e8eb202f
David Mayerich
added a new ENVI ...
|
198
|
std::vector<double> result;
|
6e257ab3
dmayerich
ENVI and colormap...
|
199
200
201
202
203
|
std::string entry;
size_t i;
do
{
i = sequence.find_first_of(',');
|
e8eb202f
David Mayerich
added a new ENVI ...
|
204
|
entry = sequence.substr(0, i);
|
6e257ab3
dmayerich
ENVI and colormap...
|
205
|
sequence = sequence.substr(i+1);
|
e8eb202f
David Mayerich
added a new ENVI ...
|
206
|
result.push_back(atof(entry.c_str()));
|
0174d823
dmayerich
bug fixes and exits
|
207
|
//std::cout<<entry<<" ";
|
6e257ab3
dmayerich
ENVI and colormap...
|
208
209
210
211
212
213
214
215
216
217
218
|
}while(i != std::string::npos);
return result;
}
bool load(std::string filename)
{
//open the header file
std::ifstream file(filename.c_str());
if(!file)
|
e8eb202f
David Mayerich
added a new ENVI ...
|
219
|
{
|
6708cc25
heziqi
Ziqi added envi c...
|
220
|
std::cout<<"ERROR: unable to open header file: "<<filename<<std::endl;
|
6e257ab3
dmayerich
ENVI and colormap...
|
221
222
223
224
225
226
227
228
229
230
|
return false;
}
//the first line should just be "ENVI"
std::string line;
file>>line;
if(line != "ENVI")
{
std::cout<<"ENVI Header Error: The header doesn't appear to be an ENVI file. The first line should be 'ENVI'."<<std::endl;
exit(1);
|
e8eb202f
David Mayerich
added a new ENVI ...
|
231
|
}
|
6e257ab3
dmayerich
ENVI and colormap...
|
232
233
|
//for each line in the file, get the token
|
e8eb202f
David Mayerich
added a new ENVI ...
|
234
235
|
std::string token;
|
0174d823
dmayerich
bug fixes and exits
|
236
237
|
//get a line
getline(file, line);
|
6e257ab3
dmayerich
ENVI and colormap...
|
238
239
240
|
while(file)
{
|
0174d823
dmayerich
bug fixes and exits
|
241
|
|
6e257ab3
dmayerich
ENVI and colormap...
|
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
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
|
//get the token
token = get_token(line);
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];
}
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")
|
e8eb202f
David Mayerich
added a new ENVI ...
|
284
|
data_type = (dataType)atoi(get_data_str(line).c_str());
|
6e257ab3
dmayerich
ENVI and colormap...
|
285
286
287
288
|
else if(token == "interleave")
{
std::string interleave_str = get_data_str(line);
if(interleave_str == "bip")
|
e8eb202f
David Mayerich
added a new ENVI ...
|
289
|
interleave = BIP;
|
6e257ab3
dmayerich
ENVI and colormap...
|
290
|
else if(interleave_str == "bil")
|
e8eb202f
David Mayerich
added a new ENVI ...
|
291
|
interleave = BIL;
|
6e257ab3
dmayerich
ENVI and colormap...
|
292
|
else if(interleave_str == "bsq")
|
e8eb202f
David Mayerich
added a new ENVI ...
|
293
|
interleave = BSQ;
|
6e257ab3
dmayerich
ENVI and colormap...
|
294
295
296
297
|
}
else if(token == "sensor type")
sensor_type = get_data_str(line);
else if(token == "byte order")
|
e8eb202f
David Mayerich
added a new ENVI ...
|
298
|
byte_order = (endianType)atoi(get_data_str(line).c_str());
|
6e257ab3
dmayerich
ENVI and colormap...
|
299
300
301
302
303
|
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")
|
e8eb202f
David Mayerich
added a new ENVI ...
|
304
305
306
|
wavelength_units = get_data_str(line);
//get the next line
|
0174d823
dmayerich
bug fixes and exits
|
307
|
getline(file, line);
|
e8eb202f
David Mayerich
added a new ENVI ...
|
308
309
310
|
}
//make sure the number of bands matches the number of wavelengths
|
9d3ba0b1
David Mayerich
added stim::hsi a...
|
311
|
size_t wavelengths = wavelength.size();
|
ee4dea28
David Mayerich
fixed errors in c...
|
312
|
if(wavelengths && bands != wavelengths)
|
e8eb202f
David Mayerich
added a new ENVI ...
|
313
314
315
|
{
std::cout<<"ENVI Header Error -- Number of wavelengths doesn't match the number of bands. Bands = "<<bands<<", Wavelengths = "<<wavelength.size()<<std::endl;
exit(1);
|
0174d823
dmayerich
bug fixes and exits
|
316
|
}
|
6e257ab3
dmayerich
ENVI and colormap...
|
317
|
|
721fd9d1
David Mayerich
fixed problems wi...
|
318
319
320
321
322
323
|
if(wavelength.size() == 0){ //if no list of wavelengths is given, create one based on band numbers
wavelength.resize(bands);
for(size_t b = 0; b < bands; b++)
wavelength[b] = (double)b;
}
|
6e257ab3
dmayerich
ENVI and colormap...
|
324
|
//close the file
|
e8eb202f
David Mayerich
added a new ENVI ...
|
325
326
327
328
329
|
file.close();
//set the file name
name = filename;
|
6e257ab3
dmayerich
ENVI and colormap...
|
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
|
return true;
}
void save(std::string filename)
{
//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 = ";
|
e8eb202f
David Mayerich
added a new ENVI ...
|
352
|
if(interleave == BIP)
|
6e257ab3
dmayerich
ENVI and colormap...
|
353
|
outfile<<"bip";
|
e8eb202f
David Mayerich
added a new ENVI ...
|
354
|
if(interleave == BIL)
|
6e257ab3
dmayerich
ENVI and colormap...
|
355
|
outfile<<"bil";
|
e8eb202f
David Mayerich
added a new ENVI ...
|
356
|
if(interleave == BSQ)
|
6e257ab3
dmayerich
ENVI and colormap...
|
357
358
359
360
361
362
363
364
365
366
367
368
369
|
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;
|
7006df5f
David Mayerich
reformat of direc...
|
370
|
for(unsigned int i=0; i<band_names.size(); i++)
|
6e257ab3
dmayerich
ENVI and colormap...
|
371
372
373
374
375
376
377
|
{
outfile<<band_names[i];
if(i < band_names.size() - 1)
outfile<<", ";
}
outfile<<"}"<<std::endl;
}
|
ee4dea28
David Mayerich
fixed errors in c...
|
378
379
380
381
382
383
384
|
//if there is a list of wavelengths, output it
if(wavelength.size()){
outfile<<"wavelength = {"<<std::endl;
for(unsigned int i=0; i<wavelength.size()-1; i++)
outfile<<wavelength[i]<<", ";
outfile<<wavelength.back()<<"}"<<std::endl;
}
|
6e257ab3
dmayerich
ENVI and colormap...
|
385
386
387
388
389
|
outfile.close();
}
void save()
|
e8eb202f
David Mayerich
added a new ENVI ...
|
390
|
{
|
6e257ab3
dmayerich
ENVI and colormap...
|
391
392
393
394
395
396
397
398
399
|
//std::cout<<"ENVI Header Name: "<<name<<std::endl;
save(name);
}
//returns the size of a single value (in bytes)
unsigned int valsize()
{
switch(data_type)
{
|
e8eb202f
David Mayerich
added a new ENVI ...
|
400
|
case int8: //1 = 8-bit byte
|
6e257ab3
dmayerich
ENVI and colormap...
|
401
|
return 1;
|
e8eb202f
David Mayerich
added a new ENVI ...
|
402
403
|
case int16: //16-bit signed integer
case uint16: //16-bit unsigned integer
|
6e257ab3
dmayerich
ENVI and colormap...
|
404
|
return 2;
|
e8eb202f
David Mayerich
added a new ENVI ...
|
405
406
407
|
case int32: //32-bit signed long integer
case float32: //32-bit floating point
case uint32: //32-bit unsigned long integer
|
6e257ab3
dmayerich
ENVI and colormap...
|
408
|
return 4;
|
e8eb202f
David Mayerich
added a new ENVI ...
|
409
410
411
412
|
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
|
6e257ab3
dmayerich
ENVI and colormap...
|
413
|
return 8;
|
e8eb202f
David Mayerich
added a new ENVI ...
|
414
|
case complex64: //64-bit complex value
|
6e257ab3
dmayerich
ENVI and colormap...
|
415
416
417
418
419
420
|
return 16;
default:
return 0;
}
}
|
a2bf1d08
David Mayerich
general bug fixes...
|
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
|
/// Convert an interleave type to a string
static std::string interleave_str(interleaveType t){
switch(t){
case stim::envi_header::BSQ:
return std::string("BSQ");
case stim::envi_header::BIL:
return std::string("BIL");
case stim::envi_header::BIP:
return std::string("BIP");
default:
std::cout<<"ERROR in stim::envi_header::typestr() - unrecognized type"<<std::endl;
exit(1);
}
}
/// Convert the current interleave type to a string
std::string interleave_str(){
return interleave_str(interleave);
}
|
dc8eb8aa
David Mayerich
added band trimmi...
|
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
|
/// Convert a wavelength to a band index (or a pair of surrounding band indices)
std::vector<size_t> band_index(double w){
std::vector<size_t> idx; //create an empty array of indices
if(w < wavelength[0] || w > wavelength[bands-1]) return idx; //if the wavelength range is outside of the file, return an empty array
for(size_t b = 0; b < bands; b++){ //for each band in the wavelength vector
if(wavelength[b] == w){ //if an exact match is found
idx.push_back(b); //add the band to the array and return it
return idx;
}
if(wavelength[b] >= w){ //if the current wavelength exceeds w
idx.resize(2);
idx[0] = b-1; //push both the previous and current band index
idx[1] = b;
return idx; //return the pair of indices
}
}
return idx;
}
/// Convert a wavelength range to a list of bands
std::vector<size_t> band_indices(double w0, double w1){
size_t idx0 = 0;
size_t idx1 = 0; //declare the interval indices for the band range
//get the indices for the first wavelength
std::vector<size_t> r;
if(w0 > wavelength[bands-1] || w1 < wavelength[0]){
std::cout<<"ERROR in envi_header::band_indices - wavelengths are completely out of range: ["<<w0<<", "<<w1<<"]"<<std::endl;
exit(1);
}
if(w0 < wavelength[0]) //if the lower wavelength is outside the file range
idx0 = 0; //just set it to zero
else{
r = band_index(w0);
if(r.size() == 1) //if there is an exact match, set the first interval index
idx0 = r[0];
else
idx0 = r[1]; //otherwise save the highest band index (this is the first band with wavelength > w0)
}
//get the indices for the second wavelength
if(w1 > wavelength[bands-1])
idx1 = bands-1;
else{
r = band_index(w1);
if(r.size() == 0)
idx1 = bands - 1;
else
idx1 = r[0]; //take the lowest band index (this is the last band < w1)
}
size_t n = idx1 - idx0 + 1; //calculate the number of bands in this interval
r.resize(n); //resize the band index array
for(size_t b = 0; b < n; b++){ //for each band in the interval
r[b] = idx0 + b; // insert the band in the array
}
return r; //return the index array
}
|
e8eb202f
David Mayerich
added a new ENVI ...
|
503
504
505
|
}; //end EnviHeader
}
#endif
|