SyncObjectsD3D11.cpp

#include "Foundation/Logging/Logger.h"
#include "SyncObjectsD3D11.h"
#include "GraphicsDeviceD3D11.h"
 
namespace {
  Cogs::Logging::Log logger = Cogs::Logging::getLogger("SyncObjectsD3D11");
}
 
 
void Cogs::SyncObjectsD3D11::setDevice(ResourcePointer<ID3D11Device5>& device5)
{
  this->device5 = device5;
}
 
Cogs::FenceHandle Cogs::SyncObjectsD3D11::createFence()
{
  if (device5) {
    FenceD3D11 fence{};
    if (HRESULT hr = device5->CreateFence(0, D3D11_FENCE_FLAG_NONE, IID_PPV_ARGS(fence.fence.internalPointer())); SUCCEEDED(hr)) {
      return fences.addResource(fence);
    }
  }
  return Cogs::FenceHandle::NoHandle;
}
 
void Cogs::SyncObjectsD3D11::releaseFence(FenceHandle fenceHandle)
{
  if (HandleIsValid(fenceHandle)) {
    FenceD3D11& fence = fences[fenceHandle];
    if (fence.event) {
      CloseHandle(fence.event);
    }
    fences.removeResource(fenceHandle);
  }
}
 
void Cogs::SyncObjectsD3D11::resetFence(FenceHandle fenceHandle)
{
  if (HandleIsValid(fenceHandle)) {
    FenceD3D11& fence = fences[fenceHandle];
    fence.value++;
  }
}
 
bool Cogs::SyncObjectsD3D11::isFenceSignalled(FenceHandle fenceHandle)
{
  if (!HandleIsValid(fenceHandle)) {
    LOG_WARNING_ONCE(logger, "isFenceSignalled: fenceHandle is invalid.");
    return false;
  }
 
  FenceD3D11& fence = fences[fenceHandle];
  return fence.value == fence.fence->GetCompletedValue();
}
 
bool Cogs::SyncObjectsD3D11::waitFence(FenceHandle fenceHandle, uint32_t timeout_ms)
{
  if (!HandleIsValid(fenceHandle)) {
    LOG_WARNING_ONCE(logger, "waitFence: fenceHandle is invalid.");
    return false;
  }
 
  // First, try early exit by checking if fence is already signalled.
  FenceD3D11& fence = fences[fenceHandle];
  if (fence.value == fence.fence->GetCompletedValue()) {
    return true;
  }
 
  // Make sure we have an unsignalled event and tie this to the fence
  if (fence.event == nullptr) {
    fence.event = CreateEvent(nullptr, TRUE, FALSE, nullptr);
    assert(fence.event);
  }
  else if (ResetEvent(fence.event) == 0) {
    LOG_WARNING(logger, "ResetEvent failed: 0x%x", GetLastError());
    return false;
  }
  fence.fence->SetEventOnCompletion(fence.value, fence.event);
 
  // Wait
  DWORD result = WaitForSingleObject(fence.event, timeout_ms);
  switch (result) {
  case WAIT_OBJECT_0:
    return true;
 
  case WAIT_TIMEOUT:
    return false;
 
  default:
    LOG_WARNING(logger, "WaitForSingleObject failed: 0x%x", GetLastError());
    return false;
  }
}