/**
 * @file tools/dxgi.cpp
 * @brief Displays information about connected displays and GPUs
 */
#define WINVER 0x0A00
#include "src/platform/windows/tools/helper.h"
#include "src/utility.h"

#include <d3dcommon.h>
#include <dxgi.h>
#include <format>
#include <iostream>

using namespace std::literals;

namespace dxgi {
  template<class T>
  void Release(T *dxgi) {
    dxgi->Release();
  }

  using factory1_t = util::safe_ptr<IDXGIFactory1, Release<IDXGIFactory1>>;
  using adapter_t = util::safe_ptr<IDXGIAdapter1, Release<IDXGIAdapter1>>;
  using output_t = util::safe_ptr<IDXGIOutput, Release<IDXGIOutput>>;
}  // namespace dxgi

int main(int argc, char *argv[]) {
  // Set ourselves as per-monitor DPI aware for accurate resolution values on High DPI systems
  SetProcessDpiAwarenessContext(DPI_AWARENESS_CONTEXT_PER_MONITOR_AWARE_V2);

  dxgi::factory1_t::pointer factory_p {};
  const HRESULT status = CreateDXGIFactory1(IID_IDXGIFactory1, static_cast<void **>(static_cast<void *>(&factory_p)));
  dxgi::factory1_t factory {factory_p};
  if (FAILED(status)) {
    std::cout << "Failed to create DXGIFactory1 [0x"sv << util::hex(status).to_string_view() << ']' << std::endl;
    return -1;
  }

  dxgi::adapter_t::pointer adapter_p {};
  for (int x = 0; factory->EnumAdapters1(x, &adapter_p) != DXGI_ERROR_NOT_FOUND; ++x) {
    dxgi::adapter_t adapter {adapter_p};

    DXGI_ADAPTER_DESC1 adapter_desc;
    adapter->GetDesc1(&adapter_desc);

    std::cout << "====== ADAPTER =====" << std::endl;
    output::output_field("Device Name      ", adapter_desc.Description);
    output::output_field("Device Vendor ID ", "0x" + util::hex(adapter_desc.VendorId).to_string());
    output::output_field("Device Device ID ", "0x" + util::hex(adapter_desc.DeviceId).to_string());
    output::output_field("Device Video Mem ", std::format("{} MiB", adapter_desc.DedicatedVideoMemory / 1048576));
    output::output_field("Device Sys Mem   ", std::format("{} MiB", adapter_desc.DedicatedSystemMemory / 1048576));
    output::output_field("Share Sys Mem    ", std::format("{} MiB", adapter_desc.SharedSystemMemory / 1048576));

    dxgi::output_t::pointer output_p {};
    bool has_outputs = false;
    for (int y = 0; adapter->EnumOutputs(y, &output_p) != DXGI_ERROR_NOT_FOUND; ++y) {
      // Print the header only when we find the first output
      if (!has_outputs) {
        std::cout << std::endl
                  << "    ====== OUTPUT ======" << std::endl;
        has_outputs = true;
      }

      dxgi::output_t output {output_p};

      DXGI_OUTPUT_DESC desc;
      output->GetDesc(&desc);

      auto width = desc.DesktopCoordinates.right - desc.DesktopCoordinates.left;
      auto height = desc.DesktopCoordinates.bottom - desc.DesktopCoordinates.top;

      output::output_field("    Output Name       ", desc.DeviceName);
      output::output_field("    AttachedToDesktop ", desc.AttachedToDesktop ? "yes" : "no");
      output::output_field("    Resolution        ", std::format("{}x{}", width, height));
    }
    std::cout << std::endl;
  }

  return 0;
}