BansheeEngine/BansheeEngine/Source/BsGUIInputSelection.cpp

#include "BsGUIInputSelection.h"
#include "BsImageSprite.h"
#include "BsGUIManager.h"

using namespace CamelotFramework;

namespace BansheeEngine
{
	GUIInputSelection::GUIInputSelection(const TEXT_SPRITE_DESC& textDesc, const CM::Int2& offset, const CM::Int2 clipOffset)
		:GUIInputTool(textDesc, offset, clipOffset), mSelectionStart(0), mSelectionEnd(0), mSelectionAnchor(0), mSelectionDragAnchor(0)
	{ }

	GUIInputSelection::~GUIInputSelection()
	{
		for(auto& sprite : mSprites)
			cm_delete<PoolAlloc>(sprite);
	}

	void GUIInputSelection::updateSprite()
	{
		mSelectionRects = getSelectionRects();

		INT32 diff = (INT32)(mSprites.size() - mSelectionRects.size());

		if(diff > 0)
		{
			for(UINT32 i = (UINT32)mSelectionRects.size(); i < (UINT32)mSprites.size(); i++)
				cm_delete(mSprites[i]);

			mSprites.erase(mSprites.begin() + mSelectionRects.size(), mSprites.end());
		}
		else if(diff < 0)
		{
			for(INT32 i = diff; i < 0; i++)
			{
				ImageSprite* newSprite = cm_new<ImageSprite>();
				mSprites.push_back(newSprite);
			}
		}

		UINT32 idx = 0;
		for(auto& sprite : mSprites)
		{
			IMAGE_SPRITE_DESC desc;
			desc.width = mSelectionRects[idx].width;
			desc.height = mSelectionRects[idx].height;
			desc.texture = GUIManager::instance().getTextSelectionTexture();

			sprite->update(desc);
			idx++;
		}
	}

	Int2 GUIInputSelection::getSelectionSpriteOffset(UINT32 spriteIdx) const
	{
		return Int2(mSelectionRects[spriteIdx].x, mSelectionRects[spriteIdx].y);
	}

	Rect GUIInputSelection::getSelectionSpriteClipRect(UINT32 spriteIdx) const
	{
		return Rect(-mSelectionRects[spriteIdx].x + mTextOffset.x - mClipOffset.x, 
			 -mSelectionRects[spriteIdx].y + mTextOffset.y - mClipOffset.y, 
			mTextDesc.width, mTextDesc.height);
	}

	Vector<Rect>::type GUIInputSelection::getSelectionRects() const
	{
		Vector<Rect>::type selectionRects;

		if(mSelectionStart == mSelectionEnd)
			return selectionRects;

		UINT32 startLine = getLineForChar(mSelectionStart);

		UINT32 endLine = startLine;
		if(mSelectionEnd > 0)
			endLine = getLineForChar(mSelectionEnd - 1, true);

		{
			const GUIInputLineDesc& lineDesc = getLineDesc(startLine);

			UINT32 startCharIdx = mSelectionStart;

			UINT32 endCharIdx = mSelectionEnd - 1;
			if(startLine != endLine)
			{
				endCharIdx = lineDesc.getEndChar(false);
				if(endCharIdx > 0)
					endCharIdx = endCharIdx - 1;
			}

			if(!isNewlineChar(startCharIdx) && !isNewlineChar(endCharIdx))
			{
				Rect startChar = getCharRect(startCharIdx);
				Rect endChar = getCharRect(endCharIdx);

				Rect selectionRect;
				selectionRect.x = startChar.x;
				selectionRect.y = lineDesc.getLineYStart();
				selectionRect.height = lineDesc.getLineHeight();
				selectionRect.width = (endChar.x + endChar.width) - startChar.x;

				selectionRects.push_back(selectionRect);
			}
		}

		for(UINT32 i = startLine + 1; i < endLine; i++)
		{
			const GUIInputLineDesc& lineDesc = getLineDesc(i);
			if(lineDesc.getStartChar() == lineDesc.getEndChar() || isNewlineChar(lineDesc.getStartChar()))
				continue;

			UINT32 endCharIdx = lineDesc.getEndChar(false);
			if(endCharIdx > 0)
				endCharIdx = endCharIdx - 1;

			Rect startChar = getCharRect(lineDesc.getStartChar());
			Rect endChar = getCharRect(endCharIdx);

			Rect selectionRect;
			selectionRect.x = startChar.x;
			selectionRect.y = lineDesc.getLineYStart();
			selectionRect.height = lineDesc.getLineHeight();
			selectionRect.width = (endChar.x + endChar.width) - startChar.x;

			selectionRects.push_back(selectionRect);
		}

		if(startLine != endLine)
		{
			const GUIInputLineDesc& lineDesc = getLineDesc(endLine);

			if(lineDesc.getStartChar() != lineDesc.getEndChar() && !isNewlineChar(lineDesc.getStartChar()))
			{
				UINT32 endCharIdx = mSelectionEnd - 1;

				if(!isNewlineChar(endCharIdx))
				{
					Rect startChar = getCharRect(lineDesc.getStartChar());
					Rect endChar = getCharRect(endCharIdx);

					Rect selectionRect;
					selectionRect.x = startChar.x;
					selectionRect.y = lineDesc.getLineYStart();
					selectionRect.height = lineDesc.getLineHeight();
					selectionRect.width = (endChar.x + endChar.width) - startChar.x;

					selectionRects.push_back(selectionRect);
				}
			}
		}

		return selectionRects;
	}

