git.sesse.net Git - casparcg/blob - core/mixer/audio/audio_util.h

   1 #pragma once\r
   2 \r
   3 #include <vector>\r
   4 \r
   5 #include <stdint.h>\r
   6 \r
   7 #include <boost/range/iterator_range.hpp>\r
   8 \r
   9 #include <tbb/cache_aligned_allocator.h>\r
  10 \r
  11 namespace caspar { namespace core {\r
  12 \r
  13 static std::vector<int16_t, tbb::cache_aligned_allocator<int16_t>> audio_32_to_16_sse(const boost::iterator_range<int32_t*>& input)\r
  14 {       \r
  15         std::vector<int16_t, tbb::cache_aligned_allocator<int16_t>> audio16(input.size());\r
  16         auto audio32_ptr = reinterpret_cast<const __m128i*>(input.begin());\r
  17         auto audio16_ptr = reinterpret_cast<__m128i*>(audio16.data());\r
  18         auto size                = input.size();\r
  19         for(int n = 0; n < size/8; ++n)         \r
  20         {\r
  21                 auto xmm0 = _mm_srai_epi32(_mm_load_si128(audio32_ptr++), 16);\r
  22                 auto xmm1 = _mm_srai_epi32(_mm_load_si128(audio32_ptr++), 16);\r
  23                 auto xmm3 = _mm_packs_epi32(xmm0, xmm1);\r
  24                 _mm_store_si128(audio16_ptr++, xmm3);\r
  25         }\r
  26         return audio16;\r
  27 }\r
  28 \r
  29 static std::vector<int8_t> audio_32_to_24(const boost::iterator_range<int32_t*>& input)\r
  30 {       \r
  31         std::vector<int8_t> audio24(input.size()*3+16);\r
  32         auto audio32_ptr = reinterpret_cast<const uint32_t*>(input.begin());\r
  33         auto audio24_ptr = reinterpret_cast<uint8_t*>(audio24.data());\r
  34         auto size                = input.size();\r
  35         for(int n = 0; n < size; ++n)           \r
  36                 *reinterpret_cast<uint32_t*>(audio24_ptr + n*3) = *(audio32_ptr + n) >> 8;      \r
  37         audio24.resize(input.size());\r
  38         return audio24;\r
  39 }\r
  40 \r
  41 }}