vpx_convolve_neon.c 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. /*
  2. * Copyright (c) 2013 The WebM project authors. All Rights Reserved.
  3. *
  4. * Use of this source code is governed by a BSD-style license
  5. * that can be found in the LICENSE file in the root of the source
  6. * tree. An additional intellectual property rights grant can be found
  7. * in the file PATENTS. All contributing project authors may
  8. * be found in the AUTHORS file in the root of the source tree.
  9. */
  10. #include <assert.h>
  11. #include "./vpx_dsp_rtcd.h"
  12. #include "vpx_dsp/vpx_dsp_common.h"
  13. #include "vpx_ports/mem.h"
  14. void vpx_convolve8_neon(const uint8_t *src, ptrdiff_t src_stride,
  15. uint8_t *dst, ptrdiff_t dst_stride,
  16. const int16_t *filter_x, int x_step_q4,
  17. const int16_t *filter_y, int y_step_q4,
  18. int w, int h) {
  19. /* Given our constraints: w <= 64, h <= 64, taps == 8 we can reduce the
  20. * maximum buffer size to 64 * 64 + 7 (+ 1 to make it divisible by 4).
  21. */
  22. DECLARE_ALIGNED(8, uint8_t, temp[64 * 72]);
  23. // Account for the vertical phase needing 3 lines prior and 4 lines post
  24. int intermediate_height = h + 7;
  25. assert(y_step_q4 == 16);
  26. assert(x_step_q4 == 16);
  27. /* Filter starting 3 lines back. The neon implementation will ignore the
  28. * given height and filter a multiple of 4 lines. Since this goes in to
  29. * the temp buffer which has lots of extra room and is subsequently discarded
  30. * this is safe if somewhat less than ideal.
  31. */
  32. vpx_convolve8_horiz_neon(src - src_stride * 3, src_stride,
  33. temp, 64,
  34. filter_x, x_step_q4, filter_y, y_step_q4,
  35. w, intermediate_height);
  36. /* Step into the temp buffer 3 lines to get the actual frame data */
  37. vpx_convolve8_vert_neon(temp + 64 * 3, 64,
  38. dst, dst_stride,
  39. filter_x, x_step_q4, filter_y, y_step_q4,
  40. w, h);
  41. }
  42. void vpx_convolve8_avg_neon(const uint8_t *src, ptrdiff_t src_stride,
  43. uint8_t *dst, ptrdiff_t dst_stride,
  44. const int16_t *filter_x, int x_step_q4,
  45. const int16_t *filter_y, int y_step_q4,
  46. int w, int h) {
  47. DECLARE_ALIGNED(8, uint8_t, temp[64 * 72]);
  48. int intermediate_height = h + 7;
  49. assert(y_step_q4 == 16);
  50. assert(x_step_q4 == 16);
  51. /* This implementation has the same issues as above. In addition, we only want
  52. * to average the values after both passes.
  53. */
  54. vpx_convolve8_horiz_neon(src - src_stride * 3, src_stride,
  55. temp, 64,
  56. filter_x, x_step_q4, filter_y, y_step_q4,
  57. w, intermediate_height);
  58. vpx_convolve8_avg_vert_neon(temp + 64 * 3,
  59. 64, dst, dst_stride,
  60. filter_x, x_step_q4, filter_y, y_step_q4,
  61. w, h);
  62. }