// Copyright (c) 2016-2018 Easy2D - 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 "render.h"
#include "logs.h"
#include "Factory.h"
#include "Image.h"
#include "Transform.hpp"

namespace easy2d
{
	RenderSystem::RenderSystem()
		: fps_text_format_(nullptr)
		, fps_text_layout_(nullptr)
		, clear_color_(D2D1::ColorF(D2D1::ColorF::Black))
		, opacity_(1.f)
		, debug_(false)
		, window_occluded_(false)
		, vsync_enabled_(true)
		, antialias_(true)
		, text_antialias_(TextAntialias::ClearType)
	{
	}

	RenderSystem::~RenderSystem()
	{
		E2D_LOG(L"Destroying graphics device");

		ClearImageCache();
	}

	HRESULT RenderSystem::Init(HWND hwnd, bool vsync, bool debug)
	{
		E2D_LOG(L"Initing graphics device");

		vsync_enabled_ = vsync;
		debug_ = debug;

		return CreateResources(hwnd);
	}

	HRESULT RenderSystem::BeginDraw(HWND hwnd)
	{
		HRESULT hr = CreateResources(hwnd);

		if (debug_)
		{
			status_.start = time::Now();
			status_.primitives = 0;
		}

		if (SUCCEEDED(hr))
		{
			window_occluded_ = !!(render_target_->CheckWindowState() & D2D1_WINDOW_STATE_OCCLUDED);

			if (!window_occluded_)
			{
				render_target_->BeginDraw();
				render_target_->Clear(clear_color_);
			}
		}
		return hr;
	}

	HRESULT RenderSystem::EndDraw()
	{
		HRESULT hr = S_OK;
		
		if (!window_occluded_)
		{
			hr = render_target_->EndDraw();

			if (hr == D2DERR_RECREATE_TARGET)
			{
				// 如果 Direct3D 设备在执行过程中消失，将丢弃当前的设备相关资源
				// 并在下一次调用时重建资源
				DiscardResources();
				hr = S_OK;
			}
		}

		if (debug_)
		{
			status_.duration = time::Now() - status_.start;
		}
		return hr;
	}

	void RenderSystem::ClearImageCache()
	{
		bitmap_cache_.clear();
	}

	D2DHwndRenderTargetPtr const & RenderSystem::GetRenderTarget() const
	{
		return render_target_;
	}

	D2DSolidColorBrushPtr const & RenderSystem::GetSolidBrush() const
	{
		return solid_brush_;
	}

	HRESULT RenderSystem::CreateLayer(D2DLayerPtr& layer)
	{
		if (!render_target_)
			return E_UNEXPECTED;

		layer = nullptr;
		return render_target_->CreateLayer(&layer);
	}

	HRESULT RenderSystem::CreateSolidColorBrush(D2DSolidColorBrushPtr & brush) const
	{
		if (!render_target_)
			return E_UNEXPECTED;

		brush = nullptr;
		return render_target_->CreateSolidColorBrush(
			D2D1::ColorF(D2D1::ColorF::White),
			&brush
		);
	}

	HRESULT RenderSystem::DrawGeometry(
		D2DGeometryPtr const& geometry,
		Color const& stroke_color,
		float stroke_width,
		StrokeStyle stroke
	)
	{
		if (!solid_brush_ ||
			!render_target_)
			return E_UNEXPECTED;

		if (window_occluded_)
			return S_OK;

		solid_brush_->SetColor(stroke_color);
		auto stroke_style = Factory::Instance()->GetStrokeStyle(stroke);
		render_target_->DrawGeometry(
			geometry.Get(),
			solid_brush_.Get(),
			stroke_width,
			stroke_style.Get()
		);

		if (debug_)
			++status_.primitives;
		return S_OK;
	}

	HRESULT RenderSystem::FillGeometry(D2DGeometryPtr const & geometry, const Color & fill_color)
	{
		if (!solid_brush_ ||
			!render_target_)
			return E_UNEXPECTED;

		if (window_occluded_)
			return S_OK;

		solid_brush_->SetColor(fill_color);
		render_target_->FillGeometry(
			geometry.Get(),
			solid_brush_.Get()
		);

		if (debug_)
			++status_.primitives;
		return S_OK;
	}

	HRESULT RenderSystem::DrawImage(ImagePtr const & image, Rect const& dest_rect)
	{
		if (!render_target_)
			return E_UNEXPECTED;

		if (!image->GetBitmap())
			return S_OK;

		if (window_occluded_)
			return S_OK;

		render_target_->DrawBitmap(
			image->GetBitmap().Get(),
			dest_rect,
			opacity_,
			D2D1_BITMAP_INTERPOLATION_MODE_LINEAR,
			image->GetCropRect()
		);

		if (debug_)
			++status_.primitives;
		return S_OK;
	}

