#include "GsCachedAreaTest.h"
#include "gs/GsCachedArea.h"
#include "gs/GSHandler.h"
#include "gs/GsPixelFormats.h"

void CGsCachedAreaTest::Execute()
{
	CheckEmptyArea();
	CheckDirtyRect();
	CheckClearDirtyPages();
	CheckInvalidate();
}

void CGsCachedAreaTest::CheckEmptyArea()
{
	CGsCachedArea area;
	area.SetArea(CGSHandler::PSMCT32, 0, 0, 0);

	auto areaRect = area.GetAreaPageRect();
	TEST_VERIFY(areaRect.width == 0);
	TEST_VERIFY(areaRect.height == 0);

	auto areaSize = area.GetSize();
	TEST_VERIFY(areaSize == 0);

	auto dirtyRect = area.GetDirtyPageRect();
	TEST_VERIFY(dirtyRect.x == 0);
	TEST_VERIFY(dirtyRect.y == 0);
	TEST_VERIFY(dirtyRect.width == 0);
	TEST_VERIFY(dirtyRect.height == 0);
}

void CGsCachedAreaTest::CheckDirtyRect()
{
	//Completely dirty
	{
		CGsCachedArea area;
		area.SetArea(CGSHandler::PSMCT32, 0, 512, 512);

		auto areaRect = area.GetAreaPageRect();
		for(uint32 y = 0; y < areaRect.height; y++)
		{
			for(uint32 x = 0; x < areaRect.width; x++)
			{
				uint32 pageIndex = x + (y * areaRect.width);
				area.SetPageDirty(pageIndex);
			}
		}

		auto dirtyRect = area.GetDirtyPageRect();
		TEST_VERIFY(dirtyRect.x == 0);
		TEST_VERIFY(dirtyRect.y == 0);
		TEST_VERIFY(dirtyRect.width == areaRect.width);
		TEST_VERIFY(dirtyRect.height == areaRect.height);
	}

	//One page
	{
		CGsCachedArea area;
		area.SetArea(CGSHandler::PSMCT32, 0, 512, 512);

		auto areaRect = area.GetAreaPageRect();
		uint32 dirtyX = 3;
		uint32 dirtyY = 2;
		area.SetPageDirty(dirtyX + (dirtyY * areaRect.width));

		auto dirtyRect = area.GetDirtyPageRect();
		TEST_VERIFY(dirtyRect.x == dirtyX);
		TEST_VERIFY(dirtyRect.y == dirtyY);
		TEST_VERIFY(dirtyRect.width == 1);
		TEST_VERIFY(dirtyRect.height == 1);
	}

	//Completely clear
	{
		CGsCachedArea area;
		area.SetArea(CGSHandler::PSMCT32, 0, 512, 512);

		auto dirtyRect = area.GetDirtyPageRect();
		TEST_VERIFY(dirtyRect.x == 0);
		TEST_VERIFY(dirtyRect.y == 0);
		TEST_VERIFY(dirtyRect.width == 0);
		TEST_VERIFY(dirtyRect.height == 0);
	}
}

void CGsCachedAreaTest::CheckClearDirtyPages()
{
	CGsCachedArea area;
	area.SetArea(CGSHandler::PSMCT32, 0, 512, 512);

	auto areaRect = area.GetAreaPageRect();
	uint32 dirtyX = 3;
	uint32 dirtyY = 2;
	area.SetPageDirty(dirtyX + (dirtyY * areaRect.width));
	TEST_VERIFY(area.HasDirtyPages());

	area.ClearDirtyPages(CGsCachedArea::PageRect{0, 0, 1, 1});
	TEST_VERIFY(area.HasDirtyPages());

	area.ClearDirtyPages(CGsCachedArea::PageRect{0, 0, 3, 2});
	TEST_VERIFY(area.HasDirtyPages());

	area.ClearDirtyPages(CGsCachedArea::PageRect{3, 2, 1, 1});
	TEST_VERIFY(!area.HasDirtyPages());
}

void CGsCachedAreaTest::CheckInvalidate()
{
	auto pixelFormat = CGSHandler::PSMCT32;
	auto pageSize = CGsPixelFormats::GetPsmPageSize(pixelFormat);
	uint32 areaWidth = 512;
	uint32 areaPageWidth = areaWidth / pageSize.first;

	//Invalidate first line
	{
		CGsCachedArea area;
		area.SetArea(pixelFormat, 0, areaWidth, 512);

		area.Invalidate(0, areaPageWidth * CGsPixelFormats::PAGESIZE);

		auto dirtyRect = area.GetDirtyPageRect();
		TEST_VERIFY(dirtyRect.x == 0);
		TEST_VERIFY(dirtyRect.y == 0);
		TEST_VERIFY(dirtyRect.width == areaPageWidth);
		TEST_VERIFY(dirtyRect.height == 1);
	}

	//Invalidate second line
	{
		CGsCachedArea area;
		area.SetArea(pixelFormat, 0, areaWidth, 512);

		area.Invalidate(areaPageWidth * CGsPixelFormats::PAGESIZE, areaPageWidth * CGsPixelFormats::PAGESIZE);

		auto dirtyRect = area.GetDirtyPageRect();
		TEST_VERIFY(dirtyRect.x == 0);
		TEST_VERIFY(dirtyRect.y == 1);
		TEST_VERIFY(dirtyRect.width == areaPageWidth);
		TEST_VERIFY(dirtyRect.height == 1);
	}

	//Invalidate two first lines
	{
		CGsCachedArea area;
		area.SetArea(pixelFormat, 0, areaWidth, 512);

		area.Invalidate(0, areaPageWidth * CGsPixelFormats::PAGESIZE * 2);

		auto dirtyRect = area.GetDirtyPageRect();
		TEST_VERIFY(dirtyRect.x == 0);
		TEST_VERIFY(dirtyRect.y == 0);
		TEST_VERIFY(dirtyRect.width == areaPageWidth);
		TEST_VERIFY(dirtyRect.height == 2);
	}
}