blob: 0c7881286c5e959125489635ea4d04d42424406b
1 | /* |
2 | * HEVC video decoder |
3 | * |
4 | * Copyright (C) 2012 - 2013 Guillaume Martres |
5 | * |
6 | * This file is part of FFmpeg. |
7 | * |
8 | * FFmpeg is free software; you can redistribute it and/or |
9 | * modify it under the terms of the GNU Lesser General Public |
10 | * License as published by the Free Software Foundation; either |
11 | * version 2.1 of the License, or (at your option) any later version. |
12 | * |
13 | * FFmpeg is distributed in the hope that it will be useful, |
14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
16 | * Lesser General Public License for more details. |
17 | * |
18 | * You should have received a copy of the GNU Lesser General Public |
19 | * License along with FFmpeg; if not, write to the Free Software |
20 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
21 | */ |
22 | |
23 | #ifndef AVCODEC_HEVCDEC_H |
24 | #define AVCODEC_HEVCDEC_H |
25 | |
26 | #include <stdatomic.h> |
27 | |
28 | #include "libavutil/buffer.h" |
29 | #include "libavutil/md5.h" |
30 | |
31 | #include "avcodec.h" |
32 | #include "bswapdsp.h" |
33 | #include "cabac.h" |
34 | #include "get_bits.h" |
35 | #include "hevcpred.h" |
36 | #include "h2645_parse.h" |
37 | #include "hevc.h" |
38 | #include "hevc_ps.h" |
39 | #include "hevcdsp.h" |
40 | #include "internal.h" |
41 | #include "thread.h" |
42 | #include "videodsp.h" |
43 | |
44 | #define MAX_NB_THREADS 16 |
45 | #define SHIFT_CTB_WPP 2 |
46 | |
47 | //TODO: check if this is really the maximum |
48 | #define MAX_TRANSFORM_DEPTH 5 |
49 | |
50 | #define MAX_TB_SIZE 32 |
51 | #define MAX_QP 51 |
52 | #define DEFAULT_INTRA_TC_OFFSET 2 |
53 | |
54 | #define HEVC_CONTEXTS 199 |
55 | |
56 | #define MRG_MAX_NUM_CANDS 5 |
57 | |
58 | #define L0 0 |
59 | #define L1 1 |
60 | |
61 | #define EPEL_EXTRA_BEFORE 1 |
62 | #define EPEL_EXTRA_AFTER 2 |
63 | #define EPEL_EXTRA 3 |
64 | #define QPEL_EXTRA_BEFORE 3 |
65 | #define QPEL_EXTRA_AFTER 4 |
66 | #define QPEL_EXTRA 7 |
67 | |
68 | #define EDGE_EMU_BUFFER_STRIDE 80 |
69 | |
70 | /** |
71 | * Value of the luma sample at position (x, y) in the 2D array tab. |
72 | */ |
73 | #define SAMPLE(tab, x, y) ((tab)[(y) * s->sps->width + (x)]) |
74 | #define SAMPLE_CTB(tab, x, y) ((tab)[(y) * min_cb_width + (x)]) |
75 | |
76 | #define IS_IDR(s) ((s)->nal_unit_type == HEVC_NAL_IDR_W_RADL || (s)->nal_unit_type == HEVC_NAL_IDR_N_LP) |
77 | #define IS_BLA(s) ((s)->nal_unit_type == HEVC_NAL_BLA_W_RADL || (s)->nal_unit_type == HEVC_NAL_BLA_W_LP || \ |
78 | (s)->nal_unit_type == HEVC_NAL_BLA_N_LP) |
79 | #define IS_IRAP(s) ((s)->nal_unit_type >= 16 && (s)->nal_unit_type <= 23) |
80 | |
81 | enum RPSType { |
82 | ST_CURR_BEF = 0, |
83 | ST_CURR_AFT, |
84 | ST_FOLL, |
85 | LT_CURR, |
86 | LT_FOLL, |
87 | NB_RPS_TYPE, |
88 | }; |
89 | |
90 | enum SyntaxElement { |
91 | SAO_MERGE_FLAG = 0, |
92 | SAO_TYPE_IDX, |
93 | SAO_EO_CLASS, |
94 | SAO_BAND_POSITION, |
95 | SAO_OFFSET_ABS, |
96 | SAO_OFFSET_SIGN, |
97 | END_OF_SLICE_FLAG, |
98 | SPLIT_CODING_UNIT_FLAG, |
99 | CU_TRANSQUANT_BYPASS_FLAG, |
100 | SKIP_FLAG, |
101 | CU_QP_DELTA, |
102 | PRED_MODE_FLAG, |
103 | PART_MODE, |
104 | PCM_FLAG, |
105 | PREV_INTRA_LUMA_PRED_FLAG, |
106 | MPM_IDX, |
107 | REM_INTRA_LUMA_PRED_MODE, |
108 | INTRA_CHROMA_PRED_MODE, |
109 | MERGE_FLAG, |
110 | MERGE_IDX, |
111 | INTER_PRED_IDC, |
112 | REF_IDX_L0, |
113 | REF_IDX_L1, |
114 | ABS_MVD_GREATER0_FLAG, |
115 | ABS_MVD_GREATER1_FLAG, |
116 | ABS_MVD_MINUS2, |
117 | MVD_SIGN_FLAG, |
118 | MVP_LX_FLAG, |
119 | NO_RESIDUAL_DATA_FLAG, |
120 | SPLIT_TRANSFORM_FLAG, |
121 | CBF_LUMA, |
122 | CBF_CB_CR, |
123 | TRANSFORM_SKIP_FLAG, |
124 | EXPLICIT_RDPCM_FLAG, |
125 | EXPLICIT_RDPCM_DIR_FLAG, |
126 | LAST_SIGNIFICANT_COEFF_X_PREFIX, |
127 | LAST_SIGNIFICANT_COEFF_Y_PREFIX, |
128 | LAST_SIGNIFICANT_COEFF_X_SUFFIX, |
129 | LAST_SIGNIFICANT_COEFF_Y_SUFFIX, |
130 | SIGNIFICANT_COEFF_GROUP_FLAG, |
131 | SIGNIFICANT_COEFF_FLAG, |
132 | COEFF_ABS_LEVEL_GREATER1_FLAG, |
133 | COEFF_ABS_LEVEL_GREATER2_FLAG, |
134 | COEFF_ABS_LEVEL_REMAINING, |
135 | COEFF_SIGN_FLAG, |
136 | LOG2_RES_SCALE_ABS, |
137 | RES_SCALE_SIGN_FLAG, |
138 | CU_CHROMA_QP_OFFSET_FLAG, |
139 | CU_CHROMA_QP_OFFSET_IDX, |
140 | }; |
141 | |
142 | enum PartMode { |
143 | PART_2Nx2N = 0, |
144 | PART_2NxN = 1, |
145 | PART_Nx2N = 2, |
146 | PART_NxN = 3, |
147 | PART_2NxnU = 4, |
148 | PART_2NxnD = 5, |
149 | PART_nLx2N = 6, |
150 | PART_nRx2N = 7, |
151 | }; |
152 | |
153 | enum PredMode { |
154 | MODE_INTER = 0, |
155 | MODE_INTRA, |
156 | MODE_SKIP, |
157 | }; |
158 | |
159 | enum InterPredIdc { |
160 | PRED_L0 = 0, |
161 | PRED_L1, |
162 | PRED_BI, |
163 | }; |
164 | |
165 | enum PredFlag { |
166 | PF_INTRA = 0, |
167 | PF_L0, |
168 | PF_L1, |
169 | PF_BI, |
170 | }; |
171 | |
172 | enum IntraPredMode { |
173 | INTRA_PLANAR = 0, |
174 | INTRA_DC, |
175 | INTRA_ANGULAR_2, |
176 | INTRA_ANGULAR_3, |
177 | INTRA_ANGULAR_4, |
178 | INTRA_ANGULAR_5, |
179 | INTRA_ANGULAR_6, |
180 | INTRA_ANGULAR_7, |
181 | INTRA_ANGULAR_8, |
182 | INTRA_ANGULAR_9, |
183 | INTRA_ANGULAR_10, |
184 | INTRA_ANGULAR_11, |
185 | INTRA_ANGULAR_12, |
186 | INTRA_ANGULAR_13, |
187 | INTRA_ANGULAR_14, |
188 | INTRA_ANGULAR_15, |
189 | INTRA_ANGULAR_16, |
190 | INTRA_ANGULAR_17, |
191 | INTRA_ANGULAR_18, |
192 | INTRA_ANGULAR_19, |
193 | INTRA_ANGULAR_20, |
194 | INTRA_ANGULAR_21, |
195 | INTRA_ANGULAR_22, |
196 | INTRA_ANGULAR_23, |
197 | INTRA_ANGULAR_24, |
198 | INTRA_ANGULAR_25, |
199 | INTRA_ANGULAR_26, |
200 | INTRA_ANGULAR_27, |
201 | INTRA_ANGULAR_28, |
202 | INTRA_ANGULAR_29, |
203 | INTRA_ANGULAR_30, |
204 | INTRA_ANGULAR_31, |
205 | INTRA_ANGULAR_32, |
206 | INTRA_ANGULAR_33, |
207 | INTRA_ANGULAR_34, |
208 | }; |
209 | |
210 | enum SAOType { |
211 | SAO_NOT_APPLIED = 0, |
212 | SAO_BAND, |
213 | SAO_EDGE, |
214 | SAO_APPLIED |
215 | }; |
216 | |
217 | enum SAOEOClass { |
218 | SAO_EO_HORIZ = 0, |
219 | SAO_EO_VERT, |
220 | SAO_EO_135D, |
221 | SAO_EO_45D, |
222 | }; |
223 | |
224 | enum ScanType { |
225 | SCAN_DIAG = 0, |
226 | SCAN_HORIZ, |
227 | SCAN_VERT, |
228 | }; |
229 | |
230 | typedef struct LongTermRPS { |
231 | int poc[32]; |
232 | uint8_t used[32]; |
233 | uint8_t nb_refs; |
234 | } LongTermRPS; |
235 | |
236 | typedef struct RefPicList { |
237 | struct HEVCFrame *ref[HEVC_MAX_REFS]; |
238 | int list[HEVC_MAX_REFS]; |
239 | int isLongTerm[HEVC_MAX_REFS]; |
240 | int nb_refs; |
241 | } RefPicList; |
242 | |
243 | typedef struct RefPicListTab { |
244 | RefPicList refPicList[2]; |
245 | } RefPicListTab; |
246 | |
247 | typedef struct SliceHeader { |
248 | unsigned int pps_id; |
249 | |
250 | ///< address (in raster order) of the first block in the current slice segment |
251 | unsigned int slice_segment_addr; |
252 | ///< address (in raster order) of the first block in the current slice |
253 | unsigned int slice_addr; |
254 | |
255 | enum HEVCSliceType slice_type; |
256 | |
257 | int pic_order_cnt_lsb; |
258 | |
259 | uint8_t first_slice_in_pic_flag; |
260 | uint8_t dependent_slice_segment_flag; |
261 | uint8_t pic_output_flag; |
262 | uint8_t colour_plane_id; |
263 | |
264 | ///< RPS coded in the slice header itself is stored here |
265 | int short_term_ref_pic_set_sps_flag; |
266 | int short_term_ref_pic_set_size; |
267 | ShortTermRPS slice_rps; |
268 | const ShortTermRPS *short_term_rps; |
269 | int long_term_ref_pic_set_size; |
270 | LongTermRPS long_term_rps; |
271 | unsigned int list_entry_lx[2][32]; |
272 | |
273 | uint8_t rpl_modification_flag[2]; |
274 | uint8_t no_output_of_prior_pics_flag; |
275 | uint8_t slice_temporal_mvp_enabled_flag; |
276 | |
277 | unsigned int nb_refs[2]; |
278 | |
279 | uint8_t slice_sample_adaptive_offset_flag[3]; |
280 | uint8_t mvd_l1_zero_flag; |
281 | |
282 | uint8_t cabac_init_flag; |
283 | uint8_t disable_deblocking_filter_flag; ///< slice_header_disable_deblocking_filter_flag |
284 | uint8_t slice_loop_filter_across_slices_enabled_flag; |
285 | uint8_t collocated_list; |
286 | |
287 | unsigned int collocated_ref_idx; |
288 | |
289 | int slice_qp_delta; |
290 | int slice_cb_qp_offset; |
291 | int slice_cr_qp_offset; |
292 | |
293 | uint8_t cu_chroma_qp_offset_enabled_flag; |
294 | |
295 | int beta_offset; ///< beta_offset_div2 * 2 |
296 | int tc_offset; ///< tc_offset_div2 * 2 |
297 | |
298 | unsigned int max_num_merge_cand; ///< 5 - 5_minus_max_num_merge_cand |
299 | |
300 | unsigned *entry_point_offset; |
301 | int * offset; |
302 | int * size; |
303 | int num_entry_point_offsets; |
304 | |
305 | int8_t slice_qp; |
306 | |
307 | uint8_t luma_log2_weight_denom; |
308 | int16_t chroma_log2_weight_denom; |
309 | |
310 | int16_t luma_weight_l0[16]; |
311 | int16_t chroma_weight_l0[16][2]; |
312 | int16_t chroma_weight_l1[16][2]; |
313 | int16_t luma_weight_l1[16]; |
314 | |
315 | int16_t luma_offset_l0[16]; |
316 | int16_t chroma_offset_l0[16][2]; |
317 | |
318 | int16_t luma_offset_l1[16]; |
319 | int16_t chroma_offset_l1[16][2]; |
320 | |
321 | int slice_ctb_addr_rs; |
322 | } SliceHeader; |
323 | |
324 | typedef struct CodingUnit { |
325 | int x; |
326 | int y; |
327 | |
328 | enum PredMode pred_mode; ///< PredMode |
329 | enum PartMode part_mode; ///< PartMode |
330 | |
331 | // Inferred parameters |
332 | uint8_t intra_split_flag; ///< IntraSplitFlag |
333 | uint8_t max_trafo_depth; ///< MaxTrafoDepth |
334 | uint8_t cu_transquant_bypass_flag; |
335 | } CodingUnit; |
336 | |
337 | typedef struct Mv { |
338 | int16_t x; ///< horizontal component of motion vector |
339 | int16_t y; ///< vertical component of motion vector |
340 | } Mv; |
341 | |
342 | typedef struct MvField { |
343 | DECLARE_ALIGNED(4, Mv, mv)[2]; |
344 | int8_t ref_idx[2]; |
345 | int8_t pred_flag; |
346 | } MvField; |
347 | |
348 | typedef struct NeighbourAvailable { |
349 | int cand_bottom_left; |
350 | int cand_left; |
351 | int cand_up; |
352 | int cand_up_left; |
353 | int cand_up_right; |
354 | int cand_up_right_sap; |
355 | } NeighbourAvailable; |
356 | |
357 | typedef struct PredictionUnit { |
358 | int mpm_idx; |
359 | int rem_intra_luma_pred_mode; |
360 | uint8_t intra_pred_mode[4]; |
361 | Mv mvd; |
362 | uint8_t merge_flag; |
363 | uint8_t intra_pred_mode_c[4]; |
364 | uint8_t chroma_mode_c[4]; |
365 | } PredictionUnit; |
366 | |
367 | typedef struct TransformUnit { |
368 | int cu_qp_delta; |
369 | |
370 | int res_scale_val; |
371 | |
372 | // Inferred parameters; |
373 | int intra_pred_mode; |
374 | int intra_pred_mode_c; |
375 | int chroma_mode_c; |
376 | uint8_t is_cu_qp_delta_coded; |
377 | uint8_t is_cu_chroma_qp_offset_coded; |
378 | int8_t cu_qp_offset_cb; |
379 | int8_t cu_qp_offset_cr; |
380 | uint8_t cross_pf; |
381 | } TransformUnit; |
382 | |
383 | typedef struct DBParams { |
384 | int beta_offset; |
385 | int tc_offset; |
386 | } DBParams; |
387 | |
388 | #define HEVC_FRAME_FLAG_OUTPUT (1 << 0) |
389 | #define HEVC_FRAME_FLAG_SHORT_REF (1 << 1) |
390 | #define HEVC_FRAME_FLAG_LONG_REF (1 << 2) |
391 | #define HEVC_FRAME_FLAG_BUMPING (1 << 3) |
392 | |
393 | typedef struct HEVCFrame { |
394 | AVFrame *frame; |
395 | ThreadFrame tf; |
396 | MvField *tab_mvf; |
397 | RefPicList *refPicList; |
398 | RefPicListTab **rpl_tab; |
399 | int ctb_count; |
400 | int poc; |
401 | struct HEVCFrame *collocated_ref; |
402 | |
403 | HEVCWindow window; |
404 | |
405 | AVBufferRef *tab_mvf_buf; |
406 | AVBufferRef *rpl_tab_buf; |
407 | AVBufferRef *rpl_buf; |
408 | |
409 | AVBufferRef *hwaccel_priv_buf; |
410 | void *hwaccel_picture_private; |
411 | |
412 | /** |
413 | * A sequence counter, so that old frames are output first |
414 | * after a POC reset |
415 | */ |
416 | uint16_t sequence; |
417 | |
418 | /** |
419 | * A combination of HEVC_FRAME_FLAG_* |
420 | */ |
421 | uint8_t flags; |
422 | } HEVCFrame; |
423 | |
424 | typedef struct HEVCLocalContext { |
425 | uint8_t cabac_state[HEVC_CONTEXTS]; |
426 | |
427 | uint8_t stat_coeff[4]; |
428 | |
429 | uint8_t first_qp_group; |
430 | |
431 | GetBitContext gb; |
432 | CABACContext cc; |
433 | |
434 | int8_t qp_y; |
435 | int8_t curr_qp_y; |
436 | |
437 | int qPy_pred; |
438 | |
439 | TransformUnit tu; |
440 | |
441 | uint8_t ctb_left_flag; |
442 | uint8_t ctb_up_flag; |
443 | uint8_t ctb_up_right_flag; |
444 | uint8_t ctb_up_left_flag; |
445 | int end_of_tiles_x; |
446 | int end_of_tiles_y; |
447 | /* +7 is for subpixel interpolation, *2 for high bit depths */ |
448 | DECLARE_ALIGNED(32, uint8_t, edge_emu_buffer)[(MAX_PB_SIZE + 7) * EDGE_EMU_BUFFER_STRIDE * 2]; |
449 | /* The extended size between the new edge emu buffer is abused by SAO */ |
450 | DECLARE_ALIGNED(32, uint8_t, edge_emu_buffer2)[(MAX_PB_SIZE + 7) * EDGE_EMU_BUFFER_STRIDE * 2]; |
451 | DECLARE_ALIGNED(32, int16_t, tmp [MAX_PB_SIZE * MAX_PB_SIZE]); |
452 | |
453 | int ct_depth; |
454 | CodingUnit cu; |
455 | PredictionUnit pu; |
456 | NeighbourAvailable na; |
457 | |
458 | #define BOUNDARY_LEFT_SLICE (1 << 0) |
459 | #define BOUNDARY_LEFT_TILE (1 << 1) |
460 | #define BOUNDARY_UPPER_SLICE (1 << 2) |
461 | #define BOUNDARY_UPPER_TILE (1 << 3) |
462 | /* properties of the boundary of the current CTB for the purposes |
463 | * of the deblocking filter */ |
464 | int boundary_flags; |
465 | } HEVCLocalContext; |
466 | |
467 | typedef struct HEVCContext { |
468 | const AVClass *c; // needed by private avoptions |
469 | AVCodecContext *avctx; |
470 | |
471 | struct HEVCContext *sList[MAX_NB_THREADS]; |
472 | |
473 | HEVCLocalContext *HEVClcList[MAX_NB_THREADS]; |
474 | HEVCLocalContext *HEVClc; |
475 | |
476 | uint8_t threads_type; |
477 | uint8_t threads_number; |
478 | |
479 | int width; |
480 | int height; |
481 | |
482 | uint8_t *cabac_state; |
483 | |
484 | /** 1 if the independent slice segment header was successfully parsed */ |
485 | uint8_t slice_initialized; |
486 | |
487 | AVFrame *frame; |
488 | AVFrame *output_frame; |
489 | uint8_t *sao_pixel_buffer_h[3]; |
490 | uint8_t *sao_pixel_buffer_v[3]; |
491 | |
492 | HEVCParamSets ps; |
493 | |
494 | AVBufferPool *tab_mvf_pool; |
495 | AVBufferPool *rpl_tab_pool; |
496 | |
497 | ///< candidate references for the current frame |
498 | RefPicList rps[5]; |
499 | |
500 | SliceHeader sh; |
501 | SAOParams *sao; |
502 | DBParams *deblock; |
503 | enum HEVCNALUnitType nal_unit_type; |
504 | int temporal_id; ///< temporal_id_plus1 - 1 |
505 | HEVCFrame *ref; |
506 | HEVCFrame DPB[32]; |
507 | int poc; |
508 | int pocTid0; |
509 | int slice_idx; ///< number of the slice being currently decoded |
510 | int eos; ///< current packet contains an EOS/EOB NAL |
511 | int last_eos; ///< last packet contains an EOS/EOB NAL |
512 | int max_ra; |
513 | int bs_width; |
514 | int bs_height; |
515 | |
516 | int is_decoded; |
517 | int no_rasl_output_flag; |
518 | |
519 | HEVCPredContext hpc; |
520 | HEVCDSPContext hevcdsp; |
521 | VideoDSPContext vdsp; |
522 | BswapDSPContext bdsp; |
523 | int8_t *qp_y_tab; |
524 | uint8_t *horizontal_bs; |
525 | uint8_t *vertical_bs; |
526 | |
527 | int32_t *tab_slice_address; |
528 | |
529 | // CU |
530 | uint8_t *skip_flag; |
531 | uint8_t *tab_ct_depth; |
532 | // PU |
533 | uint8_t *tab_ipm; |
534 | |
535 | uint8_t *cbf_luma; // cbf_luma of colocated TU |
536 | uint8_t *is_pcm; |
537 | |
538 | // CTB-level flags affecting loop filter operation |
539 | uint8_t *filter_slice_edges; |
540 | |
541 | /** used on BE to byteswap the lines for checksumming */ |
542 | uint8_t *checksum_buf; |
543 | int checksum_buf_size; |
544 | |
545 | /** |
546 | * Sequence counters for decoded and output frames, so that old |
547 | * frames are output first after a POC reset |
548 | */ |
549 | uint16_t seq_decode; |
550 | uint16_t seq_output; |
551 | |
552 | int enable_parallel_tiles; |
553 | atomic_int wpp_err; |
554 | |
555 | const uint8_t *data; |
556 | |
557 | H2645Packet pkt; |
558 | // type of the first VCL NAL of the current frame |
559 | enum HEVCNALUnitType first_nal_type; |
560 | |
561 | // for checking the frame checksums |
562 | struct AVMD5 *md5_ctx; |
563 | uint8_t md5[3][16]; |
564 | uint8_t is_md5; |
565 | |
566 | uint8_t context_initialized; |
567 | uint8_t is_nalff; ///< this flag is != 0 if bitstream is encapsulated |
568 | ///< as a format defined in 14496-15 |
569 | int apply_defdispwin; |
570 | |
571 | int active_seq_parameter_set_id; |
572 | |
573 | int nal_length_size; ///< Number of bytes used for nal length (1, 2 or 4) |
574 | int nuh_layer_id; |
575 | |
576 | /** frame packing arrangement variables */ |
577 | int sei_frame_packing_present; |
578 | int frame_packing_arrangement_type; |
579 | int content_interpretation_type; |
580 | int quincunx_subsampling; |
581 | |
582 | /** display orientation */ |
583 | int sei_display_orientation_present; |
584 | int sei_anticlockwise_rotation; |
585 | int sei_hflip, sei_vflip; |
586 | |
587 | int picture_struct; |
588 | |
589 | uint8_t* a53_caption; |
590 | int a53_caption_size; |
591 | |
592 | /** mastering display */ |
593 | int sei_mastering_display_info_present; |
594 | uint16_t display_primaries[3][2]; |
595 | uint16_t white_point[2]; |
596 | uint32_t max_mastering_luminance; |
597 | uint32_t min_mastering_luminance; |
598 | |
599 | } HEVCContext; |
600 | |
601 | int ff_hevc_decode_nal_sei(HEVCContext *s); |
602 | |
603 | /** |
604 | * Mark all frames in DPB as unused for reference. |
605 | */ |
606 | void ff_hevc_clear_refs(HEVCContext *s); |
607 | |
608 | /** |
609 | * Drop all frames currently in DPB. |
610 | */ |
611 | void ff_hevc_flush_dpb(HEVCContext *s); |
612 | |
613 | /** |
614 | * Compute POC of the current frame and return it. |
615 | */ |
616 | int ff_hevc_compute_poc(HEVCContext *s, int poc_lsb); |
617 | |
618 | RefPicList *ff_hevc_get_ref_list(HEVCContext *s, HEVCFrame *frame, |
619 | int x0, int y0); |
620 | |
621 | /** |
622 | * Construct the reference picture sets for the current frame. |
623 | */ |
624 | int ff_hevc_frame_rps(HEVCContext *s); |
625 | |
626 | /** |
627 | * Construct the reference picture list(s) for the current slice. |
628 | */ |
629 | int ff_hevc_slice_rpl(HEVCContext *s); |
630 | |
631 | void ff_hevc_save_states(HEVCContext *s, int ctb_addr_ts); |
632 | void ff_hevc_cabac_init(HEVCContext *s, int ctb_addr_ts); |
633 | int ff_hevc_sao_merge_flag_decode(HEVCContext *s); |
634 | int ff_hevc_sao_type_idx_decode(HEVCContext *s); |
635 | int ff_hevc_sao_band_position_decode(HEVCContext *s); |
636 | int ff_hevc_sao_offset_abs_decode(HEVCContext *s); |
637 | int ff_hevc_sao_offset_sign_decode(HEVCContext *s); |
638 | int ff_hevc_sao_eo_class_decode(HEVCContext *s); |
639 | int ff_hevc_end_of_slice_flag_decode(HEVCContext *s); |
640 | int ff_hevc_cu_transquant_bypass_flag_decode(HEVCContext *s); |
641 | int ff_hevc_skip_flag_decode(HEVCContext *s, int x0, int y0, |
642 | int x_cb, int y_cb); |
643 | int ff_hevc_pred_mode_decode(HEVCContext *s); |
644 | int ff_hevc_split_coding_unit_flag_decode(HEVCContext *s, int ct_depth, |
645 | int x0, int y0); |
646 | int ff_hevc_part_mode_decode(HEVCContext *s, int log2_cb_size); |
647 | int ff_hevc_pcm_flag_decode(HEVCContext *s); |
648 | int ff_hevc_prev_intra_luma_pred_flag_decode(HEVCContext *s); |
649 | int ff_hevc_mpm_idx_decode(HEVCContext *s); |
650 | int ff_hevc_rem_intra_luma_pred_mode_decode(HEVCContext *s); |
651 | int ff_hevc_intra_chroma_pred_mode_decode(HEVCContext *s); |
652 | int ff_hevc_merge_idx_decode(HEVCContext *s); |
653 | int ff_hevc_merge_flag_decode(HEVCContext *s); |
654 | int ff_hevc_inter_pred_idc_decode(HEVCContext *s, int nPbW, int nPbH); |
655 | int ff_hevc_ref_idx_lx_decode(HEVCContext *s, int num_ref_idx_lx); |
656 | int ff_hevc_mvp_lx_flag_decode(HEVCContext *s); |
657 | int ff_hevc_no_residual_syntax_flag_decode(HEVCContext *s); |
658 | int ff_hevc_split_transform_flag_decode(HEVCContext *s, int log2_trafo_size); |
659 | int ff_hevc_cbf_cb_cr_decode(HEVCContext *s, int trafo_depth); |
660 | int ff_hevc_cbf_luma_decode(HEVCContext *s, int trafo_depth); |
661 | int ff_hevc_log2_res_scale_abs(HEVCContext *s, int idx); |
662 | int ff_hevc_res_scale_sign_flag(HEVCContext *s, int idx); |
663 | |
664 | /** |
665 | * Get the number of candidate references for the current frame. |
666 | */ |
667 | int ff_hevc_frame_nb_refs(HEVCContext *s); |
668 | |
669 | int ff_hevc_set_new_ref(HEVCContext *s, AVFrame **frame, int poc); |
670 | |
671 | /** |
672 | * Find next frame in output order and put a reference to it in frame. |
673 | * @return 1 if a frame was output, 0 otherwise |
674 | */ |
675 | int ff_hevc_output_frame(HEVCContext *s, AVFrame *frame, int flush); |
676 | |
677 | void ff_hevc_bump_frame(HEVCContext *s); |
678 | |
679 | void ff_hevc_unref_frame(HEVCContext *s, HEVCFrame *frame, int flags); |
680 | |
681 | void ff_hevc_set_neighbour_available(HEVCContext *s, int x0, int y0, |
682 | int nPbW, int nPbH); |
683 | void ff_hevc_luma_mv_merge_mode(HEVCContext *s, int x0, int y0, |
684 | int nPbW, int nPbH, int log2_cb_size, |
685 | int part_idx, int merge_idx, MvField *mv); |
686 | void ff_hevc_luma_mv_mvp_mode(HEVCContext *s, int x0, int y0, |
687 | int nPbW, int nPbH, int log2_cb_size, |
688 | int part_idx, int merge_idx, |
689 | MvField *mv, int mvp_lx_flag, int LX); |
690 | void ff_hevc_set_qPy(HEVCContext *s, int xBase, int yBase, |
691 | int log2_cb_size); |
692 | void ff_hevc_deblocking_boundary_strengths(HEVCContext *s, int x0, int y0, |
693 | int log2_trafo_size); |
694 | int ff_hevc_cu_qp_delta_sign_flag(HEVCContext *s); |
695 | int ff_hevc_cu_qp_delta_abs(HEVCContext *s); |
696 | int ff_hevc_cu_chroma_qp_offset_flag(HEVCContext *s); |
697 | int ff_hevc_cu_chroma_qp_offset_idx(HEVCContext *s); |
698 | void ff_hevc_hls_filter(HEVCContext *s, int x, int y, int ctb_size); |
699 | void ff_hevc_hls_filters(HEVCContext *s, int x_ctb, int y_ctb, int ctb_size); |
700 | void ff_hevc_hls_residual_coding(HEVCContext *s, int x0, int y0, |
701 | int log2_trafo_size, enum ScanType scan_idx, |
702 | int c_idx); |
703 | |
704 | void ff_hevc_hls_mvd_coding(HEVCContext *s, int x0, int y0, int log2_cb_size); |
705 | |
706 | /** |
707 | * Reset SEI values that are stored on the Context. |
708 | * e.g. Caption data that was extracted during NAL |
709 | * parsing. |
710 | * |
711 | * @param s HEVCContext. |
712 | */ |
713 | void ff_hevc_reset_sei(HEVCContext *s); |
714 | |
715 | extern const uint8_t ff_hevc_qpel_extra_before[4]; |
716 | extern const uint8_t ff_hevc_qpel_extra_after[4]; |
717 | extern const uint8_t ff_hevc_qpel_extra[4]; |
718 | |
719 | #endif /* AVCODEC_HEVCDEC_H */ |
720 |