	HRESULT RenderSystem::DrawBitmap(
		D2DBitmapPtr const& bitmap
	)
	{
		if (!render_target_)
			return E_UNEXPECTED;

		if (window_occluded_)
			return S_OK;

		// Do not crop bitmap 
		auto rect = D2D1::RectF(0.f, 0.f, bitmap->GetSize().width, bitmap->GetSize().height);
		render_target_->DrawBitmap(
			bitmap.Get(),
			rect,
			opacity_,
			D2D1_BITMAP_INTERPOLATION_MODE_LINEAR,
			rect
		);

		if (debug_)
			++status_.primitives;
		return S_OK;
	}

	HRESULT RenderSystem::DrawTextLayout(D2DTextLayoutPtr const& text_layout)
	{
		if (!text_renderer_)
			return E_UNEXPECTED;

		if (window_occluded_)
			return S_OK;

		if (debug_)
			++status_.primitives;
		return text_layout->Draw(nullptr, text_renderer_.Get(), 0, 0);
	}

	HRESULT RenderSystem::PushClip(const Matrix & clip_matrix, const Size & clip_size)
	{
		if (!render_target_)
			return E_UNEXPECTED;

		if (window_occluded_)
			return S_OK;

		render_target_->SetTransform(clip_matrix);
		render_target_->PushAxisAlignedClip(
			D2D1::RectF(0, 0, clip_size.x, clip_size.y),
			D2D1_ANTIALIAS_MODE_PER_PRIMITIVE
		);
		return S_OK;
	}

	HRESULT RenderSystem::PopClip()
	{
		if (!render_target_)
			return E_UNEXPECTED;

		if (window_occluded_)
			return S_OK;

		render_target_->PopAxisAlignedClip();
		return S_OK;
	}

	HRESULT RenderSystem::PushLayer(D2DLayerPtr const& layer, LayerProperties const& properties)
	{
		if (!render_target_ ||
			!solid_brush_)
			return E_UNEXPECTED;

		if (window_occluded_)
			return S_OK;

		render_target_->PushLayer(
			D2D1::LayerParameters(
				properties.area,
				nullptr,
				D2D1_ANTIALIAS_MODE_PER_PRIMITIVE,
				D2D1::Matrix3x2F::Identity(),
				properties.opacity,
				solid_brush_.Get(),
				D2D1_LAYER_OPTIONS_NONE
			),
			layer.Get()
		);
		return S_OK;
	}

	HRESULT RenderSystem::PopLayer()
	{
		if (!render_target_)
			return E_UNEXPECTED;

		if (window_occluded_)
			return S_OK;

		render_target_->PopLayer();
		return S_OK;
	}

	HRESULT RenderSystem::GetSize(Size & size)
	{
		if (!render_target_)
			return E_UNEXPECTED;

		auto rtsize = render_target_->GetSize();
		size.x = rtsize.width;
		size.y = rtsize.height;
		return S_OK;
	}

	HRESULT RenderSystem::CreateBitmapFromFile(D2DBitmapPtr& bitmap, String const& file_path)
	{
		if (render_target_ == nullptr)
		{
			return E_UNEXPECTED;
		}

		size_t hash_code = std::hash<String>{}(file_path);
		if (bitmap_cache_.find(hash_code) != bitmap_cache_.end())
		{
			bitmap = bitmap_cache_[hash_code];
			return S_OK;
		}

		D2DBitmapPtr bitmap_tmp;
		HRESULT hr = Factory::Instance()->CreateBitmapFromFile(
			bitmap,
			render_target_,
			file_path
		);

		if (SUCCEEDED(hr))
		{
			bitmap_cache_.insert(std::make_pair(hash_code, bitmap));
		}

		return hr;
	}

	HRESULT RenderSystem::CreateBitmapFromResource(D2DBitmapPtr& bitmap, Resource const& res)
	{
		if (render_target_ == nullptr)
		{
			return E_UNEXPECTED;
		}

		size_t hash_code = res.GetHashCode();
		if (bitmap_cache_.find(hash_code) != bitmap_cache_.end())
		{
			bitmap = bitmap_cache_[hash_code];
			return S_OK;
		}

		HRESULT hr = Factory::Instance()->CreateBitmapFromResource(
			bitmap,
			render_target_,
			res
		);

		if (SUCCEEDED(hr))
		{
			bitmap_cache_.insert(std::make_pair(hash_code, bitmap));
		}

		return hr;
	}

