ffv1.h 5.24 KB
Newer Older
Luca Barbato's avatar
Luca Barbato committed
1 2 3
/*
 * FFV1 codec for libavcodec
 *
Luca Barbato's avatar
Luca Barbato committed
4
 * Copyright (c) 2003-2012 Michael Niedermayer <michaelni@gmx.at>
Luca Barbato's avatar
Luca Barbato committed
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
 *
 * This file is part of Libav.
 *
 * Libav 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.
 *
 * Libav 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 Libav; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

#ifndef AVCODEC_FFV1_H
#define AVCODEC_FFV1_H
25 26 27

#include <stdint.h>

Luca Barbato's avatar
Luca Barbato committed
28
#include "avcodec.h"
29 30 31
#include "get_bits.h"
#include "put_bits.h"
#include "rangecoder.h"
Luca Barbato's avatar
Luca Barbato committed
32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65

#define MAX_PLANES 4
#define CONTEXT_SIZE 32

#define MAX_QUANT_TABLES 8
#define MAX_CONTEXT_INPUTS 5

extern const uint8_t ff_log2_run[41];

extern const int8_t ffv1_quant5_10bit[256];
extern const int8_t ffv1_quant5[256];
extern const int8_t ffv1_quant9_10bit[256];
extern const int8_t ffv1_quant11[256];
extern const uint8_t ffv1_ver2_state[256];

typedef struct VlcState {
    int16_t drift;
    uint16_t error_sum;
    int8_t bias;
    uint8_t count;
} VlcState;

typedef struct PlaneContext {
    int16_t quant_table[MAX_CONTEXT_INPUTS][256];
    int quant_table_index;
    int context_count;
    uint8_t (*state)[CONTEXT_SIZE];
    VlcState *vlc_state;
    uint8_t interlace_bit_state[2];
} PlaneContext;

#define MAX_SLICES 256

typedef struct FFV1Context {
Luca Barbato's avatar
Luca Barbato committed
66
    AVClass *class;
Luca Barbato's avatar
Luca Barbato committed
67 68 69 70 71 72 73
    AVCodecContext *avctx;
    RangeCoder c;
    GetBitContext gb;
    PutBitContext pb;
    uint64_t rc_stat[256][2];
    uint64_t (*rc_stat2[MAX_QUANT_TABLES])[32][2];
    int version;
Luca Barbato's avatar
Luca Barbato committed
74
    int minor_version;
Luca Barbato's avatar
Luca Barbato committed
75
    int width, height;
Luca Barbato's avatar
Luca Barbato committed
76
    int chroma_planes;
Luca Barbato's avatar
Luca Barbato committed
77
    int chroma_h_shift, chroma_v_shift;
Luca Barbato's avatar
Luca Barbato committed
78
    int transparency;
Luca Barbato's avatar
Luca Barbato committed
79 80
    int flags;
    int picture_number;
81 82
    AVFrame *frame;
    AVFrame *last_picture;
83 84

    AVFrame *cur;
Luca Barbato's avatar
Luca Barbato committed
85 86
    int plane_count;
    int ac;     // 1 = range coder <-> 0 = golomb rice
Luca Barbato's avatar
Luca Barbato committed
87
    int ac_byte_count;      // number of bytes used for AC coding
Luca Barbato's avatar
Luca Barbato committed
88 89 90 91 92 93 94 95 96 97
    PlaneContext plane[MAX_PLANES];
    int16_t quant_table[MAX_CONTEXT_INPUTS][256];
    int16_t quant_tables[MAX_QUANT_TABLES][MAX_CONTEXT_INPUTS][256];
    int context_count[MAX_QUANT_TABLES];
    uint8_t state_transition[256];
    uint8_t (*initial_states[MAX_QUANT_TABLES])[32];
    int run_index;
    int colorspace;
    int16_t *sample_buffer;

Luca Barbato's avatar
Luca Barbato committed
98 99 100 101 102 103 104 105
    int ec;
    int slice_damaged;
    int key_frame_ok;

    int bits_per_raw_sample;
    int packed_at_lsb;

    int gob_count;
Luca Barbato's avatar
Luca Barbato committed
106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196
    int quant_table_count;

    struct FFV1Context *slice_context[MAX_SLICES];
    int slice_count;
    int num_v_slices;
    int num_h_slices;
    int slice_width;
    int slice_height;
    int slice_x;
    int slice_y;
} FFV1Context;

static av_always_inline int fold(int diff, int bits)
{
    if (bits == 8)
        diff = (int8_t)diff;
    else {
        diff +=  1 << (bits  - 1);
        diff &= (1 <<  bits) - 1;
        diff -=  1 << (bits  - 1);
    }

    return diff;
}

static inline int predict(int16_t *src, int16_t *last)
{
    const int LT = last[-1];
    const int T  = last[0];
    const int L  = src[-1];

    return mid_pred(L, L + T - LT, T);
}

static inline int get_context(PlaneContext *p, int16_t *src,
                              int16_t *last, int16_t *last2)
{
    const int LT = last[-1];
    const int T  = last[0];
    const int RT = last[1];
    const int L  = src[-1];

    if (p->quant_table[3][127]) {
        const int TT = last2[0];
        const int LL = src[-2];
        return p->quant_table[0][(L - LT) & 0xFF] +
               p->quant_table[1][(LT - T) & 0xFF] +
               p->quant_table[2][(T - RT) & 0xFF] +
               p->quant_table[3][(LL - L) & 0xFF] +
               p->quant_table[4][(TT - T) & 0xFF];
    } else
        return p->quant_table[0][(L - LT) & 0xFF] +
               p->quant_table[1][(LT - T) & 0xFF] +
               p->quant_table[2][(T - RT) & 0xFF];
}

static inline void update_vlc_state(VlcState *const state, const int v)
{
    int drift = state->drift;
    int count = state->count;
    state->error_sum += FFABS(v);
    drift            += v;

    if (count == 128) { // FIXME: variable
        count            >>= 1;
        drift            >>= 1;
        state->error_sum >>= 1;
    }
    count++;

    if (drift <= -count) {
        if (state->bias > -128)
            state->bias--;

        drift += count;
        if (drift <= -count)
            drift = -count + 1;
    } else if (drift > 0) {
        if (state->bias < 127)
            state->bias++;

        drift -= count;
        if (drift > 0)
            drift = 0;
    }

    state->drift = drift;
    state->count = count;
}

int ffv1_common_init(AVCodecContext *avctx);
Luca Barbato's avatar
Luca Barbato committed
197
int ffv1_init_slice_state(FFV1Context *f, FFV1Context *fs);
Luca Barbato's avatar
Luca Barbato committed
198 199
int ffv1_init_slice_contexts(FFV1Context *f);
int ffv1_allocate_initial_states(FFV1Context *f);
Luca Barbato's avatar
Luca Barbato committed
200
void ffv1_clear_slice_state(FFV1Context *f, FFV1Context *fs);
Luca Barbato's avatar
Luca Barbato committed
201 202 203
int ffv1_close(AVCodecContext *avctx);

#endif /* AVCODEC_FFV1_H */