// Copyright (c) 2016-2018 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 "ShapeNode.h"
#include "../base/logs.h"

namespace kiwano
{
	ShapeNode::ShapeNode()
		: fill_color_(Color::White)
		, stroke_color_(Color(Color::Black, 0))
		, stroke_width_(1.f)
		, outline_join_(StrokeStyle::Miter)
	{
	}

	ShapeNode::ShapeNode(ComPtr<ID2D1Geometry> geometry)
		: ShapeNode()
	{
		SetGeometry(geometry);
	}

	ShapeNode::~ShapeNode()
	{
	}

	Rect ShapeNode::GetBoundingBox()
	{
		if (!geo_)
			return Rect{};

		D2D1_RECT_F rect;
		// no matter it failed or not
		geo_->GetBounds(D2D1::Matrix3x2F::Identity(), &rect);
		return Rect{ rect.left, rect.top, rect.right - rect.left, rect.bottom - rect.top };
	}

	float ShapeNode::GetLength()
	{
		float length = 0.f;
		if (geo_)
		{
			// no matter it failed or not
			geo_->ComputeLength(D2D1::Matrix3x2F::Identity(), &length);
		}
		return length;
	}

	bool ShapeNode::ComputePointAtLength(float length, Point& point, Vec2& tangent)
	{
		if (geo_)
		{
			if (SUCCEEDED(geo_->ComputePointAtLength(
				length,
				D2D1::Matrix3x2F::Identity(),
				DX::ConvertToPoint2F(&point),
				DX::ConvertToPoint2F(&tangent))))
			{
				return true;
			}
		}
		return false;
	}

	float ShapeNode::ComputeArea()
	{
		if (!geo_)
			return 0.f;

		float area = 0.f;
		// no matter it failed or not
		geo_->ComputeArea(D2D1::Matrix3x2F::Identity(), &area);
		return area;
	}

	bool ShapeNode::ContainsPoint(Point const& point)
	{
		if (!geo_)
			return false;

		BOOL ret = 0;
		// no matter it failed or not
		geo_->FillContainsPoint(
			DX::ConvertToPoint2F(point),
			D2D1::Matrix3x2F::Identity(),
			&ret
		);
		return !!ret;
	}

	void ShapeNode::SetFillColor(const Color & color)
	{
		fill_color_ = color;
	}

	void ShapeNode::SetStrokeColor(const Color & color)
	{
		stroke_color_ = color;
	}

	void ShapeNode::SetStrokeWidth(float width)
	{
		stroke_width_ = std::max(width, 0.f);
	}

	void ShapeNode::SetOutlineJoinStyle(StrokeStyle outline_join)
	{
		outline_join_ = outline_join;
	}

	void ShapeNode::OnRender()
	{
		if (geo_)
		{
			Renderer::GetInstance()->FillGeometry(
				geo_,
				fill_color_
			);

			Renderer::GetInstance()->DrawGeometry(
				geo_,
				stroke_color_,
				stroke_width_,
				outline_join_
			);
		}
	}

	//-------------------------------------------------------
	// LineNode
	//-------------------------------------------------------

	LineNode::LineNode()
	{
	}

	LineNode::LineNode(Point const& begin, Point const& end)
	{
		SetLine(begin, end);
	}

	LineNode::~LineNode()
	{
	}

	void LineNode::SetLine(Point const& begin, Point const& end)
	{
		ComPtr<ID2D1PathGeometry> path_geo;
		ComPtr<ID2D1GeometrySink> path_sink;

		HRESULT hr = Renderer::GetInstance()->GetD2DDeviceResources()->GetFactory()->CreatePathGeometry(&path_geo);

		if (SUCCEEDED(hr))
		{
			hr = path_geo->Open(&path_sink);
		}

		if (SUCCEEDED(hr))
		{
			path_sink->BeginFigure(DX::ConvertToPoint2F(begin), D2D1_FIGURE_BEGIN_FILLED);
			path_sink->AddLine(DX::ConvertToPoint2F(end));
			path_sink->EndFigure(D2D1_FIGURE_END_OPEN);
			hr = path_sink->Close();
		}

		if (SUCCEEDED(hr))
		{
			geo_ = path_geo;
		}
	}

