FFmpeg
error_resilience.h
Go to the documentation of this file.
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 
28 ///< current MB is the first after a resync marker
29 #define VP_START 1
30 #define ER_AC_ERROR 2
31 #define ER_DC_ERROR 4
32 #define ER_MV_ERROR 8
33 #define ER_AC_END 16
34 #define ER_DC_END 32
35 #define ER_MV_END 64
36 
37 #define ER_MB_ERROR (ER_AC_ERROR|ER_DC_ERROR|ER_MV_ERROR)
38 #define ER_MB_END (ER_AC_END|ER_DC_END|ER_MV_END)
39 
40 typedef struct ERPicture {
42  const struct ThreadFrame *tf;
43 
44  // it is the caller's responsibility to allocate these buffers
45  int16_t (*motion_val[2])[2];
46  int8_t *ref_index[2];
47 
48  uint32_t *mb_type;
50 } ERPicture;
51 
52 typedef struct ERContext {
54 
57 
59  int mb_num;
61  ptrdiff_t mb_stride;
62  ptrdiff_t b8_stride;
63 
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 
76 
77  int8_t *ref_index[2];
78  int16_t (*motion_val_base[2])[2];
79 
80  uint16_t pp_time;
81  uint16_t pb_time;
84 
85  void (*decode_mb)(void *opaque, int ref, int mv_dir, int mv_type,
86  int (*mv)[2][4][2],
87  int mb_x, int mb_y, int mb_intra, int mb_skipped);
88  void *opaque;
89 } ERContext;
90 
92 
93 /**
94  * Indicate that a frame has finished decoding and perform error concealment
95  * in case it has been enabled and is necessary and supported.
96  *
97  * @param s ERContext in use
98  * @param decode_error_flags pointer where updated decode_error_flags are written
99  * if supplied; if not, the new flags are directly
100  * applied to the AVFrame whose errors are concealed
101  */
102 void ff_er_frame_end(ERContext *s, int *decode_error_flags);
103 void ff_er_add_slice(ERContext *s, int startx, int starty, int endx, int endy,
104  int status);
105 
106 #endif /* AVCODEC_ERROR_RESILIENCE_H */
ERContext::pb_time
uint16_t pb_time
Definition: error_resilience.h:81
ERContext::motion_val_base
int16_t(*[2] motion_val_base)[2]
Definition: error_resilience.h:78
ERPicture::ref_index
int8_t * ref_index[2]
Definition: error_resilience.h:46
ERContext::mbintra_table
uint8_t * mbintra_table
Definition: error_resilience.h:70
ERContext::mb_index2xy
int * mb_index2xy
Definition: error_resilience.h:58
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:344
ERContext::sad
me_cmp_func sad
Definition: error_resilience.h:55
atomic_int
intptr_t atomic_int
Definition: stdatomic.h:55
ERContext
Definition: error_resilience.h:52
ERPicture
Definition: error_resilience.h:40
ERContext::partitioned_frame
int partitioned_frame
Definition: error_resilience.h:83
ERPicture::motion_val
int16_t(*[2] motion_val)[2]
Definition: error_resilience.h:45
ERContext::mb_num
int mb_num
Definition: error_resilience.h:59
ERContext::avctx
AVCodecContext * avctx
Definition: error_resilience.h:53
ERContext::mv
int mv[2][4][2]
Definition: error_resilience.h:71
ERPicture::field_picture
int field_picture
Definition: error_resilience.h:49
s
#define s(width, name)
Definition: cbs_vp9.c:198
ERContext::dc_val
int16_t * dc_val[3]
Definition: error_resilience.h:68
ERPicture::f
AVFrame * f
Definition: error_resilience.h:41
ERPicture::mb_type
uint32_t * mb_type
Definition: error_resilience.h:48
ERPicture::tf
const struct ThreadFrame * tf
Definition: error_resilience.h:42
ERContext::error_occurred
int error_occurred
Definition: error_resilience.h:65
ERContext::mbskip_table
uint8_t * mbskip_table
Definition: error_resilience.h:69
ERContext::opaque
void * opaque
Definition: error_resilience.h:88
ERContext::error_count
atomic_int error_count
Definition: error_resilience.h:64
ff_er_add_slice
void ff_er_add_slice(ERContext *s, int startx, int starty, int endx, int endy, int status)
Add a slice.
Definition: error_resilience.c:822
ERContext::b8_stride
ptrdiff_t b8_stride
Definition: error_resilience.h:62
ERContext::mb_stride
ptrdiff_t mb_stride
Definition: error_resilience.h:61
ERContext::decode_mb
void(* decode_mb)(void *opaque, int ref, int mv_dir, int mv_type, int(*mv)[2][4][2], int mb_x, int mb_y, int mb_intra, int mb_skipped)
Definition: error_resilience.h:85
ERContext::mb_width
int mb_width
Definition: error_resilience.h:60
ERContext::cur_pic
ERPicture cur_pic
Definition: error_resilience.h:73
ERContext::ref_index
int8_t * ref_index[2]
Definition: error_resilience.h:77
avcodec.h
ff_er_frame_end
void ff_er_frame_end(ERContext *s, int *decode_error_flags)
Indicate that a frame has finished decoding and perform error concealment in case it has been enabled...
Definition: error_resilience.c:892
ERContext::mb_height
int mb_height
Definition: error_resilience.h:60
me_cmp_func
int(* me_cmp_func)(struct MpegEncContext *c, const uint8_t *blk1, const uint8_t *blk2, ptrdiff_t stride, int h)
Definition: me_cmp.h:50
ERContext::mecc_inited
int mecc_inited
Definition: error_resilience.h:56
ERContext::last_pic
ERPicture last_pic
Definition: error_resilience.h:74
me_cmp.h
AVCodecContext
main external API structure.
Definition: avcodec.h:445
status
ov_status_e status
Definition: dnn_backend_openvino.c:120
ThreadFrame
Definition: threadframe.h:27
ERContext::pp_time
uint16_t pp_time
Definition: error_resilience.h:80
ERContext::error_status_table
uint8_t * error_status_table
Definition: error_resilience.h:66
ERContext::quarter_sample
int quarter_sample
Definition: error_resilience.h:82
ref
static int ref[MAX_W *MAX_W]
Definition: jpeg2000dwt.c:112
ERContext::er_temp_buffer
uint8_t * er_temp_buffer
Definition: error_resilience.h:67
ff_er_frame_start
void ff_er_frame_start(ERContext *s)
Definition: error_resilience.c:787
int
int
Definition: ffmpeg_filter.c:409
ERContext::next_pic
ERPicture next_pic
Definition: error_resilience.h:75