#include "BsGUIColorField.h"
#include "BsGUIArea.h"
#include "BsGUILayout.h"
#include "BsGUILabel.h"
#include "BsGUIColor.h"
#include "BsBuiltinResources.h"
#include "BsGUIWidget.h"
#include "BsGUIMouseEvent.h"
#include "BsGUIWidget.h"

namespace BansheeEngine
{
	GUIColorField::GUIColorField(const PrivatelyConstruct& dummy, const GUIContent& labelContent, 
		const String& labelStyle, const String& colorStyle, const GUILayoutOptions& layoutOptions)
		:GUIElementContainer(layoutOptions), mLabel(nullptr), mColor(nullptr), mLabelWidth(100)
	{
		mLabel = GUILabel::create(labelContent, labelStyle);
		mColor = GUIColor::create(colorStyle);

		_registerChildElement(mLabel);
		_registerChildElement(mColor);
	}

	GUIColorField::GUIColorField(const PrivatelyConstruct& dummy, 
		const String& labelStyle, const String& colorStyle, const GUILayoutOptions& layoutOptions)
		:GUIElementContainer(layoutOptions), mLabel(nullptr), mColor(nullptr), mLabelWidth(100)
	{
		mColor = GUIColor::create(colorStyle);

		_registerChildElement(mColor);
	}

	GUIColorField::~GUIColorField()
	{

	}

	GUIColorField* GUIColorField::create(const GUIContent& labelContent, const GUIOptions& layoutOptions, 
		const String& labelStyle, const String& toggleStyle)
	{
		return bs_new<GUIColorField>(PrivatelyConstruct(), labelContent, labelStyle, toggleStyle, 
			GUILayoutOptions::create(layoutOptions));
	}

	GUIColorField* GUIColorField::create(const GUIContent& labelContent, const String& labelStyle, 
		const String& toggleStyle)
	{
		return bs_new<GUIColorField>(PrivatelyConstruct(), labelContent, labelStyle, toggleStyle, 
			GUILayoutOptions::create());
	}

	GUIColorField* GUIColorField::create(const HString& labelContent, const GUIOptions& layoutOptions, 
		const String& labelStyle, const String& toggleStyle)
	{
		return bs_new<GUIColorField>(PrivatelyConstruct(), GUIContent(labelContent), labelStyle, 
			toggleStyle, GUILayoutOptions::create(layoutOptions));
	}

	GUIColorField* GUIColorField::create( const HString& labelContent, const String& labelStyle, 
		const String& toggleStyle)
	{
		return bs_new<GUIColorField>(PrivatelyConstruct(), GUIContent(labelContent), labelStyle, toggleStyle, 
			GUILayoutOptions::create());
	}

	GUIColorField* GUIColorField::create(const GUIOptions& layoutOptions, const String& labelStyle, 
		const String& toggleStyle)
	{
		return bs_new<GUIColorField>(PrivatelyConstruct(), labelStyle, toggleStyle, 
			GUILayoutOptions::create(layoutOptions));
	}

	GUIColorField* GUIColorField::create(const String& labelStyle, const String& toggleStyle)
	{
		return bs_new<GUIColorField>(PrivatelyConstruct(), labelStyle, toggleStyle, GUILayoutOptions::create());
	}

	Color GUIColorField::getValue() const
	{
		return mColor->getColor();
	}

	void GUIColorField::setValue(const Color& color)
	{
		mColor->setColor(color);
	}

	void GUIColorField::setLabelWidth(UINT32 width)
	{
		mLabelWidth = width;

		markContentAsDirty();
	}

	void GUIColorField::_updateLayoutInternal(INT32 x, INT32 y, UINT32 width, UINT32 height,
		RectI clipRect, UINT8 widgetDepth, UINT16 areaDepth)
	{
		UINT32 colorOffset = 0;
		UINT32 colorWidth = width;

		if(mLabel != nullptr)
		{
			UINT32 labelWidth = mLabelWidth;

			Vector2I optimalSize = mLabel->_getOptimalSize();
			INT32 yOffset = Math::roundToInt((height - optimalSize.y) * 0.5f);

			Vector2I offset(x, y + yOffset);
			mLabel->_setOffset(offset);
			mLabel->_setWidth(labelWidth);
			mLabel->_setHeight(optimalSize.y);
			mLabel->_setAreaDepth(areaDepth);
			mLabel->_setWidgetDepth(widgetDepth);

			RectI elemClipRect(clipRect.x - offset.x, clipRect.y - offset.y, clipRect.width, clipRect.height);
			mLabel->_setClipRect(elemClipRect);

			colorOffset = labelWidth;
			colorWidth = width - labelWidth;
		}

		{
			Vector2I optimalSize = mColor->_getOptimalSize();
			INT32 yOffset = Math::roundToInt((height - optimalSize.y) * 0.5f);

			Vector2I offset(x + colorOffset, y + yOffset);
			mColor->_setOffset(offset);
			mColor->_setWidth(colorWidth);
			mColor->_setHeight(optimalSize.y);
			mColor->_setAreaDepth(areaDepth);
			mColor->_setWidgetDepth(widgetDepth);

			RectI elemClipRect(clipRect.x - offset.x, clipRect.y - offset.y, clipRect.width, clipRect.height);
			mColor->_setClipRect(elemClipRect);
		}
	}

	Vector2I GUIColorField::_getOptimalSize() const
	{
		Vector2I optimalsize = mColor->_getOptimalSize();

		if(mLabel != nullptr)
		{
			optimalsize.x += mLabel->_getOptimalSize().x;
			optimalsize.y = std::max(optimalsize.y, mLabel->_getOptimalSize().y);
		}

		return optimalsize;
	}

	const String& GUIColorField::getGUITypeName()
	{
		static String typeName = "GUIColorField";
		return typeName;
	}
}