// 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/renderer/RenderTarget.h>
#include <kiwano/core/Logger.h>

namespace kiwano
{
	//
	// RenderTarget
	//

	RenderTarget::RenderTarget()
		: collecting_status_(false)
		, fast_global_transform_(true)
		, brush_opacity_(1.0f)
		, antialias_(true)
		, text_antialias_(TextAntialiasMode::GrayScale)
	{
		status_.primitives = 0;
	}

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

		render_target_ = rt;
		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&>(GetRenderTarget()->GetSize()));
		}

		return hr;
	}

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

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

	void RenderTarget::BeginDraw()
	{
		if (collecting_status_)
		{
			status_.start = Time::Now();
			status_.primitives = 0;
		}

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

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

		if (collecting_status_)
		{
			status_.duration = Time::Now() - status_.start;
		}
	}

	void RenderTarget::DrawGeometry(Geometry const& geometry, float stroke_width, const StrokeStyle& stroke)
	{
		KGE_ASSERT(render_target_ && "Render target has not been initialized!");
		KGE_ASSERT(current_brush_ && "The brush used for rendering has not been set!");

		if (geometry.IsValid())
		{
			render_target_->DrawGeometry(
				geometry.GetGeometry().get(),
				current_brush_->GetBrush().get(),
				stroke_width,
				stroke.GetStrokeStyle().get()
			);

			IncreasePrimitivesCount();
		}
	}

	void RenderTarget::FillGeometry(Geometry const& geometry)
	{
		KGE_ASSERT(render_target_ && "Render target has not been initialized!");
		KGE_ASSERT(current_brush_ && "The brush used for rendering has not been set!");

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

			IncreasePrimitivesCount();
		}
	}

	void RenderTarget::DrawLine(Point const& point1, Point const& point2, float stroke_width, const StrokeStyle& stroke)
	{
		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_->DrawLine(
			DX::ConvertToPoint2F(point1),
			DX::ConvertToPoint2F(point2),
			current_brush_->GetBrush().get(),
			stroke_width,
			stroke.GetStrokeStyle().get()
		);

		IncreasePrimitivesCount();
	}

	void RenderTarget::DrawRectangle(Rect const& rect, float stroke_width, const StrokeStyle& stroke)
	{
		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_->DrawRectangle(
			DX::ConvertToRectF(rect),
			current_brush_->GetBrush().get(),
			stroke_width,
			stroke.GetStrokeStyle().get()
		);

		IncreasePrimitivesCount();
	}

	void RenderTarget::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 RenderTarget::DrawRoundedRectangle(Rect const& rect, Vec2 const& radius, float stroke_width, const StrokeStyle& stroke)
	{
		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_->DrawRoundedRectangle(
			D2D1::RoundedRect(
				DX::ConvertToRectF(rect),
				radius.x,
				radius.y
			),
			current_brush_->GetBrush().get(),
			stroke_width,
			stroke.GetStrokeStyle().get()
		);

		IncreasePrimitivesCount();
	}

	void RenderTarget::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 RenderTarget::DrawEllipse(Point const& center, Vec2 const& radius, float stroke_width, const StrokeStyle& stroke)
	{
		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_->DrawEllipse(
			D2D1::Ellipse(
				DX::ConvertToPoint2F(center),
				radius.x,
				radius.y
			),
			current_brush_->GetBrush().get(),
			stroke_width,
			stroke.GetStrokeStyle().get()
		);

		IncreasePrimitivesCount();
	}

	void RenderTarget::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 RenderTarget::DrawTexture(Texture const& texture, Rect const& src_rect, Rect const& dest_rect)
	{
		DrawTexture(texture, &src_rect, &dest_rect);
	}

	void RenderTarget::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 RenderTarget::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 = text_renderer_->DrawTextLayout(
				layout.GetTextLayout().get(),
				offset.x,
				offset.y,
				fill_brush.get(),
				outline_brush.get(),
				style.outline_width,
				style.outline_stroke.GetStrokeStyle().get()
			);

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

	void RenderTarget::CreateTexture(Texture& texture, math::Vec2T<uint32_t> size, D2D1_PIXEL_FORMAT format)
	{
		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(format),
			&saved_bitmap
		);

		if (SUCCEEDED(hr))
		{
			texture.SetBitmap(saved_bitmap);
		}
		else
		{
			win32::ThrowIfFailed(hr);
		}
	}

	void RenderTarget::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 RenderTarget::PopClipRect()
	{
		KGE_ASSERT(render_target_ && "Render target has not been initialized!");
		render_target_->PopAxisAlignedClip();
	}

	void RenderTarget::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())
		{
			render_target_->PushLayer(
				D2D1::LayerParameters(
					DX::ConvertToRectF(layer.GetAreaRect()),
					layer.GetMaskGeometry().GetGeometry().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 RenderTarget::PopLayer()
	{
		KGE_ASSERT(render_target_ && "Render target has not been initialized!");
		render_target_->PopLayer();
	}

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

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

	void RenderTarget::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 RenderTarget::SetGlobalTransform(const Matrix3x2* matrix)
	{
		if (matrix)
		{
			global_transform_ = *matrix;
			fast_global_transform_ = false;
		}
		else
		{
			fast_global_transform_ = true;
		}
	}

	void RenderTarget::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 RenderTarget::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 RenderTarget::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 RenderTarget::Resize(Size const& size)
	{
		visible_size_ = Rect(Point(), size);
	}

	void RenderTarget::SetCollectingStatus(bool enable)
	{
		collecting_status_ = enable;
	}

	void RenderTarget::IncreasePrimitivesCount(uint32_t increase) const
	{
		if (collecting_status_)
		{
			status_.primitives += increase;
		}
	}


	//
	// TextureRenderTarget
	//

	TextureRenderTarget::TextureRenderTarget()
	{
	}

	bool TextureRenderTarget::GetOutput(Texture& texture)
	{
		HRESULT hr = E_FAIL;

		if (bitmap_rt_)
		{
			ComPtr<ID2D1Bitmap> bitmap;

			hr = bitmap_rt_->GetBitmap(&bitmap);

			if (SUCCEEDED(hr))
			{
				texture.SetBitmap(bitmap);
			}
		}
		return SUCCEEDED(hr);
	}

}