Tensor1_or_Tensor1.hpp

Go to the documentation of this file.

/* Adds two Tensor1's together yielding a Tensor2_symmetric. */
 
#pragma once
 
namespace FTensor
{
  template <class A, class B, class T, class U, int Dim, char i, char j>
  class Tensor1_or_Tensor1
  {
    Tensor1_Expr<A, T, Dim, i> iterA;
    Tensor1_Expr<B, U, Dim, j> iterB;
 
  public:
    typename promote<T, U>::V operator()(const int N1, const int N2) const
    {
      return iterA(N1) + iterB(N2);
    }
 
    Tensor1_or_Tensor1(const Tensor1_Expr<A, T, Dim, i> &a,
                       const Tensor1_Expr<B, U, Dim, j> &b)
        : iterA(a), iterB(b)
    {}
  };
 
  template <class A, class B, class T, class U, int Dim, char i, char j>
  Tensor2_symmetric_Expr<Tensor1_or_Tensor1<A, B, T, U, Dim, i, j>,
                         typename promote<T, U>::V, Dim, i, j>
  operator||(const Tensor1_Expr<A, T, Dim, i> &a,
             const Tensor1_Expr<B, U, Dim, j> &b)
  {
    using TensorExpr = Tensor1_or_Tensor1<A, B, T, U, Dim, i, j>;
    return Tensor2_symmetric_Expr<TensorExpr, typename promote<T, U>::V, Dim,
                                  i, j>(TensorExpr(a, b));
  }
}