/* 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 "LayerTreeOwnerTracker.h"

#include "mozilla/StaticPtr.h"              // for StaticAutoPtr
#include "mozilla/gfx/GPUChild.h"           // for GPUChild
#include "mozilla/gfx/GPUProcessManager.h"  // for GPUProcessManager

#include <functional>

namespace mozilla {
namespace layers {

static StaticAutoPtr<LayerTreeOwnerTracker> sSingleton;

LayerTreeOwnerTracker::LayerTreeOwnerTracker()
    : mLayerIdsLock("LayerTreeOwnerTrackerLock") {}

void LayerTreeOwnerTracker::Initialize() {
  MOZ_ASSERT(!sSingleton);
  sSingleton = new LayerTreeOwnerTracker();
}

void LayerTreeOwnerTracker::Shutdown() { sSingleton = nullptr; }

LayerTreeOwnerTracker* LayerTreeOwnerTracker::Get() { return sSingleton; }

void LayerTreeOwnerTracker::Map(LayersId aLayersId,
                                base::ProcessId aProcessId) {
  MutexAutoLock lock(mLayerIdsLock);

  // Add the mapping to the list
  const auto i = mLayerIds.insert({aLayersId, aProcessId});
  MOZ_ASSERT(i.second, "Mapping already used layers ID!");
  (void)i;
}

void LayerTreeOwnerTracker::Unmap(LayersId aLayersId,
                                  base::ProcessId aProcessId) {
  MutexAutoLock lock(mLayerIdsLock);

  auto i = mLayerIds.find(aLayersId);
  if (i == mLayerIds.end()) {
    return;
  }

  if (NS_WARN_IF(i->second != aProcessId)) {
    MOZ_ASSERT_UNREACHABLE("Unmapping layers ID for another process!");
    return;
  }

  mLayerIds.erase(i);
}

bool LayerTreeOwnerTracker::IsMapped(LayersId aLayersId,
                                     base::ProcessId aProcessId) {
  MutexAutoLock lock(mLayerIdsLock);

  auto iter = mLayerIds.find(aLayersId);
  return iter != mLayerIds.end() && iter->second == aProcessId;
}

void LayerTreeOwnerTracker::Iterate(
    const std::function<void(LayersId aLayersId, base::ProcessId aProcessId)>&
        aCallback) {
  MutexAutoLock lock(mLayerIdsLock);

  for (const auto& iter : mLayerIds) {
    aCallback(iter.first, iter.second);
  }
}

}  // namespace layers
}  // namespace mozilla