#include "cru/win/graph/win_graph_factory.hpp"

#include "cru/win/exception.hpp"
#include "cru/win/graph/win_brush.hpp"
#include "cru/win/graph/win_font.hpp"
#include "cru/win/graph/win_geometry.hpp"
#include "cru/win/graph/win_text_layout.hpp"

#include <cassert>
#include <cstdlib>
#include <utility>

namespace cru::win::graph {
namespace {
WinGraphFactory* instance = nullptr;
}
}  // namespace cru::win::graph

namespace cru::platform::graph {
void GraphFactoryAutoDeleteExitHandler() {
  const auto i = ::cru::win::graph::instance;  // avoid long namespace prefix
  if (i == nullptr) return;
  if (i->IsAutoDelete()) delete i;
}

IGraphFactory* IGraphFactory::CreateInstance() {
  auto& i = ::cru::win::graph::instance;  // avoid long namespace prefix
  assert(i == nullptr);
  i = new cru::win::graph::WinGraphFactory();
  std::atexit(&GraphFactoryAutoDeleteExitHandler);
  return i;
}

IGraphFactory* IGraphFactory::GetInstance() {
  return ::cru::win::graph::instance;
}
}  // namespace cru::platform::graph

namespace cru::win::graph {
WinGraphFactory* WinGraphFactory::GetInstance() { return instance; }

WinGraphFactory::WinGraphFactory() {
  UINT creation_flags = D3D11_CREATE_DEVICE_BGRA_SUPPORT;

#ifdef CRU_DEBUG
  creation_flags |= D3D11_CREATE_DEVICE_DEBUG;
#endif

  const D3D_FEATURE_LEVEL feature_levels[] = {
      D3D_FEATURE_LEVEL_11_1, D3D_FEATURE_LEVEL_11_0, D3D_FEATURE_LEVEL_10_1,
      D3D_FEATURE_LEVEL_10_0, D3D_FEATURE_LEVEL_9_3,  D3D_FEATURE_LEVEL_9_2,
      D3D_FEATURE_LEVEL_9_1};

  Microsoft::WRL::ComPtr<ID3D11DeviceContext> d3d11_device_context;

  ThrowIfFailed(D3D11CreateDevice(
      nullptr, D3D_DRIVER_TYPE_HARDWARE, nullptr, creation_flags,
      feature_levels, ARRAYSIZE(feature_levels), D3D11_SDK_VERSION,
      &d3d11_device_, nullptr, &d3d11_device_context));

  Microsoft::WRL::ComPtr<IDXGIDevice> dxgi_device;
  ThrowIfFailed(d3d11_device_->QueryInterface(dxgi_device.GetAddressOf()));

  ThrowIfFailed(D2D1CreateFactory(D2D1_FACTORY_TYPE_SINGLE_THREADED,
                                  IID_PPV_ARGS(&d2d1_factory_)));

  Microsoft::WRL::ComPtr<ID2D1Device> d2d1_device;

  ThrowIfFailed(d2d1_factory_->CreateDevice(dxgi_device.Get(), &d2d1_device));

  ThrowIfFailed(d2d1_device->CreateDeviceContext(
      D2D1_DEVICE_CONTEXT_OPTIONS_NONE, &d2d1_device_context_));

  // Identify the physical adapter (GPU or card) this device is runs on.
  Microsoft::WRL::ComPtr<IDXGIAdapter> dxgi_adapter;
  ThrowIfFailed(dxgi_device->GetAdapter(&dxgi_adapter));

  // Get the factory object that created the DXGI device.
  ThrowIfFailed(dxgi_adapter->GetParent(IID_PPV_ARGS(&dxgi_factory_)));

  ThrowIfFailed(DWriteCreateFactory(
      DWRITE_FACTORY_TYPE_SHARED, __uuidof(IDWriteFactory),
      reinterpret_cast<IUnknown**>(dwrite_factory_.GetAddressOf())));

  ThrowIfFailed(dwrite_factory_->GetSystemFontCollection(
      &dwrite_system_font_collection_));
}

WinGraphFactory::~WinGraphFactory() { instance = nullptr; }

platform::graph::ISolidColorBrush* WinGraphFactory::CreateSolidColorBrush(
    const ui::Color& color) {
  return new WinSolidColorBrush(this, color);
}

platform::graph::IGeometryBuilder* WinGraphFactory::CreateGeometryBuilder() {
  return new WinGeometryBuilder(this);
}

platform::graph::IFontDescriptor* WinGraphFactory::CreateFontDescriptor(
    const std::wstring_view& font_family, float font_size) {
  return new WinFontDescriptor(this, font_family, font_size);
}

platform::graph::ITextLayout* WinGraphFactory::CreateTextLayout(
    std::shared_ptr<platform::graph::IFontDescriptor> font, std::wstring text) {
  const auto f = std::dynamic_pointer_cast<WinFontDescriptor>(font);
  assert(f);
  return new WinTextLayout(this, std::move(f), std::move(text));
}
}  // namespace cru::win::graph