|| || || || || || || ||

---

MatSum.h

Go to the documentation of this file.

#ifndef Impala_Core_Matrix_MatSum_h
#define Impala_Core_Matrix_MatSum_h

#include "Core/Matrix/MatFunc.h"
#include "Core/Array/SetVal.h"

namespace Impala
{
namespace Core
{
namespace Matrix
{

/* sum over the rows (result = size(R,1)) */
template<class ArrayT>
inline ArrayT*
MatSumAxis1(ArrayT* m)
{
    ArrayT* res = MatCreate<ArrayT>(MatNrRow(m), 1);
    SetVal(res, res, 0.0);
    for (int i=0 ; i<MatNrRow(m) ; i++)
        for (int j=0 ; j<MatNrCol(m) ; j++)
            *MatE(res, i, 0) = *MatE(res, i, 0) + *MatE(m, i, j);
    return res;
}

template<class ArrayT>
inline ArrayT*
MatSumRows(ArrayT* m)
{
    return MatSumAxis1(m);
}
    
/* sum over the columns (result = size(1,C)) */
template<class ArrayT>
inline ArrayT*
MatSumAxis0(ArrayT* m)
{
    ArrayT* res = MatCreate<ArrayT>(1, MatNrCol(m));
    SetVal(res, res, 0.0);
    for (int i=0 ; i<MatNrRow(m) ; i++)
        for (int j=0 ; j<MatNrCol(m) ; j++)
            *MatE(res, 0, j) = *MatE(res, 0, j) + *MatE(m, i, j);
    return res;
}

template<class ArrayT>
inline ArrayT*
MatSumColumns(ArrayT* m)
{
    return MatSumAxis0(m);
}

} // namespace Matrix
} // namespace Core
} // namespace Impala

#endif

---