SimpleCSRInternalLinearAlgebra.h

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2026 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
 
#pragma once
 
#include <alien/utils/Precomp.h>
 
#include <alien/core/backend/IInternalLinearAlgebraExprT.h>
#include <alien/core/backend/IInternalLinearAlgebraT.h>
 
#include <alien/kernels/simple_csr/SimpleCSRBackEnd.h>
#include <alien/kernels/simple_csr/SimpleCSRVector.h>
#include <alien/kernels/simple_csr/SendRecvOp.h>
#include <alien/kernels/simple_csr/LUSendRecvOp.h>
 
#include <alien/distribution/VectorDistribution.h>
#include <alien/distribution/MatrixDistribution.h>
 
#include <alien/utils/ExceptionUtils.h>
 
#include <alien/utils/StdTimer.h>
/*---------------------------------------------------------------------------*/
 
namespace Alien
{
 
typedef AlgebraTraits<BackEnd::tag::simplecsr>::matrix_type CSRMatrix;
typedef AlgebraTraits<BackEnd::tag::simplecsr>::vector_type CSRVector;
 
template <>
struct LUSendRecvTraits<BackEnd::tag::simplecsr>
{
  // clang-format off
  typedef CSRMatrix                                          matrix_type ;
  typedef CSRVector                                          vector_type ;
  typedef AlgebraTraits<BackEnd::tag::simplecsr>::value_type value_type ;
  typedef SimpleCSRInternal::LUSendRecvOp<CSRMatrix>         matrix_op_type ;
  typedef SimpleCSRInternal::SendRecvOp<value_type>          vector_op_type ;
  // clang-format on
};
 
class ALIEN_EXPORT SimpleCSRInternalLinearAlgebra
: public IInternalLinearAlgebra<CSRMatrix, CSRVector>
{
 public:
  typedef BackEnd::tag::simplecsr BackEndType;
 
  typedef std::tuple<VectorDistribution const*,Integer> ResourceType;
 
  class NullValueException
  : public Exception::NumericException
  {
   public:
    typedef Exception::NumericException BaseType;
    NullValueException(std::string const& type)
    : BaseType(type, __LINE__)
    {}
  };
 
  template <typename T>
  class Future
  {
   public:
    Future(T& value)
    : m_value(value)
    {}
 
    T& operator()()
    {
      return m_value;
    }
 
    T operator()() const
    {
      return m_value;
    }
 
    T get()
    {
      return m_value;
    }
 
   private:
    T& m_value;
  };
 
  typedef Future<Real> FutureType;
 
  typedef Alien::StdTimer TimerType;
  typedef TimerType::Sentry SentryType;
 
  SimpleCSRInternalLinearAlgebra();
  virtual ~SimpleCSRInternalLinearAlgebra();
 
 public:
  // IInternalLinearAlgebra interface.
  Real norm0(const Vector& x) const;
  Real norm1(const Vector& x) const;
  Real norm2(const Vector& x) const;
  Real normInf(const Vector& x) const;
 
  void synchronize(const Matrix& A, Vector& x) const;
 
  void mult(const Matrix& A, const Vector& x, Vector& r) const;
  void addLMult(Real alpha, const Matrix& A, const Vector& x, Vector& y) const;
  void addUMult(Real alpha, const Matrix& A, const Vector& x, Vector& y) const;
 
  void multDiag(const Matrix& A, Vector& y) const;
  void computeDiag(const Matrix& a, Vector& inv_diag) const;
 
  void multInvDiag(const Matrix& A, Vector& y) const;
  void computeInvDiag(const Matrix& a, Vector& inv_diag) const;
 
  void axpy(Real alpha, const Vector& x, Vector& r) const;
  void aypx(Real alpha, Vector& y, const Vector& x) const;
  void copy(const Vector& x, Vector& r) const;
 
  void axpy(Real alpha, const Vector& x,Integer stride_x,Vector& r,Integer stride_r) const;
  void aypx(Real alpha, Vector& y, Integer stride_y, const Vector& x,Integer stride_x) const;
  void copy(const Vector& x, Integer stride_x, Vector& r, Integer stride_r) const;
 
  Real dot(const Vector& x, const Vector& y) const;
  void dot(const Vector& x, const Vector& y, FutureType& res) const;
 
  void scal(Real alpha, Vector& x) const;
  void scal(const Vector& x, Matrix& a) const;
  void diagonal(const Matrix& a, Vector& x) const;
  void reciprocal(Vector& x) const;
  void pointwiseMult(const Vector& x, const Vector& y, Vector& w) const;
  void assign(Vector& x, Real alpha) const;
 
  template <typename LambdaT>
  void assign(Vector& x, LambdaT const& lambda) const
  {
    auto x_ptr = x.getDataPtr();
    for (Integer i = 0; i < x.getAllocSize(); ++i) {
      x_ptr[i] = lambda(i);
    }
  }
 
  template <typename PrecondT>
  void exec(PrecondT& precond, Vector const& x, Vector& y)
  {
    return precond.solve(*this, x, y);
  }
 
  Integer computeCxr(const Matrix& a, Matrix& cxr_a) const ;
  Integer computeCxr(const Matrix& a, Vector const& diag_scal, Matrix& cxr_a) const ;
 
  static ResourceType resource(Matrix const& A);
 
  void allocate(ResourceType resource, Vector& v);
 
  template <typename T0, typename... T>
  void allocate(ResourceType resource, T0& v0, T&... args)
  {
    allocate(resource, v0);
    allocate(resource, args...);
  }
 
  void free(Vector& v);
 
  template <typename T0, typename... T>
  void free(T0& v0, T&... args)
  {
    free(v0);
    free(args...);
  }
 
#ifdef ALIEN_USE_PERF_TIMER
 private:
  mutable TimerType m_timer;
#endif
};
 
class SimpleCSRInternalLinearAlgebraExpr
: public IInternalLinearAlgebraExpr<CSRMatrix, CSRVector>
{
 public:
  SimpleCSRInternalLinearAlgebraExpr();
  virtual ~SimpleCSRInternalLinearAlgebraExpr();
 
 public:
  // IInternalLinearAlgebra interface.
  Real norm0(const Vector& x) const;
  Real norm1(const Vector& x) const;
  Real norm2(const Vector& x) const;
  Real normInf(const Vector& x) const;
  Real norm2(const Matrix& x) const;
  void mult(const Matrix& a, const Vector& x, Vector& r) const;
  void axpy(Real alpha, const Vector& x, Vector& r) const;
  void aypx(Real alpha, Vector& y, const Vector& x) const;
  void copy(const Vector& x, Vector& r) const;
  Real dot(const Vector& x, const Vector& y) const;
  void scal(Real alpha, Vector& x) const;
  void diagonal(const Matrix& a, Vector& x) const;
  void reciprocal(Vector& x) const;
  void pointwiseMult(const Vector& x, const Vector& y, Vector& w) const;
 
  // IInternalLinearAlgebraExpr interface.
 
  void mult(const Matrix& a, const UniqueArray<Real>& x, UniqueArray<Real>& r) const;
  void axpy(Real alpha, UniqueArray<Real> const& x, UniqueArray<Real>& r) const;
  void aypx(Real alpha, UniqueArray<Real>& y, UniqueArray<Real> const& x) const;
  void copy(const UniqueArray<Real>& x, UniqueArray<Real>& r) const;
  Real dot(
  Integer local_size, const UniqueArray<Real>& x, const UniqueArray<Real>& y) const;
 
  void scal(Real alpha, UniqueArray<Real>& x) const;
 
  void copy(const Matrix& a, Matrix& r) const;
  void add(const Matrix& a, Matrix& r) const;
  void scal(Real alpha, Matrix& r) const;
 
 private:
  // No member.
};
 
} // namespace Alien
 
/*---------------------------------------------------------------------------*/