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/internal.h"
28 #include "libavutil/intreadwrite.h"
29
30 /** Additional profile info flags */
31 #define DNXHD_INTERLACED (1<<0)
32 #define DNXHD_MBAFF (1<<1)
33 #define DNXHD_444 (1<<2)
34
35 /** Frame headers, extra 0x00 added to end for parser */
36 #define DNXHD_HEADER_INITIAL 0x000002800100
37 #define DNXHD_HEADER_444 0x000002800200
38
39 /** Indicate that a CIDEntry value must be read in the bitstream */
40 #define DNXHD_VARIABLE 0
41
42 typedef struct CIDEntry {
43 int cid;
44 unsigned int width, height;
45 unsigned int frame_size;
46 unsigned int coding_unit_size;
47 uint16_t flags;
48 int index_bits;
49 int bit_depth;
50 int eob_index;
51 const uint8_t *luma_weight, *chroma_weight;
52 const uint8_t *dc_codes, *dc_bits;
53 const uint16_t *ac_codes;
54 const uint8_t *ac_bits, *ac_info;
55 const uint16_t *run_codes;
56 const uint8_t *run_bits, *run;
57 int bit_rates[5]; ///< Helper to choose variants, rounded to nearest 5Mb/s
58 AVRational packet_scale;
59 } CIDEntry;
60
61 const CIDEntry *ff_dnxhd_get_cid_table(int cid);
62 int ff_dnxhd_find_cid(AVCodecContext *avctx, int bit_depth);
63 void ff_dnxhd_print_profiles(AVCodecContext *avctx, int loglevel);
64
ff_dnxhd_check_header_prefix_hr(uint64_t prefix)65 static av_always_inline uint64_t ff_dnxhd_check_header_prefix_hr(uint64_t prefix)
66 {
67 uint64_t data_offset = prefix >> 16;
68 if ((prefix & 0xFFFF0000FFFFLL) == 0x0300 &&
69 data_offset >= 0x0280 && data_offset <= 0x2170 &&
70 (data_offset & 3) == 0)
71 return prefix;
72 return 0;
73 }
74
ff_dnxhd_check_header_prefix(uint64_t prefix)75 static av_always_inline uint64_t ff_dnxhd_check_header_prefix(uint64_t prefix)
76 {
77 if (prefix == DNXHD_HEADER_INITIAL ||
78 prefix == DNXHD_HEADER_444 ||
79 ff_dnxhd_check_header_prefix_hr(prefix))
80 return prefix;
81 return 0;
82 }
83
ff_dnxhd_parse_header_prefix(const uint8_t * buf)84 static av_always_inline uint64_t ff_dnxhd_parse_header_prefix(const uint8_t *buf)
85 {
86 uint64_t prefix = AV_RB32(buf);
87 prefix = (prefix << 16) | buf[4] << 8;
88 return ff_dnxhd_check_header_prefix(prefix);
89 }
90
91 int avpriv_dnxhd_get_frame_size(int cid);
92 int avpriv_dnxhd_get_hr_frame_size(int cid, int w, int h);
93 int avpriv_dnxhd_get_interlaced(int cid);
94
95 #endif /* AVCODEC_DNXHDDATA_H */
96