// Copyright (c) 2016-2019 Kiwano - Nomango
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

#include <kiwano/core/Logger.h>
#include <kiwano/render/DirectX/RenderContextImpl.h>

namespace kiwano
{

RenderContextImpl::RenderContextImpl()
{}

RenderContextImpl::~RenderContextImpl()
{
    DiscardDeviceResources();
}

HRESULT RenderContextImpl::CreateDeviceResources(ComPtr<ID2D1Factory> factory, ComPtr<ID2D1RenderTarget> ctx)
{
    if (!factory || !ctx)
        return E_INVALIDARG;

    render_target_ = ctx;
    text_renderer_.reset();
    current_brush_.reset();

    HRESULT hr = ITextRenderer::Create(&text_renderer_, render_target_.get());

    if (SUCCEEDED(hr))
    {
        SetAntialiasMode(antialias_);
        SetTextAntialiasMode(text_antialias_);

        Resize(reinterpret_cast<const Size&>(ctx->GetSize()));
    }

    // DrawingStateBlock
    if (SUCCEEDED(hr))
    {
        hr = factory->CreateDrawingStateBlock(&drawing_state_);
    }

    return hr;
}

void RenderContextImpl::DiscardDeviceResources()
{
    text_renderer_.reset();
    render_target_.reset();
    current_brush_.reset();
}

bool RenderContextImpl::IsValid() const
{
    return render_target_ != nullptr;
}

void RenderContextImpl::BeginDraw()
{
    SaveDrawingState();

    RenderContext::BeginDraw();

    if (render_target_)
    {
        render_target_->BeginDraw();
    }
}

void RenderContextImpl::EndDraw()
{
    win32::ThrowIfFailed(render_target_->EndDraw());

    RenderContext::EndDraw();

    RestoreDrawingState();
}

void RenderContextImpl::DrawTexture(Texture const& texture, const Rect* src_rect, const Rect* dest_rect)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");

    if (texture.IsValid())
    {
        auto mode = (texture.GetBitmapInterpolationMode() == InterpolationMode::Linear)
                        ? D2D1_BITMAP_INTERPOLATION_MODE_LINEAR
                        : D2D1_BITMAP_INTERPOLATION_MODE_NEAREST_NEIGHBOR;

        render_target_->DrawBitmap(texture.GetBitmap().get(), dest_rect ? &DX::ConvertToRectF(*dest_rect) : nullptr,
                                   brush_opacity_, mode, src_rect ? &DX::ConvertToRectF(*src_rect) : nullptr);

        IncreasePrimitivesCount();
    }
}

void RenderContextImpl::DrawTextLayout(TextLayout const& layout, Point const& offset)
{
    KGE_ASSERT(text_renderer_ && "Text renderer has not been initialized!");

    if (layout.IsValid())
    {
        ComPtr<ID2D1Brush> fill_brush;
        ComPtr<ID2D1Brush> outline_brush;
        const TextStyle&   style = layout.GetStyle();

        if (style.fill_brush)
        {
            fill_brush = style.fill_brush->GetBrush();
            fill_brush->SetOpacity(brush_opacity_);
        }

        if (style.outline_brush)
        {
            outline_brush = style.outline_brush->GetBrush();
            outline_brush->SetOpacity(brush_opacity_);
        }

        HRESULT hr = S_OK;

        if (style.outline_stroke)
        {
            hr = text_renderer_->DrawTextLayout(layout.GetTextLayout().get(), offset.x, offset.y, fill_brush.get(),
                                                outline_brush.get(), style.outline_width,
                                                style.outline_stroke->GetStrokeStyle().get());
        }
        else
        {
            hr = text_renderer_->DrawTextLayout(layout.GetTextLayout().get(), offset.x, offset.y, fill_brush.get(),
                                                outline_brush.get(), style.outline_width, nullptr);
        }

        if (SUCCEEDED(hr))
        {
            IncreasePrimitivesCount(text_renderer_->GetLastPrimitivesCount());
        }
        else
        {
            KGE_ERROR(L"Failed to draw text layout with HRESULT of %08X", hr);
        }
    }
}

void RenderContextImpl::DrawShape(Shape const& shape, StrokeStylePtr stroke, float stroke_width)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");
    KGE_ASSERT(current_brush_ && "The brush used for rendering has not been set!");

    if (shape.IsValid())
    {
        if (stroke)
        {
            render_target_->DrawGeometry(shape.GetGeometry().get(), current_brush_->GetBrush().get(), stroke_width,
                                         stroke->GetStrokeStyle().get());
        }
        else
        {
            render_target_->DrawGeometry(shape.GetGeometry().get(), current_brush_->GetBrush().get(), stroke_width,
                                         nullptr);
        }

        IncreasePrimitivesCount();
    }
}

