/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "nsILabelableRunnable.h"

#include "mozilla/Scheduler.h"
#include "mozilla/SchedulerGroup.h"

bool
nsILabelableRunnable::IsReadyToRun()
{
  MOZ_ASSERT(mozilla::Scheduler::AnyEventRunning());
  MOZ_ASSERT(!mozilla::Scheduler::UnlabeledEventRunning());
  SchedulerGroupSet groups;
  if (!GetAffectedSchedulerGroups(groups)) {
    // it can not be labeled right now.
    return false;
  }

  if (groups.mSingle) {
    MOZ_ASSERT(groups.mMulti.isNothing());
    return !groups.mSingle->IsRunning();
  }

  if (groups.mMulti.isSome()) {
    MOZ_ASSERT(!groups.mSingle);
    for (auto iter = groups.mMulti.ref().ConstIter(); !iter.Done(); iter.Next()) {
      if (iter.Get()->GetKey()->IsRunning()) {
        return false;
      }
    }
    return true;
  }

  // No affected groups if we are here. Then, it's ready to run.
  return true;
}

void
nsILabelableRunnable::SchedulerGroupSet::Put(mozilla::SchedulerGroup* aGroup)
{
  if (mSingle) {
    MOZ_ASSERT(mMulti.isNothing());
    mMulti.emplace();
    auto& multi = mMulti.ref();
    multi.PutEntry(mSingle);
    multi.PutEntry(aGroup);
    mSingle = nullptr;
    return;
  }

  if (mMulti.isSome()) {
    MOZ_ASSERT(!mSingle);
    mMulti.ref().PutEntry(aGroup);
    return;
  }

  mSingle = aGroup;
}

void
nsILabelableRunnable::SchedulerGroupSet::Clear()
{
  mSingle = nullptr;
  mMulti.reset();
}

void
nsILabelableRunnable::SchedulerGroupSet::SetIsRunning(bool aIsRunning)
{
  if (mSingle) {
    MOZ_ASSERT(mMulti.isNothing());
    mSingle->SetIsRunning(aIsRunning);
    return;
  }

  if (mMulti.isSome()) {
    MOZ_ASSERT(!mSingle);
    for (auto iter = mMulti.ref().ConstIter(); !iter.Done(); iter.Next()) {
      MOZ_ASSERT(iter.Get()->GetKey()->IsRunning() != aIsRunning);
      iter.Get()->GetKey()->SetIsRunning(aIsRunning);
    }
    return;
  }
}