/////////////////////////////////////////////////////////////////////
// Created by [Dennis Andersen] [2013]
/////////////////////////////////////////////////////////////////////
#include "..\..\Include\Win32\Win32Input.h"

using namespace Input;
using namespace Input::Enum;
using namespace Input::Struct;
using namespace Input::Typedefs;

void MapButton(RAWMOUSE& rawMouse, bool &isUp, Enum::SAMI& btn, int& delta, Struct::SAIPointInt2D& vel, unsigned int& mcode)
{
	if(rawMouse.lLastX != 0 || rawMouse.lLastY != 0)
	{
		vel.x = rawMouse.lLastX;
		vel.y = rawMouse.lLastY;
	}
	if( rawMouse.usButtonFlags > 0 )
	{
//--------------------------------------------------------------------------------------
		//Mouse button pressed
		if(rawMouse.usButtonFlags == RI_MOUSE_LEFT_BUTTON_DOWN)	
		{
			btn = SAMI_MouseLeftBtn;
			isUp = false;
		}
		else if(rawMouse.usButtonFlags == RI_MOUSE_MIDDLE_BUTTON_DOWN)	
		{	
			btn = SAMI_MouseMiddleBtn;
			isUp = false;
		}
		else if(rawMouse.usButtonFlags == RI_MOUSE_RIGHT_BUTTON_DOWN)	
		{	
			btn = SAMI_MouseRightBtn;
			isUp = false;
		}
//--------------------------------------------------------------------------------------
		//Mouse button Released
		else if(rawMouse.usButtonFlags == RI_MOUSE_LEFT_BUTTON_UP)	
		{
			btn = SAMI_MouseLeftBtn;
			isUp = true;
		}
		else if(rawMouse.usButtonFlags == RI_MOUSE_MIDDLE_BUTTON_UP)	
		{
			btn = SAMI_MouseMiddleBtn;
			isUp = true;
		}
		else if(rawMouse.usButtonFlags == RI_MOUSE_RIGHT_BUTTON_UP)	
		{
			btn = SAMI_MouseRightBtn;
			isUp = true;
		}
//--------------------------------------------------------------------------------------
		else if (rawMouse.usButtonFlags == RI_MOUSE_WHEEL)
		{
			delta = ((int)rawMouse.usButtonData);
	
			if(delta > 120)	delta = -1;
			else			delta = 1;
		}
	}
}

void ContainPoint(Struct::SAIPointInt2D& pixelPos, Struct::SAIPointInt2D& windowSize)
{
		 if(pixelPos.x < 0)				{ pixelPos.x = 0;				}
	else if(pixelPos.x > windowSize.x)	{ pixelPos.x = windowSize.x;	}
		 if(pixelPos.y < 0)				{ pixelPos.y = 0;				}
	else if(pixelPos.y > windowSize.y)	{ pixelPos.y = windowSize.y;	}
}

Win32Mouse::Win32Mouse(HWND target)
{
	this->isActive = false;
	this->device.usUsagePage = 0x01;
	this->device.hwndTarget = target;
	this->device.usUsage = RawInput_Usage_mouse;
	this->device.dwFlags = RIDEV_NOLEGACY | RIDEV_CAPTUREMOUSE;
	memset(&this->buttons[0], 0, sizeof(Buttons) * MAXBUTTONS);
}
Win32Mouse::~Win32Mouse()
{  }


bool Win32Mouse::IsBtnUp(Enum::SAMI btn) const
{
	if(btn == SAMI_Unknown) return false;

	return !this->buttons[btn].isDown;
}
bool Win32Mouse::IsBtnDown(Enum::SAMI btn) const
{
	if(btn == SAMI_Unknown) return false;

	return this->buttons[btn].isDown;
}
int Win32Mouse::GetWheelDelta() const
{
	return this->wheelDelta;
}
SAIPointInt2D& Win32Mouse::GetPixelPosition( Struct::SAIPointInt2D &targetMem ) const
{
	memcpy(&targetMem, &this->pixelPos, sizeof(SAIPointFloat2D));

	return targetMem;
}
SAIPointFloat2D& Win32Mouse::GetNormalizedPosition(Struct::SAIPointFloat2D& targetMem)
{
	RECT windowVertex;
	GetWindowRect( this->device.hwndTarget, &windowVertex );

	this->normalPos.x = (float)(pixelPos.x - windowVertex.left);
	this->normalPos.x /= (float)(windowVertex.right - windowVertex.left);

	this->normalPos.y = (float)(pixelPos.y - windowVertex.top);
	this->normalPos.y /= (float)(windowVertex.bottom - windowVertex.top);

	memcpy(&targetMem, &this->normalPos, sizeof(SAIPointFloat2D));

	return targetMem;
}
SAIPointFloat2D& Win32Mouse::GetDeltaPosition(Struct::SAIPointFloat2D& targetMem) const
{
	memcpy(&targetMem, &this->deltaPos, sizeof(SAIPointFloat2D));

	return targetMem;
}

