#include <vector>

#include <dxgi.h>

#include <windows.h>
#include <windowsx.h>

#include "../test_utils.h"

using namespace dxvk;

int WINAPI WinMain(HINSTANCE hInstance,
                   HINSTANCE hPrevInstance,
                   LPSTR lpCmdLine,
                   int nCmdShow) {
  Com<IDXGIFactory> factory;
  
  if (CreateDXGIFactory(__uuidof(IDXGIFactory),
      reinterpret_cast<void**>(&factory)) != S_OK) {
    std::cerr << "Failed to create DXGI factory" << std::endl;
    return 1;
  }
  
  Com<IDXGIAdapter> adapter;
  
  for (UINT i = 0; factory->EnumAdapters(i, &adapter) == S_OK; i++) {
    DXGI_ADAPTER_DESC adapterDesc;
    
    if (adapter->GetDesc(&adapterDesc) != S_OK) {
      std::cerr << "Failed to get DXGI adapter info" << std::endl;
      return 1;
    }
    
    DXGI_ADAPTER_DESC desc;
    
    if (adapter->GetDesc(&desc) != S_OK) {
      std::cerr << "Failed to get DXGI adapter info" << std::endl;
      return 1;
    }
    
    std::array<char, 257> chars;
    std::wcstombs(chars.data(), desc.Description, chars.size() - 1);
    
    std::cout << str::format("Adapter ", i, ":") << std::endl;
    std::cout << str::format(" ", chars.data()) << std::endl;
    std::cout << str::format(" Vendor: ", desc.VendorId) << std::endl;
    std::cout << str::format(" Device: ", desc.DeviceId) << std::endl;
    std::cout << str::format(" Dedicated RAM: ", desc.DedicatedVideoMemory) << std::endl;
    std::cout << str::format(" Shared RAM: ", desc.SharedSystemMemory) << std::endl;
    
    Com<IDXGIOutput> output;
    
    for (UINT j = 0; adapter->EnumOutputs(j, &output) == S_OK; j++) {
      std::vector<DXGI_MODE_DESC> modes;
      
      DXGI_OUTPUT_DESC desc;
      
      if (output->GetDesc(&desc) != S_OK) {
        std::cerr << "Failed to get DXGI output info" << std::endl;
        return 1;
      }
      
      std::array<char, 257> chars;
      std::wcstombs(chars.data(), desc.DeviceName, chars.size() - 1);
      
      std::cout << str::format(" Output ", j, ":") << std::endl;
      std::cout << str::format("  ", chars.data()) << std::endl;
      std::cout << str::format("  Coordinates: ",
        desc.DesktopCoordinates.left, ",",
        desc.DesktopCoordinates.top, ":",
        desc.DesktopCoordinates.right - desc.DesktopCoordinates.left, "x",
        desc.DesktopCoordinates.bottom - desc.DesktopCoordinates.top) << std::endl;
      
      HRESULT status = S_OK;
      UINT    displayModeCount = 0;
      
      do {
        if (output->GetDisplayModeList(
          DXGI_FORMAT_R8G8B8A8_UNORM,
          DXGI_ENUM_MODES_SCALING,
          &displayModeCount, nullptr) != S_OK) {
          std::cerr << "Failed to get DXGI output display mode count" << std::endl;
          return 1;
        }
        
        modes.resize(displayModeCount);
        
        status = output->GetDisplayModeList(
          DXGI_FORMAT_R8G8B8A8_UNORM,
          DXGI_ENUM_MODES_SCALING,
          &displayModeCount, modes.data());
      } while (status == DXGI_ERROR_MORE_DATA);
      
      if (status != S_OK) {
        std::cerr << "Failed to get DXGI output display mode list" << std::endl;
        return 1;
      }
      
      for (auto mode : modes) {
        std::cout << str::format("  ",
          mode.Width, "x", mode.Height, " @ ",
          mode.RefreshRate.Numerator / mode.RefreshRate.Denominator,
          mode.Scaling == DXGI_MODE_SCALING_CENTERED ? " (native)" : "") << std::endl;

        //test matching modes
        DXGI_MODE_DESC matched_mode{ 0 };
        status = output->FindClosestMatchingMode(&mode, &matched_mode, nullptr);

        if (status != S_OK) {
            std::cerr << "Failed to get matching mode" << std::endl;
            return 1;
        }
        
        if (matched_mode.Width != mode.Width ||
            matched_mode.Height != mode.Height ||
            matched_mode.RefreshRate.Numerator != mode.RefreshRate.Numerator ||
            matched_mode.RefreshRate.Denominator != mode.RefreshRate.Denominator ||
            matched_mode.Format != mode.Format)
        {
            std::cerr << "Matched mode is incorrect" << std::endl;
            return 1;
        }
      }
    }
  }
  
  return 0;
}