	void GUIInputSelection::showSelection(CM::UINT32 anchorCaretPos)
	{
		UINT32 charIdx = caretPosToSelectionChar(anchorCaretPos, SelectionDir::Left);

		mSelectionStart = charIdx;
		mSelectionEnd = charIdx;
		mSelectionAnchor = charIdx;
	}

	void GUIInputSelection::clearSelection()
	{
		for(auto& sprite : mSprites)
			cm_delete(sprite);

		mSprites.clear();
	}

	UINT32 GUIInputSelection::caretPosToSelectionChar(UINT32 caretPos, SelectionDir dir) const
	{
		UINT32 charIdx = getCharIdxAtInputIdx(caretPos);

		if(dir == SelectionDir::Right)
			charIdx = (UINT32)std::max(0, (INT32)(charIdx - 1));

		return charIdx;
	}

	void GUIInputSelection::selectionDragStart(UINT32 caretPos)
	{
		clearSelection();

		showSelection(caretPos); 
		mSelectionDragAnchor = caretPos;
	}

	void GUIInputSelection::selectionDragUpdate(UINT32 caretPos)
	{
		if(caretPos < mSelectionDragAnchor)
		{
			mSelectionStart = getCharIdxAtInputIdx(caretPos);
			mSelectionEnd = getCharIdxAtInputIdx(mSelectionDragAnchor);

			mSelectionAnchor = mSelectionStart;
		}

		if(caretPos > mSelectionDragAnchor)
		{
			mSelectionStart = getCharIdxAtInputIdx(mSelectionDragAnchor);
			mSelectionEnd = getCharIdxAtInputIdx(caretPos);

			mSelectionAnchor = mSelectionEnd;
		}

		if(caretPos == mSelectionDragAnchor)
		{
			mSelectionStart = mSelectionAnchor;
			mSelectionEnd = mSelectionAnchor;
		}
	}

	void GUIInputSelection::selectionDragEnd()
	{
		if(isSelectionEmpty())
			clearSelection();
	}

	void GUIInputSelection::moveSelectionToCaret(UINT32 caretPos)
	{
		UINT32 charIdx = caretPosToSelectionChar(caretPos, SelectionDir::Left);

		if(charIdx > mSelectionAnchor)
		{
			mSelectionStart = mSelectionAnchor;
			mSelectionEnd = charIdx;
		}
		else
		{
			mSelectionStart = charIdx;
			mSelectionEnd = mSelectionAnchor;
		}

		if(mSelectionStart == mSelectionEnd)
			clearSelection();
	}

	void GUIInputSelection::selectAll()
	{
		mSelectionStart = 0;
		mSelectionEnd = (UINT32)mTextDesc.text.size();
	}

	bool GUIInputSelection::isSelectionEmpty() const
	{
		return mSelectionStart == mSelectionEnd;
	}
}