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 "threadframe.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
41typedef 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
53typedef struct ERContext {
54    AVCodecContext *avctx;
55
56    me_cmp_func sad;
57    int mecc_inited;
58
59    int *mb_index2xy;
60    int mb_num;
61    int mb_width, mb_height;
62    ptrdiff_t mb_stride;
63    ptrdiff_t b8_stride;
64
65    atomic_int error_count;
66    int error_occurred;
67    uint8_t *error_status_table;
68    uint8_t *er_temp_buffer;
69    int16_t *dc_val[3];
70    uint8_t *mbskip_table;
71    uint8_t *mbintra_table;
72    int mv[2][4][2];
73
74    ERPicture cur_pic;
75    ERPicture last_pic;
76    ERPicture next_pic;
77
78    AVBufferRef *ref_index_buf[2];
79    AVBufferRef *motion_val_buf[2];
80
81    uint16_t pp_time;
82    uint16_t pb_time;
83    int quarter_sample;
84    int partitioned_frame;
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
92void ff_er_frame_start(ERContext *s);
93void ff_er_frame_end(ERContext *s);
94void ff_er_add_slice(ERContext *s, int startx, int starty, int endx, int endy,
95                     int status);
96
97#endif /* AVCODEC_ERROR_RESILIENCE_H */
98