128#ifndef STBI_INCLUDE_STB_IMAGE_H
129#define STBI_INCLUDE_STB_IMAGE_H
374#define STBI_VERSION 1
387typedef unsigned char stbi_uc;
388typedef unsigned short stbi_us;
395#ifdef STB_IMAGE_STATIC
396#define STBIDEF static
398#define STBIDEF extern
413 int (*read) (
void *user,
char *data,
int size);
414 void (*skip) (
void *user,
int n);
415 int (*eof) (
void *user);
423STBIDEF stbi_uc *stbi_load_from_memory (stbi_uc
const *buffer,
int len ,
int *x,
int *y,
int *channels_in_file,
int desired_channels);
424STBIDEF stbi_uc *stbi_load_from_callbacks(
stbi_io_callbacks const *clbk ,
void *user,
int *x,
int *y,
int *channels_in_file,
int desired_channels);
427STBIDEF stbi_uc *stbi_load (
char const *filename,
int *x,
int *y,
int *channels_in_file,
int desired_channels);
428STBIDEF stbi_uc *stbi_load_from_file (FILE *f,
int *x,
int *y,
int *channels_in_file,
int desired_channels);
433STBIDEF stbi_uc *stbi_load_gif_from_memory(stbi_uc
const *buffer,
int len,
int **delays,
int *x,
int *y,
int *z,
int *comp,
int req_comp);
436#ifdef STBI_WINDOWS_UTF8
437STBIDEF
int stbi_convert_wchar_to_utf8(
char *buffer,
size_t bufferlen,
const wchar_t* input);
445STBIDEF stbi_us *stbi_load_16_from_memory (stbi_uc
const *buffer,
int len,
int *x,
int *y,
int *channels_in_file,
int desired_channels);
446STBIDEF stbi_us *stbi_load_16_from_callbacks(
stbi_io_callbacks const *clbk,
void *user,
int *x,
int *y,
int *channels_in_file,
int desired_channels);
449STBIDEF stbi_us *stbi_load_16 (
char const *filename,
int *x,
int *y,
int *channels_in_file,
int desired_channels);
450STBIDEF stbi_us *stbi_load_from_file_16(FILE *f,
int *x,
int *y,
int *channels_in_file,
int desired_channels);
457#ifndef STBI_NO_LINEAR
458 STBIDEF
float *stbi_loadf_from_memory (stbi_uc
const *buffer,
int len,
int *x,
int *y,
int *channels_in_file,
int desired_channels);
459 STBIDEF
float *stbi_loadf_from_callbacks (
stbi_io_callbacks const *clbk,
void *user,
int *x,
int *y,
int *channels_in_file,
int desired_channels);
461 #ifndef STBI_NO_STDIO
462 STBIDEF
float *stbi_loadf (
char const *filename,
int *x,
int *y,
int *channels_in_file,
int desired_channels);
463 STBIDEF
float *stbi_loadf_from_file (FILE *f,
int *x,
int *y,
int *channels_in_file,
int desired_channels);
468 STBIDEF
void stbi_hdr_to_ldr_gamma(
float gamma);
469 STBIDEF
void stbi_hdr_to_ldr_scale(
float scale);
472#ifndef STBI_NO_LINEAR
473 STBIDEF
void stbi_ldr_to_hdr_gamma(
float gamma);
474 STBIDEF
void stbi_ldr_to_hdr_scale(
float scale);
478STBIDEF
int stbi_is_hdr_from_callbacks(
stbi_io_callbacks const *clbk,
void *user);
479STBIDEF
int stbi_is_hdr_from_memory(stbi_uc
const *buffer,
int len);
481STBIDEF
int stbi_is_hdr (
char const *filename);
482STBIDEF
int stbi_is_hdr_from_file(FILE *f);
488STBIDEF
const char *stbi_failure_reason (
void);
491STBIDEF
void stbi_image_free (
void *retval_from_stbi_load);
494STBIDEF
int stbi_info_from_memory(stbi_uc
const *buffer,
int len,
int *x,
int *y,
int *comp);
495STBIDEF
int stbi_info_from_callbacks(
stbi_io_callbacks const *clbk,
void *user,
int *x,
int *y,
int *comp);
496STBIDEF
int stbi_is_16_bit_from_memory(stbi_uc
const *buffer,
int len);
497STBIDEF
int stbi_is_16_bit_from_callbacks(
stbi_io_callbacks const *clbk,
void *user);
500STBIDEF
int stbi_info (
char const *filename,
int *x,
int *y,
int *comp);
501STBIDEF
int stbi_info_from_file (FILE *f,
int *x,
int *y,
int *comp);
502STBIDEF
int stbi_is_16_bit (
char const *filename);
503STBIDEF
int stbi_is_16_bit_from_file(FILE *f);
511STBIDEF
void stbi_set_unpremultiply_on_load(
int flag_true_if_should_unpremultiply);
515STBIDEF
void stbi_convert_iphone_png_to_rgb(
int flag_true_if_should_convert);
518STBIDEF
void stbi_set_flip_vertically_on_load(
int flag_true_if_should_flip);
523STBIDEF
void stbi_set_unpremultiply_on_load_thread(
int flag_true_if_should_unpremultiply);
524STBIDEF
void stbi_convert_iphone_png_to_rgb_thread(
int flag_true_if_should_convert);
525STBIDEF
void stbi_set_flip_vertically_on_load_thread(
int flag_true_if_should_flip);
529STBIDEF
char *stbi_zlib_decode_malloc_guesssize(
const char *buffer,
int len,
int initial_size,
int *outlen);
530STBIDEF
char *stbi_zlib_decode_malloc_guesssize_headerflag(
const char *buffer,
int len,
int initial_size,
int *outlen,
int parse_header);
531STBIDEF
char *stbi_zlib_decode_malloc(
const char *buffer,
int len,
int *outlen);
532STBIDEF
int stbi_zlib_decode_buffer(
char *obuffer,
int olen,
const char *ibuffer,
int ilen);
534STBIDEF
char *stbi_zlib_decode_noheader_malloc(
const char *buffer,
int len,
int *outlen);
535STBIDEF
int stbi_zlib_decode_noheader_buffer(
char *obuffer,
int olen,
const char *ibuffer,
int ilen);
547#ifdef STB_IMAGE_IMPLEMENTATION
549#if defined(STBI_ONLY_JPEG) || defined(STBI_ONLY_PNG) || defined(STBI_ONLY_BMP) \
550 || defined(STBI_ONLY_TGA) || defined(STBI_ONLY_GIF) || defined(STBI_ONLY_PSD) \
551 || defined(STBI_ONLY_HDR) || defined(STBI_ONLY_PIC) || defined(STBI_ONLY_PNM) \
552 || defined(STBI_ONLY_ZLIB)
553 #ifndef STBI_ONLY_JPEG
556 #ifndef STBI_ONLY_PNG
559 #ifndef STBI_ONLY_BMP
562 #ifndef STBI_ONLY_PSD
565 #ifndef STBI_ONLY_TGA
568 #ifndef STBI_ONLY_GIF
571 #ifndef STBI_ONLY_HDR
574 #ifndef STBI_ONLY_PIC
577 #ifndef STBI_ONLY_PNM
582#if defined(STBI_NO_PNG) && !defined(STBI_SUPPORT_ZLIB) && !defined(STBI_NO_ZLIB)
593#if !defined(STBI_NO_LINEAR) || !defined(STBI_NO_HDR)
603#define STBI_ASSERT(x) assert(x)
607#define STBI_EXTERN extern "C"
609#define STBI_EXTERN extern
615 #define stbi_inline inline
620 #define stbi_inline __forceinline
623#ifndef STBI_NO_THREAD_LOCALS
624 #if defined(__cplusplus) && __cplusplus >= 201103L
625 #define STBI_THREAD_LOCAL thread_local
626 #elif defined(__GNUC__) && __GNUC__ < 5
627 #define STBI_THREAD_LOCAL __thread
628 #elif defined(_MSC_VER)
629 #define STBI_THREAD_LOCAL __declspec(thread)
630 #elif defined (__STDC_VERSION__) && __STDC_VERSION__ >= 201112L && !defined(__STDC_NO_THREADS__)
631 #define STBI_THREAD_LOCAL _Thread_local
634 #ifndef STBI_THREAD_LOCAL
635 #if defined(__GNUC__)
636 #define STBI_THREAD_LOCAL __thread
641#if defined(_MSC_VER) || defined(__SYMBIAN32__)
642typedef unsigned short stbi__uint16;
643typedef signed short stbi__int16;
644typedef unsigned int stbi__uint32;
645typedef signed int stbi__int32;
648typedef uint16_t stbi__uint16;
649typedef int16_t stbi__int16;
650typedef uint32_t stbi__uint32;
651typedef int32_t stbi__int32;
655typedef unsigned char validate_uint32[
sizeof(stbi__uint32)==4 ? 1 : -1];
658#define STBI_NOTUSED(v) (void)(v)
660#define STBI_NOTUSED(v) (void)sizeof(v)
664#define STBI_HAS_LROTL
668 #define stbi_lrot(x,y) _lrotl(x,y)
670 #define stbi_lrot(x,y) (((x) << (y)) | ((x) >> (-(y) & 31)))
673#if defined(STBI_MALLOC) && defined(STBI_FREE) && (defined(STBI_REALLOC) || defined(STBI_REALLOC_SIZED))
675#elif !defined(STBI_MALLOC) && !defined(STBI_FREE) && !defined(STBI_REALLOC) && !defined(STBI_REALLOC_SIZED)
678#error "Must define all or none of STBI_MALLOC, STBI_FREE, and STBI_REALLOC (or STBI_REALLOC_SIZED)."
682#define STBI_MALLOC(sz) malloc(sz)
683#define STBI_REALLOC(p,newsz) realloc(p,newsz)
684#define STBI_FREE(p) free(p)
687#ifndef STBI_REALLOC_SIZED
688#define STBI_REALLOC_SIZED(p,oldsz,newsz) STBI_REALLOC(p,newsz)
692#if defined(__x86_64__) || defined(_M_X64)
693#define STBI__X64_TARGET
694#elif defined(__i386) || defined(_M_IX86)
695#define STBI__X86_TARGET
698#if defined(__GNUC__) && defined(STBI__X86_TARGET) && !defined(__SSE2__) && !defined(STBI_NO_SIMD)
709#if defined(__MINGW32__) && defined(STBI__X86_TARGET) && !defined(STBI_MINGW_ENABLE_SSE2) && !defined(STBI_NO_SIMD)
724#if !defined(STBI_NO_SIMD) && (defined(STBI__X86_TARGET) || defined(STBI__X64_TARGET))
726#include <emmintrin.h>
732static int stbi__cpuid3(
void)
739static int stbi__cpuid3(
void)
751#define STBI_SIMD_ALIGN(type, name) __declspec(align(16)) type name
753#if !defined(STBI_NO_JPEG) && defined(STBI_SSE2)
754static int stbi__sse2_available(
void)
756 int info3 = stbi__cpuid3();
757 return ((info3 >> 26) & 1) != 0;
762#define STBI_SIMD_ALIGN(type, name) type name __attribute__((aligned(16)))
764#if !defined(STBI_NO_JPEG) && defined(STBI_SSE2)
765static int stbi__sse2_available(
void)
778#if defined(STBI_NO_SIMD) && defined(STBI_NEON)
785#define STBI_SIMD_ALIGN(type, name) __declspec(align(16)) type name
787#define STBI_SIMD_ALIGN(type, name) type name __attribute__((aligned(16)))
791#ifndef STBI_SIMD_ALIGN
792#define STBI_SIMD_ALIGN(type, name) type name
795#ifndef STBI_MAX_DIMENSIONS
796#define STBI_MAX_DIMENSIONS (1 << 24)
807 stbi__uint32 img_x, img_y;
808 int img_n, img_out_n;
813 int read_from_callbacks;
815 stbi_uc buffer_start[128];
816 int callback_already_read;
818 stbi_uc *img_buffer, *img_buffer_end;
819 stbi_uc *img_buffer_original, *img_buffer_original_end;
823static void stbi__refill_buffer(stbi__context *s);
826static void stbi__start_mem(stbi__context *s, stbi_uc
const *buffer,
int len)
829 s->read_from_callbacks = 0;
830 s->callback_already_read = 0;
831 s->img_buffer = s->img_buffer_original = (stbi_uc *) buffer;
832 s->img_buffer_end = s->img_buffer_original_end = (stbi_uc *) buffer+len;
836static void stbi__start_callbacks(stbi__context *s,
stbi_io_callbacks *c,
void *user)
839 s->io_user_data = user;
840 s->buflen =
sizeof(s->buffer_start);
841 s->read_from_callbacks = 1;
842 s->callback_already_read = 0;
843 s->img_buffer = s->img_buffer_original = s->buffer_start;
844 stbi__refill_buffer(s);
845 s->img_buffer_original_end = s->img_buffer_end;
850static int stbi__stdio_read(
void *user,
char *data,
int size)
852 return (
int) fread(data,1,size,(FILE*) user);
855static void stbi__stdio_skip(
void *user,
int n)
858 fseek((FILE*) user, n, SEEK_CUR);
859 ch = fgetc((FILE*) user);
861 ungetc(ch, (FILE *) user);
865static int stbi__stdio_eof(
void *user)
867 return feof((FILE*) user) || ferror((FILE *) user);
877static void stbi__start_file(stbi__context *s, FILE *f)
879 stbi__start_callbacks(s, &stbi__stdio_callbacks, (
void *) f);
886static void stbi__rewind(stbi__context *s)
891 s->img_buffer = s->img_buffer_original;
892 s->img_buffer_end = s->img_buffer_original_end;
903 int bits_per_channel;
909static int stbi__jpeg_test(stbi__context *s);
910static void *stbi__jpeg_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri);
911static int stbi__jpeg_info(stbi__context *s,
int *x,
int *y,
int *comp);
915static int stbi__png_test(stbi__context *s);
916static void *stbi__png_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri);
917static int stbi__png_info(stbi__context *s,
int *x,
int *y,
int *comp);
918static int stbi__png_is16(stbi__context *s);
922static int stbi__bmp_test(stbi__context *s);
923static void *stbi__bmp_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri);
924static int stbi__bmp_info(stbi__context *s,
int *x,
int *y,
int *comp);
928static int stbi__tga_test(stbi__context *s);
929static void *stbi__tga_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri);
930static int stbi__tga_info(stbi__context *s,
int *x,
int *y,
int *comp);
934static int stbi__psd_test(stbi__context *s);
935static void *stbi__psd_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri,
int bpc);
936static int stbi__psd_info(stbi__context *s,
int *x,
int *y,
int *comp);
937static int stbi__psd_is16(stbi__context *s);
941static int stbi__hdr_test(stbi__context *s);
942static float *stbi__hdr_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri);
943static int stbi__hdr_info(stbi__context *s,
int *x,
int *y,
int *comp);
947static int stbi__pic_test(stbi__context *s);
948static void *stbi__pic_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri);
949static int stbi__pic_info(stbi__context *s,
int *x,
int *y,
int *comp);
953static int stbi__gif_test(stbi__context *s);
954static void *stbi__gif_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri);
955static void *stbi__load_gif_main(stbi__context *s,
int **delays,
int *x,
int *y,
int *z,
int *comp,
int req_comp);
956static int stbi__gif_info(stbi__context *s,
int *x,
int *y,
int *comp);
960static int stbi__pnm_test(stbi__context *s);
961static void *stbi__pnm_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri);
962static int stbi__pnm_info(stbi__context *s,
int *x,
int *y,
int *comp);
963static int stbi__pnm_is16(stbi__context *s);
967#ifdef STBI_THREAD_LOCAL
970const char *stbi__g_failure_reason;
972STBIDEF
const char *stbi_failure_reason(
void)
974 return stbi__g_failure_reason;
977#ifndef STBI_NO_FAILURE_STRINGS
978static int stbi__err(
const char *str)
980 stbi__g_failure_reason = str;
985static void *stbi__malloc(
size_t size)
987 return STBI_MALLOC(size);
1002static int stbi__addsizes_valid(
int a,
int b)
1004 if (b < 0)
return 0;
1009 return a <= INT_MAX - b;
1014static int stbi__mul2sizes_valid(
int a,
int b)
1016 if (a < 0 || b < 0)
return 0;
1017 if (b == 0)
return 1;
1019 return a <= INT_MAX/b;
1022#if !defined(STBI_NO_JPEG) || !defined(STBI_NO_PNG) || !defined(STBI_NO_TGA) || !defined(STBI_NO_HDR)
1024static int stbi__mad2sizes_valid(
int a,
int b,
int add)
1026 return stbi__mul2sizes_valid(a, b) && stbi__addsizes_valid(a*b, add);
1031static int stbi__mad3sizes_valid(
int a,
int b,
int c,
int add)
1033 return stbi__mul2sizes_valid(a, b) && stbi__mul2sizes_valid(a*b, c) &&
1034 stbi__addsizes_valid(a*b*c, add);
1038#if !defined(STBI_NO_LINEAR) || !defined(STBI_NO_HDR) || !defined(STBI_NO_PNM)
1039static int stbi__mad4sizes_valid(
int a,
int b,
int c,
int d,
int add)
1041 return stbi__mul2sizes_valid(a, b) && stbi__mul2sizes_valid(a*b, c) &&
1042 stbi__mul2sizes_valid(a*b*c, d) && stbi__addsizes_valid(a*b*c*d, add);
1046#if !defined(STBI_NO_JPEG) || !defined(STBI_NO_PNG) || !defined(STBI_NO_TGA) || !defined(STBI_NO_HDR)
1048static void *stbi__malloc_mad2(
int a,
int b,
int add)
1050 if (!stbi__mad2sizes_valid(a, b, add))
return NULL;
1051 return stbi__malloc(a*b + add);
1055static void *stbi__malloc_mad3(
int a,
int b,
int c,
int add)
1057 if (!stbi__mad3sizes_valid(a, b, c, add))
return NULL;
1058 return stbi__malloc(a*b*c + add);
1061#if !defined(STBI_NO_LINEAR) || !defined(STBI_NO_HDR) || !defined(STBI_NO_PNM)
1062static void *stbi__malloc_mad4(
int a,
int b,
int c,
int d,
int add)
1064 if (!stbi__mad4sizes_valid(a, b, c, d, add))
return NULL;
1065 return stbi__malloc(a*b*c*d + add);
1070static int stbi__addints_valid(
int a,
int b)
1072 if ((a >= 0) != (b >= 0))
return 1;
1073 if (a < 0 && b < 0)
return a >= INT_MIN - b;
1074 return a <= INT_MAX - b;
1078static int stbi__mul2shorts_valid(
int a,
int b)
1080 if (b == 0 || b == -1)
return 1;
1081 if ((a >= 0) == (b >= 0))
return a <= SHRT_MAX/b;
1082 if (b < 0)
return a <= SHRT_MIN / b;
1083 return a >= SHRT_MIN / b;
1090#ifdef STBI_NO_FAILURE_STRINGS
1091 #define stbi__err(x,y) 0
1092#elif defined(STBI_FAILURE_USERMSG)
1093 #define stbi__err(x,y) stbi__err(y)
1095 #define stbi__err(x,y) stbi__err(x)
1098#define stbi__errpf(x,y) ((float *)(size_t) (stbi__err(x,y)?NULL:NULL))
1099#define stbi__errpuc(x,y) ((unsigned char *)(size_t) (stbi__err(x,y)?NULL:NULL))
1101STBIDEF
void stbi_image_free(
void *retval_from_stbi_load)
1103 STBI_FREE(retval_from_stbi_load);
1106#ifndef STBI_NO_LINEAR
1107static float *stbi__ldr_to_hdr(stbi_uc *data,
int x,
int y,
int comp);
1111static stbi_uc *stbi__hdr_to_ldr(
float *data,
int x,
int y,
int comp);
1114static int stbi__vertically_flip_on_load_global = 0;
1116STBIDEF
void stbi_set_flip_vertically_on_load(
int flag_true_if_should_flip)
1118 stbi__vertically_flip_on_load_global = flag_true_if_should_flip;
1121#ifndef STBI_THREAD_LOCAL
1122#define stbi__vertically_flip_on_load stbi__vertically_flip_on_load_global
1124static STBI_THREAD_LOCAL
int stbi__vertically_flip_on_load_local, stbi__vertically_flip_on_load_set;
1126STBIDEF
void stbi_set_flip_vertically_on_load_thread(
int flag_true_if_should_flip)
1128 stbi__vertically_flip_on_load_local = flag_true_if_should_flip;
1129 stbi__vertically_flip_on_load_set = 1;
1132#define stbi__vertically_flip_on_load (stbi__vertically_flip_on_load_set \
1133 ? stbi__vertically_flip_on_load_local \
1134 : stbi__vertically_flip_on_load_global)
1137static void *stbi__load_main(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri,
int bpc)
1139 memset(ri, 0,
sizeof(*ri));
1140 ri->bits_per_channel = 8;
1141 ri->channel_order = STBI_ORDER_RGB;
1142 ri->num_channels = 0;
1147 if (stbi__png_test(s))
return stbi__png_load(s,x,y,comp,req_comp, ri);
1150 if (stbi__bmp_test(s))
return stbi__bmp_load(s,x,y,comp,req_comp, ri);
1153 if (stbi__gif_test(s))
return stbi__gif_load(s,x,y,comp,req_comp, ri);
1156 if (stbi__psd_test(s))
return stbi__psd_load(s,x,y,comp,req_comp, ri, bpc);
1161 if (stbi__pic_test(s))
return stbi__pic_load(s,x,y,comp,req_comp, ri);
1167 #ifndef STBI_NO_JPEG
1168 if (stbi__jpeg_test(s))
return stbi__jpeg_load(s,x,y,comp,req_comp, ri);
1171 if (stbi__pnm_test(s))
return stbi__pnm_load(s,x,y,comp,req_comp, ri);
1175 if (stbi__hdr_test(s)) {
1176 float *hdr = stbi__hdr_load(s, x,y,comp,req_comp, ri);
1177 return stbi__hdr_to_ldr(hdr, *x, *y, req_comp ? req_comp : *comp);
1183 if (stbi__tga_test(s))
1184 return stbi__tga_load(s,x,y,comp,req_comp, ri);
1187 return stbi__errpuc(
"unknown image type",
"Image not of any known type, or corrupt");
1190static stbi_uc *stbi__convert_16_to_8(stbi__uint16 *orig,
int w,
int h,
int channels)
1193 int img_len = w * h * channels;
1196 reduced = (stbi_uc *) stbi__malloc(img_len);
1197 if (reduced == NULL)
return stbi__errpuc(
"outofmem",
"Out of memory");
1199 for (i = 0; i < img_len; ++i)
1200 reduced[i] = (stbi_uc)((orig[i] >> 8) & 0xFF);
1206static stbi__uint16 *stbi__convert_8_to_16(stbi_uc *orig,
int w,
int h,
int channels)
1209 int img_len = w * h * channels;
1210 stbi__uint16 *enlarged;
1212 enlarged = (stbi__uint16 *) stbi__malloc(img_len*2);
1213 if (enlarged == NULL)
return (stbi__uint16 *) stbi__errpuc(
"outofmem",
"Out of memory");
1215 for (i = 0; i < img_len; ++i)
1216 enlarged[i] = (stbi__uint16)((orig[i] << 8) + orig[i]);
1222static void stbi__vertical_flip(
void *image,
int w,
int h,
int bytes_per_pixel)
1225 size_t bytes_per_row = (size_t)w * bytes_per_pixel;
1227 stbi_uc *bytes = (stbi_uc *)image;
1229 for (row = 0; row < (h>>1); row++) {
1230 stbi_uc *row0 = bytes + row*bytes_per_row;
1231 stbi_uc *row1 = bytes + (h - row - 1)*bytes_per_row;
1233 size_t bytes_left = bytes_per_row;
1234 while (bytes_left) {
1235 size_t bytes_copy = (bytes_left <
sizeof(temp)) ? bytes_left : sizeof(temp);
1236 memcpy(temp, row0, bytes_copy);
1237 memcpy(row0, row1, bytes_copy);
1238 memcpy(row1, temp, bytes_copy);
1241 bytes_left -= bytes_copy;
1247static void stbi__vertical_flip_slices(
void *image,
int w,
int h,
int z,
int bytes_per_pixel)
1250 int slice_size = w * h * bytes_per_pixel;
1252 stbi_uc *bytes = (stbi_uc *)image;
1253 for (slice = 0; slice < z; ++slice) {
1254 stbi__vertical_flip(bytes, w, h, bytes_per_pixel);
1255 bytes += slice_size;
1260static unsigned char *stbi__load_and_postprocess_8bit(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp)
1262 stbi__result_info ri;
1263 void *result = stbi__load_main(s, x, y, comp, req_comp, &ri, 8);
1269 STBI_ASSERT(ri.bits_per_channel == 8 || ri.bits_per_channel == 16);
1271 if (ri.bits_per_channel != 8) {
1272 result = stbi__convert_16_to_8((stbi__uint16 *) result, *x, *y, req_comp == 0 ? *comp : req_comp);
1273 ri.bits_per_channel = 8;
1278 if (stbi__vertically_flip_on_load) {
1279 int channels = req_comp ? req_comp : *comp;
1280 stbi__vertical_flip(result, *x, *y, channels *
sizeof(stbi_uc));
1283 return (
unsigned char *) result;
1286static stbi__uint16 *stbi__load_and_postprocess_16bit(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp)
1288 stbi__result_info ri;
1289 void *result = stbi__load_main(s, x, y, comp, req_comp, &ri, 16);
1295 STBI_ASSERT(ri.bits_per_channel == 8 || ri.bits_per_channel == 16);
1297 if (ri.bits_per_channel != 16) {
1298 result = stbi__convert_8_to_16((stbi_uc *) result, *x, *y, req_comp == 0 ? *comp : req_comp);
1299 ri.bits_per_channel = 16;
1305 if (stbi__vertically_flip_on_load) {
1306 int channels = req_comp ? req_comp : *comp;
1307 stbi__vertical_flip(result, *x, *y, channels *
sizeof(stbi__uint16));
1310 return (stbi__uint16 *) result;
1313#if !defined(STBI_NO_HDR) && !defined(STBI_NO_LINEAR)
1314static void stbi__float_postprocess(
float *result,
int *x,
int *y,
int *comp,
int req_comp)
1316 if (stbi__vertically_flip_on_load && result != NULL) {
1317 int channels = req_comp ? req_comp : *comp;
1318 stbi__vertical_flip(result, *x, *y, channels *
sizeof(
float));
1323#ifndef STBI_NO_STDIO
1325#if defined(_WIN32) && defined(STBI_WINDOWS_UTF8)
1326STBI_EXTERN __declspec(dllimport)
int __stdcall MultiByteToWideChar(
unsigned int cp,
unsigned long flags,
const char *str,
int cbmb,
wchar_t *widestr,
int cchwide);
1327STBI_EXTERN __declspec(dllimport)
int __stdcall WideCharToMultiByte(
unsigned int cp,
unsigned long flags,
const wchar_t *widestr,
int cchwide,
char *str,
int cbmb,
const char *defchar,
int *used_default);
1330#if defined(_WIN32) && defined(STBI_WINDOWS_UTF8)
1331STBIDEF
int stbi_convert_wchar_to_utf8(
char *buffer,
size_t bufferlen,
const wchar_t* input)
1333 return WideCharToMultiByte(65001 , 0, input, -1, buffer, (
int) bufferlen, NULL, NULL);
1337static FILE *stbi__fopen(
char const *filename,
char const *mode)
1340#if defined(_WIN32) && defined(STBI_WINDOWS_UTF8)
1342 wchar_t wFilename[1024];
1343 if (0 == MultiByteToWideChar(65001 , 0, filename, -1, wFilename,
sizeof(wFilename)/
sizeof(*wFilename)))
1346 if (0 == MultiByteToWideChar(65001 , 0, mode, -1, wMode,
sizeof(wMode)/
sizeof(*wMode)))
1349#if defined(_MSC_VER) && _MSC_VER >= 1400
1350 if (0 != _wfopen_s(&f, wFilename, wMode))
1353 f = _wfopen(wFilename, wMode);
1356#elif defined(_MSC_VER) && _MSC_VER >= 1400
1357 if (0 != fopen_s(&f, filename, mode))
1360 f = fopen(filename, mode);
1366STBIDEF stbi_uc *stbi_load(
char const *filename,
int *x,
int *y,
int *comp,
int req_comp)
1368 FILE *f = stbi__fopen(filename,
"rb");
1369 unsigned char *result;
1370 if (!f)
return stbi__errpuc(
"can't fopen",
"Unable to open file");
1371 result = stbi_load_from_file(f,x,y,comp,req_comp);
1376STBIDEF stbi_uc *stbi_load_from_file(FILE *f,
int *x,
int *y,
int *comp,
int req_comp)
1378 unsigned char *result;
1380 stbi__start_file(&s,f);
1381 result = stbi__load_and_postprocess_8bit(&s,x,y,comp,req_comp);
1384 fseek(f, - (
int) (s.img_buffer_end - s.img_buffer), SEEK_CUR);
1389STBIDEF stbi__uint16 *stbi_load_from_file_16(FILE *f,
int *x,
int *y,
int *comp,
int req_comp)
1391 stbi__uint16 *result;
1393 stbi__start_file(&s,f);
1394 result = stbi__load_and_postprocess_16bit(&s,x,y,comp,req_comp);
1397 fseek(f, - (
int) (s.img_buffer_end - s.img_buffer), SEEK_CUR);
1402STBIDEF stbi_us *stbi_load_16(
char const *filename,
int *x,
int *y,
int *comp,
int req_comp)
1404 FILE *f = stbi__fopen(filename,
"rb");
1405 stbi__uint16 *result;
1406 if (!f)
return (stbi_us *) stbi__errpuc(
"can't fopen",
"Unable to open file");
1407 result = stbi_load_from_file_16(f,x,y,comp,req_comp);
1415STBIDEF stbi_us *stbi_load_16_from_memory(stbi_uc
const *buffer,
int len,
int *x,
int *y,
int *channels_in_file,
int desired_channels)
1418 stbi__start_mem(&s,buffer,len);
1419 return stbi__load_and_postprocess_16bit(&s,x,y,channels_in_file,desired_channels);
1422STBIDEF stbi_us *stbi_load_16_from_callbacks(
stbi_io_callbacks const *clbk,
void *user,
int *x,
int *y,
int *channels_in_file,
int desired_channels)
1426 return stbi__load_and_postprocess_16bit(&s,x,y,channels_in_file,desired_channels);
1429STBIDEF stbi_uc *stbi_load_from_memory(stbi_uc
const *buffer,
int len,
int *x,
int *y,
int *comp,
int req_comp)
1432 stbi__start_mem(&s,buffer,len);
1433 return stbi__load_and_postprocess_8bit(&s,x,y,comp,req_comp);
1436STBIDEF stbi_uc *stbi_load_from_callbacks(
stbi_io_callbacks const *clbk,
void *user,
int *x,
int *y,
int *comp,
int req_comp)
1440 return stbi__load_and_postprocess_8bit(&s,x,y,comp,req_comp);
1444STBIDEF stbi_uc *stbi_load_gif_from_memory(stbi_uc
const *buffer,
int len,
int **delays,
int *x,
int *y,
int *z,
int *comp,
int req_comp)
1446 unsigned char *result;
1448 stbi__start_mem(&s,buffer,len);
1450 result = (
unsigned char*) stbi__load_gif_main(&s, delays, x, y, z, comp, req_comp);
1451 if (stbi__vertically_flip_on_load) {
1452 stbi__vertical_flip_slices( result, *x, *y, *z, *comp );
1459#ifndef STBI_NO_LINEAR
1460static float *stbi__loadf_main(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp)
1462 unsigned char *data;
1464 if (stbi__hdr_test(s)) {
1465 stbi__result_info ri;
1466 float *hdr_data = stbi__hdr_load(s,x,y,comp,req_comp, &ri);
1468 stbi__float_postprocess(hdr_data,x,y,comp,req_comp);
1472 data = stbi__load_and_postprocess_8bit(s, x, y, comp, req_comp);
1474 return stbi__ldr_to_hdr(data, *x, *y, req_comp ? req_comp : *comp);
1475 return stbi__errpf(
"unknown image type",
"Image not of any known type, or corrupt");
1478STBIDEF
float *stbi_loadf_from_memory(stbi_uc
const *buffer,
int len,
int *x,
int *y,
int *comp,
int req_comp)
1481 stbi__start_mem(&s,buffer,len);
1482 return stbi__loadf_main(&s,x,y,comp,req_comp);
1485STBIDEF
float *stbi_loadf_from_callbacks(
stbi_io_callbacks const *clbk,
void *user,
int *x,
int *y,
int *comp,
int req_comp)
1489 return stbi__loadf_main(&s,x,y,comp,req_comp);
1492#ifndef STBI_NO_STDIO
1493STBIDEF
float *stbi_loadf(
char const *filename,
int *x,
int *y,
int *comp,
int req_comp)
1496 FILE *f = stbi__fopen(filename,
"rb");
1497 if (!f)
return stbi__errpf(
"can't fopen",
"Unable to open file");
1498 result = stbi_loadf_from_file(f,x,y,comp,req_comp);
1503STBIDEF
float *stbi_loadf_from_file(FILE *f,
int *x,
int *y,
int *comp,
int req_comp)
1506 stbi__start_file(&s,f);
1507 return stbi__loadf_main(&s,x,y,comp,req_comp);
1517STBIDEF
int stbi_is_hdr_from_memory(stbi_uc
const *buffer,
int len)
1521 stbi__start_mem(&s,buffer,len);
1522 return stbi__hdr_test(&s);
1524 STBI_NOTUSED(buffer);
1530#ifndef STBI_NO_STDIO
1531STBIDEF
int stbi_is_hdr (
char const *filename)
1533 FILE *f = stbi__fopen(filename,
"rb");
1536 result = stbi_is_hdr_from_file(f);
1542STBIDEF
int stbi_is_hdr_from_file(FILE *f)
1545 long pos = ftell(f);
1548 stbi__start_file(&s,f);
1549 res = stbi__hdr_test(&s);
1550 fseek(f, pos, SEEK_SET);
1559STBIDEF
int stbi_is_hdr_from_callbacks(
stbi_io_callbacks const *clbk,
void *user)
1564 return stbi__hdr_test(&s);
1572#ifndef STBI_NO_LINEAR
1573static float stbi__l2h_gamma=2.2f, stbi__l2h_scale=1.0f;
1575STBIDEF
void stbi_ldr_to_hdr_gamma(
float gamma) { stbi__l2h_gamma = gamma; }
1576STBIDEF
void stbi_ldr_to_hdr_scale(
float scale) { stbi__l2h_scale = scale; }
1579static float stbi__h2l_gamma_i=1.0f/2.2f, stbi__h2l_scale_i=1.0f;
1581STBIDEF
void stbi_hdr_to_ldr_gamma(
float gamma) { stbi__h2l_gamma_i = 1/gamma; }
1582STBIDEF
void stbi_hdr_to_ldr_scale(
float scale) { stbi__h2l_scale_i = 1/scale; }
1597static void stbi__refill_buffer(stbi__context *s)
1599 int n = (s->io.read)(s->io_user_data,(
char*)s->buffer_start,s->buflen);
1600 s->callback_already_read += (int) (s->img_buffer - s->img_buffer_original);
1604 s->read_from_callbacks = 0;
1605 s->img_buffer = s->buffer_start;
1606 s->img_buffer_end = s->buffer_start+1;
1609 s->img_buffer = s->buffer_start;
1610 s->img_buffer_end = s->buffer_start + n;
1614stbi_inline
static stbi_uc stbi__get8(stbi__context *s)
1616 if (s->img_buffer < s->img_buffer_end)
1617 return *s->img_buffer++;
1618 if (s->read_from_callbacks) {
1619 stbi__refill_buffer(s);
1620 return *s->img_buffer++;
1625#if defined(STBI_NO_JPEG) && defined(STBI_NO_HDR) && defined(STBI_NO_PIC) && defined(STBI_NO_PNM)
1628stbi_inline
static int stbi__at_eof(stbi__context *s)
1631 if (!(s->io.eof)(s->io_user_data))
return 0;
1634 if (s->read_from_callbacks == 0)
return 1;
1637 return s->img_buffer >= s->img_buffer_end;
1641#if defined(STBI_NO_JPEG) && defined(STBI_NO_PNG) && defined(STBI_NO_BMP) && defined(STBI_NO_PSD) && defined(STBI_NO_TGA) && defined(STBI_NO_GIF) && defined(STBI_NO_PIC)
1644static void stbi__skip(stbi__context *s,
int n)
1648 s->img_buffer = s->img_buffer_end;
1652 int blen = (int) (s->img_buffer_end - s->img_buffer);
1654 s->img_buffer = s->img_buffer_end;
1655 (s->io.skip)(s->io_user_data, n - blen);
1663#if defined(STBI_NO_PNG) && defined(STBI_NO_TGA) && defined(STBI_NO_HDR) && defined(STBI_NO_PNM)
1666static int stbi__getn(stbi__context *s, stbi_uc *buffer,
int n)
1669 int blen = (int) (s->img_buffer_end - s->img_buffer);
1673 memcpy(buffer, s->img_buffer, blen);
1675 count = (s->io.read)(s->io_user_data, (
char*) buffer + blen, n - blen);
1676 res = (count == (n-blen));
1677 s->img_buffer = s->img_buffer_end;
1682 if (s->img_buffer+n <= s->img_buffer_end) {
1683 memcpy(buffer, s->img_buffer, n);
1691#if defined(STBI_NO_JPEG) && defined(STBI_NO_PNG) && defined(STBI_NO_PSD) && defined(STBI_NO_PIC)
1694static int stbi__get16be(stbi__context *s)
1696 int z = stbi__get8(s);
1697 return (z << 8) + stbi__get8(s);
1701#if defined(STBI_NO_PNG) && defined(STBI_NO_PSD) && defined(STBI_NO_PIC)
1704static stbi__uint32 stbi__get32be(stbi__context *s)
1706 stbi__uint32 z = stbi__get16be(s);
1707 return (z << 16) + stbi__get16be(s);
1711#if defined(STBI_NO_BMP) && defined(STBI_NO_TGA) && defined(STBI_NO_GIF)
1714static int stbi__get16le(stbi__context *s)
1716 int z = stbi__get8(s);
1717 return z + (stbi__get8(s) << 8);
1722static stbi__uint32 stbi__get32le(stbi__context *s)
1724 stbi__uint32 z = stbi__get16le(s);
1725 z += (stbi__uint32)stbi__get16le(s) << 16;
1730#define STBI__BYTECAST(x) ((stbi_uc) ((x) & 255))
1732#if defined(STBI_NO_JPEG) && defined(STBI_NO_PNG) && defined(STBI_NO_BMP) && defined(STBI_NO_PSD) && defined(STBI_NO_TGA) && defined(STBI_NO_GIF) && defined(STBI_NO_PIC) && defined(STBI_NO_PNM)
1746static stbi_uc stbi__compute_y(
int r,
int g,
int b)
1748 return (stbi_uc) (((r*77) + (g*150) + (29*b)) >> 8);
1752#if defined(STBI_NO_PNG) && defined(STBI_NO_BMP) && defined(STBI_NO_PSD) && defined(STBI_NO_TGA) && defined(STBI_NO_GIF) && defined(STBI_NO_PIC) && defined(STBI_NO_PNM)
1755static unsigned char *stbi__convert_format(
unsigned char *data,
int img_n,
int req_comp,
unsigned int x,
unsigned int y)
1758 unsigned char *good;
1760 if (req_comp == img_n)
return data;
1761 STBI_ASSERT(req_comp >= 1 && req_comp <= 4);
1763 good = (
unsigned char *) stbi__malloc_mad3(req_comp, x, y, 0);
1766 return stbi__errpuc(
"outofmem",
"Out of memory");
1769 for (j=0; j < (int) y; ++j) {
1770 unsigned char *src = data + j * x * img_n ;
1771 unsigned char *dest = good + j * x * req_comp;
1773 #define STBI__COMBO(a,b) ((a)*8+(b))
1774 #define STBI__CASE(a,b) case STBI__COMBO(a,b): for(i=x-1; i >= 0; --i, src += a, dest += b)
1777 switch (STBI__COMBO(img_n, req_comp)) {
1778 STBI__CASE(1,2) { dest[0]=src[0]; dest[1]=255; }
break;
1779 STBI__CASE(1,3) { dest[0]=dest[1]=dest[2]=src[0]; }
break;
1780 STBI__CASE(1,4) { dest[0]=dest[1]=dest[2]=src[0]; dest[3]=255; }
break;
1781 STBI__CASE(2,1) { dest[0]=src[0]; }
break;
1782 STBI__CASE(2,3) { dest[0]=dest[1]=dest[2]=src[0]; }
break;
1783 STBI__CASE(2,4) { dest[0]=dest[1]=dest[2]=src[0]; dest[3]=src[1]; }
break;
1784 STBI__CASE(3,4) { dest[0]=src[0];dest[1]=src[1];dest[2]=src[2];dest[3]=255; }
break;
1785 STBI__CASE(3,1) { dest[0]=stbi__compute_y(src[0],src[1],src[2]); }
break;
1786 STBI__CASE(3,2) { dest[0]=stbi__compute_y(src[0],src[1],src[2]); dest[1] = 255; }
break;
1787 STBI__CASE(4,1) { dest[0]=stbi__compute_y(src[0],src[1],src[2]); }
break;
1788 STBI__CASE(4,2) { dest[0]=stbi__compute_y(src[0],src[1],src[2]); dest[1] = src[3]; }
break;
1789 STBI__CASE(4,3) { dest[0]=src[0];dest[1]=src[1];dest[2]=src[2]; }
break;
1790 default: STBI_ASSERT(0); STBI_FREE(data); STBI_FREE(good);
return stbi__errpuc(
"unsupported",
"Unsupported format conversion");
1800#if defined(STBI_NO_PNG) && defined(STBI_NO_PSD)
1803static stbi__uint16 stbi__compute_y_16(
int r,
int g,
int b)
1805 return (stbi__uint16) (((r*77) + (g*150) + (29*b)) >> 8);
1809#if defined(STBI_NO_PNG) && defined(STBI_NO_PSD)
1812static stbi__uint16 *stbi__convert_format16(stbi__uint16 *data,
int img_n,
int req_comp,
unsigned int x,
unsigned int y)
1817 if (req_comp == img_n)
return data;
1818 STBI_ASSERT(req_comp >= 1 && req_comp <= 4);
1820 good = (stbi__uint16 *) stbi__malloc(req_comp * x * y * 2);
1823 return (stbi__uint16 *) stbi__errpuc(
"outofmem",
"Out of memory");
1826 for (j=0; j < (int) y; ++j) {
1827 stbi__uint16 *src = data + j * x * img_n ;
1828 stbi__uint16 *dest = good + j * x * req_comp;
1830 #define STBI__COMBO(a,b) ((a)*8+(b))
1831 #define STBI__CASE(a,b) case STBI__COMBO(a,b): for(i=x-1; i >= 0; --i, src += a, dest += b)
1834 switch (STBI__COMBO(img_n, req_comp)) {
1835 STBI__CASE(1,2) { dest[0]=src[0]; dest[1]=0xffff; }
break;
1836 STBI__CASE(1,3) { dest[0]=dest[1]=dest[2]=src[0]; }
break;
1837 STBI__CASE(1,4) { dest[0]=dest[1]=dest[2]=src[0]; dest[3]=0xffff; }
break;
1838 STBI__CASE(2,1) { dest[0]=src[0]; }
break;
1839 STBI__CASE(2,3) { dest[0]=dest[1]=dest[2]=src[0]; }
break;
1840 STBI__CASE(2,4) { dest[0]=dest[1]=dest[2]=src[0]; dest[3]=src[1]; }
break;
1841 STBI__CASE(3,4) { dest[0]=src[0];dest[1]=src[1];dest[2]=src[2];dest[3]=0xffff; }
break;
1842 STBI__CASE(3,1) { dest[0]=stbi__compute_y_16(src[0],src[1],src[2]); }
break;
1843 STBI__CASE(3,2) { dest[0]=stbi__compute_y_16(src[0],src[1],src[2]); dest[1] = 0xffff; }
break;
1844 STBI__CASE(4,1) { dest[0]=stbi__compute_y_16(src[0],src[1],src[2]); }
break;
1845 STBI__CASE(4,2) { dest[0]=stbi__compute_y_16(src[0],src[1],src[2]); dest[1] = src[3]; }
break;
1846 STBI__CASE(4,3) { dest[0]=src[0];dest[1]=src[1];dest[2]=src[2]; }
break;
1847 default: STBI_ASSERT(0); STBI_FREE(data); STBI_FREE(good);
return (stbi__uint16*) stbi__errpuc(
"unsupported",
"Unsupported format conversion");
1857#ifndef STBI_NO_LINEAR
1858static float *stbi__ldr_to_hdr(stbi_uc *data,
int x,
int y,
int comp)
1862 if (!data)
return NULL;
1863 output = (
float *) stbi__malloc_mad4(x, y, comp,
sizeof(
float), 0);
1864 if (output == NULL) { STBI_FREE(data);
return stbi__errpf(
"outofmem",
"Out of memory"); }
1866 if (comp & 1) n = comp;
else n = comp-1;
1867 for (i=0; i < x*y; ++i) {
1868 for (k=0; k < n; ++k) {
1869 output[i*comp + k] = (float) (pow(data[i*comp+k]/255.0f, stbi__l2h_gamma) * stbi__l2h_scale);
1873 for (i=0; i < x*y; ++i) {
1874 output[i*comp + n] = data[i*comp + n]/255.0f;
1883#define stbi__float2int(x) ((int) (x))
1884static stbi_uc *stbi__hdr_to_ldr(
float *data,
int x,
int y,
int comp)
1888 if (!data)
return NULL;
1889 output = (stbi_uc *) stbi__malloc_mad3(x, y, comp, 0);
1890 if (output == NULL) { STBI_FREE(data);
return stbi__errpuc(
"outofmem",
"Out of memory"); }
1892 if (comp & 1) n = comp;
else n = comp-1;
1893 for (i=0; i < x*y; ++i) {
1894 for (k=0; k < n; ++k) {
1895 float z = (float) pow(data[i*comp+k]*stbi__h2l_scale_i, stbi__h2l_gamma_i) * 255 + 0.5f;
1897 if (z > 255) z = 255;
1898 output[i*comp + k] = (stbi_uc) stbi__float2int(z);
1901 float z = data[i*comp+k] * 255 + 0.5f;
1903 if (z > 255) z = 255;
1904 output[i*comp + k] = (stbi_uc) stbi__float2int(z);
1940 stbi_uc fast[1 << FAST_BITS];
1942 stbi__uint16 code[256];
1943 stbi_uc values[256];
1945 unsigned int maxcode[18];
1952 stbi__huffman huff_dc[4];
1953 stbi__huffman huff_ac[4];
1954 stbi__uint16 dequant[4][64];
1955 stbi__int16 fast_ac[4][1 << FAST_BITS];
1958 int img_h_max, img_v_max;
1959 int img_mcu_x, img_mcu_y;
1960 int img_mcu_w, img_mcu_h;
1973 void *raw_data, *raw_coeff;
1976 int coeff_w, coeff_h;
1979 stbi__uint32 code_buffer;
1981 unsigned char marker;
1991 int app14_color_transform;
1994 int scan_n, order[4];
1995 int restart_interval, todo;
1998 void (*idct_block_kernel)(stbi_uc *out,
int out_stride,
short data[64]);
1999 void (*YCbCr_to_RGB_kernel)(stbi_uc *out,
const stbi_uc *y,
const stbi_uc *pcb,
const stbi_uc *pcr,
int count,
int step);
2000 stbi_uc *(*resample_row_hv_2_kernel)(stbi_uc *out, stbi_uc *in_near, stbi_uc *in_far,
int w,
int hs);
2003static int stbi__build_huffman(stbi__huffman *h,
int *count)
2008 for (i=0; i < 16; ++i) {
2009 for (j=0; j < count[i]; ++j) {
2010 h->size[k++] = (stbi_uc) (i+1);
2011 if(k >= 257)
return stbi__err(
"bad size list",
"Corrupt JPEG");
2019 for(j=1; j <= 16; ++j) {
2021 h->delta[j] = k - code;
2022 if (h->size[k] == j) {
2023 while (h->size[k] == j)
2024 h->code[k++] = (stbi__uint16) (code++);
2025 if (code-1 >= (1u << j))
return stbi__err(
"bad code lengths",
"Corrupt JPEG");
2028 h->maxcode[j] = code << (16-j);
2031 h->maxcode[j] = 0xffffffff;
2034 memset(h->fast, 255, 1 << FAST_BITS);
2035 for (i=0; i < k; ++i) {
2037 if (s <= FAST_BITS) {
2038 int c = h->code[i] << (FAST_BITS-s);
2039 int m = 1 << (FAST_BITS-s);
2040 for (j=0; j < m; ++j) {
2041 h->fast[c+j] = (stbi_uc) i;
2050static void stbi__build_fast_ac(stbi__int16 *fast_ac, stbi__huffman *h)
2053 for (i=0; i < (1 << FAST_BITS); ++i) {
2054 stbi_uc fast = h->fast[i];
2057 int rs = h->values[fast];
2058 int run = (rs >> 4) & 15;
2059 int magbits = rs & 15;
2060 int len = h->size[fast];
2062 if (magbits && len + magbits <= FAST_BITS) {
2064 int k = ((i << len) & ((1 << FAST_BITS) - 1)) >> (FAST_BITS - magbits);
2065 int m = 1 << (magbits - 1);
2066 if (k < m) k += (~0U << magbits) + 1;
2068 if (k >= -128 && k <= 127)
2069 fast_ac[i] = (stbi__int16) ((k * 256) + (run * 16) + (len + magbits));
2075static void stbi__grow_buffer_unsafe(stbi__jpeg *j)
2078 unsigned int b = j->nomore ? 0 : stbi__get8(j->s);
2080 int c = stbi__get8(j->s);
2081 while (c == 0xff) c = stbi__get8(j->s);
2083 j->marker = (
unsigned char) c;
2088 j->code_buffer |= b << (24 - j->code_bits);
2090 }
while (j->code_bits <= 24);
2094static const stbi__uint32 stbi__bmask[17]={0,1,3,7,15,31,63,127,255,511,1023,2047,4095,8191,16383,32767,65535};
2097stbi_inline
static int stbi__jpeg_huff_decode(stbi__jpeg *j, stbi__huffman *h)
2102 if (j->code_bits < 16) stbi__grow_buffer_unsafe(j);
2106 c = (j->code_buffer >> (32 - FAST_BITS)) & ((1 << FAST_BITS)-1);
2110 if (s > j->code_bits)
2112 j->code_buffer <<= s;
2114 return h->values[k];
2123 temp = j->code_buffer >> 16;
2124 for (k=FAST_BITS+1 ; ; ++k)
2125 if (temp < h->maxcode[k])
2133 if (k > j->code_bits)
2137 c = ((j->code_buffer >> (32 - k)) & stbi__bmask[k]) + h->delta[k];
2138 if(c < 0 || c >= 256)
2140 STBI_ASSERT((((j->code_buffer) >> (32 - h->size[c])) & stbi__bmask[h->size[c]]) == h->code[c]);
2144 j->code_buffer <<= k;
2145 return h->values[c];
2149static const int stbi__jbias[16] = {0,-1,-3,-7,-15,-31,-63,-127,-255,-511,-1023,-2047,-4095,-8191,-16383,-32767};
2153stbi_inline
static int stbi__extend_receive(stbi__jpeg *j,
int n)
2157 if (j->code_bits < n) stbi__grow_buffer_unsafe(j);
2158 if (j->code_bits < n)
return 0;
2160 sgn = j->code_buffer >> 31;
2161 k = stbi_lrot(j->code_buffer, n);
2162 j->code_buffer = k & ~stbi__bmask[n];
2163 k &= stbi__bmask[n];
2165 return k + (stbi__jbias[n] & (sgn - 1));
2169stbi_inline
static int stbi__jpeg_get_bits(stbi__jpeg *j,
int n)
2172 if (j->code_bits < n) stbi__grow_buffer_unsafe(j);
2173 if (j->code_bits < n)
return 0;
2174 k = stbi_lrot(j->code_buffer, n);
2175 j->code_buffer = k & ~stbi__bmask[n];
2176 k &= stbi__bmask[n];
2181stbi_inline
static int stbi__jpeg_get_bit(stbi__jpeg *j)
2184 if (j->code_bits < 1) stbi__grow_buffer_unsafe(j);
2185 if (j->code_bits < 1)
return 0;
2187 j->code_buffer <<= 1;
2189 return k & 0x80000000;
2194static const stbi_uc stbi__jpeg_dezigzag[64+15] =
2196 0, 1, 8, 16, 9, 2, 3, 10,
2197 17, 24, 32, 25, 18, 11, 4, 5,
2198 12, 19, 26, 33, 40, 48, 41, 34,
2199 27, 20, 13, 6, 7, 14, 21, 28,
2200 35, 42, 49, 56, 57, 50, 43, 36,
2201 29, 22, 15, 23, 30, 37, 44, 51,
2202 58, 59, 52, 45, 38, 31, 39, 46,
2203 53, 60, 61, 54, 47, 55, 62, 63,
2205 63, 63, 63, 63, 63, 63, 63, 63,
2206 63, 63, 63, 63, 63, 63, 63
2210static int stbi__jpeg_decode_block(stbi__jpeg *j,
short data[64], stbi__huffman *hdc, stbi__huffman *hac, stbi__int16 *fac,
int b, stbi__uint16 *dequant)
2215 if (j->code_bits < 16) stbi__grow_buffer_unsafe(j);
2216 t = stbi__jpeg_huff_decode(j, hdc);
2217 if (t < 0 || t > 15)
return stbi__err(
"bad huffman code",
"Corrupt JPEG");
2220 memset(data,0,64*
sizeof(data[0]));
2222 diff = t ? stbi__extend_receive(j, t) : 0;
2223 if (!stbi__addints_valid(j->img_comp[b].dc_pred, diff))
return stbi__err(
"bad delta",
"Corrupt JPEG");
2224 dc = j->img_comp[b].dc_pred + diff;
2225 j->img_comp[b].dc_pred = dc;
2226 if (!stbi__mul2shorts_valid(dc, dequant[0]))
return stbi__err(
"can't merge dc and ac",
"Corrupt JPEG");
2227 data[0] = (short) (dc * dequant[0]);
2234 if (j->code_bits < 16) stbi__grow_buffer_unsafe(j);
2235 c = (j->code_buffer >> (32 - FAST_BITS)) & ((1 << FAST_BITS)-1);
2240 if (s > j->code_bits)
return stbi__err(
"bad huffman code",
"Combined length longer than code bits available");
2241 j->code_buffer <<= s;
2244 zig = stbi__jpeg_dezigzag[k++];
2245 data[zig] = (short) ((r >> 8) * dequant[zig]);
2247 int rs = stbi__jpeg_huff_decode(j, hac);
2248 if (rs < 0)
return stbi__err(
"bad huffman code",
"Corrupt JPEG");
2252 if (rs != 0xf0)
break;
2257 zig = stbi__jpeg_dezigzag[k++];
2258 data[zig] = (short) (stbi__extend_receive(j,s) * dequant[zig]);
2265static int stbi__jpeg_decode_block_prog_dc(stbi__jpeg *j,
short data[64], stbi__huffman *hdc,
int b)
2269 if (j->spec_end != 0)
return stbi__err(
"can't merge dc and ac",
"Corrupt JPEG");
2271 if (j->code_bits < 16) stbi__grow_buffer_unsafe(j);
2273 if (j->succ_high == 0) {
2275 memset(data,0,64*
sizeof(data[0]));
2276 t = stbi__jpeg_huff_decode(j, hdc);
2277 if (t < 0 || t > 15)
return stbi__err(
"can't merge dc and ac",
"Corrupt JPEG");
2278 diff = t ? stbi__extend_receive(j, t) : 0;
2280 if (!stbi__addints_valid(j->img_comp[b].dc_pred, diff))
return stbi__err(
"bad delta",
"Corrupt JPEG");
2281 dc = j->img_comp[b].dc_pred + diff;
2282 j->img_comp[b].dc_pred = dc;
2283 if (!stbi__mul2shorts_valid(dc, 1 << j->succ_low))
return stbi__err(
"can't merge dc and ac",
"Corrupt JPEG");
2284 data[0] = (short) (dc * (1 << j->succ_low));
2287 if (stbi__jpeg_get_bit(j))
2288 data[0] += (short) (1 << j->succ_low);
2295static int stbi__jpeg_decode_block_prog_ac(stbi__jpeg *j,
short data[64], stbi__huffman *hac, stbi__int16 *fac)
2298 if (j->spec_start == 0)
return stbi__err(
"can't merge dc and ac",
"Corrupt JPEG");
2300 if (j->succ_high == 0) {
2301 int shift = j->succ_low;
2312 if (j->code_bits < 16) stbi__grow_buffer_unsafe(j);
2313 c = (j->code_buffer >> (32 - FAST_BITS)) & ((1 << FAST_BITS)-1);
2318 if (s > j->code_bits)
return stbi__err(
"bad huffman code",
"Combined length longer than code bits available");
2319 j->code_buffer <<= s;
2321 zig = stbi__jpeg_dezigzag[k++];
2322 data[zig] = (short) ((r >> 8) * (1 << shift));
2324 int rs = stbi__jpeg_huff_decode(j, hac);
2325 if (rs < 0)
return stbi__err(
"bad huffman code",
"Corrupt JPEG");
2330 j->eob_run = (1 << r);
2332 j->eob_run += stbi__jpeg_get_bits(j, r);
2339 zig = stbi__jpeg_dezigzag[k++];
2340 data[zig] = (short) (stbi__extend_receive(j,s) * (1 << shift));
2343 }
while (k <= j->spec_end);
2347 short bit = (short) (1 << j->succ_low);
2351 for (k = j->spec_start; k <= j->spec_end; ++k) {
2352 short *p = &data[stbi__jpeg_dezigzag[k]];
2354 if (stbi__jpeg_get_bit(j))
2355 if ((*p & bit)==0) {
2366 int rs = stbi__jpeg_huff_decode(j, hac);
2367 if (rs < 0)
return stbi__err(
"bad huffman code",
"Corrupt JPEG");
2372 j->eob_run = (1 << r) - 1;
2374 j->eob_run += stbi__jpeg_get_bits(j, r);
2382 if (s != 1)
return stbi__err(
"bad huffman code",
"Corrupt JPEG");
2384 if (stbi__jpeg_get_bit(j))
2391 while (k <= j->spec_end) {
2392 short *p = &data[stbi__jpeg_dezigzag[k++]];
2394 if (stbi__jpeg_get_bit(j))
2395 if ((*p & bit)==0) {
2409 }
while (k <= j->spec_end);
2416stbi_inline
static stbi_uc stbi__clamp(
int x)
2419 if ((
unsigned int) x > 255) {
2420 if (x < 0)
return 0;
2421 if (x > 255)
return 255;
2426#define stbi__f2f(x) ((int) (((x) * 4096 + 0.5)))
2427#define stbi__fsh(x) ((x) * 4096)
2430#define STBI__IDCT_1D(s0,s1,s2,s3,s4,s5,s6,s7) \
2431 int t0,t1,t2,t3,p1,p2,p3,p4,p5,x0,x1,x2,x3; \
2434 p1 = (p2+p3) * stbi__f2f(0.5411961f); \
2435 t2 = p1 + p3*stbi__f2f(-1.847759065f); \
2436 t3 = p1 + p2*stbi__f2f( 0.765366865f); \
2439 t0 = stbi__fsh(p2+p3); \
2440 t1 = stbi__fsh(p2-p3); \
2453 p5 = (p3+p4)*stbi__f2f( 1.175875602f); \
2454 t0 = t0*stbi__f2f( 0.298631336f); \
2455 t1 = t1*stbi__f2f( 2.053119869f); \
2456 t2 = t2*stbi__f2f( 3.072711026f); \
2457 t3 = t3*stbi__f2f( 1.501321110f); \
2458 p1 = p5 + p1*stbi__f2f(-0.899976223f); \
2459 p2 = p5 + p2*stbi__f2f(-2.562915447f); \
2460 p3 = p3*stbi__f2f(-1.961570560f); \
2461 p4 = p4*stbi__f2f(-0.390180644f); \
2467static void stbi__idct_block(stbi_uc *out,
int out_stride,
short data[64])
2469 int i,val[64],*v=val;
2474 for (i=0; i < 8; ++i,++d, ++v) {
2476 if (d[ 8]==0 && d[16]==0 && d[24]==0 && d[32]==0
2477 && d[40]==0 && d[48]==0 && d[56]==0) {
2482 int dcterm = d[0]*4;
2483 v[0] = v[8] = v[16] = v[24] = v[32] = v[40] = v[48] = v[56] = dcterm;
2485 STBI__IDCT_1D(d[ 0],d[ 8],d[16],d[24],d[32],d[40],d[48],d[56])
2488 x0 += 512; x1 += 512; x2 += 512; x3 += 512;
2489 v[ 0] = (x0+t3) >> 10;
2490 v[56] = (x0-t3) >> 10;
2491 v[ 8] = (x1+t2) >> 10;
2492 v[48] = (x1-t2) >> 10;
2493 v[16] = (x2+t1) >> 10;
2494 v[40] = (x2-t1) >> 10;
2495 v[24] = (x3+t0) >> 10;
2496 v[32] = (x3-t0) >> 10;
2500 for (i=0, v=val, o=out; i < 8; ++i,v+=8,o+=out_stride) {
2502 STBI__IDCT_1D(v[0],v[1],v[2],v[3],v[4],v[5],v[6],v[7])
2509 x0 += 65536 + (128<<17);
2510 x1 += 65536 + (128<<17);
2511 x2 += 65536 + (128<<17);
2512 x3 += 65536 + (128<<17);
2515 o[0] = stbi__clamp((x0+t3) >> 17);
2516 o[7] = stbi__clamp((x0-t3) >> 17);
2517 o[1] = stbi__clamp((x1+t2) >> 17);
2518 o[6] = stbi__clamp((x1-t2) >> 17);
2519 o[2] = stbi__clamp((x2+t1) >> 17);
2520 o[5] = stbi__clamp((x2-t1) >> 17);
2521 o[3] = stbi__clamp((x3+t0) >> 17);
2522 o[4] = stbi__clamp((x3-t0) >> 17);
2530static void stbi__idct_simd(stbi_uc *out,
int out_stride,
short data[64])
2533 __m128i row0, row1, row2, row3, row4, row5, row6, row7;
2537 #define dct_const(x,y) _mm_setr_epi16((x),(y),(x),(y),(x),(y),(x),(y))
2541 #define dct_rot(out0,out1, x,y,c0,c1) \
2542 __m128i c0##lo = _mm_unpacklo_epi16((x),(y)); \
2543 __m128i c0##hi = _mm_unpackhi_epi16((x),(y)); \
2544 __m128i out0##_l = _mm_madd_epi16(c0##lo, c0); \
2545 __m128i out0##_h = _mm_madd_epi16(c0##hi, c0); \
2546 __m128i out1##_l = _mm_madd_epi16(c0##lo, c1); \
2547 __m128i out1##_h = _mm_madd_epi16(c0##hi, c1)
2550 #define dct_widen(out, in) \
2551 __m128i out##_l = _mm_srai_epi32(_mm_unpacklo_epi16(_mm_setzero_si128(), (in)), 4); \
2552 __m128i out##_h = _mm_srai_epi32(_mm_unpackhi_epi16(_mm_setzero_si128(), (in)), 4)
2555 #define dct_wadd(out, a, b) \
2556 __m128i out##_l = _mm_add_epi32(a##_l, b##_l); \
2557 __m128i out##_h = _mm_add_epi32(a##_h, b##_h)
2560 #define dct_wsub(out, a, b) \
2561 __m128i out##_l = _mm_sub_epi32(a##_l, b##_l); \
2562 __m128i out##_h = _mm_sub_epi32(a##_h, b##_h)
2565 #define dct_bfly32o(out0, out1, a,b,bias,s) \
2567 __m128i abiased_l = _mm_add_epi32(a##_l, bias); \
2568 __m128i abiased_h = _mm_add_epi32(a##_h, bias); \
2569 dct_wadd(sum, abiased, b); \
2570 dct_wsub(dif, abiased, b); \
2571 out0 = _mm_packs_epi32(_mm_srai_epi32(sum_l, s), _mm_srai_epi32(sum_h, s)); \
2572 out1 = _mm_packs_epi32(_mm_srai_epi32(dif_l, s), _mm_srai_epi32(dif_h, s)); \
2576 #define dct_interleave8(a, b) \
2578 a = _mm_unpacklo_epi8(a, b); \
2579 b = _mm_unpackhi_epi8(tmp, b)
2582 #define dct_interleave16(a, b) \
2584 a = _mm_unpacklo_epi16(a, b); \
2585 b = _mm_unpackhi_epi16(tmp, b)
2587 #define dct_pass(bias,shift) \
2590 dct_rot(t2e,t3e, row2,row6, rot0_0,rot0_1); \
2591 __m128i sum04 = _mm_add_epi16(row0, row4); \
2592 __m128i dif04 = _mm_sub_epi16(row0, row4); \
2593 dct_widen(t0e, sum04); \
2594 dct_widen(t1e, dif04); \
2595 dct_wadd(x0, t0e, t3e); \
2596 dct_wsub(x3, t0e, t3e); \
2597 dct_wadd(x1, t1e, t2e); \
2598 dct_wsub(x2, t1e, t2e); \
2600 dct_rot(y0o,y2o, row7,row3, rot2_0,rot2_1); \
2601 dct_rot(y1o,y3o, row5,row1, rot3_0,rot3_1); \
2602 __m128i sum17 = _mm_add_epi16(row1, row7); \
2603 __m128i sum35 = _mm_add_epi16(row3, row5); \
2604 dct_rot(y4o,y5o, sum17,sum35, rot1_0,rot1_1); \
2605 dct_wadd(x4, y0o, y4o); \
2606 dct_wadd(x5, y1o, y5o); \
2607 dct_wadd(x6, y2o, y5o); \
2608 dct_wadd(x7, y3o, y4o); \
2609 dct_bfly32o(row0,row7, x0,x7,bias,shift); \
2610 dct_bfly32o(row1,row6, x1,x6,bias,shift); \
2611 dct_bfly32o(row2,row5, x2,x5,bias,shift); \
2612 dct_bfly32o(row3,row4, x3,x4,bias,shift); \
2615 __m128i rot0_0 = dct_const(stbi__f2f(0.5411961f), stbi__f2f(0.5411961f) + stbi__f2f(-1.847759065f));
2616 __m128i rot0_1 = dct_const(stbi__f2f(0.5411961f) + stbi__f2f( 0.765366865f), stbi__f2f(0.5411961f));
2617 __m128i rot1_0 = dct_const(stbi__f2f(1.175875602f) + stbi__f2f(-0.899976223f), stbi__f2f(1.175875602f));
2618 __m128i rot1_1 = dct_const(stbi__f2f(1.175875602f), stbi__f2f(1.175875602f) + stbi__f2f(-2.562915447f));
2619 __m128i rot2_0 = dct_const(stbi__f2f(-1.961570560f) + stbi__f2f( 0.298631336f), stbi__f2f(-1.961570560f));
2620 __m128i rot2_1 = dct_const(stbi__f2f(-1.961570560f), stbi__f2f(-1.961570560f) + stbi__f2f( 3.072711026f));
2621 __m128i rot3_0 = dct_const(stbi__f2f(-0.390180644f) + stbi__f2f( 2.053119869f), stbi__f2f(-0.390180644f));
2622 __m128i rot3_1 = dct_const(stbi__f2f(-0.390180644f), stbi__f2f(-0.390180644f) + stbi__f2f( 1.501321110f));
2625 __m128i bias_0 = _mm_set1_epi32(512);
2626 __m128i bias_1 = _mm_set1_epi32(65536 + (128<<17));
2629 row0 = _mm_load_si128((
const __m128i *) (data + 0*8));
2630 row1 = _mm_load_si128((
const __m128i *) (data + 1*8));
2631 row2 = _mm_load_si128((
const __m128i *) (data + 2*8));
2632 row3 = _mm_load_si128((
const __m128i *) (data + 3*8));
2633 row4 = _mm_load_si128((
const __m128i *) (data + 4*8));
2634 row5 = _mm_load_si128((
const __m128i *) (data + 5*8));
2635 row6 = _mm_load_si128((
const __m128i *) (data + 6*8));
2636 row7 = _mm_load_si128((
const __m128i *) (data + 7*8));
2639 dct_pass(bias_0, 10);
2643 dct_interleave16(row0, row4);
2644 dct_interleave16(row1, row5);
2645 dct_interleave16(row2, row6);
2646 dct_interleave16(row3, row7);
2649 dct_interleave16(row0, row2);
2650 dct_interleave16(row1, row3);
2651 dct_interleave16(row4, row6);
2652 dct_interleave16(row5, row7);
2655 dct_interleave16(row0, row1);
2656 dct_interleave16(row2, row3);
2657 dct_interleave16(row4, row5);
2658 dct_interleave16(row6, row7);
2662 dct_pass(bias_1, 17);
2666 __m128i p0 = _mm_packus_epi16(row0, row1);
2667 __m128i p1 = _mm_packus_epi16(row2, row3);
2668 __m128i p2 = _mm_packus_epi16(row4, row5);
2669 __m128i p3 = _mm_packus_epi16(row6, row7);
2672 dct_interleave8(p0, p2);
2673 dct_interleave8(p1, p3);
2676 dct_interleave8(p0, p1);
2677 dct_interleave8(p2, p3);
2680 dct_interleave8(p0, p2);
2681 dct_interleave8(p1, p3);
2684 _mm_storel_epi64((__m128i *) out, p0); out += out_stride;
2685 _mm_storel_epi64((__m128i *) out, _mm_shuffle_epi32(p0, 0x4e)); out += out_stride;
2686 _mm_storel_epi64((__m128i *) out, p2); out += out_stride;
2687 _mm_storel_epi64((__m128i *) out, _mm_shuffle_epi32(p2, 0x4e)); out += out_stride;
2688 _mm_storel_epi64((__m128i *) out, p1); out += out_stride;
2689 _mm_storel_epi64((__m128i *) out, _mm_shuffle_epi32(p1, 0x4e)); out += out_stride;
2690 _mm_storel_epi64((__m128i *) out, p3); out += out_stride;
2691 _mm_storel_epi64((__m128i *) out, _mm_shuffle_epi32(p3, 0x4e));
2700#undef dct_interleave8
2701#undef dct_interleave16
2711static void stbi__idct_simd(stbi_uc *out,
int out_stride,
short data[64])
2713 int16x8_t row0, row1, row2, row3, row4, row5, row6, row7;
2715 int16x4_t rot0_0 = vdup_n_s16(stbi__f2f(0.5411961f));
2716 int16x4_t rot0_1 = vdup_n_s16(stbi__f2f(-1.847759065f));
2717 int16x4_t rot0_2 = vdup_n_s16(stbi__f2f( 0.765366865f));
2718 int16x4_t rot1_0 = vdup_n_s16(stbi__f2f( 1.175875602f));
2719 int16x4_t rot1_1 = vdup_n_s16(stbi__f2f(-0.899976223f));
2720 int16x4_t rot1_2 = vdup_n_s16(stbi__f2f(-2.562915447f));
2721 int16x4_t rot2_0 = vdup_n_s16(stbi__f2f(-1.961570560f));
2722 int16x4_t rot2_1 = vdup_n_s16(stbi__f2f(-0.390180644f));
2723 int16x4_t rot3_0 = vdup_n_s16(stbi__f2f( 0.298631336f));
2724 int16x4_t rot3_1 = vdup_n_s16(stbi__f2f( 2.053119869f));
2725 int16x4_t rot3_2 = vdup_n_s16(stbi__f2f( 3.072711026f));
2726 int16x4_t rot3_3 = vdup_n_s16(stbi__f2f( 1.501321110f));
2728#define dct_long_mul(out, inq, coeff) \
2729 int32x4_t out##_l = vmull_s16(vget_low_s16(inq), coeff); \
2730 int32x4_t out##_h = vmull_s16(vget_high_s16(inq), coeff)
2732#define dct_long_mac(out, acc, inq, coeff) \
2733 int32x4_t out##_l = vmlal_s16(acc##_l, vget_low_s16(inq), coeff); \
2734 int32x4_t out##_h = vmlal_s16(acc##_h, vget_high_s16(inq), coeff)
2736#define dct_widen(out, inq) \
2737 int32x4_t out##_l = vshll_n_s16(vget_low_s16(inq), 12); \
2738 int32x4_t out##_h = vshll_n_s16(vget_high_s16(inq), 12)
2741#define dct_wadd(out, a, b) \
2742 int32x4_t out##_l = vaddq_s32(a##_l, b##_l); \
2743 int32x4_t out##_h = vaddq_s32(a##_h, b##_h)
2746#define dct_wsub(out, a, b) \
2747 int32x4_t out##_l = vsubq_s32(a##_l, b##_l); \
2748 int32x4_t out##_h = vsubq_s32(a##_h, b##_h)
2751#define dct_bfly32o(out0,out1, a,b,shiftop,s) \
2753 dct_wadd(sum, a, b); \
2754 dct_wsub(dif, a, b); \
2755 out0 = vcombine_s16(shiftop(sum_l, s), shiftop(sum_h, s)); \
2756 out1 = vcombine_s16(shiftop(dif_l, s), shiftop(dif_h, s)); \
2759#define dct_pass(shiftop, shift) \
2762 int16x8_t sum26 = vaddq_s16(row2, row6); \
2763 dct_long_mul(p1e, sum26, rot0_0); \
2764 dct_long_mac(t2e, p1e, row6, rot0_1); \
2765 dct_long_mac(t3e, p1e, row2, rot0_2); \
2766 int16x8_t sum04 = vaddq_s16(row0, row4); \
2767 int16x8_t dif04 = vsubq_s16(row0, row4); \
2768 dct_widen(t0e, sum04); \
2769 dct_widen(t1e, dif04); \
2770 dct_wadd(x0, t0e, t3e); \
2771 dct_wsub(x3, t0e, t3e); \
2772 dct_wadd(x1, t1e, t2e); \
2773 dct_wsub(x2, t1e, t2e); \
2775 int16x8_t sum15 = vaddq_s16(row1, row5); \
2776 int16x8_t sum17 = vaddq_s16(row1, row7); \
2777 int16x8_t sum35 = vaddq_s16(row3, row5); \
2778 int16x8_t sum37 = vaddq_s16(row3, row7); \
2779 int16x8_t sumodd = vaddq_s16(sum17, sum35); \
2780 dct_long_mul(p5o, sumodd, rot1_0); \
2781 dct_long_mac(p1o, p5o, sum17, rot1_1); \
2782 dct_long_mac(p2o, p5o, sum35, rot1_2); \
2783 dct_long_mul(p3o, sum37, rot2_0); \
2784 dct_long_mul(p4o, sum15, rot2_1); \
2785 dct_wadd(sump13o, p1o, p3o); \
2786 dct_wadd(sump24o, p2o, p4o); \
2787 dct_wadd(sump23o, p2o, p3o); \
2788 dct_wadd(sump14o, p1o, p4o); \
2789 dct_long_mac(x4, sump13o, row7, rot3_0); \
2790 dct_long_mac(x5, sump24o, row5, rot3_1); \
2791 dct_long_mac(x6, sump23o, row3, rot3_2); \
2792 dct_long_mac(x7, sump14o, row1, rot3_3); \
2793 dct_bfly32o(row0,row7, x0,x7,shiftop,shift); \
2794 dct_bfly32o(row1,row6, x1,x6,shiftop,shift); \
2795 dct_bfly32o(row2,row5, x2,x5,shiftop,shift); \
2796 dct_bfly32o(row3,row4, x3,x4,shiftop,shift); \
2800 row0 = vld1q_s16(data + 0*8);
2801 row1 = vld1q_s16(data + 1*8);
2802 row2 = vld1q_s16(data + 2*8);
2803 row3 = vld1q_s16(data + 3*8);
2804 row4 = vld1q_s16(data + 4*8);
2805 row5 = vld1q_s16(data + 5*8);
2806 row6 = vld1q_s16(data + 6*8);
2807 row7 = vld1q_s16(data + 7*8);
2810 row0 = vaddq_s16(row0, vsetq_lane_s16(1024, vdupq_n_s16(0), 0));
2813 dct_pass(vrshrn_n_s32, 10);
2819#define dct_trn16(x, y) { int16x8x2_t t = vtrnq_s16(x, y); x = t.val[0]; y = t.val[1]; }
2820#define dct_trn32(x, y) { int32x4x2_t t = vtrnq_s32(vreinterpretq_s32_s16(x), vreinterpretq_s32_s16(y)); x = vreinterpretq_s16_s32(t.val[0]); y = vreinterpretq_s16_s32(t.val[1]); }
2821#define dct_trn64(x, y) { int16x8_t x0 = x; int16x8_t y0 = y; x = vcombine_s16(vget_low_s16(x0), vget_low_s16(y0)); y = vcombine_s16(vget_high_s16(x0), vget_high_s16(y0)); }
2824 dct_trn16(row0, row1);
2825 dct_trn16(row2, row3);
2826 dct_trn16(row4, row5);
2827 dct_trn16(row6, row7);
2830 dct_trn32(row0, row2);
2831 dct_trn32(row1, row3);
2832 dct_trn32(row4, row6);
2833 dct_trn32(row5, row7);
2836 dct_trn64(row0, row4);
2837 dct_trn64(row1, row5);
2838 dct_trn64(row2, row6);
2839 dct_trn64(row3, row7);
2850 dct_pass(vshrn_n_s32, 16);
2854 uint8x8_t p0 = vqrshrun_n_s16(row0, 1);
2855 uint8x8_t p1 = vqrshrun_n_s16(row1, 1);
2856 uint8x8_t p2 = vqrshrun_n_s16(row2, 1);
2857 uint8x8_t p3 = vqrshrun_n_s16(row3, 1);
2858 uint8x8_t p4 = vqrshrun_n_s16(row4, 1);
2859 uint8x8_t p5 = vqrshrun_n_s16(row5, 1);
2860 uint8x8_t p6 = vqrshrun_n_s16(row6, 1);
2861 uint8x8_t p7 = vqrshrun_n_s16(row7, 1);
2864#define dct_trn8_8(x, y) { uint8x8x2_t t = vtrn_u8(x, y); x = t.val[0]; y = t.val[1]; }
2865#define dct_trn8_16(x, y) { uint16x4x2_t t = vtrn_u16(vreinterpret_u16_u8(x), vreinterpret_u16_u8(y)); x = vreinterpret_u8_u16(t.val[0]); y = vreinterpret_u8_u16(t.val[1]); }
2866#define dct_trn8_32(x, y) { uint32x2x2_t t = vtrn_u32(vreinterpret_u32_u8(x), vreinterpret_u32_u8(y)); x = vreinterpret_u8_u32(t.val[0]); y = vreinterpret_u8_u32(t.val[1]); }
2878 dct_trn8_16(p0, p2);
2879 dct_trn8_16(p1, p3);
2880 dct_trn8_16(p4, p6);
2881 dct_trn8_16(p5, p7);
2884 dct_trn8_32(p0, p4);
2885 dct_trn8_32(p1, p5);
2886 dct_trn8_32(p2, p6);
2887 dct_trn8_32(p3, p7);
2890 vst1_u8(out, p0); out += out_stride;
2891 vst1_u8(out, p1); out += out_stride;
2892 vst1_u8(out, p2); out += out_stride;
2893 vst1_u8(out, p3); out += out_stride;
2894 vst1_u8(out, p4); out += out_stride;
2895 vst1_u8(out, p5); out += out_stride;
2896 vst1_u8(out, p6); out += out_stride;
2915#define STBI__MARKER_none 0xff
2919static stbi_uc stbi__get_marker(stbi__jpeg *j)
2922 if (j->marker != STBI__MARKER_none) { x = j->marker; j->marker = STBI__MARKER_none;
return x; }
2923 x = stbi__get8(j->s);
2924 if (x != 0xff)
return STBI__MARKER_none;
2926 x = stbi__get8(j->s);
2932#define STBI__RESTART(x) ((x) >= 0xd0 && (x) <= 0xd7)
2936static void stbi__jpeg_reset(stbi__jpeg *j)
2941 j->img_comp[0].dc_pred = j->img_comp[1].dc_pred = j->img_comp[2].dc_pred = j->img_comp[3].dc_pred = 0;
2942 j->marker = STBI__MARKER_none;
2943 j->todo = j->restart_interval ? j->restart_interval : 0x7fffffff;
2949static int stbi__parse_entropy_coded_data(stbi__jpeg *z)
2951 stbi__jpeg_reset(z);
2952 if (!z->progressive) {
2953 if (z->scan_n == 1) {
2955 STBI_SIMD_ALIGN(
short, data[64]);
2956 int n = z->order[0];
2961 int w = (z->img_comp[n].x+7) >> 3;
2962 int h = (z->img_comp[n].y+7) >> 3;
2963 for (j=0; j < h; ++j) {
2964 for (i=0; i < w; ++i) {
2965 int ha = z->img_comp[n].ha;
2966 if (!stbi__jpeg_decode_block(z, data, z->huff_dc+z->img_comp[n].hd, z->huff_ac+ha, z->fast_ac[ha], n, z->dequant[z->img_comp[n].tq]))
return 0;
2967 z->idct_block_kernel(z->img_comp[n].data+z->img_comp[n].w2*j*8+i*8, z->img_comp[n].w2, data);
2969 if (--z->todo <= 0) {
2970 if (z->code_bits < 24) stbi__grow_buffer_unsafe(z);
2973 if (!STBI__RESTART(z->marker))
return 1;
2974 stbi__jpeg_reset(z);
2981 STBI_SIMD_ALIGN(
short, data[64]);
2982 for (j=0; j < z->img_mcu_y; ++j) {
2983 for (i=0; i < z->img_mcu_x; ++i) {
2985 for (k=0; k < z->scan_n; ++k) {
2986 int n = z->order[k];
2989 for (y=0; y < z->img_comp[n].v; ++y) {
2990 for (x=0; x < z->img_comp[n].h; ++x) {
2991 int x2 = (i*z->img_comp[n].h + x)*8;
2992 int y2 = (j*z->img_comp[n].v + y)*8;
2993 int ha = z->img_comp[n].ha;
2994 if (!stbi__jpeg_decode_block(z, data, z->huff_dc+z->img_comp[n].hd, z->huff_ac+ha, z->fast_ac[ha], n, z->dequant[z->img_comp[n].tq]))
return 0;
2995 z->idct_block_kernel(z->img_comp[n].data+z->img_comp[n].w2*y2+x2, z->img_comp[n].w2, data);
3001 if (--z->todo <= 0) {
3002 if (z->code_bits < 24) stbi__grow_buffer_unsafe(z);
3003 if (!STBI__RESTART(z->marker))
return 1;
3004 stbi__jpeg_reset(z);
3011 if (z->scan_n == 1) {
3013 int n = z->order[0];
3018 int w = (z->img_comp[n].x+7) >> 3;
3019 int h = (z->img_comp[n].y+7) >> 3;
3020 for (j=0; j < h; ++j) {
3021 for (i=0; i < w; ++i) {
3022 short *data = z->img_comp[n].coeff + 64 * (i + j * z->img_comp[n].coeff_w);
3023 if (z->spec_start == 0) {
3024 if (!stbi__jpeg_decode_block_prog_dc(z, data, &z->huff_dc[z->img_comp[n].hd], n))
3027 int ha = z->img_comp[n].ha;
3028 if (!stbi__jpeg_decode_block_prog_ac(z, data, &z->huff_ac[ha], z->fast_ac[ha]))
3032 if (--z->todo <= 0) {
3033 if (z->code_bits < 24) stbi__grow_buffer_unsafe(z);
3034 if (!STBI__RESTART(z->marker))
return 1;
3035 stbi__jpeg_reset(z);
3042 for (j=0; j < z->img_mcu_y; ++j) {
3043 for (i=0; i < z->img_mcu_x; ++i) {
3045 for (k=0; k < z->scan_n; ++k) {
3046 int n = z->order[k];
3049 for (y=0; y < z->img_comp[n].v; ++y) {
3050 for (x=0; x < z->img_comp[n].h; ++x) {
3051 int x2 = (i*z->img_comp[n].h + x);
3052 int y2 = (j*z->img_comp[n].v + y);
3053 short *data = z->img_comp[n].coeff + 64 * (x2 + y2 * z->img_comp[n].coeff_w);
3054 if (!stbi__jpeg_decode_block_prog_dc(z, data, &z->huff_dc[z->img_comp[n].hd], n))
3061 if (--z->todo <= 0) {
3062 if (z->code_bits < 24) stbi__grow_buffer_unsafe(z);
3063 if (!STBI__RESTART(z->marker))
return 1;
3064 stbi__jpeg_reset(z);
3073static void stbi__jpeg_dequantize(
short *data, stbi__uint16 *dequant)
3076 for (i=0; i < 64; ++i)
3077 data[i] *= dequant[i];
3080static void stbi__jpeg_finish(stbi__jpeg *z)
3082 if (z->progressive) {
3085 for (n=0; n < z->s->img_n; ++n) {
3086 int w = (z->img_comp[n].x+7) >> 3;
3087 int h = (z->img_comp[n].y+7) >> 3;
3088 for (j=0; j < h; ++j) {
3089 for (i=0; i < w; ++i) {
3090 short *data = z->img_comp[n].coeff + 64 * (i + j * z->img_comp[n].coeff_w);
3091 stbi__jpeg_dequantize(data, z->dequant[z->img_comp[n].tq]);
3092 z->idct_block_kernel(z->img_comp[n].data+z->img_comp[n].w2*j*8+i*8, z->img_comp[n].w2, data);
3099static int stbi__process_marker(stbi__jpeg *z,
int m)
3103 case STBI__MARKER_none:
3104 return stbi__err(
"expected marker",
"Corrupt JPEG");
3107 if (stbi__get16be(z->s) != 4)
return stbi__err(
"bad DRI len",
"Corrupt JPEG");
3108 z->restart_interval = stbi__get16be(z->s);
3112 L = stbi__get16be(z->s)-2;
3114 int q = stbi__get8(z->s);
3115 int p = q >> 4, sixteen = (p != 0);
3117 if (p != 0 && p != 1)
return stbi__err(
"bad DQT type",
"Corrupt JPEG");
3118 if (t > 3)
return stbi__err(
"bad DQT table",
"Corrupt JPEG");
3120 for (i=0; i < 64; ++i)
3121 z->dequant[t][stbi__jpeg_dezigzag[i]] = (stbi__uint16)(sixteen ? stbi__get16be(z->s) : stbi__get8(z->s));
3122 L -= (sixteen ? 129 : 65);
3127 L = stbi__get16be(z->s)-2;
3130 int sizes[16],i,n=0;
3131 int q = stbi__get8(z->s);
3134 if (tc > 1 || th > 3)
return stbi__err(
"bad DHT header",
"Corrupt JPEG");
3135 for (i=0; i < 16; ++i) {
3136 sizes[i] = stbi__get8(z->s);
3139 if(n > 256)
return stbi__err(
"bad DHT header",
"Corrupt JPEG");
3142 if (!stbi__build_huffman(z->huff_dc+th, sizes))
return 0;
3143 v = z->huff_dc[th].values;
3145 if (!stbi__build_huffman(z->huff_ac+th, sizes))
return 0;
3146 v = z->huff_ac[th].values;
3148 for (i=0; i < n; ++i)
3149 v[i] = stbi__get8(z->s);
3151 stbi__build_fast_ac(z->fast_ac[th], z->huff_ac + th);
3158 if ((m >= 0xE0 && m <= 0xEF) || m == 0xFE) {
3159 L = stbi__get16be(z->s);
3162 return stbi__err(
"bad COM len",
"Corrupt JPEG");
3164 return stbi__err(
"bad APP len",
"Corrupt JPEG");
3168 if (m == 0xE0 && L >= 5) {
3169 static const unsigned char tag[5] = {
'J',
'F',
'I',
'F',
'\0'};
3172 for (i=0; i < 5; ++i)
3173 if (stbi__get8(z->s) != tag[i])
3178 }
else if (m == 0xEE && L >= 12) {
3179 static const unsigned char tag[6] = {
'A',
'd',
'o',
'b',
'e',
'\0'};
3182 for (i=0; i < 6; ++i)
3183 if (stbi__get8(z->s) != tag[i])
3188 stbi__get16be(z->s);
3189 stbi__get16be(z->s);
3190 z->app14_color_transform = stbi__get8(z->s);
3195 stbi__skip(z->s, L);
3199 return stbi__err(
"unknown marker",
"Corrupt JPEG");
3203static int stbi__process_scan_header(stbi__jpeg *z)
3206 int Ls = stbi__get16be(z->s);
3207 z->scan_n = stbi__get8(z->s);
3208 if (z->scan_n < 1 || z->scan_n > 4 || z->scan_n > (
int) z->s->img_n)
return stbi__err(
"bad SOS component count",
"Corrupt JPEG");
3209 if (Ls != 6+2*z->scan_n)
return stbi__err(
"bad SOS len",
"Corrupt JPEG");
3210 for (i=0; i < z->scan_n; ++i) {
3211 int id = stbi__get8(z->s), which;
3212 int q = stbi__get8(z->s);
3213 for (which = 0; which < z->s->img_n; ++which)
3214 if (z->img_comp[which].id ==
id)
3216 if (which == z->s->img_n)
return 0;
3217 z->img_comp[which].hd = q >> 4;
if (z->img_comp[which].hd > 3)
return stbi__err(
"bad DC huff",
"Corrupt JPEG");
3218 z->img_comp[which].ha = q & 15;
if (z->img_comp[which].ha > 3)
return stbi__err(
"bad AC huff",
"Corrupt JPEG");
3219 z->order[i] = which;
3224 z->spec_start = stbi__get8(z->s);
3225 z->spec_end = stbi__get8(z->s);
3226 aa = stbi__get8(z->s);
3227 z->succ_high = (aa >> 4);
3228 z->succ_low = (aa & 15);
3229 if (z->progressive) {
3230 if (z->spec_start > 63 || z->spec_end > 63 || z->spec_start > z->spec_end || z->succ_high > 13 || z->succ_low > 13)
3231 return stbi__err(
"bad SOS",
"Corrupt JPEG");
3233 if (z->spec_start != 0)
return stbi__err(
"bad SOS",
"Corrupt JPEG");
3234 if (z->succ_high != 0 || z->succ_low != 0)
return stbi__err(
"bad SOS",
"Corrupt JPEG");
3242static int stbi__free_jpeg_components(stbi__jpeg *z,
int ncomp,
int why)
3245 for (i=0; i < ncomp; ++i) {
3246 if (z->img_comp[i].raw_data) {
3247 STBI_FREE(z->img_comp[i].raw_data);
3248 z->img_comp[i].raw_data = NULL;
3249 z->img_comp[i].data = NULL;
3251 if (z->img_comp[i].raw_coeff) {
3252 STBI_FREE(z->img_comp[i].raw_coeff);
3253 z->img_comp[i].raw_coeff = 0;
3254 z->img_comp[i].coeff = 0;
3256 if (z->img_comp[i].linebuf) {
3257 STBI_FREE(z->img_comp[i].linebuf);
3258 z->img_comp[i].linebuf = NULL;
3264static int stbi__process_frame_header(stbi__jpeg *z,
int scan)
3266 stbi__context *s = z->s;
3267 int Lf,p,i,q, h_max=1,v_max=1,c;
3268 Lf = stbi__get16be(s);
if (Lf < 11)
return stbi__err(
"bad SOF len",
"Corrupt JPEG");
3269 p = stbi__get8(s);
if (p != 8)
return stbi__err(
"only 8-bit",
"JPEG format not supported: 8-bit only");
3270 s->img_y = stbi__get16be(s);
if (s->img_y == 0)
return stbi__err(
"no header height",
"JPEG format not supported: delayed height");
3271 s->img_x = stbi__get16be(s);
if (s->img_x == 0)
return stbi__err(
"0 width",
"Corrupt JPEG");
3272 if (s->img_y > STBI_MAX_DIMENSIONS)
return stbi__err(
"too large",
"Very large image (corrupt?)");
3273 if (s->img_x > STBI_MAX_DIMENSIONS)
return stbi__err(
"too large",
"Very large image (corrupt?)");
3275 if (c != 3 && c != 1 && c != 4)
return stbi__err(
"bad component count",
"Corrupt JPEG");
3277 for (i=0; i < c; ++i) {
3278 z->img_comp[i].data = NULL;
3279 z->img_comp[i].linebuf = NULL;
3282 if (Lf != 8+3*s->img_n)
return stbi__err(
"bad SOF len",
"Corrupt JPEG");
3285 for (i=0; i < s->img_n; ++i) {
3286 static const unsigned char rgb[3] = {
'R',
'G',
'B' };
3287 z->img_comp[i].id = stbi__get8(s);
3288 if (s->img_n == 3 && z->img_comp[i].id == rgb[i])
3291 z->img_comp[i].h = (q >> 4);
if (!z->img_comp[i].h || z->img_comp[i].h > 4)
return stbi__err(
"bad H",
"Corrupt JPEG");
3292 z->img_comp[i].v = q & 15;
if (!z->img_comp[i].v || z->img_comp[i].v > 4)
return stbi__err(
"bad V",
"Corrupt JPEG");
3293 z->img_comp[i].tq = stbi__get8(s);
if (z->img_comp[i].tq > 3)
return stbi__err(
"bad TQ",
"Corrupt JPEG");
3296 if (scan != STBI__SCAN_load)
return 1;
3298 if (!stbi__mad3sizes_valid(s->img_x, s->img_y, s->img_n, 0))
return stbi__err(
"too large",
"Image too large to decode");
3300 for (i=0; i < s->img_n; ++i) {
3301 if (z->img_comp[i].h > h_max) h_max = z->img_comp[i].h;
3302 if (z->img_comp[i].v > v_max) v_max = z->img_comp[i].v;
3307 for (i=0; i < s->img_n; ++i) {
3308 if (h_max % z->img_comp[i].h != 0)
return stbi__err(
"bad H",
"Corrupt JPEG");
3309 if (v_max % z->img_comp[i].v != 0)
return stbi__err(
"bad V",
"Corrupt JPEG");
3313 z->img_h_max = h_max;
3314 z->img_v_max = v_max;
3315 z->img_mcu_w = h_max * 8;
3316 z->img_mcu_h = v_max * 8;
3318 z->img_mcu_x = (s->img_x + z->img_mcu_w-1) / z->img_mcu_w;
3319 z->img_mcu_y = (s->img_y + z->img_mcu_h-1) / z->img_mcu_h;
3321 for (i=0; i < s->img_n; ++i) {
3323 z->img_comp[i].x = (s->img_x * z->img_comp[i].h + h_max-1) / h_max;
3324 z->img_comp[i].y = (s->img_y * z->img_comp[i].v + v_max-1) / v_max;
3332 z->img_comp[i].w2 = z->img_mcu_x * z->img_comp[i].h * 8;
3333 z->img_comp[i].h2 = z->img_mcu_y * z->img_comp[i].v * 8;
3334 z->img_comp[i].coeff = 0;
3335 z->img_comp[i].raw_coeff = 0;
3336 z->img_comp[i].linebuf = NULL;
3337 z->img_comp[i].raw_data = stbi__malloc_mad2(z->img_comp[i].w2, z->img_comp[i].h2, 15);
3338 if (z->img_comp[i].raw_data == NULL)
3339 return stbi__free_jpeg_components(z, i+1, stbi__err(
"outofmem",
"Out of memory"));
3341 z->img_comp[i].data = (stbi_uc*) (((
size_t) z->img_comp[i].raw_data + 15) & ~15);
3342 if (z->progressive) {
3344 z->img_comp[i].coeff_w = z->img_comp[i].w2 / 8;
3345 z->img_comp[i].coeff_h = z->img_comp[i].h2 / 8;
3346 z->img_comp[i].raw_coeff = stbi__malloc_mad3(z->img_comp[i].w2, z->img_comp[i].h2,
sizeof(
short), 15);
3347 if (z->img_comp[i].raw_coeff == NULL)
3348 return stbi__free_jpeg_components(z, i+1, stbi__err(
"outofmem",
"Out of memory"));
3349 z->img_comp[i].coeff = (
short*) (((
size_t) z->img_comp[i].raw_coeff + 15) & ~15);
3357#define stbi__DNL(x) ((x) == 0xdc)
3358#define stbi__SOI(x) ((x) == 0xd8)
3359#define stbi__EOI(x) ((x) == 0xd9)
3360#define stbi__SOF(x) ((x) == 0xc0 || (x) == 0xc1 || (x) == 0xc2)
3361#define stbi__SOS(x) ((x) == 0xda)
3363#define stbi__SOF_progressive(x) ((x) == 0xc2)
3365static int stbi__decode_jpeg_header(stbi__jpeg *z,
int scan)
3369 z->app14_color_transform = -1;
3370 z->marker = STBI__MARKER_none;
3371 m = stbi__get_marker(z);
3372 if (!stbi__SOI(m))
return stbi__err(
"no SOI",
"Corrupt JPEG");
3373 if (scan == STBI__SCAN_type)
return 1;
3374 m = stbi__get_marker(z);
3375 while (!stbi__SOF(m)) {
3376 if (!stbi__process_marker(z,m))
return 0;
3377 m = stbi__get_marker(z);
3378 while (m == STBI__MARKER_none) {
3380 if (stbi__at_eof(z->s))
return stbi__err(
"no SOF",
"Corrupt JPEG");
3381 m = stbi__get_marker(z);
3384 z->progressive = stbi__SOF_progressive(m);
3385 if (!stbi__process_frame_header(z, scan))
return 0;
3389static stbi_uc stbi__skip_jpeg_junk_at_end(stbi__jpeg *j)
3393 while (!stbi__at_eof(j->s)) {
3394 stbi_uc x = stbi__get8(j->s);
3396 if (stbi__at_eof(j->s))
return STBI__MARKER_none;
3397 x = stbi__get8(j->s);
3398 if (x != 0x00 && x != 0xff) {
3408 return STBI__MARKER_none;
3412static int stbi__decode_jpeg_image(stbi__jpeg *j)
3415 for (m = 0; m < 4; m++) {
3416 j->img_comp[m].raw_data = NULL;
3417 j->img_comp[m].raw_coeff = NULL;
3419 j->restart_interval = 0;
3420 if (!stbi__decode_jpeg_header(j, STBI__SCAN_load))
return 0;
3421 m = stbi__get_marker(j);
3422 while (!stbi__EOI(m)) {
3424 if (!stbi__process_scan_header(j))
return 0;
3425 if (!stbi__parse_entropy_coded_data(j))
return 0;
3426 if (j->marker == STBI__MARKER_none ) {
3427 j->marker = stbi__skip_jpeg_junk_at_end(j);
3430 m = stbi__get_marker(j);
3431 if (STBI__RESTART(m))
3432 m = stbi__get_marker(j);
3433 }
else if (stbi__DNL(m)) {
3434 int Ld = stbi__get16be(j->s);
3435 stbi__uint32 NL = stbi__get16be(j->s);
3436 if (Ld != 4)
return stbi__err(
"bad DNL len",
"Corrupt JPEG");
3437 if (NL != j->s->img_y)
return stbi__err(
"bad DNL height",
"Corrupt JPEG");
3438 m = stbi__get_marker(j);
3440 if (!stbi__process_marker(j, m))
return 1;
3441 m = stbi__get_marker(j);
3445 stbi__jpeg_finish(j);
3451typedef stbi_uc *(*resample_row_func)(stbi_uc *out, stbi_uc *in0, stbi_uc *in1,
3454#define stbi__div4(x) ((stbi_uc) ((x) >> 2))
3456static stbi_uc *resample_row_1(stbi_uc *out, stbi_uc *in_near, stbi_uc *in_far,
int w,
int hs)
3459 STBI_NOTUSED(in_far);
3465static stbi_uc* stbi__resample_row_v_2(stbi_uc *out, stbi_uc *in_near, stbi_uc *in_far,
int w,
int hs)
3470 for (i=0; i < w; ++i)
3471 out[i] = stbi__div4(3*in_near[i] + in_far[i] + 2);
3475static stbi_uc* stbi__resample_row_h_2(stbi_uc *out, stbi_uc *in_near, stbi_uc *in_far,
int w,
int hs)
3479 stbi_uc *input = in_near;
3483 out[0] = out[1] = input[0];
3488 out[1] = stbi__div4(input[0]*3 + input[1] + 2);
3489 for (i=1; i < w-1; ++i) {
3490 int n = 3*input[i]+2;
3491 out[i*2+0] = stbi__div4(n+input[i-1]);
3492 out[i*2+1] = stbi__div4(n+input[i+1]);
3494 out[i*2+0] = stbi__div4(input[w-2]*3 + input[w-1] + 2);
3495 out[i*2+1] = input[w-1];
3497 STBI_NOTUSED(in_far);
3503#define stbi__div16(x) ((stbi_uc) ((x) >> 4))
3505static stbi_uc *stbi__resample_row_hv_2(stbi_uc *out, stbi_uc *in_near, stbi_uc *in_far,
int w,
int hs)
3510 out[0] = out[1] = stbi__div4(3*in_near[0] + in_far[0] + 2);
3514 t1 = 3*in_near[0] + in_far[0];
3515 out[0] = stbi__div4(t1+2);
3516 for (i=1; i < w; ++i) {
3518 t1 = 3*in_near[i]+in_far[i];
3519 out[i*2-1] = stbi__div16(3*t0 + t1 + 8);
3520 out[i*2 ] = stbi__div16(3*t1 + t0 + 8);
3522 out[w*2-1] = stbi__div4(t1+2);
3529#if defined(STBI_SSE2) || defined(STBI_NEON)
3530static stbi_uc *stbi__resample_row_hv_2_simd(stbi_uc *out, stbi_uc *in_near, stbi_uc *in_far,
int w,
int hs)
3536 out[0] = out[1] = stbi__div4(3*in_near[0] + in_far[0] + 2);
3540 t1 = 3*in_near[0] + in_far[0];
3544 for (; i < ((w-1) & ~7); i += 8) {
3545#if defined(STBI_SSE2)
3548 __m128i zero = _mm_setzero_si128();
3549 __m128i farb = _mm_loadl_epi64((__m128i *) (in_far + i));
3550 __m128i nearb = _mm_loadl_epi64((__m128i *) (in_near + i));
3551 __m128i farw = _mm_unpacklo_epi8(farb, zero);
3552 __m128i nearw = _mm_unpacklo_epi8(nearb, zero);
3553 __m128i diff = _mm_sub_epi16(farw, nearw);
3554 __m128i nears = _mm_slli_epi16(nearw, 2);
3555 __m128i curr = _mm_add_epi16(nears, diff);
3562 __m128i prv0 = _mm_slli_si128(curr, 2);
3563 __m128i nxt0 = _mm_srli_si128(curr, 2);
3564 __m128i prev = _mm_insert_epi16(prv0, t1, 0);
3565 __m128i next = _mm_insert_epi16(nxt0, 3*in_near[i+8] + in_far[i+8], 7);
3571 __m128i bias = _mm_set1_epi16(8);
3572 __m128i curs = _mm_slli_epi16(curr, 2);
3573 __m128i prvd = _mm_sub_epi16(prev, curr);
3574 __m128i nxtd = _mm_sub_epi16(next, curr);
3575 __m128i curb = _mm_add_epi16(curs, bias);
3576 __m128i even = _mm_add_epi16(prvd, curb);
3577 __m128i odd = _mm_add_epi16(nxtd, curb);
3580 __m128i int0 = _mm_unpacklo_epi16(even, odd);
3581 __m128i int1 = _mm_unpackhi_epi16(even, odd);
3582 __m128i de0 = _mm_srli_epi16(int0, 4);
3583 __m128i de1 = _mm_srli_epi16(int1, 4);
3586 __m128i outv = _mm_packus_epi16(de0, de1);
3587 _mm_storeu_si128((__m128i *) (out + i*2), outv);
3588#elif defined(STBI_NEON)
3591 uint8x8_t farb = vld1_u8(in_far + i);
3592 uint8x8_t nearb = vld1_u8(in_near + i);
3593 int16x8_t diff = vreinterpretq_s16_u16(vsubl_u8(farb, nearb));
3594 int16x8_t nears = vreinterpretq_s16_u16(vshll_n_u8(nearb, 2));
3595 int16x8_t curr = vaddq_s16(nears, diff);
3602 int16x8_t prv0 = vextq_s16(curr, curr, 7);
3603 int16x8_t nxt0 = vextq_s16(curr, curr, 1);
3604 int16x8_t prev = vsetq_lane_s16(t1, prv0, 0);
3605 int16x8_t next = vsetq_lane_s16(3*in_near[i+8] + in_far[i+8], nxt0, 7);
3611 int16x8_t curs = vshlq_n_s16(curr, 2);
3612 int16x8_t prvd = vsubq_s16(prev, curr);
3613 int16x8_t nxtd = vsubq_s16(next, curr);
3614 int16x8_t even = vaddq_s16(curs, prvd);
3615 int16x8_t odd = vaddq_s16(curs, nxtd);
3619 o.val[0] = vqrshrun_n_s16(even, 4);
3620 o.val[1] = vqrshrun_n_s16(odd, 4);
3621 vst2_u8(out + i*2, o);
3625 t1 = 3*in_near[i+7] + in_far[i+7];
3629 t1 = 3*in_near[i] + in_far[i];
3630 out[i*2] = stbi__div16(3*t1 + t0 + 8);
3632 for (++i; i < w; ++i) {
3634 t1 = 3*in_near[i]+in_far[i];
3635 out[i*2-1] = stbi__div16(3*t0 + t1 + 8);
3636 out[i*2 ] = stbi__div16(3*t1 + t0 + 8);
3638 out[w*2-1] = stbi__div4(t1+2);
3646static stbi_uc *stbi__resample_row_generic(stbi_uc *out, stbi_uc *in_near, stbi_uc *in_far,
int w,
int hs)
3650 STBI_NOTUSED(in_far);
3651 for (i=0; i < w; ++i)
3652 for (j=0; j < hs; ++j)
3653 out[i*hs+j] = in_near[i];
3659#define stbi__float2fixed(x) (((int) ((x) * 4096.0f + 0.5f)) << 8)
3660static void stbi__YCbCr_to_RGB_row(stbi_uc *out,
const stbi_uc *y,
const stbi_uc *pcb,
const stbi_uc *pcr,
int count,
int step)
3663 for (i=0; i < count; ++i) {
3664 int y_fixed = (y[i] << 20) + (1<<19);
3666 int cr = pcr[i] - 128;
3667 int cb = pcb[i] - 128;
3668 r = y_fixed + cr* stbi__float2fixed(1.40200f);
3669 g = y_fixed + (cr*-stbi__float2fixed(0.71414f)) + ((cb*-stbi__float2fixed(0.34414f)) & 0xffff0000);
3670 b = y_fixed + cb* stbi__float2fixed(1.77200f);
3674 if ((
unsigned) r > 255) {
if (r < 0) r = 0;
else r = 255; }
3675 if ((
unsigned) g > 255) {
if (g < 0) g = 0;
else g = 255; }
3676 if ((
unsigned) b > 255) {
if (b < 0) b = 0;
else b = 255; }
3677 out[0] = (stbi_uc)r;
3678 out[1] = (stbi_uc)g;
3679 out[2] = (stbi_uc)b;
3685#if defined(STBI_SSE2) || defined(STBI_NEON)
3686static void stbi__YCbCr_to_RGB_simd(stbi_uc *out, stbi_uc
const *y, stbi_uc
const *pcb, stbi_uc
const *pcr,
int count,
int step)
3696 __m128i signflip = _mm_set1_epi8(-0x80);
3697 __m128i cr_const0 = _mm_set1_epi16( (
short) ( 1.40200f*4096.0f+0.5f));
3698 __m128i cr_const1 = _mm_set1_epi16( - (
short) ( 0.71414f*4096.0f+0.5f));
3699 __m128i cb_const0 = _mm_set1_epi16( - (
short) ( 0.34414f*4096.0f+0.5f));
3700 __m128i cb_const1 = _mm_set1_epi16( (
short) ( 1.77200f*4096.0f+0.5f));
3701 __m128i y_bias = _mm_set1_epi8((
char) (
unsigned char) 128);
3702 __m128i xw = _mm_set1_epi16(255);
3704 for (; i+7 < count; i += 8) {
3706 __m128i y_bytes = _mm_loadl_epi64((__m128i *) (y+i));
3707 __m128i cr_bytes = _mm_loadl_epi64((__m128i *) (pcr+i));
3708 __m128i cb_bytes = _mm_loadl_epi64((__m128i *) (pcb+i));
3709 __m128i cr_biased = _mm_xor_si128(cr_bytes, signflip);
3710 __m128i cb_biased = _mm_xor_si128(cb_bytes, signflip);
3713 __m128i yw = _mm_unpacklo_epi8(y_bias, y_bytes);
3714 __m128i crw = _mm_unpacklo_epi8(_mm_setzero_si128(), cr_biased);
3715 __m128i cbw = _mm_unpacklo_epi8(_mm_setzero_si128(), cb_biased);
3718 __m128i yws = _mm_srli_epi16(yw, 4);
3719 __m128i cr0 = _mm_mulhi_epi16(cr_const0, crw);
3720 __m128i cb0 = _mm_mulhi_epi16(cb_const0, cbw);
3721 __m128i cb1 = _mm_mulhi_epi16(cbw, cb_const1);
3722 __m128i cr1 = _mm_mulhi_epi16(crw, cr_const1);
3723 __m128i rws = _mm_add_epi16(cr0, yws);
3724 __m128i gwt = _mm_add_epi16(cb0, yws);
3725 __m128i bws = _mm_add_epi16(yws, cb1);
3726 __m128i gws = _mm_add_epi16(gwt, cr1);
3729 __m128i rw = _mm_srai_epi16(rws, 4);
3730 __m128i bw = _mm_srai_epi16(bws, 4);
3731 __m128i gw = _mm_srai_epi16(gws, 4);
3734 __m128i brb = _mm_packus_epi16(rw, bw);
3735 __m128i gxb = _mm_packus_epi16(gw, xw);
3738 __m128i t0 = _mm_unpacklo_epi8(brb, gxb);
3739 __m128i t1 = _mm_unpackhi_epi8(brb, gxb);
3740 __m128i o0 = _mm_unpacklo_epi16(t0, t1);
3741 __m128i o1 = _mm_unpackhi_epi16(t0, t1);
3744 _mm_storeu_si128((__m128i *) (out + 0), o0);
3745 _mm_storeu_si128((__m128i *) (out + 16), o1);
3755 uint8x8_t signflip = vdup_n_u8(0x80);
3756 int16x8_t cr_const0 = vdupq_n_s16( (
short) ( 1.40200f*4096.0f+0.5f));
3757 int16x8_t cr_const1 = vdupq_n_s16( - (
short) ( 0.71414f*4096.0f+0.5f));
3758 int16x8_t cb_const0 = vdupq_n_s16( - (
short) ( 0.34414f*4096.0f+0.5f));
3759 int16x8_t cb_const1 = vdupq_n_s16( (
short) ( 1.77200f*4096.0f+0.5f));
3761 for (; i+7 < count; i += 8) {
3763 uint8x8_t y_bytes = vld1_u8(y + i);
3764 uint8x8_t cr_bytes = vld1_u8(pcr + i);
3765 uint8x8_t cb_bytes = vld1_u8(pcb + i);
3766 int8x8_t cr_biased = vreinterpret_s8_u8(vsub_u8(cr_bytes, signflip));
3767 int8x8_t cb_biased = vreinterpret_s8_u8(vsub_u8(cb_bytes, signflip));
3770 int16x8_t yws = vreinterpretq_s16_u16(vshll_n_u8(y_bytes, 4));
3771 int16x8_t crw = vshll_n_s8(cr_biased, 7);
3772 int16x8_t cbw = vshll_n_s8(cb_biased, 7);
3775 int16x8_t cr0 = vqdmulhq_s16(crw, cr_const0);
3776 int16x8_t cb0 = vqdmulhq_s16(cbw, cb_const0);
3777 int16x8_t cr1 = vqdmulhq_s16(crw, cr_const1);
3778 int16x8_t cb1 = vqdmulhq_s16(cbw, cb_const1);
3779 int16x8_t rws = vaddq_s16(yws, cr0);
3780 int16x8_t gws = vaddq_s16(vaddq_s16(yws, cb0), cr1);
3781 int16x8_t bws = vaddq_s16(yws, cb1);
3785 o.val[0] = vqrshrun_n_s16(rws, 4);
3786 o.val[1] = vqrshrun_n_s16(gws, 4);
3787 o.val[2] = vqrshrun_n_s16(bws, 4);
3788 o.val[3] = vdup_n_u8(255);
3797 for (; i < count; ++i) {
3798 int y_fixed = (y[i] << 20) + (1<<19);
3800 int cr = pcr[i] - 128;
3801 int cb = pcb[i] - 128;
3802 r = y_fixed + cr* stbi__float2fixed(1.40200f);
3803 g = y_fixed + cr*-stbi__float2fixed(0.71414f) + ((cb*-stbi__float2fixed(0.34414f)) & 0xffff0000);
3804 b = y_fixed + cb* stbi__float2fixed(1.77200f);
3808 if ((
unsigned) r > 255) {
if (r < 0) r = 0;
else r = 255; }
3809 if ((
unsigned) g > 255) {
if (g < 0) g = 0;
else g = 255; }
3810 if ((
unsigned) b > 255) {
if (b < 0) b = 0;
else b = 255; }
3811 out[0] = (stbi_uc)r;
3812 out[1] = (stbi_uc)g;
3813 out[2] = (stbi_uc)b;
3821static void stbi__setup_jpeg(stbi__jpeg *j)
3823 j->idct_block_kernel = stbi__idct_block;
3824 j->YCbCr_to_RGB_kernel = stbi__YCbCr_to_RGB_row;
3825 j->resample_row_hv_2_kernel = stbi__resample_row_hv_2;
3828 if (stbi__sse2_available()) {
3829 j->idct_block_kernel = stbi__idct_simd;
3830 j->YCbCr_to_RGB_kernel = stbi__YCbCr_to_RGB_simd;
3831 j->resample_row_hv_2_kernel = stbi__resample_row_hv_2_simd;
3836 j->idct_block_kernel = stbi__idct_simd;
3837 j->YCbCr_to_RGB_kernel = stbi__YCbCr_to_RGB_simd;
3838 j->resample_row_hv_2_kernel = stbi__resample_row_hv_2_simd;
3843static void stbi__cleanup_jpeg(stbi__jpeg *j)
3845 stbi__free_jpeg_components(j, j->s->img_n, 0);
3850 resample_row_func resample;
3851 stbi_uc *line0,*line1;
3859static stbi_uc stbi__blinn_8x8(stbi_uc x, stbi_uc y)
3861 unsigned int t = x*y + 128;
3862 return (stbi_uc) ((t + (t >>8)) >> 8);
3865static stbi_uc *load_jpeg_image(stbi__jpeg *z,
int *out_x,
int *out_y,
int *comp,
int req_comp)
3867 int n, decode_n, is_rgb;
3871 if (req_comp < 0 || req_comp > 4)
return stbi__errpuc(
"bad req_comp",
"Internal error");
3874 if (!stbi__decode_jpeg_image(z)) { stbi__cleanup_jpeg(z);
return NULL; }
3877 n = req_comp ? req_comp : z->s->img_n >= 3 ? 3 : 1;
3879 is_rgb = z->s->img_n == 3 && (z->rgb == 3 || (z->app14_color_transform == 0 && !z->jfif));
3881 if (z->s->img_n == 3 && n < 3 && !is_rgb)
3884 decode_n = z->s->img_n;
3888 if (decode_n <= 0) { stbi__cleanup_jpeg(z);
return NULL; }
3895 stbi_uc *coutput[4] = { NULL, NULL, NULL, NULL };
3897 stbi__resample res_comp[4];
3899 for (k=0; k < decode_n; ++k) {
3900 stbi__resample *r = &res_comp[k];
3904 z->img_comp[k].linebuf = (stbi_uc *) stbi__malloc(z->s->img_x + 3);
3905 if (!z->img_comp[k].linebuf) { stbi__cleanup_jpeg(z);
return stbi__errpuc(
"outofmem",
"Out of memory"); }
3907 r->hs = z->img_h_max / z->img_comp[k].h;
3908 r->vs = z->img_v_max / z->img_comp[k].v;
3909 r->ystep = r->vs >> 1;
3910 r->w_lores = (z->s->img_x + r->hs-1) / r->hs;
3912 r->line0 = r->line1 = z->img_comp[k].data;
3914 if (r->hs == 1 && r->vs == 1) r->resample = resample_row_1;
3915 else if (r->hs == 1 && r->vs == 2) r->resample = stbi__resample_row_v_2;
3916 else if (r->hs == 2 && r->vs == 1) r->resample = stbi__resample_row_h_2;
3917 else if (r->hs == 2 && r->vs == 2) r->resample = z->resample_row_hv_2_kernel;
3918 else r->resample = stbi__resample_row_generic;
3922 output = (stbi_uc *) stbi__malloc_mad3(n, z->s->img_x, z->s->img_y, 1);
3923 if (!output) { stbi__cleanup_jpeg(z);
return stbi__errpuc(
"outofmem",
"Out of memory"); }
3926 for (j=0; j < z->s->img_y; ++j) {
3927 stbi_uc *out = output + n * z->s->img_x * j;
3928 for (k=0; k < decode_n; ++k) {
3929 stbi__resample *r = &res_comp[k];
3930 int y_bot = r->ystep >= (r->vs >> 1);
3931 coutput[k] = r->resample(z->img_comp[k].linebuf,
3932 y_bot ? r->line1 : r->line0,
3933 y_bot ? r->line0 : r->line1,
3935 if (++r->ystep >= r->vs) {
3937 r->line0 = r->line1;
3938 if (++r->ypos < z->img_comp[k].y)
3939 r->line1 += z->img_comp[k].w2;
3943 stbi_uc *y = coutput[0];
3944 if (z->s->img_n == 3) {
3946 for (i=0; i < z->s->img_x; ++i) {
3948 out[1] = coutput[1][i];
3949 out[2] = coutput[2][i];
3954 z->YCbCr_to_RGB_kernel(out, y, coutput[1], coutput[2], z->s->img_x, n);
3956 }
else if (z->s->img_n == 4) {
3957 if (z->app14_color_transform == 0) {
3958 for (i=0; i < z->s->img_x; ++i) {
3959 stbi_uc m = coutput[3][i];
3960 out[0] = stbi__blinn_8x8(coutput[0][i], m);
3961 out[1] = stbi__blinn_8x8(coutput[1][i], m);
3962 out[2] = stbi__blinn_8x8(coutput[2][i], m);
3966 }
else if (z->app14_color_transform == 2) {
3967 z->YCbCr_to_RGB_kernel(out, y, coutput[1], coutput[2], z->s->img_x, n);
3968 for (i=0; i < z->s->img_x; ++i) {
3969 stbi_uc m = coutput[3][i];
3970 out[0] = stbi__blinn_8x8(255 - out[0], m);
3971 out[1] = stbi__blinn_8x8(255 - out[1], m);
3972 out[2] = stbi__blinn_8x8(255 - out[2], m);
3976 z->YCbCr_to_RGB_kernel(out, y, coutput[1], coutput[2], z->s->img_x, n);
3979 for (i=0; i < z->s->img_x; ++i) {
3980 out[0] = out[1] = out[2] = y[i];
3987 for (i=0; i < z->s->img_x; ++i)
3988 *out++ = stbi__compute_y(coutput[0][i], coutput[1][i], coutput[2][i]);
3990 for (i=0; i < z->s->img_x; ++i, out += 2) {
3991 out[0] = stbi__compute_y(coutput[0][i], coutput[1][i], coutput[2][i]);
3995 }
else if (z->s->img_n == 4 && z->app14_color_transform == 0) {
3996 for (i=0; i < z->s->img_x; ++i) {
3997 stbi_uc m = coutput[3][i];
3998 stbi_uc r = stbi__blinn_8x8(coutput[0][i], m);
3999 stbi_uc g = stbi__blinn_8x8(coutput[1][i], m);
4000 stbi_uc b = stbi__blinn_8x8(coutput[2][i], m);
4001 out[0] = stbi__compute_y(r, g, b);
4005 }
else if (z->s->img_n == 4 && z->app14_color_transform == 2) {
4006 for (i=0; i < z->s->img_x; ++i) {
4007 out[0] = stbi__blinn_8x8(255 - coutput[0][i], coutput[3][i]);
4012 stbi_uc *y = coutput[0];
4014 for (i=0; i < z->s->img_x; ++i) out[i] = y[i];
4016 for (i=0; i < z->s->img_x; ++i) { *out++ = y[i]; *out++ = 255; }
4020 stbi__cleanup_jpeg(z);
4021 *out_x = z->s->img_x;
4022 *out_y = z->s->img_y;
4023 if (comp) *comp = z->s->img_n >= 3 ? 3 : 1;
4028static void *stbi__jpeg_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri)
4030 unsigned char* result;
4031 stbi__jpeg* j = (stbi__jpeg*) stbi__malloc(
sizeof(stbi__jpeg));
4032 if (!j)
return stbi__errpuc(
"outofmem",
"Out of memory");
4033 memset(j, 0,
sizeof(stbi__jpeg));
4036 stbi__setup_jpeg(j);
4037 result = load_jpeg_image(j, x,y,comp,req_comp);
4042static int stbi__jpeg_test(stbi__context *s)
4045 stbi__jpeg* j = (stbi__jpeg*)stbi__malloc(
sizeof(stbi__jpeg));
4046 if (!j)
return stbi__err(
"outofmem",
"Out of memory");
4047 memset(j, 0,
sizeof(stbi__jpeg));
4049 stbi__setup_jpeg(j);
4050 r = stbi__decode_jpeg_header(j, STBI__SCAN_type);
4056static int stbi__jpeg_info_raw(stbi__jpeg *j,
int *x,
int *y,
int *comp)
4058 if (!stbi__decode_jpeg_header(j, STBI__SCAN_header)) {
4059 stbi__rewind( j->s );
4062 if (x) *x = j->s->img_x;
4063 if (y) *y = j->s->img_y;
4064 if (comp) *comp = j->s->img_n >= 3 ? 3 : 1;
4068static int stbi__jpeg_info(stbi__context *s,
int *x,
int *y,
int *comp)
4071 stbi__jpeg* j = (stbi__jpeg*) (stbi__malloc(
sizeof(stbi__jpeg)));
4072 if (!j)
return stbi__err(
"outofmem",
"Out of memory");
4073 memset(j, 0,
sizeof(stbi__jpeg));
4075 result = stbi__jpeg_info_raw(j, x, y, comp);
4091#define STBI__ZFAST_BITS 9
4092#define STBI__ZFAST_MASK ((1 << STBI__ZFAST_BITS) - 1)
4093#define STBI__ZNSYMS 288
4099 stbi__uint16 fast[1 << STBI__ZFAST_BITS];
4100 stbi__uint16 firstcode[16];
4102 stbi__uint16 firstsymbol[16];
4103 stbi_uc size[STBI__ZNSYMS];
4104 stbi__uint16 value[STBI__ZNSYMS];
4107stbi_inline
static int stbi__bitreverse16(
int n)
4109 n = ((n & 0xAAAA) >> 1) | ((n & 0x5555) << 1);
4110 n = ((n & 0xCCCC) >> 2) | ((n & 0x3333) << 2);
4111 n = ((n & 0xF0F0) >> 4) | ((n & 0x0F0F) << 4);
4112 n = ((n & 0xFF00) >> 8) | ((n & 0x00FF) << 8);
4116stbi_inline
static int stbi__bit_reverse(
int v,
int bits)
4118 STBI_ASSERT(bits <= 16);
4121 return stbi__bitreverse16(v) >> (16-bits);
4124static int stbi__zbuild_huffman(stbi__zhuffman *z,
const stbi_uc *sizelist,
int num)
4127 int code, next_code[16], sizes[17];
4130 memset(sizes, 0,
sizeof(sizes));
4131 memset(z->fast, 0,
sizeof(z->fast));
4132 for (i=0; i < num; ++i)
4133 ++sizes[sizelist[i]];
4135 for (i=1; i < 16; ++i)
4136 if (sizes[i] > (1 << i))
4137 return stbi__err(
"bad sizes",
"Corrupt PNG");
4139 for (i=1; i < 16; ++i) {
4140 next_code[i] = code;
4141 z->firstcode[i] = (stbi__uint16) code;
4142 z->firstsymbol[i] = (stbi__uint16) k;
4143 code = (code + sizes[i]);
4145 if (code-1 >= (1 << i))
return stbi__err(
"bad codelengths",
"Corrupt PNG");
4146 z->maxcode[i] = code << (16-i);
4150 z->maxcode[16] = 0x10000;
4151 for (i=0; i < num; ++i) {
4152 int s = sizelist[i];
4154 int c = next_code[s] - z->firstcode[s] + z->firstsymbol[s];
4155 stbi__uint16 fastv = (stbi__uint16) ((s << 9) | i);
4156 z->size [c] = (stbi_uc ) s;
4157 z->value[c] = (stbi__uint16) i;
4158 if (s <= STBI__ZFAST_BITS) {
4159 int j = stbi__bit_reverse(next_code[s],s);
4160 while (j < (1 << STBI__ZFAST_BITS)) {
4179 stbi_uc *zbuffer, *zbuffer_end;
4182 stbi__uint32 code_buffer;
4189 stbi__zhuffman z_length, z_distance;
4192stbi_inline
static int stbi__zeof(stbi__zbuf *z)
4194 return (z->zbuffer >= z->zbuffer_end);
4197stbi_inline
static stbi_uc stbi__zget8(stbi__zbuf *z)
4199 return stbi__zeof(z) ? 0 : *z->zbuffer++;
4202static void stbi__fill_bits(stbi__zbuf *z)
4205 if (z->code_buffer >= (1U << z->num_bits)) {
4206 z->zbuffer = z->zbuffer_end;
4209 z->code_buffer |= (
unsigned int) stbi__zget8(z) << z->num_bits;
4211 }
while (z->num_bits <= 24);
4214stbi_inline
static unsigned int stbi__zreceive(stbi__zbuf *z,
int n)
4217 if (z->num_bits < n) stbi__fill_bits(z);
4218 k = z->code_buffer & ((1 << n) - 1);
4219 z->code_buffer >>= n;
4224static int stbi__zhuffman_decode_slowpath(stbi__zbuf *a, stbi__zhuffman *z)
4229 k = stbi__bit_reverse(a->code_buffer, 16);
4230 for (s=STBI__ZFAST_BITS+1; ; ++s)
4231 if (k < z->maxcode[s])
4233 if (s >= 16)
return -1;
4235 b = (k >> (16-s)) - z->firstcode[s] + z->firstsymbol[s];
4236 if (b >= STBI__ZNSYMS)
return -1;
4237 if (z->size[b] != s)
return -1;
4238 a->code_buffer >>= s;
4243stbi_inline
static int stbi__zhuffman_decode(stbi__zbuf *a, stbi__zhuffman *z)
4246 if (a->num_bits < 16) {
4247 if (stbi__zeof(a)) {
4248 if (!a->hit_zeof_once) {
4252 a->hit_zeof_once = 1;
4263 b = z->fast[a->code_buffer & STBI__ZFAST_MASK];
4266 a->code_buffer >>= s;
4270 return stbi__zhuffman_decode_slowpath(a, z);
4273static int stbi__zexpand(stbi__zbuf *z,
char *zout,
int n)
4276 unsigned int cur, limit, old_limit;
4278 if (!z->z_expandable)
return stbi__err(
"output buffer limit",
"Corrupt PNG");
4279 cur = (
unsigned int) (z->zout - z->zout_start);
4280 limit = old_limit = (unsigned) (z->zout_end - z->zout_start);
4281 if (UINT_MAX - cur < (
unsigned) n)
return stbi__err(
"outofmem",
"Out of memory");
4282 while (cur + n > limit) {
4283 if(limit > UINT_MAX / 2)
return stbi__err(
"outofmem",
"Out of memory");
4286 q = (
char *) STBI_REALLOC_SIZED(z->zout_start, old_limit, limit);
4287 STBI_NOTUSED(old_limit);
4288 if (q == NULL)
return stbi__err(
"outofmem",
"Out of memory");
4291 z->zout_end = q + limit;
4295static const int stbi__zlength_base[31] = {
4296 3,4,5,6,7,8,9,10,11,13,
4297 15,17,19,23,27,31,35,43,51,59,
4298 67,83,99,115,131,163,195,227,258,0,0 };
4300static const int stbi__zlength_extra[31]=
4301{ 0,0,0,0,0,0,0,0,1,1,1,1,2,2,2,2,3,3,3,3,4,4,4,4,5,5,5,5,0,0,0 };
4303static const int stbi__zdist_base[32] = { 1,2,3,4,5,7,9,13,17,25,33,49,65,97,129,193,
4304257,385,513,769,1025,1537,2049,3073,4097,6145,8193,12289,16385,24577,0,0};
4306static const int stbi__zdist_extra[32] =
4307{ 0,0,0,0,1,1,2,2,3,3,4,4,5,5,6,6,7,7,8,8,9,9,10,10,11,11,12,12,13,13};
4309static int stbi__parse_huffman_block(stbi__zbuf *a)
4311 char *zout = a->zout;
4313 int z = stbi__zhuffman_decode(a, &a->z_length);
4315 if (z < 0)
return stbi__err(
"bad huffman code",
"Corrupt PNG");
4316 if (zout >= a->zout_end) {
4317 if (!stbi__zexpand(a, zout, 1))
return 0;
4326 if (a->hit_zeof_once && a->num_bits < 16) {
4331 return stbi__err(
"unexpected end",
"Corrupt PNG");
4335 if (z >= 286)
return stbi__err(
"bad huffman code",
"Corrupt PNG");
4337 len = stbi__zlength_base[z];
4338 if (stbi__zlength_extra[z]) len += stbi__zreceive(a, stbi__zlength_extra[z]);
4339 z = stbi__zhuffman_decode(a, &a->z_distance);
4340 if (z < 0 || z >= 30)
return stbi__err(
"bad huffman code",
"Corrupt PNG");
4341 dist = stbi__zdist_base[z];
4342 if (stbi__zdist_extra[z]) dist += stbi__zreceive(a, stbi__zdist_extra[z]);
4343 if (zout - a->zout_start < dist)
return stbi__err(
"bad dist",
"Corrupt PNG");
4344 if (len > a->zout_end - zout) {
4345 if (!stbi__zexpand(a, zout, len))
return 0;
4348 p = (stbi_uc *) (zout - dist);
4351 if (len) {
do *zout++ = v;
while (--len); }
4353 if (len) {
do *zout++ = *p++;
while (--len); }
4359static int stbi__compute_huffman_codes(stbi__zbuf *a)
4361 static const stbi_uc length_dezigzag[19] = { 16,17,18,0,8,7,9,6,10,5,11,4,12,3,13,2,14,1,15 };
4362 stbi__zhuffman z_codelength;
4363 stbi_uc lencodes[286+32+137];
4364 stbi_uc codelength_sizes[19];
4367 int hlit = stbi__zreceive(a,5) + 257;
4368 int hdist = stbi__zreceive(a,5) + 1;
4369 int hclen = stbi__zreceive(a,4) + 4;
4370 int ntot = hlit + hdist;
4372 memset(codelength_sizes, 0,
sizeof(codelength_sizes));
4373 for (i=0; i < hclen; ++i) {
4374 int s = stbi__zreceive(a,3);
4375 codelength_sizes[length_dezigzag[i]] = (stbi_uc) s;
4377 if (!stbi__zbuild_huffman(&z_codelength, codelength_sizes, 19))
return 0;
4381 int c = stbi__zhuffman_decode(a, &z_codelength);
4382 if (c < 0 || c >= 19)
return stbi__err(
"bad codelengths",
"Corrupt PNG");
4384 lencodes[n++] = (stbi_uc) c;
4388 c = stbi__zreceive(a,2)+3;
4389 if (n == 0)
return stbi__err(
"bad codelengths",
"Corrupt PNG");
4390 fill = lencodes[n-1];
4391 }
else if (c == 17) {
4392 c = stbi__zreceive(a,3)+3;
4393 }
else if (c == 18) {
4394 c = stbi__zreceive(a,7)+11;
4396 return stbi__err(
"bad codelengths",
"Corrupt PNG");
4398 if (ntot - n < c)
return stbi__err(
"bad codelengths",
"Corrupt PNG");
4399 memset(lencodes+n, fill, c);
4403 if (n != ntot)
return stbi__err(
"bad codelengths",
"Corrupt PNG");
4404 if (!stbi__zbuild_huffman(&a->z_length, lencodes, hlit))
return 0;
4405 if (!stbi__zbuild_huffman(&a->z_distance, lencodes+hlit, hdist))
return 0;
4409static int stbi__parse_uncompressed_block(stbi__zbuf *a)
4413 if (a->num_bits & 7)
4414 stbi__zreceive(a, a->num_bits & 7);
4417 while (a->num_bits > 0) {
4418 header[k++] = (stbi_uc) (a->code_buffer & 255);
4419 a->code_buffer >>= 8;
4422 if (a->num_bits < 0)
return stbi__err(
"zlib corrupt",
"Corrupt PNG");
4425 header[k++] = stbi__zget8(a);
4426 len = header[1] * 256 + header[0];
4427 nlen = header[3] * 256 + header[2];
4428 if (nlen != (len ^ 0xffff))
return stbi__err(
"zlib corrupt",
"Corrupt PNG");
4429 if (a->zbuffer + len > a->zbuffer_end)
return stbi__err(
"read past buffer",
"Corrupt PNG");
4430 if (a->zout + len > a->zout_end)
4431 if (!stbi__zexpand(a, a->zout, len))
return 0;
4432 memcpy(a->zout, a->zbuffer, len);
4438static int stbi__parse_zlib_header(stbi__zbuf *a)
4440 int cmf = stbi__zget8(a);
4443 int flg = stbi__zget8(a);
4444 if (stbi__zeof(a))
return stbi__err(
"bad zlib header",
"Corrupt PNG");
4445 if ((cmf*256+flg) % 31 != 0)
return stbi__err(
"bad zlib header",
"Corrupt PNG");
4446 if (flg & 32)
return stbi__err(
"no preset dict",
"Corrupt PNG");
4447 if (cm != 8)
return stbi__err(
"bad compression",
"Corrupt PNG");
4452static const stbi_uc stbi__zdefault_length[STBI__ZNSYMS] =
4454 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8, 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
4455 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8, 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
4456 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8, 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
4457 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8, 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,
4458 8,8,8,8,8,8,8,8,8,8,8,8,8,8,8,8, 9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,
4459 9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9, 9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,
4460 9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9, 9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,
4461 9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9, 9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,9,
4462 7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7, 7,7,7,7,7,7,7,7,8,8,8,8,8,8,8,8
4464static const stbi_uc stbi__zdefault_distance[32] =
4466 5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5
4481static int stbi__parse_zlib(stbi__zbuf *a,
int parse_header)
4485 if (!stbi__parse_zlib_header(a))
return 0;
4488 a->hit_zeof_once = 0;
4490 final = stbi__zreceive(a,1);
4491 type = stbi__zreceive(a,2);
4493 if (!stbi__parse_uncompressed_block(a))
return 0;
4494 }
else if (type == 3) {
4499 if (!stbi__zbuild_huffman(&a->z_length , stbi__zdefault_length , STBI__ZNSYMS))
return 0;
4500 if (!stbi__zbuild_huffman(&a->z_distance, stbi__zdefault_distance, 32))
return 0;
4502 if (!stbi__compute_huffman_codes(a))
return 0;
4504 if (!stbi__parse_huffman_block(a))
return 0;
4510static int stbi__do_zlib(stbi__zbuf *a,
char *obuf,
int olen,
int exp,
int parse_header)
4512 a->zout_start = obuf;
4514 a->zout_end = obuf + olen;
4515 a->z_expandable = exp;
4517 return stbi__parse_zlib(a, parse_header);
4520STBIDEF
char *stbi_zlib_decode_malloc_guesssize(
const char *buffer,
int len,
int initial_size,
int *outlen)
4523 char *p = (
char *) stbi__malloc(initial_size);
4524 if (p == NULL)
return NULL;
4525 a.zbuffer = (stbi_uc *) buffer;
4526 a.zbuffer_end = (stbi_uc *) buffer + len;
4527 if (stbi__do_zlib(&a, p, initial_size, 1, 1)) {
4528 if (outlen) *outlen = (int) (a.zout - a.zout_start);
4529 return a.zout_start;
4531 STBI_FREE(a.zout_start);
4536STBIDEF
char *stbi_zlib_decode_malloc(
char const *buffer,
int len,
int *outlen)
4538 return stbi_zlib_decode_malloc_guesssize(buffer, len, 16384, outlen);
4541STBIDEF
char *stbi_zlib_decode_malloc_guesssize_headerflag(
const char *buffer,
int len,
int initial_size,
int *outlen,
int parse_header)
4544 char *p = (
char *) stbi__malloc(initial_size);
4545 if (p == NULL)
return NULL;
4546 a.zbuffer = (stbi_uc *) buffer;
4547 a.zbuffer_end = (stbi_uc *) buffer + len;
4548 if (stbi__do_zlib(&a, p, initial_size, 1, parse_header)) {
4549 if (outlen) *outlen = (int) (a.zout - a.zout_start);
4550 return a.zout_start;
4552 STBI_FREE(a.zout_start);
4557STBIDEF
int stbi_zlib_decode_buffer(
char *obuffer,
int olen,
char const *ibuffer,
int ilen)
4560 a.zbuffer = (stbi_uc *) ibuffer;
4561 a.zbuffer_end = (stbi_uc *) ibuffer + ilen;
4562 if (stbi__do_zlib(&a, obuffer, olen, 0, 1))
4563 return (
int) (a.zout - a.zout_start);
4568STBIDEF
char *stbi_zlib_decode_noheader_malloc(
char const *buffer,
int len,
int *outlen)
4571 char *p = (
char *) stbi__malloc(16384);
4572 if (p == NULL)
return NULL;
4573 a.zbuffer = (stbi_uc *) buffer;
4574 a.zbuffer_end = (stbi_uc *) buffer+len;
4575 if (stbi__do_zlib(&a, p, 16384, 1, 0)) {
4576 if (outlen) *outlen = (int) (a.zout - a.zout_start);
4577 return a.zout_start;
4579 STBI_FREE(a.zout_start);
4584STBIDEF
int stbi_zlib_decode_noheader_buffer(
char *obuffer,
int olen,
const char *ibuffer,
int ilen)
4587 a.zbuffer = (stbi_uc *) ibuffer;
4588 a.zbuffer_end = (stbi_uc *) ibuffer + ilen;
4589 if (stbi__do_zlib(&a, obuffer, olen, 0, 0))
4590 return (
int) (a.zout - a.zout_start);
4609 stbi__uint32 length;
4613static stbi__pngchunk stbi__get_chunk_header(stbi__context *s)
4616 c.length = stbi__get32be(s);
4617 c.type = stbi__get32be(s);
4621static int stbi__check_png_header(stbi__context *s)
4623 static const stbi_uc png_sig[8] = { 137,80,78,71,13,10,26,10 };
4625 for (i=0; i < 8; ++i)
4626 if (stbi__get8(s) != png_sig[i])
return stbi__err(
"bad png sig",
"Not a PNG");
4633 stbi_uc *idata, *expanded, *out;
4648static stbi_uc first_row_filter[5] =
4657static int stbi__paeth(
int a,
int b,
int c)
4662 int thresh = c*3 - (a + b);
4663 int lo = a < b ? a : b;
4664 int hi = a < b ? b : a;
4665 int t0 = (hi <= thresh) ? lo : c;
4666 int t1 = (thresh <= lo) ? hi : t0;
4670static const stbi_uc stbi__depth_scale_table[9] = { 0, 0xff, 0x55, 0, 0x11, 0,0,0, 0x01 };
4675static void stbi__create_png_alpha_expand8(stbi_uc *dest, stbi_uc *src, stbi__uint32 x,
int img_n)
4680 for (i=x-1; i >= 0; --i) {
4682 dest[i*2+0] = src[i];
4685 STBI_ASSERT(img_n == 3);
4686 for (i=x-1; i >= 0; --i) {
4688 dest[i*4+2] = src[i*3+2];
4689 dest[i*4+1] = src[i*3+1];
4690 dest[i*4+0] = src[i*3+0];
4696static int stbi__create_png_image_raw(stbi__png *a, stbi_uc *raw, stbi__uint32 raw_len,
int out_n, stbi__uint32 x, stbi__uint32 y,
int depth,
int color)
4698 int bytes = (depth == 16 ? 2 : 1);
4699 stbi__context *s = a->s;
4700 stbi__uint32 i,j,stride = x*out_n*bytes;
4701 stbi__uint32 img_len, img_width_bytes;
4702 stbi_uc *filter_buf;
4705 int img_n = s->img_n;
4707 int output_bytes = out_n*bytes;
4708 int filter_bytes = img_n*bytes;
4711 STBI_ASSERT(out_n == s->img_n || out_n == s->img_n+1);
4712 a->out = (stbi_uc *) stbi__malloc_mad3(x, y, output_bytes, 0);
4713 if (!a->out)
return stbi__err(
"outofmem",
"Out of memory");
4717 if (!stbi__mad3sizes_valid(img_n, x, depth, 7))
return stbi__err(
"too large",
"Corrupt PNG");
4718 img_width_bytes = (((img_n * x * depth) + 7) >> 3);
4719 if (!stbi__mad2sizes_valid(img_width_bytes, y, img_width_bytes))
return stbi__err(
"too large",
"Corrupt PNG");
4720 img_len = (img_width_bytes + 1) * y;
4725 if (raw_len < img_len)
return stbi__err(
"not enough pixels",
"Corrupt PNG");
4728 filter_buf = (stbi_uc *) stbi__malloc_mad2(img_width_bytes, 2, 0);
4729 if (!filter_buf)
return stbi__err(
"outofmem",
"Out of memory");
4734 width = img_width_bytes;
4737 for (j=0; j < y; ++j) {
4739 stbi_uc *cur = filter_buf + (j & 1)*img_width_bytes;
4740 stbi_uc *prior = filter_buf + (~j & 1)*img_width_bytes;
4741 stbi_uc *dest = a->out + stride*j;
4742 int nk = width * filter_bytes;
4743 int filter = *raw++;
4747 all_ok = stbi__err(
"invalid filter",
"Corrupt PNG");
4752 if (j == 0) filter = first_row_filter[filter];
4757 memcpy(cur, raw, nk);
4760 memcpy(cur, raw, filter_bytes);
4761 for (k = filter_bytes; k < nk; ++k)
4762 cur[k] = STBI__BYTECAST(raw[k] + cur[k-filter_bytes]);
4765 for (k = 0; k < nk; ++k)
4766 cur[k] = STBI__BYTECAST(raw[k] + prior[k]);
4769 for (k = 0; k < filter_bytes; ++k)
4770 cur[k] = STBI__BYTECAST(raw[k] + (prior[k]>>1));
4771 for (k = filter_bytes; k < nk; ++k)
4772 cur[k] = STBI__BYTECAST(raw[k] + ((prior[k] + cur[k-filter_bytes])>>1));
4775 for (k = 0; k < filter_bytes; ++k)
4776 cur[k] = STBI__BYTECAST(raw[k] + prior[k]);
4777 for (k = filter_bytes; k < nk; ++k)
4778 cur[k] = STBI__BYTECAST(raw[k] + stbi__paeth(cur[k-filter_bytes], prior[k], prior[k-filter_bytes]));
4780 case STBI__F_avg_first:
4781 memcpy(cur, raw, filter_bytes);
4782 for (k = filter_bytes; k < nk; ++k)
4783 cur[k] = STBI__BYTECAST(raw[k] + (cur[k-filter_bytes] >> 1));
4791 stbi_uc scale = (color == 0) ? stbi__depth_scale_table[depth] : 1;
4793 stbi_uc *out = dest;
4795 stbi__uint32 nsmp = x*img_n;
4799 for (i=0; i < nsmp; ++i) {
4800 if ((i & 1) == 0) inb = *in++;
4801 *out++ = scale * (inb >> 4);
4804 }
else if (depth == 2) {
4805 for (i=0; i < nsmp; ++i) {
4806 if ((i & 3) == 0) inb = *in++;
4807 *out++ = scale * (inb >> 6);
4811 STBI_ASSERT(depth == 1);
4812 for (i=0; i < nsmp; ++i) {
4813 if ((i & 7) == 0) inb = *in++;
4814 *out++ = scale * (inb >> 7);
4821 stbi__create_png_alpha_expand8(dest, dest, x, img_n);
4822 }
else if (depth == 8) {
4824 memcpy(dest, cur, x*img_n);
4826 stbi__create_png_alpha_expand8(dest, cur, x, img_n);
4827 }
else if (depth == 16) {
4829 stbi__uint16 *dest16 = (stbi__uint16*)dest;
4830 stbi__uint32 nsmp = x*img_n;
4832 if (img_n == out_n) {
4833 for (i = 0; i < nsmp; ++i, ++dest16, cur += 2)
4834 *dest16 = (cur[0] << 8) | cur[1];
4836 STBI_ASSERT(img_n+1 == out_n);
4838 for (i = 0; i < x; ++i, dest16 += 2, cur += 2) {
4839 dest16[0] = (cur[0] << 8) | cur[1];
4843 STBI_ASSERT(img_n == 3);
4844 for (i = 0; i < x; ++i, dest16 += 4, cur += 6) {
4845 dest16[0] = (cur[0] << 8) | cur[1];
4846 dest16[1] = (cur[2] << 8) | cur[3];
4847 dest16[2] = (cur[4] << 8) | cur[5];
4855 STBI_FREE(filter_buf);
4856 if (!all_ok)
return 0;
4861static int stbi__create_png_image(stbi__png *a, stbi_uc *image_data, stbi__uint32 image_data_len,
int out_n,
int depth,
int color,
int interlaced)
4863 int bytes = (depth == 16 ? 2 : 1);
4864 int out_bytes = out_n * bytes;
4868 return stbi__create_png_image_raw(a, image_data, image_data_len, out_n, a->s->img_x, a->s->img_y, depth, color);
4871 final = (stbi_uc *) stbi__malloc_mad3(a->s->img_x, a->s->img_y, out_bytes, 0);
4872 if (!
final)
return stbi__err(
"outofmem",
"Out of memory");
4873 for (p=0; p < 7; ++p) {
4874 int xorig[] = { 0,4,0,2,0,1,0 };
4875 int yorig[] = { 0,0,4,0,2,0,1 };
4876 int xspc[] = { 8,8,4,4,2,2,1 };
4877 int yspc[] = { 8,8,8,4,4,2,2 };
4880 x = (a->s->img_x - xorig[p] + xspc[p]-1) / xspc[p];
4881 y = (a->s->img_y - yorig[p] + yspc[p]-1) / yspc[p];
4883 stbi__uint32 img_len = ((((a->s->img_n * x * depth) + 7) >> 3) + 1) * y;
4884 if (!stbi__create_png_image_raw(a, image_data, image_data_len, out_n, x, y, depth, color)) {
4888 for (j=0; j < y; ++j) {
4889 for (i=0; i < x; ++i) {
4890 int out_y = j*yspc[p]+yorig[p];
4891 int out_x = i*xspc[p]+xorig[p];
4892 memcpy(
final + out_y*a->s->img_x*out_bytes + out_x*out_bytes,
4893 a->out + (j*x+i)*out_bytes, out_bytes);
4897 image_data += img_len;
4898 image_data_len -= img_len;
4906static int stbi__compute_transparency(stbi__png *z, stbi_uc tc[3],
int out_n)
4908 stbi__context *s = z->s;
4909 stbi__uint32 i, pixel_count = s->img_x * s->img_y;
4910 stbi_uc *p = z->out;
4914 STBI_ASSERT(out_n == 2 || out_n == 4);
4917 for (i=0; i < pixel_count; ++i) {
4918 p[1] = (p[0] == tc[0] ? 0 : 255);
4922 for (i=0; i < pixel_count; ++i) {
4923 if (p[0] == tc[0] && p[1] == tc[1] && p[2] == tc[2])
4931static int stbi__compute_transparency16(stbi__png *z, stbi__uint16 tc[3],
int out_n)
4933 stbi__context *s = z->s;
4934 stbi__uint32 i, pixel_count = s->img_x * s->img_y;
4935 stbi__uint16 *p = (stbi__uint16*) z->out;
4939 STBI_ASSERT(out_n == 2 || out_n == 4);
4942 for (i = 0; i < pixel_count; ++i) {
4943 p[1] = (p[0] == tc[0] ? 0 : 65535);
4947 for (i = 0; i < pixel_count; ++i) {
4948 if (p[0] == tc[0] && p[1] == tc[1] && p[2] == tc[2])
4956static int stbi__expand_png_palette(stbi__png *a, stbi_uc *palette,
int len,
int pal_img_n)
4958 stbi__uint32 i, pixel_count = a->s->img_x * a->s->img_y;
4959 stbi_uc *p, *temp_out, *orig = a->out;
4961 p = (stbi_uc *) stbi__malloc_mad2(pixel_count, pal_img_n, 0);
4962 if (p == NULL)
return stbi__err(
"outofmem",
"Out of memory");
4967 if (pal_img_n == 3) {
4968 for (i=0; i < pixel_count; ++i) {
4971 p[1] = palette[n+1];
4972 p[2] = palette[n+2];
4976 for (i=0; i < pixel_count; ++i) {
4979 p[1] = palette[n+1];
4980 p[2] = palette[n+2];
4981 p[3] = palette[n+3];
4993static int stbi__unpremultiply_on_load_global = 0;
4994static int stbi__de_iphone_flag_global = 0;
4996STBIDEF
void stbi_set_unpremultiply_on_load(
int flag_true_if_should_unpremultiply)
4998 stbi__unpremultiply_on_load_global = flag_true_if_should_unpremultiply;
5001STBIDEF
void stbi_convert_iphone_png_to_rgb(
int flag_true_if_should_convert)
5003 stbi__de_iphone_flag_global = flag_true_if_should_convert;
5006#ifndef STBI_THREAD_LOCAL
5007#define stbi__unpremultiply_on_load stbi__unpremultiply_on_load_global
5008#define stbi__de_iphone_flag stbi__de_iphone_flag_global
5010static STBI_THREAD_LOCAL
int stbi__unpremultiply_on_load_local, stbi__unpremultiply_on_load_set;
5011static STBI_THREAD_LOCAL
int stbi__de_iphone_flag_local, stbi__de_iphone_flag_set;
5013STBIDEF
void stbi_set_unpremultiply_on_load_thread(
int flag_true_if_should_unpremultiply)
5015 stbi__unpremultiply_on_load_local = flag_true_if_should_unpremultiply;
5016 stbi__unpremultiply_on_load_set = 1;
5019STBIDEF
void stbi_convert_iphone_png_to_rgb_thread(
int flag_true_if_should_convert)
5021 stbi__de_iphone_flag_local = flag_true_if_should_convert;
5022 stbi__de_iphone_flag_set = 1;
5025#define stbi__unpremultiply_on_load (stbi__unpremultiply_on_load_set \
5026 ? stbi__unpremultiply_on_load_local \
5027 : stbi__unpremultiply_on_load_global)
5028#define stbi__de_iphone_flag (stbi__de_iphone_flag_set \
5029 ? stbi__de_iphone_flag_local \
5030 : stbi__de_iphone_flag_global)
5033static void stbi__de_iphone(stbi__png *z)
5035 stbi__context *s = z->s;
5036 stbi__uint32 i, pixel_count = s->img_x * s->img_y;
5037 stbi_uc *p = z->out;
5039 if (s->img_out_n == 3) {
5040 for (i=0; i < pixel_count; ++i) {
5047 STBI_ASSERT(s->img_out_n == 4);
5048 if (stbi__unpremultiply_on_load) {
5050 for (i=0; i < pixel_count; ++i) {
5054 stbi_uc half = a / 2;
5055 p[0] = (p[2] * 255 + half) / a;
5056 p[1] = (p[1] * 255 + half) / a;
5057 p[2] = ( t * 255 + half) / a;
5066 for (i=0; i < pixel_count; ++i) {
5076#define STBI__PNG_TYPE(a,b,c,d) (((unsigned) (a) << 24) + ((unsigned) (b) << 16) + ((unsigned) (c) << 8) + (unsigned) (d))
5078static int stbi__parse_png_file(stbi__png *z,
int scan,
int req_comp)
5080 stbi_uc palette[1024], pal_img_n=0;
5081 stbi_uc has_trans=0, tc[3]={0};
5082 stbi__uint16 tc16[3];
5083 stbi__uint32 ioff=0, idata_limit=0, i, pal_len=0;
5084 int first=1,k,interlace=0, color=0, is_iphone=0;
5085 stbi__context *s = z->s;
5091 if (!stbi__check_png_header(s))
return 0;
5093 if (scan == STBI__SCAN_type)
return 1;
5096 stbi__pngchunk c = stbi__get_chunk_header(s);
5098 case STBI__PNG_TYPE(
'C',
'g',
'B',
'I'):
5100 stbi__skip(s, c.length);
5102 case STBI__PNG_TYPE(
'I',
'H',
'D',
'R'): {
5104 if (!first)
return stbi__err(
"multiple IHDR",
"Corrupt PNG");
5106 if (c.length != 13)
return stbi__err(
"bad IHDR len",
"Corrupt PNG");
5107 s->img_x = stbi__get32be(s);
5108 s->img_y = stbi__get32be(s);
5109 if (s->img_y > STBI_MAX_DIMENSIONS)
return stbi__err(
"too large",
"Very large image (corrupt?)");
5110 if (s->img_x > STBI_MAX_DIMENSIONS)
return stbi__err(
"too large",
"Very large image (corrupt?)");
5111 z->depth = stbi__get8(s);
if (z->depth != 1 && z->depth != 2 && z->depth != 4 && z->depth != 8 && z->depth != 16)
return stbi__err(
"1/2/4/8/16-bit only",
"PNG not supported: 1/2/4/8/16-bit only");
5112 color = stbi__get8(s);
if (color > 6)
return stbi__err(
"bad ctype",
"Corrupt PNG");
5113 if (color == 3 && z->depth == 16)
return stbi__err(
"bad ctype",
"Corrupt PNG");
5114 if (color == 3) pal_img_n = 3;
else if (color & 1)
return stbi__err(
"bad ctype",
"Corrupt PNG");
5115 comp = stbi__get8(s);
if (comp)
return stbi__err(
"bad comp method",
"Corrupt PNG");
5116 filter= stbi__get8(s);
if (filter)
return stbi__err(
"bad filter method",
"Corrupt PNG");
5117 interlace = stbi__get8(s);
if (interlace>1)
return stbi__err(
"bad interlace method",
"Corrupt PNG");
5118 if (!s->img_x || !s->img_y)
return stbi__err(
"0-pixel image",
"Corrupt PNG");
5120 s->img_n = (color & 2 ? 3 : 1) + (color & 4 ? 1 : 0);
5121 if ((1 << 30) / s->img_x / s->img_n < s->img_y)
return stbi__err(
"too large",
"Image too large to decode");
5126 if ((1 << 30) / s->img_x / 4 < s->img_y)
return stbi__err(
"too large",
"Corrupt PNG");
5132 case STBI__PNG_TYPE(
'P',
'L',
'T',
'E'): {
5133 if (first)
return stbi__err(
"first not IHDR",
"Corrupt PNG");
5134 if (c.length > 256*3)
return stbi__err(
"invalid PLTE",
"Corrupt PNG");
5135 pal_len = c.length / 3;
5136 if (pal_len * 3 != c.length)
return stbi__err(
"invalid PLTE",
"Corrupt PNG");
5137 for (i=0; i < pal_len; ++i) {
5138 palette[i*4+0] = stbi__get8(s);
5139 palette[i*4+1] = stbi__get8(s);
5140 palette[i*4+2] = stbi__get8(s);
5141 palette[i*4+3] = 255;
5146 case STBI__PNG_TYPE(
't',
'R',
'N',
'S'): {
5147 if (first)
return stbi__err(
"first not IHDR",
"Corrupt PNG");
5148 if (z->idata)
return stbi__err(
"tRNS after IDAT",
"Corrupt PNG");
5150 if (scan == STBI__SCAN_header) { s->img_n = 4;
return 1; }
5151 if (pal_len == 0)
return stbi__err(
"tRNS before PLTE",
"Corrupt PNG");
5152 if (c.length > pal_len)
return stbi__err(
"bad tRNS len",
"Corrupt PNG");
5154 for (i=0; i < c.length; ++i)
5155 palette[i*4+3] = stbi__get8(s);
5157 if (!(s->img_n & 1))
return stbi__err(
"tRNS with alpha",
"Corrupt PNG");
5158 if (c.length != (stbi__uint32) s->img_n*2)
return stbi__err(
"bad tRNS len",
"Corrupt PNG");
5161 if (scan == STBI__SCAN_header) { ++s->img_n;
return 1; }
5162 if (z->depth == 16) {
5163 for (k = 0; k < s->img_n; ++k) tc16[k] = (stbi__uint16)stbi__get16be(s);
5165 for (k = 0; k < s->img_n; ++k) tc[k] = (stbi_uc)(stbi__get16be(s) & 255) * stbi__depth_scale_table[z->depth];
5171 case STBI__PNG_TYPE(
'I',
'D',
'A',
'T'): {
5172 if (first)
return stbi__err(
"first not IHDR",
"Corrupt PNG");
5173 if (pal_img_n && !pal_len)
return stbi__err(
"no PLTE",
"Corrupt PNG");
5174 if (scan == STBI__SCAN_header) {
5177 s->img_n = pal_img_n;
5180 if (c.length > (1u << 30))
return stbi__err(
"IDAT size limit",
"IDAT section larger than 2^30 bytes");
5181 if ((
int)(ioff + c.length) < (
int)ioff)
return 0;
5182 if (ioff + c.length > idata_limit) {
5183 stbi__uint32 idata_limit_old = idata_limit;
5185 if (idata_limit == 0) idata_limit = c.length > 4096 ? c.length : 4096;
5186 while (ioff + c.length > idata_limit)
5188 STBI_NOTUSED(idata_limit_old);
5189 p = (stbi_uc *) STBI_REALLOC_SIZED(z->idata, idata_limit_old, idata_limit);
if (p == NULL)
return stbi__err(
"outofmem",
"Out of memory");
5192 if (!stbi__getn(s, z->idata+ioff,c.length))
return stbi__err(
"outofdata",
"Corrupt PNG");
5197 case STBI__PNG_TYPE(
'I',
'E',
'N',
'D'): {
5198 stbi__uint32 raw_len, bpl;
5199 if (first)
return stbi__err(
"first not IHDR",
"Corrupt PNG");
5200 if (scan != STBI__SCAN_load)
return 1;
5201 if (z->idata == NULL)
return stbi__err(
"no IDAT",
"Corrupt PNG");
5203 bpl = (s->img_x * z->depth + 7) / 8;
5204 raw_len = bpl * s->img_y * s->img_n + s->img_y ;
5205 z->expanded = (stbi_uc *) stbi_zlib_decode_malloc_guesssize_headerflag((
char *) z->idata, ioff, raw_len, (
int *) &raw_len, !is_iphone);
5206 if (z->expanded == NULL)
return 0;
5207 STBI_FREE(z->idata); z->idata = NULL;
5208 if ((req_comp == s->img_n+1 && req_comp != 3 && !pal_img_n) || has_trans)
5209 s->img_out_n = s->img_n+1;
5211 s->img_out_n = s->img_n;
5212 if (!stbi__create_png_image(z, z->expanded, raw_len, s->img_out_n, z->depth, color, interlace))
return 0;
5214 if (z->depth == 16) {
5215 if (!stbi__compute_transparency16(z, tc16, s->img_out_n))
return 0;
5217 if (!stbi__compute_transparency(z, tc, s->img_out_n))
return 0;
5220 if (is_iphone && stbi__de_iphone_flag && s->img_out_n > 2)
5224 s->img_n = pal_img_n;
5225 s->img_out_n = pal_img_n;
5226 if (req_comp >= 3) s->img_out_n = req_comp;
5227 if (!stbi__expand_png_palette(z, palette, pal_len, s->img_out_n))
5229 }
else if (has_trans) {
5233 STBI_FREE(z->expanded); z->expanded = NULL;
5241 if (first)
return stbi__err(
"first not IHDR",
"Corrupt PNG");
5242 if ((c.type & (1 << 29)) == 0) {
5243 #ifndef STBI_NO_FAILURE_STRINGS
5245 static char invalid_chunk[] =
"XXXX PNG chunk not known";
5246 invalid_chunk[0] = STBI__BYTECAST(c.type >> 24);
5247 invalid_chunk[1] = STBI__BYTECAST(c.type >> 16);
5248 invalid_chunk[2] = STBI__BYTECAST(c.type >> 8);
5249 invalid_chunk[3] = STBI__BYTECAST(c.type >> 0);
5251 return stbi__err(invalid_chunk,
"PNG not supported: unknown PNG chunk type");
5253 stbi__skip(s, c.length);
5261static void *stbi__do_png(stbi__png *p,
int *x,
int *y,
int *n,
int req_comp, stbi__result_info *ri)
5264 if (req_comp < 0 || req_comp > 4)
return stbi__errpuc(
"bad req_comp",
"Internal error");
5265 if (stbi__parse_png_file(p, STBI__SCAN_load, req_comp)) {
5267 ri->bits_per_channel = 8;
5268 else if (p->depth == 16)
5269 ri->bits_per_channel = 16;
5271 return stbi__errpuc(
"bad bits_per_channel",
"PNG not supported: unsupported color depth");
5274 if (req_comp && req_comp != p->s->img_out_n) {
5275 if (ri->bits_per_channel == 8)
5276 result = stbi__convert_format((
unsigned char *) result, p->s->img_out_n, req_comp, p->s->img_x, p->s->img_y);
5278 result = stbi__convert_format16((stbi__uint16 *) result, p->s->img_out_n, req_comp, p->s->img_x, p->s->img_y);
5279 p->s->img_out_n = req_comp;
5280 if (result == NULL)
return result;
5284 if (n) *n = p->s->img_n;
5286 STBI_FREE(p->out); p->out = NULL;
5287 STBI_FREE(p->expanded); p->expanded = NULL;
5288 STBI_FREE(p->idata); p->idata = NULL;
5293static void *stbi__png_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri)
5297 return stbi__do_png(&p, x,y,comp,req_comp, ri);
5300static int stbi__png_test(stbi__context *s)
5303 r = stbi__check_png_header(s);
5308static int stbi__png_info_raw(stbi__png *p,
int *x,
int *y,
int *comp)
5310 if (!stbi__parse_png_file(p, STBI__SCAN_header, 0)) {
5311 stbi__rewind( p->s );
5314 if (x) *x = p->s->img_x;
5315 if (y) *y = p->s->img_y;
5316 if (comp) *comp = p->s->img_n;
5320static int stbi__png_info(stbi__context *s,
int *x,
int *y,
int *comp)
5324 return stbi__png_info_raw(&p, x, y, comp);
5327static int stbi__png_is16(stbi__context *s)
5331 if (!stbi__png_info_raw(&p, NULL, NULL, NULL))
5333 if (p.depth != 16) {
5344static int stbi__bmp_test_raw(stbi__context *s)
5348 if (stbi__get8(s) !=
'B')
return 0;
5349 if (stbi__get8(s) !=
'M')
return 0;
5354 sz = stbi__get32le(s);
5355 r = (sz == 12 || sz == 40 || sz == 56 || sz == 108 || sz == 124);
5359static int stbi__bmp_test(stbi__context *s)
5361 int r = stbi__bmp_test_raw(s);
5368static int stbi__high_bit(
unsigned int z)
5371 if (z == 0)
return -1;
5372 if (z >= 0x10000) { n += 16; z >>= 16; }
5373 if (z >= 0x00100) { n += 8; z >>= 8; }
5374 if (z >= 0x00010) { n += 4; z >>= 4; }
5375 if (z >= 0x00004) { n += 2; z >>= 2; }
5376 if (z >= 0x00002) { n += 1; }
5380static int stbi__bitcount(
unsigned int a)
5382 a = (a & 0x55555555) + ((a >> 1) & 0x55555555);
5383 a = (a & 0x33333333) + ((a >> 2) & 0x33333333);
5384 a = (a + (a >> 4)) & 0x0f0f0f0f;
5386 a = (a + (a >> 16));
5393static int stbi__shiftsigned(
unsigned int v,
int shift,
int bits)
5395 static unsigned int mul_table[9] = {
5397 0xff, 0x55, 0x49, 0x11,
5398 0x21, 0x41, 0x81, 0x01,
5400 static unsigned int shift_table[9] = {
5407 STBI_ASSERT(v < 256);
5409 STBI_ASSERT(bits >= 0 && bits <= 8);
5410 return (
int) ((unsigned) v * mul_table[bits]) >> shift_table[bits];
5415 int bpp, offset, hsz;
5416 unsigned int mr,mg,mb,ma, all_a;
5420static int stbi__bmp_set_mask_defaults(stbi__bmp_data *info,
int compress)
5426 if (compress == 0) {
5427 if (info->bpp == 16) {
5428 info->mr = 31u << 10;
5429 info->mg = 31u << 5;
5430 info->mb = 31u << 0;
5431 }
else if (info->bpp == 32) {
5432 info->mr = 0xffu << 16;
5433 info->mg = 0xffu << 8;
5434 info->mb = 0xffu << 0;
5435 info->ma = 0xffu << 24;
5439 info->mr = info->mg = info->mb = info->ma = 0;
5446static void *stbi__bmp_parse_header(stbi__context *s, stbi__bmp_data *info)
5449 if (stbi__get8(s) !=
'B' || stbi__get8(s) !=
'M')
return stbi__errpuc(
"not BMP",
"Corrupt BMP");
5453 info->offset = stbi__get32le(s);
5454 info->hsz = hsz = stbi__get32le(s);
5455 info->mr = info->mg = info->mb = info->ma = 0;
5456 info->extra_read = 14;
5458 if (info->offset < 0)
return stbi__errpuc(
"bad BMP",
"bad BMP");
5460 if (hsz != 12 && hsz != 40 && hsz != 56 && hsz != 108 && hsz != 124)
return stbi__errpuc(
"unknown BMP",
"BMP type not supported: unknown");
5462 s->img_x = stbi__get16le(s);
5463 s->img_y = stbi__get16le(s);
5465 s->img_x = stbi__get32le(s);
5466 s->img_y = stbi__get32le(s);
5468 if (stbi__get16le(s) != 1)
return stbi__errpuc(
"bad BMP",
"bad BMP");
5469 info->bpp = stbi__get16le(s);
5471 int compress = stbi__get32le(s);
5472 if (compress == 1 || compress == 2)
return stbi__errpuc(
"BMP RLE",
"BMP type not supported: RLE");
5473 if (compress >= 4)
return stbi__errpuc(
"BMP JPEG/PNG",
"BMP type not supported: unsupported compression");
5474 if (compress == 3 && info->bpp != 16 && info->bpp != 32)
return stbi__errpuc(
"bad BMP",
"bad BMP");
5480 if (hsz == 40 || hsz == 56) {
5487 if (info->bpp == 16 || info->bpp == 32) {
5488 if (compress == 0) {
5489 stbi__bmp_set_mask_defaults(info, compress);
5490 }
else if (compress == 3) {
5491 info->mr = stbi__get32le(s);
5492 info->mg = stbi__get32le(s);
5493 info->mb = stbi__get32le(s);
5494 info->extra_read += 12;
5496 if (info->mr == info->mg && info->mg == info->mb) {
5498 return stbi__errpuc(
"bad BMP",
"bad BMP");
5501 return stbi__errpuc(
"bad BMP",
"bad BMP");
5506 if (hsz != 108 && hsz != 124)
5507 return stbi__errpuc(
"bad BMP",
"bad BMP");
5508 info->mr = stbi__get32le(s);
5509 info->mg = stbi__get32le(s);
5510 info->mb = stbi__get32le(s);
5511 info->ma = stbi__get32le(s);
5513 stbi__bmp_set_mask_defaults(info, compress);
5515 for (i=0; i < 12; ++i)
5529static void *stbi__bmp_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri)
5532 unsigned int mr=0,mg=0,mb=0,ma=0, all_a;
5533 stbi_uc pal[256][4];
5534 int psize=0,i,j,width;
5535 int flip_vertically, pad, target;
5536 stbi__bmp_data info;
5540 if (stbi__bmp_parse_header(s, &info) == NULL)
5543 flip_vertically = ((int) s->img_y) > 0;
5544 s->img_y = abs((
int) s->img_y);
5546 if (s->img_y > STBI_MAX_DIMENSIONS)
return stbi__errpuc(
"too large",
"Very large image (corrupt?)");
5547 if (s->img_x > STBI_MAX_DIMENSIONS)
return stbi__errpuc(
"too large",
"Very large image (corrupt?)");
5555 if (info.hsz == 12) {
5557 psize = (info.offset - info.extra_read - 24) / 3;
5560 psize = (info.offset - info.extra_read - info.hsz) >> 2;
5565 int bytes_read_so_far = s->callback_already_read + (int)(s->img_buffer - s->img_buffer_original);
5566 int header_limit = 1024;
5567 int extra_data_limit = 256*4;
5568 if (bytes_read_so_far <= 0 || bytes_read_so_far > header_limit) {
5569 return stbi__errpuc(
"bad header",
"Corrupt BMP");
5575 if (info.offset < bytes_read_so_far || info.offset - bytes_read_so_far > extra_data_limit) {
5576 return stbi__errpuc(
"bad offset",
"Corrupt BMP");
5578 stbi__skip(s, info.offset - bytes_read_so_far);
5582 if (info.bpp == 24 && ma == 0xff000000)
5585 s->img_n = ma ? 4 : 3;
5586 if (req_comp && req_comp >= 3)
5592 if (!stbi__mad3sizes_valid(target, s->img_x, s->img_y, 0))
5593 return stbi__errpuc(
"too large",
"Corrupt BMP");
5595 out = (stbi_uc *) stbi__malloc_mad3(target, s->img_x, s->img_y, 0);
5596 if (!out)
return stbi__errpuc(
"outofmem",
"Out of memory");
5597 if (info.bpp < 16) {
5599 if (psize == 0 || psize > 256) { STBI_FREE(out);
return stbi__errpuc(
"invalid",
"Corrupt BMP"); }
5600 for (i=0; i < psize; ++i) {
5601 pal[i][2] = stbi__get8(s);
5602 pal[i][1] = stbi__get8(s);
5603 pal[i][0] = stbi__get8(s);
5604 if (info.hsz != 12) stbi__get8(s);
5607 stbi__skip(s, info.offset - info.extra_read - info.hsz - psize * (info.hsz == 12 ? 3 : 4));
5608 if (info.bpp == 1) width = (s->img_x + 7) >> 3;
5609 else if (info.bpp == 4) width = (s->img_x + 1) >> 1;
5610 else if (info.bpp == 8) width = s->img_x;
5611 else { STBI_FREE(out);
return stbi__errpuc(
"bad bpp",
"Corrupt BMP"); }
5613 if (info.bpp == 1) {
5614 for (j=0; j < (int) s->img_y; ++j) {
5615 int bit_offset = 7, v = stbi__get8(s);
5616 for (i=0; i < (int) s->img_x; ++i) {
5617 int color = (v>>bit_offset)&0x1;
5618 out[z++] = pal[color][0];
5619 out[z++] = pal[color][1];
5620 out[z++] = pal[color][2];
5621 if (target == 4) out[z++] = 255;
5622 if (i+1 == (
int) s->img_x)
break;
5623 if((--bit_offset) < 0) {
5631 for (j=0; j < (int) s->img_y; ++j) {
5632 for (i=0; i < (int) s->img_x; i += 2) {
5633 int v=stbi__get8(s),v2=0;
5634 if (info.bpp == 4) {
5638 out[z++] = pal[v][0];
5639 out[z++] = pal[v][1];
5640 out[z++] = pal[v][2];
5641 if (target == 4) out[z++] = 255;
5642 if (i+1 == (
int) s->img_x)
break;
5643 v = (info.bpp == 8) ? stbi__get8(s) : v2;
5644 out[z++] = pal[v][0];
5645 out[z++] = pal[v][1];
5646 out[z++] = pal[v][2];
5647 if (target == 4) out[z++] = 255;
5653 int rshift=0,gshift=0,bshift=0,ashift=0,rcount=0,gcount=0,bcount=0,acount=0;
5656 stbi__skip(s, info.offset - info.extra_read - info.hsz);
5657 if (info.bpp == 24) width = 3 * s->img_x;
5658 else if (info.bpp == 16) width = 2*s->img_x;
5661 if (info.bpp == 24) {
5663 }
else if (info.bpp == 32) {
5664 if (mb == 0xff && mg == 0xff00 && mr == 0x00ff0000 && ma == 0xff000000)
5668 if (!mr || !mg || !mb) { STBI_FREE(out);
return stbi__errpuc(
"bad masks",
"Corrupt BMP"); }
5670 rshift = stbi__high_bit(mr)-7; rcount = stbi__bitcount(mr);
5671 gshift = stbi__high_bit(mg)-7; gcount = stbi__bitcount(mg);
5672 bshift = stbi__high_bit(mb)-7; bcount = stbi__bitcount(mb);
5673 ashift = stbi__high_bit(ma)-7; acount = stbi__bitcount(ma);
5674 if (rcount > 8 || gcount > 8 || bcount > 8 || acount > 8) { STBI_FREE(out);
return stbi__errpuc(
"bad masks",
"Corrupt BMP"); }
5676 for (j=0; j < (int) s->img_y; ++j) {
5678 for (i=0; i < (int) s->img_x; ++i) {
5680 out[z+2] = stbi__get8(s);
5681 out[z+1] = stbi__get8(s);
5682 out[z+0] = stbi__get8(s);
5684 a = (easy == 2 ? stbi__get8(s) : 255);
5686 if (target == 4) out[z++] = a;
5690 for (i=0; i < (int) s->img_x; ++i) {
5691 stbi__uint32 v = (bpp == 16 ? (stbi__uint32) stbi__get16le(s) : stbi__get32le(s));
5693 out[z++] = STBI__BYTECAST(stbi__shiftsigned(v & mr, rshift, rcount));
5694 out[z++] = STBI__BYTECAST(stbi__shiftsigned(v & mg, gshift, gcount));
5695 out[z++] = STBI__BYTECAST(stbi__shiftsigned(v & mb, bshift, bcount));
5696 a = (ma ? stbi__shiftsigned(v & ma, ashift, acount) : 255);
5698 if (target == 4) out[z++] = STBI__BYTECAST(a);
5706 if (target == 4 && all_a == 0)
5707 for (i=4*s->img_x*s->img_y-1; i >= 0; i -= 4)
5710 if (flip_vertically) {
5712 for (j=0; j < (int) s->img_y>>1; ++j) {
5713 stbi_uc *p1 = out + j *s->img_x*target;
5714 stbi_uc *p2 = out + (s->img_y-1-j)*s->img_x*target;
5715 for (i=0; i < (int) s->img_x*target; ++i) {
5716 t = p1[i]; p1[i] = p2[i]; p2[i] = t;
5721 if (req_comp && req_comp != target) {
5722 out = stbi__convert_format(out, target, req_comp, s->img_x, s->img_y);
5723 if (out == NULL)
return out;
5728 if (comp) *comp = s->img_n;
5737static int stbi__tga_get_comp(
int bits_per_pixel,
int is_grey,
int* is_rgb16)
5740 if (is_rgb16) *is_rgb16 = 0;
5741 switch(bits_per_pixel) {
5742 case 8:
return STBI_grey;
5743 case 16:
if(is_grey)
return STBI_grey_alpha;
5745 case 15:
if(is_rgb16) *is_rgb16 = 1;
5748 case 32:
return bits_per_pixel/8;
5753static int stbi__tga_info(stbi__context *s,
int *x,
int *y,
int *comp)
5755 int tga_w, tga_h, tga_comp, tga_image_type, tga_bits_per_pixel, tga_colormap_bpp;
5756 int sz, tga_colormap_type;
5758 tga_colormap_type = stbi__get8(s);
5759 if( tga_colormap_type > 1 ) {
5763 tga_image_type = stbi__get8(s);
5764 if ( tga_colormap_type == 1 ) {
5765 if (tga_image_type != 1 && tga_image_type != 9) {
5771 if ( (sz != 8) && (sz != 15) && (sz != 16) && (sz != 24) && (sz != 32) ) {
5776 tga_colormap_bpp = sz;
5778 if ( (tga_image_type != 2) && (tga_image_type != 3) && (tga_image_type != 10) && (tga_image_type != 11) ) {
5783 tga_colormap_bpp = 0;
5785 tga_w = stbi__get16le(s);
5790 tga_h = stbi__get16le(s);
5795 tga_bits_per_pixel = stbi__get8(s);
5797 if (tga_colormap_bpp != 0) {
5798 if((tga_bits_per_pixel != 8) && (tga_bits_per_pixel != 16)) {
5804 tga_comp = stbi__tga_get_comp(tga_colormap_bpp, 0, NULL);
5806 tga_comp = stbi__tga_get_comp(tga_bits_per_pixel, (tga_image_type == 3) || (tga_image_type == 11), NULL);
5814 if (comp) *comp = tga_comp;
5818static int stbi__tga_test(stbi__context *s)
5821 int sz, tga_color_type;
5823 tga_color_type = stbi__get8(s);
5824 if ( tga_color_type > 1 )
goto errorEnd;
5826 if ( tga_color_type == 1 ) {
5827 if (sz != 1 && sz != 9)
goto errorEnd;
5830 if ( (sz != 8) && (sz != 15) && (sz != 16) && (sz != 24) && (sz != 32) )
goto errorEnd;
5833 if ( (sz != 2) && (sz != 3) && (sz != 10) && (sz != 11) )
goto errorEnd;
5836 if ( stbi__get16le(s) < 1 )
goto errorEnd;
5837 if ( stbi__get16le(s) < 1 )
goto errorEnd;
5839 if ( (tga_color_type == 1) && (sz != 8) && (sz != 16) )
goto errorEnd;
5840 if ( (sz != 8) && (sz != 15) && (sz != 16) && (sz != 24) && (sz != 32) )
goto errorEnd;
5850static void stbi__tga_read_rgb16(stbi__context *s, stbi_uc* out)
5852 stbi__uint16 px = (stbi__uint16)stbi__get16le(s);
5853 stbi__uint16 fiveBitMask = 31;
5855 int r = (px >> 10) & fiveBitMask;
5856 int g = (px >> 5) & fiveBitMask;
5857 int b = px & fiveBitMask;
5859 out[0] = (stbi_uc)((r * 255)/31);
5860 out[1] = (stbi_uc)((g * 255)/31);
5861 out[2] = (stbi_uc)((b * 255)/31);
5869static void *stbi__tga_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri)
5872 int tga_offset = stbi__get8(s);
5873 int tga_indexed = stbi__get8(s);
5874 int tga_image_type = stbi__get8(s);
5876 int tga_palette_start = stbi__get16le(s);
5877 int tga_palette_len = stbi__get16le(s);
5878 int tga_palette_bits = stbi__get8(s);
5879 int tga_x_origin = stbi__get16le(s);
5880 int tga_y_origin = stbi__get16le(s);
5881 int tga_width = stbi__get16le(s);
5882 int tga_height = stbi__get16le(s);
5883 int tga_bits_per_pixel = stbi__get8(s);
5884 int tga_comp, tga_rgb16=0;
5885 int tga_inverted = stbi__get8(s);
5888 unsigned char *tga_data;
5889 unsigned char *tga_palette = NULL;
5891 unsigned char raw_data[4] = {0};
5893 int RLE_repeating = 0;
5894 int read_next_pixel = 1;
5896 STBI_NOTUSED(tga_x_origin);
5897 STBI_NOTUSED(tga_y_origin);
5899 if (tga_height > STBI_MAX_DIMENSIONS)
return stbi__errpuc(
"too large",
"Very large image (corrupt?)");
5900 if (tga_width > STBI_MAX_DIMENSIONS)
return stbi__errpuc(
"too large",
"Very large image (corrupt?)");
5903 if ( tga_image_type >= 8 )
5905 tga_image_type -= 8;
5908 tga_inverted = 1 - ((tga_inverted >> 5) & 1);
5911 if ( tga_indexed ) tga_comp = stbi__tga_get_comp(tga_palette_bits, 0, &tga_rgb16);
5912 else tga_comp = stbi__tga_get_comp(tga_bits_per_pixel, (tga_image_type == 3), &tga_rgb16);
5915 return stbi__errpuc(
"bad format",
"Can't find out TGA pixelformat");
5920 if (comp) *comp = tga_comp;
5922 if (!stbi__mad3sizes_valid(tga_width, tga_height, tga_comp, 0))
5923 return stbi__errpuc(
"too large",
"Corrupt TGA");
5925 tga_data = (
unsigned char*)stbi__malloc_mad3(tga_width, tga_height, tga_comp, 0);
5926 if (!tga_data)
return stbi__errpuc(
"outofmem",
"Out of memory");
5929 stbi__skip(s, tga_offset );
5931 if ( !tga_indexed && !tga_is_RLE && !tga_rgb16 ) {
5932 for (i=0; i < tga_height; ++i) {
5933 int row = tga_inverted ? tga_height -i - 1 : i;
5934 stbi_uc *tga_row = tga_data + row*tga_width*tga_comp;
5935 stbi__getn(s, tga_row, tga_width * tga_comp);
5941 if (tga_palette_len == 0) {
5942 STBI_FREE(tga_data);
5943 return stbi__errpuc(
"bad palette",
"Corrupt TGA");
5947 stbi__skip(s, tga_palette_start );
5949 tga_palette = (
unsigned char*)stbi__malloc_mad2(tga_palette_len, tga_comp, 0);
5951 STBI_FREE(tga_data);
5952 return stbi__errpuc(
"outofmem",
"Out of memory");
5955 stbi_uc *pal_entry = tga_palette;
5956 STBI_ASSERT(tga_comp == STBI_rgb);
5957 for (i=0; i < tga_palette_len; ++i) {
5958 stbi__tga_read_rgb16(s, pal_entry);
5959 pal_entry += tga_comp;
5961 }
else if (!stbi__getn(s, tga_palette, tga_palette_len * tga_comp)) {
5962 STBI_FREE(tga_data);
5963 STBI_FREE(tga_palette);
5964 return stbi__errpuc(
"bad palette",
"Corrupt TGA");
5968 for (i=0; i < tga_width * tga_height; ++i)
5973 if ( RLE_count == 0 )
5976 int RLE_cmd = stbi__get8(s);
5977 RLE_count = 1 + (RLE_cmd & 127);
5978 RLE_repeating = RLE_cmd >> 7;
5979 read_next_pixel = 1;
5980 }
else if ( !RLE_repeating )
5982 read_next_pixel = 1;
5986 read_next_pixel = 1;
5989 if ( read_next_pixel )
5995 int pal_idx = (tga_bits_per_pixel == 8) ? stbi__get8(s) : stbi__get16le(s);
5996 if ( pal_idx >= tga_palette_len ) {
6000 pal_idx *= tga_comp;
6001 for (j = 0; j < tga_comp; ++j) {
6002 raw_data[j] = tga_palette[pal_idx+j];
6004 }
else if(tga_rgb16) {
6005 STBI_ASSERT(tga_comp == STBI_rgb);
6006 stbi__tga_read_rgb16(s, raw_data);
6009 for (j = 0; j < tga_comp; ++j) {
6010 raw_data[j] = stbi__get8(s);
6014 read_next_pixel = 0;
6018 for (j = 0; j < tga_comp; ++j)
6019 tga_data[i*tga_comp+j] = raw_data[j];
6027 for (j = 0; j*2 < tga_height; ++j)
6029 int index1 = j * tga_width * tga_comp;
6030 int index2 = (tga_height - 1 - j) * tga_width * tga_comp;
6031 for (i = tga_width * tga_comp; i > 0; --i)
6033 unsigned char temp = tga_data[index1];
6034 tga_data[index1] = tga_data[index2];
6035 tga_data[index2] = temp;
6042 if ( tga_palette != NULL )
6044 STBI_FREE( tga_palette );
6049 if (tga_comp >= 3 && !tga_rgb16)
6051 unsigned char* tga_pixel = tga_data;
6052 for (i=0; i < tga_width * tga_height; ++i)
6054 unsigned char temp = tga_pixel[0];
6055 tga_pixel[0] = tga_pixel[2];
6056 tga_pixel[2] = temp;
6057 tga_pixel += tga_comp;
6062 if (req_comp && req_comp != tga_comp)
6063 tga_data = stbi__convert_format(tga_data, tga_comp, req_comp, tga_width, tga_height);
6067 tga_palette_start = tga_palette_len = tga_palette_bits =
6068 tga_x_origin = tga_y_origin = 0;
6069 STBI_NOTUSED(tga_palette_start);
6079static int stbi__psd_test(stbi__context *s)
6081 int r = (stbi__get32be(s) == 0x38425053);
6086static int stbi__psd_decode_rle(stbi__context *s, stbi_uc *p,
int pixelCount)
6088 int count, nleft, len;
6091 while ((nleft = pixelCount - count) > 0) {
6092 len = stbi__get8(s);
6095 }
else if (len < 128) {
6098 if (len > nleft)
return 0;
6105 }
else if (len > 128) {
6110 if (len > nleft)
return 0;
6111 val = stbi__get8(s);
6124static void *stbi__psd_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri,
int bpc)
6127 int channelCount, compression;
6135 if (stbi__get32be(s) != 0x38425053)
6136 return stbi__errpuc(
"not PSD",
"Corrupt PSD image");
6139 if (stbi__get16be(s) != 1)
6140 return stbi__errpuc(
"wrong version",
"Unsupported version of PSD image");
6146 channelCount = stbi__get16be(s);
6147 if (channelCount < 0 || channelCount > 16)
6148 return stbi__errpuc(
"wrong channel count",
"Unsupported number of channels in PSD image");
6151 h = stbi__get32be(s);
6152 w = stbi__get32be(s);
6154 if (h > STBI_MAX_DIMENSIONS)
return stbi__errpuc(
"too large",
"Very large image (corrupt?)");
6155 if (w > STBI_MAX_DIMENSIONS)
return stbi__errpuc(
"too large",
"Very large image (corrupt?)");
6158 bitdepth = stbi__get16be(s);
6159 if (bitdepth != 8 && bitdepth != 16)
6160 return stbi__errpuc(
"unsupported bit depth",
"PSD bit depth is not 8 or 16 bit");
6172 if (stbi__get16be(s) != 3)
6173 return stbi__errpuc(
"wrong color format",
"PSD is not in RGB color format");
6176 stbi__skip(s,stbi__get32be(s) );
6179 stbi__skip(s, stbi__get32be(s) );
6182 stbi__skip(s, stbi__get32be(s) );
6188 compression = stbi__get16be(s);
6189 if (compression > 1)
6190 return stbi__errpuc(
"bad compression",
"PSD has an unknown compression format");
6193 if (!stbi__mad3sizes_valid(4, w, h, 0))
6194 return stbi__errpuc(
"too large",
"Corrupt PSD");
6198 if (!compression && bitdepth == 16 && bpc == 16) {
6199 out = (stbi_uc *) stbi__malloc_mad3(8, w, h, 0);
6200 ri->bits_per_channel = 16;
6202 out = (stbi_uc *) stbi__malloc(4 * w*h);
6204 if (!out)
return stbi__errpuc(
"outofmem",
"Out of memory");
6222 stbi__skip(s, h * channelCount * 2 );
6225 for (channel = 0; channel < 4; channel++) {
6229 if (channel >= channelCount) {
6231 for (i = 0; i < pixelCount; i++, p += 4)
6232 *p = (channel == 3 ? 255 : 0);
6235 if (!stbi__psd_decode_rle(s, p, pixelCount)) {
6237 return stbi__errpuc(
"corrupt",
"bad RLE data");
6247 for (channel = 0; channel < 4; channel++) {
6248 if (channel >= channelCount) {
6250 if (bitdepth == 16 && bpc == 16) {
6251 stbi__uint16 *q = ((stbi__uint16 *) out) + channel;
6252 stbi__uint16 val = channel == 3 ? 65535 : 0;
6253 for (i = 0; i < pixelCount; i++, q += 4)
6256 stbi_uc *p = out+channel;
6257 stbi_uc val = channel == 3 ? 255 : 0;
6258 for (i = 0; i < pixelCount; i++, p += 4)
6262 if (ri->bits_per_channel == 16) {
6263 stbi__uint16 *q = ((stbi__uint16 *) out) + channel;
6264 for (i = 0; i < pixelCount; i++, q += 4)
6265 *q = (stbi__uint16) stbi__get16be(s);
6267 stbi_uc *p = out+channel;
6268 if (bitdepth == 16) {
6269 for (i = 0; i < pixelCount; i++, p += 4)
6270 *p = (stbi_uc) (stbi__get16be(s) >> 8);
6272 for (i = 0; i < pixelCount; i++, p += 4)
6281 if (channelCount >= 4) {
6282 if (ri->bits_per_channel == 16) {
6283 for (i=0; i < w*h; ++i) {
6284 stbi__uint16 *pixel = (stbi__uint16 *) out + 4*i;
6285 if (pixel[3] != 0 && pixel[3] != 65535) {
6286 float a = pixel[3] / 65535.0f;
6287 float ra = 1.0f / a;
6288 float inv_a = 65535.0f * (1 - ra);
6289 pixel[0] = (stbi__uint16) (pixel[0]*ra + inv_a);
6290 pixel[1] = (stbi__uint16) (pixel[1]*ra + inv_a);
6291 pixel[2] = (stbi__uint16) (pixel[2]*ra + inv_a);
6295 for (i=0; i < w*h; ++i) {
6296 unsigned char *pixel = out + 4*i;
6297 if (pixel[3] != 0 && pixel[3] != 255) {
6298 float a = pixel[3] / 255.0f;
6299 float ra = 1.0f / a;
6300 float inv_a = 255.0f * (1 - ra);
6301 pixel[0] = (
unsigned char) (pixel[0]*ra + inv_a);
6302 pixel[1] = (
unsigned char) (pixel[1]*ra + inv_a);
6303 pixel[2] = (
unsigned char) (pixel[2]*ra + inv_a);
6310 if (req_comp && req_comp != 4) {
6311 if (ri->bits_per_channel == 16)
6312 out = (stbi_uc *) stbi__convert_format16((stbi__uint16 *) out, 4, req_comp, w, h);
6314 out = stbi__convert_format(out, 4, req_comp, w, h);
6315 if (out == NULL)
return out;
6318 if (comp) *comp = 4;
6334static int stbi__pic_is4(stbi__context *s,
const char *str)
6338 if (stbi__get8(s) != (stbi_uc)str[i])
6344static int stbi__pic_test_core(stbi__context *s)
6348 if (!stbi__pic_is4(s,
"\x53\x80\xF6\x34"))
6354 if (!stbi__pic_is4(s,
"PICT"))
6362 stbi_uc size,type,channel;
6365static stbi_uc *stbi__readval(stbi__context *s,
int channel, stbi_uc *dest)
6369 for (i=0; i<4; ++i, mask>>=1) {
6370 if (channel & mask) {
6371 if (stbi__at_eof(s))
return stbi__errpuc(
"bad file",
"PIC file too short");
6372 dest[i]=stbi__get8(s);
6379static void stbi__copyval(
int channel,stbi_uc *dest,
const stbi_uc *src)
6383 for (i=0;i<4; ++i, mask>>=1)
6388static stbi_uc *stbi__pic_load_core(stbi__context *s,
int width,
int height,
int *comp, stbi_uc *result)
6390 int act_comp=0,num_packets=0,y,chained;
6391 stbi__pic_packet packets[10];
6396 stbi__pic_packet *packet;
6398 if (num_packets==
sizeof(packets)/
sizeof(packets[0]))
6399 return stbi__errpuc(
"bad format",
"too many packets");
6401 packet = &packets[num_packets++];
6403 chained = stbi__get8(s);
6404 packet->size = stbi__get8(s);
6405 packet->type = stbi__get8(s);
6406 packet->channel = stbi__get8(s);
6408 act_comp |= packet->channel;
6410 if (stbi__at_eof(s))
return stbi__errpuc(
"bad file",
"file too short (reading packets)");
6411 if (packet->size != 8)
return stbi__errpuc(
"bad format",
"packet isn't 8bpp");
6414 *comp = (act_comp & 0x10 ? 4 : 3);
6416 for(y=0; y<height; ++y) {
6419 for(packet_idx=0; packet_idx < num_packets; ++packet_idx) {
6420 stbi__pic_packet *packet = &packets[packet_idx];
6421 stbi_uc *dest = result+y*width*4;
6423 switch (packet->type) {
6425 return stbi__errpuc(
"bad format",
"packet has bad compression type");
6430 for(x=0;x<width;++x, dest+=4)
6431 if (!stbi__readval(s,packet->channel,dest))
6441 stbi_uc count,value[4];
6443 count=stbi__get8(s);
6444 if (stbi__at_eof(s))
return stbi__errpuc(
"bad file",
"file too short (pure read count)");
6447 count = (stbi_uc) left;
6449 if (!stbi__readval(s,packet->channel,value))
return 0;
6451 for(i=0; i<count; ++i,dest+=4)
6452 stbi__copyval(packet->channel,dest,value);
6461 int count = stbi__get8(s), i;
6462 if (stbi__at_eof(s))
return stbi__errpuc(
"bad file",
"file too short (mixed read count)");
6468 count = stbi__get16be(s);
6472 return stbi__errpuc(
"bad file",
"scanline overrun");
6474 if (!stbi__readval(s,packet->channel,value))
6477 for(i=0;i<count;++i, dest += 4)
6478 stbi__copyval(packet->channel,dest,value);
6481 if (count>left)
return stbi__errpuc(
"bad file",
"scanline overrun");
6483 for(i=0;i<count;++i, dest+=4)
6484 if (!stbi__readval(s,packet->channel,dest))
6498static void *stbi__pic_load(stbi__context *s,
int *px,
int *py,
int *comp,
int req_comp, stbi__result_info *ri)
6501 int i, x,y, internal_comp;
6504 if (!comp) comp = &internal_comp;
6506 for (i=0; i<92; ++i)
6509 x = stbi__get16be(s);
6510 y = stbi__get16be(s);
6512 if (y > STBI_MAX_DIMENSIONS)
return stbi__errpuc(
"too large",
"Very large image (corrupt?)");
6513 if (x > STBI_MAX_DIMENSIONS)
return stbi__errpuc(
"too large",
"Very large image (corrupt?)");
6515 if (stbi__at_eof(s))
return stbi__errpuc(
"bad file",
"file too short (pic header)");
6516 if (!stbi__mad3sizes_valid(x, y, 4, 0))
return stbi__errpuc(
"too large",
"PIC image too large to decode");
6523 result = (stbi_uc *) stbi__malloc_mad3(x, y, 4, 0);
6524 if (!result)
return stbi__errpuc(
"outofmem",
"Out of memory");
6525 memset(result, 0xff, x*y*4);
6527 if (!stbi__pic_load_core(s,x,y,comp, result)) {
6533 if (req_comp == 0) req_comp = *comp;
6534 result=stbi__convert_format(result,4,req_comp,x,y);
6539static int stbi__pic_test(stbi__context *s)
6541 int r = stbi__pic_test_core(s);
6562 stbi_uc *background;
6564 int flags, bgindex, ratio, transparent, eflags;
6565 stbi_uc pal[256][4];
6566 stbi_uc lpal[256][4];
6567 stbi__gif_lzw codes[8192];
6568 stbi_uc *color_table;
6571 int start_x, start_y;
6578static int stbi__gif_test_raw(stbi__context *s)
6581 if (stbi__get8(s) !=
'G' || stbi__get8(s) !=
'I' || stbi__get8(s) !=
'F' || stbi__get8(s) !=
'8')
return 0;
6583 if (sz !=
'9' && sz !=
'7')
return 0;
6584 if (stbi__get8(s) !=
'a')
return 0;
6588static int stbi__gif_test(stbi__context *s)
6590 int r = stbi__gif_test_raw(s);
6595static void stbi__gif_parse_colortable(stbi__context *s, stbi_uc pal[256][4],
int num_entries,
int transp)
6598 for (i=0; i < num_entries; ++i) {
6599 pal[i][2] = stbi__get8(s);
6600 pal[i][1] = stbi__get8(s);
6601 pal[i][0] = stbi__get8(s);
6602 pal[i][3] = transp == i ? 0 : 255;
6606static int stbi__gif_header(stbi__context *s, stbi__gif *g,
int *comp,
int is_info)
6609 if (stbi__get8(s) !=
'G' || stbi__get8(s) !=
'I' || stbi__get8(s) !=
'F' || stbi__get8(s) !=
'8')
6610 return stbi__err(
"not GIF",
"Corrupt GIF");
6612 version = stbi__get8(s);
6613 if (version !=
'7' && version !=
'9')
return stbi__err(
"not GIF",
"Corrupt GIF");
6614 if (stbi__get8(s) !=
'a')
return stbi__err(
"not GIF",
"Corrupt GIF");
6616 stbi__g_failure_reason =
"";
6617 g->w = stbi__get16le(s);
6618 g->h = stbi__get16le(s);
6619 g->flags = stbi__get8(s);
6620 g->bgindex = stbi__get8(s);
6621 g->ratio = stbi__get8(s);
6622 g->transparent = -1;
6624 if (g->w > STBI_MAX_DIMENSIONS)
return stbi__err(
"too large",
"Very large image (corrupt?)");
6625 if (g->h > STBI_MAX_DIMENSIONS)
return stbi__err(
"too large",
"Very large image (corrupt?)");
6627 if (comp != 0) *comp = 4;
6629 if (is_info)
return 1;
6631 if (g->flags & 0x80)
6632 stbi__gif_parse_colortable(s,g->pal, 2 << (g->flags & 7), -1);
6637static int stbi__gif_info_raw(stbi__context *s,
int *x,
int *y,
int *comp)
6639 stbi__gif* g = (stbi__gif*) stbi__malloc(
sizeof(stbi__gif));
6640 if (!g)
return stbi__err(
"outofmem",
"Out of memory");
6641 if (!stbi__gif_header(s, g, comp, 1)) {
6652static void stbi__out_gif_code(stbi__gif *g, stbi__uint16 code)
6659 if (g->codes[code].prefix >= 0)
6660 stbi__out_gif_code(g, g->codes[code].prefix);
6662 if (g->cur_y >= g->max_y)
return;
6664 idx = g->cur_x + g->cur_y;
6666 g->history[idx / 4] = 1;
6668 c = &g->color_table[g->codes[code].suffix * 4];
6677 if (g->cur_x >= g->max_x) {
6678 g->cur_x = g->start_x;
6679 g->cur_y += g->step;
6681 while (g->cur_y >= g->max_y && g->parse > 0) {
6682 g->step = (1 << g->parse) * g->line_size;
6683 g->cur_y = g->start_y + (g->step >> 1);
6689static stbi_uc *stbi__process_gif_raster(stbi__context *s, stbi__gif *g)
6692 stbi__int32 len, init_code;
6694 stbi__int32 codesize, codemask, avail, oldcode, bits, valid_bits, clear;
6697 lzw_cs = stbi__get8(s);
6698 if (lzw_cs > 12)
return NULL;
6699 clear = 1 << lzw_cs;
6701 codesize = lzw_cs + 1;
6702 codemask = (1 << codesize) - 1;
6705 for (init_code = 0; init_code < clear; init_code++) {
6706 g->codes[init_code].prefix = -1;
6707 g->codes[init_code].first = (stbi_uc) init_code;
6708 g->codes[init_code].suffix = (stbi_uc) init_code;
6717 if (valid_bits < codesize) {
6719 len = stbi__get8(s);
6724 bits |= (stbi__int32) stbi__get8(s) << valid_bits;
6727 stbi__int32 code = bits & codemask;
6729 valid_bits -= codesize;
6731 if (code == clear) {
6732 codesize = lzw_cs + 1;
6733 codemask = (1 << codesize) - 1;
6737 }
else if (code == clear + 1) {
6739 while ((len = stbi__get8(s)) > 0)
6742 }
else if (code <= avail) {
6744 return stbi__errpuc(
"no clear code",
"Corrupt GIF");
6748 p = &g->codes[avail++];
6750 return stbi__errpuc(
"too many codes",
"Corrupt GIF");
6753 p->prefix = (stbi__int16) oldcode;
6754 p->first = g->codes[oldcode].first;
6755 p->suffix = (code == avail) ? p->first : g->codes[code].first;
6756 }
else if (code == avail)
6757 return stbi__errpuc(
"illegal code in raster",
"Corrupt GIF");
6759 stbi__out_gif_code(g, (stbi__uint16) code);
6761 if ((avail & codemask) == 0 && avail <= 0x0FFF) {
6763 codemask = (1 << codesize) - 1;
6768 return stbi__errpuc(
"illegal code in raster",
"Corrupt GIF");
6776static stbi_uc *stbi__gif_load_next(stbi__context *s, stbi__gif *g,
int *comp,
int req_comp, stbi_uc *two_back)
6782 STBI_NOTUSED(req_comp);
6787 if (!stbi__gif_header(s, g, comp,0))
return 0;
6788 if (!stbi__mad3sizes_valid(4, g->w, g->h, 0))
6789 return stbi__errpuc(
"too large",
"GIF image is too large");
6790 pcount = g->w * g->h;
6791 g->out = (stbi_uc *) stbi__malloc(4 * pcount);
6792 g->background = (stbi_uc *) stbi__malloc(4 * pcount);
6793 g->history = (stbi_uc *) stbi__malloc(pcount);
6794 if (!g->out || !g->background || !g->history)
6795 return stbi__errpuc(
"outofmem",
"Out of memory");
6800 memset(g->out, 0x00, 4 * pcount);
6801 memset(g->background, 0x00, 4 * pcount);
6802 memset(g->history, 0x00, pcount);
6806 dispose = (g->eflags & 0x1C) >> 2;
6807 pcount = g->w * g->h;
6809 if ((dispose == 3) && (two_back == 0)) {
6814 for (pi = 0; pi < pcount; ++pi) {
6815 if (g->history[pi]) {
6816 memcpy( &g->out[pi * 4], &two_back[pi * 4], 4 );
6819 }
else if (dispose == 2) {
6821 for (pi = 0; pi < pcount; ++pi) {
6822 if (g->history[pi]) {
6823 memcpy( &g->out[pi * 4], &g->background[pi * 4], 4 );
6834 memcpy( g->background, g->out, 4 * g->w * g->h );
6838 memset( g->history, 0x00, g->w * g->h );
6841 int tag = stbi__get8(s);
6845 stbi__int32 x, y, w, h;
6848 x = stbi__get16le(s);
6849 y = stbi__get16le(s);
6850 w = stbi__get16le(s);
6851 h = stbi__get16le(s);
6852 if (((x + w) > (g->w)) || ((y + h) > (g->h)))
6853 return stbi__errpuc(
"bad Image Descriptor",
"Corrupt GIF");
6855 g->line_size = g->w * 4;
6857 g->start_y = y * g->line_size;
6858 g->max_x = g->start_x + w * 4;
6859 g->max_y = g->start_y + h * g->line_size;
6860 g->cur_x = g->start_x;
6861 g->cur_y = g->start_y;
6868 g->cur_y = g->max_y;
6870 g->lflags = stbi__get8(s);
6872 if (g->lflags & 0x40) {
6873 g->step = 8 * g->line_size;
6876 g->step = g->line_size;
6880 if (g->lflags & 0x80) {
6881 stbi__gif_parse_colortable(s,g->lpal, 2 << (g->lflags & 7), g->eflags & 0x01 ? g->transparent : -1);
6882 g->color_table = (stbi_uc *) g->lpal;
6883 }
else if (g->flags & 0x80) {
6884 g->color_table = (stbi_uc *) g->pal;
6886 return stbi__errpuc(
"missing color table",
"Corrupt GIF");
6888 o = stbi__process_gif_raster(s, g);
6889 if (!o)
return NULL;
6892 pcount = g->w * g->h;
6893 if (first_frame && (g->bgindex > 0)) {
6895 for (pi = 0; pi < pcount; ++pi) {
6896 if (g->history[pi] == 0) {
6897 g->pal[g->bgindex][3] = 255;
6898 memcpy( &g->out[pi * 4], &g->pal[g->bgindex], 4 );
6909 int ext = stbi__get8(s);
6911 len = stbi__get8(s);
6913 g->eflags = stbi__get8(s);
6914 g->delay = 10 * stbi__get16le(s);
6917 if (g->transparent >= 0) {
6918 g->pal[g->transparent][3] = 255;
6920 if (g->eflags & 0x01) {
6921 g->transparent = stbi__get8(s);
6922 if (g->transparent >= 0) {
6923 g->pal[g->transparent][3] = 0;
6928 g->transparent = -1;
6935 while ((len = stbi__get8(s)) != 0) {
6942 return (stbi_uc *) s;
6945 return stbi__errpuc(
"unknown code",
"Corrupt GIF");
6950static void *stbi__load_gif_main_outofmem(stbi__gif *g, stbi_uc *out,
int **delays)
6953 STBI_FREE(g->history);
6954 STBI_FREE(g->background);
6956 if (out) STBI_FREE(out);
6957 if (delays && *delays) STBI_FREE(*delays);
6958 return stbi__errpuc(
"outofmem",
"Out of memory");
6961static void *stbi__load_gif_main(stbi__context *s,
int **delays,
int *x,
int *y,
int *z,
int *comp,
int req_comp)
6963 if (stbi__gif_test(s)) {
6967 stbi_uc *two_back = 0;
6971 int delays_size = 0;
6973 STBI_NOTUSED(out_size);
6974 STBI_NOTUSED(delays_size);
6976 memset(&g, 0,
sizeof(g));
6982 u = stbi__gif_load_next(s, &g, comp, req_comp, two_back);
6983 if (u == (stbi_uc *) s) u = 0;
6989 stride = g.w * g.h * 4;
6992 void *tmp = (stbi_uc*) STBI_REALLOC_SIZED( out, out_size, layers * stride );
6994 return stbi__load_gif_main_outofmem(&g, out, delays);
6996 out = (stbi_uc*) tmp;
6997 out_size = layers * stride;
7001 int *new_delays = (
int*) STBI_REALLOC_SIZED( *delays, delays_size,
sizeof(
int) * layers );
7003 return stbi__load_gif_main_outofmem(&g, out, delays);
7004 *delays = new_delays;
7005 delays_size = layers *
sizeof(int);
7008 out = (stbi_uc*)stbi__malloc( layers * stride );
7010 return stbi__load_gif_main_outofmem(&g, out, delays);
7011 out_size = layers * stride;
7013 *delays = (
int*) stbi__malloc( layers *
sizeof(
int) );
7015 return stbi__load_gif_main_outofmem(&g, out, delays);
7016 delays_size = layers *
sizeof(int);
7019 memcpy( out + ((layers - 1) * stride), u, stride );
7021 two_back = out - 2 * stride;
7025 (*delays)[layers - 1U] = g.delay;
7032 STBI_FREE(g.history);
7033 STBI_FREE(g.background);
7036 if (req_comp && req_comp != 4)
7037 out = stbi__convert_format(out, 4, req_comp, layers * g.w, g.h);
7042 return stbi__errpuc(
"not GIF",
"Image was not as a gif type.");
7046static void *stbi__gif_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri)
7050 memset(&g, 0,
sizeof(g));
7053 u = stbi__gif_load_next(s, &g, comp, req_comp, 0);
7054 if (u == (stbi_uc *) s) u = 0;
7061 if (req_comp && req_comp != 4)
7062 u = stbi__convert_format(u, 4, req_comp, g.w, g.h);
7069 STBI_FREE(g.history);
7070 STBI_FREE(g.background);
7075static int stbi__gif_info(stbi__context *s,
int *x,
int *y,
int *comp)
7077 return stbi__gif_info_raw(s,x,y,comp);
7085static int stbi__hdr_test_core(stbi__context *s,
const char *signature)
7088 for (i=0; signature[i]; ++i)
7089 if (stbi__get8(s) != signature[i])
7095static int stbi__hdr_test(stbi__context* s)
7097 int r = stbi__hdr_test_core(s,
"#?RADIANCE\n");
7100 r = stbi__hdr_test_core(s,
"#?RGBE\n");
7106#define STBI__HDR_BUFLEN 1024
7107static char *stbi__hdr_gettoken(stbi__context *z,
char *buffer)
7112 c = (char) stbi__get8(z);
7114 while (!stbi__at_eof(z) && c !=
'\n') {
7116 if (len == STBI__HDR_BUFLEN-1) {
7118 while (!stbi__at_eof(z) && stbi__get8(z) !=
'\n')
7122 c = (char) stbi__get8(z);
7129static void stbi__hdr_convert(
float *output, stbi_uc *input,
int req_comp)
7131 if ( input[3] != 0 ) {
7134 f1 = (float) ldexp(1.0f, input[3] - (
int)(128 + 8));
7136 output[0] = (input[0] + input[1] + input[2]) * f1 / 3;
7138 output[0] = input[0] * f1;
7139 output[1] = input[1] * f1;
7140 output[2] = input[2] * f1;
7142 if (req_comp == 2) output[1] = 1;
7143 if (req_comp == 4) output[3] = 1;
7146 case 4: output[3] = 1;
7147 case 3: output[0] = output[1] = output[2] = 0;
7149 case 2: output[1] = 1;
7150 case 1: output[0] = 0;
7156static float *stbi__hdr_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri)
7158 char buffer[STBI__HDR_BUFLEN];
7165 unsigned char count, value;
7166 int i, j, k, c1,c2, z;
7167 const char *headerToken;
7171 headerToken = stbi__hdr_gettoken(s,buffer);
7172 if (strcmp(headerToken,
"#?RADIANCE") != 0 && strcmp(headerToken,
"#?RGBE") != 0)
7173 return stbi__errpf(
"not HDR",
"Corrupt HDR image");
7177 token = stbi__hdr_gettoken(s,buffer);
7178 if (token[0] == 0)
break;
7179 if (strcmp(token,
"FORMAT=32-bit_rle_rgbe") == 0) valid = 1;
7182 if (!valid)
return stbi__errpf(
"unsupported format",
"Unsupported HDR format");
7186 token = stbi__hdr_gettoken(s,buffer);
7187 if (strncmp(token,
"-Y ", 3))
return stbi__errpf(
"unsupported data layout",
"Unsupported HDR format");
7189 height = (int) strtol(token, &token, 10);
7190 while (*token ==
' ') ++token;
7191 if (strncmp(token,
"+X ", 3))
return stbi__errpf(
"unsupported data layout",
"Unsupported HDR format");
7193 width = (int) strtol(token, NULL, 10);
7195 if (height > STBI_MAX_DIMENSIONS)
return stbi__errpf(
"too large",
"Very large image (corrupt?)");
7196 if (width > STBI_MAX_DIMENSIONS)
return stbi__errpf(
"too large",
"Very large image (corrupt?)");
7201 if (comp) *comp = 3;
7202 if (req_comp == 0) req_comp = 3;
7204 if (!stbi__mad4sizes_valid(width, height, req_comp,
sizeof(
float), 0))
7205 return stbi__errpf(
"too large",
"HDR image is too large");
7208 hdr_data = (
float *) stbi__malloc_mad4(width, height, req_comp,
sizeof(
float), 0);
7210 return stbi__errpf(
"outofmem",
"Out of memory");
7214 if ( width < 8 || width >= 32768) {
7216 for (j=0; j < height; ++j) {
7217 for (i=0; i < width; ++i) {
7220 stbi__getn(s, rgbe, 4);
7221 stbi__hdr_convert(hdr_data + j * width * req_comp + i * req_comp, rgbe, req_comp);
7228 for (j = 0; j < height; ++j) {
7231 len = stbi__get8(s);
7232 if (c1 != 2 || c2 != 2 || (len & 0x80)) {
7236 rgbe[0] = (stbi_uc) c1;
7237 rgbe[1] = (stbi_uc) c2;
7238 rgbe[2] = (stbi_uc) len;
7239 rgbe[3] = (stbi_uc) stbi__get8(s);
7240 stbi__hdr_convert(hdr_data, rgbe, req_comp);
7243 STBI_FREE(scanline);
7244 goto main_decode_loop;
7247 len |= stbi__get8(s);
7248 if (len != width) { STBI_FREE(hdr_data); STBI_FREE(scanline);
return stbi__errpf(
"invalid decoded scanline length",
"corrupt HDR"); }
7249 if (scanline == NULL) {
7250 scanline = (stbi_uc *) stbi__malloc_mad2(width, 4, 0);
7252 STBI_FREE(hdr_data);
7253 return stbi__errpf(
"outofmem",
"Out of memory");
7257 for (k = 0; k < 4; ++k) {
7260 while ((nleft = width - i) > 0) {
7261 count = stbi__get8(s);
7264 value = stbi__get8(s);
7266 if ((count == 0) || (count > nleft)) { STBI_FREE(hdr_data); STBI_FREE(scanline);
return stbi__errpf(
"corrupt",
"bad RLE data in HDR"); }
7267 for (z = 0; z < count; ++z)
7268 scanline[i++ * 4 + k] = value;
7271 if ((count == 0) || (count > nleft)) { STBI_FREE(hdr_data); STBI_FREE(scanline);
return stbi__errpf(
"corrupt",
"bad RLE data in HDR"); }
7272 for (z = 0; z < count; ++z)
7273 scanline[i++ * 4 + k] = stbi__get8(s);
7277 for (i=0; i < width; ++i)
7278 stbi__hdr_convert(hdr_data+(j*width + i)*req_comp, scanline + i*4, req_comp);
7281 STBI_FREE(scanline);
7287static int stbi__hdr_info(stbi__context *s,
int *x,
int *y,
int *comp)
7289 char buffer[STBI__HDR_BUFLEN];
7296 if (!comp) comp = &dummy;
7298 if (stbi__hdr_test(s) == 0) {
7304 token = stbi__hdr_gettoken(s,buffer);
7305 if (token[0] == 0)
break;
7306 if (strcmp(token,
"FORMAT=32-bit_rle_rgbe") == 0) valid = 1;
7313 token = stbi__hdr_gettoken(s,buffer);
7314 if (strncmp(token,
"-Y ", 3)) {
7319 *y = (int) strtol(token, &token, 10);
7320 while (*token ==
' ') ++token;
7321 if (strncmp(token,
"+X ", 3)) {
7326 *x = (int) strtol(token, NULL, 10);
7333static int stbi__bmp_info(stbi__context *s,
int *x,
int *y,
int *comp)
7336 stbi__bmp_data info;
7339 p = stbi__bmp_parse_header(s, &info);
7344 if (x) *x = s->img_x;
7345 if (y) *y = s->img_y;
7347 if (info.bpp == 24 && info.ma == 0xff000000)
7350 *comp = info.ma ? 4 : 3;
7357static int stbi__psd_info(stbi__context *s,
int *x,
int *y,
int *comp)
7359 int channelCount, dummy, depth;
7362 if (!comp) comp = &dummy;
7363 if (stbi__get32be(s) != 0x38425053) {
7367 if (stbi__get16be(s) != 1) {
7372 channelCount = stbi__get16be(s);
7373 if (channelCount < 0 || channelCount > 16) {
7377 *y = stbi__get32be(s);
7378 *x = stbi__get32be(s);
7379 depth = stbi__get16be(s);
7380 if (depth != 8 && depth != 16) {
7384 if (stbi__get16be(s) != 3) {
7392static int stbi__psd_is16(stbi__context *s)
7394 int channelCount, depth;
7395 if (stbi__get32be(s) != 0x38425053) {
7399 if (stbi__get16be(s) != 1) {
7404 channelCount = stbi__get16be(s);
7405 if (channelCount < 0 || channelCount > 16) {
7409 STBI_NOTUSED(stbi__get32be(s));
7410 STBI_NOTUSED(stbi__get32be(s));
7411 depth = stbi__get16be(s);
7421static int stbi__pic_info(stbi__context *s,
int *x,
int *y,
int *comp)
7423 int act_comp=0,num_packets=0,chained,dummy;
7424 stbi__pic_packet packets[10];
7428 if (!comp) comp = &dummy;
7430 if (!stbi__pic_is4(s,
"\x53\x80\xF6\x34")) {
7437 *x = stbi__get16be(s);
7438 *y = stbi__get16be(s);
7439 if (stbi__at_eof(s)) {
7443 if ( (*x) != 0 && (1 << 28) / (*x) < (*y)) {
7451 stbi__pic_packet *packet;
7453 if (num_packets==
sizeof(packets)/
sizeof(packets[0]))
7456 packet = &packets[num_packets++];
7457 chained = stbi__get8(s);
7458 packet->size = stbi__get8(s);
7459 packet->type = stbi__get8(s);
7460 packet->channel = stbi__get8(s);
7461 act_comp |= packet->channel;
7463 if (stbi__at_eof(s)) {
7467 if (packet->size != 8) {
7473 *comp = (act_comp & 0x10 ? 4 : 3);
7492static int stbi__pnm_test(stbi__context *s)
7495 p = (char) stbi__get8(s);
7496 t = (char) stbi__get8(s);
7497 if (p !=
'P' || (t !=
'5' && t !=
'6')) {
7504static void *stbi__pnm_load(stbi__context *s,
int *x,
int *y,
int *comp,
int req_comp, stbi__result_info *ri)
7509 ri->bits_per_channel = stbi__pnm_info(s, (
int *)&s->img_x, (
int *)&s->img_y, (
int *)&s->img_n);
7510 if (ri->bits_per_channel == 0)
7513 if (s->img_y > STBI_MAX_DIMENSIONS)
return stbi__errpuc(
"too large",
"Very large image (corrupt?)");
7514 if (s->img_x > STBI_MAX_DIMENSIONS)
return stbi__errpuc(
"too large",
"Very large image (corrupt?)");
7518 if (comp) *comp = s->img_n;
7520 if (!stbi__mad4sizes_valid(s->img_n, s->img_x, s->img_y, ri->bits_per_channel / 8, 0))
7521 return stbi__errpuc(
"too large",
"PNM too large");
7523 out = (stbi_uc *) stbi__malloc_mad4(s->img_n, s->img_x, s->img_y, ri->bits_per_channel / 8, 0);
7524 if (!out)
return stbi__errpuc(
"outofmem",
"Out of memory");
7525 if (!stbi__getn(s, out, s->img_n * s->img_x * s->img_y * (ri->bits_per_channel / 8))) {
7527 return stbi__errpuc(
"bad PNM",
"PNM file truncated");
7530 if (req_comp && req_comp != s->img_n) {
7531 if (ri->bits_per_channel == 16) {
7532 out = (stbi_uc *) stbi__convert_format16((stbi__uint16 *) out, s->img_n, req_comp, s->img_x, s->img_y);
7534 out = stbi__convert_format(out, s->img_n, req_comp, s->img_x, s->img_y);
7536 if (out == NULL)
return out;
7541static int stbi__pnm_isspace(
char c)
7543 return c ==
' ' || c ==
'\t' || c ==
'\n' || c ==
'\v' || c ==
'\f' || c ==
'\r';
7546static void stbi__pnm_skip_whitespace(stbi__context *s,
char *c)
7549 while (!stbi__at_eof(s) && stbi__pnm_isspace(*c))
7550 *c = (char) stbi__get8(s);
7552 if (stbi__at_eof(s) || *c !=
'#')
7555 while (!stbi__at_eof(s) && *c !=
'\n' && *c !=
'\r' )
7556 *c = (char) stbi__get8(s);
7560static int stbi__pnm_isdigit(
char c)
7562 return c >=
'0' && c <=
'9';
7565static int stbi__pnm_getinteger(stbi__context *s,
char *c)
7569 while (!stbi__at_eof(s) && stbi__pnm_isdigit(*c)) {
7570 value = value*10 + (*c -
'0');
7571 *c = (char) stbi__get8(s);
7572 if((value > 214748364) || (value == 214748364 && *c >
'7'))
7573 return stbi__err(
"integer parse overflow",
"Parsing an integer in the PPM header overflowed a 32-bit int");
7579static int stbi__pnm_info(stbi__context *s,
int *x,
int *y,
int *comp)
7586 if (!comp) comp = &dummy;
7591 p = (char) stbi__get8(s);
7592 t = (char) stbi__get8(s);
7593 if (p !=
'P' || (t !=
'5' && t !=
'6')) {
7598 *comp = (t ==
'6') ? 3 : 1;
7600 c = (char) stbi__get8(s);
7601 stbi__pnm_skip_whitespace(s, &c);
7603 *x = stbi__pnm_getinteger(s, &c);
7605 return stbi__err(
"invalid width",
"PPM image header had zero or overflowing width");
7606 stbi__pnm_skip_whitespace(s, &c);
7608 *y = stbi__pnm_getinteger(s, &c);
7610 return stbi__err(
"invalid width",
"PPM image header had zero or overflowing width");
7611 stbi__pnm_skip_whitespace(s, &c);
7613 maxv = stbi__pnm_getinteger(s, &c);
7615 return stbi__err(
"max value > 65535",
"PPM image supports only 8-bit and 16-bit images");
7616 else if (maxv > 255)
7622static int stbi__pnm_is16(stbi__context *s)
7624 if (stbi__pnm_info(s, NULL, NULL, NULL) == 16)
7630static int stbi__info_main(stbi__context *s,
int *x,
int *y,
int *comp)
7632 #ifndef STBI_NO_JPEG
7633 if (stbi__jpeg_info(s, x, y, comp))
return 1;
7637 if (stbi__png_info(s, x, y, comp))
return 1;
7641 if (stbi__gif_info(s, x, y, comp))
return 1;
7645 if (stbi__bmp_info(s, x, y, comp))
return 1;
7649 if (stbi__psd_info(s, x, y, comp))
return 1;
7653 if (stbi__pic_info(s, x, y, comp))
return 1;
7657 if (stbi__pnm_info(s, x, y, comp))
return 1;
7661 if (stbi__hdr_info(s, x, y, comp))
return 1;
7666 if (stbi__tga_info(s, x, y, comp))
7669 return stbi__err(
"unknown image type",
"Image not of any known type, or corrupt");
7672static int stbi__is_16_main(stbi__context *s)
7675 if (stbi__png_is16(s))
return 1;
7679 if (stbi__psd_is16(s))
return 1;
7683 if (stbi__pnm_is16(s))
return 1;
7688#ifndef STBI_NO_STDIO
7689STBIDEF
int stbi_info(
char const *filename,
int *x,
int *y,
int *comp)
7691 FILE *f = stbi__fopen(filename,
"rb");
7693 if (!f)
return stbi__err(
"can't fopen",
"Unable to open file");
7694 result = stbi_info_from_file(f, x, y, comp);
7699STBIDEF
int stbi_info_from_file(FILE *f,
int *x,
int *y,
int *comp)
7703 long pos = ftell(f);
7704 stbi__start_file(&s, f);
7705 r = stbi__info_main(&s,x,y,comp);
7706 fseek(f,pos,SEEK_SET);
7710STBIDEF
int stbi_is_16_bit(
char const *filename)
7712 FILE *f = stbi__fopen(filename,
"rb");
7714 if (!f)
return stbi__err(
"can't fopen",
"Unable to open file");
7715 result = stbi_is_16_bit_from_file(f);
7720STBIDEF
int stbi_is_16_bit_from_file(FILE *f)
7724 long pos = ftell(f);
7725 stbi__start_file(&s, f);
7726 r = stbi__is_16_main(&s);
7727 fseek(f,pos,SEEK_SET);
7732STBIDEF
int stbi_info_from_memory(stbi_uc
const *buffer,
int len,
int *x,
int *y,
int *comp)
7735 stbi__start_mem(&s,buffer,len);
7736 return stbi__info_main(&s,x,y,comp);
7739STBIDEF
int stbi_info_from_callbacks(
stbi_io_callbacks const *c,
void *user,
int *x,
int *y,
int *comp)
7743 return stbi__info_main(&s,x,y,comp);
7746STBIDEF
int stbi_is_16_bit_from_memory(stbi_uc
const *buffer,
int len)
7749 stbi__start_mem(&s,buffer,len);
7750 return stbi__is_16_main(&s);
7753STBIDEF
int stbi_is_16_bit_from_callbacks(
stbi_io_callbacks const *c,
void *user)
7757 return stbi__is_16_main(&s);
Definition stb_image.hpp:412