	void LineNode::SetBegin(Point const& begin)
	{
		SetLine(begin, end_);
	}

	void LineNode::SetEnd(Point const& end)
	{
		SetLine(begin_, end);
	}


	//-------------------------------------------------------
	// RectNode
	//-------------------------------------------------------

	RectNode::RectNode()
	{
	}

	RectNode::RectNode(Rect const& rect)
	{
		SetRect(rect);
	}

	RectNode::RectNode(Point const& left_top, Size const& size)
	{
		SetRect(Rect{ left_top, size });
	}

	RectNode::~RectNode()
	{
	}

	void RectNode::SetRect(Rect const& rect)
	{
		ComPtr<ID2D1RectangleGeometry> geo;
		auto factory = Renderer::GetInstance()->GetD2DDeviceResources()->GetFactory();

		if (SUCCEEDED(factory->CreateRectangleGeometry(DX::ConvertToRectF(rect), &geo)))
		{
			geo_ = geo;
			rect_ = rect;
		}
	}


	//-------------------------------------------------------
	// RoundedRectNode
	//-------------------------------------------------------

	RoundedRectNode::RoundedRectNode()
		: radius_x_(0.f)
		, radius_y_(0.f)
	{
	}

	RoundedRectNode::RoundedRectNode(Rect const& rect, float radius_x, float radius_y)
	{
		SetRoundedRect(rect, radius_x, radius_y);
	}

	RoundedRectNode::~RoundedRectNode()
	{
	}

	void RoundedRectNode::SetRadius(float radius_x, float radius_y)
	{
		SetRoundedRect(rect_, radius_x, radius_y);
	}

	void RoundedRectNode::SetRect(Rect const& rect)
	{
		SetRoundedRect(rect, radius_x_, radius_y_);
	}

	void RoundedRectNode::SetRoundedRect(Rect const& rect, float radius_x, float radius_y)
	{
		ComPtr<ID2D1RoundedRectangleGeometry> geo;
		auto factory = Renderer::GetInstance()->GetD2DDeviceResources()->GetFactory();

		if (SUCCEEDED(factory->CreateRoundedRectangleGeometry(
			D2D1::RoundedRect(
				DX::ConvertToRectF(rect),
				radius_x,
				radius_y
			),
			&geo)))
		{
			geo_ = geo;
			rect_ = rect;
			radius_x_ = radius_x;
			radius_y_ = radius_y;
		}
	}


	//-------------------------------------------------------
	// CircleNode
	//-------------------------------------------------------

	CircleNode::CircleNode()
		: radius_(0.f)
	{
	}

	CircleNode::CircleNode(Point const& center, float radius)
	{
		SetCircle(center, radius);
	}

	CircleNode::~CircleNode()
	{
	}

	void CircleNode::SetRadius(float radius)
	{
		SetCircle(center_, radius);
	}

	void CircleNode::SetCenter(Point const& center)
	{
		SetCircle(center, radius_);
	}

	void CircleNode::SetCircle(Point const& center, float radius)
	{
		ComPtr<ID2D1EllipseGeometry> geo;
		auto factory = Renderer::GetInstance()->GetD2DDeviceResources()->GetFactory();

		if (SUCCEEDED(factory->CreateEllipseGeometry(
			D2D1::Ellipse(
				DX::ConvertToPoint2F(center),
				radius,
				radius),
			&geo)))
		{
			geo_ = geo;
			center_ = center;
			radius_ = radius;
		}
	}


	//-------------------------------------------------------
	// EllipseNode
	//-------------------------------------------------------

