NoAlias.h 3.5 KB
Newer Older
LM's avatar
LM committed
1 2 3 4 5
// This file is part of Eigen, a lightweight C++ template library
// for linear algebra.
//
// Copyright (C) 2009 Gael Guennebaud <gael.guennebaud@inria.fr>
//
Don Gagne's avatar
Don Gagne committed
6 7 8
// 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/.
LM's avatar
LM committed
9 10 11 12

#ifndef EIGEN_NOALIAS_H
#define EIGEN_NOALIAS_H

Don Gagne's avatar
Don Gagne committed
13 14
namespace Eigen {

LM's avatar
LM committed
15 16 17 18 19
/** \class NoAlias
  * \ingroup Core_Module
  *
  * \brief Pseudo expression providing an operator = assuming no aliasing
  *
20
  * \tparam ExpressionType the type of the object on which to do the lazy assignment
LM's avatar
LM committed
21 22 23 24 25 26 27 28 29 30 31 32 33
  *
  * 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
{
  public:
34 35 36 37
    typedef typename ExpressionType::Scalar Scalar;
    
    explicit NoAlias(ExpressionType& expression) : m_expression(expression) {}
    
LM's avatar
LM committed
38
    template<typename OtherDerived>
39
    EIGEN_DEVICE_FUNC
LM's avatar
LM committed
40
    EIGEN_STRONG_INLINE ExpressionType& operator=(const StorageBase<OtherDerived>& other)
41 42 43 44 45
    {
      call_assignment_no_alias(m_expression, other.derived(), internal::assign_op<Scalar,typename OtherDerived::Scalar>());
      return m_expression;
    }
    
LM's avatar
LM committed
46
    template<typename OtherDerived>
47
    EIGEN_DEVICE_FUNC
LM's avatar
LM committed
48 49
    EIGEN_STRONG_INLINE ExpressionType& operator+=(const StorageBase<OtherDerived>& other)
    {
50
      call_assignment_no_alias(m_expression, other.derived(), internal::add_assign_op<Scalar,typename OtherDerived::Scalar>());
LM's avatar
LM committed
51 52
      return m_expression;
    }
53
    
LM's avatar
LM committed
54
    template<typename OtherDerived>
55
    EIGEN_DEVICE_FUNC
LM's avatar
LM committed
56 57
    EIGEN_STRONG_INLINE ExpressionType& operator-=(const StorageBase<OtherDerived>& other)
    {
58
      call_assignment_no_alias(m_expression, other.derived(), internal::sub_assign_op<Scalar,typename OtherDerived::Scalar>());
LM's avatar
LM committed
59 60 61
      return m_expression;
    }

62
    EIGEN_DEVICE_FUNC
Don Gagne's avatar
Don Gagne committed
63 64 65 66 67
    ExpressionType& expression() const
    {
      return m_expression;
    }

LM's avatar
LM committed
68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102
  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()
{
103
  return NoAlias<Derived, Eigen::MatrixBase >(derived());
LM's avatar
LM committed
104 105
}

Don Gagne's avatar
Don Gagne committed
106 107
} // end namespace Eigen

LM's avatar
LM committed
108
#endif // EIGEN_NOALIAS_H