arm_mult_q31.c 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119
  1. /* ----------------------------------------------------------------------
  2. * Project: CMSIS DSP Library
  3. * Title: arm_mult_q31.c
  4. * Description: Q31 vector multiplication
  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 BasicMult
  34. @{
  35. */
  36. /**
  37. @brief Q31 vector multiplication.
  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 Q31 range[0x80000000 0x7FFFFFFF] are saturated.
  46. */
  47. void arm_mult_q31(
  48. const q31_t * pSrcA,
  49. const q31_t * pSrcB,
  50. q31_t * pDst,
  51. uint32_t blockSize)
  52. {
  53. uint32_t blkCnt; /* Loop counter */
  54. q31_t out; /* Temporary output variable */
  55. #if defined (ARM_MATH_LOOPUNROLL)
  56. /* Loop unrolling: Compute 4 outputs at a time */
  57. blkCnt = blockSize >> 2U;
  58. while (blkCnt > 0U)
  59. {
  60. /* C = A * B */
  61. /* Multiply inputs and store result in destination buffer. */
  62. out = ((q63_t) *pSrcA++ * *pSrcB++) >> 32;
  63. out = __SSAT(out, 31);
  64. *pDst++ = out << 1U;
  65. out = ((q63_t) *pSrcA++ * *pSrcB++) >> 32;
  66. out = __SSAT(out, 31);
  67. *pDst++ = out << 1U;
  68. out = ((q63_t) *pSrcA++ * *pSrcB++) >> 32;
  69. out = __SSAT(out, 31);
  70. *pDst++ = out << 1U;
  71. out = ((q63_t) *pSrcA++ * *pSrcB++) >> 32;
  72. out = __SSAT(out, 31);
  73. *pDst++ = out << 1U;
  74. /* Decrement loop counter */
  75. blkCnt--;
  76. }
  77. /* Loop unrolling: Compute remaining outputs */
  78. blkCnt = blockSize % 0x4U;
  79. #else
  80. /* Initialize blkCnt with number of samples */
  81. blkCnt = blockSize;
  82. #endif /* #if defined (ARM_MATH_LOOPUNROLL) */
  83. while (blkCnt > 0U)
  84. {
  85. /* C = A * B */
  86. /* Multiply inputs and store result in destination buffer. */
  87. out = ((q63_t) *pSrcA++ * *pSrcB++) >> 32;
  88. out = __SSAT(out, 31);
  89. *pDst++ = out << 1U;
  90. /* Decrement loop counter */
  91. blkCnt--;
  92. }
  93. }
  94. /**
  95. @} end of BasicMult group
  96. */