codebase / stimlib

Commit 4a6f666c0f5db7de6bb74221eb8306f699658ccc

Authored by heziqi
1 parent 646a89c5

Added mask method

Showing 4 changed files with 85 additions and 4 deletions   Show diff stats
Inline Side-by-side
envi/bil.h
  Show/Hide comments   View file @4a6f666
... ... @@ -387,6 +387,22 @@ public:
387 387 return true;
388 388 }
389 389  
  390 + //create mask file
  391 + bool mask(unsigned char* p, double mask_band, double threshold){
  392 +
  393 + T* temp = (T*)malloc(R[0] * R[1] * sizeof(T)); //allocate memory for the certain band
  394 + band(temp, mask_band);
  395 +
  396 + for (unsigned int i = 0; i < R[0] * R[1]; i++) {
  397 + if (temp[i] < threshold)
  398 + p[i] = 0;
  399 + else
  400 + p[i] = 255;
  401 + }
  402 +
  403 + return true;
  404 + }
  405 +
390 406 //close the file
391 407 bool close(){
392 408 file.close();
... ...
envi/bip.h
  Show/Hide comments   View file @4a6f666
... ... @@ -477,6 +477,23 @@ public:
477 477 return true;
478 478 }
479 479  
  480 + //create mask file
  481 + bool mask(unsigned char* p, double mask_band, double threshold){
  482 +
  483 + T* temp = (T*)malloc(R[0] * R[1] * sizeof(T)); //allocate memory for the certain band
  484 + band(temp, mask_band);
  485 +
  486 + for (unsigned int i = 0; i < R[0] * R[1];i++) {
  487 + if (temp[i] < threshold)
  488 + p[i] = 0;
  489 + else
  490 + p[i] = 255;
  491 + }
  492 +
  493 + return true;
  494 +
  495 + }
  496 +
480 497 //close the file
481 498 bool close(){
482 499 file.close();
... ...
envi/bsq.h
  Show/Hide comments   View file @4a6f666
... ... @@ -310,6 +310,22 @@ public:
310 310 return true;
311 311 }
312 312  
  313 + //create mask file
  314 + bool mask(unsigned char* p, double mask_band, double threshold){
  315 +
  316 + T* temp = (T*)malloc(R[0] * R[1] * sizeof(T)); //allocate memory for the certain band
  317 + band(temp, mask_band);
  318 +
  319 + for (unsigned int i = 0; i < R[0] * R[1]; i++) {
  320 + if (temp[i] < threshold)
  321 + p[i] = 0;
  322 + else
  323 + p[i] = 255;
  324 + }
  325 +
  326 + return true;
  327 +
  328 + }
313 329 //close the file
314 330 bool close(){
315 331 file.close();
... ...
envi/envi.h
  Show/Hide comments   View file @4a6f666
... ... @@ -9,14 +9,14 @@
9 9 namespace rts{
10 10  
11 11 //container class for an ENVI binary file reader
12   -class envi{
13   -
14   - envi_header header;
  12 +class envi{
15 13  
16 14 void* file; //void pointer to the relevant file reader (bip, bsq, or bil - with appropriate data type)
17 15  
18 16 public:
19 17  
  18 + envi_header header;
  19 +
20 20 bool allocate(){
21 21  
22 22 file = NULL; //set file to a NULL pointer
... ... @@ -274,8 +274,40 @@ public:
274 274 }
275 275  
276 276 //get the mask
277   - bool mask(unsigned char* mask, mask_band, threshold) {
  277 + bool mask(unsigned char* p, double mask_band, double threshold) {
  278 +
  279 + if(header.interleave == envi_header::BSQ){ //if the infile is bsq file
  280 + if(header.data_type ==envi_header::float32)
  281 + return ((bsq<float>*)file)->mask(p, mask_band, threshold);
  282 + else if(header.data_type == envi_header::float64)
  283 + return ((bsq<double>*)file)->mask(p, mask_band, threshold);
  284 + else
  285 + std::cout<<"ERROR: unidentified data type"<<std::endl;
  286 + }
  287 +
  288 + else if(header.interleave == envi_header::BIL){ //if the infile is bil file
  289 + if(header.data_type ==envi_header::float32)
  290 + return ((bil<float>*)file)->mask(p, mask_band, threshold);
  291 + else if(header.data_type == envi_header::float64)
  292 + return ((bil<double>*)file)->mask(p, mask_band, threshold);
  293 + else
  294 + std::cout<<"ERROR: unidentified data type"<<std::endl;
  295 + }
278 296  
  297 + else if(header.interleave == envi_header::BIP){ //if the infile is bip file
  298 + if(header.data_type ==envi_header::float32)
  299 + return ((bip<float>*)file)->mask(p, mask_band, threshold);
  300 + else if(header.data_type == envi_header::float64)
  301 + return ((bip<double>*)file)->mask(p, mask_band, threshold);
  302 + else
  303 + std::cout<<"ERROR: unidentified data type"<<std::endl;
  304 + }
  305 +
  306 + else{
  307 + std::cout<<"ERROR: unidentified file type"<<std::endl;
  308 + exit(1);
  309 + }
  310 + return false;
279 311 }
280 312  
281 313 bool close(){
... ...