/* ---------------------------------------------------------------------- * Project: CMSIS DSP Library * Title: arm_power_f64.c * Description: Sum of the squares of the elements of a floating-point vector * * $Date: 10 August 2022 * $Revision: V1.10.1 * * Target Processor: Cortex-M and Cortex-A cores * -------------------------------------------------------------------- */ /* * Copyright (C) 2010-2021 ARM Limited or its affiliates. All rights reserved. * * SPDX-License-Identifier: Apache-2.0 * * Licensed under the Apache License, Version 2.0 (the License); you may * not use this file except in compliance with the License. * You may obtain a copy of the License at * * www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an AS IS BASIS, WITHOUT * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #include "dsp/statistics_functions.h" /** @ingroup groupStats */ /** @addtogroup power @{ */ /** @brief Sum of the squares of the elements of a floating-point vector. @param[in] pSrc points to the input vector @param[in] blockSize number of samples in input vector @param[out] pResult sum of the squares value returned here */ #if defined(ARM_MATH_NEON) && defined(__aarch64__) ARM_DSP_ATTRIBUTE void arm_power_f64( const float64_t * pSrc, uint32_t blockSize, float64_t * pResult) { uint32_t blkCnt; /* Loop counter */ float64_t sum = 0.; /* Temporary result storage */ float64x2_t sumV ; /* Temporary variable to store input value */ sumV = vdupq_n_f64(0.0); float64x2_t pSrcV; /* Initialize blkCnt with number of samples */ blkCnt = blockSize >> 1U; while (blkCnt > 0U) { /* C = A[0] * A[0] + A[1] * A[1] + ... + A[blockSize-1] * A[blockSize-1] */ /* Compute Power and store result in a temporary variable, sum. */ pSrcV = vld1q_f64(pSrc); sumV = vmlaq_f64(sumV, pSrcV, pSrcV); pSrc+= 2 ; /* Decrement loop counter */ blkCnt--; } sum = vaddvq_f64(sumV); float64_t in; blkCnt = blockSize & 1; while (blkCnt > 0U) { /* C = A[0] * A[0] + A[1] * A[1] + ... + A[blockSize-1] * A[blockSize-1] */ /* Compute Power and store result in a temporary variable, sum. */ in = *pSrc++; sum += in * in; /* Decrement loop counter */ blkCnt--; } /* Store result to destination */ *pResult = sum; } #else ARM_DSP_ATTRIBUTE void arm_power_f64( const float64_t * pSrc, uint32_t blockSize, float64_t * pResult) { uint32_t blkCnt; /* Loop counter */ float64_t sum = 0.; /* Temporary result storage */ float64_t in; /* Temporary variable to store input value */ /* Initialize blkCnt with number of samples */ blkCnt = blockSize; while (blkCnt > 0U) { /* C = A[0] * A[0] + A[1] * A[1] + ... + A[blockSize-1] * A[blockSize-1] */ /* Compute Power and store result in a temporary variable, sum. */ in = *pSrc++; sum += in * in; /* Decrement loop counter */ blkCnt--; } /* Store result to destination */ *pResult = sum; } #endif /** @} end of power group */