dnxhdenc.h 3.01 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
/*
 * VC3/DNxHD encoder structure definitions and prototypes
 * Copyright (c) 2007 Baptiste Coudurier <baptiste dot coudurier at smartjog dot com>
 *
 * VC-3 encoder funded by the British Broadcasting Corporation
 *
 * 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
 */

#ifndef AVCODEC_DNXHDENC_H
#define AVCODEC_DNXHDENC_H

#include <stdint.h>
28

29 30
#include "config.h"

31 32
#include "mpegvideo.h"
#include "dnxhddata.h"
33

34
typedef struct RCCMPEntry {
35 36 37 38
    uint16_t mb;
    int value;
} RCCMPEntry;

39
typedef struct RCEntry {
40 41 42 43 44
    int ssd;
    int bits;
} RCEntry;

typedef struct DNXHDEncContext {
45
    AVClass *class;
46
    BlockDSPContext bdsp;
47 48 49
    MpegEncContext m; ///< Used for quantization dsp functions

    int cid;
50
    int profile;
51 52
    int bit_depth;
    int is_444;
53 54 55
    const CIDEntry *cid_table;
    uint8_t *msip; ///< Macroblock Scan Indexes Payload
    uint32_t *slice_size;
56
    uint32_t *slice_offs;
57 58 59

    struct DNXHDEncContext *thread[MAX_THREADS];

60 61
    // Because our samples are either 8 or 16 bits for 8-bit and 10-bit
    // encoding respectively, these refer either to bytes or to two-byte words.
62 63
    unsigned dct_y_offset;
    unsigned dct_uv_offset;
64 65
    unsigned block_width_l2;

66 67 68 69
    int frame_size;
    int coding_unit_size;
    int data_offset;

70 71 72
    int interlaced;
    int cur_field;

73 74
    int nitris_compat;
    unsigned min_padding;
75
    int intra_quant_bias;
76

77
    DECLARE_ALIGNED(32, int16_t, blocks)[12][64];
78 79
    DECLARE_ALIGNED(16, uint8_t, edge_buf_y)[512]; // has to hold 16x16 uint16 when depth=10
    DECLARE_ALIGNED(16, uint8_t, edge_buf_uv)[2][512]; // has to hold 16x16 uint16_t when depth=10
80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102

    int      (*qmatrix_c)     [64];
    int      (*qmatrix_l)     [64];
    uint16_t (*qmatrix_l16)[2][64];
    uint16_t (*qmatrix_c16)[2][64];

    unsigned frame_bits;
    uint8_t *src[3];

    uint32_t *vlc_codes;
    uint8_t  *vlc_bits;
    uint16_t *run_codes;
    uint8_t  *run_bits;

    /** Rate control */
    unsigned slice_bits;
    unsigned qscale;
    unsigned lambda;

    uint16_t *mb_bits;
    uint8_t  *mb_qscale;

    RCCMPEntry *mb_cmp;
103
    RCCMPEntry *mb_cmp_tmp;
104
    RCEntry    *mb_rc;
105

106
    void (*get_pixels_8x4_sym)(int16_t *av_restrict /* align 16 */ block,
107
                               const uint8_t *pixels, ptrdiff_t line_size);
108 109
} DNXHDEncContext;

110
void ff_dnxhdenc_init_x86(DNXHDEncContext *ctx);
111 112

#endif /* AVCODEC_DNXHDENC_H */