arm_negate_f32.c 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192
  1. /* ----------------------------------------------------------------------
  2. * Project: CMSIS DSP Library
  3. * Title: arm_negate_f32.c
  4. * Description: Negates floating-point vectors
  5. *
  6. * $Date: 18. March 2019
  7. * $Revision: V1.6.0
  8. *
  9. * Target Processor: Cortex-M cores
  10. * -------------------------------------------------------------------- */
  11. /*
  12. * Copyright (C) 2010-2019 ARM Limited or its affiliates. All rights reserved.
  13. *
  14. * SPDX-License-Identifier: Apache-2.0
  15. *
  16. * Licensed under the Apache License, Version 2.0 (the License); you may
  17. * not use this file except in compliance with the License.
  18. * You may obtain a copy of the License at
  19. *
  20. * www.apache.org/licenses/LICENSE-2.0
  21. *
  22. * Unless required by applicable law or agreed to in writing, software
  23. * distributed under the License is distributed on an AS IS BASIS, WITHOUT
  24. * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  25. * See the License for the specific language governing permissions and
  26. * limitations under the License.
  27. */
  28. #include "arm_math.h"
  29. /**
  30. @ingroup groupMath
  31. */
  32. /**
  33. @defgroup BasicNegate Vector Negate
  34. Negates the elements of a vector.
  35. <pre>
  36. pDst[n] = -pSrc[n], 0 <= n < blockSize.
  37. </pre>
  38. The functions support in-place computation allowing the source and
  39. destination pointers to reference the same memory buffer.
  40. There are separate functions for floating-point, Q7, Q15, and Q31 data types.
  41. */
  42. /**
  43. @addtogroup BasicNegate
  44. @{
  45. */
  46. /**
  47. @brief Negates the elements of a floating-point vector.
  48. @param[in] pSrc points to input vector.
  49. @param[out] pDst points to output vector.
  50. @param[in] blockSize number of samples in each vector.
  51. @return none
  52. */
  53. #if defined(ARM_MATH_MVEF) && !defined(ARM_MATH_AUTOVECTORIZE)
  54. #include "arm_helium_utils.h"
  55. void arm_negate_f32(
  56. const float32_t * pSrc,
  57. float32_t * pDst,
  58. uint32_t blockSize)
  59. {
  60. uint32_t blkCnt; /* Loop counter */
  61. f32x4_t vec1;
  62. f32x4_t res;
  63. /* Compute 4 outputs at a time */
  64. blkCnt = blockSize >> 2U;
  65. while (blkCnt > 0U)
  66. {
  67. /* C = |A| */
  68. /* Calculate absolute values and then store the results in the destination buffer. */
  69. vec1 = vld1q(pSrc);
  70. res = vnegq(vec1);
  71. vst1q(pDst, res);
  72. /* Increment pointers */
  73. pSrc += 4;
  74. pDst += 4;
  75. /* Decrement the loop counter */
  76. blkCnt--;
  77. }
  78. /* Tail */
  79. blkCnt = blockSize & 0x3;
  80. if (blkCnt > 0U)
  81. {
  82. /* C = |A| */
  83. mve_pred16_t p0 = vctp32q(blkCnt);
  84. vec1 = vld1q((float32_t const *) pSrc);
  85. vstrwq_p(pDst, vnegq(vec1), p0);
  86. }
  87. }
  88. #else
  89. void arm_negate_f32(
  90. const float32_t * pSrc,
  91. float32_t * pDst,
  92. uint32_t blockSize)
  93. {
  94. uint32_t blkCnt; /* Loop counter */
  95. #if defined(ARM_MATH_NEON_EXPERIMENTAL) && !defined(ARM_MATH_AUTOVECTORIZE)
  96. f32x4_t vec1;
  97. f32x4_t res;
  98. /* Compute 4 outputs at a time */
  99. blkCnt = blockSize >> 2U;
  100. while (blkCnt > 0U)
  101. {
  102. /* C = -A */
  103. /* Negate and then store the results in the destination buffer. */
  104. vec1 = vld1q_f32(pSrc);
  105. res = vnegq_f32(vec1);
  106. vst1q_f32(pDst, res);
  107. /* Increment pointers */
  108. pSrc += 4;
  109. pDst += 4;
  110. /* Decrement the loop counter */
  111. blkCnt--;
  112. }
  113. /* Tail */
  114. blkCnt = blockSize & 0x3;
  115. #else
  116. #if defined (ARM_MATH_LOOPUNROLL) && !defined(ARM_MATH_AUTOVECTORIZE)
  117. /* Loop unrolling: Compute 4 outputs at a time */
  118. blkCnt = blockSize >> 2U;
  119. while (blkCnt > 0U)
  120. {
  121. /* C = -A */
  122. /* Negate and store result in destination buffer. */
  123. *pDst++ = -*pSrc++;
  124. *pDst++ = -*pSrc++;
  125. *pDst++ = -*pSrc++;
  126. *pDst++ = -*pSrc++;
  127. /* Decrement loop counter */
  128. blkCnt--;
  129. }
  130. /* Loop unrolling: Compute remaining outputs */
  131. blkCnt = blockSize % 0x4U;
  132. #else
  133. /* Initialize blkCnt with number of samples */
  134. blkCnt = blockSize;
  135. #endif /* #if defined (ARM_MATH_LOOPUNROLL) */
  136. #endif /* #if defined(ARM_MATH_NEON_EXPERIMENTAL) */
  137. while (blkCnt > 0U)
  138. {
  139. /* C = -A */
  140. /* Negate and store result in destination buffer. */
  141. *pDst++ = -*pSrc++;
  142. /* Decrement loop counter */
  143. blkCnt--;
  144. }
  145. }
  146. #endif /* defined(ARM_MATH_MVEF) && !defined(ARM_MATH_AUTOVECTORIZE) */
  147. /**
  148. @} end of BasicNegate group
  149. */