]> git.sesse.net Git - ffmpeg/blob - libavcodec/error_resilience.h
lavc: Add coded bitstream read/write support for AV1
[ffmpeg] / libavcodec / error_resilience.h
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg 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  * FFmpeg 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 FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18
19 #ifndef AVCODEC_ERROR_RESILIENCE_H
20 #define AVCODEC_ERROR_RESILIENCE_H
21
22 #include <stdint.h>
23 #include <stdatomic.h>
24
25 #include "avcodec.h"
26 #include "me_cmp.h"
27 #include "thread.h"
28
29 ///< current MB is the first after a resync marker
30 #define VP_START               1
31 #define ER_AC_ERROR            2
32 #define ER_DC_ERROR            4
33 #define ER_MV_ERROR            8
34 #define ER_AC_END              16
35 #define ER_DC_END              32
36 #define ER_MV_END              64
37
38 #define ER_MB_ERROR (ER_AC_ERROR|ER_DC_ERROR|ER_MV_ERROR)
39 #define ER_MB_END   (ER_AC_END|ER_DC_END|ER_MV_END)
40
41 typedef struct ERPicture {
42     AVFrame *f;
43     ThreadFrame *tf;
44
45     // it is the caller's responsibility to allocate these buffers
46     int16_t (*motion_val[2])[2];
47     int8_t *ref_index[2];
48
49     uint32_t *mb_type;
50     int field_picture;
51 } ERPicture;
52
53 typedef struct ERContext {
54     AVCodecContext *avctx;
55     MECmpContext mecc;
56     int mecc_inited;
57
58     int *mb_index2xy;
59     int mb_num;
60     int mb_width, mb_height;
61     ptrdiff_t mb_stride;
62     ptrdiff_t b8_stride;
63
64     atomic_int error_count;
65     int error_occurred;
66     uint8_t *error_status_table;
67     uint8_t *er_temp_buffer;
68     int16_t *dc_val[3];
69     uint8_t *mbskip_table;
70     uint8_t *mbintra_table;
71     int mv[2][4][2];
72
73     ERPicture cur_pic;
74     ERPicture last_pic;
75     ERPicture next_pic;
76
77     AVBufferRef *ref_index_buf[2];
78     AVBufferRef *motion_val_buf[2];
79
80     uint16_t pp_time;
81     uint16_t pb_time;
82     int quarter_sample;
83     int partitioned_frame;
84     int ref_count;
85
86     void (*decode_mb)(void *opaque, int ref, int mv_dir, int mv_type,
87                       int (*mv)[2][4][2],
88                       int mb_x, int mb_y, int mb_intra, int mb_skipped);
89     void *opaque;
90 } ERContext;
91
92 void ff_er_frame_start(ERContext *s);
93 void ff_er_frame_end(ERContext *s);
94 void ff_er_add_slice(ERContext *s, int startx, int starty, int endx, int endy,
95                      int status);
96
97 #endif /* AVCODEC_ERROR_RESILIENCE_H */