commit | 70f9af8e0894b2a04d5fff910e20a00598984cba | [log] [tgz] |
---|---|---|
author | Joe Drago <jdrago@netflix.com> | Thu Apr 16 16:32:06 2020 -0700 |
committer | Joe Drago <jdrago@netflix.com> | Thu Apr 16 16:32:06 2020 -0700 |
tree | 870bc57429e96fe5b119459d6ed930fedec1d130 | |
parent | c74f70c047aada1af7bf650a138ec48faf037fca [diff] |
avifenc: Set nclx/range values in avifImage earlier so proper YUV coefficients are used when converting JPEG/PNG
This library aims to be a friendly, portable C implementation of the AV1 Image File Format, as described here:
https://aomediacodec.github.io/av1-avif/
It is a work-in-progress, but can already encode and decode all AOM supported YUV formats and bit depths (with alpha).
For now, it is recommended that you checkout/use tagged releases instead of just using the master branch. I will regularly create new versions as bugfixes and features are added.
#include "avif/avif.h" // NOTE: avifDecoderRead() offers the simplest means to get an avifImage that is complete independent of // an avifDecoder, but at the cost of additional allocations and copies, and no support for image sequences. // If you don't mind keeping around the avifDecoder while you read in the image and/or need image sequence // support, skip ahead to the Advanced Decoding example. It is only one additional function call, and the // avifImage is owned by the avifDecoder. // point raw.data and raw.size to the contents of an .avif(s) avifROData raw; raw.data = ...; raw.size = ...; avifImage * image = avifImageCreateEmpty(); avifDecoder * decoder = avifDecoderCreate(); avifResult decodeResult = avifDecoderRead(decoder, image, &raw); if (decodeResult == AVIF_RESULT_OK) { // image is an independent copy of decoded data, decoder may be destroyed here ... image->width; ... image->height; ... image->depth; // If >8, all plane ptrs below are uint16_t* ... image->yuvFormat; // U and V planes might be smaller than Y based on format, // use avifGetPixelFormatInfo() to find out in a generic way // Option 1: Use YUV planes directly ... image->yuvPlanes; ... image->yuvRowBytes; // Option 2: Convert to interleaved RGB(A)/BGR(A) using a libavif-allocated buffer. avifRGBImage rgb; avifRGBImageSetDefaults(&rgb, image); rgb.format = ...; // See choices in avif.h rgb.depth = ...; // [8, 10, 12, 16]; Does not need to match image->depth. // If >8, rgb->pixels is uint16_t* avifRGBImageAllocatePixels(&rgb); // You can supply your own pixels/rowBytes, see Option 3 avifImageYUVToRGB(image, &rgb); ... rgb.pixels; // Pixels in interleaved rgbFormat chosen above; ... rgb.rowBytes; // all channels are always full range avifRGBImageFreePixels(&rgb); // Option 3: Convert directly into your own pre-existing interleaved RGB(A)/BGR(A) buffer avifRGBImage rgb; avifRGBImageSetDefaults(&rgb, image); rgb.format = ...; // See choices in avif.h rgb.depth = ...; // [8, 10, 12, 16]; Does not need to match image->depth. // If >8, rgb->pixels is uint16_t* rgb.pixels = ...; // Point at your RGB(A)/BGR(A) pixels here rgb.rowBytes = ...; avifImageYUVToRGB(image, &rgb); ... rgb.pixels; // Pixels in interleaved rgbFormat chosen above; ... rgb.rowBytes; // all channels are always full range // Use your own buffer; no need to call avifRGBImageFreePixels() // Use alpha plane, if present. // Note: This might be limited range! if (image->alphaPlane) { ... image->alphaPlane; ... image->alphaRowBytes; ... image->alphaRange; } // Optional: query color profile if (image->profileFormat == AVIF_PROFILE_FORMAT_ICC) { // ICC profile present ... image->icc.data; ... image->icc.size; } else if (image->profileFormat == AVIF_PROFILE_FORMAT_NCLX) { // NCLX profile present ... image->nclx.colourPrimaries; ... image->nclx.transferCharacteristics; ... image->nclx.matrixCoefficients; ... image->nclx.fullRangeFlag; } // Optional: Exif and XMP metadata querying if(image->exif.size > 0) { // Parse Exif payload ... image->exif.data; ... image->exif.size; } if(image->xmp.size > 0) { // Parse XMP document ... image->xmp.data; ... image->xmp.size; } } else { printf("ERROR: Failed to decode: %s\n", avifResultToString(result)); } avifImageDestroy(image); avifDecoderDestroy(decoder);
#include "avif/avif.h" // point raw.data and raw.size to the contents of an .avif(s) avifROData raw; raw.data = ...; raw.size = ...; avifDecoder * decoder = avifDecoderCreate(); avifResult decodeResult = avifDecoderParse(decoder, &raw); if (decodeResult == AVIF_RESULT_OK) { // Timing and frame information ... decoder->imageCount; // Total images expected to decode ... decoder->duration; // Duration of entire sequence (seconds) for (;;) { avifResult nextImageResult = avifDecoderNextImage(decoder); if (nextImageResult == AVIF_RESULT_NO_IMAGES_REMAINING) { // No more images, bail out. Verify that you got the expected amount of images decoded. break; } else if (nextImageResult != AVIF_RESULT_OK) { printf("ERROR: Failed to decode all frames: %s\n", avifResultToString(nextImageResult)); break; } // decoder->image now points at decoder owned planes, and the image itself // is also owned and dependent on decoder. decoder->image's data/pointers are // likely to be completely different after each call to avifDecoderNextImage(). ... decoder->image->width; ... decoder->image->height; ... decoder->image->depth; // If >8, all plane ptrs below are uint16_t* ... decoder->image->yuvFormat; // U and V planes might be smaller than Y based on format, // use avifGetPixelFormatInfo() to find out in a generic way // See Basic Decoding example for color profile and metadata querying // Option 1: Use YUV planes directly ... decoder->image->yuvPlanes; ... decoder->image->yuvRowBytes; // Option 2: Convert to interleaved RGB(A)/BGR(A) using a libavif-allocated buffer. avifRGBImage rgb; avifRGBImageSetDefaults(&rgb, image); rgb.format = ...; // See choices in avif.h rgb.depth = ...; // [8, 10, 12, 16]; Does not need to match image->depth. // If >8, rgb->pixels is uint16_t* avifRGBImageAllocatePixels(&rgb); // You can supply your own pixels/rowBytes, see Option 3 avifImageYUVToRGB(image, &rgb); ... rgb.pixels; // Pixels in interleaved rgbFormat chosen above; ... rgb.rowBytes; // all channels are always full range avifRGBImageFreePixels(&rgb); // Option 3: Convert directly into your own pre-existing interleaved RGB(A)/BGR(A) buffer avifRGBImage rgb; avifRGBImageSetDefaults(&rgb, image); rgb.format = ...; // See choices in avif.h rgb.depth = ...; // [8, 10, 12, 16]; Does not need to match image->depth. // If >8, rgb->pixels is uint16_t* rgb.pixels = ...; // Point at your RGB(A)/BGR(A) pixels here rgb.rowBytes = ...; avifImageYUVToRGB(image, &rgb); ... rgb.pixels; // Pixels in interleaved rgbFormat chosen above; ... rgb.rowBytes; // all channels are always full range // Use your own buffer; no need to call avifRGBImageFreePixels() // Use alpha plane, if present. // Note: This might be limited range! if (decoder->image->alphaPlane) { ... image->alphaPlane; ... image->alphaRowBytes; ... image->alphaRange; } // Timing and frame information ... decoder->imageIndex; // Current index (0-based) ... decoder->imageTiming.pts; // Current image's presentation timestamp (seconds) ... decoder->imageTiming.duration; // Current image's duration (seconds) // Optional: If you want to have a decoder-independent copy of image data avifImage * image = avifImageCreateEmpty(); avifImageCopy(image, decoder->image); ... image; // do something with image avifImageDestroy(image); // destroy later } } else { printf("ERROR: Failed to decode: %s\n", avifResultToString(result)); } avifDecoderDestroy(decoder);
#include "avif/avif.h" int width = 32; int height = 32; int depth = 8; avifPixelFormat format = AVIF_PIXEL_FORMAT_YUV420; avifImage * image = avifImageCreate(width, height, depth, format); // Option 1: Populate YUV planes avifImageAllocatePlanes(image, AVIF_PLANES_YUV); ... image->yuvPlanes; ... image->yuvRowBytes; // Option 2: Populate RGB planes (if YUV planes are absent, RGB->YUV conversion will automatically happen) avifImageAllocatePlanes(image, AVIF_PLANES_RGB); ... image->rgbPlanes; ... image->rgbRowBytes; // Option 2: Convert from interleaved RGB(A)/BGR(A) using a libavif-allocated buffer. uint32_t rgbDepth = ...; // [8, 10, 12, 16]; Does not need to match image->depth. // If >8, rgb->pixels is uint16_t* avifRGBFormat rgbFormat = AVIF_RGB_FORMAT_RGBA; // See choices in avif.h avifRGBImage * rgb = avifRGBImageCreate(image->width, image->width, rgbDepth, rgbFormat); ... rgb->pixels; // fill these pixels; all channel data must be full range ... rgb->rowBytes; avifImageRGBToYUV(image, rgb); // if alpha is present, it will also be copied/converted avifRGBImageDestroy(rgb); // Option 3: Convert directly from your own pre-existing interleaved RGB(A)/BGR(A) buffer avifRGBImage rgb; rgb.width = image->width; rgb.height = image->height; rgb.depth = ...; // [8, 10, 12, 16]; Does not need to match image->depth. // If >8, rgb->pixels is uint16_t* rgb.format = ...; // See choices in avif.h, match to your buffer's pixel format rgb.pixels = ...; // Point at your RGB(A)/BGR(A) pixels here rgb.rowBytes = ...; avifImageRGBToYUV(image, rgb); // if alpha is present, it will also be copied/converted // no need to cleanup avifRGBImage // Optional: Populate alpha plane // Note: This step is unnecessary if you used avifImageRGBToYUV from an // avifRGBImage containing an alpha channel. avifImageAllocatePlanes(image, AVIF_PLANES_A); ... image->alphaPlane; ... image->alphaRowBytes; // Optional: Set color profile based on NCLX box avifNclxColorProfile nclx; nclx.colourPrimaries = AVIF_NCLX_COLOUR_PRIMARIES_BT709; nclx.transferCharacteristics = AVIF_NCLX_TRANSFER_CHARACTERISTICS_GAMMA22; nclx.matrixCoefficients = AVIF_NCLX_MATRIX_COEFFICIENTS_BT709; nclx.fullRangeFlag = AVIF_NCLX_FULL_RANGE; avifImageSetProfileNCLX(image, &nclx); // Optional: Set color profile based on ICC profile uint8_t * icc = ...; // raw ICC profile data size_t iccSize = ...; // Length of raw ICC profile data avifImageSetProfileICC(image, icc, iccSize); // Optional: Set Exif and/or XMP metadata uint8_t * exif = ...; // raw Exif payload size_t exifSize = ...; // Length of raw Exif payload avifImageSetMetadataExif(image, exif, exifSize); uint8_t * xmp = ...; // raw XMP document size_t xmpSize = ...; // Length of raw XMP document avifImageSetMetadataXMP(image, xmp, xmpSize); avifRWData output = AVIF_DATA_EMPTY; avifEncoder * encoder = avifEncoderCreate(); encoder->maxThreads = ...; // Choose max encoder threads, 1 to disable multithreading encoder->minQuantizer = AVIF_QUANTIZER_LOSSLESS; encoder->maxQuantizer = AVIF_QUANTIZER_LOSSLESS; avifResult encodeResult = avifEncoderWrite(encoder, image, &output); if (encodeResult == AVIF_RESULT_OK) { // output contains a valid .avif file's contents ... output.data; ... output.size; } else { printf("ERROR: Failed to encode: %s\n", avifResultToString(encodeResult)); } avifImageDestroy(image); avifRWDataFree(&output); avifEncoderDestroy(encoder);
Building libavif requires CMake.
No AV1 codecs are enabled by default. Enable them by enabling any of the following CMake options:
AVIF_CODEC_AOM
- requires CMake, NASMAVIF_CODEC_DAV1D
- requires Meson, Ninja, NASMAVIF_CODEC_LIBGAV1
- requires CMake, NinjaAVIF_CODEC_RAV1E
- requires cargo (Rust), NASMThese libraries (in their C API form) must be externally available (discoverable via CMake‘s FIND_LIBRARY
) to use them, or if libavif is a child CMake project, the appropriate CMake target must already exist by the time libavif’s CMake scripts are executed.
The ext/
subdirectory contains a handful of basic scripts which each pull down a known-good copy of an AV1 codec and make a local static library build. If you want to statically link any codec into your local (static) build of libavif, building using one of these scripts and then enabling the associated AVIF_LOCAL_*
is a convenient method, but you must make sure to disable BUILD_SHARED_LIBS
in CMake to instruct it to make a static libavif library.
If you want to build/install shared libraries for AV1 codecs, you can still peek inside of each script to see where the current known-good SHA is for each codec.
If you're building on Windows with Visual Studio 2019 and want to try out libavif without going through the build process, static library builds for both Debug and Release are available on AppVeyor.
Released under the BSD License.
Copyright 2019 Joe Drago. All rights reserved. Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.