error_resilience: decouple ER from MpegEncContext
[ffmpeg.git] / libavcodec / error_resilience.h
1 /*
2  *
3  * This file is part of Libav.
4  *
5  * Libav is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Lesser General Public
7  * License as published by the Free Software Foundation; either
8  * version 2.1 of the License, or (at your option) any later version.
9  *
10  * Libav is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * Lesser General Public License for more details.
14  *
15  * You should have received a copy of the GNU Lesser General Public
16  * License along with Libav; if not, write to the Free Software
17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18  */
19
20 #ifndef AVCODEC_ERROR_RESILIENCE_H
21 #define AVCODEC_ERROR_RESILIENCE_H
22
23 ///< current MB is the first after a resync marker
24 #define VP_START               1
25 #define ER_AC_ERROR            2
26 #define ER_DC_ERROR            4
27 #define ER_MV_ERROR            8
28 #define ER_AC_END              16
29 #define ER_DC_END              32
30 #define ER_MV_END              64
31
32 #define ER_MB_ERROR (ER_AC_ERROR|ER_DC_ERROR|ER_MV_ERROR)
33 #define ER_MB_END   (ER_AC_END|ER_DC_END|ER_MV_END)
34
35 typedef struct ERContext {
36     AVCodecContext *avctx;
37     DSPContext *dsp;
38
39     int *mb_index2xy;
40     int mb_num;
41     int mb_width, mb_height;
42     int mb_stride;
43     int b8_stride;
44
45     int error_count, error_occurred;
46     uint8_t *error_status_table;
47     uint8_t *er_temp_buffer;
48     int16_t *dc_val[3];
49     uint8_t *mbskip_table;
50     uint8_t *mbintra_table;
51     int mv[2][4][2];
52
53     struct Picture *cur_pic;
54     struct Picture *last_pic;
55     struct Picture *next_pic;
56
57     uint16_t pp_time;
58     uint16_t pb_time;
59     int quarter_sample;
60     int partitioned_frame;
61     int ref_count;
62
63     void (*decode_mb)(void *opaque, int ref, int mv_dir, int mv_type,
64                       int (*mv)[2][4][2],
65                       int mb_x, int mb_y, int mb_intra, int mb_skipped);
66     void *opaque;
67 } ERContext;
68
69 void ff_er_frame_start(ERContext *s);
70 void ff_er_frame_end(ERContext *s);
71 void ff_er_add_slice(ERContext *s, int startx, int starty, int endx, int endy,
72                      int status);
73
74 #endif /* AVCODEC_ERROR_RESILIENCE_H */