xref: /third_party/ffmpeg/libavcodec/dnxhddata.h (revision cabdff1a)
1/*
2 * VC3/DNxHD decoder.
3 * Copyright (c) 2007 SmartJog S.A., Baptiste Coudurier <baptiste dot coudurier at smartjog dot com>
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22#ifndef AVCODEC_DNXHDDATA_H
23#define AVCODEC_DNXHDDATA_H
24
25#include <stdint.h>
26#include "avcodec.h"
27#include "libavutil/attributes.h"
28#include "libavutil/intreadwrite.h"
29#include "libavutil/rational.h"
30
31/** Additional profile info flags */
32#define DNXHD_INTERLACED   (1<<0)
33#define DNXHD_MBAFF        (1<<1)
34#define DNXHD_444          (1<<2)
35
36/** Frame headers, extra 0x00 added to end for parser */
37#define DNXHD_HEADER_INITIAL 0x000002800100
38#define DNXHD_HEADER_444     0x000002800200
39
40/** Indicate that a CIDEntry value must be read in the bitstream */
41#define DNXHD_VARIABLE 0
42
43typedef struct CIDEntry {
44    int cid;
45    unsigned int width, height;
46    unsigned int frame_size;
47    unsigned int coding_unit_size;
48    uint16_t flags;
49    int index_bits;
50    int bit_depth;
51    int eob_index;
52    const uint8_t *luma_weight, *chroma_weight;
53    const uint8_t *dc_codes, *dc_bits;
54    const uint16_t *ac_codes;
55    const uint8_t *ac_bits, *ac_info;
56    const uint16_t *run_codes;
57    const uint8_t *run_bits, *run;
58    int bit_rates[5]; ///< Helper to choose variants, rounded to nearest 5Mb/s
59    AVRational packet_scale;
60} CIDEntry;
61
62const CIDEntry *ff_dnxhd_get_cid_table(int cid);
63int ff_dnxhd_find_cid(AVCodecContext *avctx, int bit_depth);
64void ff_dnxhd_print_profiles(AVCodecContext *avctx, int loglevel);
65
66static av_always_inline uint64_t ff_dnxhd_check_header_prefix_hr(uint64_t prefix)
67{
68    uint64_t data_offset = prefix >> 16;
69    if ((prefix & 0xFFFF0000FFFFLL) == 0x0300 &&
70         data_offset >= 0x0280 && data_offset <= 0x2170 &&
71         (data_offset & 3) == 0)
72        return prefix;
73    return 0;
74}
75
76static av_always_inline uint64_t ff_dnxhd_check_header_prefix(uint64_t prefix)
77{
78    if (prefix == DNXHD_HEADER_INITIAL ||
79        prefix == DNXHD_HEADER_444     ||
80        ff_dnxhd_check_header_prefix_hr(prefix))
81        return prefix;
82    return 0;
83}
84
85static av_always_inline uint64_t ff_dnxhd_parse_header_prefix(const uint8_t *buf)
86{
87    uint64_t prefix = AV_RB32(buf);
88    prefix = (prefix << 16) | buf[4] << 8;
89    return ff_dnxhd_check_header_prefix(prefix);
90}
91
92int ff_dnxhd_get_frame_size(int cid);
93int ff_dnxhd_get_hr_frame_size(int cid, int w, int h);
94
95#endif /* AVCODEC_DNXHDDATA_H */
96