libavcodec/hapenc.c - third_party/ffmpeg - Git at Google

 /*
  * Vidvox Hap encoder
  * Copyright (C) 2015 Vittorio Giovara <vittorio.giovara@gmail.com>
  * Copyright (C) 2015 Tom Butterworth <bangnoise@gmail.com>
  *
  * This file is part of FFmpeg.
  *
  * FFmpeg is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public
  * License as published by the Free Software Foundation; either
  * version 2.1 of the License, or (at your option) any later version.
  *
  * FFmpeg is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  * Lesser General Public License for more details.
  *
  * You should have received a copy of the GNU Lesser General Public
  * License along with FFmpeg; if not, write to the Free Software
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */

 /**
  * @file
  * Hap encoder
  *
  * Fourcc: Hap1, Hap5, HapY
  *
  * https://github.com/Vidvox/hap/blob/master/documentation/HapVideoDRAFT.md
  */

 #include <stdint.h>
 #include "snappy-c.h"

 #include "libavutil/frame.h"
 #include "libavutil/imgutils.h"
 #include "libavutil/mem.h"
 #include "libavutil/opt.h"

 #include "avcodec.h"
 #include "bytestream.h"
 #include "codec_internal.h"
 #include "encode.h"
 #include "hap.h"
 #include "texturedsp.h"

 #define HAP_MAX_CHUNKS 64

 enum HapHeaderLength {
     /* Short header: four bytes with a 24 bit size value */
     HAP_HDR_SHORT = 4,
     /* Long header: eight bytes with a 32 bit size value */
     HAP_HDR_LONG = 8,
 };

 static int compress_texture(AVCodecContext *avctx, uint8_t *out, int out_length, const AVFrame *f)
 {
     HapContext *ctx = avctx->priv_data;

     if (ctx->tex_size > out_length)
         return AVERROR_BUFFER_TOO_SMALL;

     ctx->enc.tex_data.out = out;
     ctx->enc.frame_data.in = f->data[0];
     ctx->enc.stride = f->linesize[0];
     ctx->enc.width  = avctx->width;
     ctx->enc.height = avctx->height;
     ff_texturedsp_exec_compress_threads(avctx, &ctx->enc);

     return 0;
 }

 /* section_length does not include the header */
 static void hap_write_section_header(PutByteContext *pbc,
                                      enum HapHeaderLength header_length,
                                      int section_length,
                                      enum HapSectionType section_type)
 {
     /* The first three bytes are the length of the section (not including the
      * header) or zero if using an eight-byte header.
      * For an eight-byte header, the length is in the last four bytes.
      * The fourth byte stores the section type. */
     bytestream2_put_le24(pbc, header_length == HAP_HDR_LONG ? 0 : section_length);
     bytestream2_put_byte(pbc, section_type);

     if (header_length == HAP_HDR_LONG) {
         bytestream2_put_le32(pbc, section_length);
     }
 }

 static int hap_compress_frame(AVCodecContext *avctx, uint8_t *dst)
 {
     HapContext *ctx = avctx->priv_data;
     int i, final_size = 0;

     for (i = 0; i < ctx->chunk_count; i++) {
         HapChunk *chunk = &ctx->chunks[i];
         uint8_t *chunk_src, *chunk_dst;
         int ret;

         if (i == 0) {
             chunk->compressed_offset = 0;
         } else {
             chunk->compressed_offset = ctx->chunks[i-1].compressed_offset
                                        + ctx->chunks[i-1].compressed_size;
         }
         chunk->uncompressed_size = ctx->tex_size / ctx->chunk_count;
         chunk->uncompressed_offset = i * chunk->uncompressed_size;
         chunk->compressed_size = ctx->max_snappy;
         chunk_src = ctx->tex_buf + chunk->uncompressed_offset;
         chunk_dst = dst + chunk->compressed_offset;

         /* Compress with snappy too, write directly on packet buffer. */
         ret = snappy_compress(chunk_src, chunk->uncompressed_size,
                               chunk_dst, &chunk->compressed_size);
         if (ret != SNAPPY_OK) {
             av_log(avctx, AV_LOG_ERROR, "Snappy compress error.\n");
             return AVERROR_BUG;
         }

         /* If there is no gain from snappy, just use the raw texture. */
         if (chunk->compressed_size >= chunk->uncompressed_size) {
             av_log(avctx, AV_LOG_VERBOSE,
                    "Snappy buffer bigger than uncompressed (%"SIZE_SPECIFIER" >= %"SIZE_SPECIFIER" bytes).\n",
                    chunk->compressed_size, chunk->uncompressed_size);
             memcpy(chunk_dst, chunk_src, chunk->uncompressed_size);
             chunk->compressor = HAP_COMP_NONE;
             chunk->compressed_size = chunk->uncompressed_size;
         } else {
             chunk->compressor = HAP_COMP_SNAPPY;
         }

         final_size += chunk->compressed_size;
     }

     return final_size;
 }

 static int hap_decode_instructions_length(HapContext *ctx)
 {
     /*    Second-Stage Compressor Table (one byte per entry)
      *  + Chunk Size Table (four bytes per entry)
      *  + headers for both sections (short versions)
      *  = chunk_count + (4 * chunk_count) + 4 + 4 */
     return (5 * ctx->chunk_count) + 8;
 }

 static int hap_header_length(HapContext *ctx)
 {
     /* Top section header (long version) */
     int length = HAP_HDR_LONG;

     if (ctx->chunk_count > 1) {
         /* Decode Instructions header (short) + Decode Instructions Container */
         length += HAP_HDR_SHORT + hap_decode_instructions_length(ctx);
     }

     return length;
 }

 static void hap_write_frame_header(HapContext *ctx, uint8_t *dst, int frame_length)
 {
     PutByteContext pbc;
     int i;

     bytestream2_init_writer(&pbc, dst, frame_length);
     if (ctx->chunk_count == 1) {
         /* Write a simple header */
         hap_write_section_header(&pbc, HAP_HDR_LONG, frame_length - 8,
                                  ctx->chunks[0].compressor | ctx->opt_tex_fmt);
     } else {
         /* Write a complex header with Decode Instructions Container */
         hap_write_section_header(&pbc, HAP_HDR_LONG, frame_length - 8,
                                  HAP_COMP_COMPLEX | ctx->opt_tex_fmt);
         hap_write_section_header(&pbc, HAP_HDR_SHORT, hap_decode_instructions_length(ctx),
                                  HAP_ST_DECODE_INSTRUCTIONS);
         hap_write_section_header(&pbc, HAP_HDR_SHORT, ctx->chunk_count,
                                  HAP_ST_COMPRESSOR_TABLE);

         for (i = 0; i < ctx->chunk_count; i++) {
             bytestream2_put_byte(&pbc, ctx->chunks[i].compressor >> 4);
         }

         hap_write_section_header(&pbc, HAP_HDR_SHORT, ctx->chunk_count * 4,
                                  HAP_ST_SIZE_TABLE);

         for (i = 0; i < ctx->chunk_count; i++) {
             bytestream2_put_le32(&pbc, ctx->chunks[i].compressed_size);
         }
     }
 }

 static int hap_encode(AVCodecContext *avctx, AVPacket *pkt,
                       const AVFrame *frame, int *got_packet)
 {
     HapContext *ctx = avctx->priv_data;
     int header_length = hap_header_length(ctx);
     int final_data_size, ret;
     int pktsize = FFMAX(ctx->tex_size, ctx->max_snappy * ctx->chunk_count) + header_length;

     /* Allocate maximum size packet, shrink later. */
     ret = ff_alloc_packet(avctx, pkt, pktsize);
     if (ret < 0)
         return ret;

     if (ctx->opt_compressor == HAP_COMP_NONE) {
         /* DXTC compression directly to the packet buffer. */
         ret = compress_texture(avctx, pkt->data + header_length, pkt->size - header_length, frame);
         if (ret < 0)
             return ret;

         ctx->chunks[0].compressor = HAP_COMP_NONE;
         final_data_size = ctx->tex_size;
     } else {
         /* DXTC compression. */
         ret = compress_texture(avctx, ctx->tex_buf, ctx->tex_size, frame);
         if (ret < 0)
             return ret;

         /* Compress (using Snappy) the frame */
         final_data_size = hap_compress_frame(avctx, pkt->data + header_length);
         if (final_data_size < 0)
             return final_data_size;
     }

     /* Write header at the start. */
     hap_write_frame_header(ctx, pkt->data, final_data_size + header_length);

     av_shrink_packet(pkt, final_data_size + header_length);
     *got_packet = 1;
     return 0;
 }

 static av_cold int hap_init(AVCodecContext *avctx)
 {
     HapContext *ctx = avctx->priv_data;
     TextureDSPEncContext dxtc;
     int corrected_chunk_count;
     int ret = av_image_check_size(avctx->width, avctx->height, 0, avctx);

     if (ret < 0) {
         av_log(avctx, AV_LOG_ERROR, "Invalid video size %dx%d.\n",
                avctx->width, avctx->height);
         return ret;
     }

     if (avctx->width % 4 || avctx->height % 4) {
         av_log(avctx, AV_LOG_ERROR, "Video size %dx%d is not multiple of 4.\n",
                avctx->width, avctx->height);
         return AVERROR_INVALIDDATA;
     }

     ff_texturedspenc_init(&dxtc);

     switch (ctx->opt_tex_fmt) {
     case HAP_FMT_RGBDXT1:
         ctx->enc.tex_ratio = 8;
         avctx->codec_tag = MKTAG('H', 'a', 'p', '1');
         avctx->bits_per_coded_sample = 24;
         ctx->enc.tex_funct = dxtc.dxt1_block;
         break;
     case HAP_FMT_RGBADXT5:
         ctx->enc.tex_ratio = 16;
         avctx->codec_tag = MKTAG('H', 'a', 'p', '5');
         avctx->bits_per_coded_sample = 32;
         ctx->enc.tex_funct = dxtc.dxt5_block;
         break;
     case HAP_FMT_YCOCGDXT5:
         ctx->enc.tex_ratio = 16;
         avctx->codec_tag = MKTAG('H', 'a', 'p', 'Y');
         avctx->bits_per_coded_sample = 24;
         ctx->enc.tex_funct = dxtc.dxt5ys_block;
         break;
     default:
         av_log(avctx, AV_LOG_ERROR, "Invalid format %02X\n", ctx->opt_tex_fmt);
         return AVERROR_INVALIDDATA;
     }
     ctx->enc.raw_ratio = 16;
     ctx->enc.slice_count = av_clip(avctx->thread_count, 1, avctx->height / TEXTURE_BLOCK_H);

     /* Texture compression ratio is constant, so can we computer
      * beforehand the final size of the uncompressed buffer. */
     ctx->tex_size   = avctx->width  / TEXTURE_BLOCK_W *
                       avctx->height / TEXTURE_BLOCK_H * ctx->enc.tex_ratio;

     switch (ctx->opt_compressor) {
     case HAP_COMP_NONE:
         /* No benefit chunking uncompressed data */
         corrected_chunk_count = 1;

         ctx->max_snappy = ctx->tex_size;
         ctx->tex_buf = NULL;
         break;
     case HAP_COMP_SNAPPY:
         /* Round the chunk count to divide evenly on DXT block edges */
         corrected_chunk_count = av_clip(ctx->opt_chunk_count, 1, HAP_MAX_CHUNKS);
         while ((ctx->tex_size / ctx->enc.tex_ratio) % corrected_chunk_count != 0) {
             corrected_chunk_count--;
         }

         ctx->max_snappy = snappy_max_compressed_length(ctx->tex_size / corrected_chunk_count);
         ctx->tex_buf = av_malloc(ctx->tex_size);
         if (!ctx->tex_buf) {
             return AVERROR(ENOMEM);
         }
         break;
     default:
         av_log(avctx, AV_LOG_ERROR, "Invalid compresor %02X\n", ctx->opt_compressor);
         return AVERROR_INVALIDDATA;
     }
     if (corrected_chunk_count != ctx->opt_chunk_count) {
         av_log(avctx, AV_LOG_INFO, "%d chunks requested but %d used.\n",
                                     ctx->opt_chunk_count, corrected_chunk_count);
     }
     ret = ff_hap_set_chunk_count(ctx, corrected_chunk_count, 1);
     if (ret != 0)
         return ret;

     return 0;
 }

 static av_cold int hap_close(AVCodecContext *avctx)
 {
     HapContext *ctx = avctx->priv_data;

     ff_hap_free_context(ctx);

     return 0;
 }

 #define OFFSET(x) offsetof(HapContext, x)
 #define FLAGS     AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
 static const AVOption options[] = {
     { "format", NULL, OFFSET(opt_tex_fmt), AV_OPT_TYPE_INT, { .i64 = HAP_FMT_RGBDXT1 }, HAP_FMT_RGBDXT1, HAP_FMT_YCOCGDXT5, FLAGS, .unit = "format" },
         { "hap",       "Hap 1 (DXT1 textures)", 0, AV_OPT_TYPE_CONST, { .i64 = HAP_FMT_RGBDXT1   }, 0, 0, FLAGS, .unit = "format" },
         { "hap_alpha", "Hap Alpha (DXT5 textures)", 0, AV_OPT_TYPE_CONST, { .i64 = HAP_FMT_RGBADXT5  }, 0, 0, FLAGS, .unit = "format" },
         { "hap_q",     "Hap Q (DXT5-YCoCg textures)", 0, AV_OPT_TYPE_CONST, { .i64 = HAP_FMT_YCOCGDXT5 }, 0, 0, FLAGS, .unit = "format" },
     { "chunks", "chunk count", OFFSET(opt_chunk_count), AV_OPT_TYPE_INT, {.i64 = 1 }, 1, HAP_MAX_CHUNKS, FLAGS, },
     { "compressor", "second-stage compressor", OFFSET(opt_compressor), AV_OPT_TYPE_INT, { .i64 = HAP_COMP_SNAPPY }, HAP_COMP_NONE, HAP_COMP_SNAPPY, FLAGS, .unit = "compressor" },
         { "none",       "None", 0, AV_OPT_TYPE_CONST, { .i64 = HAP_COMP_NONE }, 0, 0, FLAGS, .unit = "compressor" },
         { "snappy",     "Snappy", 0, AV_OPT_TYPE_CONST, { .i64 = HAP_COMP_SNAPPY }, 0, 0, FLAGS, .unit = "compressor" },
     { NULL },
 };

 static const AVClass hapenc_class = {
     .class_name = "Hap encoder",
     .item_name  = av_default_item_name,
     .option     = options,
     .version    = LIBAVUTIL_VERSION_INT,
 };

 const FFCodec ff_hap_encoder = {
     .p.name         = "hap",
     CODEC_LONG_NAME("Vidvox Hap"),
     .p.type         = AVMEDIA_TYPE_VIDEO,
     .p.id           = AV_CODEC_ID_HAP,
     .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_SLICE_THREADS |
                       AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE,
     .priv_data_size = sizeof(HapContext),
     .p.priv_class   = &hapenc_class,
     .init           = hap_init,
     FF_CODEC_ENCODE_CB(hap_encode),
     .close          = hap_close,
     .p.pix_fmts     = (const enum AVPixelFormat[]) {
         AV_PIX_FMT_RGBA, AV_PIX_FMT_NONE,
     },
     .caps_internal  = FF_CODEC_CAP_INIT_CLEANUP,
 };
	/*
	* Vidvox Hap encoder
	* Copyright (C) 2015 Vittorio Giovara <vittorio.giovara@gmail.com>
	* Copyright (C) 2015 Tom Butterworth <bangnoise@gmail.com>
	*
	* This file is part of FFmpeg.
	*
	* FFmpeg is free software; you can redistribute it and/or
	* modify it under the terms of the GNU Lesser General Public
	* License as published by the Free Software Foundation; either
	* version 2.1 of the License, or (at your option) any later version.
	*
	* FFmpeg is distributed in the hope that it will be useful,
	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	* Lesser General Public License for more details.
	*
	* You should have received a copy of the GNU Lesser General Public
	* License along with FFmpeg; if not, write to the Free Software
	* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
	*/

	/**
	* @file
	* Hap encoder
	*
	* Fourcc: Hap1, Hap5, HapY
	*
	* https://github.com/Vidvox/hap/blob/master/documentation/HapVideoDRAFT.md
	*/

	#include <stdint.h>
	#include "snappy-c.h"

	#include "libavutil/frame.h"
	#include "libavutil/imgutils.h"
	#include "libavutil/mem.h"
	#include "libavutil/opt.h"

	#include "avcodec.h"
	#include "bytestream.h"
	#include "codec_internal.h"
	#include "encode.h"
	#include "hap.h"
	#include "texturedsp.h"

	#define HAP_MAX_CHUNKS 64

	enum HapHeaderLength {
	/* Short header: four bytes with a 24 bit size value */
	HAP_HDR_SHORT = 4,
	/* Long header: eight bytes with a 32 bit size value */
	HAP_HDR_LONG = 8,
	};

	static int compress_texture(AVCodecContext avctx, uint8_t out, int out_length, const AVFrame *f)
	{
	HapContext *ctx = avctx->priv_data;

	if (ctx->tex_size > out_length)
	return AVERROR_BUFFER_TOO_SMALL;

	ctx->enc.tex_data.out = out;
	ctx->enc.frame_data.in = f->data[0];
	ctx->enc.stride = f->linesize[0];
	ctx->enc.width = avctx->width;
	ctx->enc.height = avctx->height;
	ff_texturedsp_exec_compress_threads(avctx, &ctx->enc);

	return 0;
	}

	/* section_length does not include the header */
	static void hap_write_section_header(PutByteContext *pbc,
	enum HapHeaderLength header_length,
	int section_length,
	enum HapSectionType section_type)
	{
	/* The first three bytes are the length of the section (not including the
	* header) or zero if using an eight-byte header.
	* For an eight-byte header, the length is in the last four bytes.
	* The fourth byte stores the section type. */
	bytestream2_put_le24(pbc, header_length == HAP_HDR_LONG ? 0 : section_length);
	bytestream2_put_byte(pbc, section_type);

	if (header_length == HAP_HDR_LONG) {
	bytestream2_put_le32(pbc, section_length);
	}
	}

	static int hap_compress_frame(AVCodecContext avctx, uint8_t dst)
	{
	HapContext *ctx = avctx->priv_data;
	int i, final_size = 0;

	for (i = 0; i < ctx->chunk_count; i++) {
	HapChunk *chunk = &ctx->chunks[i];
	uint8_t chunk_src, chunk_dst;
	int ret;

	if (i == 0) {
	chunk->compressed_offset = 0;
	} else {
	chunk->compressed_offset = ctx->chunks[i-1].compressed_offset
	+ ctx->chunks[i-1].compressed_size;
	}
	chunk->uncompressed_size = ctx->tex_size / ctx->chunk_count;
	chunk->uncompressed_offset = i * chunk->uncompressed_size;
	chunk->compressed_size = ctx->max_snappy;
	chunk_src = ctx->tex_buf + chunk->uncompressed_offset;
	chunk_dst = dst + chunk->compressed_offset;

	/* Compress with snappy too, write directly on packet buffer. */
	ret = snappy_compress(chunk_src, chunk->uncompressed_size,
	chunk_dst, &chunk->compressed_size);
	if (ret != SNAPPY_OK) {
	av_log(avctx, AV_LOG_ERROR, "Snappy compress error.\n");
	return AVERROR_BUG;
	}

	/* If there is no gain from snappy, just use the raw texture. */
	if (chunk->compressed_size >= chunk->uncompressed_size) {
	av_log(avctx, AV_LOG_VERBOSE,
	"Snappy buffer bigger than uncompressed (%"SIZE_SPECIFIER" >= %"SIZE_SPECIFIER" bytes).\n",
	chunk->compressed_size, chunk->uncompressed_size);
	memcpy(chunk_dst, chunk_src, chunk->uncompressed_size);
	chunk->compressor = HAP_COMP_NONE;
	chunk->compressed_size = chunk->uncompressed_size;
	} else {
	chunk->compressor = HAP_COMP_SNAPPY;
	}

	final_size += chunk->compressed_size;
	}

	return final_size;
	}

	static int hap_decode_instructions_length(HapContext *ctx)
	{
	/* Second-Stage Compressor Table (one byte per entry)
	* + Chunk Size Table (four bytes per entry)
	* + headers for both sections (short versions)
	* = chunk_count + (4 * chunk_count) + 4 + 4 */
	return (5 * ctx->chunk_count) + 8;
	}

	static int hap_header_length(HapContext *ctx)
	{
	/* Top section header (long version) */
	int length = HAP_HDR_LONG;

	if (ctx->chunk_count > 1) {
	/* Decode Instructions header (short) + Decode Instructions Container */
	length += HAP_HDR_SHORT + hap_decode_instructions_length(ctx);
	}

	return length;
	}

	static void hap_write_frame_header(HapContext ctx, uint8_t dst, int frame_length)
	{
	PutByteContext pbc;
	int i;

	bytestream2_init_writer(&pbc, dst, frame_length);
	if (ctx->chunk_count == 1) {
	/* Write a simple header */
	hap_write_section_header(&pbc, HAP_HDR_LONG, frame_length - 8,
	ctx->chunks[0].compressor \| ctx->opt_tex_fmt);
	} else {
	/* Write a complex header with Decode Instructions Container */
	hap_write_section_header(&pbc, HAP_HDR_LONG, frame_length - 8,
	HAP_COMP_COMPLEX \| ctx->opt_tex_fmt);
	hap_write_section_header(&pbc, HAP_HDR_SHORT, hap_decode_instructions_length(ctx),
	HAP_ST_DECODE_INSTRUCTIONS);
	hap_write_section_header(&pbc, HAP_HDR_SHORT, ctx->chunk_count,
	HAP_ST_COMPRESSOR_TABLE);

	for (i = 0; i < ctx->chunk_count; i++) {
	bytestream2_put_byte(&pbc, ctx->chunks[i].compressor >> 4);
	}

	hap_write_section_header(&pbc, HAP_HDR_SHORT, ctx->chunk_count * 4,
	HAP_ST_SIZE_TABLE);

	for (i = 0; i < ctx->chunk_count; i++) {
	bytestream2_put_le32(&pbc, ctx->chunks[i].compressed_size);
	}
	}
	}

	static int hap_encode(AVCodecContext avctx, AVPacket pkt,
	const AVFrame frame, int got_packet)
	{
	HapContext *ctx = avctx->priv_data;
	int header_length = hap_header_length(ctx);
	int final_data_size, ret;
	int pktsize = FFMAX(ctx->tex_size, ctx->max_snappy * ctx->chunk_count) + header_length;

	/* Allocate maximum size packet, shrink later. */
	ret = ff_alloc_packet(avctx, pkt, pktsize);
	if (ret < 0)
	return ret;

	if (ctx->opt_compressor == HAP_COMP_NONE) {
	/* DXTC compression directly to the packet buffer. */
	ret = compress_texture(avctx, pkt->data + header_length, pkt->size - header_length, frame);
	if (ret < 0)
	return ret;

	ctx->chunks[0].compressor = HAP_COMP_NONE;
	final_data_size = ctx->tex_size;
	} else {
	/* DXTC compression. */
	ret = compress_texture(avctx, ctx->tex_buf, ctx->tex_size, frame);
	if (ret < 0)
	return ret;

	/* Compress (using Snappy) the frame */
	final_data_size = hap_compress_frame(avctx, pkt->data + header_length);
	if (final_data_size < 0)
	return final_data_size;
	}

	/* Write header at the start. */
	hap_write_frame_header(ctx, pkt->data, final_data_size + header_length);

	av_shrink_packet(pkt, final_data_size + header_length);
	*got_packet = 1;
	return 0;
	}

	static av_cold int hap_init(AVCodecContext *avctx)
	{
	HapContext *ctx = avctx->priv_data;
	TextureDSPEncContext dxtc;
	int corrected_chunk_count;
	int ret = av_image_check_size(avctx->width, avctx->height, 0, avctx);

	if (ret < 0) {
	av_log(avctx, AV_LOG_ERROR, "Invalid video size %dx%d.\n",
	avctx->width, avctx->height);
	return ret;
	}

	if (avctx->width % 4 \|\| avctx->height % 4) {
	av_log(avctx, AV_LOG_ERROR, "Video size %dx%d is not multiple of 4.\n",
	avctx->width, avctx->height);
	return AVERROR_INVALIDDATA;
	}

	ff_texturedspenc_init(&dxtc);

	switch (ctx->opt_tex_fmt) {
	case HAP_FMT_RGBDXT1:
	ctx->enc.tex_ratio = 8;
	avctx->codec_tag = MKTAG('H', 'a', 'p', '1');
	avctx->bits_per_coded_sample = 24;
	ctx->enc.tex_funct = dxtc.dxt1_block;
	break;
	case HAP_FMT_RGBADXT5:
	ctx->enc.tex_ratio = 16;
	avctx->codec_tag = MKTAG('H', 'a', 'p', '5');
	avctx->bits_per_coded_sample = 32;
	ctx->enc.tex_funct = dxtc.dxt5_block;
	break;
	case HAP_FMT_YCOCGDXT5:
	ctx->enc.tex_ratio = 16;
	avctx->codec_tag = MKTAG('H', 'a', 'p', 'Y');
	avctx->bits_per_coded_sample = 24;
	ctx->enc.tex_funct = dxtc.dxt5ys_block;
	break;
	default:
	av_log(avctx, AV_LOG_ERROR, "Invalid format %02X\n", ctx->opt_tex_fmt);
	return AVERROR_INVALIDDATA;
	}
	ctx->enc.raw_ratio = 16;
	ctx->enc.slice_count = av_clip(avctx->thread_count, 1, avctx->height / TEXTURE_BLOCK_H);

	/* Texture compression ratio is constant, so can we computer
	* beforehand the final size of the uncompressed buffer. */
	ctx->tex_size = avctx->width / TEXTURE_BLOCK_W *
	avctx->height / TEXTURE_BLOCK_H * ctx->enc.tex_ratio;

	switch (ctx->opt_compressor) {
	case HAP_COMP_NONE:
	/* No benefit chunking uncompressed data */
	corrected_chunk_count = 1;

	ctx->max_snappy = ctx->tex_size;
	ctx->tex_buf = NULL;
	break;
	case HAP_COMP_SNAPPY:
	/* Round the chunk count to divide evenly on DXT block edges */
	corrected_chunk_count = av_clip(ctx->opt_chunk_count, 1, HAP_MAX_CHUNKS);
	while ((ctx->tex_size / ctx->enc.tex_ratio) % corrected_chunk_count != 0) {
	corrected_chunk_count--;
	}

	ctx->max_snappy = snappy_max_compressed_length(ctx->tex_size / corrected_chunk_count);
	ctx->tex_buf = av_malloc(ctx->tex_size);
	if (!ctx->tex_buf) {
	return AVERROR(ENOMEM);
	}
	break;
	default:
	av_log(avctx, AV_LOG_ERROR, "Invalid compresor %02X\n", ctx->opt_compressor);
	return AVERROR_INVALIDDATA;
	}
	if (corrected_chunk_count != ctx->opt_chunk_count) {
	av_log(avctx, AV_LOG_INFO, "%d chunks requested but %d used.\n",
	ctx->opt_chunk_count, corrected_chunk_count);
	}
	ret = ff_hap_set_chunk_count(ctx, corrected_chunk_count, 1);
	if (ret != 0)
	return ret;

	return 0;
	}

	static av_cold int hap_close(AVCodecContext *avctx)
	{
	HapContext *ctx = avctx->priv_data;

	ff_hap_free_context(ctx);

	return 0;
	}

	#define OFFSET(x) offsetof(HapContext, x)
	#define FLAGS AV_OPT_FLAG_VIDEO_PARAM \| AV_OPT_FLAG_ENCODING_PARAM
	static const AVOption options[] = {
	{ "format", NULL, OFFSET(opt_tex_fmt), AV_OPT_TYPE_INT, { .i64 = HAP_FMT_RGBDXT1 }, HAP_FMT_RGBDXT1, HAP_FMT_YCOCGDXT5, FLAGS, .unit = "format" },
	{ "hap", "Hap 1 (DXT1 textures)", 0, AV_OPT_TYPE_CONST, { .i64 = HAP_FMT_RGBDXT1 }, 0, 0, FLAGS, .unit = "format" },
	{ "hap_alpha", "Hap Alpha (DXT5 textures)", 0, AV_OPT_TYPE_CONST, { .i64 = HAP_FMT_RGBADXT5 }, 0, 0, FLAGS, .unit = "format" },
	{ "hap_q", "Hap Q (DXT5-YCoCg textures)", 0, AV_OPT_TYPE_CONST, { .i64 = HAP_FMT_YCOCGDXT5 }, 0, 0, FLAGS, .unit = "format" },
	{ "chunks", "chunk count", OFFSET(opt_chunk_count), AV_OPT_TYPE_INT, {.i64 = 1 }, 1, HAP_MAX_CHUNKS, FLAGS, },
	{ "compressor", "second-stage compressor", OFFSET(opt_compressor), AV_OPT_TYPE_INT, { .i64 = HAP_COMP_SNAPPY }, HAP_COMP_NONE, HAP_COMP_SNAPPY, FLAGS, .unit = "compressor" },
	{ "none", "None", 0, AV_OPT_TYPE_CONST, { .i64 = HAP_COMP_NONE }, 0, 0, FLAGS, .unit = "compressor" },
	{ "snappy", "Snappy", 0, AV_OPT_TYPE_CONST, { .i64 = HAP_COMP_SNAPPY }, 0, 0, FLAGS, .unit = "compressor" },
	{ NULL },
	};

	static const AVClass hapenc_class = {
	.class_name = "Hap encoder",
	.item_name = av_default_item_name,
	.option = options,
	.version = LIBAVUTIL_VERSION_INT,
	};

	const FFCodec ff_hap_encoder = {
	.p.name = "hap",
	CODEC_LONG_NAME("Vidvox Hap"),
	.p.type = AVMEDIA_TYPE_VIDEO,
	.p.id = AV_CODEC_ID_HAP,
	.p.capabilities = AV_CODEC_CAP_DR1 \| AV_CODEC_CAP_SLICE_THREADS \|
	AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE,
	.priv_data_size = sizeof(HapContext),
	.p.priv_class = &hapenc_class,
	.init = hap_init,
	FF_CODEC_ENCODE_CB(hap_encode),
	.close = hap_close,
	.p.pix_fmts = (const enum AVPixelFormat[]) {
	AV_PIX_FMT_RGBA, AV_PIX_FMT_NONE,
	},
	.caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
	};