/*
  ==============================================================================

   This file is part of the JUCE framework.
   Copyright (c) Raw Material Software Limited

   JUCE is an open source framework subject to commercial or open source
   licensing.

   By downloading, installing, or using the JUCE framework, or combining the
   JUCE framework with any other source code, object code, content or any other
   copyrightable work, you agree to the terms of the JUCE End User Licence
   Agreement, and all incorporated terms including the JUCE Privacy Policy and
   the JUCE Website Terms of Service, as applicable, which will bind you. If you
   do not agree to the terms of these agreements, we will not license the JUCE
   framework to you, and you must discontinue the installation or download
   process and cease use of the JUCE framework.

   JUCE End User Licence Agreement: https://juce.com/legal/juce-8-licence/
   JUCE Privacy Policy: https://juce.com/juce-privacy-policy
   JUCE Website Terms of Service: https://juce.com/juce-website-terms-of-service/

   Or:

   You may also use this code under the terms of the AGPLv3:
   https://www.gnu.org/licenses/agpl-3.0.en.html

   THE JUCE FRAMEWORK IS PROVIDED "AS IS" WITHOUT ANY WARRANTY, AND ALL
   WARRANTIES, WHETHER EXPRESSED OR IMPLIED, INCLUDING WARRANTY OF
   MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE, ARE DISCLAIMED.

  ==============================================================================
*/

