135 lines
5.2 KiB
C
135 lines
5.2 KiB
C
|
// This file is part of Eigen, a lightweight C++ template library
|
||
|
// for linear algebra.
|
||
|
//
|
||
|
// Copyright (C) 2009 Gael Guennebaud <gael.guennebaud@inria.fr>
|
||
|
//
|
||
|
// This Source Code Form is subject to the terms of the Mozilla
|
||
|
// Public License v. 2.0. If a copy of the MPL was not distributed
|
||
|
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||
|
|
||
|
#ifndef EIGEN_NOALIAS_H
|
||
|
#define EIGEN_NOALIAS_H
|
||
|
|
||
|
namespace Eigen {
|
||
|
|
||
|
/** \class NoAlias
|
||
|
* \ingroup Core_Module
|
||
|
*
|
||
|
* \brief Pseudo expression providing an operator = assuming no aliasing
|
||
|
*
|
||
|
* \param ExpressionType the type of the object on which to do the lazy assignment
|
||
|
*
|
||
|
* This class represents an expression with special assignment operators
|
||
|
* assuming no aliasing between the target expression and the source expression.
|
||
|
* More precisely it alloas to bypass the EvalBeforeAssignBit flag of the source expression.
|
||
|
* It is the return type of MatrixBase::noalias()
|
||
|
* and most of the time this is the only way it is used.
|
||
|
*
|
||
|
* \sa MatrixBase::noalias()
|
||
|
*/
|
||
|
template<typename ExpressionType, template <typename> class StorageBase>
|
||
|
class NoAlias
|
||
|
{
|
||
|
typedef typename ExpressionType::Scalar Scalar;
|
||
|
public:
|
||
|
NoAlias(ExpressionType& expression) : m_expression(expression) {}
|
||
|
|
||
|
/** Behaves like MatrixBase::lazyAssign(other)
|
||
|
* \sa MatrixBase::lazyAssign() */
|
||
|
template<typename OtherDerived>
|
||
|
EIGEN_STRONG_INLINE ExpressionType& operator=(const StorageBase<OtherDerived>& other)
|
||
|
{ return internal::assign_selector<ExpressionType,OtherDerived,false>::run(m_expression,other.derived()); }
|
||
|
|
||
|
/** \sa MatrixBase::operator+= */
|
||
|
template<typename OtherDerived>
|
||
|
EIGEN_STRONG_INLINE ExpressionType& operator+=(const StorageBase<OtherDerived>& other)
|
||
|
{
|
||
|
typedef SelfCwiseBinaryOp<internal::scalar_sum_op<Scalar>, ExpressionType, OtherDerived> SelfAdder;
|
||
|
SelfAdder tmp(m_expression);
|
||
|
typedef typename internal::nested<OtherDerived>::type OtherDerivedNested;
|
||
|
typedef typename internal::remove_all<OtherDerivedNested>::type _OtherDerivedNested;
|
||
|
internal::assign_selector<SelfAdder,_OtherDerivedNested,false>::run(tmp,OtherDerivedNested(other.derived()));
|
||
|
return m_expression;
|
||
|
}
|
||
|
|
||
|
/** \sa MatrixBase::operator-= */
|
||
|
template<typename OtherDerived>
|
||
|
EIGEN_STRONG_INLINE ExpressionType& operator-=(const StorageBase<OtherDerived>& other)
|
||
|
{
|
||
|
typedef SelfCwiseBinaryOp<internal::scalar_difference_op<Scalar>, ExpressionType, OtherDerived> SelfAdder;
|
||
|
SelfAdder tmp(m_expression);
|
||
|
typedef typename internal::nested<OtherDerived>::type OtherDerivedNested;
|
||
|
typedef typename internal::remove_all<OtherDerivedNested>::type _OtherDerivedNested;
|
||
|
internal::assign_selector<SelfAdder,_OtherDerivedNested,false>::run(tmp,OtherDerivedNested(other.derived()));
|
||
|
return m_expression;
|
||
|
}
|
||
|
|
||
|
#ifndef EIGEN_PARSED_BY_DOXYGEN
|
||
|
template<typename ProductDerived, typename Lhs, typename Rhs>
|
||
|
EIGEN_STRONG_INLINE ExpressionType& operator+=(const ProductBase<ProductDerived, Lhs,Rhs>& other)
|
||
|
{ other.derived().addTo(m_expression); return m_expression; }
|
||
|
|
||
|
template<typename ProductDerived, typename Lhs, typename Rhs>
|
||
|
EIGEN_STRONG_INLINE ExpressionType& operator-=(const ProductBase<ProductDerived, Lhs,Rhs>& other)
|
||
|
{ other.derived().subTo(m_expression); return m_expression; }
|
||
|
|
||
|
template<typename Lhs, typename Rhs, int NestingFlags>
|
||
|
EIGEN_STRONG_INLINE ExpressionType& operator+=(const CoeffBasedProduct<Lhs,Rhs,NestingFlags>& other)
|
||
|
{ return m_expression.derived() += CoeffBasedProduct<Lhs,Rhs,NestByRefBit>(other.lhs(), other.rhs()); }
|
||
|
|
||
|
template<typename Lhs, typename Rhs, int NestingFlags>
|
||
|
EIGEN_STRONG_INLINE ExpressionType& operator-=(const CoeffBasedProduct<Lhs,Rhs,NestingFlags>& other)
|
||
|
{ return m_expression.derived() -= CoeffBasedProduct<Lhs,Rhs,NestByRefBit>(other.lhs(), other.rhs()); }
|
||
|
|
||
|
template<typename OtherDerived>
|
||
|
ExpressionType& operator=(const ReturnByValue<OtherDerived>& func)
|
||
|
{ return m_expression = func; }
|
||
|
#endif
|
||
|
|
||
|
ExpressionType& expression() const
|
||
|
{
|
||
|
return m_expression;
|
||
|
}
|
||
|
|
||
|
protected:
|
||
|
ExpressionType& m_expression;
|
||
|
};
|
||
|
|
||
|
/** \returns a pseudo expression of \c *this with an operator= assuming
|
||
|
* no aliasing between \c *this and the source expression.
|
||
|
*
|
||
|
* More precisely, noalias() allows to bypass the EvalBeforeAssignBit flag.
|
||
|
* Currently, even though several expressions may alias, only product
|
||
|
* expressions have this flag. Therefore, noalias() is only usefull when
|
||
|
* the source expression contains a matrix product.
|
||
|
*
|
||
|
* Here are some examples where noalias is usefull:
|
||
|
* \code
|
||
|
* D.noalias() = A * B;
|
||
|
* D.noalias() += A.transpose() * B;
|
||
|
* D.noalias() -= 2 * A * B.adjoint();
|
||
|
* \endcode
|
||
|
*
|
||
|
* On the other hand the following example will lead to a \b wrong result:
|
||
|
* \code
|
||
|
* A.noalias() = A * B;
|
||
|
* \endcode
|
||
|
* because the result matrix A is also an operand of the matrix product. Therefore,
|
||
|
* there is no alternative than evaluating A * B in a temporary, that is the default
|
||
|
* behavior when you write:
|
||
|
* \code
|
||
|
* A = A * B;
|
||
|
* \endcode
|
||
|
*
|
||
|
* \sa class NoAlias
|
||
|
*/
|
||
|
template<typename Derived>
|
||
|
NoAlias<Derived,MatrixBase> MatrixBase<Derived>::noalias()
|
||
|
{
|
||
|
return derived();
|
||
|
}
|
||
|
|
||
|
} // end namespace Eigen
|
||
|
|
||
|
#endif // EIGEN_NOALIAS_H
|