1  /* ----------------------------------------------------------------------
  2   * Project:      CMSIS DSP Library
  3   * Title:        arm_fill_q15.c
  4   * Description:  Fills a constant value into a Q15 vector
  5   *
  6   * $Date:        23 April 2021
  7   * $Revision:    V1.9.0
  8   *
  9   * Target Processor: Cortex-M and Cortex-A cores
 10   * -------------------------------------------------------------------- */
 11  /*
 12   * Copyright (C) 2010-2021 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  
 29  #include "dsp/support_functions.h"
 30  
 31  /**
 32    @ingroup groupSupport
 33   */
 34  
 35  /**
 36    @addtogroup Fill
 37    @{
 38   */
 39  
 40  /**
 41    @brief         Fills a constant value into a Q15 vector.
 42    @param[in]     value      input value to be filled
 43    @param[out]    pDst       points to output vector
 44    @param[in]     blockSize  number of samples in each vector
 45    @return        none
 46   */
 47  #if defined(ARM_MATH_MVEI) && !defined(ARM_MATH_AUTOVECTORIZE)
 48  void arm_fill_q15(
 49    q15_t value,
 50    q15_t * pDst,
 51    uint32_t blockSize)
 52  {
 53    uint32_t blkCnt;  
 54    blkCnt = blockSize >> 3;
 55    while (blkCnt > 0U)
 56    {
 57  
 58          vstrhq_s16(pDst,vdupq_n_s16(value));
 59          /*
 60           * Decrement the blockSize loop counter
 61           * Advance vector source and destination pointers
 62           */
 63          pDst += 8;
 64          blkCnt --;
 65      }
 66  
 67    blkCnt = blockSize & 7;
 68    while (blkCnt > 0U)
 69    {
 70      /* C = value */
 71  
 72      /* Fill value in destination buffer */
 73      *pDst++ = value;
 74  
 75      /* Decrement loop counter */
 76      blkCnt--;
 77    }
 78  }
 79  
 80  #else
 81  void arm_fill_q15(
 82    q15_t value,
 83    q15_t * pDst,
 84    uint32_t blockSize)
 85  {
 86    uint32_t blkCnt;                               /* Loop counter */
 87  
 88  #if defined (ARM_MATH_LOOPUNROLL)
 89    q31_t packedValue;                             /* value packed to 32 bits */
 90  
 91    /* Packing two 16 bit values to 32 bit value in order to use SIMD */
 92    packedValue = __PKHBT(value, value, 16U);
 93  
 94    /* Loop unrolling: Compute 4 outputs at a time */
 95    blkCnt = blockSize >> 2U;
 96  
 97    while (blkCnt > 0U)
 98    {
 99      /* C = value */
100  
101      /* fill 2 times 2 samples at a time */
102      write_q15x2_ia (&pDst, packedValue);
103      write_q15x2_ia (&pDst, packedValue);
104  
105      /* Decrement loop counter */
106      blkCnt--;
107    }
108  
109    /* Loop unrolling: Compute remaining outputs */
110    blkCnt = blockSize % 0x4U;
111  
112  #else
113  
114    /* Initialize blkCnt with number of samples */
115    blkCnt = blockSize;
116  
117  #endif /* #if defined (ARM_MATH_LOOPUNROLL) */
118  
119    while (blkCnt > 0U)
120    {
121      /* C = value */
122  
123      /* Fill value in destination buffer */
124      *pDst++ = value;
125  
126      /* Decrement loop counter */
127      blkCnt--;
128    }
129  }
130  #endif /* defined(ARM_MATH_MVEI) */
131  
132  /**
133    @} end of Fill group
134   */