arm_mat_sub_f32.c 6.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209
  1. /* ----------------------------------------------------------------------
  2. * Copyright (C) 2010-2014 ARM Limited. All rights reserved.
  3. *
  4. * $Date: 31. July 2014
  5. * $Revision: V1.4.4
  6. *
  7. * Project: CMSIS DSP Library
  8. * Title: arm_mat_sub_f32.c
  9. *
  10. * Description: Floating-point matrix subtraction.
  11. *
  12. * Target Processor: Cortex-M4/Cortex-M3/Cortex-M0
  13. *
  14. * Redistribution and use in source and binary forms, with or without
  15. * modification, are permitted provided that the following conditions
  16. * are met:
  17. * - Redistributions of source code must retain the above copyright
  18. * notice, this list of conditions and the following disclaimer.
  19. * - Redistributions in binary form must reproduce the above copyright
  20. * notice, this list of conditions and the following disclaimer in
  21. * the documentation and/or other materials provided with the
  22. * distribution.
  23. * - Neither the name of ARM LIMITED nor the names of its contributors
  24. * may be used to endorse or promote products derived from this
  25. * software without specific prior written permission.
  26. *
  27. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  28. * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  29. * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  30. * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  31. * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  32. * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
  33. * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  34. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  35. * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  36. * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  37. * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  38. * POSSIBILITY OF SUCH DAMAGE.
  39. * -------------------------------------------------------------------- */
  40. #include "arm_math.h"
  41. /**
  42. * @ingroup groupMatrix
  43. */
  44. /**
  45. * @defgroup MatrixSub Matrix Subtraction
  46. *
  47. * Subtract two matrices.
  48. * \image html MatrixSubtraction.gif "Subraction of two 3 x 3 matrices"
  49. *
  50. * The functions check to make sure that
  51. * <code>pSrcA</code>, <code>pSrcB</code>, and <code>pDst</code> have the same
  52. * number of rows and columns.
  53. */
  54. /**
  55. * @addtogroup MatrixSub
  56. * @{
  57. */
  58. /**
  59. * @brief Floating-point matrix subtraction
  60. * @param[in] *pSrcA points to the first input matrix structure
  61. * @param[in] *pSrcB points to the second input matrix structure
  62. * @param[out] *pDst points to output matrix structure
  63. * @return The function returns either
  64. * <code>ARM_MATH_SIZE_MISMATCH</code> or <code>ARM_MATH_SUCCESS</code> based on the outcome of size checking.
  65. */
  66. arm_status arm_mat_sub_f32(
  67. const arm_matrix_instance_f32 * pSrcA,
  68. const arm_matrix_instance_f32 * pSrcB,
  69. arm_matrix_instance_f32 * pDst)
  70. {
  71. float32_t *pIn1 = pSrcA->pData; /* input data matrix pointer A */
  72. float32_t *pIn2 = pSrcB->pData; /* input data matrix pointer B */
  73. float32_t *pOut = pDst->pData; /* output data matrix pointer */
  74. #ifndef ARM_MATH_CM0_FAMILY
  75. float32_t inA1, inA2, inB1, inB2, out1, out2; /* temporary variables */
  76. #endif // #ifndef ARM_MATH_CM0_FAMILY
  77. uint32_t numSamples; /* total number of elements in the matrix */
  78. uint32_t blkCnt; /* loop counters */
  79. arm_status status; /* status of matrix subtraction */
  80. #ifdef ARM_MATH_MATRIX_CHECK
  81. /* Check for matrix mismatch condition */
  82. if((pSrcA->numRows != pSrcB->numRows) ||
  83. (pSrcA->numCols != pSrcB->numCols) ||
  84. (pSrcA->numRows != pDst->numRows) || (pSrcA->numCols != pDst->numCols))
  85. {
  86. /* Set status as ARM_MATH_SIZE_MISMATCH */
  87. status = ARM_MATH_SIZE_MISMATCH;
  88. }
  89. else
  90. #endif /* #ifdef ARM_MATH_MATRIX_CHECK */
  91. {
  92. /* Total number of samples in the input matrix */
  93. numSamples = (uint32_t) pSrcA->numRows * pSrcA->numCols;
  94. #ifndef ARM_MATH_CM0_FAMILY
  95. /* Run the below code for Cortex-M4 and Cortex-M3 */
  96. /* Loop Unrolling */
  97. blkCnt = numSamples >> 2u;
  98. /* First part of the processing with loop unrolling. Compute 4 outputs at a time.
  99. ** a second loop below computes the remaining 1 to 3 samples. */
  100. while(blkCnt > 0u)
  101. {
  102. /* C(m,n) = A(m,n) - B(m,n) */
  103. /* Subtract and then store the results in the destination buffer. */
  104. /* Read values from source A */
  105. inA1 = pIn1[0];
  106. /* Read values from source B */
  107. inB1 = pIn2[0];
  108. /* Read values from source A */
  109. inA2 = pIn1[1];
  110. /* out = sourceA - sourceB */
  111. out1 = inA1 - inB1;
  112. /* Read values from source B */
  113. inB2 = pIn2[1];
  114. /* Read values from source A */
  115. inA1 = pIn1[2];
  116. /* out = sourceA - sourceB */
  117. out2 = inA2 - inB2;
  118. /* Read values from source B */
  119. inB1 = pIn2[2];
  120. /* Store result in destination */
  121. pOut[0] = out1;
  122. pOut[1] = out2;
  123. /* Read values from source A */
  124. inA2 = pIn1[3];
  125. /* Read values from source B */
  126. inB2 = pIn2[3];
  127. /* out = sourceA - sourceB */
  128. out1 = inA1 - inB1;
  129. /* out = sourceA - sourceB */
  130. out2 = inA2 - inB2;
  131. /* Store result in destination */
  132. pOut[2] = out1;
  133. /* Store result in destination */
  134. pOut[3] = out2;
  135. /* update pointers to process next sampels */
  136. pIn1 += 4u;
  137. pIn2 += 4u;
  138. pOut += 4u;
  139. /* Decrement the loop counter */
  140. blkCnt--;
  141. }
  142. /* If the numSamples is not a multiple of 4, compute any remaining output samples here.
  143. ** No loop unrolling is used. */
  144. blkCnt = numSamples % 0x4u;
  145. #else
  146. /* Run the below code for Cortex-M0 */
  147. /* Initialize blkCnt with number of samples */
  148. blkCnt = numSamples;
  149. #endif /* #ifndef ARM_MATH_CM0_FAMILY */
  150. while(blkCnt > 0u)
  151. {
  152. /* C(m,n) = A(m,n) - B(m,n) */
  153. /* Subtract and then store the results in the destination buffer. */
  154. *pOut++ = (*pIn1++) - (*pIn2++);
  155. /* Decrement the loop counter */
  156. blkCnt--;
  157. }
  158. /* Set status as ARM_MATH_SUCCESS */
  159. status = ARM_MATH_SUCCESS;
  160. }
  161. /* Return to application */
  162. return (status);
  163. }
  164. /**
  165. * @} end of MatrixSub group
  166. */