arm_sub_q7.c 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158
  1. /* ----------------------------------------------------------------------
  2. * Project: CMSIS DSP Library
  3. * Title: arm_sub_q7.c
  4. * Description: Q7 vector subtraction
  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. @addtogroup BasicSub
  34. @{
  35. */
  36. /**
  37. @brief Q7 vector subtraction.
  38. @param[in] pSrcA points to the first input vector
  39. @param[in] pSrcB points to the second input vector
  40. @param[out] pDst points to the output vector
  41. @param[in] blockSize number of samples in each vector
  42. @return none
  43. @par Scaling and Overflow Behavior
  44. The function uses saturating arithmetic.
  45. Results outside of the allowable Q7 range [0x80 0x7F] will be saturated.
  46. */
  47. #if defined(ARM_MATH_MVEI)
  48. #include "arm_helium_utils.h"
  49. void arm_sub_q7(
  50. const q7_t * pSrcA,
  51. const q7_t * pSrcB,
  52. q7_t * pDst,
  53. uint32_t blockSize)
  54. {
  55. uint32_t blkCnt; /* loop counters */
  56. q7x16_t vecA;
  57. q7x16_t vecB;
  58. /* Compute 16 outputs at a time */
  59. blkCnt = blockSize >> 4;
  60. while (blkCnt > 0U)
  61. {
  62. /*
  63. * C = A - B
  64. * Subtract and then store the results in the destination buffer.
  65. */
  66. vecA = vld1q(pSrcA);
  67. vecB = vld1q(pSrcB);
  68. vst1q(pDst, vqsubq(vecA, vecB));
  69. /*
  70. * Decrement the blockSize loop counter
  71. */
  72. blkCnt--;
  73. /*
  74. * advance vector source and destination pointers
  75. */
  76. pSrcA += 16;
  77. pSrcB += 16;
  78. pDst += 16;
  79. }
  80. /*
  81. * tail
  82. */
  83. blkCnt = blockSize & 0xF;
  84. if (blkCnt > 0U)
  85. {
  86. mve_pred16_t p0 = vctp8q(blkCnt);
  87. vecA = vld1q(pSrcA);
  88. vecB = vld1q(pSrcB);
  89. vstrbq_p(pDst, vqsubq(vecA, vecB), p0);
  90. }
  91. }
  92. #else
  93. void arm_sub_q7(
  94. const q7_t * pSrcA,
  95. const q7_t * pSrcB,
  96. q7_t * pDst,
  97. uint32_t blockSize)
  98. {
  99. uint32_t blkCnt; /* Loop counter */
  100. #if defined (ARM_MATH_LOOPUNROLL)
  101. /* Loop unrolling: Compute 4 outputs at a time */
  102. blkCnt = blockSize >> 2U;
  103. while (blkCnt > 0U)
  104. {
  105. /* C = A - B */
  106. #if defined (ARM_MATH_DSP)
  107. /* Subtract and store result in destination buffer (4 samples at a time). */
  108. write_q7x4_ia (&pDst, __QSUB8(read_q7x4_ia ((q7_t **) &pSrcA), read_q7x4_ia ((q7_t **) &pSrcB)));
  109. #else
  110. *pDst++ = (q7_t) __SSAT((q15_t) *pSrcA++ - *pSrcB++, 8);
  111. *pDst++ = (q7_t) __SSAT((q15_t) *pSrcA++ - *pSrcB++, 8);
  112. *pDst++ = (q7_t) __SSAT((q15_t) *pSrcA++ - *pSrcB++, 8);
  113. *pDst++ = (q7_t) __SSAT((q15_t) *pSrcA++ - *pSrcB++, 8);
  114. #endif
  115. /* Decrement loop counter */
  116. blkCnt--;
  117. }
  118. /* Loop unrolling: Compute remaining outputs */
  119. blkCnt = blockSize % 0x4U;
  120. #else
  121. /* Initialize blkCnt with number of samples */
  122. blkCnt = blockSize;
  123. #endif /* #if defined (ARM_MATH_LOOPUNROLL) */
  124. while (blkCnt > 0U)
  125. {
  126. /* C = A - B */
  127. /* Subtract and store result in destination buffer. */
  128. *pDst++ = (q7_t) __SSAT((q15_t) *pSrcA++ - *pSrcB++, 8);
  129. /* Decrement loop counter */
  130. blkCnt--;
  131. }
  132. }
  133. #endif /* defined(ARM_MATH_MVEI) */
  134. /**
  135. @} end of BasicSub group
  136. */