#include "Render/DisplayMode.h"

#include <algorithm>
#include <d3d9.h>
#include <utility>

// OFFSET: 0x0056A590
// FUNCTION: fn_02291
std::vector<DisplayMode> EnumerateDisplayModes(int backend, unsigned deviceIndex, DirectDrawEnumeration& legacy) {
  if (backend == 1) {
    return EnumerateDisplayModesDX7(legacy, deviceIndex);
  }

  if (backend == 2) {
    return EnumerateDisplayModesDX9(deviceIndex);
  }

  return {};
}

// OFFSET: 0x0056DCF0
// FUNCTION: fn_02372
std::vector<DisplayMode> EnumerateDisplayModesDX9(UINT adapter) {
  IDirect3D9Ex* direct3D = nullptr;
  std::vector<DisplayMode> result;

  if (SUCCEEDED(Direct3DCreate9Ex(D3D_SDK_VERSION, &direct3D))) {
    const UINT count = direct3D->GetAdapterModeCount(adapter, D3DFMT_X8R8G8B8);
    std::vector<std::pair<unsigned, unsigned>> dimensions;
    dimensions.reserve(count);

    for (UINT index = 0; index < count; ++index) {
      D3DDISPLAYMODE mode;

      if (SUCCEEDED(direct3D->EnumAdapterModes(adapter, D3DFMT_X8R8G8B8, index, &mode)) && mode.Height >= 664) {
        dimensions.emplace_back(mode.Width, mode.Height);
      }
    }

    std::sort(dimensions.begin(), dimensions.end());
    dimensions.erase(std::unique(dimensions.begin(), dimensions.end()), dimensions.end());
    result.reserve(dimensions.size());

    for (const auto& [width, height] : dimensions) {
      result.push_back({width, height, 32, std::to_string(width) + " x " + std::to_string(height)});
    }
  }

  if (direct3D) {
    direct3D->Release();
  }

  return result;
}

// STATIC: 0x01348F80
// GLOBAL: gl_21922
extern const char* g_subKey;

// STATIC: 0x0173A878
// GLOBAL: gl_24397
extern unsigned char g_guidDevice[16];

// STATIC: 0x0173A888
// GLOBAL: gl_24513
extern unsigned char g_guidDriver[16];

// STATIC: 0x0173A898
// GLOBAL: gl_24514
extern char g_deviceName[40];

// OFFSET: 0x0056A7E0
// FUNCTION: fn_02297
LSTATUS SaveDisplayDeviceSettings() {
  HKEY key = nullptr;
  LSTATUS status = RegCreateKeyExA(
    HKEY_LOCAL_MACHINE,
    g_subKey,
    0,
    nullptr,
    0,
    KEY_ALL_ACCESS,
    nullptr,
    &key,
    nullptr
  );

  if (status == ERROR_SUCCESS) {
    RegSetValueExA(key, "GUIDDEVICE", 0, REG_BINARY, g_guidDevice, 16);
    RegSetValueExA(key, "GUIDDRIVER", 0, REG_BINARY, g_guidDriver, 16);
    status = RegSetValueExA(key, "DEVICENAME", 0, REG_SZ, reinterpret_cast<const BYTE*>(g_deviceName), 40);
  }

  return status;
}