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  const struct ThreadProgress *progress;
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;
51 } ERPicture;
52 
53 typedef struct ERContext {
55 
58 
60  int mb_num;
62  ptrdiff_t mb_stride;
63  ptrdiff_t b8_stride;
64 
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 
77 
78  int8_t *ref_index[2];
79  int16_t (*motion_val_base[2])[2];
80 
81  uint16_t pp_time;
82  uint16_t pb_time;
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 
93 
94 /**
95  * Indicate that a frame has finished decoding and perform error concealment
96  * in case it has been enabled and is necessary and supported.
97  *
98  * @param s ERContext in use
99  * @param decode_error_flags pointer where updated decode_error_flags are written
100  * if supplied; if not, the new flags are directly
101  * applied to the AVFrame whose errors are concealed
102  */
103 void ff_er_frame_end(ERContext *s, int *decode_error_flags);
104 void ff_er_add_slice(ERContext *s, int startx, int starty, int endx, int endy,
105  int status);
106 
107 #endif /* AVCODEC_ERROR_RESILIENCE_H */
ThreadProgress
ThreadProgress is an API to easily notify other threads about progress of any kind as long as it can ...
Definition: threadprogress.h:43
ERContext::pb_time
uint16_t pb_time
Definition: error_resilience.h:82
ERContext::motion_val_base
int16_t(*[2] motion_val_base)[2]
Definition: error_resilience.h:79
ERPicture::ref_index
int8_t * ref_index[2]
Definition: error_resilience.h:47
ERContext::mbintra_table
uint8_t * mbintra_table
Definition: error_resilience.h:71
ERContext::mb_index2xy
int * mb_index2xy
Definition: error_resilience.h:59
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:389
ERContext::sad
me_cmp_func sad
Definition: error_resilience.h:56
atomic_int
intptr_t atomic_int
Definition: stdatomic.h:55
ERContext
Definition: error_resilience.h:53
ERPicture
Definition: error_resilience.h:40
ERContext::partitioned_frame
int partitioned_frame
Definition: error_resilience.h:84
ERPicture::motion_val
int16_t(*[2] motion_val)[2]
Definition: error_resilience.h:46
ERContext::mb_num
int mb_num
Definition: error_resilience.h:60
ERContext::avctx
AVCodecContext * avctx
Definition: error_resilience.h:54
ERContext::mv
int mv[2][4][2]
Definition: error_resilience.h:72
ERPicture::field_picture
int field_picture
Definition: error_resilience.h:50
s
#define s(width, name)
Definition: cbs_vp9.c:198
ERContext::dc_val
int16_t * dc_val[3]
Definition: error_resilience.h:69
ERPicture::f
AVFrame * f
Definition: error_resilience.h:41
ERPicture::mb_type
uint32_t * mb_type
Definition: error_resilience.h:49
ERPicture::tf
const struct ThreadFrame * tf
Definition: error_resilience.h:42
ERContext::error_occurred
int error_occurred
Definition: error_resilience.h:66
ERContext::mbskip_table
uint8_t * mbskip_table
Definition: error_resilience.h:70
ERPicture::progress
const struct ThreadProgress * progress
Definition: error_resilience.h:43
ERContext::opaque
void * opaque
Definition: error_resilience.h:89
ERContext::error_count
atomic_int error_count
Definition: error_resilience.h:65
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:826
ERContext::b8_stride
ptrdiff_t b8_stride
Definition: error_resilience.h:63
ERContext::mb_stride
ptrdiff_t mb_stride
Definition: error_resilience.h:62
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:86
ERContext::mb_width
int mb_width
Definition: error_resilience.h:61
ERContext::cur_pic
ERPicture cur_pic
Definition: error_resilience.h:74
ERContext::ref_index
int8_t * ref_index[2]
Definition: error_resilience.h:78
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:896
ERContext::mb_height
int mb_height
Definition: error_resilience.h:61
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:57
ERContext::last_pic
ERPicture last_pic
Definition: error_resilience.h:75
me_cmp.h
AVCodecContext
main external API structure.
Definition: avcodec.h:451
status
ov_status_e status
Definition: dnn_backend_openvino.c:100
ThreadFrame
Definition: threadframe.h:27
ERContext::pp_time
uint16_t pp_time
Definition: error_resilience.h:81
ERContext::error_status_table
uint8_t * error_status_table
Definition: error_resilience.h:67
ERContext::quarter_sample
int quarter_sample
Definition: error_resilience.h:83
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:68
ff_er_frame_start
void ff_er_frame_start(ERContext *s)
Definition: error_resilience.c:791
ERContext::next_pic
ERPicture next_pic
Definition: error_resilience.h:76