Blame view

image/image.h 2 KB
d32a1854   David Mayerich   added framework f...
1
2
  #ifndef STIM_IMAGE_H
  #define STIM_IMAGE_H
32c433c7   Pavel Govyadinov   recovered the maj...
3
  #define cimg_use_jpeg         //necessary for JPG files
fbf0ab02   David Mayerich   added support for...
4
5
  #include "CImg.h"
  
d32a1854   David Mayerich   added framework f...
6
  #include <iostream>
7b3948ab   David Mayerich   added support for...
7
  namespace stim{
d32a1854   David Mayerich   added framework f...
8
9
10
11
12
13
14
15
  //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
  
  //currently this interface uses CImg
  //	T = data type (usually unsigned char)
  template <class T>
  class image{
  
7b3948ab   David Mayerich   added support for...
16
  	cimg_library::CImg<T> img;
d32a1854   David Mayerich   added framework f...
17
18
19
  
  public:
  
7b3948ab   David Mayerich   added support for...
20
21
22
23
24
25
26
27
28
  	//default constructor
  	image(){
  	}
  
  	//constructor (load an image file)
  	image(std::string filename){
  		img.load(filename.c_str());
  	}
  
d32a1854   David Mayerich   added framework f...
29
30
  	//Load an image from a file
  	void load(std::string filename){
7b3948ab   David Mayerich   added support for...
31
  		img.load(filename.c_str());
d32a1854   David Mayerich   added framework f...
32
33
34
35
  	}
  
  	//save a file
  	void save(std::string filename){
7b3948ab   David Mayerich   added support for...
36
  		img.save(filename.c_str());
d32a1854   David Mayerich   added framework f...
37
38
  	}
  
8b7be670   David Mayerich   implemented savin...
39
  	//create an image from an interleaved buffer
fbf0ab02   David Mayerich   added support for...
40
41
42
43
44
45
46
47
48
49
  	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];
  			}
  		}
d32a1854   David Mayerich   added framework f...
50
  
7b3948ab   David Mayerich   added support for...
51
52
53
  		img = cimg_library::CImg<unsigned char>(non_interleaved, width, height, 1, channels);
  	}
  
32504c2c   David Mayerich   fixed some functi...
54
  	//fills an allocated region of memory with non-interleaved data
8b7be670   David Mayerich   implemented savin...
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
  	void data_noninterleaved(T* data){
  		memcpy(data, img.data(), sizeof(T) * size());
  	}
  
  	void data_interleaved(T* data){
  
  		unsigned int C = channels();
  		unsigned int X = size();
  
  		T* ptr = img.data();
  
  		//for each channel
  		for(unsigned int c = 0; c < C; c++)
  			//convert each pixel
  			for(unsigned int x = 0; x < X; x++)
  				data[x * C + c] = ptr[c * X + x];
  	}
  
  	unsigned int channels(){
  		return (unsigned int)img.spectrum();
  	}
  
  	unsigned int width(){
  		return img.width();
  	}
  
  	unsigned int height(){
  		return img.height();
7b3948ab   David Mayerich   added support for...
83
  	}
d32a1854   David Mayerich   added framework f...
84
  
7b3948ab   David Mayerich   added support for...
85
86
87
  	//returns the size (number of values) of the image
  	unsigned long size(){
  		return img.size();
d32a1854   David Mayerich   added framework f...
88
89
90
91
92
93
94
95
  	}
  
  	
  
  
  
  };
  
7b3948ab   David Mayerich   added support for...
96
97
  };		//end namespace stim
  
d32a1854   David Mayerich   added framework f...
98
99
  
  #endif