added support for the new image interface

David Mayerich
1 parent d32a1854
Showing 3 changed files with 24 additions and 9 deletions Show diff stats
envi/envi.h
image/image.h
visualization/colormap.h
@@ -6,7 +6,7 @@
 #include "../envi/bip.h"
 #include "../envi/bil.h"
 #include <iostream>
-#include "../CImg/CImg.h"
+//#include "../image/CImg.h"
 namespace stim{
 #ifndef STIM_IMAGE_H
 #define STIM_IMAGE_H
+#include "CImg.h"
+
 #include <iostream>
 //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
@@ -10,7 +12,7 @@
 template <class T>
 class image{
-	CImg<T> data;
+	cimg_library::CImg<T> data;
 public:
@@ -25,8 +27,18 @@ public:
 	}
 	//create an image from an interlaced buffer
-	void create_interlaced(T* buffer, unsigned int channels = 1){
+	void set_interleaved(T* buffer, unsigned int width, unsigned int height, unsigned int channels = 1){
+
+		unsigned char* non_interleaved = (unsigned char*)malloc(width * height * 3);
+		unsigned int S = width * height;
+
+		for(unsigned int i = 0; i < S; i++){
+			for(unsigned int c = 0; c < channels; c++){
+				non_interleaved[i + c * S] = buffer[i * channels + c];
+			}
+		}
+		data = cimg_library::CImg<unsigned char>(non_interleaved, width, height, 1, channels);
 	}
@@ -7,7 +7,7 @@
 //saving an image to a file uses the CImg library
 	//this currently throws a lot of "unreachable" warnings (as of GCC 4.8.2, nvcc 6.5.12)
-#include "../CImg/CImg.h"
+#include "../image/image.h"
 #define BREWER_CTRL_PTS 11
@@ -34,20 +34,23 @@ namespace stim{
 enum colormapType {cmBrewer, cmGrayscale, cmRainbow};
-static void buffer2image(unsigned char* buffer, std::string filename, unsigned int x_size, unsigned int y_size)
+static void buffer2image(unsigned char* buffer, std::string filename, unsigned int width, unsigned int height)
 {
-	unsigned char* non_interleaved = (unsigned char*)malloc(x_size * y_size * 3);
+	/*unsigned char* non_interleaved = (unsigned char*)malloc(x_size * y_size * 3);
 	unsigned int S = x_size * y_size;
 	for(unsigned int i = 0; i < S; i++){
 		non_interleaved[i + 0 * S] = buffer[i * 3 + 0];
 		non_interleaved[i + 1 * S] = buffer[i * 3 + 1];
 		non_interleaved[i + 2 * S] = buffer[i * 3 + 2];
-	}
+	}*/
 	//create an image object
-	cimg_library::CImg<unsigned char> image(non_interleaved, x_size, y_size, 1, 3);
-	image.save(filename.c_str());
+	//cimg_library::CImg<unsigned char> image(non_interleaved, x_size, y_size, 1, 3);
+	//image.save(filename.c_str());
+    image<unsigned char> I;
+    I.set_interleaved(buffer, width, height, 3);
+	I.save(filename);
 }
 #ifdef __CUDACC__