#pragma once

#include "BsEditorPrerequisites.h"
#include "BsGUIElementContainer.h"
#include "BsRectI.h"

namespace BansheeEngine
{
	enum class DockLocation
	{
		Left, Right, Top, Bottom, Center
	};

	class DockManager : public GUIElementContainer
	{
	public:
		class DockContainer
		{
		public:
			DockContainer();
			DockContainer(DockContainer* parent);
			~DockContainer();

			void setArea(INT32 x, INT32 y, UINT32 width, UINT32 height);
			void makeLeaf(GUIWidget* widgetParent, RenderWindow* parentWindow);
			void makeLeaf(EditorWidgetContainer* existingContainer);
			void splitContainer(GUIWidget* widgetParent, RenderWindow* parentWindow, bool horizontal, bool newChildIsFirst, float splitPosition = 0.5f);
			void addLeft(GUIWidget* widgetParent, RenderWindow* parentWindow, EditorWidgetBase* widget);
			void addRight(GUIWidget* widgetParent, RenderWindow* parentWindow, EditorWidgetBase* widget);
			void addTop(GUIWidget* widgetParent, RenderWindow* parentWindow, EditorWidgetBase* widget);
			void addBottom(GUIWidget* widgetParent, RenderWindow* parentWindow, EditorWidgetBase* widget);

			void addWidget(EditorWidgetBase* widget);
			void addWidget(const String& name);

			DockContainer* find(EditorWidgetContainer* widgetContainer);

			/**
			 * @brief	Searches for a DockContainer at the specified position.
			 *
			 * @param	pos	Position in the same space as DockContainer. 
			 *
			 * @return	null if it fails, else the found DockContainer at position.
			 */
			DockContainer* findAtPos(const Vector2I& pos);

			RectI getContentBounds() const;

			bool mIsLeaf;
			DockContainer* mChildren[2];
			DockContainer* mParent;
			EditorWidgetContainer* mWidgets;
			GUIDockSlider* mSlider;
			RectI mArea;
			float mSplitPosition;
			bool mIsHorizontal;

			static const UINT32 SLIDER_SIZE;
			static const UINT32 MIN_CHILD_SIZE;

		private:
			void updateChildAreas();

			void sliderDragged(const Vector2I& delta);
			void widgetRemoved();
		};

		enum class DockLocation
		{
			Top,
			Bottom,
			Left,
			Right,
			None
		};
	public:
		static DockManager* create(RenderWindow* parentWindow);

		/**
		 * @brief	Internal method. Called once every frame.
		 */
		void update();

		void render(const Viewport* viewport, DrawList& renderQueue);
		void insert(EditorWidgetContainer* relativeTo, EditorWidgetBase* widgetToInsert, DockLocation location);

		DockManagerLayoutPtr getLayout() const;
		void setLayout(const DockManagerLayoutPtr& layout);

		void setArea(INT32 x, INT32 y, UINT32 width, UINT32 height);

	protected:
		~DockManager();

		void updateClippedBounds();

	private:
		DockManager(RenderWindow* parentWindow, const GUILayoutOptions& layoutOptions);

		static const Color TINT_COLOR;
		static const Color HIGHLIGHT_COLOR;

		RenderWindow* mParentWindow;
		DockContainer mRootContainer;
		RectI mArea;

		HMesh mDropOverlayMesh;
		HMaterial mDropOverlayMat;
		RectI mLastOverlayBounds;

		DockContainer* mMouseOverContainer;
		DockLocation mHighlightedDropLoc;
		bool mShowOverlay;
		Vector2* mTopDropPolygon;
		Vector2* mBotDropPolygon;
		Vector2* mLeftDropPolygon;
		Vector2* mRightDropPolygon;
		bool mAddedRenderCallback;

		void updateDropOverlay(INT32 x, INT32 y, UINT32 width, UINT32 height);

		bool mouseEvent(const GUIMouseEvent& event);
		bool insidePolygon(Vector2* polyPoints, UINT32 numPoints, Vector2 point) const;

		void _changeParentWidget(GUIWidget* widget);
	};
}