solveu.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_SOLVEU_H
#define MYRAMATH_MULTIFRONTAL_LU_SOLVEU_H

#include <myramath/multifrontal/detail/solveu.h>

namespace myra {
namespace multifrontal {
namespace lu {
namespace solveu {

template<class Number> class JobGraphBase : public ::myra::multifrontal::detail::solveu::JobGraphBase2<Number>
  {
  public:

    // Structural typedefs for base class.
    typedef ::myra::multifrontal::detail::solveu::JobGraphBase2<Number> Base;

    // Typedefs related to numeric storage on LUContainer
    typedef LUKernel<Number> Kernel;
    typedef ::myra::multifrontal::detail::lu::LUContainer<Kernel> LUContainer;    
  
    // Constructor - requires const reference to LUContainer, mutable reference to right hand side.
    JobGraphBase(const LUContainer* in_lucontainer, const MatrixRange<Number>& in_B, int in_blocksize)
      : Base(&in_lucontainer->tree(), in_B, in_blocksize), lucontainer(in_lucontainer) { }

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

  protected:

    // Accesors for U Kernel's/Block's
    const Kernel& u(int n, int ij) const
      { return lucontainer->lu(n,ij); }
    const CMatrixRange<Number> u(int n, int i, int j) const
      { return lucontainer->lu(n,i,j); }

    // Solves Un(k,k)*Xn(k,j) = Bn(k,j)
    virtual uint64_t backsolve(int n, int k, int j)
      {
      const Kernel& Un_kk = this->u(n,k);
      MatrixRange<Number> Bn_kj = this->b(n,k,j);
      return Un_kk.solveU(Bn_kj,'L','N');
      }
    
    // Downdates Bn(i,j) -= Un(i,k)*Xn(k,j)
    virtual uint64_t downdate(int n, int k, int i, int j)
      {
      CMatrixRange<Number> Un_ik = this->u(n,i,k);
      CMatrixRange<Number> Xn_kj = this->x(n,k,j);
       MatrixRange<Number> Bn_ij = this->b(n,i,j);
      Number one(1);
      return gemm_nwork(Bn_ij, Un_ik, 'N', Xn_kj, 'N', -one, one);
      }

  private:
  
    // Reference to LUContainer (const).
    const LUContainer* lucontainer;
    
  }; // class JobGraph

} } } } // namespace

#endif