snow_dwt.h 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130
  1. /*
  2. * Copyright (C) 2004-2010 Michael Niedermayer <michaelni@gmx.at>
  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_SNOW_DWT_H
  21. #define AVCODEC_SNOW_DWT_H
  22. #include <stddef.h>
  23. #include <stdint.h>
  24. struct MpegEncContext;
  25. typedef int DWTELEM;
  26. typedef short IDWTELEM;
  27. #define MAX_DECOMPOSITIONS 8
  28. typedef struct DWTCompose {
  29. IDWTELEM *b0;
  30. IDWTELEM *b1;
  31. IDWTELEM *b2;
  32. IDWTELEM *b3;
  33. int y;
  34. } DWTCompose;
  35. /** Used to minimize the amount of memory used in order to
  36. * optimize cache performance. **/
  37. typedef struct slice_buffer_s {
  38. IDWTELEM **line; ///< For use by idwt and predict_slices.
  39. IDWTELEM **data_stack; ///< Used for internal purposes.
  40. int data_stack_top;
  41. int line_count;
  42. int line_width;
  43. int data_count;
  44. IDWTELEM *base_buffer; ///< Buffer that this structure is caching.
  45. } slice_buffer;
  46. struct SnowDWTContext;
  47. typedef struct SnowDWTContext {
  48. void (*vertical_compose97i)(IDWTELEM *b0, IDWTELEM *b1, IDWTELEM *b2,
  49. IDWTELEM *b3, IDWTELEM *b4, IDWTELEM *b5,
  50. int width);
  51. void (*horizontal_compose97i)(IDWTELEM *b, IDWTELEM *temp, int width);
  52. void (*inner_add_yblock)(const uint8_t *obmc, const int obmc_stride,
  53. uint8_t **block, int b_w, int b_h, int src_x,
  54. int src_y, int src_stride, slice_buffer *sb,
  55. int add, uint8_t *dst8);
  56. } SnowDWTContext;
  57. #define DWT_97 0
  58. #define DWT_53 1
  59. #define liftS lift
  60. #define W_AM 3
  61. #define W_AO 0
  62. #define W_AS 1
  63. #undef liftS
  64. #define W_BM 1
  65. #define W_BO 8
  66. #define W_BS 4
  67. #define W_CM 1
  68. #define W_CO 0
  69. #define W_CS 0
  70. #define W_DM 3
  71. #define W_DO 4
  72. #define W_DS 3
  73. #define slice_buffer_get_line(slice_buf, line_num) \
  74. ((slice_buf)->line[line_num] ? (slice_buf)->line[line_num] \
  75. : ff_slice_buffer_load_line((slice_buf), \
  76. (line_num)))
  77. int ff_slice_buffer_init(slice_buffer *buf, int line_count,
  78. int max_allocated_lines, int line_width,
  79. IDWTELEM *base_buffer);
  80. void ff_slice_buffer_release(slice_buffer *buf, int line);
  81. void ff_slice_buffer_flush(slice_buffer *buf);
  82. void ff_slice_buffer_destroy(slice_buffer *buf);
  83. IDWTELEM *ff_slice_buffer_load_line(slice_buffer *buf, int line);
  84. void ff_snow_vertical_compose97i(IDWTELEM *b0, IDWTELEM *b1, IDWTELEM *b2,
  85. IDWTELEM *b3, IDWTELEM *b4, IDWTELEM *b5,
  86. int width);
  87. void ff_snow_horizontal_compose97i(IDWTELEM *b, IDWTELEM *temp, int width);
  88. void ff_snow_inner_add_yblock(const uint8_t *obmc, const int obmc_stride,
  89. uint8_t **block, int b_w, int b_h, int src_x,
  90. int src_y, int src_stride, slice_buffer *sb,
  91. int add, uint8_t *dst8);
  92. int ff_w53_32_c(struct MpegEncContext *v, uint8_t *pix1, uint8_t *pix2, ptrdiff_t line_size, int h);
  93. int ff_w97_32_c(struct MpegEncContext *v, uint8_t *pix1, uint8_t *pix2, ptrdiff_t line_size, int h);
  94. void ff_spatial_dwt(int *buffer, int *temp, int width, int height, int stride,
  95. int type, int decomposition_count);
  96. void ff_spatial_idwt_buffered_init(DWTCompose *cs, slice_buffer *sb, int width,
  97. int height, int stride_line, int type,
  98. int decomposition_count);
  99. void ff_spatial_idwt_buffered_slice(SnowDWTContext *dsp, DWTCompose *cs,
  100. slice_buffer *slice_buf, IDWTELEM *temp,
  101. int width, int height, int stride_line,
  102. int type, int decomposition_count, int y);
  103. void ff_spatial_idwt(IDWTELEM *buffer, IDWTELEM *temp, int width, int height,
  104. int stride, int type, int decomposition_count);
  105. void ff_dwt_init(SnowDWTContext *c);
  106. void ff_dwt_init_x86(SnowDWTContext *c);
  107. #endif /* AVCODEC_DWT_H */