StdThreadImplementation.cc

// -*- 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
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* StdThreadImplementation.cc                                  (C) 2000-2026 */
/*                                                                           */
/* Implementation of threads using the standard C++ library.                 */
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arccore/base/NotImplementedException.h"
#include "arccore/base/IFunctor.h"
#include "arccore/base/ReferenceCounterImpl.h"
#include "arccore/base/NotSupportedException.h"
#include "arccore/base/Ref.h"
 
#include "arccore/concurrency/internal/ConcurrencyGlobalInternal.h"
#include "arccore/concurrency/IThreadBarrier.h"
#include "arccore/concurrency/Mutex.h"
 
#include <thread>
#include <condition_variable>
#include <mutex>
#include <barrier>
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace Arcane::Concurrency
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

class ARCCORE_CONCURRENCY_EXPORT StdThreadImplementation
: public IThreadImplementation
, public ReferenceCounterImpl
{
  ARCCORE_DEFINE_REFERENCE_COUNTED_INCLASS_METHODS();
 
 public:
 
  explicit StdThreadImplementation(bool use_legacy_barrier);
  ~StdThreadImplementation() override;
 
 public:
 
  void initialize() override;
 
 public:
 
  ThreadImpl* createThread(IFunctor* f) override;
  void joinThread(ThreadImpl* t) override;
  void destroyThread(ThreadImpl* t) override;
 
  void createSpinLock(Int64* spin_lock_addr) override;
  void lockSpinLock(Int64* spin_lock_addr, Int64* scoped_spin_lock_addr) override;
  void unlockSpinLock(Int64* spin_lock_addr, Int64* scoped_spin_lock_addr) override;
 
  MutexImpl* createMutex() override;
  void destroyMutex(MutexImpl*) override;
  void lockMutex(MutexImpl* mutex) override;
  void unlockMutex(MutexImpl* mutex) override;
 
  Int64 currentThread() override;
 
  IThreadBarrier* createBarrier() override;
 
 public:
 
  void addReference() override { ReferenceCounterImpl::addReference(); }
  void removeReference() override { ReferenceCounterImpl::removeReference(); }
 
 private:
 
  MutexImpl* m_global_mutex_impl = nullptr;
  bool m_use_legacy_barrier = false;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
namespace
{
  void* _StdStartFunc(void* f)
  {
    IFunctor* ff = static_cast<IFunctor*>(f);
    ff->executeFunctor();
    return nullptr;
  }
} // namespace
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

class LegacyStdThreadBarrier
: public IThreadBarrier
{
 public:
 
  void init(Integer nb_thread) override
  {
    m_nb_thread = nb_thread;
    m_current_reached = 0;
  }
 
  void destroy() override
  {
    m_nb_thread = 0;
    m_current_reached = 0;
    delete this;
  }
 
  void wait() override
  {
    std::unique_lock<std::mutex> lk(m_wait_mutex);
    ++m_current_reached;
    Int32 generation = m_generation;
    //cout << "ADD BARRIER N=" << m_current_reached << '\n';
    if (m_current_reached == m_nb_thread) {
      ++m_generation;
      m_current_reached = 0;
      //cout << "BROADCAST BARRIER N=" << m_current_reached << '\n';
      lk.unlock();
      m_wait.notify_all();
    }
    while (generation == m_generation)
      m_wait.wait(lk);
  }
 
 private:
 
  std::mutex m_wait_mutex;
  std::condition_variable m_wait;
  Integer m_nb_thread = 0;
  Integer m_current_reached = 0;
  Int32 m_generation = 0;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/

class StdThreadBarrier
: public IThreadBarrier
{
  class NullFunc
  {
   public:
 
    void operator()() const noexcept { /* Nothing to do */ }
  };
 
 public:
 
  ~StdThreadBarrier() override { delete m_barrier; }
 
 public:
 
  void init(Integer nb_thread) override
  {
    m_barrier = new std::barrier<NullFunc>(nb_thread);
  }
 
  void destroy() override
  {
    delete this;
  }
 
  void wait() override
  {
    ARCCORE_CHECK_POINTER(m_barrier);
    m_barrier->arrive_and_wait();
  }
 
 private:
 
  std::barrier<NullFunc>* m_barrier = nullptr;
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
StdThreadImplementation::
StdThreadImplementation(bool use_legacy_barrier)
: m_use_legacy_barrier(use_legacy_barrier)
{
}
 
StdThreadImplementation::
~StdThreadImplementation()
{
  GlobalMutex::destroy();
  if (m_global_mutex_impl)
    destroyMutex(m_global_mutex_impl);
}
 
void StdThreadImplementation::
initialize()
{
  m_global_mutex_impl = createMutex();
  GlobalMutex::init(m_global_mutex_impl);
}
 
ThreadImpl* StdThreadImplementation::
createThread(IFunctor* f)
{
  return reinterpret_cast<ThreadImpl*>(new std::thread(&_StdStartFunc, f));
}
 
void StdThreadImplementation::
joinThread(ThreadImpl* t)
{
  std::thread* tt = reinterpret_cast<std::thread*>(t);
  tt->join();
}
 
void StdThreadImplementation::
destroyThread(ThreadImpl* t)
{
  std::thread* tt = reinterpret_cast<std::thread*>(t);
  delete tt;
}
 
void StdThreadImplementation::
createSpinLock(Int64* spin_lock_addr)
{
  ARCCORE_THROW(NotSupportedException, "Spin lock. Use std::atomic_flag instead()");
}
 
void StdThreadImplementation::
lockSpinLock(Int64* spin_lock_addr, Int64* scoped_spin_lock_addr)
{
  ARCCORE_THROW(NotSupportedException, "Spin lock. Use std::atomic_flag instead()");
}
 
void StdThreadImplementation::
unlockSpinLock(Int64* spin_lock_addr, Int64* scoped_spin_lock_addr)
{
  ARCCORE_THROW(NotSupportedException, "Spin lock. Use std::atomic_flag instead()");
}
 
MutexImpl* StdThreadImplementation::
createMutex()
{
  std::mutex* m = new std::mutex();
  return reinterpret_cast<MutexImpl*>(m);
}
 
void StdThreadImplementation::
destroyMutex(MutexImpl* mutex)
{
  std::mutex* m = reinterpret_cast<std::mutex*>(mutex);
  delete m;
}
 
void StdThreadImplementation::
lockMutex(MutexImpl* mutex)
{
  std::mutex* m = reinterpret_cast<std::mutex*>(mutex);
  m->lock();
}
 
void StdThreadImplementation::
unlockMutex(MutexImpl* mutex)
{
  std::mutex* m = reinterpret_cast<std::mutex*>(mutex);
  m->unlock();
}
 
Int64 StdThreadImplementation::
currentThread()
{
  Int64 v = std::hash<std::thread::id>{}(std::this_thread::get_id());
  return v;
}
 
IThreadBarrier* StdThreadImplementation::
createBarrier()
{
  if (m_use_legacy_barrier)
    return new LegacyStdThreadBarrier();
  return new StdThreadBarrier();
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
Ref<IThreadImplementation>
createStdThreadImplementation()
{
  return makeRef<IThreadImplementation>(new StdThreadImplementation(false));
}
 
Ref<IThreadImplementation>
createLegacyStdThreadImplementation()
{
  return makeRef<IThreadImplementation>(new StdThreadImplementation(true));
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // namespace Arcane::Concurrency
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/