Audio plugin host https://kx.studio/carla
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

juce_neon_SIMDNativeOps.h 38KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501
  1. /*
  2. ==============================================================================
  3. This file is part of the JUCE library.
  4. Copyright (c) 2022 - Raw Material Software Limited
  5. JUCE is an open source library subject to commercial or open-source
  6. licensing.
  7. By using JUCE, you agree to the terms of both the JUCE 7 End-User License
  8. Agreement and JUCE Privacy Policy.
  9. End User License Agreement: www.juce.com/juce-7-licence
  10. Privacy Policy: www.juce.com/juce-privacy-policy
  11. Or: You may also use this code under the terms of the GPL v3 (see
  12. www.gnu.org/licenses).
  13. JUCE IS PROVIDED "AS IS" WITHOUT ANY WARRANTY, AND ALL WARRANTIES, WHETHER
  14. EXPRESSED OR IMPLIED, INCLUDING MERCHANTABILITY AND FITNESS FOR PURPOSE, ARE
  15. DISCLAIMED.
  16. ==============================================================================
  17. */
  18. namespace juce
  19. {
  20. namespace dsp
  21. {
  22. #ifndef DOXYGEN
  23. JUCE_BEGIN_IGNORE_WARNINGS_GCC_LIKE ("-Wignored-attributes")
  24. #ifdef _MSC_VER
  25. #define DECLARE_NEON_SIMD_CONST(type, name) \
  26. static __declspec(align(16)) const type name [16 / sizeof (type)]
  27. #define DEFINE_NEON_SIMD_CONST(type, class_type, name) \
  28. __declspec(align(16)) const type SIMDNativeOps<class_type>:: name [16 / sizeof (type)]
  29. #else
  30. #define DECLARE_NEON_SIMD_CONST(type, name) \
  31. static const type name [16 / sizeof (type)] __attribute__((aligned(16)))
  32. #define DEFINE_NEON_SIMD_CONST(type, class_type, name) \
  33. const type SIMDNativeOps<class_type>:: name [16 / sizeof (type)] __attribute__((aligned(16)))
  34. #endif
  35. template <typename type>
  36. struct SIMDNativeOps;
  37. //==============================================================================
  38. /** Unsigned 32-bit integer NEON intrinsics.
  39. @tags{DSP}
  40. */
  41. template <>
  42. struct SIMDNativeOps<uint32_t>
  43. {
  44. //==============================================================================
  45. using vSIMDType = uint32x4_t;
  46. using fb = SIMDFallbackOps<uint32_t, vSIMDType>;
  47. //==============================================================================
  48. DECLARE_NEON_SIMD_CONST (uint32_t, kAllBitsSet);
  49. //==============================================================================
  50. static forcedinline vSIMDType expand (uint32_t s) noexcept { return vdupq_n_u32 (s); }
  51. static forcedinline vSIMDType load (const uint32_t* a) noexcept { return vld1q_u32 (a); }
  52. static forcedinline void store (vSIMDType value, uint32_t* a) noexcept { vst1q_u32 (a, value); }
  53. static forcedinline uint32_t get (vSIMDType v, size_t i) noexcept { return v[i]; }
  54. static forcedinline vSIMDType set (vSIMDType v, size_t i, uint32_t s) noexcept { v[i] = s; return v; }
  55. static forcedinline vSIMDType add (vSIMDType a, vSIMDType b) noexcept { return vaddq_u32 (a, b); }
  56. static forcedinline vSIMDType sub (vSIMDType a, vSIMDType b) noexcept { return vsubq_u32 (a, b); }
  57. static forcedinline vSIMDType mul (vSIMDType a, vSIMDType b) noexcept { return vmulq_u32 (a, b); }
  58. static forcedinline vSIMDType bit_and (vSIMDType a, vSIMDType b) noexcept { return vandq_u32 (a, b); }
  59. static forcedinline vSIMDType bit_or (vSIMDType a, vSIMDType b) noexcept { return vorrq_u32 (a, b); }
  60. static forcedinline vSIMDType bit_xor (vSIMDType a, vSIMDType b) noexcept { return veorq_u32 (a, b); }
  61. static forcedinline vSIMDType bit_notand (vSIMDType a, vSIMDType b) noexcept { return vbicq_u32 (b, a); }
  62. static forcedinline vSIMDType bit_not (vSIMDType a) noexcept { return bit_notand (a, vld1q_u32 ((uint32_t*) kAllBitsSet)); }
  63. static forcedinline vSIMDType min (vSIMDType a, vSIMDType b) noexcept { return vminq_u32 (a, b); }
  64. static forcedinline vSIMDType max (vSIMDType a, vSIMDType b) noexcept { return vmaxq_u32 (a, b); }
  65. static forcedinline vSIMDType equal (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vceqq_u32 (a, b); }
  66. static forcedinline bool allEqual (vSIMDType a, vSIMDType b) noexcept { return (sum (notEqual (a, b)) == 0); }
  67. static forcedinline vSIMDType notEqual (vSIMDType a, vSIMDType b) noexcept { return bit_not (equal (a, b)); }
  68. static forcedinline vSIMDType greaterThan (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vcgtq_u32 (a, b); }
  69. static forcedinline vSIMDType greaterThanOrEqual (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vcgeq_u32 (a, b); }
  70. static forcedinline vSIMDType multiplyAdd (vSIMDType a, vSIMDType b, vSIMDType c) noexcept { return vmlaq_u32 (a, b, c); }
  71. static forcedinline vSIMDType truncate (vSIMDType a) noexcept { return a; }
  72. static forcedinline uint32_t sum (vSIMDType a) noexcept
  73. {
  74. auto rr = vadd_u32 (vget_high_u32 (a), vget_low_u32 (a));
  75. return vget_lane_u32 (vpadd_u32 (rr, rr), 0);
  76. }
  77. };
  78. //==============================================================================
  79. /** Signed 32-bit integer NEON intrinsics.
  80. @tags{DSP}
  81. */
  82. template <>
  83. struct SIMDNativeOps<int32_t>
  84. {
  85. //==============================================================================
  86. using vSIMDType = int32x4_t;
  87. using fb = SIMDFallbackOps<int32_t, vSIMDType>;
  88. //==============================================================================
  89. DECLARE_NEON_SIMD_CONST (int32_t, kAllBitsSet);
  90. //==============================================================================
  91. static forcedinline vSIMDType expand (int32_t s) noexcept { return vdupq_n_s32 (s); }
  92. static forcedinline vSIMDType load (const int32_t* a) noexcept { return vld1q_s32 (a); }
  93. static forcedinline void store (vSIMDType value, int32_t* a) noexcept { vst1q_s32 (a, value); }
  94. static forcedinline int32_t get (vSIMDType v, size_t i) noexcept { return v[i]; }
  95. static forcedinline vSIMDType set (vSIMDType v, size_t i, int32_t s) noexcept { v[i] = s; return v; }
  96. static forcedinline vSIMDType add (vSIMDType a, vSIMDType b) noexcept { return vaddq_s32 (a, b); }
  97. static forcedinline vSIMDType sub (vSIMDType a, vSIMDType b) noexcept { return vsubq_s32 (a, b); }
  98. static forcedinline vSIMDType mul (vSIMDType a, vSIMDType b) noexcept { return vmulq_s32 (a, b); }
  99. static forcedinline vSIMDType bit_and (vSIMDType a, vSIMDType b) noexcept { return vandq_s32 (a, b); }
  100. static forcedinline vSIMDType bit_or (vSIMDType a, vSIMDType b) noexcept { return vorrq_s32 (a, b); }
  101. static forcedinline vSIMDType bit_xor (vSIMDType a, vSIMDType b) noexcept { return veorq_s32 (a, b); }
  102. static forcedinline vSIMDType bit_notand (vSIMDType a, vSIMDType b) noexcept { return vbicq_s32 (b, a); }
  103. static forcedinline vSIMDType bit_not (vSIMDType a) noexcept { return bit_notand (a, vld1q_s32 ((int32_t*) kAllBitsSet)); }
  104. static forcedinline vSIMDType min (vSIMDType a, vSIMDType b) noexcept { return vminq_s32 (a, b); }
  105. static forcedinline vSIMDType max (vSIMDType a, vSIMDType b) noexcept { return vmaxq_s32 (a, b); }
  106. static forcedinline vSIMDType equal (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vceqq_s32 (a, b); }
  107. static forcedinline bool allEqual (vSIMDType a, vSIMDType b) noexcept { return (sum (notEqual (a, b)) == 0); }
  108. static forcedinline vSIMDType notEqual (vSIMDType a, vSIMDType b) noexcept { return bit_not (equal (a, b)); }
  109. static forcedinline vSIMDType greaterThan (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vcgtq_s32 (a, b); }
  110. static forcedinline vSIMDType greaterThanOrEqual (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vcgeq_s32 (a, b); }
  111. static forcedinline vSIMDType multiplyAdd (vSIMDType a, vSIMDType b, vSIMDType c) noexcept { return vmlaq_s32 (a, b, c); }
  112. static forcedinline vSIMDType truncate (vSIMDType a) noexcept { return a; }
  113. static forcedinline int32_t sum (vSIMDType a) noexcept
  114. {
  115. auto rr = vadd_s32 (vget_high_s32 (a), vget_low_s32 (a));
  116. rr = vpadd_s32 (rr, rr);
  117. return vget_lane_s32 (rr, 0);
  118. }
  119. };
  120. //==============================================================================
  121. /** Signed 8-bit integer NEON intrinsics.
  122. @tags{DSP}
  123. */
  124. template <>
  125. struct SIMDNativeOps<int8_t>
  126. {
  127. //==============================================================================
  128. using vSIMDType = int8x16_t;
  129. using fb = SIMDFallbackOps<int8_t, vSIMDType>;
  130. //==============================================================================
  131. DECLARE_NEON_SIMD_CONST (int8_t, kAllBitsSet);
  132. //==============================================================================
  133. static forcedinline vSIMDType expand (int8_t s) noexcept { return vdupq_n_s8 (s); }
  134. static forcedinline vSIMDType load (const int8_t* a) noexcept { return vld1q_s8 (a); }
  135. static forcedinline void store (vSIMDType value, int8_t* a) noexcept { vst1q_s8 (a, value); }
  136. static forcedinline int8_t get (vSIMDType v, size_t i) noexcept { return v[i]; }
  137. static forcedinline vSIMDType set (vSIMDType v, size_t i, int8_t s) noexcept { v[i] = s; return v; }
  138. static forcedinline vSIMDType add (vSIMDType a, vSIMDType b) noexcept { return vaddq_s8 (a, b); }
  139. static forcedinline vSIMDType sub (vSIMDType a, vSIMDType b) noexcept { return vsubq_s8 (a, b); }
  140. static forcedinline vSIMDType mul (vSIMDType a, vSIMDType b) noexcept { return vmulq_s8 (a, b); }
  141. static forcedinline vSIMDType bit_and (vSIMDType a, vSIMDType b) noexcept { return vandq_s8 (a, b); }
  142. static forcedinline vSIMDType bit_or (vSIMDType a, vSIMDType b) noexcept { return vorrq_s8 (a, b); }
  143. static forcedinline vSIMDType bit_xor (vSIMDType a, vSIMDType b) noexcept { return veorq_s8 (a, b); }
  144. static forcedinline vSIMDType bit_notand (vSIMDType a, vSIMDType b) noexcept { return vbicq_s8 (b, a); }
  145. static forcedinline vSIMDType bit_not (vSIMDType a) noexcept { return bit_notand (a, vld1q_s8 ((int8_t*) kAllBitsSet)); }
  146. static forcedinline vSIMDType min (vSIMDType a, vSIMDType b) noexcept { return vminq_s8 (a, b); }
  147. static forcedinline vSIMDType max (vSIMDType a, vSIMDType b) noexcept { return vmaxq_s8 (a, b); }
  148. static forcedinline vSIMDType equal (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vceqq_s8 (a, b); }
  149. static forcedinline vSIMDType notEqual (vSIMDType a, vSIMDType b) noexcept { return bit_not (equal (a, b)); }
  150. static forcedinline vSIMDType greaterThan (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vcgtq_s8 (a, b); }
  151. static forcedinline vSIMDType greaterThanOrEqual (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vcgeq_s8 (a, b); }
  152. static forcedinline bool allEqual (vSIMDType a, vSIMDType b) noexcept { return (SIMDNativeOps<int32_t>::sum ((SIMDNativeOps<int32_t>::vSIMDType) notEqual (a, b)) == 0); }
  153. static forcedinline vSIMDType multiplyAdd (vSIMDType a, vSIMDType b, vSIMDType c) noexcept { return vmlaq_s8 (a, b, c); }
  154. static forcedinline int8_t sum (vSIMDType a) noexcept { return fb::sum (a); }
  155. static forcedinline vSIMDType truncate (vSIMDType a) noexcept { return a; }
  156. };
  157. //==============================================================================
  158. /** Unsigned 8-bit integer NEON intrinsics.
  159. @tags{DSP}
  160. */
  161. template <>
  162. struct SIMDNativeOps<uint8_t>
  163. {
  164. //==============================================================================
  165. using vSIMDType = uint8x16_t;
  166. using fb = SIMDFallbackOps<uint8_t, vSIMDType>;
  167. //==============================================================================
  168. DECLARE_NEON_SIMD_CONST (uint8_t, kAllBitsSet);
  169. //==============================================================================
  170. static forcedinline vSIMDType expand (uint8_t s) noexcept { return vdupq_n_u8 (s); }
  171. static forcedinline vSIMDType load (const uint8_t* a) noexcept { return vld1q_u8 (a); }
  172. static forcedinline void store (vSIMDType value, uint8_t* a) noexcept { vst1q_u8 (a, value); }
  173. static forcedinline uint8_t get (vSIMDType v, size_t i) noexcept { return v[i]; }
  174. static forcedinline vSIMDType set (vSIMDType v, size_t i, uint8_t s) noexcept { v[i] = s; return v; }
  175. static forcedinline vSIMDType add (vSIMDType a, vSIMDType b) noexcept { return vaddq_u8 (a, b); }
  176. static forcedinline vSIMDType sub (vSIMDType a, vSIMDType b) noexcept { return vsubq_u8 (a, b); }
  177. static forcedinline vSIMDType mul (vSIMDType a, vSIMDType b) noexcept { return vmulq_u8 (a, b); }
  178. static forcedinline vSIMDType bit_and (vSIMDType a, vSIMDType b) noexcept { return vandq_u8 (a, b); }
  179. static forcedinline vSIMDType bit_or (vSIMDType a, vSIMDType b) noexcept { return vorrq_u8 (a, b); }
  180. static forcedinline vSIMDType bit_xor (vSIMDType a, vSIMDType b) noexcept { return veorq_u8 (a, b); }
  181. static forcedinline vSIMDType bit_notand (vSIMDType a, vSIMDType b) noexcept { return vbicq_u8 (b, a); }
  182. static forcedinline vSIMDType bit_not (vSIMDType a) noexcept { return bit_notand (a, vld1q_u8 ((uint8_t*) kAllBitsSet)); }
  183. static forcedinline vSIMDType min (vSIMDType a, vSIMDType b) noexcept { return vminq_u8 (a, b); }
  184. static forcedinline vSIMDType max (vSIMDType a, vSIMDType b) noexcept { return vmaxq_u8 (a, b); }
  185. static forcedinline vSIMDType equal (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vceqq_u8 (a, b); }
  186. static forcedinline vSIMDType notEqual (vSIMDType a, vSIMDType b) noexcept { return bit_not (equal (a, b)); }
  187. static forcedinline vSIMDType greaterThan (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vcgtq_u8 (a, b); }
  188. static forcedinline vSIMDType greaterThanOrEqual (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vcgeq_u8 (a, b); }
  189. static forcedinline bool allEqual (vSIMDType a, vSIMDType b) noexcept { return (SIMDNativeOps<uint32_t>::sum ((SIMDNativeOps<uint32_t>::vSIMDType) notEqual (a, b)) == 0); }
  190. static forcedinline vSIMDType multiplyAdd (vSIMDType a, vSIMDType b, vSIMDType c) noexcept { return vmlaq_u8 (a, b, c); }
  191. static forcedinline uint8_t sum (vSIMDType a) noexcept { return fb::sum (a); }
  192. static forcedinline vSIMDType truncate (vSIMDType a) noexcept { return a; }
  193. };
  194. //==============================================================================
  195. /** Signed 16-bit integer NEON intrinsics.
  196. @tags{DSP}
  197. */
  198. template <>
  199. struct SIMDNativeOps<int16_t>
  200. {
  201. //==============================================================================
  202. using vSIMDType = int16x8_t;
  203. using fb = SIMDFallbackOps<int16_t, vSIMDType>;
  204. //==============================================================================
  205. DECLARE_NEON_SIMD_CONST (int16_t, kAllBitsSet);
  206. //==============================================================================
  207. static forcedinline vSIMDType expand (int16_t s) noexcept { return vdupq_n_s16 (s); }
  208. static forcedinline vSIMDType load (const int16_t* a) noexcept { return vld1q_s16 (a); }
  209. static forcedinline void store (vSIMDType value, int16_t* a) noexcept { vst1q_s16 (a, value); }
  210. static forcedinline int16_t get (vSIMDType v, size_t i) noexcept { return v[i]; }
  211. static forcedinline vSIMDType set (vSIMDType v, size_t i, int16_t s) noexcept { v[i] = s; return v; }
  212. static forcedinline vSIMDType add (vSIMDType a, vSIMDType b) noexcept { return vaddq_s16 (a, b); }
  213. static forcedinline vSIMDType sub (vSIMDType a, vSIMDType b) noexcept { return vsubq_s16 (a, b); }
  214. static forcedinline vSIMDType mul (vSIMDType a, vSIMDType b) noexcept { return vmulq_s16 (a, b); }
  215. static forcedinline vSIMDType bit_and (vSIMDType a, vSIMDType b) noexcept { return vandq_s16 (a, b); }
  216. static forcedinline vSIMDType bit_or (vSIMDType a, vSIMDType b) noexcept { return vorrq_s16 (a, b); }
  217. static forcedinline vSIMDType bit_xor (vSIMDType a, vSIMDType b) noexcept { return veorq_s16 (a, b); }
  218. static forcedinline vSIMDType bit_notand (vSIMDType a, vSIMDType b) noexcept { return vbicq_s16 (b, a); }
  219. static forcedinline vSIMDType bit_not (vSIMDType a) noexcept { return bit_notand (a, vld1q_s16 ((int16_t*) kAllBitsSet)); }
  220. static forcedinline vSIMDType min (vSIMDType a, vSIMDType b) noexcept { return vminq_s16 (a, b); }
  221. static forcedinline vSIMDType max (vSIMDType a, vSIMDType b) noexcept { return vmaxq_s16 (a, b); }
  222. static forcedinline vSIMDType equal (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vceqq_s16 (a, b); }
  223. static forcedinline vSIMDType notEqual (vSIMDType a, vSIMDType b) noexcept { return bit_not (equal (a, b)); }
  224. static forcedinline vSIMDType greaterThan (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vcgtq_s16 (a, b); }
  225. static forcedinline vSIMDType greaterThanOrEqual (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vcgeq_s16 (a, b); }
  226. static forcedinline bool allEqual (vSIMDType a, vSIMDType b) noexcept { return (SIMDNativeOps<int32_t>::sum ((SIMDNativeOps<int32_t>::vSIMDType) notEqual (a, b)) == 0); }
  227. static forcedinline vSIMDType multiplyAdd (vSIMDType a, vSIMDType b, vSIMDType c) noexcept { return vmlaq_s16 (a, b, c); }
  228. static forcedinline int16_t sum (vSIMDType a) noexcept { return fb::sum (a); }
  229. static forcedinline vSIMDType truncate (vSIMDType a) noexcept { return a; }
  230. };
  231. //==============================================================================
  232. /** Unsigned 16-bit integer NEON intrinsics.
  233. @tags{DSP}
  234. */
  235. template <>
  236. struct SIMDNativeOps<uint16_t>
  237. {
  238. //==============================================================================
  239. using vSIMDType = uint16x8_t;
  240. using fb = SIMDFallbackOps<uint16_t, vSIMDType>;
  241. //==============================================================================
  242. DECLARE_NEON_SIMD_CONST (uint16_t, kAllBitsSet);
  243. //==============================================================================
  244. static forcedinline vSIMDType expand (uint16_t s) noexcept { return vdupq_n_u16 (s); }
  245. static forcedinline vSIMDType load (const uint16_t* a) noexcept { return vld1q_u16 (a); }
  246. static forcedinline void store (vSIMDType value, uint16_t* a) noexcept { vst1q_u16 (a, value); }
  247. static forcedinline uint16_t get (vSIMDType v, size_t i) noexcept { return v[i]; }
  248. static forcedinline vSIMDType set (vSIMDType v, size_t i, uint16_t s) noexcept { v[i] = s; return v; }
  249. static forcedinline vSIMDType add (vSIMDType a, vSIMDType b) noexcept { return vaddq_u16 (a, b); }
  250. static forcedinline vSIMDType sub (vSIMDType a, vSIMDType b) noexcept { return vsubq_u16 (a, b); }
  251. static forcedinline vSIMDType mul (vSIMDType a, vSIMDType b) noexcept { return vmulq_u16 (a, b); }
  252. static forcedinline vSIMDType bit_and (vSIMDType a, vSIMDType b) noexcept { return vandq_u16 (a, b); }
  253. static forcedinline vSIMDType bit_or (vSIMDType a, vSIMDType b) noexcept { return vorrq_u16 (a, b); }
  254. static forcedinline vSIMDType bit_xor (vSIMDType a, vSIMDType b) noexcept { return veorq_u16 (a, b); }
  255. static forcedinline vSIMDType bit_notand (vSIMDType a, vSIMDType b) noexcept { return vbicq_u16 (b, a); }
  256. static forcedinline vSIMDType bit_not (vSIMDType a) noexcept { return bit_notand (a, vld1q_u16 ((uint16_t*) kAllBitsSet)); }
  257. static forcedinline vSIMDType min (vSIMDType a, vSIMDType b) noexcept { return vminq_u16 (a, b); }
  258. static forcedinline vSIMDType max (vSIMDType a, vSIMDType b) noexcept { return vmaxq_u16 (a, b); }
  259. static forcedinline vSIMDType equal (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vceqq_u16 (a, b); }
  260. static forcedinline vSIMDType notEqual (vSIMDType a, vSIMDType b) noexcept { return bit_not (equal (a, b)); }
  261. static forcedinline vSIMDType greaterThan (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vcgtq_u16 (a, b); }
  262. static forcedinline vSIMDType greaterThanOrEqual (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vcgeq_u16 (a, b); }
  263. static forcedinline bool allEqual (vSIMDType a, vSIMDType b) noexcept { return (SIMDNativeOps<uint32_t>::sum ((SIMDNativeOps<uint32_t>::vSIMDType) notEqual (a, b)) == 0); }
  264. static forcedinline vSIMDType multiplyAdd (vSIMDType a, vSIMDType b, vSIMDType c) noexcept { return vmlaq_u16 (a, b, c); }
  265. static forcedinline uint16_t sum (vSIMDType a) noexcept { return fb::sum (a); }
  266. static forcedinline vSIMDType truncate (vSIMDType a) noexcept { return a; }
  267. };
  268. //==============================================================================
  269. /** Signed 64-bit integer NEON intrinsics.
  270. @tags{DSP}
  271. */
  272. template <>
  273. struct SIMDNativeOps<int64_t>
  274. {
  275. //==============================================================================
  276. using vSIMDType = int64x2_t;
  277. using fb = SIMDFallbackOps<int64_t, vSIMDType>;
  278. //==============================================================================
  279. DECLARE_NEON_SIMD_CONST (int64_t, kAllBitsSet);
  280. //==============================================================================
  281. static forcedinline vSIMDType expand (int64_t s) noexcept { return vdupq_n_s64 (s); }
  282. static forcedinline vSIMDType load (const int64_t* a) noexcept { return vld1q_s64 (a); }
  283. static forcedinline void store (vSIMDType value, int64_t* a) noexcept { vst1q_s64 (a, value); }
  284. static forcedinline int64_t get (vSIMDType v, size_t i) noexcept { return v[i]; }
  285. static forcedinline vSIMDType set (vSIMDType v, size_t i, int64_t s) noexcept { v[i] = s; return v; }
  286. static forcedinline vSIMDType add (vSIMDType a, vSIMDType b) noexcept { return vaddq_s64 (a, b); }
  287. static forcedinline vSIMDType sub (vSIMDType a, vSIMDType b) noexcept { return vsubq_s64 (a, b); }
  288. static forcedinline vSIMDType mul (vSIMDType a, vSIMDType b) noexcept { return fb::mul (a, b); }
  289. static forcedinline vSIMDType bit_and (vSIMDType a, vSIMDType b) noexcept { return vandq_s64 (a, b); }
  290. static forcedinline vSIMDType bit_or (vSIMDType a, vSIMDType b) noexcept { return vorrq_s64 (a, b); }
  291. static forcedinline vSIMDType bit_xor (vSIMDType a, vSIMDType b) noexcept { return veorq_s64 (a, b); }
  292. static forcedinline vSIMDType bit_notand (vSIMDType a, vSIMDType b) noexcept { return vbicq_s64 (b, a); }
  293. static forcedinline vSIMDType bit_not (vSIMDType a) noexcept { return bit_notand (a, vld1q_s64 ((int64_t*) kAllBitsSet)); }
  294. static forcedinline vSIMDType min (vSIMDType a, vSIMDType b) noexcept { return fb::min (a, b); }
  295. static forcedinline vSIMDType max (vSIMDType a, vSIMDType b) noexcept { return fb::max (a, b); }
  296. static forcedinline vSIMDType equal (vSIMDType a, vSIMDType b) noexcept { return fb::equal (a, b); }
  297. static forcedinline vSIMDType notEqual (vSIMDType a, vSIMDType b) noexcept { return fb::notEqual (a, b); }
  298. static forcedinline vSIMDType greaterThan (vSIMDType a, vSIMDType b) noexcept { return fb::greaterThan (a, b); }
  299. static forcedinline vSIMDType greaterThanOrEqual (vSIMDType a, vSIMDType b) noexcept { return fb::greaterThanOrEqual (a, b); }
  300. static forcedinline bool allEqual (vSIMDType a, vSIMDType b) noexcept { return (SIMDNativeOps<int32_t>::sum ((SIMDNativeOps<int32_t>::vSIMDType) notEqual (a, b)) == 0); }
  301. static forcedinline vSIMDType multiplyAdd (vSIMDType a, vSIMDType b, vSIMDType c) noexcept { return fb::multiplyAdd (a, b, c); }
  302. static forcedinline int64_t sum (vSIMDType a) noexcept { return fb::sum (a); }
  303. static forcedinline vSIMDType truncate (vSIMDType a) noexcept { return a; }
  304. };
  305. //==============================================================================
  306. /** Unsigned 64-bit integer NEON intrinsics.
  307. @tags{DSP}
  308. */
  309. template <>
  310. struct SIMDNativeOps<uint64_t>
  311. {
  312. //==============================================================================
  313. using vSIMDType = uint64x2_t;
  314. using fb = SIMDFallbackOps<uint64_t, vSIMDType>;
  315. //==============================================================================
  316. DECLARE_NEON_SIMD_CONST (uint64_t, kAllBitsSet);
  317. //==============================================================================
  318. static forcedinline vSIMDType expand (uint64_t s) noexcept { return vdupq_n_u64 (s); }
  319. static forcedinline vSIMDType load (const uint64_t* a) noexcept { return vld1q_u64 (a); }
  320. static forcedinline void store (vSIMDType value, uint64_t* a) noexcept { vst1q_u64 (a, value); }
  321. static forcedinline uint64_t get (vSIMDType v, size_t i) noexcept { return v[i]; }
  322. static forcedinline vSIMDType set (vSIMDType v, size_t i, uint64_t s) noexcept { v[i] = s; return v; }
  323. static forcedinline vSIMDType add (vSIMDType a, vSIMDType b) noexcept { return vaddq_u64 (a, b); }
  324. static forcedinline vSIMDType sub (vSIMDType a, vSIMDType b) noexcept { return vsubq_u64 (a, b); }
  325. static forcedinline vSIMDType mul (vSIMDType a, vSIMDType b) noexcept { return fb::mul (a, b); }
  326. static forcedinline vSIMDType bit_and (vSIMDType a, vSIMDType b) noexcept { return vandq_u64 (a, b); }
  327. static forcedinline vSIMDType bit_or (vSIMDType a, vSIMDType b) noexcept { return vorrq_u64 (a, b); }
  328. static forcedinline vSIMDType bit_xor (vSIMDType a, vSIMDType b) noexcept { return veorq_u64 (a, b); }
  329. static forcedinline vSIMDType bit_notand (vSIMDType a, vSIMDType b) noexcept { return vbicq_u64 (b, a); }
  330. static forcedinline vSIMDType bit_not (vSIMDType a) noexcept { return bit_notand (a, vld1q_u64 ((uint64_t*) kAllBitsSet)); }
  331. static forcedinline vSIMDType min (vSIMDType a, vSIMDType b) noexcept { return fb::min (a, b); }
  332. static forcedinline vSIMDType max (vSIMDType a, vSIMDType b) noexcept { return fb::max (a, b); }
  333. static forcedinline vSIMDType equal (vSIMDType a, vSIMDType b) noexcept { return fb::equal (a, b); }
  334. static forcedinline vSIMDType notEqual (vSIMDType a, vSIMDType b) noexcept { return fb::notEqual (a, b); }
  335. static forcedinline vSIMDType greaterThan (vSIMDType a, vSIMDType b) noexcept { return fb::greaterThan (a, b); }
  336. static forcedinline vSIMDType greaterThanOrEqual (vSIMDType a, vSIMDType b) noexcept { return fb::greaterThanOrEqual (a, b); }
  337. static forcedinline bool allEqual (vSIMDType a, vSIMDType b) noexcept { return (SIMDNativeOps<uint32_t>::sum ((SIMDNativeOps<uint32_t>::vSIMDType) notEqual (a, b)) == 0); }
  338. static forcedinline vSIMDType multiplyAdd (vSIMDType a, vSIMDType b, vSIMDType c) noexcept { return fb::multiplyAdd (a, b, c); }
  339. static forcedinline uint64_t sum (vSIMDType a) noexcept { return fb::sum (a); }
  340. static forcedinline vSIMDType truncate (vSIMDType a) noexcept { return a; }
  341. };
  342. //==============================================================================
  343. /** Single-precision floating point NEON intrinsics.
  344. @tags{DSP}
  345. */
  346. template <>
  347. struct SIMDNativeOps<float>
  348. {
  349. //==============================================================================
  350. using vSIMDType = float32x4_t;
  351. using vMaskType = uint32x4_t;
  352. using fb = SIMDFallbackOps<float, vSIMDType>;
  353. //==============================================================================
  354. DECLARE_NEON_SIMD_CONST (int32_t, kAllBitsSet);
  355. DECLARE_NEON_SIMD_CONST (int32_t, kEvenHighBit);
  356. DECLARE_NEON_SIMD_CONST (float, kOne);
  357. //==============================================================================
  358. static forcedinline vSIMDType expand (float s) noexcept { return vdupq_n_f32 (s); }
  359. static forcedinline vSIMDType load (const float* a) noexcept { return vld1q_f32 (a); }
  360. static forcedinline float get (vSIMDType v, size_t i) noexcept { return v[i]; }
  361. static forcedinline vSIMDType set (vSIMDType v, size_t i, float s) noexcept { v[i] = s; return v; }
  362. static forcedinline void store (vSIMDType value, float* a) noexcept { vst1q_f32 (a, value); }
  363. static forcedinline vSIMDType add (vSIMDType a, vSIMDType b) noexcept { return vaddq_f32 (a, b); }
  364. static forcedinline vSIMDType sub (vSIMDType a, vSIMDType b) noexcept { return vsubq_f32 (a, b); }
  365. static forcedinline vSIMDType mul (vSIMDType a, vSIMDType b) noexcept { return vmulq_f32 (a, b); }
  366. static forcedinline vSIMDType bit_and (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vandq_u32 ((vMaskType) a, (vMaskType) b); }
  367. static forcedinline vSIMDType bit_or (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vorrq_u32 ((vMaskType) a, (vMaskType) b); }
  368. static forcedinline vSIMDType bit_xor (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) veorq_u32 ((vMaskType) a, (vMaskType) b); }
  369. static forcedinline vSIMDType bit_notand (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vbicq_u32 ((vMaskType) b, (vMaskType) a); }
  370. static forcedinline vSIMDType bit_not (vSIMDType a) noexcept { return bit_notand (a, vld1q_f32 ((float*) kAllBitsSet)); }
  371. static forcedinline vSIMDType min (vSIMDType a, vSIMDType b) noexcept { return vminq_f32 (a, b); }
  372. static forcedinline vSIMDType max (vSIMDType a, vSIMDType b) noexcept { return vmaxq_f32 (a, b); }
  373. static forcedinline vSIMDType equal (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vceqq_f32 (a, b); }
  374. static forcedinline vSIMDType notEqual (vSIMDType a, vSIMDType b) noexcept { return bit_not (equal (a, b)); }
  375. static forcedinline vSIMDType greaterThan (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vcgtq_f32 (a, b); }
  376. static forcedinline vSIMDType greaterThanOrEqual (vSIMDType a, vSIMDType b) noexcept { return (vSIMDType) vcgeq_f32 (a, b); }
  377. static forcedinline bool allEqual (vSIMDType a, vSIMDType b) noexcept { return (SIMDNativeOps<uint32_t>::sum ((SIMDNativeOps<uint32_t>::vSIMDType) notEqual (a, b)) == 0); }
  378. static forcedinline vSIMDType multiplyAdd (vSIMDType a, vSIMDType b, vSIMDType c) noexcept { return vmlaq_f32 (a, b, c); }
  379. static forcedinline vSIMDType dupeven (vSIMDType a) noexcept { return fb::shuffle<(0 << 0) | (0 << 2) | (2 << 4) | (2 << 6)> (a); }
  380. static forcedinline vSIMDType dupodd (vSIMDType a) noexcept { return fb::shuffle<(1 << 0) | (1 << 2) | (3 << 4) | (3 << 6)> (a); }
  381. static forcedinline vSIMDType swapevenodd (vSIMDType a) noexcept { return fb::shuffle<(1 << 0) | (0 << 2) | (3 << 4) | (2 << 6)> (a); }
  382. static forcedinline vSIMDType oddevensum (vSIMDType a) noexcept { return add (fb::shuffle<(2 << 0) | (3 << 2) | (0 << 4) | (1 << 6)> (a), a); }
  383. static forcedinline vSIMDType truncate (vSIMDType a) noexcept { return vcvtq_f32_s32 (vcvtq_s32_f32 (a)); }
  384. //==============================================================================
  385. static forcedinline vSIMDType cmplxmul (vSIMDType a, vSIMDType b) noexcept
  386. {
  387. vSIMDType rr_ir = mul (a, dupeven (b));
  388. vSIMDType ii_ri = mul (swapevenodd (a), dupodd (b));
  389. return add (rr_ir, bit_xor (ii_ri, vld1q_f32 ((float*) kEvenHighBit)));
  390. }
  391. static forcedinline float sum (vSIMDType a) noexcept
  392. {
  393. auto rr = vadd_f32 (vget_high_f32 (a), vget_low_f32 (a));
  394. return vget_lane_f32 (vpadd_f32 (rr, rr), 0);
  395. }
  396. };
  397. //==============================================================================
  398. /** Double-precision floating point NEON intrinsics does not exist in NEON
  399. so we need to emulate this.
  400. @tags{DSP}
  401. */
  402. template <>
  403. struct SIMDNativeOps<double>
  404. {
  405. //==============================================================================
  406. using vSIMDType = struct { double v[2]; };
  407. using fb = SIMDFallbackOps<double, vSIMDType>;
  408. static forcedinline vSIMDType expand (double s) noexcept { return {{s, s}}; }
  409. static forcedinline vSIMDType load (const double* a) noexcept { return {{a[0], a[1]}}; }
  410. static forcedinline void store (vSIMDType v, double* a) noexcept { a[0] = v.v[0]; a[1] = v.v[1]; }
  411. static forcedinline double get (vSIMDType v, size_t i) noexcept { return v.v[i]; }
  412. static forcedinline vSIMDType set (vSIMDType v, size_t i, double s) noexcept { v.v[i] = s; return v; }
  413. static forcedinline vSIMDType add (vSIMDType a, vSIMDType b) noexcept { return {{a.v[0] + b.v[0], a.v[1] + b.v[1]}}; }
  414. static forcedinline vSIMDType sub (vSIMDType a, vSIMDType b) noexcept { return {{a.v[0] - b.v[0], a.v[1] - b.v[1]}}; }
  415. static forcedinline vSIMDType mul (vSIMDType a, vSIMDType b) noexcept { return {{a.v[0] * b.v[0], a.v[1] * b.v[1]}}; }
  416. static forcedinline vSIMDType bit_and (vSIMDType a, vSIMDType b) noexcept { return fb::bit_and (a, b); }
  417. static forcedinline vSIMDType bit_or (vSIMDType a, vSIMDType b) noexcept { return fb::bit_or (a, b); }
  418. static forcedinline vSIMDType bit_xor (vSIMDType a, vSIMDType b) noexcept { return fb::bit_xor (a, b); }
  419. static forcedinline vSIMDType bit_notand (vSIMDType a, vSIMDType b) noexcept { return fb::bit_notand (a, b); }
  420. static forcedinline vSIMDType bit_not (vSIMDType a) noexcept { return fb::bit_not (a); }
  421. static forcedinline vSIMDType min (vSIMDType a, vSIMDType b) noexcept { return fb::min (a, b); }
  422. static forcedinline vSIMDType max (vSIMDType a, vSIMDType b) noexcept { return fb::max (a, b); }
  423. static forcedinline vSIMDType equal (vSIMDType a, vSIMDType b) noexcept { return fb::equal (a, b); }
  424. static forcedinline vSIMDType notEqual (vSIMDType a, vSIMDType b) noexcept { return fb::notEqual (a, b); }
  425. static forcedinline vSIMDType greaterThan (vSIMDType a, vSIMDType b) noexcept { return fb::greaterThan (a, b); }
  426. static forcedinline vSIMDType greaterThanOrEqual (vSIMDType a, vSIMDType b) noexcept { return fb::greaterThanOrEqual (a, b); }
  427. static forcedinline bool allEqual (vSIMDType a, vSIMDType b) noexcept { return fb::allEqual (a, b); }
  428. static forcedinline vSIMDType multiplyAdd (vSIMDType a, vSIMDType b, vSIMDType c) noexcept { return fb::multiplyAdd (a, b, c); }
  429. static forcedinline vSIMDType cmplxmul (vSIMDType a, vSIMDType b) noexcept { return fb::cmplxmul (a, b); }
  430. static forcedinline double sum (vSIMDType a) noexcept { return fb::sum (a); }
  431. static forcedinline vSIMDType oddevensum (vSIMDType a) noexcept { return a; }
  432. static forcedinline vSIMDType truncate (vSIMDType a) noexcept { return fb::truncate (a); }
  433. };
  434. #endif
  435. JUCE_END_IGNORE_WARNINGS_GCC_LIKE
  436. } // namespace dsp
  437. } // namespace juce