dnxhdenc.h 2.61 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 "mpegvideo.h"
#include "dnxhddata.h"
31

32
typedef struct RCCMPEntry {
33 34 35 36
    uint16_t mb;
    int value;
} RCCMPEntry;

37
typedef struct RCEntry {
38 39 40 41 42
    int ssd;
    int bits;
} RCEntry;

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

    int cid;
    const CIDEntry *cid_table;
    uint8_t *msip; ///< Macroblock Scan Indexes Payload
    uint32_t *slice_size;
51
    uint32_t *slice_offs;
52 53 54

    struct DNXHDEncContext *thread[MAX_THREADS];

55 56
    // 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.
57 58
    unsigned dct_y_offset;
    unsigned dct_uv_offset;
59 60
    unsigned block_width_l2;

61 62 63
    int interlaced;
    int cur_field;

64 65 66
    int nitris_compat;
    unsigned min_padding;

Diego Biurrun's avatar
Diego Biurrun committed
67
    DECLARE_ALIGNED(16, int16_t, blocks)[8][64];
68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91

    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;
    RCEntry   (*mb_rc)[8160];
92

93 94
    void (*get_pixels_8x4_sym)(int16_t * /* align 16 */,
                               const uint8_t *, ptrdiff_t);
95 96
} DNXHDEncContext;

97
void ff_dnxhdenc_init_x86(DNXHDEncContext *ctx);
98 99

#endif /* AVCODEC_DNXHDENC_H */