Function write_huff_codes

libavcodec/utvideoenc.c:364–392

Write huffman bit codes to a memory block */

/* Write huffman bit codes to a memory block */
static int write_huff_codes(uint8_t *src, uint8_t *dst, int dst_size,
                            int width, int height, HuffEntry *he)
{
    PutBitContext pb;
    int i, j;
    int count;

    init_put_bits(&pb, dst, dst_size);

    /* Write the codes */
    for (j = 0; j < height; j++) {
        for (i = 0; i < width; i++)
            put_bits(&pb, he[src[i]].len, he[src[i]].code);

        src += width;
    }

    /* Pad output to a 32-bit boundary */
    count = put_bits_count(&pb) & 0x1F;

    if (count)
        put_bits(&pb, 32 - count, 0);

    /* Flush the rest with zeroes */
    flush_put_bits(&pb);

    /* Return the amount of bytes written */
    return put_bytes_output(&pb);
}

static int encode_plane(AVCodecContext *avctx, const uint8_t *src,
                        uint8_t *dst, ptrdiff_t stride, int plane_no,