#include <cstdlib>
#include <cstring>

#include "dxgi_adapter.h"
#include "dxgi_enums.h"
#include "dxgi_factory.h"
#include "dxgi_output.h"

#include "../dxvk/vulkan/dxvk_vulkan_names.h"

namespace dxvk {

  DxgiAdapter::DxgiAdapter(
          DxgiFactory*      factory,
    const Rc<DxvkAdapter>&  adapter)
  : m_factory (factory),
    m_adapter (adapter) {
    SetupFormatTable();
  }
  
  
  DxgiAdapter::~DxgiAdapter() {
    
  }
  
  
  HRESULT DxgiAdapter::QueryInterface(
          REFIID riid,
          void **ppvObject) {
    COM_QUERY_IFACE(riid, ppvObject, IUnknown);
    COM_QUERY_IFACE(riid, ppvObject, IDXGIObject);
    COM_QUERY_IFACE(riid, ppvObject, IDXGIAdapter);
    COM_QUERY_IFACE(riid, ppvObject, IDXGIAdapter1);
    COM_QUERY_IFACE(riid, ppvObject, IDXGIAdapterPrivate);
    
    Logger::warn("DxgiAdapter::QueryInterface: Unknown interface query");
    return E_NOINTERFACE;
  }
  
  
  HRESULT DxgiAdapter::GetParent(
          REFIID riid,
          void   **ppParent) {
    return m_factory->QueryInterface(riid, ppParent);
  }
  
  
  HRESULT DxgiAdapter::CheckInterfaceSupport(
          REFGUID       InterfaceName,
          LARGE_INTEGER *pUMDVersion) {
    Logger::err("DxgiAdapter::CheckInterfaceSupport: No D3D10 support");
    return DXGI_ERROR_UNSUPPORTED;
  }
  
  
  HRESULT DxgiAdapter::EnumOutputs(
          UINT        Output,
          IDXGIOutput **ppOutput) {
    if (ppOutput == nullptr)
      return DXGI_ERROR_INVALID_CALL;
    
    int numDisplays = SDL_GetNumVideoDisplays();
    
    if (numDisplays < 0) {
      Logger::err("DxgiAdapter::EnumOutputs: Failed to query display count");
      return DXGI_ERROR_DRIVER_INTERNAL_ERROR;
    }
    
    if (Output >= static_cast<uint32_t>(numDisplays))
      return DXGI_ERROR_NOT_FOUND;
    
    *ppOutput = ref(new DxgiOutput(this, Output));
    return S_OK;
  }
  
  
  HRESULT DxgiAdapter::GetDesc(DXGI_ADAPTER_DESC* pDesc) {
    DXGI_ADAPTER_DESC1 desc1;
    HRESULT hr = this->GetDesc1(&desc1);
    
    if (SUCCEEDED(hr)) {
      std::memcpy(
        pDesc->Description,
        desc1.Description,
        sizeof(pDesc->Description));
      
      pDesc->VendorId               = desc1.VendorId;
      pDesc->DeviceId               = desc1.DeviceId;
      pDesc->SubSysId               = desc1.SubSysId;
      pDesc->Revision               = desc1.Revision;
      pDesc->DedicatedVideoMemory   = desc1.DedicatedVideoMemory;
      pDesc->DedicatedSystemMemory  = desc1.DedicatedSystemMemory;
      pDesc->SharedSystemMemory     = desc1.SharedSystemMemory;
      pDesc->AdapterLuid            = desc1.AdapterLuid;
    }
    
    return hr;
  }
  
  
  HRESULT DxgiAdapter::GetDesc1(DXGI_ADAPTER_DESC1* pDesc) {
    if (pDesc == nullptr)
      return DXGI_ERROR_INVALID_CALL;
    
    const auto deviceProp = m_adapter->deviceProperties();
    const auto memoryProp = m_adapter->memoryProperties();
    
    std::memset(pDesc->Description, 0, sizeof(pDesc->Description));
    std::mbstowcs(pDesc->Description, deviceProp.deviceName, _countof(pDesc->Description) - 1);
    
    VkDeviceSize deviceMemory = 0;
    VkDeviceSize sharedMemory = 0;
    
    for (uint32_t i = 0; i < memoryProp.memoryHeapCount; i++) {
      VkMemoryHeap heap = memoryProp.memoryHeaps[i];
      
      if (heap.flags & VK_MEMORY_HEAP_DEVICE_LOCAL_BIT)
        deviceMemory += heap.size;
      else
        sharedMemory += heap.size;
    }
    
    pDesc->VendorId               = deviceProp.vendorID;
    pDesc->DeviceId               = deviceProp.deviceID;
    pDesc->SubSysId               = 0;
    pDesc->Revision               = 0;
    pDesc->DedicatedVideoMemory   = deviceMemory;
    pDesc->DedicatedSystemMemory  = 0;
    pDesc->SharedSystemMemory     = sharedMemory;
    pDesc->AdapterLuid            = LUID { 0, 0 };  // TODO implement
    pDesc->Flags                  = 0;
    return S_OK;
  }
  
  
  Rc<DxvkAdapter> DxgiAdapter::GetDXVKAdapter() {
    return m_adapter;
  }
  
  
  DxgiFormatPair DxgiAdapter::LookupFormat(DXGI_FORMAT format) {
    auto pair = m_formats.find(format);
    
    return pair != m_formats.end()
      ? pair->second
      : DxgiFormatPair();
  }
  
  
  void DxgiAdapter::AddFormat(
          DXGI_FORMAT                       srcFormat,
          VkFormat                          dstFormat) {
    DxgiFormatPair formatPair;
    formatPair.wanted = dstFormat;
    formatPair.actual = dstFormat;
    m_formats.insert(std::make_pair(srcFormat, formatPair));
  }
  
  
  void DxgiAdapter::AddFormat(
          DXGI_FORMAT                       srcFormat,
          VkFormat                          dstFormat,
    const std::initializer_list<VkFormat>&  fallbacks,
          VkFormatFeatureFlags              features) {
    DxgiFormatPair formatPair;
    formatPair.wanted = dstFormat;
    formatPair.actual = VK_FORMAT_UNDEFINED;
    
    if (this->HasFormatSupport(dstFormat, features)) {
      formatPair.actual = dstFormat;
    } else {
      for (VkFormat fmt : fallbacks) {
        if (this->HasFormatSupport(fmt, features)) {
          formatPair.actual = fmt;
          break;
        }
      }
    }
    
    if (formatPair.actual == VK_FORMAT_UNDEFINED)
      Logger::err(str::format("DxgiAdapter: ", srcFormat, " not supported"));
    else if (formatPair.actual != formatPair.wanted)
      Logger::warn(str::format("DxgiAdapter: ", srcFormat, " -> ", formatPair.actual));
    
    m_formats.insert(std::make_pair(srcFormat, formatPair));
  }
  
  
  void DxgiAdapter::SetupFormatTable() {
    AddFormat(DXGI_FORMAT_UNKNOWN,                    VK_FORMAT_UNDEFINED);
    
    AddFormat(DXGI_FORMAT_R32G32B32A32_TYPELESS,      VK_FORMAT_R32G32B32A32_UINT);
    AddFormat(DXGI_FORMAT_R32G32B32A32_FLOAT,         VK_FORMAT_R32G32B32A32_SFLOAT);
    AddFormat(DXGI_FORMAT_R32G32B32A32_UINT,          VK_FORMAT_R32G32B32A32_UINT);
    AddFormat(DXGI_FORMAT_R32G32B32A32_SINT,          VK_FORMAT_R32G32B32A32_SINT);
    
    AddFormat(DXGI_FORMAT_R32G32B32_TYPELESS,         VK_FORMAT_R32G32B32_UINT);
    AddFormat(DXGI_FORMAT_R32G32B32_FLOAT,            VK_FORMAT_R32G32B32_SFLOAT);
    AddFormat(DXGI_FORMAT_R32G32B32_UINT,             VK_FORMAT_R32G32B32_UINT);
    AddFormat(DXGI_FORMAT_R32G32B32_SINT,             VK_FORMAT_R32G32B32_SINT);
    
    AddFormat(DXGI_FORMAT_R16G16B16A16_TYPELESS,      VK_FORMAT_R16G16B16A16_UINT);
    AddFormat(DXGI_FORMAT_R16G16B16A16_FLOAT,         VK_FORMAT_R16G16B16A16_SFLOAT);
    AddFormat(DXGI_FORMAT_R16G16B16A16_UNORM,         VK_FORMAT_R16G16B16A16_UNORM);
    AddFormat(DXGI_FORMAT_R16G16B16A16_UINT,          VK_FORMAT_R16G16B16A16_UINT);
    AddFormat(DXGI_FORMAT_R16G16B16A16_SNORM,         VK_FORMAT_R16G16B16A16_SNORM);
    AddFormat(DXGI_FORMAT_R16G16B16A16_SINT,          VK_FORMAT_R16G16B16A16_SINT);
    
    AddFormat(DXGI_FORMAT_R32G32_TYPELESS,            VK_FORMAT_R32G32_UINT);
    AddFormat(DXGI_FORMAT_R32G32_FLOAT,               VK_FORMAT_R32G32_SFLOAT);
    AddFormat(DXGI_FORMAT_R32G32_UINT,                VK_FORMAT_R32G32_UINT);
    AddFormat(DXGI_FORMAT_R32G32_SINT,                VK_FORMAT_R32G32_SINT);
    
    AddFormat(DXGI_FORMAT_D32_FLOAT_S8X24_UINT,       VK_FORMAT_D32_SFLOAT_S8_UINT);
//     AddFormat(DXGI_FORMAT_R32G8X24_TYPELESS,          VK_FORMAT_UNDEFINED);
//     AddFormat(DXGI_FORMAT_R32_FLOAT_X8X24_TYPELESS,   VK_FORMAT_UNDEFINED);
//     AddFormat(DXGI_FORMAT_X32_TYPELESS_G8X24_UINT,    VK_FORMAT_UNDEFINED);
    
    AddFormat(DXGI_FORMAT_R10G10B10A2_TYPELESS,       VK_FORMAT_A2R10G10B10_UINT_PACK32);
    AddFormat(DXGI_FORMAT_R10G10B10A2_UINT,           VK_FORMAT_A2R10G10B10_UINT_PACK32);
    AddFormat(DXGI_FORMAT_R10G10B10A2_UNORM,          VK_FORMAT_A2R10G10B10_UNORM_PACK32);
    
    AddFormat(DXGI_FORMAT_R11G11B10_FLOAT,            VK_FORMAT_B10G11R11_UFLOAT_PACK32);
    
    AddFormat(DXGI_FORMAT_R8G8B8A8_TYPELESS,          VK_FORMAT_R8G8B8A8_UINT);
    AddFormat(DXGI_FORMAT_R8G8B8A8_UNORM,             VK_FORMAT_R8G8B8A8_UNORM);
    AddFormat(DXGI_FORMAT_R8G8B8A8_UNORM_SRGB,        VK_FORMAT_R8G8B8A8_SRGB);
    AddFormat(DXGI_FORMAT_R8G8B8A8_UINT,              VK_FORMAT_R8G8B8A8_UINT);
    AddFormat(DXGI_FORMAT_R8G8B8A8_SNORM,             VK_FORMAT_R8G8B8A8_SNORM);
    AddFormat(DXGI_FORMAT_R8G8B8A8_SINT,              VK_FORMAT_R8G8B8A8_SINT);
    
    AddFormat(DXGI_FORMAT_R16G16_TYPELESS,            VK_FORMAT_R16G16_UINT);
    AddFormat(DXGI_FORMAT_R16G16_FLOAT,               VK_FORMAT_R16G16_SFLOAT);
    AddFormat(DXGI_FORMAT_R16G16_UNORM,               VK_FORMAT_R16G16_UNORM);
    AddFormat(DXGI_FORMAT_R16G16_UINT,                VK_FORMAT_R16G16_UINT);
    AddFormat(DXGI_FORMAT_R16G16_SNORM,               VK_FORMAT_R16G16_SNORM);
    AddFormat(DXGI_FORMAT_R16G16_SINT,                VK_FORMAT_R16G16_SINT);
    
    AddFormat(DXGI_FORMAT_R32_TYPELESS,               VK_FORMAT_R32_UINT);
    AddFormat(DXGI_FORMAT_R32_FLOAT,                  VK_FORMAT_R32_SFLOAT);
    AddFormat(DXGI_FORMAT_R32_UINT,                   VK_FORMAT_R32_UINT);
    AddFormat(DXGI_FORMAT_R32_SINT,                   VK_FORMAT_R32_SINT);
    
    AddFormat(DXGI_FORMAT_R8G8_TYPELESS,              VK_FORMAT_R8G8_UINT);
    AddFormat(DXGI_FORMAT_R8G8_UNORM,                 VK_FORMAT_R8G8_UNORM);
    AddFormat(DXGI_FORMAT_R8G8_UINT,                  VK_FORMAT_R8G8_UINT);
    AddFormat(DXGI_FORMAT_R8G8_SNORM,                 VK_FORMAT_R8G8_SNORM);
    AddFormat(DXGI_FORMAT_R8G8_SINT,                  VK_FORMAT_R8G8_SINT);
    
    AddFormat(DXGI_FORMAT_R16_TYPELESS,               VK_FORMAT_R16_UINT);
    AddFormat(DXGI_FORMAT_R16_FLOAT,                  VK_FORMAT_R16_SFLOAT);
    AddFormat(DXGI_FORMAT_R16_UNORM,                  VK_FORMAT_R16_UNORM);
    AddFormat(DXGI_FORMAT_R16_UINT,                   VK_FORMAT_R16_UINT);
    AddFormat(DXGI_FORMAT_R16_SNORM,                  VK_FORMAT_R16_SNORM);
    AddFormat(DXGI_FORMAT_R16_SINT,                   VK_FORMAT_R16_SINT);
    
    AddFormat(DXGI_FORMAT_R8_TYPELESS,                VK_FORMAT_R8_UINT);
    AddFormat(DXGI_FORMAT_R8_UNORM,                   VK_FORMAT_R8_UNORM);
    AddFormat(DXGI_FORMAT_R8_UINT,                    VK_FORMAT_R8_UINT);
    AddFormat(DXGI_FORMAT_R8_SNORM,                   VK_FORMAT_R8_SNORM);
    AddFormat(DXGI_FORMAT_R8_SINT,                    VK_FORMAT_R8_SINT);
//     AddFormat(DXGI_FORMAT_A8_UNORM,                   VK_FORMAT_UNDEFINED);
    
//     AddFormat(DXGI_FORMAT_R1_UNORM,                   VK_FORMAT_UNDEFINED);
    
    AddFormat(DXGI_FORMAT_R9G9B9E5_SHAREDEXP,         VK_FORMAT_E5B9G9R9_UFLOAT_PACK32);
//     AddFormat(DXGI_FORMAT_R8G8_B8G8_UNORM,            VK_FORMAT_UNDEFINED);
    AddFormat(DXGI_FORMAT_G8R8_G8B8_UNORM,            VK_FORMAT_UNDEFINED);
    
    AddFormat(DXGI_FORMAT_BC1_TYPELESS,               VK_FORMAT_BC1_RGBA_UNORM_BLOCK);
    AddFormat(DXGI_FORMAT_BC1_UNORM,                  VK_FORMAT_BC1_RGBA_UNORM_BLOCK);
    AddFormat(DXGI_FORMAT_BC1_UNORM_SRGB,             VK_FORMAT_BC1_RGBA_SRGB_BLOCK);
    
    AddFormat(DXGI_FORMAT_BC2_TYPELESS,               VK_FORMAT_BC2_UNORM_BLOCK);
    AddFormat(DXGI_FORMAT_BC2_UNORM,                  VK_FORMAT_BC2_UNORM_BLOCK);
    AddFormat(DXGI_FORMAT_BC2_UNORM_SRGB,             VK_FORMAT_BC2_SRGB_BLOCK);
    
    AddFormat(DXGI_FORMAT_BC3_TYPELESS,               VK_FORMAT_BC3_UNORM_BLOCK);
    AddFormat(DXGI_FORMAT_BC3_UNORM,                  VK_FORMAT_BC3_UNORM_BLOCK);
    AddFormat(DXGI_FORMAT_BC3_UNORM_SRGB,             VK_FORMAT_BC3_SRGB_BLOCK);
    
    AddFormat(DXGI_FORMAT_BC4_TYPELESS,               VK_FORMAT_BC4_UNORM_BLOCK);
    AddFormat(DXGI_FORMAT_BC4_UNORM,                  VK_FORMAT_BC4_UNORM_BLOCK);
    AddFormat(DXGI_FORMAT_BC4_SNORM,                  VK_FORMAT_BC4_SNORM_BLOCK);
    
    AddFormat(DXGI_FORMAT_BC5_TYPELESS,               VK_FORMAT_BC5_UNORM_BLOCK);
    AddFormat(DXGI_FORMAT_BC5_UNORM,                  VK_FORMAT_BC5_UNORM_BLOCK);
    AddFormat(DXGI_FORMAT_BC5_SNORM,                  VK_FORMAT_BC5_SNORM_BLOCK);
    
    AddFormat(DXGI_FORMAT_B5G6R5_UNORM,               VK_FORMAT_B5G6R5_UNORM_PACK16);
    AddFormat(DXGI_FORMAT_B5G5R5A1_UNORM,             VK_FORMAT_B5G5R5A1_UNORM_PACK16);
    
    AddFormat(DXGI_FORMAT_B8G8R8A8_TYPELESS,          VK_FORMAT_B8G8R8A8_UNORM);
    AddFormat(DXGI_FORMAT_B8G8R8A8_UNORM,             VK_FORMAT_B8G8R8A8_UNORM);
    AddFormat(DXGI_FORMAT_B8G8R8A8_UNORM_SRGB,        VK_FORMAT_B8G8R8A8_SRGB);
    
    // TODO implement component swizzle
    AddFormat(DXGI_FORMAT_B8G8R8X8_UNORM,             VK_FORMAT_B8G8R8A8_UNORM);
    AddFormat(DXGI_FORMAT_B8G8R8X8_TYPELESS,          VK_FORMAT_B8G8R8A8_UNORM);
    AddFormat(DXGI_FORMAT_B8G8R8X8_UNORM_SRGB,        VK_FORMAT_B8G8R8A8_SRGB);
    
//     AddFormat(DXGI_FORMAT_R10G10B10_XR_BIAS_A2_UNORM, VK_FORMAT_UNDEFINED);
    
    AddFormat(DXGI_FORMAT_BC6H_TYPELESS,              VK_FORMAT_BC6H_UFLOAT_BLOCK);
    AddFormat(DXGI_FORMAT_BC6H_UF16,                  VK_FORMAT_BC6H_UFLOAT_BLOCK);
    AddFormat(DXGI_FORMAT_BC6H_SF16,                  VK_FORMAT_BC6H_SFLOAT_BLOCK);
    
    AddFormat(DXGI_FORMAT_BC7_TYPELESS,               VK_FORMAT_BC7_UNORM_BLOCK);
    AddFormat(DXGI_FORMAT_BC7_UNORM,                  VK_FORMAT_BC7_UNORM_BLOCK);
    AddFormat(DXGI_FORMAT_BC7_UNORM_SRGB,             VK_FORMAT_BC7_SRGB_BLOCK);
    
    AddFormat(
      DXGI_FORMAT_D32_FLOAT,
      VK_FORMAT_D32_SFLOAT, {
        VK_FORMAT_D32_SFLOAT_S8_UINT,
      },
      VK_FORMAT_FEATURE_SAMPLED_IMAGE_BIT |
      VK_FORMAT_FEATURE_DEPTH_STENCIL_ATTACHMENT_BIT);
    
    AddFormat(
      DXGI_FORMAT_D24_UNORM_S8_UINT,
      VK_FORMAT_D24_UNORM_S8_UINT, {
        VK_FORMAT_D32_SFLOAT_S8_UINT,
      },
      VK_FORMAT_FEATURE_SAMPLED_IMAGE_BIT |
      VK_FORMAT_FEATURE_DEPTH_STENCIL_ATTACHMENT_BIT);
    
    AddFormat(DXGI_FORMAT_D16_UNORM,                  VK_FORMAT_D16_UNORM);
    
//     AddFormat(DXGI_FORMAT_D24_UNORM_S8_UINT,          VK_FORMAT_UNDEFINED);
//     AddFormat(DXGI_FORMAT_R24_UNORM_X8_TYPELESS,      VK_FORMAT_UNDEFINED);
//     AddFormat(DXGI_FORMAT_X24_TYPELESS_G8_UINT,       VK_FORMAT_UNDEFINED);
    
    // TODO finish me
  }
  
  
  bool DxgiAdapter::HasFormatSupport(
          VkFormat                          format,
          VkFormatFeatureFlags              features) const {
    VkFormatProperties info = m_adapter->formatProperties(format);
    return ((info.optimalTilingFeatures | info.bufferFeatures) & features) == features;
  }
  
}