00001 /* ---------------------------------------------------------------------- 00002 * Copyright (C) 2010 ARM Limited. All rights reserved. 00003 * 00004 * $Date: 15. July 2011 00005 * $Revision: V1.0.10 00006 * 00007 * Project: CMSIS DSP Library 00008 * Title: arm_mat_add_q31.c 00009 * 00010 * Description: Q31 matrix addition 00011 * 00012 * Target Processor: Cortex-M4/Cortex-M3/Cortex-M0 00013 * 00014 * Version 1.0.10 2011/7/15 00015 * Big Endian support added and Merged M0 and M3/M4 Source code. 00016 * 00017 * Version 1.0.3 2010/11/29 00018 * Re-organized the CMSIS folders and updated documentation. 00019 * 00020 * Version 1.0.2 2010/11/11 00021 * Documentation updated. 00022 * 00023 * Version 1.0.1 2010/10/05 00024 * Production release and review comments incorporated. 00025 * 00026 * Version 1.0.0 2010/09/20 00027 * Production release and review comments incorporated. 00028 * 00029 * Version 0.0.5 2010/04/26 00030 * incorporated review comments and updated with latest CMSIS layer 00031 * 00032 * Version 0.0.3 2010/03/10 00033 * Initial version 00034 * -------------------------------------------------------------------- */ 00035 00036 #include "arm_math.h" 00037 00061 arm_status arm_mat_add_q31( 00062 const arm_matrix_instance_q31 * pSrcA, 00063 const arm_matrix_instance_q31 * pSrcB, 00064 arm_matrix_instance_q31 * pDst) 00065 { 00066 q31_t *pIn1 = pSrcA->pData; /* input data matrix pointer A */ 00067 q31_t *pIn2 = pSrcB->pData; /* input data matrix pointer B */ 00068 q31_t *pOut = pDst->pData; /* output data matrix pointer */ 00069 uint32_t numSamples; /* total number of elements in the matrix */ 00070 uint32_t blkCnt; /* loop counters */ 00071 arm_status status; /* status of matrix addition */ 00072 00073 #ifdef ARM_MATH_MATRIX_CHECK 00074 00075 00076 /* Check for matrix mismatch condition */ 00077 if((pSrcA->numRows != pSrcB->numRows) || 00078 (pSrcA->numCols != pSrcB->numCols) || 00079 (pSrcA->numRows != pDst->numRows) || (pSrcA->numCols != pDst->numCols)) 00080 { 00081 /* Set status as ARM_MATH_SIZE_MISMATCH */ 00082 status = ARM_MATH_SIZE_MISMATCH; 00083 } 00084 else 00085 #endif /* #ifdef ARM_MATH_MATRIX_CHECK */ 00086 00087 { 00088 /* Total number of samples in the input matrix */ 00089 numSamples = (uint32_t) pSrcA->numRows * pSrcA->numCols; 00090 00091 #ifndef ARM_MATH_CM0 00092 00093 /* Run the below code for Cortex-M4 and Cortex-M3 */ 00094 00095 /* Loop Unrolling */ 00096 blkCnt = numSamples >> 2u; 00097 00098 00099 /* First part of the processing with loop unrolling. Compute 4 outputs at a time. 00100 ** a second loop below computes the remaining 1 to 3 samples. */ 00101 while(blkCnt > 0u) 00102 { 00103 /* C(m,n) = A(m,n) + B(m,n) */ 00104 /* Add, saturate and then store the results in the destination buffer. */ 00105 *pOut++ = __QADD(*pIn1++, *pIn2++); 00106 *pOut++ = __QADD(*pIn1++, *pIn2++); 00107 *pOut++ = __QADD(*pIn1++, *pIn2++); 00108 *pOut++ = __QADD(*pIn1++, *pIn2++); 00109 00110 /* Decrement the loop counter */ 00111 blkCnt--; 00112 } 00113 00114 /* If the numSamples is not a multiple of 4, compute any remaining output samples here. 00115 ** No loop unrolling is used. */ 00116 blkCnt = numSamples % 0x4u; 00117 00118 while(blkCnt > 0u) 00119 { 00120 /* C(m,n) = A(m,n) + B(m,n) */ 00121 /* Add, saturate and then store the results in the destination buffer. */ 00122 *pOut++ = __QADD(*pIn1++, *pIn2++); 00123 00124 /* Decrement the loop counter */ 00125 blkCnt--; 00126 } 00127 00128 #else 00129 00130 /* Run the below code for Cortex-M0 */ 00131 00132 /* Initialize blkCnt with number of samples */ 00133 blkCnt = numSamples; 00134 00135 while(blkCnt > 0u) 00136 { 00137 /* C(m,n) = A(m,n) + B(m,n) */ 00138 /* Add, saturate and then store the results in the destination buffer. */ 00139 *pOut++ = clip_q63_to_q31(((q63_t) (*pIn1++)) + (*pIn2++)); 00140 00141 /* Decrement the loop counter */ 00142 blkCnt--; 00143 } 00144 00145 #endif /* #ifndef ARM_MATH_CM0 */ 00146 00147 /* set status as ARM_MATH_SUCCESS */ 00148 status = ARM_MATH_SUCCESS; 00149 } 00150 00151 /* Return to application */ 00152 return (status); 00153 } 00154