/home/brandes/workspace/LAMA/src/lama/expression/MatrixExpressions.hpp

Go to the documentation of this file.

#ifndef LAMA_MATRIXEXPRESSIONS_HPP_
#define LAMA_MATRIXEXPRESSIONS_HPP_

#include <lama/matrix/Matrix.hpp>

#include <lama/Vector.hpp>
#include <lama/Scalar.hpp>

#include <lama/expression/Expression.hpp>

namespace lama
{

inline Expression<Scalar,Matrix,Times> operator*(
    const Scalar& scalar,
    const Matrix& matrix)
{
    return Expression<Scalar,Matrix,Times>(scalar,matrix);
}

inline Expression<Scalar,Matrix,Times> operator*(
    const Matrix& matrix,
    const Scalar& scalar)
{
    return Expression<Scalar,Matrix,Times>(scalar,matrix);
}

//A*B
inline Expression<Matrix,Matrix,Times> operator*(const Matrix& m1, const Matrix& m2)
{
    return Expression<Matrix,Matrix,Times>(m1, m2);
}

//alpha times A times B

inline Expression<Scalar,Expression<Matrix,Matrix,Times>,Times> operator*(
    const Matrix& m1,
    const Expression<Scalar,Matrix,Times>& exp)
{
    return Expression<Scalar,Expression<Matrix,Matrix,Times>,Times>(exp.getArg1(),Expression<Matrix,Matrix,Times>(m1,exp.getArg2()) );
}

inline Expression<Scalar,Expression<Matrix,Matrix,Times>,Times> operator*(
    const Expression<Scalar,Matrix,Times>& exp,
    const Matrix& m1)
{
    return Expression<Scalar,Expression<Matrix,Matrix,Times>,Times>( exp.getArg1(), Expression<Matrix,Matrix,Times>(exp.getArg2(),m1) );
}

inline Expression<Scalar,Expression<Matrix,Matrix,Times>,Times> operator*(
    const Scalar& s1,
    const Expression<Matrix,Matrix,Times>& exp)
{
    return Expression<Scalar,Expression<Matrix,Matrix,Times>,Times>( s1, exp );
}

inline Expression<Scalar,Expression<Matrix,Matrix,Times>,Times> operator*(
    const Expression<Matrix,Matrix,Times>& exp,
    const Scalar& s1)
{
    return Expression<Scalar,Expression<Matrix,Matrix,Times>,Times>( s1, exp );
}

//alpha*A*B + beta*C

inline Expression< Expression< Scalar, Expression< Matrix, Matrix, Times>, Times>,
                   Expression< Scalar, Matrix, Times>, Plus> operator+(
    const Expression< Scalar, Expression< Matrix, Matrix, Times>, Times>& exp1,
    const Expression< Scalar, Matrix, Times>& exp2)
{
    return Expression< Expression< Scalar, Expression< Matrix, Matrix, Times>, Times>,
                       Expression< Scalar, Matrix, Times>, Plus>(exp1,exp2);
}

inline Expression< Expression< Scalar, Expression< Matrix, Matrix, Times>, Times>,
                   Expression< Scalar, Matrix, Times>, Plus> operator+(
    const Expression< Scalar, Matrix, Times>& exp2,
    const Expression< Scalar, Expression< Matrix, Matrix, Times>, Times>& exp1)
{
    return Expression< Expression< Scalar, Expression< Matrix, Matrix, Times>, Times>,
                       Expression< Scalar, Matrix, Times>, Plus>(exp1,exp2);
}

inline Expression< Expression< Scalar, Expression< Matrix, Matrix, Times>, Times>,
                   Expression< Scalar, Matrix, Times>, Plus> operator-(
    const Expression< Scalar, Expression< Matrix, Matrix, Times>, Times>& exp1,
    const Expression< Scalar, Matrix, Times>& exp2)
{
    return Expression< Expression< Scalar, Expression< Matrix, Matrix, Times>, Times>,
                       Expression< Scalar, Matrix, Times>, Plus>(exp1,
            Expression< Scalar, Matrix, Times>( exp2.getArg1()*-1.0, exp2.getArg2() )
        );
}

inline Expression< Expression< Scalar, Expression< Matrix, Matrix, Times>, Times>,
                   Expression< Scalar, Matrix, Times>, Plus> operator-(
    const Expression< Scalar, Matrix, Times>& exp2,
    const Expression< Scalar, Expression< Matrix, Matrix, Times>, Times>& exp1)
{
    Expression<Scalar,Expression<Matrix,Matrix,Times>,Times>  e(exp1.getArg1()*-1.0,exp1.getArg2());
    return Expression< Expression< Scalar, Expression< Matrix, Matrix, Times>, Times>,
                       Expression< Scalar, Matrix, Times>, Plus>( e, exp2 );
}

}

#endif // LAMA_MATRIXEXPRESSIONS_HPP_