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  */
46 #if defined(ARM_MATH_MVEI) && !defined(ARM_MATH_AUTOVECTORIZE)
arm_fill_q15(q15_t value,q15_t * pDst,uint32_t blockSize)47 ARM_DSP_ATTRIBUTE void arm_fill_q15(
48   q15_t value,
49   q15_t * pDst,
50   uint32_t blockSize)
51 {
52   uint32_t blkCnt;
53   blkCnt = blockSize >> 3;
54   while (blkCnt > 0U)
55   {
56 
57         vstrhq_s16(pDst,vdupq_n_s16(value));
58         /*
59          * Decrement the blockSize loop counter
60          * Advance vector source and destination pointers
61          */
62         pDst += 8;
63         blkCnt --;
64     }
65 
66   blkCnt = blockSize & 7;
67   while (blkCnt > 0U)
68   {
69     /* C = value */
70 
71     /* Fill value in destination buffer */
72     *pDst++ = value;
73 
74     /* Decrement loop counter */
75     blkCnt--;
76   }
77 }
78 
79 #else
arm_fill_q15(q15_t value,q15_t * pDst,uint32_t blockSize)80 ARM_DSP_ATTRIBUTE void arm_fill_q15(
81   q15_t value,
82   q15_t * pDst,
83   uint32_t blockSize)
84 {
85   uint32_t blkCnt;                               /* Loop counter */
86 
87 #if defined (ARM_MATH_LOOPUNROLL)
88   q31_t packedValue;                             /* value packed to 32 bits */
89 
90   /* Packing two 16 bit values to 32 bit value in order to use SIMD */
91   packedValue = __PKHBT(value, value, 16U);
92 
93   /* Loop unrolling: Compute 4 outputs at a time */
94   blkCnt = blockSize >> 2U;
95 
96   while (blkCnt > 0U)
97   {
98     /* C = value */
99 
100     /* fill 2 times 2 samples at a time */
101     write_q15x2_ia (&pDst, packedValue);
102     write_q15x2_ia (&pDst, packedValue);
103 
104     /* Decrement loop counter */
105     blkCnt--;
106   }
107 
108   /* Loop unrolling: Compute remaining outputs */
109   blkCnt = blockSize % 0x4U;
110 
111 #else
112 
113   /* Initialize blkCnt with number of samples */
114   blkCnt = blockSize;
115 
116 #endif /* #if defined (ARM_MATH_LOOPUNROLL) */
117 
118   while (blkCnt > 0U)
119   {
120     /* C = value */
121 
122     /* Fill value in destination buffer */
123     *pDst++ = value;
124 
125     /* Decrement loop counter */
126     blkCnt--;
127   }
128 }
129 #endif /* defined(ARM_MATH_MVEI) */
130 
131 /**
132   @} end of Fill group
133  */
134