#include "cru/win/graph/direct/graph_factory.hpp"

#include "cru/win/graph/direct/brush.hpp"
#include "cru/win/graph/direct/exception.hpp"
#include "cru/win/graph/direct/font.hpp"
#include "cru/win/graph/direct/geometry.hpp"
#include "cru/win/graph/direct/text_layout.hpp"

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

namespace cru::platform::graph::win::direct {
namespace {
DirectGraphFactory* instance = nullptr;
}
}  // namespace cru::platform::graph::win::direct

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

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

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

namespace cru::platform::graph::win::direct {
DirectGraphFactory* DirectGraphFactory::GetInstance() { return instance; }

DirectGraphFactory::DirectGraphFactory() {
  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_));
}

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

D2DSolidColorBrush* DirectGraphFactory::CreateSolidColorBrush() {
  return new D2DSolidColorBrush(this);
}

D2DGeometryBuilder* DirectGraphFactory::CreateGeometryBuilder() {
  return new D2DGeometryBuilder(this);
}

DWriteFont* DirectGraphFactory::CreateFont(
    const std::wstring_view& font_family, float font_size) {
  return new DWriteFont(this, font_family, font_size);
}

DWriteTextLayout* DirectGraphFactory::CreateTextLayout(
    std::shared_ptr<Font> font, std::wstring text) {
  return new DWriteTextLayout(this, std::move(font), std::move(text));
}
}  // namespace cru::platform::graph::win::direct