| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190 |
- // Copyright 2012 Google Inc. All Rights Reserved.
- //
- // Use of this source code is governed by a BSD-style license
- // that can be found in the COPYING file in the root of the source
- // tree. An additional intellectual property rights grant can be found
- // in the file PATENTS. All contributing project authors may
- // be found in the AUTHORS file in the root of the source tree.
- // -----------------------------------------------------------------------------
- //
- // Misc. common utility functions
- //
- // Authors: Skal ([email protected])
- // Urvang ([email protected])
- #ifndef WEBP_UTILS_UTILS_H_
- #define WEBP_UTILS_UTILS_H_
- #ifdef HAVE_CONFIG_H
- #include "../webp/config.h"
- #endif
- #include <assert.h>
- #include <limits.h>
- #include "../dsp/dsp.h"
- #include "../webp/types.h"
- #ifdef __cplusplus
- extern "C" {
- #endif
- //------------------------------------------------------------------------------
- // Memory allocation
- // This is the maximum memory amount that libwebp will ever try to allocate.
- #ifndef WEBP_MAX_ALLOCABLE_MEMORY
- #if SIZE_MAX > (1ULL << 34)
- #define WEBP_MAX_ALLOCABLE_MEMORY (1ULL << 34)
- #else
- // For 32-bit targets keep this below INT_MAX to avoid valgrind warnings.
- #define WEBP_MAX_ALLOCABLE_MEMORY ((1ULL << 31) - (1 << 16))
- #endif
- #endif // WEBP_MAX_ALLOCABLE_MEMORY
- // size-checking safe malloc/calloc: verify that the requested size is not too
- // large, or return NULL. You don't need to call these for constructs like
- // malloc(sizeof(foo)), but only if there's picture-dependent size involved
- // somewhere (like: malloc(num_pixels * sizeof(*something))). That's why this
- // safe malloc() borrows the signature from calloc(), pointing at the dangerous
- // underlying multiply involved.
- WEBP_EXTERN(void*) WebPSafeMalloc(uint64_t nmemb, size_t size);
- // Note that WebPSafeCalloc() expects the second argument type to be 'size_t'
- // in order to favor the "calloc(num_foo, sizeof(foo))" pattern.
- WEBP_EXTERN(void*) WebPSafeCalloc(uint64_t nmemb, size_t size);
- // Companion deallocation function to the above allocations.
- WEBP_EXTERN(void) WebPSafeFree(void* const ptr);
- //------------------------------------------------------------------------------
- // Alignment
- #define WEBP_ALIGN_CST 31
- #define WEBP_ALIGN(PTR) (((uintptr_t)(PTR) + WEBP_ALIGN_CST) & ~WEBP_ALIGN_CST)
- #if defined(WEBP_FORCE_ALIGNED)
- #include <string.h>
- // memcpy() is the safe way of moving potentially unaligned 32b memory.
- static WEBP_INLINE uint32_t WebPMemToUint32(const uint8_t* const ptr) {
- uint32_t A;
- memcpy(&A, (const int*)ptr, sizeof(A));
- return A;
- }
- static WEBP_INLINE void WebPUint32ToMem(uint8_t* const ptr, uint32_t val) {
- memcpy(ptr, &val, sizeof(val));
- }
- #else
- static WEBP_UBSAN_IGNORE_UNDEF WEBP_INLINE
- uint32_t WebPMemToUint32(const uint8_t* const ptr) {
- return *(const uint32_t*)ptr;
- }
- static WEBP_UBSAN_IGNORE_UNDEF WEBP_INLINE
- void WebPUint32ToMem(uint8_t* const ptr, uint32_t val) {
- *(uint32_t*)ptr = val;
- }
- #endif
- //------------------------------------------------------------------------------
- // Reading/writing data.
- // Read 16, 24 or 32 bits stored in little-endian order.
- static WEBP_INLINE int GetLE16(const uint8_t* const data) {
- return (int)(data[0] << 0) | (data[1] << 8);
- }
- static WEBP_INLINE int GetLE24(const uint8_t* const data) {
- return GetLE16(data) | (data[2] << 16);
- }
- static WEBP_INLINE uint32_t GetLE32(const uint8_t* const data) {
- return GetLE16(data) | ((uint32_t)GetLE16(data + 2) << 16);
- }
- // Store 16, 24 or 32 bits in little-endian order.
- static WEBP_INLINE void PutLE16(uint8_t* const data, int val) {
- assert(val < (1 << 16));
- data[0] = (val >> 0);
- data[1] = (val >> 8);
- }
- static WEBP_INLINE void PutLE24(uint8_t* const data, int val) {
- assert(val < (1 << 24));
- PutLE16(data, val & 0xffff);
- data[2] = (val >> 16);
- }
- static WEBP_INLINE void PutLE32(uint8_t* const data, uint32_t val) {
- PutLE16(data, (int)(val & 0xffff));
- PutLE16(data + 2, (int)(val >> 16));
- }
- // Returns (int)floor(log2(n)). n must be > 0.
- // use GNU builtins where available.
- #if defined(__GNUC__) && \
- ((__GNUC__ == 3 && __GNUC_MINOR__ >= 4) || __GNUC__ >= 4)
- static WEBP_INLINE int BitsLog2Floor(uint32_t n) {
- return 31 ^ __builtin_clz(n);
- }
- #elif defined(_MSC_VER) && _MSC_VER > 1310 && \
- (defined(_M_X64) || defined(_M_IX86))
- #include <intrin.h>
- #pragma intrinsic(_BitScanReverse)
- static WEBP_INLINE int BitsLog2Floor(uint32_t n) {
- unsigned long first_set_bit;
- _BitScanReverse(&first_set_bit, n);
- return first_set_bit;
- }
- #else
- static WEBP_INLINE int BitsLog2Floor(uint32_t n) {
- int log = 0;
- uint32_t value = n;
- int i;
- for (i = 4; i >= 0; --i) {
- const int shift = (1 << i);
- const uint32_t x = value >> shift;
- if (x != 0) {
- value = x;
- log += shift;
- }
- }
- return log;
- }
- #endif
- //------------------------------------------------------------------------------
- // Pixel copying.
- struct WebPPicture;
- // Copy width x height pixels from 'src' to 'dst' honoring the strides.
- WEBP_EXTERN(void) WebPCopyPlane(const uint8_t* src, int src_stride,
- uint8_t* dst, int dst_stride,
- int width, int height);
- // Copy ARGB pixels from 'src' to 'dst' honoring strides. 'src' and 'dst' are
- // assumed to be already allocated and using ARGB data.
- WEBP_EXTERN(void) WebPCopyPixels(const struct WebPPicture* const src,
- struct WebPPicture* const dst);
- //------------------------------------------------------------------------------
- // Unique colors.
- // Returns count of unique colors in 'pic', assuming pic->use_argb is true.
- // If the unique color count is more than MAX_COLOR_COUNT, returns
- // MAX_COLOR_COUNT+1.
- // If 'palette' is not NULL and number of unique colors is less than or equal to
- // MAX_COLOR_COUNT, also outputs the actual unique colors into 'palette'.
- // Note: 'palette' is assumed to be an array already allocated with at least
- // MAX_COLOR_COUNT elements.
- WEBP_EXTERN(int) WebPGetColorPalette(const struct WebPPicture* const pic,
- uint32_t* const palette);
- //------------------------------------------------------------------------------
- #ifdef __cplusplus
- } // extern "C"
- #endif
- #endif /* WEBP_UTILS_UTILS_H_ */
|