arm_scale_q7.c 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
  1. /* ----------------------------------------------------------------------
  2. * Project: CMSIS DSP Library
  3. * Title: arm_scale_q7.c
  4. * Description: Multiplies a Q7 vector by a scalar
  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 BasicScale
  34. @{
  35. */
  36. /**
  37. @brief Multiplies a Q7 vector by a scalar.
  38. @param[in] pSrc points to the input vector
  39. @param[in] scaleFract fractional portion of the scale value
  40. @param[in] shift number of bits to shift the result by
  41. @param[out] pDst points to the output vector
  42. @param[in] blockSize number of samples in each vector
  43. @return none
  44. @par Scaling and Overflow Behavior
  45. The input data <code>*pSrc</code> and <code>scaleFract</code> are in 1.7 format.
  46. These are multiplied to yield a 2.14 intermediate result and this is shifted with saturation to 1.7 format.
  47. */
  48. void arm_scale_q7(
  49. const q7_t * pSrc,
  50. q7_t scaleFract,
  51. int8_t shift,
  52. q7_t * pDst,
  53. uint32_t blockSize)
  54. {
  55. uint32_t blkCnt; /* Loop counter */
  56. int8_t kShift = 7 - shift; /* Shift to apply after scaling */
  57. #if defined (ARM_MATH_LOOPUNROLL)
  58. #if defined (ARM_MATH_DSP)
  59. q7_t in1, in2, in3, in4; /* Temporary input variables */
  60. q7_t out1, out2, out3, out4; /* Temporary output variables */
  61. #endif
  62. /* Loop unrolling: Compute 4 outputs at a time */
  63. blkCnt = blockSize >> 2U;
  64. while (blkCnt > 0U)
  65. {
  66. /* C = A * scale */
  67. #if defined (ARM_MATH_DSP)
  68. /* Reading 4 inputs from memory */
  69. in1 = *pSrc++;
  70. in2 = *pSrc++;
  71. in3 = *pSrc++;
  72. in4 = *pSrc++;
  73. /* Scale inputs and store result in the temporary variable. */
  74. out1 = (q7_t) (__SSAT(((in1) * scaleFract) >> kShift, 8));
  75. out2 = (q7_t) (__SSAT(((in2) * scaleFract) >> kShift, 8));
  76. out3 = (q7_t) (__SSAT(((in3) * scaleFract) >> kShift, 8));
  77. out4 = (q7_t) (__SSAT(((in4) * scaleFract) >> kShift, 8));
  78. /* Pack and store result in destination buffer (in single write) */
  79. write_q7x4_ia (&pDst, __PACKq7(out1, out2, out3, out4));
  80. #else
  81. *pDst++ = (q7_t) (__SSAT((((q15_t) *pSrc++ * scaleFract) >> kShift), 8));
  82. *pDst++ = (q7_t) (__SSAT((((q15_t) *pSrc++ * scaleFract) >> kShift), 8));
  83. *pDst++ = (q7_t) (__SSAT((((q15_t) *pSrc++ * scaleFract) >> kShift), 8));
  84. *pDst++ = (q7_t) (__SSAT((((q15_t) *pSrc++ * scaleFract) >> kShift), 8));
  85. #endif
  86. /* Decrement loop counter */
  87. blkCnt--;
  88. }
  89. /* Loop unrolling: Compute remaining outputs */
  90. blkCnt = blockSize % 0x4U;
  91. #else
  92. /* Initialize blkCnt with number of samples */
  93. blkCnt = blockSize;
  94. #endif /* #if defined (ARM_MATH_LOOPUNROLL) */
  95. while (blkCnt > 0U)
  96. {
  97. /* C = A * scale */
  98. /* Scale input and store result in destination buffer. */
  99. *pDst++ = (q7_t) (__SSAT((((q15_t) *pSrc++ * scaleFract) >> kShift), 8));
  100. /* Decrement loop counter */
  101. blkCnt--;
  102. }
  103. }
  104. /**
  105. @} end of BasicScale group
  106. */