251 lines
		
	
	
		
			8.1 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			251 lines
		
	
	
		
			8.1 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| // Copyright 2011 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.
 | |
| // -----------------------------------------------------------------------------
 | |
| //
 | |
| // Internal header for mux library.
 | |
| //
 | |
| // Author: Urvang (urvang@google.com)
 | |
| 
 | |
| #ifndef WEBP_MUX_MUXI_H_
 | |
| #define WEBP_MUX_MUXI_H_
 | |
| 
 | |
| #include <stdlib.h>
 | |
| #include "../dec/vp8i.h"
 | |
| #include "../dec/vp8li.h"
 | |
| #include "../webp/mux.h"
 | |
| 
 | |
| #if defined(__cplusplus) || defined(c_plusplus)
 | |
| extern "C" {
 | |
| #endif
 | |
| 
 | |
| //------------------------------------------------------------------------------
 | |
| // Defines and constants.
 | |
| 
 | |
| #define MUX_MAJ_VERSION 0
 | |
| #define MUX_MIN_VERSION 1
 | |
| #define MUX_REV_VERSION 1
 | |
| 
 | |
| // Chunk object.
 | |
| typedef struct WebPChunk WebPChunk;
 | |
| struct WebPChunk {
 | |
|   uint32_t        tag_;
 | |
|   int             owner_;  // True if *data_ memory is owned internally.
 | |
|                            // VP8X, ANIM, and other internally created chunks
 | |
|                            // like ANMF/FRGM are always owned.
 | |
|   WebPData        data_;
 | |
|   WebPChunk*      next_;
 | |
| };
 | |
| 
 | |
| // MuxImage object. Store a full WebP image (including ANMF/FRGM chunk, ALPH
 | |
| // chunk and VP8/VP8L chunk),
 | |
| typedef struct WebPMuxImage WebPMuxImage;
 | |
| struct WebPMuxImage {
 | |
|   WebPChunk*  header_;      // Corresponds to WEBP_CHUNK_ANMF/WEBP_CHUNK_FRGM.
 | |
|   WebPChunk*  alpha_;       // Corresponds to WEBP_CHUNK_ALPHA.
 | |
|   WebPChunk*  img_;         // Corresponds to WEBP_CHUNK_IMAGE.
 | |
|   int         is_partial_;  // True if only some of the chunks are filled.
 | |
|   WebPMuxImage* next_;
 | |
| };
 | |
| 
 | |
| // Main mux object. Stores data chunks.
 | |
| struct WebPMux {
 | |
|   WebPMuxImage*   images_;
 | |
|   WebPChunk*      iccp_;
 | |
|   WebPChunk*      exif_;
 | |
|   WebPChunk*      xmp_;
 | |
|   WebPChunk*      anim_;
 | |
|   WebPChunk*      vp8x_;
 | |
| 
 | |
|   WebPChunk*  unknown_;
 | |
| };
 | |
| 
 | |
| // CHUNK_INDEX enum: used for indexing within 'kChunks' (defined below) only.
 | |
| // Note: the reason for having two enums ('WebPChunkId' and 'CHUNK_INDEX') is to
 | |
| // allow two different chunks to have the same id (e.g. WebPChunkId
 | |
| // 'WEBP_CHUNK_IMAGE' can correspond to CHUNK_INDEX 'IDX_VP8' or 'IDX_VP8L').
 | |
| typedef enum {
 | |
|   IDX_VP8X = 0,
 | |
|   IDX_ICCP,
 | |
|   IDX_ANIM,
 | |
|   IDX_ANMF,
 | |
|   IDX_FRGM,
 | |
|   IDX_ALPHA,
 | |
|   IDX_VP8,
 | |
|   IDX_VP8L,
 | |
|   IDX_EXIF,
 | |
|   IDX_XMP,
 | |
|   IDX_UNKNOWN,
 | |
| 
 | |
|   IDX_NIL,
 | |
|   IDX_LAST_CHUNK
 | |
| } CHUNK_INDEX;
 | |
| 
 | |
| #define NIL_TAG 0x00000000u  // To signal void chunk.
 | |
| 
 | |
| typedef struct {
 | |
|   uint32_t      tag;
 | |
|   WebPChunkId   id;
 | |
|   uint32_t      size;
 | |
| } ChunkInfo;
 | |
| 
 | |
| extern const ChunkInfo kChunks[IDX_LAST_CHUNK];
 | |
| 
 | |
| //------------------------------------------------------------------------------
 | |
| // Chunk object management.
 | |
| 
 | |
| // Initialize.
 | |
| void ChunkInit(WebPChunk* const chunk);
 | |
| 
 | |
| // Get chunk index from chunk tag. Returns IDX_NIL if not found.
 | |
| CHUNK_INDEX ChunkGetIndexFromTag(uint32_t tag);
 | |
| 
 | |
| // Get chunk id from chunk tag. Returns WEBP_CHUNK_NIL if not found.
 | |
| WebPChunkId ChunkGetIdFromTag(uint32_t tag);
 | |
| 
 | |
| // Convert a fourcc string to a tag.
 | |
| uint32_t ChunkGetTagFromFourCC(const char fourcc[4]);
 | |
| 
 | |
| // Get chunk index from fourcc. Returns IDX_UNKNOWN if given fourcc is unknown.
 | |
| CHUNK_INDEX ChunkGetIndexFromFourCC(const char fourcc[4]);
 | |
| 
 | |
| // Search for nth chunk with given 'tag' in the chunk list.
 | |
| // nth = 0 means "last of the list".
 | |
| WebPChunk* ChunkSearchList(WebPChunk* first, uint32_t nth, uint32_t tag);
 | |
| 
 | |
| // Fill the chunk with the given data.
 | |
| WebPMuxError ChunkAssignData(WebPChunk* chunk, const WebPData* const data,
 | |
|                              int copy_data, uint32_t tag);
 | |
| 
 | |
| // Sets 'chunk' at nth position in the 'chunk_list'.
 | |
| // nth = 0 has the special meaning "last of the list".
 | |
| // On success ownership is transferred from 'chunk' to the 'chunk_list'.
 | |
| WebPMuxError ChunkSetNth(WebPChunk* chunk, WebPChunk** chunk_list,
 | |
|                          uint32_t nth);
 | |
| 
 | |
| // Releases chunk and returns chunk->next_.
 | |
| WebPChunk* ChunkRelease(WebPChunk* const chunk);
 | |
| 
 | |
| // Deletes given chunk & returns chunk->next_.
 | |
| WebPChunk* ChunkDelete(WebPChunk* const chunk);
 | |
| 
 | |
| // Returns size of the chunk including chunk header and padding byte (if any).
 | |
| static WEBP_INLINE size_t SizeWithPadding(size_t chunk_size) {
 | |
|   return CHUNK_HEADER_SIZE + ((chunk_size + 1) & ~1U);
 | |
| }
 | |
| 
 | |
| // Size of a chunk including header and padding.
 | |
| static WEBP_INLINE size_t ChunkDiskSize(const WebPChunk* chunk) {
 | |
|   const size_t data_size = chunk->data_.size;
 | |
|   assert(data_size < MAX_CHUNK_PAYLOAD);
 | |
|   return SizeWithPadding(data_size);
 | |
| }
 | |
| 
 | |
| // Total size of a list of chunks.
 | |
| size_t ChunksListDiskSize(const WebPChunk* chunk_list);
 | |
| 
 | |
| // Write out the given list of chunks into 'dst'.
 | |
| uint8_t* ChunkListEmit(const WebPChunk* chunk_list, uint8_t* dst);
 | |
| 
 | |
| // Get the width & height of image stored in 'image_chunk'.
 | |
| WebPMuxError MuxGetImageWidthHeight(const WebPChunk* const image_chunk,
 | |
|                                     int* const width, int* const height);
 | |
| 
 | |
| //------------------------------------------------------------------------------
 | |
| // MuxImage object management.
 | |
| 
 | |
| // Initialize.
 | |
| void MuxImageInit(WebPMuxImage* const wpi);
 | |
| 
 | |
| // Releases image 'wpi' and returns wpi->next.
 | |
| WebPMuxImage* MuxImageRelease(WebPMuxImage* const wpi);
 | |
| 
 | |
| // Delete image 'wpi' and return the next image in the list or NULL.
 | |
| // 'wpi' can be NULL.
 | |
| WebPMuxImage* MuxImageDelete(WebPMuxImage* const wpi);
 | |
| 
 | |
| // Delete all images in 'wpi_list'.
 | |
| void MuxImageDeleteAll(WebPMuxImage** const wpi_list);
 | |
| 
 | |
| // Count number of images matching the given tag id in the 'wpi_list'.
 | |
| // If id == WEBP_CHUNK_NIL, all images will be matched.
 | |
| int MuxImageCount(const WebPMuxImage* wpi_list, WebPChunkId id);
 | |
| 
 | |
| // Check if given ID corresponds to an image related chunk.
 | |
| static WEBP_INLINE int IsWPI(WebPChunkId id) {
 | |
|   switch (id) {
 | |
|     case WEBP_CHUNK_ANMF:
 | |
|     case WEBP_CHUNK_FRGM:
 | |
|     case WEBP_CHUNK_ALPHA:
 | |
|     case WEBP_CHUNK_IMAGE:  return 1;
 | |
|     default:        return 0;
 | |
|   }
 | |
| }
 | |
| 
 | |
| // Get a reference to appropriate chunk list within an image given chunk tag.
 | |
| static WEBP_INLINE WebPChunk** MuxImageGetListFromId(
 | |
|     const WebPMuxImage* const wpi, WebPChunkId id) {
 | |
|   assert(wpi != NULL);
 | |
|   switch (id) {
 | |
|     case WEBP_CHUNK_ANMF:
 | |
|     case WEBP_CHUNK_FRGM:  return (WebPChunk**)&wpi->header_;
 | |
|     case WEBP_CHUNK_ALPHA: return (WebPChunk**)&wpi->alpha_;
 | |
|     case WEBP_CHUNK_IMAGE: return (WebPChunk**)&wpi->img_;
 | |
|     default: return NULL;
 | |
|   }
 | |
| }
 | |
| 
 | |
| // Pushes 'wpi' at the end of 'wpi_list'.
 | |
| WebPMuxError MuxImagePush(const WebPMuxImage* wpi, WebPMuxImage** wpi_list);
 | |
| 
 | |
| // Delete nth image in the image list.
 | |
| WebPMuxError MuxImageDeleteNth(WebPMuxImage** wpi_list, uint32_t nth);
 | |
| 
 | |
| // Get nth image in the image list.
 | |
| WebPMuxError MuxImageGetNth(const WebPMuxImage** wpi_list, uint32_t nth,
 | |
|                             WebPMuxImage** wpi);
 | |
| 
 | |
| // Total size of the given image.
 | |
| size_t MuxImageDiskSize(const WebPMuxImage* const wpi);
 | |
| 
 | |
| // Total size of a list of images.
 | |
| size_t MuxImageListDiskSize(const WebPMuxImage* wpi_list);
 | |
| 
 | |
| // Write out the given image into 'dst'.
 | |
| uint8_t* MuxImageEmit(const WebPMuxImage* const wpi, uint8_t* dst);
 | |
| 
 | |
| // Write out the given list of images into 'dst'.
 | |
| uint8_t* MuxImageListEmit(const WebPMuxImage* wpi_list, uint8_t* dst);
 | |
| 
 | |
| //------------------------------------------------------------------------------
 | |
| // Helper methods for mux.
 | |
| 
 | |
| // Checks if the given image list contains at least one lossless image.
 | |
| int MuxHasLosslessImages(const WebPMuxImage* images);
 | |
| 
 | |
| // Write out RIFF header into 'data', given total data size 'size'.
 | |
| uint8_t* MuxEmitRiffHeader(uint8_t* const data, size_t size);
 | |
| 
 | |
| // Returns the list where chunk with given ID is to be inserted in mux.
 | |
| // Return value is NULL if this chunk should be inserted in mux->images_ list
 | |
| // or if 'id' is not known.
 | |
| WebPChunk** MuxGetChunkListFromId(const WebPMux* mux, WebPChunkId id);
 | |
| 
 | |
| // Validates that the given mux has a single image.
 | |
| WebPMuxError MuxValidateForImage(const WebPMux* const mux);
 | |
| 
 | |
| // Validates the given mux object.
 | |
| WebPMuxError MuxValidate(const WebPMux* const mux);
 | |
| 
 | |
| //------------------------------------------------------------------------------
 | |
| 
 | |
| #if defined(__cplusplus) || defined(c_plusplus)
 | |
| }    // extern "C"
 | |
| #endif
 | |
| 
 | |
| #endif  /* WEBP_MUX_MUXI_H_ */
 | 
