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
9
10
|
#include <sstream>
#include <vector>
#include <stdlib.h>
|
6e257ab3
dmayerich
ENVI and colormap...
|
11
|
//information from an ENVI header file
|
e8eb202f
David Mayerich
added a new ENVI ...
|
12
13
|
//A good resource can be found here: http://www.exelisvis.com/docs/enviheaderfiles.html
|
8a86bd56
David Mayerich
changed rts names...
|
14
|
namespace stim{
|
e8eb202f
David Mayerich
added a new ENVI ...
|
15
16
|
struct envi_header
|
6e257ab3
dmayerich
ENVI and colormap...
|
17
|
{
|
e8eb202f
David Mayerich
added a new ENVI ...
|
18
19
20
21
|
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...
|
22
23
24
25
26
27
28
29
|
std::string name;
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)
|
e8eb202f
David Mayerich
added a new ENVI ...
|
30
31
32
|
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...
|
33
34
|
|
e8eb202f
David Mayerich
added a new ENVI ...
|
35
|
envi_header::interleaveType interleave;
|
6e257ab3
dmayerich
ENVI and colormap...
|
36
37
38
|
std::string sensor_type; //not really used
|
e8eb202f
David Mayerich
added a new ENVI ...
|
39
|
envi_header::endianType byte_order; //little = least significant bit first (most systems)
|
6e257ab3
dmayerich
ENVI and colormap...
|
40
41
42
43
44
45
46
47
48
49
|
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 ...
|
50
51
|
void init(){
name = "";
|
6e257ab3
dmayerich
ENVI and colormap...
|
52
53
54
55
56
57
|
//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 ...
|
58
59
60
|
data_type = float32;
interleave = BSQ;
byte_order = endianLittle;
|
6e257ab3
dmayerich
ENVI and colormap...
|
61
62
63
64
65
66
67
68
69
70
|
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 ...
|
71
72
73
74
75
76
77
78
79
|
envi_header(){
init();
}
envi_header(std::string name){
init();
load(name);
}
std::string trim(std::string line){
|
6e257ab3
dmayerich
ENVI and colormap...
|
80
|
//trims whitespace from the beginning and end of line
|
7006df5f
David Mayerich
reformat of direc...
|
81
|
unsigned int start_i, end_i;
|
6e257ab3
dmayerich
ENVI and colormap...
|
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
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);
}
|
e8eb202f
David Mayerich
added a new ENVI ...
|
98
|
std::string get_token(std::string line){
|
6e257ab3
dmayerich
ENVI and colormap...
|
99
100
101
102
103
104
105
106
107
108
|
//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 ...
|
109
|
std::string get_data_str(std::string line){
|
6e257ab3
dmayerich
ENVI and colormap...
|
110
111
112
113
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
|
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 ...
|
194
|
std::vector<double> result;
|
6e257ab3
dmayerich
ENVI and colormap...
|
195
196
197
198
199
|
std::string entry;
size_t i;
do
{
i = sequence.find_first_of(',');
|
e8eb202f
David Mayerich
added a new ENVI ...
|
200
|
entry = sequence.substr(0, i);
|
6e257ab3
dmayerich
ENVI and colormap...
|
201
|
sequence = sequence.substr(i+1);
|
e8eb202f
David Mayerich
added a new ENVI ...
|
202
|
result.push_back(atof(entry.c_str()));
|
0174d823
dmayerich
bug fixes and exits
|
203
|
//std::cout<<entry<<" ";
|
6e257ab3
dmayerich
ENVI and colormap...
|
204
205
206
207
208
209
210
211
212
213
214
|
}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 ...
|
215
|
{
|
6708cc25
heziqi
Ziqi added envi c...
|
216
|
std::cout<<"ERROR: unable to open header file: "<<filename<<std::endl;
|
6e257ab3
dmayerich
ENVI and colormap...
|
217
218
219
220
221
222
223
224
225
226
|
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 ...
|
227
|
}
|
6e257ab3
dmayerich
ENVI and colormap...
|
228
229
|
//for each line in the file, get the token
|
e8eb202f
David Mayerich
added a new ENVI ...
|
230
231
|
std::string token;
|
0174d823
dmayerich
bug fixes and exits
|
232
233
|
//get a line
getline(file, line);
|
6e257ab3
dmayerich
ENVI and colormap...
|
234
235
236
|
while(file)
{
|
0174d823
dmayerich
bug fixes and exits
|
237
|
|
6e257ab3
dmayerich
ENVI and colormap...
|
238
239
240
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
267
268
269
270
271
272
273
274
275
276
277
278
279
|
//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 ...
|
280
|
data_type = (dataType)atoi(get_data_str(line).c_str());
|
6e257ab3
dmayerich
ENVI and colormap...
|
281
282
283
284
|
else if(token == "interleave")
{
std::string interleave_str = get_data_str(line);
if(interleave_str == "bip")
|
e8eb202f
David Mayerich
added a new ENVI ...
|
285
|
interleave = BIP;
|
6e257ab3
dmayerich
ENVI and colormap...
|
286
|
else if(interleave_str == "bil")
|
e8eb202f
David Mayerich
added a new ENVI ...
|
287
|
interleave = BIL;
|
6e257ab3
dmayerich
ENVI and colormap...
|
288
|
else if(interleave_str == "bsq")
|
e8eb202f
David Mayerich
added a new ENVI ...
|
289
|
interleave = BSQ;
|
6e257ab3
dmayerich
ENVI and colormap...
|
290
291
292
293
|
}
else if(token == "sensor type")
sensor_type = get_data_str(line);
else if(token == "byte order")
|
e8eb202f
David Mayerich
added a new ENVI ...
|
294
|
byte_order = (endianType)atoi(get_data_str(line).c_str());
|
6e257ab3
dmayerich
ENVI and colormap...
|
295
296
297
298
299
|
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 ...
|
300
301
302
|
wavelength_units = get_data_str(line);
//get the next line
|
0174d823
dmayerich
bug fixes and exits
|
303
|
getline(file, line);
|
e8eb202f
David Mayerich
added a new ENVI ...
|
304
305
306
307
|
}
//make sure the number of bands matches the number of wavelengths
unsigned int wavelengths = wavelength.size();
|
ee4dea28
David Mayerich
fixed errors in c...
|
308
|
if(wavelengths && bands != wavelengths)
|
e8eb202f
David Mayerich
added a new ENVI ...
|
309
310
311
|
{
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
|
312
|
}
|
6e257ab3
dmayerich
ENVI and colormap...
|
313
314
|
//close the file
|
e8eb202f
David Mayerich
added a new ENVI ...
|
315
316
317
318
319
|
file.close();
//set the file name
name = filename;
|
6e257ab3
dmayerich
ENVI and colormap...
|
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
|
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 ...
|
342
|
if(interleave == BIP)
|
6e257ab3
dmayerich
ENVI and colormap...
|
343
|
outfile<<"bip";
|
e8eb202f
David Mayerich
added a new ENVI ...
|
344
|
if(interleave == BIL)
|
6e257ab3
dmayerich
ENVI and colormap...
|
345
|
outfile<<"bil";
|
e8eb202f
David Mayerich
added a new ENVI ...
|
346
|
if(interleave == BSQ)
|
6e257ab3
dmayerich
ENVI and colormap...
|
347
348
349
350
351
352
353
354
355
356
357
358
359
|
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...
|
360
|
for(unsigned int i=0; i<band_names.size(); i++)
|
6e257ab3
dmayerich
ENVI and colormap...
|
361
362
363
364
365
366
367
|
{
outfile<<band_names[i];
if(i < band_names.size() - 1)
outfile<<", ";
}
outfile<<"}"<<std::endl;
}
|
ee4dea28
David Mayerich
fixed errors in c...
|
368
369
370
371
372
373
374
|
//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...
|
375
376
377
378
379
|
outfile.close();
}
void save()
|
e8eb202f
David Mayerich
added a new ENVI ...
|
380
|
{
|
6e257ab3
dmayerich
ENVI and colormap...
|
381
382
383
384
385
386
387
388
389
|
//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 ...
|
390
|
case int8: //1 = 8-bit byte
|
6e257ab3
dmayerich
ENVI and colormap...
|
391
|
return 1;
|
e8eb202f
David Mayerich
added a new ENVI ...
|
392
393
|
case int16: //16-bit signed integer
case uint16: //16-bit unsigned integer
|
6e257ab3
dmayerich
ENVI and colormap...
|
394
|
return 2;
|
e8eb202f
David Mayerich
added a new ENVI ...
|
395
396
397
|
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...
|
398
|
return 4;
|
e8eb202f
David Mayerich
added a new ENVI ...
|
399
400
401
402
|
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...
|
403
|
return 8;
|
e8eb202f
David Mayerich
added a new ENVI ...
|
404
|
case complex64: //64-bit complex value
|
6e257ab3
dmayerich
ENVI and colormap...
|
405
406
407
408
409
410
|
return 16;
default:
return 0;
}
}
|
e8eb202f
David Mayerich
added a new ENVI ...
|
411
412
413
|
}; //end EnviHeader
}
#endif
|