namespace juce
{

constexpr auto enableDirectXDebugLayer = false;

//==============================================================================
void DirectWriteGlyphRun::replace (Span<const Point<float>> positions, float scale)
{
    advances.resize (positions.size(), 0.0f);
    offsets.resize (positions.size());
    std::transform (positions.begin(), positions.end(), offsets.begin(), [&] (auto& g)
    {
        return DWRITE_GLYPH_OFFSET { g.x / scale, -g.y };
    });
}

//==============================================================================
auto DxgiAdapter::create (ComSmartPtr<ID2D1Factory2> d2dFactory,
                          ComSmartPtr<IDXGIAdapter1> dxgiAdapterIn) -> Ptr
{
    if (dxgiAdapterIn == nullptr || d2dFactory == nullptr)
        return {};

    Ptr result = new DxgiAdapter;
    result->dxgiAdapter = dxgiAdapterIn;

    for (UINT i = 0;; ++i)
    {
        ComSmartPtr<IDXGIOutput> output;
        const auto hr = result->dxgiAdapter->EnumOutputs (i, output.resetAndGetPointerAddress());

        if (hr == DXGI_ERROR_NOT_FOUND || hr == DXGI_ERROR_NOT_CURRENTLY_AVAILABLE)
            break;

        result->dxgiOutputs.push_back (output);
    }

    // This flag adds support for surfaces with a different color channel ordering
    // than the API default. It is required for compatibility with Direct2D.
    const auto creationFlags = D3D11_CREATE_DEVICE_BGRA_SUPPORT
                             | (enableDirectXDebugLayer ? D3D11_CREATE_DEVICE_DEBUG : 0);

    if (const auto hr = D3D11CreateDevice (result->dxgiAdapter,
                                           D3D_DRIVER_TYPE_UNKNOWN,
                                           nullptr,
                                           creationFlags,
                                           nullptr,
                                           0,
                                           D3D11_SDK_VERSION,
                                           result->direct3DDevice.resetAndGetPointerAddress(),
                                           nullptr,
                                           nullptr); FAILED (hr))
    {
        return {};
    }

    if (const auto hr = result->direct3DDevice->QueryInterface (result->dxgiDevice.resetAndGetPointerAddress()); FAILED (hr))
        return {};

    if (const auto hr = d2dFactory->CreateDevice (result->dxgiDevice, result->direct2DDevice.resetAndGetPointerAddress()); FAILED (hr))
        return {};

    return result;
}

bool DxgiAdapter::uniqueIDMatches (ReferenceCountedObjectPtr<DxgiAdapter> other) const
{
    if (other == nullptr)
        return false;

    auto luid = getAdapterUniqueID();
    auto otherLuid = other->getAdapterUniqueID();
    return (luid.HighPart == otherLuid.HighPart) && (luid.LowPart == otherLuid.LowPart);
}

LUID DxgiAdapter::getAdapterUniqueID() const
{
    DXGI_ADAPTER_DESC1 desc;

    if (auto hr = dxgiAdapter->GetDesc1 (&desc); SUCCEEDED (hr))
        return desc.AdapterLuid;

    return LUID { 0, 0 };
}

DxgiAdapters::DxgiAdapters (ComSmartPtr<ID2D1Factory2> d2dFactoryIn)
    : d2dFactory (d2dFactoryIn)
{
    updateAdapters();
}

DxgiAdapters::~DxgiAdapters()
{
    releaseAdapters();
}

void DxgiAdapters::updateAdapters()
{
    if (factory != nullptr && factory->IsCurrent() && ! adapterArray.isEmpty())
        return;

    releaseAdapters();

    if (factory == nullptr || ! factory->IsCurrent())
        factory = makeDxgiFactory();

    if (factory == nullptr)
    {
        // If you hit this, we were unable to create a DXGI Factory, so we won't be able to
        // render anything using Direct2D.
        // Maybe this version of Windows doesn't have Direct2D support.
        jassertfalse;
        return;
    }

    for (UINT i = 0;; ++i)
    {
        ComSmartPtr<IDXGIAdapter1> dxgiAdapter;

        if (factory->EnumAdapters1 (i, dxgiAdapter.resetAndGetPointerAddress()) == DXGI_ERROR_NOT_FOUND)
            break;

        if (const auto adapter = DxgiAdapter::create (d2dFactory, dxgiAdapter))
        {
            adapterArray.add (adapter);
            listeners.call ([adapter] (DxgiAdapterListener& l) { l.adapterCreated (adapter); });
        }
    }
}

void DxgiAdapters::releaseAdapters()
{
    for (const auto& adapter : adapterArray)
        listeners.call ([adapter] (DxgiAdapterListener& l) { l.adapterRemoved (adapter); });

    adapterArray.clear();
}

DxgiAdapter::Ptr DxgiAdapters::getAdapterForHwnd (HWND hwnd) const
{
    const auto monitor = MonitorFromWindow (hwnd, MONITOR_DEFAULTTONULL);

    if (monitor == nullptr)
        return getDefaultAdapter();

    for (auto& adapter : adapterArray)
    {
        for (const auto& dxgiOutput : adapter->dxgiOutputs)
        {
            DXGI_OUTPUT_DESC desc{};

            if (FAILED (dxgiOutput->GetDesc (&desc)))
                continue;

            if (desc.Monitor == monitor)
                return adapter;
        }
    }

    return getDefaultAdapter();
}

DxgiAdapter::Ptr DxgiAdapters::getDefaultAdapter() const
{
    return adapterArray.getFirst();
}

void DxgiAdapters::addListener (DxgiAdapterListener& l)
{
    listeners.add (&l);
}

void DxgiAdapters::removeListener (DxgiAdapterListener& l)
{
    listeners.remove (&l);
}

ComSmartPtr<IDXGIFactory2> DxgiAdapters::makeDxgiFactory()
{
    JUCE_BEGIN_IGNORE_WARNINGS_GCC_LIKE ("-Wlanguage-extension-token")
    ComSmartPtr<IDXGIFactory2> result;
    if (const auto hr = CreateDXGIFactory2 (0, __uuidof (IDXGIFactory2), (void**) result.resetAndGetPointerAddress()); SUCCEEDED (hr))
        return result;
    JUCE_END_IGNORE_WARNINGS_GCC_LIKE

    // If CreateDXGIFactory fails, check to see if this is being called in the context of DllMain.
    // CreateDXGIFactory will always fail if called from the context of DllMain. In this case, the renderer
    // will create a software image instead as a fallback, but that won't perform as well.
    //
    // You may be creating an Image as a static object, which will likely be created in the context of DllMain.
    // Consider deferring your Image creation until later.
    jassertfalse;
    return {};
}

//==============================================================================
DirectX::DirectX()
    : d2dSharedFactory { std::invoke ([]
    {
        D2D1_FACTORY_OPTIONS options;
        options.debugLevel = enableDirectXDebugLayer ? D2D1_DEBUG_LEVEL_INFORMATION : D2D1_DEBUG_LEVEL_NONE;
        JUCE_BEGIN_IGNORE_WARNINGS_GCC_LIKE ("-Wlanguage-extension-token")
        ComSmartPtr<ID2D1Factory2> result;
        auto hr = D2D1CreateFactory (D2D1_FACTORY_TYPE_MULTI_THREADED,
                                     __uuidof (ID2D1Factory2),
                                     &options,
                                     (void**) result.resetAndGetPointerAddress());
        jassertquiet (SUCCEEDED (hr));
        JUCE_END_IGNORE_WARNINGS_GCC_LIKE

        return result;
    }) }
{}

//==============================================================================
ComSmartPtr<ID2D1Device1> D2DUtilities::getDeviceForContext (ComSmartPtr<ID2D1DeviceContext1> context)
{
    if (context == nullptr)
        return {};

    ComSmartPtr<ID2D1Device> device;
    context->GetDevice (device.resetAndGetPointerAddress());
    return device.getInterface<ID2D1Device1>();
}

//==============================================================================
ComSmartPtr<ID2D1DeviceContext1> Direct2DDeviceContext::create (ComSmartPtr<ID2D1Device1> device)
{
    ComSmartPtr<ID2D1DeviceContext1> result;

    if (const auto hr = device->CreateDeviceContext (D2D1_DEVICE_CONTEXT_OPTIONS_ENABLE_MULTITHREADED_OPTIMIZATIONS,
                                                     result.resetAndGetPointerAddress());
        FAILED (hr))
    {
        jassertfalse;
        return {};
    }

    result->SetTextAntialiasMode (D2D1_TEXT_ANTIALIAS_MODE_GRAYSCALE);
    result->SetAntialiasMode (D2D1_ANTIALIAS_MODE_PER_PRIMITIVE);
    result->SetUnitMode (D2D1_UNIT_MODE_PIXELS);

    return result;
}

//==============================================================================
ComSmartPtr<ID2D1Bitmap1> Direct2DBitmap::toBitmap (const Image& image,
                                                    ComSmartPtr<ID2D1DeviceContext1> deviceContext,
                                                    Image::PixelFormat outputFormat)
{
    JUCE_D2DMETRICS_SCOPED_ELAPSED_TIME (Direct2DMetricsHub::getInstance()->imageContextMetrics, createBitmapTime);

    jassert (outputFormat == Image::ARGB || outputFormat == Image::SingleChannel);

    JUCE_TRACE_LOG_D2D_PAINT_CALL (etw::createDirect2DBitmapFromImage, etw::graphicsKeyword);

    // Calling Image::convertedToFormat could cause unchecked recursion since convertedToFormat
    // calls Graphics::drawImageAt which calls Direct2DGraphicsContext::drawImage which calls this function...
    //
    // Use a software image for the conversion instead so the Graphics::drawImageAt call doesn't go
    // through the Direct2D renderer
    //
    // Be sure to explicitly set the DPI to 96.0 for the image; otherwise it will default to the screen DPI
    // and may be scaled incorrectly
    const auto convertedImage = SoftwareImageType{}.convert (image).convertedToFormat (outputFormat);

    if (! convertedImage.isValid())
        return {};

    Image::BitmapData bitmapData { convertedImage, Image::BitmapData::readWrite };

    D2D1_BITMAP_PROPERTIES1 bitmapProperties{};
    bitmapProperties.pixelFormat.format = outputFormat == Image::SingleChannel
                                          ? DXGI_FORMAT_A8_UNORM
                                          : DXGI_FORMAT_B8G8R8A8_UNORM;
    bitmapProperties.pixelFormat.alphaMode = outputFormat == Image::RGB
                                             ? D2D1_ALPHA_MODE_IGNORE
                                             : D2D1_ALPHA_MODE_PREMULTIPLIED;
    bitmapProperties.dpiX = USER_DEFAULT_SCREEN_DPI;
    bitmapProperties.dpiY = USER_DEFAULT_SCREEN_DPI;

    const D2D1_SIZE_U size { (UINT32) image.getWidth(), (UINT32) image.getHeight() };

    ComSmartPtr<ID2D1Bitmap1> bitmap;
    deviceContext->CreateBitmap (size,
                                 bitmapData.data,
                                 (UINT32) bitmapData.lineStride,
                                 bitmapProperties,
                                 bitmap.resetAndGetPointerAddress());
    return bitmap;
}

ComSmartPtr<ID2D1Bitmap1> Direct2DBitmap::createBitmap (ComSmartPtr<ID2D1DeviceContext1> deviceContext,
                                                        Image::PixelFormat format,
                                                        D2D_SIZE_U size,
                                                        D2D1_BITMAP_OPTIONS options)
{
    JUCE_TRACE_LOG_D2D_PAINT_CALL (etw::createDirect2DBitmap, etw::graphicsKeyword);

    JUCE_D2DMETRICS_SCOPED_ELAPSED_TIME (Direct2DMetricsHub::getInstance()->imageContextMetrics, createBitmapTime);

    // Verify that the GPU can handle a bitmap of this size
    //
    // If you need a bitmap larger than this, you'll need to either split it up into multiple bitmaps
    // or use a software image (see SoftwareImageType).
    const auto maxBitmapSize = deviceContext->GetMaximumBitmapSize();
    jassertquiet (size.width <= maxBitmapSize && size.height <= maxBitmapSize);

    const auto pixelFormat = D2D1::PixelFormat (format == Image::SingleChannel
                                                    ? DXGI_FORMAT_A8_UNORM
                                                    : DXGI_FORMAT_B8G8R8A8_UNORM,
                                                format == Image::RGB
                                                    ? D2D1_ALPHA_MODE_IGNORE
                                                    : D2D1_ALPHA_MODE_PREMULTIPLIED);
    const auto bitmapProperties = D2D1::BitmapProperties1 (options, pixelFormat);

    ComSmartPtr<ID2D1Bitmap1> bitmap;
    const auto hr = deviceContext->CreateBitmap (size,
                                                 {},
                                                 {},
                                                 bitmapProperties,
                                                 bitmap.resetAndGetPointerAddress());

    jassertquiet (SUCCEEDED (hr) && bitmap != nullptr);
    return bitmap;
}

//==============================================================================
void UpdateRegion::findRECTAndValidate (HWND windowHandle)
{
    numRect = 0;

    auto* regionHandle = CreateRectRgn (0, 0, 0, 0);

    if (regionHandle == nullptr)
    {
        ValidateRect (windowHandle, nullptr);
        return;
    }

    const ScopeGuard regionDeleter { [&] { DeleteObject (regionHandle); } };
    const auto regionType = GetUpdateRgn (windowHandle, regionHandle, false);

    if (regionType == ERROR)
        return;

    if (regionType == NULLREGION)
    {
        ValidateRect (windowHandle, nullptr);
        return;
    }

    const auto requiredBytes = GetRegionData (regionHandle, 0, nullptr);
    block.ensureSize (requiredBytes);

    const auto regionDataResult = GetRegionData (regionHandle,
                                                 (DWORD) block.getSize(),
                                                 static_cast<RGNDATA*> (block.getData()));

    numRect = std::invoke ([&]() -> decltype (numRect)
    {
        if (regionDataResult < sizeof (RGNDATAHEADER))
            return 0;

        const auto* header = static_cast<RGNDATAHEADER const* const> (block.getData());

        if (header->iType != RDH_RECTANGLES)
            return 0;

        return header->nCount;
    });

    if (numRect > 0)
        ValidateRgn (windowHandle, regionHandle);
    else
        ValidateRect (windowHandle, nullptr);
}

//==============================================================================
static ComSmartPtr<ID2D1GradientStopCollection> makeGradientStopCollection (const ColourGradient& gradient,
                                                                            ComSmartPtr<ID2D1DeviceContext1> deviceContext,
                                                                            [[maybe_unused]] Direct2DMetrics* metrics) noexcept
{
    JUCE_D2DMETRICS_SCOPED_ELAPSED_TIME (metrics, createGradientTime);

    const int numColors = gradient.getNumColours();

    std::vector<D2D1_GRADIENT_STOP> stops ((size_t) numColors);

    for (auto [index, stop] : enumerate (stops, int{}))
    {
        stop.color = D2DUtilities::toCOLOR_F (gradient.getColour (index));
        stop.position = (FLOAT) gradient.getColourPosition (index);
    }

    ComSmartPtr<ID2D1GradientStopCollection> result;
    deviceContext->CreateGradientStopCollection (stops.data(), (UINT32) stops.size(), result.resetAndGetPointerAddress());
    return result;
}

ComSmartPtr<ID2D1LinearGradientBrush> LinearGradientCache::get (const ColourGradient& gradient,
                                                                ComSmartPtr<ID2D1DeviceContext1> deviceContext,
                                                                Direct2DMetrics* metrics)
{
    jassert (! gradient.isRadial);

    return cache.get (gradient, [&deviceContext, &metrics] (const auto& key)
    {
        const auto gradientStops = makeGradientStopCollection (key, deviceContext, metrics);
        const auto p1 = key.point1;
        const auto p2 = key.point2;
        const auto linearGradientBrushProperties = D2D1::LinearGradientBrushProperties ({ p1.x, p1.y }, { p2.x, p2.y });
        const D2D1_BRUSH_PROPERTIES brushProps { 1.0f, D2D1::IdentityMatrix() };

        ComSmartPtr<ID2D1LinearGradientBrush> result;
        deviceContext->CreateLinearGradientBrush (linearGradientBrushProperties,
                                                  brushProps,
                                                  gradientStops,
                                                  result.resetAndGetPointerAddress());
        return result;
    });
}

ComSmartPtr<ID2D1RadialGradientBrush> RadialGradientCache::get (const ColourGradient& gradient,
                                                                ComSmartPtr<ID2D1DeviceContext1> deviceContext,
                                                                Direct2DMetrics* metrics)
{
    jassert (gradient.isRadial);

    return cache.get (gradient, [&deviceContext, &metrics] (const auto& key)
    {
        const auto gradientStops = makeGradientStopCollection (key, deviceContext, metrics);

        const auto p1 = key.point1;
        const auto p2 = key.point2;
        const auto r = p1.getDistanceFrom (p2);
        const auto radialGradientBrushProperties = D2D1::RadialGradientBrushProperties ({ p1.x, p1.y }, {}, r, r);
        const D2D1_BRUSH_PROPERTIES brushProps { 1.0F, D2D1::IdentityMatrix() };

        ComSmartPtr<ID2D1RadialGradientBrush> result;
        deviceContext->CreateRadialGradientBrush (radialGradientBrushProperties,
                                                  brushProps,
                                                  gradientStops,
                                                  result.resetAndGetPointerAddress());
        return result;
    });
}

RectangleListSpriteBatch::~RectangleListSpriteBatch()
{
    release();
}

void RectangleListSpriteBatch::release()
{
    whiteRectangle = nullptr;
    spriteBatches.clear();
    destinations.free();
    destinationsCapacity = 0;
}

ComSmartPtr<ID2D1SpriteBatch> RectangleListSpriteBatch::getSpriteBatch (ID2D1DeviceContext3& dc, uint32 key)
{
    return spriteBatches.get ((uint32) key, [&dc] (auto) -> ComSmartPtr<ID2D1SpriteBatch>
    {
        ComSmartPtr<ID2D1SpriteBatch> result;
        if (const auto hr = dc.CreateSpriteBatch (result.resetAndGetPointerAddress()); SUCCEEDED (hr))
            return result;

        return nullptr;
    });
}

bool RectangleListSpriteBatch::fillRectanglesImpl (ComSmartPtr<ID2D1DeviceContext1> deviceContext,
                                                   const RectangleList<float>& rectangles,
                                                   const Colour colour,
                                                   const TransformCallback& transformRectangle,
                                                   [[maybe_unused]] Direct2DMetrics* metrics)
{
    if (rectangles.isEmpty())
        return true;

    JUCE_D2DMETRICS_SCOPED_ELAPSED_TIME (metrics, spriteBatchTime)

    auto numRectanglesPainted = 0;
    while (numRectanglesPainted < rectangles.getNumRectangles())
    {
        auto numRectanglesRemaining = rectangles.getNumRectangles() - numRectanglesPainted;
        auto spriteBatchSize = isPowerOfTwo (numRectanglesRemaining) ? numRectanglesRemaining : (nextPowerOfTwo (numRectanglesRemaining) >> 1);

        {
            JUCE_D2DMETRICS_SCOPED_ELAPSED_TIME (metrics, spriteBatchSetupTime);

            if (destinationsCapacity < (size_t) spriteBatchSize)
            {
                destinations.calloc (spriteBatchSize);
                destinationsCapacity = (size_t) spriteBatchSize;
            }

            auto destination = destinations.getData();

            for (int i = numRectanglesPainted; i < numRectanglesPainted + spriteBatchSize; ++i)
            {
                auto r = rectangles.getRectangle (i);
                r = transformRectangle.transform (r);

                if (r.getWidth() < 1.0f || r.getHeight() < 1.0f)
                    return false;

                *destination = D2DUtilities::toRECT_F (r);
                ++destination;
            }
        }

        if (! whiteRectangle)
        {
            JUCE_D2DMETRICS_SCOPED_ELAPSED_TIME (metrics, createSpriteSourceTime);

            auto hr = deviceContext->CreateCompatibleRenderTarget (D2D1_SIZE_F { (float) rectangleSize, (float) rectangleSize },
                                                                   D2D1_SIZE_U { rectangleSize, rectangleSize },
                                                                   D2D1_PIXEL_FORMAT { DXGI_FORMAT_B8G8R8A8_UNORM, D2D1_ALPHA_MODE_PREMULTIPLIED },
                                                                   whiteRectangle.resetAndGetPointerAddress());
            if (FAILED (hr))
                return false;

            whiteRectangle->BeginDraw();
            whiteRectangle->Clear (D2D1_COLOR_F { 1.0f, 1.0f, 1.0f, 1.0f });
            whiteRectangle->EndDraw();
        }

        ComSmartPtr<ID2D1Bitmap> bitmap;

        if (auto hr = whiteRectangle->GetBitmap (bitmap.resetAndGetPointerAddress()); SUCCEEDED (hr))
        {
            ComSmartPtr<ID2D1DeviceContext3> deviceContext3;

            if (hr = deviceContext->QueryInterface<ID2D1DeviceContext3> (deviceContext3.resetAndGetPointerAddress()); SUCCEEDED (hr))
            {
                auto d2dColour = D2DUtilities::toCOLOR_F (colour);
                auto spriteBatch = getSpriteBatch (*deviceContext3, (uint32) spriteBatchSize);

                if (spriteBatch == nullptr)
                    return false;

                auto setCount = jmin ((uint32) spriteBatchSize, spriteBatch->GetSpriteCount());
                auto addCount = (uint32) spriteBatchSize > setCount ? (uint32) spriteBatchSize - setCount : 0;

                if (setCount != 0)
                {
                    JUCE_D2DMETRICS_SCOPED_ELAPSED_TIME (metrics, setSpritesTime);

                    spriteBatch->SetSprites (0, setCount, destinations.getData(), nullptr, &d2dColour, nullptr, sizeof (D2D1_RECT_F), 0, 0, 0);
                }

                if (addCount != 0)
                {
                    JUCE_D2DMETRICS_SCOPED_ELAPSED_TIME (metrics, addSpritesTime);

                    spriteBatch->AddSprites (addCount, destinations.getData() + setCount, nullptr, &d2dColour, nullptr, sizeof (D2D1_RECT_F), 0, 0, 0);
                }

                JUCE_D2DMETRICS_SCOPED_ELAPSED_TIME (metrics, drawSpritesTime);

                deviceContext3->SetAntialiasMode (D2D1_ANTIALIAS_MODE_ALIASED);
                deviceContext3->DrawSpriteBatch (spriteBatch, bitmap);
                deviceContext3->SetAntialiasMode (D2D1_ANTIALIAS_MODE_PER_PRIMITIVE);
            }
        }

        numRectanglesPainted += spriteBatchSize;
    }

    return true;
}

//==============================================================================
DxgiAdapter::Ptr Direct2DDeviceResources::findAdapter (const DxgiAdapters& adapters,
                                                       ID2D1Bitmap1* bitmap)
{
    if (bitmap == nullptr)
        return {};

    ComSmartPtr<IDXGISurface> surface;
    bitmap->GetSurface (surface.resetAndGetPointerAddress());

    if (surface == nullptr)
        return {};

    ComSmartPtr<IDXGIDevice> device;
    JUCE_BEGIN_IGNORE_WARNINGS_GCC_LIKE ("-Wlanguage-extension-token")
    surface->GetDevice (__uuidof (device), (void**) device.resetAndGetPointerAddress());
    JUCE_END_IGNORE_WARNINGS_GCC_LIKE

    return findAdapter (adapters, device);
}

DxgiAdapter::Ptr Direct2DDeviceResources::findAdapter (const DxgiAdapters& dxgiAdapters,
                                                       IDXGIDevice* dxgiDevice)
{
    if (dxgiDevice == nullptr)
        return {};

    ComSmartPtr<IDXGIAdapter> adapter;
    dxgiDevice->GetAdapter (adapter.resetAndGetPointerAddress());

    if (adapter == nullptr)
        return {};

    ComSmartPtr<IDXGIAdapter1> adapter1;
    adapter.QueryInterface (adapter1);

    if (adapter1 == nullptr)
        return {};

    const auto adapterLuid = getLUID (adapter1);

    const auto& adapters = dxgiAdapters.getAdapterArray();

    const auto it = std::find_if (adapters.begin(), adapters.end(), [&] (DxgiAdapter::Ptr ptr)
    {
        const auto tie = [] (const LUID& x) { return std::tie (x.LowPart, x.HighPart); };

        const auto thisLuid = getLUID (ptr->dxgiAdapter);
        return tie (thisLuid) == tie (adapterLuid);
    });

    if (it == adapters.end())
        return {};

    return *it;
}

DxgiAdapter::Ptr Direct2DDeviceResources::findAdapter (const DxgiAdapters& dxgiAdapters,
                                                       ID2D1DeviceContext1* context)
{
    if (context == nullptr)
        return {};

    ComSmartPtr<ID2D1Device> device;
    context->GetDevice (device.resetAndGetPointerAddress());

    if (device == nullptr)
        return {};

    ComSmartPtr<IDXGIDevice> dxgiDevice;
    device.QueryInterface (dxgiDevice);

    return findAdapter (dxgiAdapters, dxgiDevice);
}

LUID Direct2DDeviceResources::getLUID (ComSmartPtr<IDXGIAdapter1> adapter)
{
    DXGI_ADAPTER_DESC1 desc{};
    adapter->GetDesc1 (&desc);
    return desc.AdapterLuid;
}

std::optional<Direct2DDeviceResources> Direct2DDeviceResources::create (ComSmartPtr<ID2D1DeviceContext1> context)
{
    if (context == nullptr)
        return {};

    Direct2DDeviceResources result;

    if (const auto hr = context->CreateSolidColorBrush (D2D1::ColorF (0.0f, 0.0f, 0.0f, 1.0f),
                                                        result.colourBrush.resetAndGetPointerAddress());
        FAILED (hr))
    {
        jassertfalse;
        return {};
    }

    result.rectangleListSpriteBatch = std::make_unique<RectangleListSpriteBatch>();

    return result;
}

//==============================================================================
static String getLocalisedName (IDWriteLocalizedStrings* names)
{
    jassert (names != nullptr);

    uint32 index = 0;
    BOOL exists = false;
    [[maybe_unused]] auto hr = names->FindLocaleName (L"en-us", &index, &exists);

    if (! exists)
        index = 0;

    uint32 length = 0;
    hr = names->GetStringLength (index, &length);

    HeapBlock<wchar_t> name (length + 1);
    hr = names->GetString (index, name, length + 1);

    return static_cast<const wchar_t*> (name);
}

static String getFontFamilyName (IDWriteFontFamily* family)
{
    jassert (family != nullptr);
    ComSmartPtr<IDWriteLocalizedStrings> familyNames;
    auto hr = family->GetFamilyNames (familyNames.resetAndGetPointerAddress());
    jassertquiet (SUCCEEDED (hr));
    return getLocalisedName (familyNames);
}

static String getFontFaceName (IDWriteFont* font)
{
    jassert (font != nullptr);
    ComSmartPtr<IDWriteLocalizedStrings> faceNames;
    auto hr = font->GetFaceNames (faceNames.resetAndGetPointerAddress());
    jassertquiet (SUCCEEDED (hr));
    return getLocalisedName (faceNames);
}

template <typename Range>
static StringArray stringArrayFromRange (Range&& range)
{
    StringArray result;

    for (const auto& item : range)
        result.add (item);

    return result;
}

//==============================================================================
AggregateFontCollection::AggregateFontCollection (ComSmartPtr<IDWriteFontCollection> baseCollection)
    : collections { std::move (baseCollection) } {}

StringArray AggregateFontCollection::findAllTypefaceNames()
{
    const std::scoped_lock lock { mutex };

    std::set<String> strings;

    for (const auto& collection : collections)
    {
        const auto count = collection->GetFontFamilyCount();

        for (auto i = decltype (count){}; i < count; ++i)
        {
            ComSmartPtr<IDWriteFontFamily> family;

            if (FAILED (collection->GetFontFamily (i, family.resetAndGetPointerAddress())) || family == nullptr)
                continue;

            strings.insert (getFontFamilyName (family));
        }
    }

    return stringArrayFromRange (strings);
}

std::vector<ComSmartPtr<IDWriteFont>> AggregateFontCollection::getAllFontsInFamily (IDWriteFontFamily* fontFamily)
{
    const auto fontFacesCount = fontFamily->GetFontCount();
    std::vector<ComSmartPtr<IDWriteFont>> result;
    result.reserve (fontFacesCount);

    for (UINT32 i = 0; i < fontFacesCount; ++i)
    {
        ComSmartPtr<IDWriteFont> dwFont;

        if (FAILED (fontFamily->GetFont (i, dwFont.resetAndGetPointerAddress())))
            continue;

        if (dwFont->GetSimulations() != DWRITE_FONT_SIMULATIONS_NONE)
            continue;

        result.push_back (dwFont);
    }

    return result;
}

StringArray AggregateFontCollection::findAllTypefaceStyles (const String& family)
{
    const std::scoped_lock lock { mutex };

    for (const auto& collection : collections)
    {
        BOOL fontFound = false;
        uint32 fontIndex = 0;

        if (FAILED (collection->FindFamilyName (family.toWideCharPointer(), &fontIndex, &fontFound)) || ! fontFound)
            continue;

        ComSmartPtr<IDWriteFontFamily> fontFamily;

        if (FAILED (collection->GetFontFamily (fontIndex, fontFamily.resetAndGetPointerAddress())) || fontFamily == nullptr)
            continue;

        std::set<String> uniqueResults;
        StringArray orderedResults;

        for (const auto& font : getAllFontsInFamily (fontFamily))
        {
            const auto name = getFontFaceName (font);

            if (uniqueResults.insert (name).second)
                orderedResults.add (name);
        }

        return orderedResults;
    }

    return {};
}

ComSmartPtr<IDWriteFontFamily> AggregateFontCollection::getFamilyByName (const wchar_t* name)
{
    const std::scoped_lock lock { mutex };

    for (const auto& collection : collections)
    {
        const auto fontIndex = [&]
        {
            BOOL   found = false;
            UINT32 index = 0;

            return (SUCCEEDED (collection->FindFamilyName (name, &index, &found)) && found)
                   ? index
                   : (UINT32) -1;
        }();

        if (fontIndex == (UINT32) -1)
            continue;

        ComSmartPtr<IDWriteFontFamily> family;

        if (FAILED (collection->GetFontFamily (fontIndex, family.resetAndGetPointerAddress())) || family == nullptr)
            continue;

        return family;
    }

    return {};
}

ComSmartPtr<IDWriteFont> AggregateFontCollection::findFontForFace (IDWriteFontFace* face)
{
    for (const auto& collection : collections)
    {
        ComSmartPtr<IDWriteFont> result;

        if (SUCCEEDED (collection->GetFontFromFontFace (face, result.resetAndGetPointerAddress())))
            return result;
    }

    return {};
}

void AggregateFontCollection::addCollection (ComSmartPtr<IDWriteFontCollection> collection)
{
    const std::scoped_lock lock { mutex };
    collections.push_back (std::move (collection));
}

void AggregateFontCollection::removeCollection (ComSmartPtr<IDWriteFontCollection> collection)
{
    const std::scoped_lock lock { mutex };
    const auto iter = std::find (collections.begin(), collections.end(), collection);

    if (iter != collections.end())
        collections.erase (iter);
}

auto AggregateFontCollection::mapCharacters (IDWriteFontFallback* fallback,
                                             IDWriteTextAnalysisSource* analysisSource,
                                             UINT32 textPosition,
                                             UINT32 textLength,
                                             wchar_t const* baseFamilyName,
                                             DWRITE_FONT_WEIGHT baseWeight,
                                             DWRITE_FONT_STYLE baseStyle,
                                             DWRITE_FONT_STRETCH baseStretch) noexcept -> MapResult
{
    const std::scoped_lock lock { mutex };

    // For reasons I don't understand, the system may pick better substitutions when passing
    // nullptr, instead of the system collection, as the "default collection to use".
    auto collectionsToCheck = collections;
    collectionsToCheck.insert (collectionsToCheck.begin(), nullptr);

    MapResult bestMatch;
    for (const auto& collection : collectionsToCheck)
    {
        MapResult result;
        const auto status = fallback->MapCharacters (analysisSource,
                                                     textPosition,
                                                     textLength,
                                                     collection,
                                                     baseFamilyName,
                                                     baseWeight,
                                                     baseStyle,
                                                     baseStretch,
                                                     &result.length,
                                                     result.font.resetAndGetPointerAddress(),
                                                     &result.scale);

        if (FAILED (status) || result.font == nullptr)
            continue;

        if (result.length == textLength)
            return result;

        if (result.length >= bestMatch.length)
            bestMatch = result;
    }

    return bestMatch;
}

//==============================================================================
class MemoryFontFileStream final : public ComBaseClassHelper<IDWriteFontFileStream>
{
public:
    explicit MemoryFontFileStream (std::shared_ptr<const MemoryBlock> blockIn)
        : block (std::move (blockIn))
    {
    }

