added support for the image interface to envi

David Mayerich
1 parent fbf0ab02
Showing 2 changed files with 29 additions and 7 deletions Show diff stats
envi/envi.h
image/image.h
@@ -6,7 +6,7 @@
 #include "../envi/bip.h"
 #include "../envi/bil.h"
 #include <iostream>
-//#include "../image/CImg.h"
+#include "../image/image.h"
 namespace stim{
@@ -641,10 +641,10 @@ public:
 	//load mask from maskfile and save it into memory
 	bool load_mask(unsigned char * mask, std::string maskname){
 		//open the mask file
-		cimg_library::CImg<unsigned char> mask_image(maskname.c_str());
+		stim::image<unsigned char> mask_image(maskname);
 		//save mask file into memory
 		memcpy(mask, mask_image.data(), mask_image.size());
-		mask_image.clear();
+		//mask_image.clear();
 		return true;
 	}
@@ -4,6 +4,8 @@
 #include "CImg.h"
 #include <iostream>
+
+namespace stim{
 //This static class provides the STIM interface for loading images
 //	Use this interface for all image management - that way the actual library can be changed without problems
@@ -12,18 +14,27 @@
 template <class T>
 class image{
-	cimg_library::CImg<T> data;
+	cimg_library::CImg<T> img;
 public:
+	//default constructor
+	image(){
+	}
+
+	//constructor (load an image file)
+	image(std::string filename){
+		img.load(filename.c_str());
+	}
+
 	//Load an image from a file
 	void load(std::string filename){
-		data.load(filename.c_str());
+		img.load(filename.c_str());
 	}
 	//save a file
 	void save(std::string filename){
-		data.save(filename.c_str());
+		img.save(filename.c_str());
 	}
 	//create an image from an interlaced buffer
@@ -38,8 +49,17 @@ public:
 			}
 		}
-		data = cimg_library::CImg<unsigned char>(non_interleaved, width, height, 1, channels);
+		img = cimg_library::CImg<unsigned char>(non_interleaved, width, height, 1, channels);
+	}
+
+	//returns a pointer to the first pixel of the image data
+	T* data(){
+		return img.data();
+	}
+	//returns the size (number of values) of the image
+	unsigned long size(){
+		return img.size();
 	}
@@ -48,5 +68,7 @@ public:
 };
+};		//end namespace stim
+
 #endif