Blame view

stim/image/image.h 26.3 KB
d32a1854   David Mayerich   added framework f...
1
2
  #ifndef STIM_IMAGE_H
  #define STIM_IMAGE_H
fbf0ab02   David Mayerich   added support for...
3
  
4d80c1dd   David Mayerich   added a GLUT temp...
4
5
6
7
  #ifdef _WIN32
  #undef max
  #endif
  
2e5e3a26   David Mayerich   added 2D convolut...
8
  #ifdef USING_OPENCV
dd5aab2f   David Mayerich   fixed errors in s...
9
10
11
  	//#include <opencv2/core/core.hpp>
  	//#include <opencv2/highgui/highgui.hpp>
  	#include <opencv2/opencv.hpp>
7d01bb90   David Mayerich   matlab matrix sav...
12
13
  #else
  	#include <stim/image/bmp.h>
2e5e3a26   David Mayerich   added 2D convolut...
14
  #endif
912d9073   David Mayerich   added vector.h to...
15
  #include <vector>
d32a1854   David Mayerich   added framework f...
16
  #include <iostream>
e98c381d   David Mayerich   Changed image.h t...
17
  #include <limits>							//use limits and remove the MIN and MAX macros
e252a6d6   David Mayerich   somehow the C++11...
18
  #include <typeinfo>
cfcf8619   David Mayerich   changes to filena...
19
  #include <fstream>
b4cea01a   David Mayerich   removed automated...
20
  #include <cstring>
6d30a707   Tianshu Cheng   add cuda/array_ad...
21
  
7d01bb90   David Mayerich   matlab matrix sav...
22
  
d9b2b2a8   David Mayerich   enabled basic rea...
23
24
  #include <stim/parser/filename.h>
  
