partialsolveut_right.h

Go to the documentation of this file.

// ========================================================================= //
// This file is part of MyraMath, copyright (c) 2014-2019 by Ryan A Chilton  //
// and distributed by MyraCore, LLC. See LICENSE.txt for license terms.      //
// ========================================================================= //

#ifndef MYRAMATH_MULTIFRONTAL_LU_PARTIALSOLVEUT_RIGHT_H
#define MYRAMATH_MULTIFRONTAL_LU_PARTIALSOLVEUT_RIGHT_H

#include <myramath/utility/Number.h>

#include <myramath/dense/MatrixRange.h>
#include <myramath/dense/gemm.h>
#include <myramath/dense/detail/nwork.h>

#include <myramath/multifrontal/detail/partialsolvel_right.h>
#include <myramath/multifrontal/detail/QContainer.h>
#include <myramath/multifrontal/detail/lu/LUContainer.h>

namespace myra {
namespace multifrontal {
namespace lu {
namespace partialsolveut_right {

template<class Kernel> class JobGraphBase : public ::myra::multifrontal::detail::partialsolvel_right::JobGraphBase2<typename ReflectNumber<Kernel>::type>
  {
  public:

    // Structural typedefs for base class.
    typedef typename ReflectNumber<Kernel>::type Number;
    typedef ::myra::multifrontal::detail::partialsolvel_right::JobGraphBase2<Number> Base;

    // Typedefs related to numeric storage.
    typedef ::myra::multifrontal::detail::lu::LUContainer<Kernel> LUContainer;
    typedef ::myra::multifrontal::detail::QContainer<Number> QContainer;
    typedef CMatrixRange<Number> CRange;
    typedef  MatrixRange<Number>  Range;
    
    // Constructor - requires const reference to LUContainer, mutable reference to QContainer
    JobGraphBase(const LUContainer* in_lucontainer, const SchurTree* in_stree, QContainer* in_qcontainer, char in_op)
      : Base(in_stree,in_qcontainer), lucontainer(in_lucontainer), op(in_op) { }

    // Virtual copy constructor.
    virtual ::myra::JobGraphBase* clone() const
      { return new JobGraphBase(*this); }

  protected:

    // Accessors for U Kernel's / Block's
    const Kernel& u(int n, int ij) const
      { return lucontainer->lu(this->s2a(n),ij); }
    const CRange u(int n, int i, int j) const
      { return lucontainer->lu(this->s2a(n),i,j); }
  
    // Solves Xn(i,k)*Un(k,k)' = Bn(i,k)
    virtual uint64_t backsolve(int n, int k, int i)
      {
      const Kernel& Un_kk = this->u(n,k);
      MatrixRange<Number> Bn_ik = this->b(n,i,k);
      return Un_kk.solveU(Bn_ik,'R',op);
      }

    // Downdates Bn(i,j) -= Xn(i,k)*Un(j,k)'
    virtual uint64_t downdate(int n, int k, int i, int j)
      {
      CMatrixRange<Number> Un_jk = this->u(n,j,k);
      CMatrixRange<Number> Xn_ik = this->x(n,i,k);
       MatrixRange<Number> Bn_ij = this->b(n,i,j);
      Number one(1);
      return gemm_nwork(Bn_ij, Xn_ik, 'N', Un_jk, op, -one, one);
      }
  private:
  
    // Reference to LUContainer (const).
    const LUContainer* lucontainer;

    // Op modifier, either 'T' or 'H'
    char op;

  }; // class JobGraph

} } } } // namespace

#endif