allowed masking during normalization

David Mayerich
1 parent 81e0d221
Showing 4 changed files with 21 additions and 9 deletions Show diff stats
envi/bil.h
envi/bip.h
envi/bsq.h
envi/envi.h
@@ -309,7 +309,7 @@ public:
 	}
 	// normalize the BIL file
-	bool normalize(std::string outname, double w)
+	bool normalize(std::string outname, double w, double t = 0.0)
 	{
 		unsigned int B = R[2];		//calculate the number of bands
 		unsigned int Y = R[1];
@@ -337,7 +337,10 @@ public:
 			{
 				for(unsigned m = 0; m < X; m++)
 				{
-					c[m + i * X] = c[m + i * X] / b[m + j * X];
+					if( b[m + j * X] < t )
+						c[m + i * X] = (T)0.0;
+					else
+						c[m + i * X] = c[m + i * X] / b[m + j * X];
 				}								
 			}
 			target.write(reinterpret_cast<const char*>(c), L);   //write normalized data into destination
@@ -401,7 +401,7 @@ public:
 	}
 	// normalize the BIP file
-	bool normalize(std::string outname, double w)
+	bool normalize(std::string outname, double w, double t = 0.0)
 	{
 		unsigned int B = R[2];		//calculate the number of bands
 		unsigned int Y = R[1];
@@ -431,7 +431,10 @@ public:
 				unsigned iB = i * B;		
 				for(unsigned m = 0; m < B; m++)
 				{
-					c[m + iB] = c[m + iB] / b[i + jX];			//perform normalization
+					if( b[i+jX] < t )
+						c[m + iB] = (T)0.0;
+					else
+						c[m + iB] = c[m + iB] / b[i + jX];			//perform normalization
 				}								
 			}
 			target.write(reinterpret_cast<const char*>(c), L);   //write normalized data into destination
@@ -252,7 +252,10 @@ public:
 	}
 	// normalize the BSQ file
-	bool normalize(std::string outname, double w)
+	//	outname - output file name
+	//	w - band to normalize to
+	//	t - threshold (if w<t, all pixel bands are set to zero)
+	bool normalize(std::string outname, double w, double t = 0.0)
 	{
 		unsigned int B = R[2];		//calculate the number of bands
 		unsigned int XY = R[0] * R[1];	//calculate the number of pixels in a band
@@ -274,7 +277,10 @@ public:
 			band_index(c, j);                     //get the current band into memory
 			for(unsigned i = 0; i < XY; i++)
 			{
-				c[i] = c[i] / b[i];
+				if(b[i] < t)
+					c[i] = (T)0.0;
+				else
+					c[i] = c[i] / b[i];
 			}
 			target.write(reinterpret_cast<const char*>(c), S);   //write normalized data into destination
 		}
@@ -96,13 +96,13 @@ public:
 	}
 	//perform normalization
-	bool normalize(std::string outfile, double band){
+	bool normalize(std::string outfile, double band, double threshold = 0.0){
 		if(header.interleave == envi_header::BSQ){		//if the infile is bsq file
 			if(header.data_type ==envi_header::float32)
-				return ((bsq<float>*)file)->normalize(outfile, band);
+				return ((bsq<float>*)file)->normalize(outfile, band, threshold);
 			else if(header.data_type == envi_header::float64)
-				return ((bsq<double>*)file)->normalize(outfile,band);
+				return ((bsq<double>*)file)->normalize(outfile,band, threshold);
 			else
 				std::cout<<"ERROR: unidentified data type"<<std::endl;
 		}