libavcodec/dnxhd_parser: add parser and probe support raw 444 and dnxhr formats
[ffmpeg.git] / libavcodec / dnxhddata.h
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
29 /** Additional profile info flags */
30 #define DNXHD_INTERLACED   (1<<0)
31 #define DNXHD_MBAFF        (1<<1)
32 #define DNXHD_444          (1<<2)
33
34 /** Frame headers, extra 0x00 added to end for parser */
35 #define DNXHD_HEADER_INITIAL 0x000002800100
36 #define DNXHD_HEADER_444     0x000002800200
37 #define DNXHD_HEADER_HR1     0x000002800300
38 #define DNXHD_HEADER_HR2     0x0000038C0300
39
40 /** Indicate that a CIDEntry value must be read in the bitstream */
41 #define DNXHD_VARIABLE 0
42
43 typedef 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 frame_rates[5];
60 } CIDEntry;
61
62 extern const CIDEntry ff_dnxhd_cid_table[];
63
64 int ff_dnxhd_get_cid_table(int cid);
65 int ff_dnxhd_find_cid(AVCodecContext *avctx, int bit_depth);
66 void ff_dnxhd_print_profiles(AVCodecContext *avctx, int loglevel);
67
68 static av_always_inline uint64_t ff_dnxhd_check_header_prefix(uint64_t prefix)
69 {
70     if (prefix == DNXHD_HEADER_INITIAL ||
71         prefix == DNXHD_HEADER_444     ||
72         prefix == DNXHD_HEADER_HR1     ||
73         prefix == DNXHD_HEADER_HR2)
74         return prefix;
75     return 0;
76 }
77
78 int avpriv_dnxhd_get_frame_size(int cid);
79 int avpriv_dnxhd_get_interlaced(int cid);
80 uint64_t avpriv_dnxhd_parse_header_prefix(const uint8_t *buf);
81 #endif /* AVCODEC_DNXHDDATA_H */