// Copyright (C) 2008-2012 Christian Stehno, Colin MacDonald
// No rights reserved: this software is in the public domain.

#include "testUtils.h"

using namespace irr;

//! Tests rendering MRTs
static bool testWithDriver(video::E_DRIVER_TYPE driverType)
{
	IrrlichtDevice *device = createDevice (driverType, core::dimension2d < u32 > (220, 80));
	if (!device)
		return true; // No error if device does not exist

	video::IVideoDriver* driver = device->getVideoDriver();
	// We need at least GLSL 1.10 or HLSL 1.1
//	if (driver->getDriverAttributes().getAttributeAsInt("ShaderLanguageVersion")<=100)
//		return true;

	stabilizeScreenBackground(driver);

	logTestString("Testing driver %ls\n", driver->getName());

	// write white to first render-texture, green to second, blue to third
	const char* const psHLSL =
		"struct PS_INPUT\n"\
		"{\n"\
		"	float4 Position     : POSITION0;\n"\
		"};\n"\
		"struct PS_OUTPUT\n"\
		"{\n "\
		"	float4   Color      : COLOR0;\n"\
		"	float4   Normal     : COLOR1;\n"\
		"	float4   Depth      : COLOR2;\n"\
		"};\n"\
		"PS_OUTPUT pixelMain( PS_INPUT Input )\n"\
		"{\n"\
		"	PS_OUTPUT Output;\n"\
		"	Output.Color = float4(1.0,1.0,1.0,1.0);\n"\
		"	Output.Normal = float4(0.0,1.0,0.0,1.0);\n"\
		"	Output.Depth = float4(0.0,0.0,1.0,1.0);\n"\
		"	return Output;\n"\
		"}";

	const char* const psGLSL = 
		"void main(void)\n"\
		"{\n"\
		"	gl_FragData[0] = vec4(1.0,1.0,1.0,1.0);\n"\
		"	gl_FragData[1] = vec4(0.0,1.0,0.0,1.0);\n"\
		"	gl_FragData[2] = vec4(0.0,0.0,1.0,1.0);\n"\
		"}";

	// variable
	video::IRenderTarget* renderTarget = 0;
	core::array<video::ITexture*> renderTargetTex;
	video::ITexture* renderTargetDepth = 0;

	const core::dimension2du texsize(64,64);
	bool result=true;
	s32 newMaterialType = -1;

	if (device->getVideoDriver()->getDriverAttributes().getAttributeAsInt("MaxMultipleRenderTargets") > 2)
	{
		renderTargetTex.set_used(3);
		renderTargetTex[0] = driver->addRenderTargetTexture(texsize, "rta", video::ECF_A8R8G8B8);
		renderTargetTex[1] = driver->addRenderTargetTexture(texsize, "rtb", video::ECF_A8R8G8B8);
		renderTargetTex[2] = driver->addRenderTargetTexture(texsize, "rtc", video::ECF_A8R8G8B8);

		renderTargetDepth = driver->addRenderTargetTexture(texsize, "rtd", video::ECF_D16);

		renderTarget = driver->addRenderTarget();
		renderTarget->setTexture(renderTargetTex, renderTargetDepth);

		video::IGPUProgrammingServices* gpu = driver->getGPUProgrammingServices();

		if (gpu)
		{
			newMaterialType = gpu->addHighLevelShaderMaterial(
				0, "vertexMain", video::EVST_VS_1_1,
				driverType==video::EDT_DIRECT3D9?psHLSL:psGLSL, "pixelMain", video::EPST_PS_1_1);
		}
	}

	// shader creation succeeded
	if (newMaterialType!=-1)
	{
		scene::ISceneNode* node = device->getSceneManager()->addCubeSceneNode();
		node->setMaterialType((video::E_MATERIAL_TYPE)newMaterialType);
		device->getSceneManager()->addCameraSceneNode(0, core::vector3df(0,0,-10));

		driver->beginScene(video::ECBF_COLOR, video::SColor(255, 0, 0, 0));
		// render
		driver->setRenderTargetEx(renderTarget, video::ECBF_COLOR | video::ECBF_DEPTH, video::SColor(255,0,0,0));
		device->getSceneManager()->drawAll();
		driver->setRenderTargetEx(0, 0, video::SColor(255, 0, 0, 0));

		// draw debug rt
		driver->draw2DImage(renderTargetTex[0], core::position2d<s32>(0,0));
		driver->draw2DImage(renderTargetTex[1], core::position2d<s32>(64,0));
		driver->draw2DImage(renderTargetTex[2], core::position2d<s32>(128,0)); 

		driver->endScene();

		result = takeScreenshotAndCompareAgainstReference(driver, "-mrt.png");

		driver->beginScene(video::ECBF_COLOR, video::SColor(255, 0, 0, 0));
		// render
		device->getSceneManager()->getActiveCamera()->setPosition(core::vector3df(0,0,-15));
		driver->setRenderTargetEx(renderTarget, video::ECBF_COLOR | video::ECBF_DEPTH, video::SColor(255,0,0,0));
		device->getSceneManager()->drawAll();
		driver->setRenderTargetEx(0, 0, video::SColor(255,0,0,0));

		// draw debug rt
		driver->draw2DImage(renderTargetTex[0], core::position2d<s32>(0,0));
		driver->draw2DImage(renderTargetTex[1], core::position2d<s32>(64,0));
		driver->draw2DImage(renderTargetTex[2], core::position2d<s32>(128,0)); 

		driver->endScene();

		result &= takeScreenshotAndCompareAgainstReference(driver, "-mrt2.png");
	}

	device->closeDevice();
	device->run();
	device->drop();

	return result;
}


bool mrt(void)
{
	bool result = true;

	TestWithAllHWDrivers(testWithDriver);

	return result;
}