    JUCE_COMRESULT GetFileSize (UINT64* fileSize) noexcept override
    {
        *fileSize = block->getSize();
        return S_OK;
    }

    JUCE_COMRESULT GetLastWriteTime (UINT64* lastWriteTime) noexcept override
    {
        *lastWriteTime = 0;
        return S_OK;
    }

    JUCE_COMRESULT ReadFileFragment (const void** fragmentStart,
                                     UINT64 fileOffset,
                                     UINT64 fragmentSize,
                                     void** fragmentContext) noexcept override
    {
        if (fileOffset + fragmentSize > block->getSize())
        {
            *fragmentStart   = nullptr;
            *fragmentContext = nullptr;
            return E_INVALIDARG;
        }

        *fragmentStart   = addBytesToPointer (block->getData(), fileOffset);
        *fragmentContext = this;
        return S_OK;
    }

    void WINAPI ReleaseFileFragment (void*) noexcept override {}

private:
    std::shared_ptr<const MemoryBlock> block;
};

//==============================================================================
class DirectWriteCustomFontCollectionLoader::MemoryFontFileLoader final : public ComBaseClassHelper<IDWriteFontFileLoader>
{
public:
    explicit MemoryFontFileLoader (MemoryBlock blob)
        : block (std::make_shared<const MemoryBlock> (std::move (blob)))
    {
    }

