Redistributor.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/kernels/redistributor/RedistributorCommPlan.h>
#include <alien/utils/Precomp.h>
#include <memory>
 
namespace Alien
{
 
class MultiMatrixImpl;
class MultiVectorImpl;

class ALIEN_EXPORT Redistributor
{
 public:
  enum class Method { dok, csr };
 
  Redistributor(int globalSize, IMessagePassingMng* super, IMessagePassingMng* target, Method method = Method::dok);
  virtual ~Redistributor() = default;

  std::shared_ptr<MultiMatrixImpl> redistribute(MultiMatrixImpl* mat);

  std::shared_ptr<MultiVectorImpl> redistribute(MultiVectorImpl* vect);

  void redistributeBack(MultiVectorImpl* vect);
 
  const RedistributorCommPlan* commPlan() const;
 
 private:
  IMessagePassingMng* m_super_pm;
  std::unique_ptr<RedistributorCommPlan> m_distributor;
  Method m_method = Method::dok;
};
 
} // namespace Alien