Ddg_or_Tensor2_symmetric.hpp

Go to the documentation of this file.

/* This file has all of the declarations for expressions like
   Ddg|Tensor2_symmetric and Tensor2_symmetric|Ddg,
   yielding a Tensor2_symmetric. */
 
#pragma once
 
namespace FTensor
{
  /* A(i,j,k,l)*B(j,l) */
 
  template <class A, class B, class T, class U, int Dim, char i, char j,
            char k, char l>
  class Ddg_or_Tensor2_symmetric_13
  {
    Ddg_Expr<A, T, Dim, Dim, i, j, k, l> iterA;
    Tensor2_symmetric_Expr<B, U, Dim, j, l> iterB;
 
    template <int Current_Dim0, int Current_Dim1>
    typename promote<T, U>::V
    eval(const int N1, const int N2, const Number<Current_Dim0> &,
         const Number<Current_Dim1> &) const
    {
      return iterA(Current_Dim0 - 1, N1, Current_Dim1 - 1, N2)
               * iterB(Current_Dim0 - 1, Current_Dim1 - 1)
             + eval(N1, N2, Number<Current_Dim0 - 1>(),
                    Number<Current_Dim1>());
    }
    template <int Current_Dim1>
    typename promote<T, U>::V
    eval(const int N1, const int N2, const Number<1> &,
         const Number<Current_Dim1> &) const
    {
      return iterA(0, N1, Current_Dim1 - 1, N2) * iterB(0, Current_Dim1 - 1)
             + eval(N1, N2, Number<Dim>(), Number<Current_Dim1 - 1>());
    }
    typename promote<T, U>::V eval(const int N1, const int N2,
                                   const Number<1> &, const Number<1> &) const
    {
      return iterA(0, N1, 0, N2) * iterB(0, 0);
    }
 
  public:
    Ddg_or_Tensor2_symmetric_13(
      const Ddg_Expr<A, T, Dim, Dim, i, j, k, l> &a,
      const Tensor2_symmetric_Expr<B, U, Dim, j, l> &b)
        : iterA(a), iterB(b)
    {}
    typename promote<T, U>::V operator()(const int N1, const int N2) const
    {
      return eval(N1, N2, Number<Dim>(), Number<Dim>());
    }
  };
 
  template <class A, class B, class T, class U, int Dim, char i, char j,
            char k, char l>
  Tensor2_symmetric_Expr<
    Ddg_or_Tensor2_symmetric_13<A, B, T, U, Dim, i, j, k, l>,
    typename promote<T, U>::V, Dim, i, k>
  operator|(const Ddg_Expr<A, T, Dim, Dim, i, j, k, l> &a,
            const Tensor2_symmetric_Expr<B, U, Dim, j, l> &b)
  {
    using TensorExpr
      = Ddg_or_Tensor2_symmetric_13<A, B, T, U, Dim, i, j, k, l>;
    return Tensor2_symmetric_Expr<TensorExpr, typename promote<T, U>::V, Dim,
                                  i, k>(TensorExpr(a, b));
  }
 
  /* B(j,l)*A(i,j,k,l) */
 
  template <class A, class B, class T, class U, int Dim, char i, char j,
            char k, char l>
  Tensor2_symmetric_Expr<
    Ddg_or_Tensor2_symmetric_13<A, B, T, U, Dim, i, j, k, l>,
    typename promote<T, U>::V, Dim, i, k>
  operator|(const Tensor2_symmetric_Expr<B, U, Dim, j, l> &b,
            const Ddg_Expr<A, T, Dim, Dim, i, j, k, l> &a)
  {
    using TensorExpr
      = Ddg_or_Tensor2_symmetric_13<A, B, T, U, Dim, i, j, k, l>;
    return Tensor2_symmetric_Expr<TensorExpr, typename promote<T, U>::V, Dim,
                                  i, k>(TensorExpr(a, b));
  }
}