h264: factor out calculating the POC count into a separate file
[ffmpeg.git] / libavcodec / h264_parse.h
1 /*
2  * This file is part of Libav.
3  *
4  * Libav is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * Libav is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with Libav; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18
19 /**
20  * @file
21  * H.264 decoder/parser shared code
22  */
23
24 #ifndef AVCODEC_H264_PARSE_H
25 #define AVCODEC_H264_PARSE_H
26
27 #include "get_bits.h"
28
29 typedef struct H264PredWeightTable {
30     int use_weight;
31     int use_weight_chroma;
32     int luma_log2_weight_denom;
33     int chroma_log2_weight_denom;
34     int luma_weight_flag[2];    ///< 7.4.3.2 luma_weight_lX_flag
35     int chroma_weight_flag[2];  ///< 7.4.3.2 chroma_weight_lX_flag
36     // The following 2 can be changed to int8_t but that causes 10cpu cycles speedloss
37     int luma_weight[48][2][2];
38     int chroma_weight[48][2][2][2];
39     int implicit_weight[48][48][2];
40 } H264PredWeightTable;
41
42 typedef struct H264POCContext {
43     int poc_lsb;
44     int poc_msb;
45     int delta_poc_bottom;
46     int delta_poc[2];
47     int frame_num;
48     int prev_poc_msb;           ///< poc_msb of the last reference pic for POC type 0
49     int prev_poc_lsb;           ///< poc_lsb of the last reference pic for POC type 0
50     int frame_num_offset;       ///< for POC type 2
51     int prev_frame_num_offset;  ///< for POC type 2
52     int prev_frame_num;         ///< frame_num of the last pic for POC type 1/2
53 } H264POCContext;
54
55 struct SPS;
56 struct PPS;
57
58 int ff_h264_pred_weight_table(GetBitContext *gb, const struct SPS *sps,
59                               const int *ref_count, int slice_type_nos,
60                               H264PredWeightTable *pwt);
61
62 /**
63  * Check if the top & left blocks are available if needed & change the
64  * dc mode so it only uses the available blocks.
65  */
66 int ff_h264_check_intra4x4_pred_mode(int8_t *pred_mode_cache, void *logctx,
67                                      int top_samples_available, int left_samples_available);
68
69 /**
70  * Check if the top & left blocks are available if needed & change the
71  * dc mode so it only uses the available blocks.
72  */
73 int ff_h264_check_intra_pred_mode(void *logctx, int top_samples_available,
74                                   int left_samples_available,
75                                   int mode, int is_chroma);
76
77 int ff_h264_parse_ref_count(int *plist_count, int ref_count[2],
78                             GetBitContext *gb, const struct PPS *pps,
79                             int slice_type_nos, int picture_structure);
80
81 int ff_h264_init_poc(int pic_field_poc[2], int *pic_poc,
82                      const struct SPS *sps, H264POCContext *poc,
83                      int picture_structure, int nal_ref_idc);
84
85 #endif /* AVCODEC_H264_PARSE_H */