	EllipseNode::EllipseNode()
		: radius_x_(0.f)
		, radius_y_(0.f)
	{
	}

	EllipseNode::EllipseNode(Point const& center, float radius_x, float radius_y)
	{
		SetEllipse(center, radius_x, radius_y);
	}

	EllipseNode::~EllipseNode()
	{
	}

	void EllipseNode::SetRadius(float radius_x, float radius_y)
	{
		SetEllipse(center_, radius_x, radius_y);
	}

	void EllipseNode::SetCenter(Point const& center)
	{
		SetEllipse(center, radius_x_, radius_y_);
	}

	void EllipseNode::SetEllipse(Point const& center, float radius_x, float radius_y)
	{
		ComPtr<ID2D1EllipseGeometry> geo;
		auto factory = Renderer::GetInstance()->GetD2DDeviceResources()->GetFactory();

		if (SUCCEEDED(factory->CreateEllipseGeometry(
			D2D1::Ellipse(
				DX::ConvertToPoint2F(center),
				radius_x,
				radius_y),
			&geo)))
		{
			geo_ = geo;
			radius_x_ = radius_x;
			radius_y_ = radius_y;
		}
	}


	//-------------------------------------------------------
	// PathNode
	//-------------------------------------------------------

	PathNode::PathNode()
	{
	}

	PathNode::~PathNode()
	{
	}

	void PathNode::BeginPath(Point const& begin_pos)
	{
		current_geometry_ = nullptr;

		auto factory = Renderer::GetInstance()->GetD2DDeviceResources()->GetFactory();

		ThrowIfFailed(
			factory->CreatePathGeometry(&current_geometry_)
		);

		ThrowIfFailed(
			current_geometry_->Open(&current_sink_)
		);

		current_sink_->BeginFigure(DX::ConvertToPoint2F(begin_pos), D2D1_FIGURE_BEGIN_FILLED);
	}

	void PathNode::EndPath(bool closed)
	{
		if (current_sink_)
		{
			current_sink_->EndFigure(closed ? D2D1_FIGURE_END_CLOSED : D2D1_FIGURE_END_OPEN);
			ThrowIfFailed(
				current_sink_->Close()
			);

			geo_ = current_geometry_;

			current_sink_ = nullptr;
			current_geometry_ = nullptr;
		}
	}

	void PathNode::AddLine(Point const& point)
	{
		if (current_sink_)
			current_sink_->AddLine(DX::ConvertToPoint2F(point));
	}

	void PathNode::AddLines(Vector<Point> const& points)
	{
		if (current_sink_ && !points.empty())
		{
			current_sink_->AddLines(
				reinterpret_cast<const D2D_POINT_2F*>(&points[0]),
				static_cast<UINT32>(points.size())
			);
		}
	}

	void PathNode::AddBezier(Point const& point1, Point const& point2, Point const& point3)
	{
		if (current_sink_)
		{
			current_sink_->AddBezier(
				D2D1::BezierSegment(
					DX::ConvertToPoint2F(point1),
					DX::ConvertToPoint2F(point2),
					DX::ConvertToPoint2F(point3)
				)
			);
		}
	}

	void PathNode::AddArc(Point const& point, Size const& radius, float rotation, bool clockwise, bool is_small)
	{
		if (current_sink_)
		{
			current_sink_->AddArc(
				D2D1::ArcSegment(
					DX::ConvertToPoint2F(point),
					DX::ConvertToSizeF(radius),
					rotation,
					clockwise ? D2D1_SWEEP_DIRECTION_CLOCKWISE : D2D1_SWEEP_DIRECTION_COUNTER_CLOCKWISE,
					is_small ? D2D1_ARC_SIZE_SMALL : D2D1_ARC_SIZE_LARGE
				)
			);
		}
	}

	void PathNode::ClearPath()
	{
		geo_ = nullptr;
		current_sink_ = nullptr;
		current_geometry_ = nullptr;
	}

}