]> git.tdb.fi Git - libs/gl.git/blob - source/core/texturecube.cpp
Guard against hitting the end of the array in get_gl_type
[libs/gl.git] / source / core / texturecube.cpp
1 #include <msp/datafile/collection.h>
2 #include <msp/gl/extensions/arb_texture_cube_map.h>
3 #include <msp/gl/extensions/arb_texture_storage.h>
4 #include <msp/io/memory.h>
5 #include <msp/strings/format.h>
6 #include "bindable.h"
7 #include "error.h"
8 #include "pixelstore.h"
9 #include "texturecube.h"
10
11 using namespace std;
12
13 namespace Msp {
14 namespace GL {
15
16 TextureCubeFace TextureCube::face_order[6] =
17 {
18         POSITIVE_X,
19         NEGATIVE_X,
20         POSITIVE_Y,
21         NEGATIVE_Y,
22         POSITIVE_Z,
23         NEGATIVE_Z
24 };
25
26 Vector3 TextureCube::directions[6] =
27 {
28         Vector3(1, 0, 0),
29         Vector3(-1, 0, 0),
30         Vector3(0, 1, 0),
31         Vector3(0, -1, 0),
32         Vector3(0, 0, 1),
33         Vector3(0, 0, -1)
34 };
35
36 unsigned TextureCube::orientations[12] =
37 {
38         5, 3,
39         4, 3,
40         0, 4,
41         0, 5,
42         0, 3,
43         1, 3
44 };
45
46 TextureCube::TextureCube():
47         Texture(GL_TEXTURE_CUBE_MAP),
48         size(0),
49         allocated(0)
50 {
51         static Require _req(ARB_texture_cube_map);
52 }
53
54 void TextureCube::storage(PixelFormat fmt, unsigned sz, unsigned lv)
55 {
56         if(size>0)
57                 throw invalid_operation("TextureCube::storage");
58         if(sz==0)
59                 throw invalid_argument("TextureCube::storage");
60
61         set_format(fmt);
62         size = sz;
63         levels = get_n_levels();
64         if(lv>0)
65                 levels = min(levels, lv);
66 }
67
68 void TextureCube::allocate(unsigned level)
69 {
70         if(size==0)
71                 throw invalid_operation("TextureCube::allocate");
72         if(level>=levels)
73                 throw invalid_argument("TextureCube::allocate");
74         if(allocated&(64<<level))
75                 return;
76
77         if(ARB_texture_storage)
78         {
79                 BindRestore _bind(this);
80                 glTexStorage2D(target, levels, storage_fmt, size, size);
81                 apply_swizzle();
82                 allocated |= (64<<levels)-1;
83         }
84         else
85         {
86                 for(unsigned i=0; i<6; ++i)
87                         image(enumerate_faces(i), level, 0);
88         }
89 }
90
91 void TextureCube::image(TextureCubeFace face, unsigned level, const void *data)
92 {
93         if(size==0)
94                 throw invalid_operation("TextureCube::image");
95
96         unsigned s = get_level_size(level);
97         if(s==0)
98                 throw out_of_range("TextureCube::image");
99
100         if(ARB_texture_storage)
101                 return sub_image(face, level, 0, 0, s, s, data);
102
103         BindRestore _bind(this);
104
105         if(!allocated)
106         {
107                 glTexParameteri(target, GL_TEXTURE_MAX_LEVEL, levels-1);
108                 apply_swizzle();
109         }
110
111         PixelComponents comp = get_components(storage_fmt);
112         GLenum type = get_gl_type(get_component_type(storage_fmt));
113         glTexImage2D(face, level, storage_fmt, s, s, 0, comp, type, data);
114
115         if(level==0)
116         {
117                 allocated |= 1<<get_face_index(face);
118                 if((allocated&63)==63)
119                 {
120                         allocated |= 64;
121                         if(auto_gen_mipmap)
122                         {
123                                 generate_mipmap();
124                                 allocated |= (64<<levels)-1;
125                         }
126                 }
127         }
128         else if(!(allocated&(64<<level)))
129         {
130                 for(unsigned i=0; i<6; ++i)
131                         if(enumerate_faces(i)!=face)
132                                 glTexImage2D(enumerate_faces(i), level, storage_fmt, s, s, 0, comp, type, 0);
133
134                 allocated |= 64<<level;
135         }
136 }
137
138 void TextureCube::image(TextureCubeFace face, unsigned level, PixelComponents comp, DataType type, const void *data)
139 {
140         if(comp!=get_components(format) || type!=get_component_type(format))
141                 throw incompatible_data("TextureCube::image");
142         image(face, level, data);
143 }
144
145 void TextureCube::sub_image(TextureCubeFace face, unsigned level, int x, int y, unsigned wd, unsigned ht, const void *data)
146 {
147         if(size==0)
148                 throw invalid_operation("TextureCube::sub_image");
149
150         BindRestore _bind(this);
151         allocate(level);
152
153         PixelComponents comp = get_components(storage_fmt);
154         GLenum type = get_gl_type(get_component_type(storage_fmt));
155         glTexSubImage2D(face, level, x, y, wd, ht, comp, type, data);
156
157         if(auto_gen_mipmap && level==0)
158                 generate_mipmap();
159 }
160
161 void TextureCube::sub_image(TextureCubeFace face, unsigned level, int x, int y, unsigned wd, unsigned ht, PixelComponents comp, DataType type, const void *data)
162 {
163         if(comp!=get_components(format) || type!=get_component_type(format))
164                 throw incompatible_data("TextureCube::subimage");
165         sub_image(face, level, x, y, wd, ht, data);
166 }
167
168 void TextureCube::image(TextureCubeFace face, const Graphics::Image &img)
169 {
170         unsigned w = img.get_width();
171         unsigned h = img.get_height();
172         PixelFormat fmt = pixelformat_from_image(img);
173         if(size==0)
174         {
175                 if(w!=h)
176                         throw incompatible_data("TextureCube::image");
177
178                 storage(make_pixelformat(get_components(fmt), get_component_type(fmt), use_srgb_format), w);
179         }
180         else if(w!=size || h!=size)
181                 throw incompatible_data("TextureCube::image");
182
183         PixelStore pstore = PixelStore::from_image(img);
184         BindRestore _bind_ps(pstore);
185
186         image(face, 0, img.get_pixels());
187 }
188
189 void TextureCube::image(TextureCubeFace face, const Graphics::Image &img, bool)
190 {
191         image(face, img);
192 }
193
194 void TextureCube::image(const Graphics::Image &img, unsigned lv)
195 {
196         unsigned w = img.get_width();
197         unsigned h = img.get_height();
198
199         if(h!=w*6)
200                 throw incompatible_data("TextureCube::image");
201         h /= 6;
202
203         PixelFormat fmt = pixelformat_from_image(img);
204         if(size==0)
205                 storage(make_pixelformat(get_components(fmt), get_component_type(fmt), use_srgb_format), w, lv);
206         else if(w!=size || h!=size)
207                 throw incompatible_data("TextureCube::image");
208
209         PixelStore pstore = PixelStore::from_image(img);
210         BindRestore _bind_ps(pstore);
211
212         const char *pixels = reinterpret_cast<const char *>(img.get_pixels());
213         unsigned face_size = img.get_stride()*size;
214         for(unsigned i=0; i<6; ++i)
215                 image(enumerate_faces(i), 0, pixels+i*face_size);
216 }
217
218 unsigned TextureCube::get_n_levels() const
219 {
220         unsigned n = 0;
221         for(unsigned s=size; s; s>>=1, ++n) ;
222         return n;
223 }
224
225 unsigned TextureCube::get_level_size(unsigned level) const
226 {
227         return size>>level;
228 }
229
230 TextureCubeFace TextureCube::enumerate_faces(unsigned i)
231 {
232         if(i>=6)
233                 throw out_of_range("TextureCube::enumerate_faces");
234         return face_order[i];
235 }
236
237 unsigned TextureCube::get_face_index(TextureCubeFace face)
238 {
239         switch(face)
240         {
241         case POSITIVE_X: return 0;
242         case NEGATIVE_X: return 1;
243         case POSITIVE_Y: return 2;
244         case NEGATIVE_Y: return 3;
245         case POSITIVE_Z: return 4;
246         case NEGATIVE_Z: return 5;
247         default: throw invalid_argument("TextureCube::get_face_index");
248         }
249 }
250
251 const Vector3 &TextureCube::get_face_direction(TextureCubeFace face)
252 {
253         return directions[get_face_index(face)];
254 }
255
256 const Vector3 &TextureCube::get_s_direction(TextureCubeFace face)
257 {
258         return directions[orientations[get_face_index(face)*2]];
259 }
260
261 const Vector3 &TextureCube::get_t_direction(TextureCubeFace face)
262 {
263         return directions[orientations[get_face_index(face)*2+1]];
264 }
265
266 Vector3 TextureCube::get_texel_direction(TextureCubeFace face, unsigned u, unsigned v)
267 {
268         float s = (u+0.5f)*2.0f/size-1.0f;
269         float t = (v+0.5f)*2.0f/size-1.0f;
270         const Vector3 &fv = get_face_direction(face);
271         const Vector3 &sv = get_s_direction(face);
272         const Vector3 &tv = get_t_direction(face);
273         return fv+s*sv+t*tv;
274 }
275
276 UInt64 TextureCube::get_data_size() const
277 {
278         return id ? size*size*6*get_pixel_size(storage_fmt) : 0;
279 }
280
281
282 TextureCube::Loader::Loader(TextureCube &t):
283         DataFile::DerivedObjectLoader<TextureCube, Texture::Loader>(t)
284 {
285         init();
286 }
287
288 TextureCube::Loader::Loader(TextureCube &t, Collection &c):
289         DataFile::DerivedObjectLoader<TextureCube, Texture::Loader>(t, c)
290 {
291         init();
292 }
293
294 void TextureCube::Loader::init()
295 {
296         add("external_image", &Loader::external_image);
297         add("image_data", &Loader::image_data);
298         add("raw_data", &Loader::raw_data);
299         add("storage", &Loader::storage);
300         add("storage", &Loader::storage_levels);
301 }
302
303 void TextureCube::Loader::external_image(TextureCubeFace face, const string &fn)
304 {
305         Graphics::Image img;
306         RefPtr<IO::Seekable> io = get_collection().open_raw(fn);
307         img.load_io(*io);
308
309         obj.image(face, img);
310 }
311
312 void TextureCube::Loader::image_data(TextureCubeFace face, const string &data)
313 {
314         Graphics::Image img;
315         IO::Memory mem(data.data(), data.size());
316         img.load_io(mem);
317
318         obj.image(face, img);
319 }
320
321 void TextureCube::Loader::raw_data(TextureCubeFace face, const string &data)
322 {
323         obj.image(face, 0, data.data());
324 }
325
326 void TextureCube::Loader::storage(PixelFormat fmt, unsigned s)
327 {
328         obj.storage(fmt, s);
329 }
330
331 void TextureCube::Loader::storage_levels(PixelFormat fmt, unsigned s, unsigned l)
332 {
333         obj.storage(fmt, s, l);
334 }
335
336
337 void operator>>(const LexicalConverter &conv, TextureCubeFace &face)
338 {
339         const string &str = conv.get();
340         if(str=="POSITIVE_X")
341                 face = POSITIVE_X;
342         else if(str=="NEGATIVE_X")
343                 face = NEGATIVE_X;
344         else if(str=="POSITIVE_Y")
345                 face = POSITIVE_Y;
346         else if(str=="NEGATIVE_Y")
347                 face = NEGATIVE_Y;
348         else if(str=="POSITIVE_Z")
349                 face = POSITIVE_Z;
350         else if(str=="NEGATIVE_Z")
351                 face = NEGATIVE_Z;
352         else
353                 throw lexical_error(format("conversion of '%s' to TextureCubeFace", str));
354 }
355
356 } // namespace GL
357 } // namespace Msp