    HRESULT WINAPI CreateStreamFromKey (const void* fontFileReferenceKey,
                                        UINT32 keySize,
                                        IDWriteFontFileStream** fontFileStream) noexcept override
    {
        if (keySize != Uuid::size())
            return E_INVALIDARG;

        Uuid requestedKey { static_cast<const uint8*> (fontFileReferenceKey) };

        if (requestedKey == uuid)
        {
            *fontFileStream = new MemoryFontFileStream { block };
            return S_OK;
        }

        return E_INVALIDARG;
    }

    Uuid getUuid() const { return uuid; }

    Span<const std::byte> asSpan() const&& = delete;
    Span<const std::byte> asSpan() const&
    {
        return { static_cast<const std::byte*> (block->getData()), block->getSize() };
    }

private:
    std::shared_ptr<const MemoryBlock> block;
    Uuid uuid;

    JUCE_DECLARE_NON_COPYABLE_WITH_LEAK_DETECTOR (MemoryFontFileLoader)
};

//==============================================================================
class DirectWriteCustomFontCollectionLoader::FontFileEnumerator final : public ComBaseClassHelper<IDWriteFontFileEnumerator>
{
public:
    FontFileEnumerator (IDWriteFactory& factoryIn, ComSmartPtr<MemoryFontFileLoader> loaderIn)
        : factory (factoryIn), loader (loaderIn) {}

