Blame view

envi/bip.h 23.7 KB
e8eb202f   David Mayerich   added a new ENVI ...
1
2
3
4
  #ifndef STIM_BIP_H
  #define STIM_BIP_H
  
  #include "../envi/envi_header.h"
6aa04ba2   David Mayerich   interleave types
5
  #include "../envi/bil.h"
88c3e636   heziqi   Ziqi added big.h
6
7
8
9
  #include "../envi/binary.h"
  #include <cstring>
  #include <utility>
  
8a86bd56   David Mayerich   changed rts names...
10
  namespace stim{
88c3e636   heziqi   Ziqi added big.h
11
12
13
14
15
16
17
  
  template <typename T>
  
  class bip: public binary<T> {
  
  protected:
  	
6708cc25   heziqi   Ziqi added envi c...
18
19
20
  	
  	std::vector<double> w; //band wavelength
  	unsigned int offset;		//header offset
88c3e636   heziqi   Ziqi added big.h
21
22
23
24
25
  
  public:
  
  	using binary<T>::open;
  	using binary<T>::file;
6708cc25   heziqi   Ziqi added envi c...
26
  	using binary<T>::R;
88c3e636   heziqi   Ziqi added big.h
27
28
  
  	//open a file, given the file and its header's names
6708cc25   heziqi   Ziqi added envi c...
29
  	bool open(std::string filename, unsigned int X, unsigned int Y, unsigned int B, unsigned int header_offset, std::vector<double> wavelengths){
88c3e636   heziqi   Ziqi added big.h
30
  
6708cc25   heziqi   Ziqi added envi c...
31
32
33
34
  		//copy the wavelengths to the BSQ file structure
  		w = wavelengths;
  		//copy the offset to the structure
  		offset = header_offset;
88c3e636   heziqi   Ziqi added big.h
35
  
6708cc25   heziqi   Ziqi added envi c...
36
  		return open(filename, vec<unsigned int>(X, Y, B), header_offset);
88c3e636   heziqi   Ziqi added big.h
37
38
39
40
41
42
  		
  	}
  
  	//save one band of the file into the memory, and return the pointer
  	bool band_index( T * p, unsigned int page){
  
6708cc25   heziqi   Ziqi added envi c...
43
  		if (page >= R[2]){										//make sure the bank number is right
88c3e636   heziqi   Ziqi added big.h
44
45
46
47
  			std::cout<<"ERROR: page out of range"<<std::endl;
  			return false;
  		}
  
11f177d5   heziqi   Ziqi added functi...
48
49
  		//binary::getSlice(p, 0, page);         //I met some problems when I called this function?????
  		file.seekg(page * sizeof(T), std::ios::beg);
6708cc25   heziqi   Ziqi added envi c...
50
  		for (unsigned i = 0; i < R[0] * R[1]; i++)
11f177d5   heziqi   Ziqi added functi...
51
52
  		{
  			file.read((char *)(p + i), sizeof(T));
6708cc25   heziqi   Ziqi added envi c...
53
  			file.seekg( (R[2] - 1) * sizeof(T), std::ios::cur);
11f177d5   heziqi   Ziqi added functi...
54
55
  		}
  
88c3e636   heziqi   Ziqi added big.h
56
57
58
  		return true;
  	}
  
e933c14e   David Mayerich   cleaned up the code
59
  	bool band( T * p, double wavelength){
88c3e636   heziqi   Ziqi added big.h
60
  
ee4dea28   David Mayerich   fixed errors in c...
61
62
63
64
  		//if there are no wavelengths in the BSQ file
  		if(w.size() == 0)
  			return band_index(p, (unsigned int)wavelength);
  
6708cc25   heziqi   Ziqi added envi c...
65
  		unsigned int XY = R[0] * R[1];	//calculate the number of pixels in a band
88c3e636   heziqi   Ziqi added big.h
66
67
  
  		unsigned page=0;                      //bands around the wavelength
517876d6   heziqi   metrics finished ...
68
  
88c3e636   heziqi   Ziqi added big.h
69
70
71
72
  
  		//get the bands numbers around the wavelength
  
  		//if wavelength is smaller than the first one in header file
6708cc25   heziqi   Ziqi added envi c...
73
  		if ( w[page] > wavelength ){
88c3e636   heziqi   Ziqi added big.h
74
75
76
77
  			band_index(p, page);
  			return true;
  		}
  
6708cc25   heziqi   Ziqi added envi c...
78
  		while( w[page] < wavelength )
88c3e636   heziqi   Ziqi added big.h
79
80
81
  		{
  			page++;
  			//if wavelength is larger than the last wavelength in header file
6708cc25   heziqi   Ziqi added envi c...
82
83
  			if (page == R[2]) {
  				band_index(p, R[2]-1);
88c3e636   heziqi   Ziqi added big.h
84
85
86
  				return true;
  			}
  		}
6708cc25   heziqi   Ziqi added envi c...
87
  		if ( wavelength < w[page] )
88c3e636   heziqi   Ziqi added big.h
88
  		{
517876d6   heziqi   metrics finished ...
89
90
  			T * p1;
  			T * p2;
88c3e636   heziqi   Ziqi added big.h
91
92
93
94
95
  			p1=(T*)malloc( XY * sizeof(T));                     //memory allocation
  			p2=(T*)malloc( XY * sizeof(T));
  			band_index(p1, page - 1);
  			band_index(p2, page );
  			for(unsigned i=0; i < XY; i++){
6708cc25   heziqi   Ziqi added envi c...
96
  				double r = (double) (wavelength - w[page-1]) / (double) (w[page] - w[page-1]);
88c3e636   heziqi   Ziqi added big.h
97
98
  				p[i] = (p2[i] - p1[i]) * r + p1[i];
  			}
517876d6   heziqi   metrics finished ...
99
100
  			free(p1);
  			free(p2);
88c3e636   heziqi   Ziqi added big.h
101
102
103
104
105
106
  		}
  		else                           //if the wavelength is equal to a wavelength in header file
  		{
  			band_index(p, page);
  		}
  
88c3e636   heziqi   Ziqi added big.h
107
108
  		return true;
  	}
11f177d5   heziqi   Ziqi added functi...
109
110
111
  	//get YZ line from the a Y slice, Y slice data should be already IN the MEMORY
  	bool getYZ(T* p, T* c, double wavelength)
  	{
6708cc25   heziqi   Ziqi added envi c...
112
113
  		unsigned int X = R[0];	//calculate the number of pixels in a sample
  		unsigned int B = R[2];
11f177d5   heziqi   Ziqi added functi...
114
115
  
  		unsigned page=0;                      //samples around the wavelength
517876d6   heziqi   metrics finished ...
116
  
11f177d5   heziqi   Ziqi added functi...
117
118
119
120
  
  		//get the bands numbers around the wavelength
  
  		//if wavelength is smaller than the first one in header file
6708cc25   heziqi   Ziqi added envi c...
121
122
  		if ( w[page] > wavelength ){
  			for(unsigned j = 0; j < R[0]; j++)
11f177d5   heziqi   Ziqi added functi...
123
124
125
126
127
128
  			{
  				p[j] = c[j * B];
  			}		
  			return true;
  		}
  
6708cc25   heziqi   Ziqi added envi c...
129
  		while( w[page] < wavelength )
11f177d5   heziqi   Ziqi added functi...
130
131
132
133
  		{
  			page++;
  			//if wavelength is larger than the last wavelength in header file
  			if (page == B) {
6708cc25   heziqi   Ziqi added envi c...
134
  				for(unsigned j = 0; j < R[0]; j++)
11f177d5   heziqi   Ziqi added functi...
135
136
137
138
139
140
  				{
  					p[j] = c[(j + 1) * B - 1];
  				}
  				return true;
  			}
  		}
6708cc25   heziqi   Ziqi added envi c...
141
  		if ( wavelength < w[page] )
11f177d5   heziqi   Ziqi added functi...
142
  		{
517876d6   heziqi   metrics finished ...
143
144
  			T * p1;
  			T * p2;
11f177d5   heziqi   Ziqi added functi...
145
146
147
148
149
150
151
152
153
154
155
156
157
158
  			p1=(T*)malloc( X * sizeof(T));                     //memory allocation
  			p2=(T*)malloc( X * sizeof(T));
  			//band_index(p1, page - 1);
  			for(unsigned j = 0; j < X; j++)
  			{
  				p1[j] = c[j * B + page - 1];
  			}
  			//band_index(p2, page );
  			for(unsigned j = 0; j < X; j++)
  			{
  				p2[j] = c[j * B + page];
  			}
  			
  			for(unsigned i=0; i < X; i++){
6708cc25   heziqi   Ziqi added envi c...
159
  				double r = (double) (wavelength - w[page-1]) / (double) (w[page] - w[page-1]);
11f177d5   heziqi   Ziqi added functi...
160
161
  				p[i] = (p2[i] - p1[i]) * r + p1[i];
  			}
517876d6   heziqi   metrics finished ...
162
163
  			free(p1);
  			free(p2);
11f177d5   heziqi   Ziqi added functi...
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
  		}
  		else                           //if the wavelength is equal to a wavelength in header file
  		{
  			//band_index(p, page);
  			for(unsigned j = 0; j < X; j++)
  			{
  				p[j] = c[j * B + page];
  			}
  		}
  
  		return true;		
  	}
  	//given a y and a wavelength, return the y-band data
  	//I do not use it right now, to accelerate the processing speed, I try to read a YZ whole slice into memory first, and then we can call "getYZ"
  	bool get_x_wavelength(T * p, unsigned y, double wavelength)
  	{
6708cc25   heziqi   Ziqi added envi c...
180
  		unsigned int X = R[0];	//calculate the number of pixels in a sample
11f177d5   heziqi   Ziqi added functi...
181
182
183
184
185
186
187
188
  
  		unsigned page=0;                      //samples around the wavelength
  		T * p1;
  		T * p2;
  
  		//get the bands numbers around the wavelength
  
  		//if wavelength is smaller than the first one in header file
6708cc25   heziqi   Ziqi added envi c...
189
190
191
  		if ( w[page] > wavelength ){
  			file.seekg( y * R[2] * R[0] * sizeof(T), std::ios::beg);
  			for(unsigned j = 0; j < R[0]; j++)
11f177d5   heziqi   Ziqi added functi...
192
193
  			{
  				file.read((char *)(p + j), sizeof(T));
6708cc25   heziqi   Ziqi added envi c...
194
  				file.seekg((R[2] - 1) * sizeof(T), std::ios::cur);
11f177d5   heziqi   Ziqi added functi...
195
196
197
198
  			}		
  			return true;
  		}
  
6708cc25   heziqi   Ziqi added envi c...
199
  		while( w[page] < wavelength )
11f177d5   heziqi   Ziqi added functi...
200
201
202
  		{
  			page++;
  			//if wavelength is larger than the last wavelength in header file
6708cc25   heziqi   Ziqi added envi c...
203
204
205
  			if (page == R[2]) {
  				file.seekg( (y * R[2] * R[0] + R[2] - 1)* sizeof(T), std::ios::beg);
  				for(unsigned j = 0; j < R[0]; j++)
11f177d5   heziqi   Ziqi added functi...
206
207
  				{
  					file.read((char *)(p + j), sizeof(T));
6708cc25   heziqi   Ziqi added envi c...
208
  					file.seekg((R[2] - 1) * sizeof(T), std::ios::cur);
11f177d5   heziqi   Ziqi added functi...
209
210
211
212
  				}
  				return true;
  			}
  		}
6708cc25   heziqi   Ziqi added envi c...
213
  		if ( wavelength < w[page] )
11f177d5   heziqi   Ziqi added functi...
214
215
216
217
  		{
  			p1=(T*)malloc( X * sizeof(T));                     //memory allocation
  			p2=(T*)malloc( X * sizeof(T));
  			//band_index(p1, page - 1);
6708cc25   heziqi   Ziqi added envi c...
218
219
  			file.seekg( (y * R[2] * R[0] + page - 1)* sizeof(T), std::ios::beg);
  			for(unsigned j = 0; j < R[0]; j++)
11f177d5   heziqi   Ziqi added functi...
220
221
  			{
  				file.read((char *)(p1 + j), sizeof(T));
6708cc25   heziqi   Ziqi added envi c...
222
  				file.seekg((R[2] - 1) * sizeof(T), std::ios::cur);
11f177d5   heziqi   Ziqi added functi...
223
224
  			}
  			//band_index(p2, page );
6708cc25   heziqi   Ziqi added envi c...
225
226
  			file.seekg( (y * R[2] * R[0] + page)* sizeof(T), std::ios::beg);
  			for(unsigned j = 0; j < R[0]; j++)
11f177d5   heziqi   Ziqi added functi...
227
228
  			{
  				file.read((char *)(p2 + j), sizeof(T));
6708cc25   heziqi   Ziqi added envi c...
229
  				file.seekg((R[2] - 1) * sizeof(T), std::ios::cur);
11f177d5   heziqi   Ziqi added functi...
230
231
  			}
  			
6708cc25   heziqi   Ziqi added envi c...
232
233
  			for(unsigned i=0; i < R[0]; i++){
  				double r = (double) (wavelength - w[page-1]) / (double) (w[page] - w[page-1]);
11f177d5   heziqi   Ziqi added functi...
234
235
236
237
238
239
  				p[i] = (p2[i] - p1[i]) * r + p1[i];
  			}
  		}
  		else                           //if the wavelength is equal to a wavelength in header file
  		{
  			//band_index(p, page);
6708cc25   heziqi   Ziqi added envi c...
240
241
  			file.seekg( (y * R[2] * R[0] + page) * sizeof(T), std::ios::beg);
  			for(unsigned j = 0; j < R[0]; j++)
11f177d5   heziqi   Ziqi added functi...
242
243
  			{
  				file.read((char *)(p + j), sizeof(T));
6708cc25   heziqi   Ziqi added envi c...
244
  				file.seekg((R[2] - 1) * sizeof(T), std::ios::cur);
11f177d5   heziqi   Ziqi added functi...
245
246
  			}
  		}
11f177d5   heziqi   Ziqi added functi...
247
248
  		return true;
  	}
88c3e636   heziqi   Ziqi added big.h
249
250
  
  	//save one pixel of the BIP file into the memory, and return the pointer
e933c14e   David Mayerich   cleaned up the code
251
  	bool spectrum(T * p, unsigned x, unsigned y){
88c3e636   heziqi   Ziqi added big.h
252
  
6708cc25   heziqi   Ziqi added envi c...
253
  		if ( x >= R[0] || y >= R[1]){							//make sure the sample and line number is right
88c3e636   heziqi   Ziqi added big.h
254
  			std::cout<<"ERROR: sample or line out of range"<<std::endl;
11f177d5   heziqi   Ziqi added functi...
255
  			exit(1);
88c3e636   heziqi   Ziqi added big.h
256
257
  		}
  
6708cc25   heziqi   Ziqi added envi c...
258
  		file.seekg((x + y * R[0]) * R[2] * sizeof(T), std::ios::beg);           //point to the certain sample and line
88c3e636   heziqi   Ziqi added big.h
259
  		
6708cc25   heziqi   Ziqi added envi c...
260
  		file.read((char *)p, sizeof(T) * R[2]);
88c3e636   heziqi   Ziqi added big.h
261
262
263
  			
  		return true;	
  	}
bfe9c6db   heziqi   added feature_mat...
264
265
266
267
268
269
270
271
272
273
274
275
276
277
  
  	//save one pixel into memory
  	bool pixel(T * p, unsigned n){
  
  		unsigned bandnum = R[0] * R[1];		//calculate numbers in one band
  		if ( n >= bandnum){							//make sure the pixel number is right
  			std::cout<<"ERROR: sample or line out of range"<<std::endl;
  			return false;
  		}
  
  		file.seekg(n * R[2] * sizeof(T), std::ios::beg);           //point to the certain pixel
  		file.read((char *)p, sizeof(T) * R[2]);
  		return true;	
  	}
11f177d5   heziqi   Ziqi added functi...
278
279
280
  	
  	//given a Y ,return a ZX slice
  	bool getY(T * p, unsigned y)
88c3e636   heziqi   Ziqi added big.h
281
  	{
6708cc25   heziqi   Ziqi added envi c...
282
  		if ( y >= R[1]){							//make sure the line number is right
11f177d5   heziqi   Ziqi added functi...
283
284
285
286
  			std::cout<<"ERROR: line out of range"<<std::endl;
  			exit(1);
  		}	
  		
6708cc25   heziqi   Ziqi added envi c...
287
288
  		file.seekg(y * R[2] * R[0] * sizeof(T), std::ios::beg);
  		file.read((char *)p, sizeof(T) * R[2] * R[0]);
11f177d5   heziqi   Ziqi added functi...
289
290
291
292
293
294
  		
  		return true;
  		
  	}
  		
  	//(BIP) baseline correction
11f177d5   heziqi   Ziqi added functi...
295
296
  	bool baseline(std::string outname, std::vector<double> wls){
  	
88c3e636   heziqi   Ziqi added big.h
297
298
299
300
301
302
  		unsigned N = wls.size();			//get the number of baseline points
  		
  		std::ofstream target(outname.c_str(), std::ios::binary);	//open the target binary file
  		std::string headername = outname + ".hdr";              //the header file name		
  		
  		//simplify image resolution
6708cc25   heziqi   Ziqi added envi c...
303
  		unsigned int ZX = R[2] * R[0];		//calculate the number of points in a Y slice
11f177d5   heziqi   Ziqi added functi...
304
  		unsigned int L = ZX * sizeof(T);			//calculate the number of bytes of a Y slice
6708cc25   heziqi   Ziqi added envi c...
305
306
  		unsigned int B = R[2];
  		unsigned int X = R[0];
88c3e636   heziqi   Ziqi added big.h
307
  		
11f177d5   heziqi   Ziqi added functi...
308
309
  		T* c;			//pointer to the current Y slice
  		c = (T*)malloc(L);  //memory allocation
88c3e636   heziqi   Ziqi added big.h
310
  		
11f177d5   heziqi   Ziqi added functi...
311
312
  		T* a;			//pointer to the two YZ lines surrounding the current YZ line
  		T* b;
88c3e636   heziqi   Ziqi added big.h
313
  		
11f177d5   heziqi   Ziqi added functi...
314
315
  		a = (T*)malloc(X * sizeof(T));
  		b = (T*)malloc(X * sizeof(T));
88c3e636   heziqi   Ziqi added big.h
316
  
88c3e636   heziqi   Ziqi added big.h
317
318
319
  
  		double ai, bi;	//stores the two baseline points wavelength surrounding the current band
  		double ci;		//stores the current band's wavelength
f6169dea   heziqi   Ziqi completed bi...
320
  		unsigned control;
88c3e636   heziqi   Ziqi added big.h
321
  
88c3e636   heziqi   Ziqi added big.h
322
323
324
325
  		if (a == NULL || b == NULL || c == NULL){
  			std::cout<<"ERROR: error allocating memory";
  			exit(1);
  		}
11f177d5   heziqi   Ziqi added functi...
326
327
  	//	loop start	correct every y slice
  		
6708cc25   heziqi   Ziqi added envi c...
328
  		for (unsigned k =0; k < R[1]; k++)
11f177d5   heziqi   Ziqi added functi...
329
330
331
332
333
  		{
  			//get the current y slice
  			getY(c, k);
  		
  			//initialize lownum, highnum, low, high		
f6169dea   heziqi   Ziqi completed bi...
334
  			control=0;
6708cc25   heziqi   Ziqi added envi c...
335
  			ai = w[0];
11f177d5   heziqi   Ziqi added functi...
336
  			//if no baseline point is specified at band 0,
88c3e636   heziqi   Ziqi added big.h
337
  			//set the baseline point at band 0 to 0
6708cc25   heziqi   Ziqi added envi c...
338
  			if(wls[0] != w[0]){
11f177d5   heziqi   Ziqi added functi...
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
  				bi = wls[control];			
  				memset(a, (char)0, X * sizeof(T) );
  			}
  			//else get the low band
  			else{
  				control++;
  				getYZ(a, c, ai);
  				bi = wls[control];
  			}
  			//get the high band
  			getYZ(b, c, bi);
  		
  			//correct every YZ line
  			
  			for(unsigned cii = 0; cii < B; cii++){
11f177d5   heziqi   Ziqi added functi...
354
  				//update baseline points, if necessary
6708cc25   heziqi   Ziqi added envi c...
355
  				if( w[cii] >= bi && cii != B - 1) {
11f177d5   heziqi   Ziqi added functi...
356
357
358
359
360
361
362
363
364
365
366
367
368
  					//if the high band is now on the last BL point?
  					if (control != N-1) {
  	
  						control++;		//increment the index
  	
  						std::swap(a, b);	//swap the baseline band pointers
  	
  						ai = bi;
  						bi = wls[control];
  						getYZ(b, c, bi);
  	
  					}
  					//if the last BL point on the last band of the file?
6708cc25   heziqi   Ziqi added envi c...
369
  					else if ( wls[control] < w[B - 1]) {
11f177d5   heziqi   Ziqi added functi...
370
371
372
373
374
375
  	
  						std::swap(a, b);	//swap the baseline band pointers
  	
  						memset(b, (char)0, X * sizeof(T) );	//clear the high band
  	
  						ai = bi;
6708cc25   heziqi   Ziqi added envi c...
376
  						bi = w[B - 1];
11f177d5   heziqi   Ziqi added functi...
377
  					}
88c3e636   heziqi   Ziqi added big.h
378
  				}
88c3e636   heziqi   Ziqi added big.h
379
  
6708cc25   heziqi   Ziqi added envi c...
380
  				ci = w[cii];
11f177d5   heziqi   Ziqi added functi...
381
382
383
384
385
  			
  				//perform the baseline correction
  				for(unsigned i=0; i < X; i++)
  				{
  					double r = (double) (ci - ai) / (double) (bi - ai);
92e4cf05   heziqi   include file fixed
386
  					c[i * B + cii] =(T) ( c[i * B + cii] - (b[i] - a[i]) * r - a[i] );
11f177d5   heziqi   Ziqi added functi...
387
388
389
  				}
  				
  			}//loop for YZ line end  
11f177d5   heziqi   Ziqi added functi...
390
391
  			target.write(reinterpret_cast<const char*>(c), L);   //write the corrected data into destination	
  		}//loop for Y slice end
88c3e636   heziqi   Ziqi added big.h
392
  		
6708cc25   heziqi   Ziqi added envi c...
393
  
88c3e636   heziqi   Ziqi added big.h
394
395
396
397
398
  		
  		free(a);
  		free(b);
  		free(c);
  		target.close();
11f177d5   heziqi   Ziqi added functi...
399
400
  		return true;	
  		
88c3e636   heziqi   Ziqi added big.h
401
  	}
11f177d5   heziqi   Ziqi added functi...
402
403
  		
  	// normalize the BIP file
93503ec6   David Mayerich   allowed masking d...
404
  	bool normalize(std::string outname, double w, double t = 0.0)
88c3e636   heziqi   Ziqi added big.h
405
  	{
6708cc25   heziqi   Ziqi added envi c...
406
407
408
409
410
  		unsigned int B = R[2];		//calculate the number of bands
  		unsigned int Y = R[1];
  		unsigned int X = R[0];
  		unsigned int ZX = R[2] * R[0];
  		unsigned int XY = R[0] * R[1];	//calculate the number of pixels in a band
11f177d5   heziqi   Ziqi added functi...
411
412
  		unsigned int S = XY * sizeof(T);		//calculate the number of bytes in a band		
  		unsigned int L = ZX * sizeof(T);
88c3e636   heziqi   Ziqi added big.h
413
414
415
416
  
  		std::ofstream target(outname.c_str(), std::ios::binary);	//open the target binary file
  		std::string headername = outname + ".hdr";              //the header file name
  
11f177d5   heziqi   Ziqi added functi...
417
418
  		T * c;				//pointer to the current ZX slice
  		T * b;				//pointer to the standard band
88c3e636   heziqi   Ziqi added big.h
419
420
  
  		b = (T*)malloc( S );     //memory allocation
11f177d5   heziqi   Ziqi added functi...
421
  		c = (T*)malloc( L ); 
88c3e636   heziqi   Ziqi added big.h
422
  
e933c14e   David Mayerich   cleaned up the code
423
  		band(b, w);             //get the certain band into memory
88c3e636   heziqi   Ziqi added big.h
424
  
11f177d5   heziqi   Ziqi added functi...
425
  		for(unsigned j = 0; j < Y; j++)
88c3e636   heziqi   Ziqi added big.h
426
  		{
11f177d5   heziqi   Ziqi added functi...
427
  			getY(c, j);
f6169dea   heziqi   Ziqi completed bi...
428
  			unsigned jX = j * X;		//to avoid calculating it many times
11f177d5   heziqi   Ziqi added functi...
429
  			for(unsigned i = 0; i < X; i++)
88c3e636   heziqi   Ziqi added big.h
430
  			{
f6169dea   heziqi   Ziqi completed bi...
431
  				unsigned iB = i * B;		
11f177d5   heziqi   Ziqi added functi...
432
433
  				for(unsigned m = 0; m < B; m++)
  				{
93503ec6   David Mayerich   allowed masking d...
434
435
436
437
  					if( b[i+jX] < t )
  						c[m + iB] = (T)0.0;
  					else
  						c[m + iB] = c[m + iB] / b[i + jX];			//perform normalization
11f177d5   heziqi   Ziqi added functi...
438
  				}								
88c3e636   heziqi   Ziqi added big.h
439
  			}
11f177d5   heziqi   Ziqi added functi...
440
  			target.write(reinterpret_cast<const char*>(c), L);   //write normalized data into destination
88c3e636   heziqi   Ziqi added big.h
441
  		}
6708cc25   heziqi   Ziqi added envi c...
442
  
88c3e636   heziqi   Ziqi added big.h
443
444
445
446
447
448
449
  		
  		free(b);
  		free(c);
  		target.close();
  		return true;
  	}
  	
11f177d5   heziqi   Ziqi added functi...
450
451
  	//convert BIP file to BSQ file and save it
  	bool bsq(std::string outname)
f6169dea   heziqi   Ziqi completed bi...
452
453
454
455
456
457
  	{	
  		std::string temp = outname + "_temp";
  		std::string headtemp = temp + ".hdr";
  		//first creat a temporary bil file and convert bip file to bil file
  		bil(temp);
  
8a86bd56   David Mayerich   changed rts names...
458
  		stim::bil<T> n;
e933c14e   David Mayerich   cleaned up the code
459
  		if(n.open(temp, R[0], R[1], R[2], offset, w)==false){        //open infile
f6169dea   heziqi   Ziqi completed bi...
460
461
462
463
464
465
466
467
468
469
470
471
472
  			std::cout<<"ERROR: unable to open input file"<<std::endl;
  			exit(1);
  		}
  		//then convert bil file to bsq file
  		n.bsq(outname);
  		n.close();
  		remove(temp.c_str());
  		remove(headtemp.c_str());
  		return true;
  	}
  
  	//convert bip file to bil file and save it
  	bool bil(std::string outname)
11f177d5   heziqi   Ziqi added functi...
473
  	{
6708cc25   heziqi   Ziqi added envi c...
474
  		unsigned int S = R[0] * R[2] * sizeof(T);		//calculate the number of bytes in a ZX slice
11f177d5   heziqi   Ziqi added functi...
475
476
477
478
  		
  		std::ofstream target(outname.c_str(), std::ios::binary);
  		std::string headername = outname + ".hdr";
  		
f6169dea   heziqi   Ziqi completed bi...
479
  		T * p;			//pointer to the current ZX slice for bip file
11f177d5   heziqi   Ziqi added functi...
480
  		p = (T*)malloc(S);
f6169dea   heziqi   Ziqi completed bi...
481
482
  		T * q;			//pointer to the current XZ slice for bil file
  		q = (T*)malloc(S);
11f177d5   heziqi   Ziqi added functi...
483
  		
6708cc25   heziqi   Ziqi added envi c...
484
  		for ( unsigned i = 0; i < R[1]; i++)
11f177d5   heziqi   Ziqi added functi...
485
  		{			
f6169dea   heziqi   Ziqi completed bi...
486
  			getY(p, i);
6708cc25   heziqi   Ziqi added envi c...
487
  			for ( unsigned k = 0; k < R[2]; k++)
f6169dea   heziqi   Ziqi completed bi...
488
  			{
6708cc25   heziqi   Ziqi added envi c...
489
490
491
  				unsigned ks = k * R[0];
  				for ( unsigned j = 0; j < R[0]; j++)
  					q[ks + j] = p[k + j * R[2]];
f6169dea   heziqi   Ziqi completed bi...
492
493
  			}
  			target.write(reinterpret_cast<const char*>(q), S);   //write a band data into target file	
11f177d5   heziqi   Ziqi added functi...
494
  		}
f6169dea   heziqi   Ziqi completed bi...
495
  
11f177d5   heziqi   Ziqi added functi...
496
497
  		
  		free(p);
f6169dea   heziqi   Ziqi completed bi...
498
  		free(q);
11f177d5   heziqi   Ziqi added functi...
499
500
501
  		target.close();
  		return true;
  	}
88c3e636   heziqi   Ziqi added big.h
502
  
20c212c0   heziqi   Ziqi added functi...
503
504
505
506
507
508
509
510
511
512
513
514
515
  	//providing the left and the right bound data, return baseline-corrected band height
  	bool baseline_band(double lb, double rb, T* lp, T* rp, double wavelength, T* result){
  
  		unsigned XY = R[0] * R[1];
  		band(result, wavelength);		//get band
  
  		//perform the baseline correction
  		double r = (double) (wavelength - lb) / (double) (rb - lb);
  		for(unsigned i=0; i < XY; i++){
  			result[i] =(T) (result[i] - (rp[i] - lp[i]) * r - lp[i] );
  		}
  		return true;
  	}
517876d6   heziqi   metrics finished ...
516
  	
70407ea9   heziqi   Ziqi completed he...
517
  	bool height(double lb, double rb, double bandwavelength, T* result){
20c212c0   heziqi   Ziqi added functi...
518
519
520
  
  		T* lp;
  		T* rp;
70407ea9   heziqi   Ziqi completed he...
521
522
523
524
  		unsigned XY = R[0] * R[1];
  		unsigned S = XY * sizeof(T);
  		lp = (T*) malloc(S);			//memory allocation
  		rp = (T*) malloc(S);
20c212c0   heziqi   Ziqi added functi...
525
  
70407ea9   heziqi   Ziqi completed he...
526
  		band(lp, lb);
20c212c0   heziqi   Ziqi added functi...
527
528
529
530
  		band(rp, rb);		
  
  		baseline_band(lb, rb, lp, rp, bandwavelength, result);
  
517876d6   heziqi   metrics finished ...
531
532
  		free(lp);
  		free(rp);
20c212c0   heziqi   Ziqi added functi...
533
534
535
  		return true;
  	}
  
c359422d   heziqi   Ziqi added functi...
536
  
20c212c0   heziqi   Ziqi added functi...
537
  	//calculate the area between two bound point(including baseline correction)
70407ea9   heziqi   Ziqi completed he...
538
  	bool area(double lb, double rb, double lab, double rab, T* result){
20c212c0   heziqi   Ziqi added functi...
539
540
541
542
543
  
  		T* lp;	//left band pointer
  		T* rp;	//right band pointer
  		T* cur;		//current band 1
  		T* cur2;	//current band 2
20c212c0   heziqi   Ziqi added functi...
544
545
546
547
548
549
550
551
  
  		unsigned XY = R[0] * R[1];
  		unsigned S = XY * sizeof(T);
  
  		lp = (T*) malloc(S);			//memory allocation
  		rp = (T*) malloc(S);
  		cur = (T*) malloc(S);
  		cur2 = (T*) malloc(S);
20c212c0   heziqi   Ziqi added functi...
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
  
  		memset(result, (char)0, S);
  
  		//find the wavelenght position in the whole band
  		unsigned int n = w.size();
  		unsigned int ai = 0;		//left bound position
  		unsigned int bi = n - 1;		//right bound position
  
  
  
  		//to make sure the left and the right bound are in the bandwidth
  		if (lb < w[0] || rb < w[0] || lb > w[n-1] || rb >w[n-1]){
  			std::cout<<"ERROR: left bound or right bound out of bandwidth"<<std::endl;
  			exit(1);
  		}
  		//to make sure rigth bound is bigger than left bound
  		else if(lb > rb){
  			std::cout<<"ERROR: right bound should be bigger than left bound"<<std::endl;
  			exit(1);
  		}
  
  		//get the position of lb and rb
70407ea9   heziqi   Ziqi completed he...
574
  		while (lab >= w[ai]){
20c212c0   heziqi   Ziqi added functi...
575
576
  			ai++;
  		}
70407ea9   heziqi   Ziqi completed he...
577
  		while (rab <= w[bi]){
20c212c0   heziqi   Ziqi added functi...
578
579
580
581
582
583
584
  			bi--;
  		}
  
  		band(lp, lb);
  		band(rp, rb);
  
  		//calculate the beginning and the ending part
70407ea9   heziqi   Ziqi completed he...
585
586
  		baseline_band(lb, rb, lp, rp, rab, cur2);		//ending part
  		baseline_band(lb, rb, lp, rp, w[bi], cur);
20c212c0   heziqi   Ziqi added functi...
587
  		for(unsigned j = 0; j < XY; j++){
70407ea9   heziqi   Ziqi completed he...
588
  			result[j] += (rab - w[bi]) * (cur[j] + cur2[j]) / 2.0;
20c212c0   heziqi   Ziqi added functi...
589
  		}
70407ea9   heziqi   Ziqi completed he...
590
591
  		baseline_band(lb, rb, lp, rp, lab, cur2);		//beginnning part
  		baseline_band(lb, rb, lp, rp, w[ai], cur);
20c212c0   heziqi   Ziqi added functi...
592
  		for(unsigned j = 0; j < XY; j++){	
70407ea9   heziqi   Ziqi completed he...
593
  			result[j] += (w[ai] - lab) * (cur[j] + cur2[j]) / 2.0;
20c212c0   heziqi   Ziqi added functi...
594
  		}
517876d6   heziqi   metrics finished ...
595
  
20c212c0   heziqi   Ziqi added functi...
596
597
598
599
600
601
602
603
604
605
606
  		//calculate the area
  		ai++;
  		for(unsigned i = ai; i <= bi ;i++)
  		{
  			baseline_band(lb, rb, lp, rp, w[ai], cur2);
  			for(unsigned j = 0; j < XY; j++)
  			{
  				result[j] += (w[ai] - w[ai-1]) * (cur[j] + cur2[j]) / 2.0; 
  			}
  			std::swap(cur,cur2);		//swap the band pointers
  		}
70407ea9   heziqi   Ziqi completed he...
607
  
517876d6   heziqi   metrics finished ...
608
609
610
611
  		free(lp);
  		free(rp);
  		free(cur);
  		free(cur2);
20c212c0   heziqi   Ziqi added functi...
612
613
614
  		return true;
  	}
  
70407ea9   heziqi   Ziqi completed he...
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
  	//peak height ratio
  	bool ph_to_ph(double lb1, double rb1, double pos1, double lb2, double rb2, double pos2, T * result){
  
  		T* p1 = (T*)malloc(R[0] * R[1] * sizeof(T));	
  		T* p2 = (T*)malloc(R[0] * R[1] * sizeof(T));	
  		
  		//get the two peak band
  		height(lb1, rb1, pos1, p1);
  		height(lb2, rb2, pos2, p2);
  		//calculate the ratio in result
  		for(unsigned i = 0; i < R[0] * R[1]; i++){
  			if(p1[i] == 0 && p2[i] ==0)
  				result[i] = 1;
  			else
  				result[i] = p1[i] / p2[i];
  		}
517876d6   heziqi   metrics finished ...
631
632
633
  
  		free(p1);
  		free(p2);
70407ea9   heziqi   Ziqi completed he...
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
  		return true;	
  	}
  	
  	//peak are to peak height ratio
  	bool pa_to_ph(double lb1, double rb1, double lab1, double rab1,
  					double lb2, double rb2, double pos, T* result){
  		
  		T* p1 = (T*)malloc(R[0] * R[1] * sizeof(T));	
  		T* p2 = (T*)malloc(R[0] * R[1] * sizeof(T));	
  		
  		//get the area and the peak band
  		area(lb1, rb1, lab1, rab1, p1);
  		height(lb2, rb2, pos, p2);
  		//calculate the ratio in result
  		for(unsigned i = 0; i < R[0] * R[1]; i++){
  			if(p1[i] == 0 && p2[i] ==0)
  				result[i] = 1;
  			else
  				result[i] = p1[i] / p2[i];
  		}
517876d6   heziqi   metrics finished ...
654
655
656
  
  		free(p1);
  		free(p2);
70407ea9   heziqi   Ziqi completed he...
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
  		return true;	
  	}		
  	
  	//peak area to peak area ratio
  	bool pa_to_pa(double lb1, double rb1, double lab1, double rab1,
  					double lb2, double rb2, double lab2, double rab2, T* result){
  		
  		T* p1 = (T*)malloc(R[0] * R[1] * sizeof(T));	
  		T* p2 = (T*)malloc(R[0] * R[1] * sizeof(T));	
  		
  		//get the area and the peak band
  		area(lb1, rb1, lab1, rab1, p1);
  		area(lb2, rb2, lab2, rab2, p2);
  		//calculate the ratio in result
  		for(unsigned i = 0; i < R[0] * R[1]; i++){
  			if(p1[i] == 0 && p2[i] ==0)
  				result[i] = 1;
  			else
  				result[i] = p1[i] / p2[i];
  		}
517876d6   heziqi   metrics finished ...
677
678
679
  
  		free(p1);
  		free(p2);
70407ea9   heziqi   Ziqi completed he...
680
681
  		return true;	
  	}		
517876d6   heziqi   metrics finished ...
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
  
  	//x * f(x)
  	bool x_area(double lb, double rb, double lab, double rab, T* result){
  		T* lp;	//left band pointer
  		T* rp;	//right band pointer
  		T* cur;		//current band 1
  		T* cur2;	//current band 2
  
  		unsigned XY = R[0] * R[1];
  		unsigned S = XY * sizeof(T);
  
  		lp = (T*) malloc(S);			//memory allocation
  		rp = (T*) malloc(S);
  		cur = (T*) malloc(S);
  		cur2 = (T*) malloc(S);
  
  		memset(result, (char)0, S);
  
  		//find the wavelenght position in the whole band
  		unsigned int n = w.size();
  		unsigned int ai = 0;		//left bound position
  		unsigned int bi = n - 1;		//right bound position
  
  		//to make sure the left and the right bound are in the bandwidth
  		if (lb < w[0] || rb < w[0] || lb > w[n-1] || rb >w[n-1]){
  			std::cout<<"ERROR: left bound or right bound out of bandwidth"<<std::endl;
  			exit(1);
  		}
  		//to make sure rigth bound is bigger than left bound
  		else if(lb > rb){
  			std::cout<<"ERROR: right bound should be bigger than left bound"<<std::endl;
  			exit(1);
  		}
  
  		//get the position of lb and rb
  		while (lab >= w[ai]){
  			ai++;
  		}
  		while (rab <= w[bi]){
  			bi--;
  		}
  
  		band(lp, lb);
  		band(rp, rb);
  
  		//calculate the beginning and the ending part
  		baseline_band(lb, rb, lp, rp, rab, cur2);		//ending part
  		baseline_band(lb, rb, lp, rp, w[bi], cur);
  		for(unsigned j = 0; j < XY; j++){
  			result[j] += (rab - w[bi]) * (rab + w[bi]) * (cur[j] + cur2[j]) / 4.0;
  		}
  		baseline_band(lb, rb, lp, rp, lab, cur2);		//beginnning part
  		baseline_band(lb, rb, lp, rp, w[ai], cur);
  		for(unsigned j = 0; j < XY; j++){	
  			result[j] += (w[ai] - lab) * (w[ai] + lab) * (cur[j] + cur2[j]) / 4.0;
  		}
  
  		//calculate f(x) times x
  		ai++;
  		for(unsigned i = ai; i <= bi ;i++)
  		{
  			baseline_band(lb, rb, lp, rp, w[ai], cur2);
  			for(unsigned j = 0; j < XY; j++)
  			{
  				result[j] += (w[ai] - w[ai-1]) * (w[ai] + w[ai-1]) * (cur[j] + cur2[j]) / 4.0; 
  			}
  			std::swap(cur,cur2);		//swap the band pointers
  		}
  
  		free(lp);
  		free(rp);
  		free(cur);
  		free(cur2);
  		return true;
  	}
  
  	//centroid point
  	bool cpoint(double lb, double rb, double lab, double rab, T* result){
  		T* p1 = (T*)malloc(R[0] * R[1] * sizeof(T));	
  		T* p2 = (T*)malloc(R[0] * R[1] * sizeof(T));	
  		
  		//get the area and the peak band
  		x_area(lb, rb, lab, rab, p1);
  		area(lb, rb, lab, rab, p2);
  		//calculate the ratio in result
  		for(unsigned i = 0; i < R[0] * R[1]; i++){
  			if(p1[i] == 0 && p2[i] ==0)
  				result[i] = 1;
  			else
  				result[i] = p1[i] / p2[i];
  		}
  
  		free(p1);
  		free(p2);
  		return true;	
  	}
  
4a6f666c   heziqi   Added mask method
779
  	//create mask file
f4c5d71b   David Mayerich   started working w...
780
  	bool build_mask(double mask_band, double threshold, unsigned char* p){
4a6f666c   heziqi   Added mask method
781
782
783
784
785
786
787
788
789
790
791
  
  		T* temp = (T*)malloc(R[0] * R[1] * sizeof(T));		//allocate memory for the certain band
  		band(temp, mask_band);
  
  		for (unsigned int i = 0; i < R[0] * R[1];i++) {
  				if (temp[i] < threshold)
  					p[i] = 0;
  				else
  					p[i] = 255;
  		}
  
517876d6   heziqi   metrics finished ...
792
  		free(temp);
4a6f666c   heziqi   Added mask method
793
794
795
796
  		return true;
  
  	}
  
740f8cd2   heziqi   added apply_mask
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
  	//apply mask
  	bool apply_mask(std::string outfile, unsigned char* p){
  
  		std::ofstream target(outfile.c_str(), std::ios::binary);
  
  		unsigned ZX = R[2] * R[0];		//calculate number of a band
  		unsigned L = ZX * sizeof(T);
  
  		T * temp = (T*)malloc(L);
  
  		for (unsigned i = 0; i < R[1]; i++)
  		{
  			getY(temp, i);
  			for ( unsigned j = 0; j < R[0]; j++)
  			{
  				for (unsigned k = 0; k < R[2]; k++)
  				{
  				if(p[i * R[0] + k] == 0)
  					temp[j * R[2] + k] = 0;
  				else
  					continue;
  				}
  			}
  			target.write(reinterpret_cast<const char*>(temp), L);   //write a band data into target file	
  		}
  		target.close();
  		free(temp);
  		return true;
  	}
  
a43c4fe1   heziqi   Added crop in env...
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
  	//calculate the average band of the file
  	bool band_avg(T* p){
  		unsigned long long XY = R[0] * R[1];
  		//get every pixel and calculate average value
  		T* temp = (T*)malloc(sizeof(T) * R[2]);
  		T sum;
  		for (unsigned i = 0; i < XY; i++){
  			pixel(temp, i);
  			//calculate the sum value of every value
  			sum = 0;		//initialize sum value
  			for (unsigned j = 0; j < R[2]; j++){
  				sum += temp[j]/(T)R[2];
  			}
  			p[i] = sum;
  		}
  		free(temp);
  		return true;
  	}
  
  	//calculate the average number of every band
  	bool avg_band(T*p, unsigned char* mask){
  		unsigned long long XY = R[0] * R[1];
  		T* temp = (T*)malloc(sizeof(T) * R[2]);
  		//Iinitialize
  		for (unsigned j = 0; j < R[2]; j++){
  			p[j] = 0;
  		}
  		//calculate vaild number in a band
  		unsigned count = 0;
  		for (unsigned j = 0; j < XY; j++){
  			if (mask[j] != 0){
  				count++;
  			}
  		}
  		//calculate average number of a band
  		for (unsigned i = 0; i < XY; i++){
  			if (mask[i] != 0){			
  				pixel(temp, i);
  				for (unsigned j = 0; j < R[2]; j++){
  					p[j] += temp[j] / (T)count;
  				}
  			}
  		}
  		free(temp);
  		return true;
  	}
  	//calculate correlation coefficient matrix
  	bool co_matrix(T* co, T* avg, unsigned char *mask){
  		//memory allocation
  		unsigned long long xy = R[0] * R[1];
  		unsigned int B = R[2];
  		T* temp = (T*)malloc(sizeof(T) * B);
  		//count vaild pixels in a band
  		unsigned count = 0;
  		for (unsigned j = 0; j < xy; j++){
  			if (mask[j] != 0){
  				count++;
  			}
  		}
  		//initialize correlation matrix
  		for (unsigned i = 0; i < B; i++){
  			for (unsigned k = 0; k < B; k++){
  				co[i * B + k] = 0;
  			}
  		}
  		//calculate correlation coefficient matrix
  		for (unsigned j = 0; j < xy; j++){
  			if (mask[j] != 0){
  				pixel(temp, j);
  				for (unsigned i = 0; i < B; i++){
  					for (unsigned k = i; k < B; k++){
  						co[i * B + k] += (temp[i] - avg[i]) * (temp[k] - avg[k]) / count;
  					}
  				}
  			}
  		}
  		//because correlation matrix is symmetric
  		for (unsigned i = 0; i < B; i++){
  			for (unsigned k = i + 1; k < B; k++){
  				co[k * B + i] = co[i * B + k];
  			}
  		}
  
  		free(temp);
  		return true;
  	}
  
0df38ff3   heziqi   fixed head detached
914
  
a43c4fe1   heziqi   Added crop in env...
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
  	//crop specified area the of the original file
  	bool crop(std::string outfile, unsigned x0, unsigned y0, unsigned x1, unsigned y1){
  
  		//calculate the new number of samples and lines
  		unsigned long long sam = x1 - x0;		//samples
  		unsigned long long lin = y1 - y0;		//lines
  		unsigned long long L = R[2] * sizeof(T);
  		//get specified band and save
  		T* temp = (T*)malloc(L);
  		std::ofstream out(outfile.c_str(), std::ios::binary);
  		//get start
  		unsigned long long sp = y0 * R[0] + x0;		//start pixel
  		for (unsigned i = 0; i < lin; i++)
  		{
  			for (unsigned j = 0; j < sam; j++)
  			{
  				pixel(temp, sp + j + i * R[0]);
  				out.write(reinterpret_cast<const char*>(temp), L);   //write slice data into target file	
  			}
  		}
  		free(temp);
  		return true;
  	}
  
0df38ff3   heziqi   fixed head detached
939
  
f6169dea   heziqi   Ziqi completed bi...
940
941
942
943
944
945
  	//close the file
  	bool close(){
  		file.close();
  		return true;
  	}
  
88c3e636   heziqi   Ziqi added big.h
946
  	};
6aa04ba2   David Mayerich   interleave types
947
  }
e8eb202f   David Mayerich   added a new ENVI ...
948
949
  
  #endif