wmv2.h 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485
  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. #ifndef AVCODEC_WMV2_H
  21. #define AVCODEC_WMV2_H
  22. #include "avcodec.h"
  23. #include "intrax8.h"
  24. #include "mpegvideo.h"
  25. #include "wmv2dsp.h"
  26. #define SKIP_TYPE_NONE 0
  27. #define SKIP_TYPE_MPEG 1
  28. #define SKIP_TYPE_ROW 2
  29. #define SKIP_TYPE_COL 3
  30. typedef struct Wmv2Context {
  31. MpegEncContext s;
  32. IntraX8Context x8;
  33. WMV2DSPContext wdsp;
  34. int j_type_bit;
  35. int j_type;
  36. int abt_flag;
  37. int abt_type;
  38. int abt_type_table[6];
  39. int per_mb_abt;
  40. int per_block_abt;
  41. int mspel_bit;
  42. int cbp_table_index;
  43. int top_left_mv_flag;
  44. int per_mb_rl_bit;
  45. int skip_type;
  46. int hshift;
  47. ScanTable abt_scantable[2];
  48. DECLARE_ALIGNED(32, int16_t, abt_block2)[6][64];
  49. } Wmv2Context;
  50. void ff_wmv2_common_init(Wmv2Context *w);
  51. int ff_wmv2_decode_mb(MpegEncContext *s, int16_t block[6][64]);
  52. int ff_wmv2_encode_picture_header(MpegEncContext * s, int picture_number);
  53. void ff_wmv2_encode_mb(MpegEncContext * s, int16_t block[6][64],
  54. int motion_x, int motion_y);
  55. int ff_wmv2_decode_picture_header(MpegEncContext * s);
  56. int ff_wmv2_decode_secondary_picture_header(MpegEncContext * s);
  57. void ff_wmv2_add_mb(MpegEncContext *s, int16_t block[6][64],
  58. uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr);
  59. void ff_mspel_motion(MpegEncContext *s,
  60. uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
  61. uint8_t **ref_picture, op_pixels_func (*pix_op)[4],
  62. int motion_x, int motion_y, int h);
  63. static av_always_inline int wmv2_get_cbp_table_index(MpegEncContext *s, int cbp_index)
  64. {
  65. static const uint8_t map[3][3] = {
  66. { 0, 2, 1 },
  67. { 1, 0, 2 },
  68. { 2, 1, 0 },
  69. };
  70. return map[(s->qscale > 10) + (s->qscale > 20)][cbp_index];
  71. }
  72. #endif /* AVCODEC_WMV2_H */