SolveAction.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_ITERATIVE_SOLVEACTION_H
#define MYRAMATH_ITERATIVE_SOLVEACTION_H

#include <myramath/utility/Number.h>

#include <myramath/dense/MatrixRange.h>

#include <myramath/iterative/Action.h>
#include <myramath/iterative/ActionBase.h>

namespace myra {
namespace detail {

template<class Solver> class SolveAction : public ActionBase<typename ReflectNumber<Solver>::type>
  {
  public:

    // Useful typedefs.
    typedef typename ReflectNumber<Solver>::type Number;
    typedef CMatrixRange<Number> X;
    typedef MatrixRange<Number> B;
  
    // Constructor, requires reference to a Solver instance.
    SolveAction(const Solver& in_solver, char in_op) 
      : solver(&in_solver), op(in_op) { }

  protected:

    // Size inspector.
    virtual std::pair<int,int> size() const
      { int N = solver->size(); return std::pair<int,int>(N,N); }

    // Assign B = solve(X)
    virtual void multiply(const X& x, const B& b) const
      { b.assign(x); solver->solve(b,'L',op); }

    // Note, the other multiply(x,b,alpha,beta) just uses ActionBase's implementation.

    // Virtual copy constructor.
    virtual ActionBase<Number>* clone() const
      { return new SolveAction<Solver>(*solver,op); }
    
  private:
  
    // Underlying solver.
    const Solver* solver;

    // op modifier, if any.
    char op;

  };

} // namespace detail

template<class Solver> Action<typename ReflectNumber<Solver>::type> make_SolveAction(const Solver& solver, char op = 'N')
  { return detail::SolveAction<Solver>(solver,op); }

} // namespace myra

#endif