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

   1 /*
   2  * Copyright (c) 2002 The FFmpeg Project
   3  *
   4  * This file is part of FFmpeg.
   5  *
   6  * FFmpeg is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * FFmpeg is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with FFmpeg; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19  */
  20
  21 #ifndef AVCODEC_WMV2_H
  22 #define AVCODEC_WMV2_H
  23
  24 #include "avcodec.h"
  25 #include "intrax8.h"
  26 #include "mpegvideo.h"
  27 #include "wmv2dsp.h"
  28
  29 #define SKIP_TYPE_NONE 0
  30 #define SKIP_TYPE_MPEG 1
  31 #define SKIP_TYPE_ROW  2
  32 #define SKIP_TYPE_COL  3
  33
  34
  35 typedef struct Wmv2Context {
  36     MpegEncContext s;
  37     IntraX8Context x8;
  38     WMV2DSPContext wdsp;
  39     int j_type_bit;
  40     int j_type;
  41     int abt_flag;
  42     int abt_type;
  43     int abt_type_table[6];
  44     int per_mb_abt;
  45     int per_block_abt;
  46     int mspel_bit;
  47     int cbp_table_index;
  48     int top_left_mv_flag;
  49     int per_mb_rl_bit;
  50     int skip_type;
  51     int hshift;
  52
  53     ScanTable abt_scantable[2];
  54     DECLARE_ALIGNED(32, int16_t, abt_block2)[6][64];
  55 } Wmv2Context;
  56
  57 void ff_wmv2_common_init(Wmv2Context *w);
  58
  59 int ff_wmv2_decode_mb(MpegEncContext *s, int16_t block[6][64]);
  60 int ff_wmv2_encode_picture_header(MpegEncContext * s, int picture_number);
  61 void ff_wmv2_encode_mb(MpegEncContext * s, int16_t block[6][64],
  62                        int motion_x, int motion_y);
  63 int ff_wmv2_decode_picture_header(MpegEncContext * s);
  64 int ff_wmv2_decode_secondary_picture_header(MpegEncContext * s);
  65 void ff_wmv2_add_mb(MpegEncContext *s, int16_t block[6][64],
  66                     uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr);
  67
  68 void ff_mspel_motion(MpegEncContext *s,
  69                      uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
  70                      uint8_t **ref_picture, op_pixels_func (*pix_op)[4],
  71                      int motion_x, int motion_y, int h);
  72
  73
  74 static av_always_inline int wmv2_get_cbp_table_index(MpegEncContext *s, int cbp_index)
  75 {
  76     static const uint8_t map[3][3] = {
  77         { 0, 2, 1 },
  78         { 1, 0, 2 },
  79         { 2, 1, 0 },
  80     };
  81
  82     return map[(s->qscale > 10) + (s->qscale > 20)][cbp_index];
  83 }
  84
  85 #endif /* AVCODEC_WMV2_H */