/home/brandes/workspace/LAMA/src/lama/matrix/CRTPMatrix.hpp

Go to the documentation of this file.

#ifndef LAMA_CRTP_MATRIX_HPP_
#define LAMA_CRTP_MATRIX_HPP_

// for dll_import
#include <lama/config.hpp>

// base classes
#include <lama/matrix/Matrix.hpp>

// others
#include <lama/DenseVector.hpp>
#include <lama/Scalar.hpp>

// tracing
#include <lama/tracing.hpp>

// assert
#include <lama/exception/LAMAAssert.hpp>

namespace lama
{

template<class Derived, typename ValueType>
class CRTPMatrix : public Matrix
{
public:

    CRTPMatrix() : Matrix()
    {
    }

    CRTPMatrix( const IndexType numRows, const IndexType numColumns )

        : Matrix( numRows, numColumns )
    {
    }

    CRTPMatrix( DistributionPtr rowDistribution, DistributionPtr colDistribution )

        : Matrix( rowDistribution, colDistribution )
    {
    }

    void matrixTimesVector ( Vector& result,
                             const Scalar alpha, const Vector& x,
                             const Scalar beta,  const Vector& y ) const
    {
        LAMA_REGION( "MatrixTimesVector" );
    
        LAMA_LOG_INFO( logger, result << " = "
                  << alpha << " * " << *this << " * " << x
                  << " + " << beta << " * " << y );
    
        if ( &result == &y )
        {
            LAMA_LOG_DEBUG( logger, "alias: result = y is well handled" );
        }
        else if ( &result == &x )
        {
            LAMA_THROWEXCEPTION( "alias: result = x is not handled, use temporary" );
        }
        else
        {
            // we inherit the row distribution of this matrix to result
    
            result.resize( getDistributionPtr() );
    
            // no more to check: result.size() == mNumRows, getDistirubtion() == result.getDistribution() 
        }

        LAMA_ASSERT_EQUAL( x.getDistribution(), getColDistribution() );
        LAMA_ASSERT_EQUAL( y.getDistribution(), getDistribution() );
    
        const DenseVector<ValueType>* denseX = dynamic_cast<const DenseVector<ValueType>* > ( &x );
        const DenseVector<ValueType>* denseY = dynamic_cast<const DenseVector<ValueType>* > ( &y );
        DenseVector<ValueType>* denseResult = dynamic_cast<DenseVector<ValueType>* > ( &result );
    
        LAMA_ASSERT( denseX, x << ": must be DenseVector<" << Scalar::getType<ValueType>() << ">" );

        // Note: in case of beta == 0, we might skip this test

        LAMA_ASSERT( denseY, y << ": must be DenseVector<" << Scalar::getType<ValueType>() << ">" );

        LAMA_ASSERT( denseResult, result << ": must be DenseVector<" << Scalar::getType<ValueType>() << ">" );

        static_cast<const Derived*>(this)->matrixTimesVectorImpl( *denseResult, alpha.getValue<ValueType>(), *denseX,
                                                                  beta.getValue<ValueType>(), *denseY   );
    }

protected:
#ifndef LAMA_LOG_LEVEL_OFF
    using Matrix::logger;
#endif
};

}

#endif // LAMA_CRTP_MATRIX_HPP_