void RenderContextImpl::DrawLine(Point const& point1, Point const& point2, StrokeStylePtr stroke, float stroke_width)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");
    KGE_ASSERT(current_brush_ && "The brush used for rendering has not been set!");

    if (stroke)
    {
        render_target_->DrawLine(DX::ConvertToPoint2F(point1), DX::ConvertToPoint2F(point2),
                                 current_brush_->GetBrush().get(), stroke_width, stroke->GetStrokeStyle().get());
    }
    else
    {
        render_target_->DrawLine(DX::ConvertToPoint2F(point1), DX::ConvertToPoint2F(point2),
                                 current_brush_->GetBrush().get(), stroke_width, nullptr);
    }

    IncreasePrimitivesCount();
}

void RenderContextImpl::DrawRectangle(Rect const& rect, StrokeStylePtr stroke, float stroke_width)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");
    KGE_ASSERT(current_brush_ && "The brush used for rendering has not been set!");

    if (stroke)
    {
        render_target_->DrawRectangle(DX::ConvertToRectF(rect), current_brush_->GetBrush().get(), stroke_width,
                                      stroke->GetStrokeStyle().get());
    }
    else
    {
        render_target_->DrawRectangle(DX::ConvertToRectF(rect), current_brush_->GetBrush().get(), stroke_width,
                                      nullptr);
    }

    IncreasePrimitivesCount();
}

void RenderContextImpl::DrawRoundedRectangle(Rect const& rect, Vec2 const& radius, StrokeStylePtr stroke,
                                         float stroke_width)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");
    KGE_ASSERT(current_brush_ && "The brush used for rendering has not been set!");

    if (stroke)
    {
        render_target_->DrawRoundedRectangle(D2D1::RoundedRect(DX::ConvertToRectF(rect), radius.x, radius.y),
                                             current_brush_->GetBrush().get(), stroke_width,
                                             stroke->GetStrokeStyle().get());
    }
    else
    {
        render_target_->DrawRoundedRectangle(D2D1::RoundedRect(DX::ConvertToRectF(rect), radius.x, radius.y),
                                             current_brush_->GetBrush().get(), stroke_width, nullptr);
    }
    IncreasePrimitivesCount();
}

void RenderContextImpl::DrawEllipse(Point const& center, Vec2 const& radius, StrokeStylePtr stroke, float stroke_width)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");
    KGE_ASSERT(current_brush_ && "The brush used for rendering has not been set!");

    if (stroke)
    {
        render_target_->DrawEllipse(D2D1::Ellipse(DX::ConvertToPoint2F(center), radius.x, radius.y),
                                    current_brush_->GetBrush().get(), stroke_width, stroke->GetStrokeStyle().get());
    }
    else
    {
        render_target_->DrawEllipse(D2D1::Ellipse(DX::ConvertToPoint2F(center), radius.x, radius.y),
                                    current_brush_->GetBrush().get(), stroke_width, nullptr);
    }

    IncreasePrimitivesCount();
}

void RenderContextImpl::FillShape(Shape const& shape)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");
    KGE_ASSERT(current_brush_ && "The brush used for rendering has not been set!");

    if (shape.IsValid())
    {
        render_target_->FillGeometry(shape.GetGeometry().get(), current_brush_->GetBrush().get());

        IncreasePrimitivesCount();
    }
}

void RenderContextImpl::FillRectangle(Rect const& rect)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");
    KGE_ASSERT(current_brush_ && "The brush used for rendering has not been set!");

    render_target_->FillRectangle(DX::ConvertToRectF(rect), current_brush_->GetBrush().get());

    IncreasePrimitivesCount();
}

void RenderContextImpl::FillRoundedRectangle(Rect const& rect, Vec2 const& radius)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");
    KGE_ASSERT(current_brush_ && "The brush used for rendering has not been set!");

    render_target_->FillRoundedRectangle(D2D1::RoundedRect(DX::ConvertToRectF(rect), radius.x, radius.y),
                                         current_brush_->GetBrush().get());

    IncreasePrimitivesCount();
}

void RenderContextImpl::FillEllipse(Point const& center, Vec2 const& radius)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");
    KGE_ASSERT(current_brush_ && "The brush used for rendering has not been set!");

    render_target_->FillEllipse(D2D1::Ellipse(DX::ConvertToPoint2F(center), radius.x, radius.y),
                                current_brush_->GetBrush().get());

    IncreasePrimitivesCount();
}

void RenderContextImpl::CreateTexture(Texture& texture, math::Vec2T<uint32_t> size)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");

    ComPtr<ID2D1Bitmap> saved_bitmap;

    HRESULT hr = render_target_->CreateBitmap(D2D1::SizeU(size.x, size.y), D2D1::BitmapProperties(), &saved_bitmap);

    if (SUCCEEDED(hr))
    {
        texture.SetBitmap(saved_bitmap);
    }

    win32::ThrowIfFailed(hr);
}