void Win32Mouse::Activate ()
{
	if(this->isActive) return;

	this->Create();
}
void Win32Mouse::Deactivate ()
{
	if(!this->isActive) return;

	RAWINPUTDEVICE d;
	d.dwFlags = RIDEV_REMOVE;
	d.hwndTarget = 0;
	d.usUsage = RawInput_Usage_mouse;
	d.usUsagePage = 0x01;
	if(RegisterRawInputDevices(&d, 1, sizeof(RAWINPUTDEVICE)))
	{
		this->isActive = false;
		SetCursorPos(this->winCursPos.x, this->winCursPos.y);
		ShowCursor(TRUE);
	}
}

void Win32Mouse::ProccessMouseData (RAWMOUSE mouse)
{
	static MouseEventData mouseEventData;
	memset(&mouseEventData, 0, sizeof(MouseEventData));


	bool isUp = true;
	Enum::SAMI btn = Enum::SAMI_Unknown;
	int delta = 0;
	Struct::SAIPointInt2D velocity;
	unsigned int makeCode = 0;
	MapButton(mouse, isUp, btn, delta, velocity, makeCode);

	if(velocity.Length() != 0)
	{
		this->pixelPos.x += velocity.x;
		this->pixelPos.y += velocity.y;

		ContainPoint(this->pixelPos, this->windowSize);

		InternalOnMove(this->pixelPos, velocity);

		GetNormalizedPosition( mouseEventData.normalizedPos );
		mouseEventData.pixelPos			= this->pixelPos;
		mouseEventData.velocity			= velocity;
		mouseEventData.buttonState		= Enum::ButtonState_Unknown;
		mouseEventData.scrollDelta		= 0;
		mouseEventData.sender			= this;
		mouseEventData.type				= SAMI::SAMI_MouseMove;
		
		InternalOnEvent(mouseEventData);
	}

	if(delta != 0)
	{
		InternalOnScroll(delta);

		GetNormalizedPosition( mouseEventData.normalizedPos );
		mouseEventData.pixelPos			= this->pixelPos;
		mouseEventData.buttonState		= Enum::ButtonState_Unknown;
		mouseEventData.scrollDelta		= delta;
		mouseEventData.sender			= this;
		mouseEventData.type				= SAMI::SAMI_MouseScroll;
		
		InternalOnEvent(mouseEventData);
	}


	if(btn == SAMI_Unknown) return;

	this->buttons[btn].isDown = !isUp;
	this->buttons[btn].makeCode = makeCode;

	//The btn is released.
	if(isUp)
	{
		InternalOnBtnRelease(btn);

		GetNormalizedPosition( mouseEventData.normalizedPos );
		mouseEventData.pixelPos			= this->pixelPos;
		mouseEventData.buttonState		= Enum::ButtonState_Release;
		mouseEventData.type				= btn;
		mouseEventData.sender			= this;
		
		InternalOnEvent(mouseEventData);
	}
	//The btn is pressed.
	else
	{
		//The btn is down since last frame
		if(this->buttons[btn].isDown)
		{
			InternalOnBtnDown(btn);

			GetNormalizedPosition( mouseEventData.normalizedPos );
			mouseEventData.pixelPos			= this->pixelPos;
			mouseEventData.buttonState		= Enum::ButtonState_Down;
			mouseEventData.type				= btn;
			mouseEventData.sender			= this;
		
			InternalOnEvent(mouseEventData);
		}
		else
		{
			InternalOnBtnPress(btn);

			GetNormalizedPosition( mouseEventData.normalizedPos );
			mouseEventData.pixelPos			= this->pixelPos;
			mouseEventData.buttonState		= Enum::ButtonState_Press;
			mouseEventData.type				= btn;
			mouseEventData.sender			= this;
		
			InternalOnEvent(mouseEventData);
		}
	}
}
bool Win32Mouse::Create()
{
	if(!this->device.hwndTarget)
	{
		RECT desktop;
		const HWND hDesktop = GetDesktopWindow();
		GetClientRect(hDesktop, &desktop);
		windowSize.x = desktop.right;
		windowSize.y = desktop.bottom;
		this->device.dwFlags = 0;
	}
	else
	{
		RECT re;
		GetClientRect(this->device.hwndTarget, &re);
		windowSize.x = re.right - re.left;
		windowSize.y = re.bottom - re.top;
	}

	if(RegisterRawInputDevices(&this->device, 1, sizeof(RAWINPUTDEVICE)) == TRUE)
	{
		this->isActive = true;
		POINT p;
		GetCursorPos(&p);
		this->winCursPos.x = p.x;
		this->winCursPos.y = p.y;
		ShowCursor(FALSE);

		return true;
	}

	return false;
}
void Win32Mouse::ToggleDefault( bool toggler )
{
	if( toggler )
	{
		SetCursorPos(this->winCursPos.x, this->winCursPos.y);
		ShowCursor(TRUE);
	}
	else
	{
		POINT p;
		GetCursorPos(&p);
		this->winCursPos.x = p.x;
		this->winCursPos.y = p.y;
		ShowCursor(FALSE);
	}
}