MachineShMemWin.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
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* MachineShMemWin.h                                           (C) 2000-2026 */
/*                                                                           */
/* Class allowing the creation of memory windows for a compute node.         */
/* The segments of these windows are not contiguous in memory and can        */
/* be resized.                                                               */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_CORE_MACHINESHMEMWIN_H
#define ARCANE_CORE_MACHINESHMEMWIN_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/core/MachineShMemWinBase.h"
#include "arcane/core/ArcaneTypes.h"
#include "arcane/utils/Ref.h"
 
#include "arccore/base/Span.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

template <class Type>
class MachineShMemWin
{
 
 public:

  MachineShMemWin(IParallelMng* pm, Int64 nb_elem_segment)
  : m_impl(pm, nb_elem_segment, static_cast<Int32>(sizeof(Type)))
  {}

  explicit MachineShMemWin(IParallelMng* pm)
  : m_impl(pm, 0, static_cast<Int32>(sizeof(Type)))
  {}
 
 public:

  ConstArrayView<Int32> machineRanks() const
  {
    return m_impl.machineRanks();
  }

  void barrier() const
  {
    m_impl.barrier();
  }

  Span<Type> segmentView()
  {
    return asSpan<Type>(m_impl.segmentView());
  }

  Span<Type> segmentView(Int32 rank)
  {
    return asSpan<Type>(m_impl.segmentView(rank));
  }

  Span<const Type> segmentConstView() const
  {
    return asSpan<const Type>(m_impl.segmentConstView());
  }

  Span<const Type> segmentConstView(Int32 rank) const
  {
    return asSpan<const Type>(m_impl.segmentConstView(rank));
  }

  void add(Span<const Type> elem)
  {
    const Span<const std::byte> span_bytes(reinterpret_cast<const std::byte*>(elem.data()), elem.sizeBytes());
    m_impl.add(span_bytes);
  }

  void add()
  {
    m_impl.add();
  }

  void addToAnotherSegment(Int32 rank, Span<const Type> elem)
  {
    const Span<const std::byte> span_bytes(reinterpret_cast<const std::byte*>(elem.data()), elem.sizeBytes());
    m_impl.addToAnotherSegment(rank, span_bytes);
  }

  void addToAnotherSegment()
  {
    m_impl.addToAnotherSegment();
  }

  void reserve(Int64 new_capacity)
  {
    m_impl.reserve(new_capacity);
  }

  void reserve()
  {
    m_impl.reserve();
  }

  void resize(Int64 new_nb_elem)
  {
    m_impl.resize(new_nb_elem);
  }

  void resize()
  {
    m_impl.resize();
  }

  void shrink()
  {
    m_impl.shrink();
  }
 
 private:
 
  MachineShMemWinBase m_impl;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // End namespace Arcane
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif