// file      : libbuild2/context.ixx -*- C++ -*-
// license   : MIT; see accompanying LICENSE file

namespace build2
{
  // wait_guard
  //
  inline wait_guard::
  wait_guard ()
      : ctx (nullptr), start_count (0), task_count (nullptr), phase (false)
  {
  }

  inline wait_guard::
  wait_guard (context& c, atomic_count& tc, bool p)
      : wait_guard (c, 0, tc, p)
  {
  }

  inline wait_guard::
  wait_guard (context& c, size_t sc, atomic_count& tc, bool p)
      : ctx (&c), start_count (sc), task_count (&tc), phase (p)
  {
  }

  inline wait_guard::
  ~wait_guard () noexcept (false)
  {
    if (task_count != nullptr)
      wait ();
  }

  inline wait_guard::
  wait_guard (wait_guard&& x)
      : ctx (x.ctx),
        start_count (x.start_count),
        task_count (x.task_count),
        phase (x.phase)
  {
    x.task_count = nullptr;
  }

  inline wait_guard& wait_guard::
  operator= (wait_guard&& x)
  {
    if (&x != this)
    {
      assert (task_count == nullptr);
      ctx = x.ctx;
      start_count = x.start_count; task_count = x.task_count; phase = x.phase;
      x.task_count = nullptr;
    }
    return *this;
  }

  inline void wait_guard::
  wait ()
  {
    if (task_count->load (memory_order_acquire) > start_count)
    {
      phase_unlock u (*ctx, phase);
      ctx->sched.wait (start_count, *task_count);
    }

    task_count = nullptr;
  }
}