	HRESULT RenderSystem::CreateBitmapRenderTarget(D2DBitmapRenderTargetPtr & brt)
	{
		if (!render_target_)
			return E_UNEXPECTED;

		brt = nullptr;
		return render_target_->CreateCompatibleRenderTarget(&brt);
	}

	HRESULT RenderSystem::Resize(UINT32 width, UINT32 height)
	{
		if (!render_target_)
			return E_UNEXPECTED;

		render_target_->Resize(D2D1::SizeU(width, height));
		return S_OK;
	}

	HRESULT RenderSystem::SetTransform(const Matrix & matrix)
	{
		if (!render_target_)
			return E_UNEXPECTED;

		render_target_->SetTransform(matrix);
		return S_OK;
	}

	HRESULT RenderSystem::SetOpacity(float opacity)
	{
		if (!render_target_)
			return E_UNEXPECTED;

		opacity_ = opacity;
		solid_brush_->SetOpacity(opacity);
		return S_OK;
	}

	HRESULT RenderSystem::SetTextStyle(
		Color const& color,
		bool has_outline,
		Color const& outline_color,
		float outline_width,
		StrokeStyle outline_stroke
	)
	{
		if (!text_renderer_)
			return E_UNEXPECTED;

		auto stroke_style = Factory::Instance()->GetStrokeStyle(outline_stroke);
		text_renderer_->SetTextStyle(
			color,
			has_outline,
			outline_color,
			outline_width,
			stroke_style.Get()
		);
		return S_OK;
	}

	void RenderSystem::SetClearColor(const Color& color)
	{
		clear_color_ = color;
	}

	HRESULT RenderSystem::SetAntialiasMode(bool enabled)
	{
		if (!render_target_)
			return E_UNEXPECTED;

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

	HRESULT RenderSystem::SetTextAntialiasMode(TextAntialias mode)
	{
		if (!render_target_)
			return E_UNEXPECTED;

		text_antialias_ = mode;
		D2D1_TEXT_ANTIALIAS_MODE antialias_mode = D2D1_TEXT_ANTIALIAS_MODE_CLEARTYPE;
		switch (text_antialias_)
		{
		case TextAntialias::Default:
			antialias_mode = D2D1_TEXT_ANTIALIAS_MODE_DEFAULT;
			break;
		case TextAntialias::ClearType:
			antialias_mode = D2D1_TEXT_ANTIALIAS_MODE_CLEARTYPE;
			break;
		case TextAntialias::GrayScale:
			antialias_mode = D2D1_TEXT_ANTIALIAS_MODE_GRAYSCALE;
			break;
		case TextAntialias::None:
			antialias_mode = D2D1_TEXT_ANTIALIAS_MODE_ALIASED;
			break;
		default:
			break;
		}
		render_target_->SetTextAntialiasMode(antialias_mode);
		return S_OK;
	}

	RenderSystem::Status const & RenderSystem::GetStatus() const
	{
		return status_;
	}

	HRESULT RenderSystem::CreateResources(HWND hwnd)
	{
		HRESULT hr = S_OK;

		if (!render_target_)
		{
			RECT rc;
			::GetClientRect(hwnd, &rc);

			D2D1_SIZE_U size = D2D1::SizeU(
				rc.right - rc.left,
				rc.bottom - rc.top
			);

			hr = Factory::Instance()->CreateHwndRenderTarget(
				render_target_,
				D2D1::RenderTargetProperties(
					D2D1_RENDER_TARGET_TYPE_DEFAULT,
					D2D1::PixelFormat(),
					96.f,
					96.f
				),
				D2D1::HwndRenderTargetProperties(
					hwnd,
					size,
					vsync_enabled_ ? D2D1_PRESENT_OPTIONS_NONE : D2D1_PRESENT_OPTIONS_IMMEDIATELY
				)
			);

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

			if (SUCCEEDED(hr))
			{
				hr = render_target_->CreateSolidColorBrush(
					D2D1::ColorF(D2D1::ColorF::White),
					&solid_brush_
				);
			}

			if (SUCCEEDED(hr))
			{
				hr = Factory::Instance()->CreateTextRenderer(
					text_renderer_,
					render_target_,
					solid_brush_
				);
			}
		}
		return hr;
	}

	void RenderSystem::DiscardResources()
	{
		// FIXME! 应通知 Application 类销毁所有节点的 device resources
		fps_text_format_ = nullptr;
		fps_text_layout_ = nullptr;
		text_renderer_ = nullptr;
		solid_brush_ = nullptr;
		render_target_ = nullptr;
	}

}