/
IMG_webp.c
executable file
·320 lines (270 loc) · 8.75 KB
1
/*
2
SDL_image: An example image loading library for use with SDL
3
Copyright (C) 1997-2017 Sam Lantinga <slouken@libsdl.org>
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
This software is provided 'as-is', without any express or implied
warranty. In no event will the authors be held liable for any damages
arising from the use of this software.
Permission is granted to anyone to use this software for any purpose,
including commercial applications, and to alter it and redistribute it
freely, subject to the following restrictions:
1. The origin of this software must not be misrepresented; you must not
claim that you wrote the original software. If you use this software
in a product, an acknowledgment in the product documentation would be
appreciated but is not required.
2. Altered source versions must be plainly marked as such, and must not be
misrepresented as being the original software.
3. This notice may not be removed or altered from any source distribution.
20
21
22
23
24
25
26
27
28
29
*/
/* This is a WEBP image file loading framework */
#include "SDL_image.h"
#ifdef LOAD_WEBP
/*=============================================================================
File: SDL_webp.c
30
31
Purpose: A WEBP loader for the SDL library
Revision:
32
33
34
35
36
37
38
39
40
41
42
43
44
Created by: Michael Bonfils (Murlock) (26 November 2011)
murlock42@gmail.com
=============================================================================*/
#include "SDL_endian.h"
#ifdef macintosh
#define MACOS
#endif
#include <webp/decode.h>
static struct {
45
46
int loaded;
void *handle;
47
48
49
VP8StatusCode (*webp_get_features_internal) (const uint8_t *data, size_t data_size, WebPBitstreamFeatures* features, int decoder_abi_version);
uint8_t* (*webp_decode_rgb_into) (const uint8_t* data, size_t data_size, uint8_t* output_buffer, size_t output_buffer_size, int output_stride);
uint8_t* (*webp_decode_rgba_into) (const uint8_t* data, size_t data_size, uint8_t* output_buffer, size_t output_buffer_size, int output_stride);
50
51
52
53
54
} lib;
#ifdef LOAD_WEBP_DYNAMIC
int IMG_InitWEBP()
{
55
56
57
58
59
if ( lib.loaded == 0 ) {
lib.handle = SDL_LoadObject(LOAD_WEBP_DYNAMIC);
if ( lib.handle == NULL ) {
return -1;
}
60
61
lib.webp_get_features_internal =
62
( VP8StatusCode (*) (const uint8_t *, size_t, WebPBitstreamFeatures*, int) )
63
64
65
66
67
68
69
SDL_LoadFunction(lib.handle, "WebPGetFeaturesInternal" );
if ( lib.webp_get_features_internal == NULL ) {
SDL_UnloadObject(lib.handle);
return -1;
}
lib.webp_decode_rgb_into =
70
( uint8_t* (*) (const uint8_t*, size_t, uint8_t*, size_t, int ) )
71
72
73
74
75
76
77
SDL_LoadFunction(lib.handle, "WebPDecodeRGBInto" );
if ( lib.webp_decode_rgb_into == NULL ) {
SDL_UnloadObject(lib.handle);
return -1;
}
lib.webp_decode_rgba_into =
78
( uint8_t* (*) (const uint8_t*, size_t, uint8_t*, size_t, int ) )
79
80
81
82
83
84
85
86
87
SDL_LoadFunction(lib.handle, "WebPDecodeRGBAInto" );
if ( lib.webp_decode_rgba_into == NULL ) {
SDL_UnloadObject(lib.handle);
return -1;
}
}
++lib.loaded;
return 0;
88
89
90
}
void IMG_QuitWEBP()
{
91
92
93
94
95
96
97
if ( lib.loaded == 0 ) {
return;
}
if ( lib.loaded == 1 ) {
SDL_UnloadObject(lib.handle);
}
--lib.loaded;
98
99
100
101
}
#else
int IMG_InitWEBP()
{
102
if ( lib.loaded == 0 ) {
103
104
105
106
107
108
109
110
111
112
#ifdef __MACOSX__
extern VP8StatusCode WebPGetFeaturesInternal(const uint8_t*, size_t, WebPBitstreamFeatures*, int) __attribute__((weak_import));
if ( WebPGetFeaturesInternal == NULL )
{
/* Missing weakly linked framework */
IMG_SetError("Missing webp.framework");
return -1;
}
#endif // __MACOSX__
113
114
115
116
117
118
119
lib.webp_get_features_internal = WebPGetFeaturesInternal;
lib.webp_decode_rgb_into = WebPDecodeRGBInto;
lib.webp_decode_rgba_into = WebPDecodeRGBAInto;
}
++lib.loaded;
return 0;
120
121
122
}
void IMG_QuitWEBP()
{
123
124
125
126
127
128
if ( lib.loaded == 0 ) {
return;
}
if ( lib.loaded == 1 ) {
}
--lib.loaded;
129
130
131
132
}
#endif /* LOAD_WEBP_DYNAMIC */
static int webp_getinfo( SDL_RWops *src, int *datasize ) {
133
134
135
136
137
138
139
140
141
142
Sint64 start;
int is_WEBP;
Uint8 magic[20];
if ( !src )
return 0;
start = SDL_RWtell(src);
is_WEBP = 0;
if ( SDL_RWread(src, magic, 1, sizeof(magic)) == sizeof(magic) ) {
if ( magic[ 0] == 'R' &&
143
144
145
magic[ 1] == 'I' &&
magic[ 2] == 'F' &&
magic[ 3] == 'F' &&
146
147
148
149
150
151
152
magic[ 8] == 'W' &&
magic[ 9] == 'E' &&
magic[10] == 'B' &&
magic[11] == 'P' &&
magic[12] == 'V' &&
magic[13] == 'P' &&
magic[14] == '8' &&
153
#if WEBP_DECODER_ABI_VERSION < 0x0003 /* old versions don't support WEBPVP8X and WEBPVP8L */
154
magic[15] == ' ') {
155
#else
156
(magic[15] == ' ' || magic[15] == 'X' || magic[15] == 'L')) {
157
#endif
158
159
is_WEBP = 1;
if ( datasize ) {
160
*datasize = (int)SDL_RWseek(src, 0, RW_SEEK_END);
161
162
163
164
165
}
}
}
SDL_RWseek(src, start, RW_SEEK_SET);
return(is_WEBP);
166
167
168
169
170
}
/* See if an image is contained in a data source */
int IMG_isWEBP(SDL_RWops *src)
{
171
return webp_getinfo( src, NULL );
172
173
174
175
}
SDL_Surface *IMG_LoadWEBP_RW(SDL_RWops *src)
{
176
177
178
179
180
181
182
183
184
185
Sint64 start;
const char *error = NULL;
SDL_Surface *volatile surface = NULL;
Uint32 Rmask;
Uint32 Gmask;
Uint32 Bmask;
Uint32 Amask;
WebPBitstreamFeatures features;
int raw_data_size;
uint8_t *raw_data = NULL;
186
int r;
187
188
189
190
191
192
193
194
195
uint8_t *ret;
if ( !src ) {
/* The error message has been set in SDL_RWFromFile */
return NULL;
}
start = SDL_RWtell(src);
196
if ( (IMG_Init(IMG_INIT_WEBP) & IMG_INIT_WEBP) == 0 ) {
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
goto error;
}
raw_data_size = -1;
if ( !webp_getinfo( src, &raw_data_size ) ) {
error = "Invalid WEBP";
goto error;
}
// seek to start of file
SDL_RWseek(src, 0, RW_SEEK_SET );
raw_data = (uint8_t*) SDL_malloc( raw_data_size );
if ( raw_data == NULL ) {
error = "Failed to allocate enought buffer for WEBP";
goto error;
}
r = SDL_RWread(src, raw_data, 1, raw_data_size );
if ( r != raw_data_size ) {
error = "Failed to read WEBP";
goto error;
}
221
#if 0
222
223
224
225
226
227
// extract size of picture, not interesting since we don't know about alpha channel
int width = -1, height = -1;
if ( !WebPGetInfo( raw_data, raw_data_size, &width, &height ) ) {
printf("WebPGetInfo has failed\n" );
return NULL;
}
228
229
#endif
230
231
232
233
if ( lib.webp_get_features_internal( raw_data, raw_data_size, &features, WEBP_DECODER_ABI_VERSION ) != VP8_STATUS_OK ) {
error = "WebPGetFeatures has failed";
goto error;
}
234
235
/* Check if it's ok !*/
236
#if SDL_BYTEORDER == SDL_LIL_ENDIAN
237
238
239
240
Rmask = 0x000000FF;
Gmask = 0x0000FF00;
Bmask = 0x00FF0000;
Amask = (features.has_alpha) ? 0xFF000000 : 0;
241
#else
242
243
244
245
246
247
248
{
int s = (features.has_alpha) ? 0 : 8;
Rmask = 0xFF000000 >> s;
Gmask = 0x00FF0000 >> s;
Bmask = 0x0000FF00 >> s;
Amask = 0x000000FF >> s;
}
249
#endif
250
251
252
253
surface = SDL_CreateRGBSurface(SDL_SWSURFACE,
features.width, features.height,
features.has_alpha?32:24, Rmask,Gmask,Bmask,Amask);
254
255
256
257
258
if ( surface == NULL ) {
error = "Failed to allocate SDL_Surface";
goto error;
}
259
260
261
262
263
264
if ( features.has_alpha ) {
ret = lib.webp_decode_rgba_into( raw_data, raw_data_size, (uint8_t *)surface->pixels, surface->pitch * surface->h, surface->pitch );
} else {
ret = lib.webp_decode_rgb_into( raw_data, raw_data_size, (uint8_t *)surface->pixels, surface->pitch * surface->h, surface->pitch );
}
265
266
267
268
269
if ( !ret ) {
error = "Failed to decode WEBP";
goto error;
}
270
271
272
273
274
if ( raw_data ) {
SDL_free( raw_data );
}
275
return surface;
276
277
278
279
error:
280
281
282
if ( raw_data ) {
SDL_free( raw_data );
}
283
284
285
286
287
if ( surface ) {
SDL_FreeSurface( surface );
}
288
if ( error ) {
289
IMG_SetError( "%s", error );
290
}
291
292
293
SDL_RWseek(src, start, RW_SEEK_SET);
return(NULL);
294
295
296
297
298
299
}
#else
int IMG_InitWEBP()
{
300
301
IMG_SetError("WEBP images are not supported");
return(-1);
302
303
304
305
306
307
308
309
310
}
void IMG_QuitWEBP()
{
}
/* See if an image is contained in a data source */
int IMG_isWEBP(SDL_RWops *src)
{
311
return(0);
312
313
314
315
316
}
/* Load a WEBP type image from an SDL datasource */
SDL_Surface *IMG_LoadWEBP_RW(SDL_RWops *src)
{
317
return(NULL);
318
319
320
}
#endif /* LOAD_WEBP */