7b3948ab   David Mayerich   added support for...
25
  namespace stim{
c8c976a9   David Mayerich   replaced CImg wit...
26
27
  /// This static class provides the STIM interface for loading, saving, and storing 2D images.
  /// Data is stored in an interleaved (BIP) format (default for saving and loading is RGB).
d32a1854   David Mayerich   added framework f...
28
29
30
31
32
33
  
  //currently this interface uses CImg
  //	T = data type (usually unsigned char)
  template <class T>
  class image{
  
cfcf8619   David Mayerich   changes to filena...
34
  	T* img;										//pointer to the image data (interleaved RGB for color)
c8c976a9   David Mayerich   replaced CImg wit...
35
  	size_t R[3];
fdc7f32c   David Mayerich   Added a coupled w...
36
  	bool interleaved = true;					//by default the data is interleaved
c8c976a9   David Mayerich   replaced CImg wit...
37
  
dbeb83f2   David Mayerich   added separable c...
38
39
40
  	inline size_t X() const { return R[1]; }
  	inline size_t Y() const { return R[2]; }
  	inline size_t C() const { return R[0]; }
c8c976a9   David Mayerich   replaced CImg wit...
41
  
c8c976a9   David Mayerich   replaced CImg wit...
42
43
44
45
46
47
48
  	void init(){								//initializes all variables, assumes no memory is allocated
  		memset(R, 0, sizeof(size_t) * 3);		//set the resolution and number of channels to zero
  		img = NULL;
  	}
  
  	void unalloc(){								//frees any resources associated with the image
  		if(img)	free(img);						//if memory has been allocated, free it
c8c976a9   David Mayerich   replaced CImg wit...
49
  	}
1a224b6a   David Mayerich   fixed linux compa...
50
  
c8c976a9   David Mayerich   replaced CImg wit...
51
52
53
54
55
56
57
  
  	void clear(){								//clears all image data
  		unalloc();								//unallocate previous memory
  		init();									//re-initialize the variables
  	}
  
  	void allocate(){
e9a1b974   Pavel Govyadinov   imlemented Dr. Ma...
58
  		unalloc();
814eb271   David Mayerich   fixed problems wi...
59
  		img = (T*) malloc( sizeof(T) * R[0] * R[1] * R[2] );	//allocate memory
55417e1e   David Mayerich   fixed data type l...
60
61
62
63
  		if (img == NULL) {
  			std::cout << "stim::image ERROR - failed to allocate memory for image" << std::endl;
  			exit(1);
  		}
c8c976a9   David Mayerich   replaced CImg wit...
64
65
66
  	}
  
  	void allocate(size_t x, size_t y, size_t c){	//allocate memory based on the resolution
c8c976a9   David Mayerich   replaced CImg wit...
67
68
69
70
  		R[0] = c; R[1] = x; R[2] = y;				//set the resolution
  		allocate();									//allocate memory
  	}
  
86dfa80b   David Mayerich   added 2D gaussian...
71
  	inline size_t idx(size_t x, size_t y, size_t c = 0) const {
dbeb83f2   David Mayerich   added separable c...
72
  		return y * R[0] * R[1] + x * R[0] + c;
c8c976a9   David Mayerich   replaced CImg wit...
73
  	}
1a224b6a   David Mayerich   fixed linux compa...
74
  
2e5e3a26   David Mayerich   added 2D convolut...
75
  #ifdef USING_OPENCV
c8c976a9   David Mayerich   replaced CImg wit...
76
  	int cv_type(){
fdc7f32c   David Mayerich   Added a coupled w...
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
  		if (C() == 1 || C() == 3) {																	//if the image has 1 or 3 channels, this will work
  			if (typeid(T) == typeid(unsigned char))		return CV_MAKETYPE(CV_8U, (int)C());
  			if (typeid(T) == typeid(char))				return CV_MAKETYPE(CV_8S, (int)C());
  			if (typeid(T) == typeid(unsigned short))		return CV_MAKETYPE(CV_16U, (int)C());
  			if (typeid(T) == typeid(short))				return CV_MAKETYPE(CV_16S, (int)C());
  			if (typeid(T) == typeid(int))				return CV_MAKETYPE(CV_32S, (int)C());
  			if (typeid(T) == typeid(float))				return CV_MAKETYPE(CV_32F, (int)C());
  			if (typeid(T) == typeid(double))				return CV_MAKETYPE(CV_64F, (int)C());
  		}
  		else if (C() == 4) {																		//OpenCV only supports saving BGR images - 4-channel isn't supported
  			if (typeid(T) == typeid(unsigned char))		return CV_MAKETYPE(CV_8U, 3);
  			if (typeid(T) == typeid(char))				return CV_MAKETYPE(CV_8S, 3);
  			if (typeid(T) == typeid(unsigned short))		return CV_MAKETYPE(CV_16U, 3);
  			if (typeid(T) == typeid(short))				return CV_MAKETYPE(CV_16S, 3);
  			if (typeid(T) == typeid(int))				return CV_MAKETYPE(CV_32S, 3);
  			if (typeid(T) == typeid(float))				return CV_MAKETYPE(CV_32F, 3);
  			if (typeid(T) == typeid(double))				return CV_MAKETYPE(CV_64F, 3);
  		}
1a224b6a   David Mayerich   fixed linux compa...
95
  
c8c976a9   David Mayerich   replaced CImg wit...
96
  		std::cout<<"ERROR in stim::image::cv_type - no valid data type found"<<std::endl;
3ff136b3   David Mayerich   replaced C++11 co...
97
  		exit(1);
c8c976a9   David Mayerich   replaced CImg wit...
98
  	}
2e5e3a26   David Mayerich   added 2D convolut...
99
  #endif
a2bf1d08   David Mayerich   general bug fixes...
100
101
  	/// Returns the value for "white" based on the dynamic range (assumes white is 1.0 for floating point images)
  	T white(){
11ff9d7e   David Mayerich   improved user out...
102
103
104
105
  		if (typeid(T) == typeid(double) || typeid(T) == typeid(float))
  			return (T)1.0;
  		else
  			return std::numeric_limits<T>::max();
a2bf1d08   David Mayerich   general bug fixes...
106
107
  	}
  
d32a1854   David Mayerich   added framework f...
108
109
  public:
  
a7ed4b2e   David Mayerich   ivote updates
110
111
  	size_t bytes() { return size() * sizeof(T); }
  
a2bf1d08   David Mayerich   general bug fixes...
112
  	/// Default constructor - creates an empty image object
814eb271   David Mayerich   fixed problems wi...
113
  	image(){ init(); }							//initialize all variables to zero, don't allocate any memory
7b3948ab   David Mayerich   added support for...
114
  
a2bf1d08   David Mayerich   general bug fixes...
115
  	/// Constructor with a filename - loads the specified file
c8c976a9   David Mayerich   replaced CImg wit...
116
  	image(std::string filename){				//constructor initialize the image with an image file
fba6e7e3   Pavel Govyadinov   fixed image.h bug...
117
  		init();
c8c976a9   David Mayerich   replaced CImg wit...
118
  		load(filename);
7b3948ab   David Mayerich   added support for...
119
120
  	}
  
c8c976a9   David Mayerich   replaced CImg wit...
121
  	/// Create a new image from scratch given a number of samples and channels
1a224b6a   David Mayerich   fixed linux compa...
122
  	image(size_t x, size_t y = 1, size_t c = 1){
e9a1b974   Pavel Govyadinov   imlemented Dr. Ma...
123
  		init();
a2bf1d08   David Mayerich   general bug fixes...
124
  		allocate(x, y, c);
c8c976a9   David Mayerich   replaced CImg wit...
125
  	}
f186dbda   Tianshu Cheng   header file for b...
126
  
a2bf1d08   David Mayerich   general bug fixes...
127
  	/// Create a new image with the data given in 'data'
c8c976a9   David Mayerich   replaced CImg wit...
128
  	image(T* data, size_t x, size_t y, size_t c = 1){
e9a1b974   Pavel Govyadinov   imlemented Dr. Ma...
129
  		init();
c8c976a9   David Mayerich   replaced CImg wit...
130
131
  		allocate(x, y, c);
  		memcpy(img, data, bytes());
41acaf5d   David Mayerich   added a CUDA Gaus...
132
133
  	}
  
a2bf1d08   David Mayerich   general bug fixes...
134
  	/// Copy constructor - duplicates an image object
814eb271   David Mayerich   fixed problems wi...
135
  	image(const stim::image<T>& I){
e9a1b974   Pavel Govyadinov   imlemented Dr. Ma...
136
  		init();
a2bf1d08   David Mayerich   general bug fixes...
137
  		allocate(I.X(), I.Y(), I.C());
a2bf1d08   David Mayerich   general bug fixes...
138
139
140
141
142
143
144
145
  		memcpy(img, I.img, bytes());
  	}
  
  	/// Destructor - clear memory
  	~image(){
  		free(img);
  	}
  
a7ed4b2e   David Mayerich   ivote updates
146
  	///Resize an image - this function looks like it hasn't been implemented
86dfa80b   David Mayerich   added 2D gaussian...
147
148
149
150
  	void resize(size_t x, size_t y, size_t c = 1) {
  		allocate(x, y, c);
  	}
  
735a2a24   David Mayerich   started testing o...
151
  	stim::image<T>& operator=(const stim::image<T>& I){
a2bf1d08   David Mayerich   general bug fixes...
152
153
  		if(&I == this)									//handle self-assignment
  			return *this;
276f9b23   David Mayerich   forced matching t...
154
  		init();
a2bf1d08   David Mayerich   general bug fixes...
155
156
157
158
  		allocate(I.X(), I.Y(), I.C());
  		memcpy(img, I.img, bytes());
  		return *this;
  	}
d661ab7d   David Mayerich   linux fixes for s...
159
  #ifndef USING_OPENCV
d9b2b2a8   David Mayerich   enabled basic rea...
160
161
162
163
164
165
166
167
168
169
  	void load_bmp(std::string filename) {
  		stim::bmp bitmap;
  		bitmap.open(filename);											//load the bitmap and read the headers
  		resize(bitmap.width, bitmap.height, 3);							//resize the current image to match the bitmap
  		if (!bitmap.read((char*)img)) {										//read the bits from file
  			std::cout << "stim::image ERROR: problem loading bitmap image." << std::endl;
  			exit(1);
  		}
  		bitmap.close();													//close the bitmap file
  	}
7c396e30   David Mayerich   fixed Linux stim:...
170
  #endif
d9b2b2a8   David Mayerich   enabled basic rea...
171
  
08ada8c2   David Mayerich   updated image to ...
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
  	//determines if a filename represents a valid file format that can be loaded/saved
  	static bool test_filename(std::string f) {
  		stim::filename fname = f;
  		std::string ext = fname.extension();
  #ifdef USING_OPENCV
  		if (ext == "bmp" ||
  			ext == "jpg" ||
  			ext == "png" ||
  			ext == "pbm" ||
  			ext == "tif" )
  			return true;
  #else
  		if (ext == "pbm" || ext == "bmp")
  			return true;
  #endif
  		return false;
  	}
  
cfcf8619   David Mayerich   changes to filena...
190
191
  	//save a Netpbm file
  	void load_netpbm(std::string filename) {
683f216a   David Mayerich   fixed? Linux comp...
192
  		std::ifstream infile(filename.c_str(), std::ios::in | std::ios::binary);		//open an output file
cfcf8619   David Mayerich   changes to filena...
193
194
195
196
  		if (!infile) {
  			std::cout << "Error opening input file in image::load_netpbm()" << std::endl;
  			exit(1);
  		}
cfcf8619   David Mayerich   changes to filena...
197
  
2e5e3a26   David Mayerich   added 2D convolut...
198
199
  		size_t nc;													//allocate space for the number of channels
  		char format[2];												//allocate space to hold the image format tag
cfcf8619   David Mayerich   changes to filena...
200
  		infile.read(format, 2);										//read the image format tag
2e5e3a26   David Mayerich   added 2D convolut...
201
  		infile.seekg(1, std::ios::cur);								//skip the newline character
cfcf8619   David Mayerich   changes to filena...
202
203
204
205
206
  
  		if (format[0] != 'P') {
  			std::cout << "Error in image::load_netpbm() - file format tag is invalid: " << format[0] << format[1] << std::endl;
  			exit(1);
  		}
2e5e3a26   David Mayerich   added 2D convolut...
207
208
  		if (format[1] == '5') nc = 1;								//get the number of channels from the format flag
  		else if (format[1] == '6') nc = 3;
cfcf8619   David Mayerich   changes to filena...
209
210
211
212
  		else {
  			std::cout << "Error in image::load_netpbm() - file format tag is invalid: " << format[0] << format[1] << std::endl;
  			exit(1);
  		}
2e5e3a26   David Mayerich   added 2D convolut...
213
214
215
216
217
218
219
220
221
  
  		unsigned char c;								//stores a character
  		while (infile.peek() == '#') {					//if the next character indicates the start of a comment
  			while (true) {
  				c = infile.get();
  				if (c == 0x0A) break;
  			}
  		}
  
cfcf8619   David Mayerich   changes to filena...
222
  		std::string sw;									//create a string to store the width of the image
cfcf8619   David Mayerich   changes to filena...
223
224
225
226
227
228
229
230
231
232
233
234
235
  		while(true){
  			c = infile.get();							//get a single character
  			if (c == ' ') break;						//exit if we've encountered a space
  			sw.push_back(c);							//push the character on to the string
  		}
  		size_t w = atoi(sw.c_str());					//convert the string into an integer
  
  		std::string sh;
  		while (true) {
  			c = infile.get();
  			if (c == 0x0A) break;
  			sh.push_back(c);
  		}
cfcf8619   David Mayerich   changes to filena...
236
  
2e5e3a26   David Mayerich   added 2D convolut...
237
238
239
240
241
242
  		while (true) {									//skip the maximum value
  			c = infile.get();
  			if (c == 0x0A) break;
  		}
  		size_t h = atoi(sh.c_str());					//convert the string into an integer
  
01707489   David Mayerich   implemented a new...
243
244
245
246
247
248
  		allocate(w, h, nc);													//allocate space for the image
  		unsigned char* buffer = (unsigned char*)malloc(w * h * nc);			//create a buffer to store the read data
  		infile.read((char*)buffer, size());									//copy the binary data from the file to the image
  		infile.close();														//close the file
  		for (size_t n = 0; n < size(); n++) img[n] = (T)buffer[n];			//copy the buffer data into the image
  		free(buffer);														//free the buffer array
cfcf8619   David Mayerich   changes to filena...
249
250
251
  	}
  	
  
2e5e3a26   David Mayerich   added 2D convolut...
252
  #ifdef USING_OPENCV
8cd258c6   David Mayerich   updated for 4-cha...
253
  	void from_opencv_888(unsigned char* buffer, size_t width, size_t height) {
2e5e3a26   David Mayerich   added 2D convolut...
254
255
256
257
258
259
260
261
262
263
264
265
266
  		allocate(width, height, 3);
  		T value;
  		size_t i;
  		for (size_t c = 0; c < C(); c++) {								//copy directly
  			for (size_t y = 0; y < Y(); y++) {
  				for (size_t x = 0; x < X(); x++) {
  					i = y * X() * C() + x * C() + (2 - c);
  					value = buffer[i];
  					img[idx(x, y, c)] = value;
  				}
  			}
  		}
  	}
8cd258c6   David Mayerich   updated for 4-cha...
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
  	void from_opencv_8888(unsigned char* buffer, size_t width, size_t height) {
  		//std::cerr << "ERROR - this software is not yet enabled to handle RGBA images (alpha channel)." << std::endl;
  		//exit(1);
  		allocate(width, height, 4);
  		T value;
  		size_t i;
  		for (size_t c = 0; c < C(); c++) {								//copy directly
  			for (size_t y = 0; y < Y(); y++) {
  				for (size_t x = 0; x < X(); x++) {
  					i = y * X() * C() + x * C() + (3 - c);
  					value = buffer[i];
  					img[idx(x, y, c)] = value;
  				}
  			}
  		}
  	}
2e5e3a26   David Mayerich   added 2D convolut...
283
  #endif
55417e1e   David Mayerich   fixed data type l...
284
285
286
287
288
289
290
291
  	//Copy N data points from source to dest, casting while doing so
  	template<typename S, typename D>
  	void type_copy(S* source, D* dest, size_t N) {
  		if (typeid(S) == typeid(D))						//if both types are the same
  			memcpy(dest, source, N * sizeof(S));		//just use a memcpy
  		for (size_t n = 0; n < N; n++)					//otherwise, iterate through each element
  			dest[n] = (D)source[n];							//copy and cast
  	}
c8c976a9   David Mayerich   replaced CImg wit...
292
  	/// Load an image from a file
d32a1854   David Mayerich   added framework f...
293
  	void load(std::string filename){
2e5e3a26   David Mayerich   added 2D convolut...
294
  #ifdef USING_OPENCV
c8c976a9   David Mayerich   replaced CImg wit...
295
  		cv::Mat cvImage = cv::imread(filename, CV_LOAD_IMAGE_UNCHANGED);	//use OpenCV to open the image file
a2bf1d08   David Mayerich   general bug fixes...
296
  		if(!cvImage.data){
814eb271   David Mayerich   fixed problems wi...
297
  			std::cout<<"ERROR stim::image::load() - unable to find image "<<filename<<std::endl;
a2bf1d08   David Mayerich   general bug fixes...
298
299
  			exit(1);
  		}
55417e1e   David Mayerich   fixed data type l...
300
  		int cv_type = cvImage.type();
26da82c8   David Mayerich   finalized stim::g...
301
302
303
304
  		int cols = cvImage.cols;
  		int rows = cvImage.rows;
  		int channels = cvImage.channels();
  		allocate(cols, rows, channels);			//allocate space for the image
55417e1e   David Mayerich   fixed data type l...
305
  		size_t img_bytes = bytes();
a2c755c8   Jiabing Li   start new network...
306
307
308
309
  		unsigned char* cv_ptr = (unsigned char*)cvImage.data;		
  		if (C() == 1)														//if this is a single-color image, just copy the data
  			type_copy<unsigned char, T>(cv_ptr, img, size());
  		//memcpy(img, cv_ptr, bytes());
814eb271   David Mayerich   fixed problems wi...
310
  		if(C() == 3)														//if this is a 3-color image, OpenCV uses BGR interleaving
8cd258c6   David Mayerich   updated for 4-cha...
311
312
313
  			from_opencv_888(cv_ptr, X(), Y());
  		if (C() == 4)
  			from_opencv_8888(cv_ptr, X(), Y());
2e5e3a26   David Mayerich   added 2D convolut...
314
  #else
d9b2b2a8   David Mayerich   enabled basic rea...
315
316
317
318
319
  		stim::filename file(filename);
  		if (file.extension() == "ppm")
  			load_netpbm(filename);
  		else if (file.extension() == "bmp")
  			load_bmp(filename);
2e5e3a26   David Mayerich   added 2D convolut...
320
  #endif
ca99f951   David Mayerich   faster implementa...
321
322
  	}
  
2e5e3a26   David Mayerich   added 2D convolut...
323
  
d32a1854   David Mayerich   added framework f...
324
  
cfcf8619   David Mayerich   changes to filena...
325
326
  	//save a Netpbm file
  	void save_netpbm(std::string filename) {
683f216a   David Mayerich   fixed? Linux comp...
327
  		std::ofstream outfile(filename.c_str(), std::ios::out | std::ios::binary);		//open an output file
cfcf8619   David Mayerich   changes to filena...
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
  		if(!outfile) {
  			std::cout << "Error generating output file in image::save_netpbm()" << std::endl;
  			exit(1);
  		}
  		if (sizeof(T) != 1) {
  			std::cout << "Error in image::save_netpbm() - data type must be 8-bit integer." << std::endl;
  			exit(1);
  		}
  		std::string format;
  		if (channels() == 1) outfile << "P5" << (char)0x0A;			//output P5 if the file is grayscale
  		else if (channels() == 3) outfile << "P6" << (char)0x0A;		//output P6 if the file is color
  		else {
  			std::cout << "Error in image::save_netpbm() - data must be grayscale or RGB." << std::endl;
  			exit(1);
  		}
2e5e3a26   David Mayerich   added 2D convolut...
343
344
345
  		size_t w = width();
  		size_t h = height();
  		outfile << w << " " << h << (char)0x0A;			//save the width and height
cfcf8619   David Mayerich   changes to filena...
346
347
348
349
  		outfile << "255" << (char)0x0A;								//output the maximum value
  		outfile.write((const char*)img, size());			//write the binary data
  		outfile.close();
  	}
d661ab7d   David Mayerich   linux fixes for s...
350
  #ifndef USING_OPENCV
d9b2b2a8   David Mayerich   enabled basic rea...
351
352
353
  	void save_bmp(std::string filename) {
  		stim::save_bmp(filename, (char*)img, width(), height());
  	}
d661ab7d   David Mayerich   linux fixes for s...
354
  #endif
d9b2b2a8   David Mayerich   enabled basic rea...
355
  
d32a1854   David Mayerich   added framework f...
356
357
  	//save a file
  	void save(std::string filename){
2a10ecf4   David Mayerich   updated files and...
358
359
  		stim::filename file(filename);
  		if (file.extension() == "raw" || file.extension() == "") {
bd78d658   David Mayerich   Pranathi's image....
360
  			std::ofstream outfile(filename.c_str(), std::ios::binary);
2a10ecf4   David Mayerich   updated files and...
361
362
363
  			outfile.write((char*)img, sizeof(T) * R[0] * R[1] * R[2]);
  			outfile.close();
  		}
2e5e3a26   David Mayerich   added 2D convolut...
364
  #ifdef USING_OPENCV
c8c976a9   David Mayerich   replaced CImg wit...
365
  		//OpenCV uses an interleaved format, so convert first and then output
fdc7f32c   David Mayerich   Added a coupled w...
366
367
368
369
  		
  		//single-channel image
  		if (C() == 1) {
  			T* buffer = (T*)malloc(bytes());
a2bf1d08   David Mayerich   general bug fixes...
370
  			memcpy(buffer, img, bytes());
fdc7f32c   David Mayerich   Added a coupled w...
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
  			cv::Mat cvImage((int)Y(), (int)X(), cv_type(), buffer);
  			cv::imwrite(filename, cvImage);
  			free(buffer);
  		}
  		//3-channel image
  		else if (C() == 3) {
  			T* buffer = (T*)malloc(bytes());
  			get_interleaved_bgr(buffer);
  			cv::Mat cvImage((int)Y(), (int)X(), cv_type(), buffer);
  			cv::imwrite(filename, cvImage);
  			free(buffer);
  		}
  		//ARGB image
  		else if (C() == 4) {
  			size_t channelsize = X() * Y() * sizeof(T);
  			T* buffer = (T*)malloc(bytes() - channelsize);
9b766f1f   Pavel Govyadinov   completed merge f...
387
  			get_interleaved_bgr(buffer);
fdc7f32c   David Mayerich   Added a coupled w...
388
389
390
391
392
393
394
395
396
  			cv::Mat cvImage((int)Y(), (int)X(), cv_type(), buffer);
  			cv::imwrite(filename, cvImage);
  			free(buffer);
  		}
  		else {
  			throw std::runtime_error("ERROR: number of channels not supported for image saving.");
  		}
  			
  		
2e5e3a26   David Mayerich   added 2D convolut...
397
  #else
d9b2b2a8   David Mayerich   enabled basic rea...
398
399
400
401
402
403
404
405
  		if (file.extension() == "ppm")
  			save_netpbm(filename);
  		else if (file.extension() == "bmp")
  			save_bmp(filename);
  		else {
  			std::cout << "stim::image ERROR: File type not supported without OpenCV. Make sure to link OpenCV and define USING_OPENCV" << std::endl;
  			exit(1);
  		}
2e5e3a26   David Mayerich   added 2D convolut...
406
  #endif
d32a1854   David Mayerich   added framework f...
407
408
  	}
  
e98c381d   David Mayerich   Changed image.h t...
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
  	/// Returns an image cast to the specified format
  	template<typename U>
  	image<U> convert() {
  		
  		image<U> new_image(R[1], R[2], R[0]);					//create a new image with the destination data type
  
  		size_t ni = R[0] * R[1] * R[2];							//calculate the number of data points in the image
  
  		double inmax = (std::numeric_limits<T>::max)();				//get the maximum value for the input image
  		double outmax = (std::numeric_limits<U>::max)();				//get the maximum value for the output image
  		for (size_t i = 0; i < ni; i++) {							//for each pixel in the image
  			if (img[i] > outmax) new_image(i) = outmax;			//if the source pixel is greater than the maximum destination pixel, set the output to maximum
  			else new_image(i) = img[i];							//otherwise, copy the source value and cast it to the destination value		
  		}
  		return new_image;
  	}
  
ca99f951   David Mayerich   faster implementa...
426
427
428
429
430
  	void set_interleaved(T* buffer, size_t width, size_t height, size_t channels){
  		allocate(width, height, channels);
  		memcpy(img, buffer, bytes());
  	}
  
8b7be670   David Mayerich   implemented savin...
431
  	//create an image from an interleaved buffer
814eb271   David Mayerich   fixed problems wi...
432
  	void set_interleaved_rgb(T* buffer, size_t width, size_t height){
ca99f951   David Mayerich   faster implementa...
433
  		set_interleaved(buffer, width, height, 3);
c8c976a9   David Mayerich   replaced CImg wit...
434
  	}
fbf0ab02   David Mayerich   added support for...
435
  
814eb271   David Mayerich   fixed problems wi...
436
437
  	void set_interleaved_bgr(T* buffer, size_t width, size_t height){
  		allocate(width, height, 3);
ca99f951   David Mayerich   faster implementa...
438
439
  		T value;
  		size_t i;
c8c976a9   David Mayerich   replaced CImg wit...
440
441
442
  		for(size_t c = 0; c < C(); c++){								//copy directly
  			for(size_t y = 0; y < Y(); y++){
  				for(size_t x = 0; x < X(); x++){
ca99f951   David Mayerich   faster implementa...
443
444
445
  					i = y * X() * C() + x * C() + (2-c);
  					value = buffer[i];
  					img[idx(x, y, c)] = value;
c8c976a9   David Mayerich   replaced CImg wit...
446
  				}
fbf0ab02   David Mayerich   added support for...
447
448
  			}
  		}
8b7be670   David Mayerich   implemented savin...
449
450
  	}
  
ca99f951   David Mayerich   faster implementa...
451
452
453
454
  	void set_interleaved(T* buffer, size_t width, size_t height){
  		set_interleaved_rgb(buffer, width, height);
  	}
  
9b766f1f   Pavel Govyadinov   completed merge f...
455
  	void get_interleaved_bgr(T* data){
8b7be670   David Mayerich   implemented savin...
456
  		//for each channel
fdc7f32c   David Mayerich   Added a coupled w...
457
458
459
460
461
462
463
464
465
466
  		T* source;
  		if (C() == 3) {
  			source = img;														//if the image has 3 channels, interleave all three
  		}
  		else if (C() == 4) {
  			source = img + X() * Y();											//if the image has 4 channels, skip the alpha channel
  		}
  		else {
  			throw std::runtime_error("ERROR: a BGR image must be 3 or 4 channels");	//throw an error if any other number of channels is provided
  		}
c8c976a9   David Mayerich   replaced CImg wit...
467
468
  		for(size_t y = 0; y < Y(); y++){
  			for(size_t x = 0; x < X(); x++){
fdc7f32c   David Mayerich   Added a coupled w...
469
470
  				for(size_t c = 0; c < 3; c++){
  					data[y * X() * 3 + x * 3 + (2-c)] = source[y*3*R[1] + x*3 + c];
c8c976a9   David Mayerich   replaced CImg wit...
471
472
473
  				}
  			}
  		}
8b7be670   David Mayerich   implemented savin...
474
475
  	}
  
9b766f1f   Pavel Govyadinov   completed merge f...
476
477
478
479
  	void get_interleaved_rgb(T* data){
  		memcpy(data, img, bytes());
  	}
  
904614c3   David Mayerich   added normalizati...
480
481
482
483
484
485
486
487
488
489
490
491
  	//copies data in the given channel order as a non-interleaved image
  	void get_noninterleaved(T* data){
  		//for each channel
  		for(size_t y = 0; y < Y(); y++){
  			for(size_t x = 0; x < X(); x++){
  				for(size_t c = 0; c < C(); c++){
  					data[c * Y() * X() + y * X() + x] = img[idx(x, y, c)];
  				}
  			}
  		}
  	}
  
dbeb83f2   David Mayerich   added separable c...
492
493
  	/// Return an image representing a specified channel
  	/// @param c is the channel to be returned
86dfa80b   David Mayerich   added 2D gaussian...
494
  	image<T> channel(size_t c) const {		
dbeb83f2   David Mayerich   added separable c...
495
  		image<T> r(X(), Y(), 1);				//create a new image
c8c976a9   David Mayerich   replaced CImg wit...
496
497
  		for(size_t x = 0; x < X(); x++){
  			for(size_t y = 0; y < Y(); y++){
8e2fb2b4   David Mayerich   fixed a segmentat...
498
  				r.img[r.idx(x, y, 0)] = img[idx(x, y, c)];
c8c976a9   David Mayerich   replaced CImg wit...
499
500
  			}
  		}
c8c976a9   David Mayerich   replaced CImg wit...
501
  		return r;
dbeb83f2   David Mayerich   added separable c...
502
503
504
  	}
  
  	/// Returns an std::vector containing each channel as a separate image
683f216a   David Mayerich   fixed? Linux comp...
505
506
  	std::vector< image<T> > split() const {
  		std::vector< image<T> > r;			//create an image array
dbeb83f2   David Mayerich   added separable c...
507
  		r.resize(C());						//create images for each channel
41acaf5d   David Mayerich   added a CUDA Gaus...
508
  
dbeb83f2   David Mayerich   added separable c...
509
510
511
512
  		for (size_t c = 0; c < C(); c++) {	//for each channel
  			r[c] = channel(c);				//copy the channel image to the array
  		}
  		return r;
41acaf5d   David Mayerich   added a CUDA Gaus...
513
514
  	}
  
86dfa80b   David Mayerich   added 2D gaussian...
515
  	/// Merge a series of single-channel images into a multi-channel image
683f216a   David Mayerich   fixed? Linux comp...
516
  	void merge(std::vector< image<T> >& list) {
86dfa80b   David Mayerich   added 2D gaussian...
517
518
519
520
521
522
523
  		size_t x = list[0].width();				//calculate the size of the image
  		size_t y = list[0].height();
  		allocate(x, y, list.size());			//re-allocate the image
  		for (size_t c = 0; c < list.size(); c++)		//for each channel
  			set_channel(list[c].channel(0).data(), c);	//insert the channel into the output image
  	}
  
c8c976a9   David Mayerich   replaced CImg wit...
524
525
  	T& operator()(size_t x, size_t y, size_t c = 0){
  		return img[idx(x, y, c)];
3d0b6243   David Mayerich   fixed hsiproc bug...
526
527
  	}
  
08ada8c2   David Mayerich   updated image to ...
528
529
530
531
532
  	/// This function returns a pixel reference based on a 1D index into the image
  	T& operator()(size_t i) {
  		return img[i];
  	}
  
3d0b6243   David Mayerich   fixed hsiproc bug...
533
534
535
  	/// Set all elements in the image to a given scalar value
  
  	/// @param v is the value used to set all values in the image
a8ad9192   David Mayerich   removed warnings ...
536
  	void set_all(T v) {														//set all elements of the image to a given value v
c8c976a9   David Mayerich   replaced CImg wit...
537
  		size_t N = size();
a8ad9192   David Mayerich   removed warnings ...
538
539
540
541
  		for (size_t n = 0; n < N; n++) img[n] = v;
  	}
  	image<T> operator=(T v){
  		set_all(v);
3d0b6243   David Mayerich   fixed hsiproc bug...
542
  		return *this;
3d0b6243   David Mayerich   fixed hsiproc bug...
543
544
  	}
  
a7ed4b2e   David Mayerich   ivote updates
545
546
547
548
549
550
551
552
553
  	/// invert the image, given a specified maximum value (ex. maxval = 255, I' = 255 - I)
  	/*image<T> invert(T maxval) {
  		image<T> result(width(), height(), channels());		//create a new image
  		size_t N = size();									//get the number of elements in the image
  		for (size_t n = 0; n < N; n++)
  			result.data()[n] = maxval - img[n];				//perform the inversion and save the result to the new image
  		return result;
  	}*/
  
5bd6c411   David Mayerich   added contrast st...
554
555
556
557
  	/// Stretch the contrast of the image such that the minimum and maximum intensity match the given values
  	image<T> stretch(T low, T high) {
  		T maxval = maxv();
  		T minval = minv();
5bd6c411   David Mayerich   added contrast st...
558
  		image<T> result = *this;				//create a new image for output
a8ad9192   David Mayerich   removed warnings ...
559
560
561
562
563
  		if (maxval == minval) {					//if the minimum and maximum values are the same, return an image composed of low
  			result = low;
  			return result;
  		}	
  		
5bd6c411   David Mayerich   added contrast st...
564
  		size_t N = size();						//get the number of values in the image
d9b2b2a8   David Mayerich   enabled basic rea...
565
566
  		T range = maxval - minval;			//calculate the current range of the image
  		T desired_range = high - low;		//calculate the desired range of the image
5bd6c411   David Mayerich   added contrast st...
567
  		for (size_t n = 0; n < N; n++) {		//for each element in the image
ec0fdb47   Mahsa Lotfollahi   changes in stretc...
568
  			result.data()[n] = desired_range * (img[n] - minval) / range + low;
5bd6c411   David Mayerich   added contrast st...
569
570
571
572
  		}
  		return result;
  	}
  
70f0baaf   Laila Saadatifard   update the image....
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
  	/// Add a border of width w with the given value around the image
  	/// @param w specifies the total size of the border
  	/// @param T is the pixel value (all channels will be the same)
  	image<T> border(size_t w, T value = 0) {
  		image<T> result(width() + w * 2, height() + w * 2, channels());						//create an output image
  		result = value;														//assign the border value to all pixels in the new image
  		for (size_t y = 0; y < height(); y++) {								//for each pixel in the original image
  			for (size_t x = 0; x < width(); x++) {
  				size_t n = (y + w) * (width() + w * 2) + x + w;				//calculate the index of the corresponding pixel in the result image
  				size_t n0 = idx(x,y);										//calculate the index for this pixel in the original image
  				result.data()[n] = img[n0];									// copy the original image to the result image afer the border area
  			}
  		}
  		return result;
  	}
  
276f9b23   David Mayerich   forced matching t...
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
  	/// Adds curcular padding for the specified number of pixels - in this case replicating the boundary pixels
  	image<T> pad_replicate(size_t p) {
  		image<T> result(width() + p * 2, height() + p * 2, channels());						//create an output image
  		result = 0;
  		//result = value;														//assign the border value to all pixels in the new image
  		for (size_t y = 0; y < height(); y++) {								//for each pixel in the original image
  			for (size_t x = 0; x < width(); x++) {
  				size_t n = (y + p) * (width() + p * 2) + x + p;				//calculate the index of the corresponding pixel in the result image
  				size_t n0 = idx(x, y);										//calculate the index for this pixel in the original image
  				result.data()[n] = img[n0];									// copy the original image to the result image afer the border area
  			}
  		}
  		size_t l = p;
  		size_t r = p + width() - 1;
  		size_t t = p;
  		size_t b = p + height() - 1;
  		for (size_t y = 0; y < p; y++) for (size_t x = l; x <= r; x++) result(x, y) = result(x, t);						//pad the top
  		for (size_t y = b + 1; y < result.height(); y++) for (size_t x = l; x <= r; x++) result(x, y) = result(x, b);	//pad the bottom
  		for (size_t y = t; y <= b; y++) for (size_t x = 0; x < l; x++) result(x, y) = result(l, y);						//pad the left
  		for (size_t y = t; y <= b; y++) for (size_t x = r+1; x < result.width(); x++) result(x, y) = result(r, y);		//pad the right
  		for (size_t y = 0; y < t; y++) for (size_t x = 0; x < l; x++) result(x, y) = result(l, t);						//pad the top left
  		for (size_t y = 0; y < t; y++) for (size_t x = r+1; x < result.width(); x++) result(x, y) = result(r, t);		//pad the top right
  		for (size_t y = b+1; y < result.height(); y++) for (size_t x = 0; x < l; x++) result(x, y) = result(l, b);		//pad the bottom left
  		for (size_t y = b+1; y < result.height(); y++) for (size_t x = r + 1; x < result.width(); x++) result(x, y) = result(r, b);		//pad the bottom right
  		return result;
  	}
  
f186dbda   Tianshu Cheng   header file for b...
616
617
618
619
620
  	/// Copy the given data to the specified channel
  
  	/// @param c is the channel number that the data will be copied to
  	/// @param buffer is a pointer to the image to be copied to channel c
  
c8c976a9   David Mayerich   replaced CImg wit...
621
  	void set_channel(T* buffer, size_t c){
6d30a707   Tianshu Cheng   add cuda/array_ad...
622
  
c8c976a9   David Mayerich   replaced CImg wit...
623
624
625
  		size_t x, y;
  		for(y = 0; y < Y(); y++){
  			for(x = 0; x < X(); x++){
2e5e3a26   David Mayerich   added 2D convolut...
626
  				img[idx(x, y, c)] = buffer[y * X() + x];
c8c976a9   David Mayerich   replaced CImg wit...
627
628
  			}
  		}
6d30a707   Tianshu Cheng   add cuda/array_ad...
629
630
  	}
  
86dfa80b   David Mayerich   added 2D gaussian...
631
  	size_t channels() const{
c8c976a9   David Mayerich   replaced CImg wit...
632
  		return C();
8b7be670   David Mayerich   implemented savin...
633
634
  	}
  
86dfa80b   David Mayerich   added 2D gaussian...
635
  	size_t width() const{
c8c976a9   David Mayerich   replaced CImg wit...
636
  		return X();
8b7be670   David Mayerich   implemented savin...
637
638
  	}
  
86dfa80b   David Mayerich   added 2D gaussian...
639
  	size_t height() const{
c8c976a9   David Mayerich   replaced CImg wit...
640
  		return Y();
3d0b6243   David Mayerich   fixed hsiproc bug...
641
642
  	}
  
6d30a707   Tianshu Cheng   add cuda/array_ad...
643
  	T* data(){
c8c976a9   David Mayerich   replaced CImg wit...
644
  		return img;
6d30a707   Tianshu Cheng   add cuda/array_ad...
645
646
  	}
  
7b3948ab   David Mayerich   added support for...
647
  	//returns the size (number of values) of the image
c8c976a9   David Mayerich   replaced CImg wit...
648
  	size_t size(){ return C() * X() * Y(); }
d32a1854   David Mayerich   added framework f...
649
  
faef7718   David Mayerich   updates to stim::...
650
  	/// Returns the number of nonzero values
9d3ba0b1   David Mayerich   added stim::hsi a...
651
  	size_t nnz(){
faef7718   David Mayerich   updates to stim::...
652
  
c8c976a9   David Mayerich   replaced CImg wit...
653
  		size_t N = X() * Y() * C();
faef7718   David Mayerich   updates to stim::...
654
  
c8c976a9   David Mayerich   replaced CImg wit...
655
656
657
  		size_t nz = 0;
  		for(size_t n = 0; n < N; n++)
  			if(img[n] != 0) nz++;
faef7718   David Mayerich   updates to stim::...
658
  
c8c976a9   David Mayerich   replaced CImg wit...
659
  		return nz;	//return the number of nonzero pixels
faef7718   David Mayerich   updates to stim::...
660
661
662
663
  
  	}
  
  	//this function returns indices of pixels that have nonzero values
9d3ba0b1   David Mayerich   added stim::hsi a...
664
  	std::vector<size_t> sparse_idx(){
faef7718   David Mayerich   updates to stim::...
665
  
c8c976a9   David Mayerich   replaced CImg wit...
666
  		std::vector<size_t> s;				//allocate an array
faef7718   David Mayerich   updates to stim::...
667
668
  		s.resize(nnz());					//allocate space in the array
  
c8c976a9   David Mayerich   replaced CImg wit...
669
670
  		size_t N = size();
  		//size_t C = channels();
faef7718   David Mayerich   updates to stim::...
671
  
c8c976a9   David Mayerich   replaced CImg wit...
672
  		//T* ptr = img.data();				//get a pointer to the image data
faef7718   David Mayerich   updates to stim::...
673
  
9d3ba0b1   David Mayerich   added stim::hsi a...
674
  		size_t i = 0;
c8c976a9   David Mayerich   replaced CImg wit...
675
676
677
678
  		for(size_t n = 0; n < N; n++){
  			if(img[n] != 0){
  				s[i] = n;
  				i++;
faef7718   David Mayerich   updates to stim::...
679
680
681
682
683
  			}
  		}
  
  		return s;			//return the index list
  	}
6d30a707   Tianshu Cheng   add cuda/array_ad...
684
  
1a224b6a   David Mayerich   fixed linux compa...
685
  
6d30a707   Tianshu Cheng   add cuda/array_ad...
686
  	/// Returns the maximum pixel value in the image
96f9b10f   Laila Saadatifard   change the header...
687
  	T maxv(){
c8c976a9   David Mayerich   replaced CImg wit...
688
689
  		T max_val = img[0];				//initialize the maximum value to the first one
  		size_t N = size();	//get the number of pixels
6d30a707   Tianshu Cheng   add cuda/array_ad...
690
  
c8c976a9   David Mayerich   replaced CImg wit...
691
  		for (size_t n=0; n<N; n++){		//for every value
6d30a707   Tianshu Cheng   add cuda/array_ad...
692
  
c8c976a9   David Mayerich   replaced CImg wit...
693
694
  			if (img[n] > max_val){			//if the value is higher than the current max
  				max_val = img[n];
1a224b6a   David Mayerich   fixed linux compa...
695
  			}
6d30a707   Tianshu Cheng   add cuda/array_ad...
696
  		}
1a224b6a   David Mayerich   fixed linux compa...
697
  		return max_val;
6d30a707   Tianshu Cheng   add cuda/array_ad...
698
  	}
d32a1854   David Mayerich   added framework f...
699
  
c8c976a9   David Mayerich   replaced CImg wit...
700
  	/// Returns the maximum pixel value in the image
96f9b10f   Laila Saadatifard   change the header...
701
  	T minv(){
c8c976a9   David Mayerich   replaced CImg wit...
702
703
  		T min_val = img[0];				//initialize the maximum value to the first one
  		size_t N = size();	//get the number of pixels
6d30a707   Tianshu Cheng   add cuda/array_ad...
704
  
c8c976a9   David Mayerich   replaced CImg wit...
705
706
707
  		for (size_t n=0; n<N; n++){		//for every value
  			if (img[n] < min_val){			//if the value is higher than the current max
  				min_val = img[n];
1a224b6a   David Mayerich   fixed linux compa...
708
  			}
6d30a707   Tianshu Cheng   add cuda/array_ad...
709
710
  		}
  
1a224b6a   David Mayerich   fixed linux compa...
711
  		return min_val;
6d30a707   Tianshu Cheng   add cuda/array_ad...
712
713
  	}
  
a2bf1d08   David Mayerich   general bug fixes...
714
715
716
717
718
719
  	/// Invert an image by calculating I1 = alpha - I0, where alpha is the maximum image value
  	image<T> invert(T white_val){
  		size_t N = size();						//calculate the total number of values in the image
  		image<T> r(X(), Y(), C());				//allocate space for the resulting image
  		for(size_t n = 0; n < N; n++)
  			r.img[n] = white_val - img[n];		//perform the inversion
1a224b6a   David Mayerich   fixed linux compa...
720
  
a2bf1d08   David Mayerich   general bug fixes...
721
722
  		return r;								//return the inverted image
  	}
a2bf1d08   David Mayerich   general bug fixes...
723
  
3c17c706   David Mayerich   added a crop func...
724
725
726
727
728
729
730
731
732
733
734
735
736
737
  	image<T> crop(size_t x0, size_t y0, size_t w, size_t h){
  		image<T> result(w, h, C());								//create the output cropped image
  
  		size_t srci;
  		size_t dsti;
  		size_t line_bytes = w * C();							//calculate the number of bytes in a line
  		for (size_t yi = 0; yi < h; yi++) {						//for each row in the cropped image
  			srci = (y0 + yi) * X() * C() + x0 * C();			//calculate the source index
  			dsti = yi * w * C();								//calculate the destination index
  			memcpy(&result.img[dsti], &img[srci], line_bytes);	//copy the data
  		}
  		return result;
  	}
  
e702aa4e   David Mayerich   implemented array...
738
  	//crop regions given by an array of 1D index values
276f9b23   David Mayerich   forced matching t...
739
  	std::vector< image<T> > crop_idx(size_t w, size_t h, std::vector<size_t> idx) {
0012d210   David Mayerich   fixed compiler er...
740
  		std::vector< image<T> > result(idx.size());										//create an array of image files to return
e702aa4e   David Mayerich   implemented array...
741
742
743
744
745
746
747
748
749
750
  		for (size_t i = 0; i < idx.size(); i++) {										//for each specified index point
  			size_t y = idx[i] / X();													//calculate the y coordinate from the 1D index (center of ROI)
  			size_t x = idx[i] - y * X();												//calculate the x coordinate (center of ROI)
  			y -= w / 2;																	//update x and y values to reflect the lower corner of the ROI
  			x -= h / 2;
  			result[i] = crop(x, y, w, h);												//get the cropped image and store it in the result array
  		}
  		return result;
  	}
  
66304f4f   David Mayerich   added an addition...
751
752
753
754
755
756
757
758
759
  	//operator functions
  	image<T> operator+(image<T> rhs) {
  		size_t N = size();						//calculate the total number of values in the image
  		image<T> r(X(), Y(), C());				//allocate space for the resulting image
  		for (size_t n = 0; n < N; n++)
  			r.img[n] = img[n] + rhs.img[n];		//perform the inversion
  		return r;								//return the inverted image
  	}
  
6d30a707   Tianshu Cheng   add cuda/array_ad...
760
  	image<T> srgb2lab(){
c8c976a9   David Mayerich   replaced CImg wit...
761
762
  		std::cout<<"ERROR stim::image::srgb2lab - function has been broken, re-implement."<<std::endl;
  		exit(1);
6d30a707   Tianshu Cheng   add cuda/array_ad...
763
764
765
  	}
  
  	image<T> convolve2(image<T> mask){
c8c976a9   David Mayerich   replaced CImg wit...
766
767
  		std::cout<<"ERROR stim::image::convolve2 - function has been broken, and shouldn't really be in here."<<std::endl;
  		exit(1);
6d30a707   Tianshu Cheng   add cuda/array_ad...
768
  	}
d32a1854   David Mayerich   added framework f...
769
770
  
  
f186dbda   Tianshu Cheng   header file for b...
771
  	image<T> rotate(float angle, float cx, float cy){
c8c976a9   David Mayerich   replaced CImg wit...
772
773
  		std::cout<<"ERROR stim::image::rotate - function has been broken, and shouldn't really be in here."<<std::endl;
  		exit(1);
f186dbda   Tianshu Cheng   header file for b...
774
  	}
945ee13c   Laila Saadatifard   the get_list func...
775
  
945ee13c   Laila Saadatifard   the get_list func...
776
777
  	// leila's code for non_interleaving data in 3D
  	//create an data set from an interleaved buffer
9d3ba0b1   David Mayerich   added stim::hsi a...
778
  	void set_interleaved3(T* buffer, size_t width, size_t height, size_t depth, size_t channels = 3){
c8c976a9   David Mayerich   replaced CImg wit...
779
780
  		std::cout<<"ERROR stim::image::set_interleaved3 - stim::image no longer supports 3D images."<<std::endl;
  		exit(1);
945ee13c   Laila Saadatifard   the get_list func...
781
  	}
1a224b6a   David Mayerich   fixed linux compa...
782
  
dbeb83f2   David Mayerich   added separable c...
783
  	/// Casting operator, casts every value in an image to a different data type V
2e5e3a26   David Mayerich   added 2D convolut...
784
785
  	template<typename V>
  	operator image<V>() {
dbeb83f2   David Mayerich   added separable c...
786
787
788
  		image<V> r(X(), Y(), C());					//create a new image
  		std::copy(img, img + size(), r.data());		//copy and cast the data
  		return r;									//return the new image
2e5e3a26   David Mayerich   added 2D convolut...
789
790
  	}
  
d32a1854   David Mayerich   added framework f...
791
792
  };
  
7b3948ab   David Mayerich   added support for...
793
794
  };		//end namespace stim
  
d32a1854   David Mayerich   added framework f...
795
796
  
  #endif