    HRESULT WINAPI GetCurrentFontFile (IDWriteFontFile** fontFile) noexcept override
    {
        *fontFile = nullptr;

        if (! isPositiveAndBelow (rawDataIndex, 1))
            return E_FAIL;

        const auto uuid = loader->getUuid();
        return factory.CreateCustomFontFileReference (uuid.getRawData(),
                                                      (UINT32) uuid.size(),
                                                      loader,
                                                      fontFile);
    }

    HRESULT WINAPI MoveNext (BOOL* hasCurrentFile) noexcept override
    {
        ++rawDataIndex;
        *hasCurrentFile = rawDataIndex < 1 ? TRUE : FALSE;
        return S_OK;
    }

private:
    IDWriteFactory& factory;
    ComSmartPtr<MemoryFontFileLoader> loader;
    size_t rawDataIndex = std::numeric_limits<size_t>::max();

    JUCE_DECLARE_NON_COPYABLE_WITH_LEAK_DETECTOR (FontFileEnumerator)
};

//==============================================================================
DirectWriteCustomFontCollectionLoader::DirectWriteCustomFontCollectionLoader (IDWriteFactory& factoryIn)
    : factory (factoryIn)
{
}

DirectWriteCustomFontCollectionLoader::~DirectWriteCustomFontCollectionLoader()
{
    for (auto& loader : loaders)
        factory.UnregisterFontFileLoader (loader);
}

auto DirectWriteCustomFontCollectionLoader::findLoaderForUuid (const Uuid& uuid) const -> ComSmartPtr<MemoryFontFileLoader>
{
    const auto compareUuidAndLoader = [] (ComSmartPtr<MemoryFontFileLoader> loader, const Uuid& comparisonUuid)
    {
        return loader->getUuid() < comparisonUuid;
    };

    const auto iter = std::lower_bound (loaders.begin(), loaders.end(), uuid, compareUuidAndLoader);

    if (iter == loaders.end() || iter->get()->getUuid() != uuid)
        return {};

    return *iter;
}

Uuid DirectWriteCustomFontCollectionLoader::addRawFontData (Span<const std::byte> blob)
{
    const std::string_view blobAsString (reinterpret_cast<const char*> (blob.data()), blob.size());
    const auto hashValue = std::hash<std::string_view>{} (blobAsString);
    auto& uuids = uuidsForHash[hashValue];

    for (const auto& uuid : uuids)
    {
        const auto matchingLoader = findLoaderForUuid (uuid);

        if (matchingLoader == nullptr)
            continue;

        const auto loaderData = matchingLoader->asSpan();

        if (! std::equal (blob.begin(), blob.end(), loaderData.begin(), loaderData.end()))
            continue;

        return matchingLoader->getUuid();
    }

    ComSmartPtr loader { new MemoryFontFileLoader { { blob.data(), blob.size() } },
                         IncrementRef::no };
    factory.RegisterFontFileLoader (loader);

    const auto compareUuids = [] (const auto& a, const auto& b) { return a->getUuid() < b->getUuid(); };
    OrderedContainerHelpers::insertOrAssign (loaders, loader, compareUuids);
    uuids.push_back (loader->getUuid());
    return loader->getUuid();
}

HRESULT WINAPI DirectWriteCustomFontCollectionLoader::CreateEnumeratorFromKey (IDWriteFactory* factoryIn,
                                                                               const void* collectionKey,
                                                                               UINT32 collectionKeySize,
                                                                               IDWriteFontFileEnumerator** fontFileEnumerator) noexcept
{
    if (collectionKeySize != Uuid::size())
        return E_INVALIDARG;

    const Uuid requestedCollectionKey { static_cast<const uint8*> (collectionKey) };

    const auto matchingLoader = findLoaderForUuid (requestedCollectionKey);

    if (matchingLoader == nullptr)
        return E_INVALIDARG;

    *fontFileEnumerator = new FontFileEnumerator { *factoryIn, matchingLoader };
    return S_OK;
}

//==============================================================================
Direct2DFactories::Direct2DFactories()
    : d2dFactory { std::invoke ([&]() -> ComSmartPtr<ID2D1Factory>
      {
          JUCE_LOAD_WINAPI_FUNCTION (direct2dDll,
                                     D2D1CreateFactory,
                                     d2d1CreateFactory,
                                     HRESULT,
                                     (D2D1_FACTORY_TYPE, REFIID, D2D1_FACTORY_OPTIONS*, void**))

          if (d2d1CreateFactory == nullptr)
              return {};

          D2D1_FACTORY_OPTIONS options;
          options.debugLevel = D2D1_DEBUG_LEVEL_NONE;

          ComSmartPtr<ID2D1Factory> result;

          JUCE_BEGIN_IGNORE_WARNINGS_GCC_LIKE ("-Wlanguage-extension-token")
          d2d1CreateFactory (D2D1_FACTORY_TYPE_SINGLE_THREADED,
                             __uuidof (ID2D1Factory),
                             &options,
                             (void**) result.resetAndGetPointerAddress());
          JUCE_END_IGNORE_WARNINGS_GCC_LIKE

          return result;
      }) },
      directWriteFactory { std::invoke ([&]() -> ComSmartPtr<IDWriteFactory>
      {
          JUCE_LOAD_WINAPI_FUNCTION (directWriteDll,
                                     DWriteCreateFactory,
                                     dWriteCreateFactory,
                                     HRESULT,
                                     (DWRITE_FACTORY_TYPE, REFIID, IUnknown**))

          if (dWriteCreateFactory == nullptr)
              return {};

          ComSmartPtr<IDWriteFactory> result;

          JUCE_BEGIN_IGNORE_WARNINGS_GCC_LIKE ("-Wlanguage-extension-token")
          dWriteCreateFactory (DWRITE_FACTORY_TYPE_SHARED,
                               __uuidof (IDWriteFactory),
                               (IUnknown**) result.resetAndGetPointerAddress());
          JUCE_END_IGNORE_WARNINGS_GCC_LIKE

          return result;
      }) },
      collectionLoader { std::invoke ([&]() -> ComSmartPtr<DirectWriteCustomFontCollectionLoader>
      {
          ComSmartPtr result { new DirectWriteCustomFontCollectionLoader { *directWriteFactory },
                               IncrementRef::no };

          directWriteFactory->RegisterFontCollectionLoader (result);

          return result;
      }) }
{
    ComSmartPtr<IDWriteFontCollection> collection;

    if (SUCCEEDED (directWriteFactory->GetSystemFontCollection (collection.resetAndGetPointerAddress(), FALSE)) && collection != nullptr)
        fonts.emplace (collection);
    else
        jassertfalse;
}

Direct2DFactories::~Direct2DFactories()
{
    if (directWriteFactory == nullptr)
        return;

    directWriteFactory->UnregisterFontCollectionLoader (collectionLoader);
}

} // namespace juce
