error_resilience.h (3196B)
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 { 41 AVFrame *f; 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; 50 int field_picture; 51 } ERPicture; 52 53 typedef 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 int8_t *ref_index[2]; 79 int16_t (*motion_val_base[2])[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 92 void ff_er_frame_start(ERContext *s); 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 */