libavcodec/error_resilience.h

   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 */