git.sesse.net Git - ffmpeg/blob - libavcodec/error_resilience.h

   1 /*
   2  *
   3  * This file is part of FFmpeg.
   4  *
   5  * FFmpeg is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * FFmpeg is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with FFmpeg; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  18  */
  19
  20 #ifndef AVCODEC_ERROR_RESILIENCE_H
  21 #define AVCODEC_ERROR_RESILIENCE_H
  22
  23 ///< current MB is the first after a resync marker
  24 #define VP_START               1
  25 #define ER_AC_ERROR            2
  26 #define ER_DC_ERROR            4
  27 #define ER_MV_ERROR            8
  28 #define ER_AC_END              16
  29 #define ER_DC_END              32
  30 #define ER_MV_END              64
  31
  32 #define ER_MB_ERROR (ER_AC_ERROR|ER_DC_ERROR|ER_MV_ERROR)
  33 #define ER_MB_END   (ER_AC_END|ER_DC_END|ER_MV_END)
  34
  35 typedef struct ERContext {
  36     AVCodecContext *avctx;
  37     DSPContext *dsp;
  38
  39     int *mb_index2xy;
  40     int mb_num;
  41     int mb_width, mb_height;
  42     int mb_stride;
  43     int b8_stride;
  44
  45     int error_count, error_occurred;
  46     uint8_t *error_status_table;
  47     uint8_t *er_temp_buffer;
  48     int16_t *dc_val[3];
  49     uint8_t *mbskip_table;
  50     uint8_t *mbintra_table;
  51     int mv[2][4][2];
  52
  53     struct Picture *cur_pic;
  54     struct Picture *last_pic;
  55     struct Picture *next_pic;
  56
  57     uint16_t pp_time;
  58     uint16_t pb_time;
  59     int quarter_sample;
  60     int partitioned_frame;
  61     int ref_count;
  62
  63     void (*decode_mb)(void *opaque, int ref, int mv_dir, int mv_type,
  64                       int (*mv)[2][4][2],
  65                       int mb_x, int mb_y, int mb_intra, int mb_skipped);
  66     void *opaque;
  67 } ERContext;
  68
  69 void ff_er_frame_start(ERContext *s);
  70 void ff_er_frame_end(ERContext *s);
  71 void ff_er_add_slice(ERContext *s, int startx, int starty, int endx, int endy,
  72                      int status);
  73
  74 #endif /* AVCODEC_ERROR_RESILIENCE_H */