bitdepth_conversion_sse2.h 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. /*
  2. * Copyright (c) 2017 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. #ifndef VPX_VPX_DSP_X86_BITDEPTH_CONVERSION_SSE2_H_
  11. #define VPX_VPX_DSP_X86_BITDEPTH_CONVERSION_SSE2_H_
  12. #include <xmmintrin.h>
  13. #include "./vpx_config.h"
  14. #include "vpx/vpx_integer.h"
  15. #include "vpx_dsp/vpx_dsp_common.h"
  16. // Load 8 16 bit values. If the source is 32 bits then pack down with
  17. // saturation.
  18. static INLINE __m128i load_tran_low(const tran_low_t *a) {
  19. #if CONFIG_VP9_HIGHBITDEPTH
  20. const __m128i a_low = _mm_load_si128((const __m128i *)a);
  21. return _mm_packs_epi32(a_low, *(const __m128i *)(a + 4));
  22. #else
  23. return _mm_load_si128((const __m128i *)a);
  24. #endif
  25. }
  26. // Store 8 16 bit values. If the destination is 32 bits then sign extend the
  27. // values by multiplying by 1.
  28. static INLINE void store_tran_low(__m128i a, tran_low_t *b) {
  29. #if CONFIG_VP9_HIGHBITDEPTH
  30. const __m128i one = _mm_set1_epi16(1);
  31. const __m128i a_hi = _mm_mulhi_epi16(a, one);
  32. const __m128i a_lo = _mm_mullo_epi16(a, one);
  33. const __m128i a_1 = _mm_unpacklo_epi16(a_lo, a_hi);
  34. const __m128i a_2 = _mm_unpackhi_epi16(a_lo, a_hi);
  35. _mm_store_si128((__m128i *)(b), a_1);
  36. _mm_store_si128((__m128i *)(b + 4), a_2);
  37. #else
  38. _mm_store_si128((__m128i *)(b), a);
  39. #endif
  40. }
  41. // Zero fill 8 positions in the output buffer.
  42. static INLINE void store_zero_tran_low(tran_low_t *a) {
  43. const __m128i zero = _mm_setzero_si128();
  44. #if CONFIG_VP9_HIGHBITDEPTH
  45. _mm_store_si128((__m128i *)(a), zero);
  46. _mm_store_si128((__m128i *)(a + 4), zero);
  47. #else
  48. _mm_store_si128((__m128i *)(a), zero);
  49. #endif
  50. }
  51. #endif // VPX_VPX_DSP_X86_BITDEPTH_CONVERSION_SSE2_H_