void RenderContextImpl::PushClipRect(Rect const& clip_rect)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");
    render_target_->PushAxisAlignedClip(DX::ConvertToRectF(clip_rect),
                                        antialias_ ? D2D1_ANTIALIAS_MODE_PER_PRIMITIVE : D2D1_ANTIALIAS_MODE_ALIASED);
}

void RenderContextImpl::PopClipRect()
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");
    render_target_->PopAxisAlignedClip();
}

void RenderContextImpl::PushLayer(LayerArea& layer)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");
    if (!layer.IsValid())
    {
        ComPtr<ID2D1Layer> output;
        HRESULT            hr = render_target_->CreateLayer(&output);

        if (SUCCEEDED(hr))
        {
            layer.SetLayer(output);
        }
        else
        {
            win32::ThrowIfFailed(hr);
        }
    }

    if (layer.IsValid())
    {
        ComPtr<ID2D1Geometry> mask;
        if (layer.GetMaskShape())
            mask = layer.GetMaskShape()->GetGeometry();

        render_target_->PushLayer(
            D2D1::LayerParameters(DX::ConvertToRectF(layer.GetAreaRect()), mask.get(),
                                  antialias_ ? D2D1_ANTIALIAS_MODE_PER_PRIMITIVE : D2D1_ANTIALIAS_MODE_ALIASED,
                                  DX::ConvertToMatrix3x2F(layer.GetMaskTransform()), layer.GetOpacity(), nullptr,
                                  D2D1_LAYER_OPTIONS_NONE),
            layer.GetLayer().get());
    }
}

void RenderContextImpl::PopLayer()
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");
    render_target_->PopLayer();
}

void RenderContextImpl::Clear()
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");
    render_target_->Clear();
}

void RenderContextImpl::Clear(Color const& clear_color)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");
    render_target_->Clear(DX::ConvertToColorF(clear_color));
}

Size RenderContextImpl::GetSize() const
{
    if (render_target_)
    {
        return reinterpret_cast<const Size&>(render_target_->GetSize());
    }
    return Size();
}

void RenderContextImpl::SetCurrentBrush(BrushPtr brush)
{
    RenderContext::SetCurrentBrush(brush);

    if (current_brush_ && current_brush_->IsValid())
    {
        current_brush_->GetBrush()->SetOpacity(brush_opacity_);
    }
}

void RenderContextImpl::SetTransform(const Matrix3x2& matrix)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");

    if (fast_global_transform_)
    {
        render_target_->SetTransform(DX::ConvertToMatrix3x2F(&matrix));
    }
    else
    {
        Matrix3x2 result = matrix * global_transform_;
        render_target_->SetTransform(DX::ConvertToMatrix3x2F(&result));
    }
}

void RenderContextImpl::SetAntialiasMode(bool enabled)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");

    render_target_->SetAntialiasMode(enabled ? D2D1_ANTIALIAS_MODE_PER_PRIMITIVE : D2D1_ANTIALIAS_MODE_ALIASED);
    antialias_ = enabled;
}

void RenderContextImpl::SetTextAntialiasMode(TextAntialiasMode mode)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");

    D2D1_TEXT_ANTIALIAS_MODE antialias_mode = D2D1_TEXT_ANTIALIAS_MODE_CLEARTYPE;
    switch (mode)
    {
    case TextAntialiasMode::Default:
        antialias_mode = D2D1_TEXT_ANTIALIAS_MODE_DEFAULT;
        break;
    case TextAntialiasMode::ClearType:
        antialias_mode = D2D1_TEXT_ANTIALIAS_MODE_CLEARTYPE;
        break;
    case TextAntialiasMode::GrayScale:
        antialias_mode = D2D1_TEXT_ANTIALIAS_MODE_GRAYSCALE;
        break;
    case TextAntialiasMode::None:
        antialias_mode = D2D1_TEXT_ANTIALIAS_MODE_ALIASED;
        break;
    default:
        break;
    }

    text_antialias_ = mode;
    render_target_->SetTextAntialiasMode(antialias_mode);
}

bool RenderContextImpl::CheckVisibility(Rect const& bounds, Matrix3x2 const& transform)
{
    KGE_ASSERT(render_target_ && "Render target has not been initialized!");

    if (fast_global_transform_)
    {
        return visible_size_.Intersects(transform.Transform(bounds));
    }
    return visible_size_.Intersects(Matrix3x2(transform * global_transform_).Transform(bounds));
}

void RenderContextImpl::Resize(Size const& size)
{
    visible_size_ = Rect(Point(), size);
}

void RenderContextImpl::SaveDrawingState()
{
    KGE_ASSERT(IsValid());

    if (drawing_state_)
    {
        render_target_->SaveDrawingState(drawing_state_.get());
    }
}

void RenderContextImpl::RestoreDrawingState()
{
    KGE_ASSERT(IsValid());

    if (drawing_state_)
    {
        render_target_->RestoreDrawingState(drawing_state_.get());
    }
}

}  // namespace kiwano