er: move relevant fields from Picture to ERPicture
[libav.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 #include <stdint.h>
24
25 #include "avcodec.h"
26 #include "dsputil.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's the caller responsability 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 DSPContext *dsp;
56
57 int *mb_index2xy;
58 int mb_num;
59 int mb_width, mb_height;
60 int mb_stride;
61 int b8_stride;
62
63 int error_count, error_occurred;
64 uint8_t *error_status_table;
65 uint8_t *er_temp_buffer;
66 int16_t *dc_val[3];
67 uint8_t *mbskip_table;
68 uint8_t *mbintra_table;
69 int mv[2][4][2];
70
71 ERPicture cur_pic;
72 ERPicture last_pic;
73 ERPicture next_pic;
74
75 uint16_t pp_time;
76 uint16_t pb_time;
77 int quarter_sample;
78 int partitioned_frame;
79 int ref_count;
80
81 void (*decode_mb)(void *opaque, int ref, int mv_dir, int mv_type,
82 int (*mv)[2][4][2],
83 int mb_x, int mb_y, int mb_intra, int mb_skipped);
84 void *opaque;
85 } ERContext;
86
87 void ff_er_frame_start(ERContext *s);
88 void ff_er_frame_end(ERContext *s);
89 void ff_er_add_slice(ERContext *s, int startx, int starty, int endx, int endy,
90 int status);
91
92 #endif /* AVCODEC_ERROR_RESILIENCE_H */