#include <gtest/gtest.h>
#include "Bullet3Common/b3Logging.h"

#include "Bullet3OpenCL/Initialize/b3OpenCLUtils.h"
TEST(b3OpenCLUtils, getNumPlatforms)
{
	int numPlatforms = b3OpenCLUtils::getNumPlatforms();
	ASSERT_GT(numPlatforms, 0);
}

TEST(b3OpenCLUtils, getSdkVendorName)
{
	const char* vendorSDK = b3OpenCLUtils::getSdkVendorName();
	b3Printf("getSdkVendorName=%s\n", vendorSDK);
	ASSERT_FALSE(vendorSDK == NULL);
}

TEST(b3OpenCLUtils, getPlatformInfo)
{
	int numPlatforms = b3OpenCLUtils::getNumPlatforms();
	ASSERT_GT(numPlatforms, 0);

	b3Printf("Num Platforms = %d\n", numPlatforms);
	for (int i = 0; i < numPlatforms; i++)
	{
		cl_platform_id platform = b3OpenCLUtils::getPlatform(i);
		ASSERT_FALSE(platform == NULL);

		b3OpenCLPlatformInfo platformInfo;
		b3OpenCLUtils::getPlatformInfo(platform, &platformInfo);
		ASSERT_FALSE(platformInfo.m_platformName == NULL);
		ASSERT_FALSE(platformInfo.m_platformVendor == NULL);
		ASSERT_FALSE(platformInfo.m_platformVersion == NULL);
	}
}

TEST(b3OpenCLUtils, createContextFromPlatform)
{
	int numPlatforms = b3OpenCLUtils::getNumPlatforms();
	b3Printf("Num Platforms = %d\n", numPlatforms);

	cl_device_type deviceType = CL_DEVICE_TYPE_ALL;
	int ciErrNum = 0;

	for (int i = 0; i < numPlatforms; i++)
	{
		cl_platform_id platform = b3OpenCLUtils::getPlatform(i);
		b3OpenCLPlatformInfo platformInfo;
		b3OpenCLUtils::getPlatformInfo(platform, &platformInfo);
		b3Printf("--------------------------------\n");
		b3Printf("Platform info for platform nr %d:\n", i);
		b3Printf("  CL_PLATFORM_VENDOR: \t\t\t%s\n", platformInfo.m_platformVendor);
		b3Printf("  CL_PLATFORM_NAME: \t\t\t%s\n", platformInfo.m_platformName);
		b3Printf("  CL_PLATFORM_VERSION: \t\t\t%s\n", platformInfo.m_platformVersion);

		cl_context ctx = b3OpenCLUtils::createContextFromPlatform(platform, deviceType, &ciErrNum);
		ASSERT_FALSE(ctx == 0);
		ASSERT_EQ(CL_SUCCESS, ciErrNum);
		clReleaseContext(ctx);
	}
}

TEST(b3OpenCLUtils, getDeviceAndQueue)
{
	int numPlatforms = b3OpenCLUtils::getNumPlatforms();
	b3Printf("Num Platforms = %d\n", numPlatforms);
	cl_device_type deviceType = CL_DEVICE_TYPE_ALL;
	int ciErrNum = 0;
	for (int i = 0; i < numPlatforms; i++)
	{
		cl_platform_id platform = b3OpenCLUtils::getPlatform(i);
		b3OpenCLPlatformInfo platformInfo;
		b3OpenCLUtils::getPlatformInfo(platform, &platformInfo);
		b3Printf("--------------------------------\n");
		b3Printf("Platform info for platform nr %d:\n", i);
		b3Printf("  CL_PLATFORM_VENDOR: \t\t\t%s\n", platformInfo.m_platformVendor);
		b3Printf("  CL_PLATFORM_NAME: \t\t\t%s\n", platformInfo.m_platformName);
		b3Printf("  CL_PLATFORM_VERSION: \t\t\t%s\n", platformInfo.m_platformVersion);
		cl_context ctx = b3OpenCLUtils::createContextFromPlatform(platform, deviceType, &ciErrNum);
		ASSERT_FALSE(ctx == 0);
		ASSERT_EQ(CL_SUCCESS, ciErrNum);
		int numDevices = b3OpenCLUtils::getNumDevices(ctx);
		ASSERT_GT(numDevices, 0);

		b3Printf("Num Devices = %d\n", numDevices);
		for (int j = 0; j < numDevices; j++)
		{
			cl_device_id device = b3OpenCLUtils::getDevice(ctx, j);
			b3OpenCLDeviceInfo devInfo;
			b3OpenCLUtils::getDeviceInfo(device, &devInfo);
			ASSERT_GT(devInfo.m_clockFrequency, 0);
			ASSERT_GT(devInfo.m_addressBits, 0);
			ASSERT_GT(devInfo.m_computeUnits, 0);
			ASSERT_GT(devInfo.m_constantBufferSize, 0);
			ASSERT_FALSE(devInfo.m_deviceName == NULL);
			ASSERT_FALSE(devInfo.m_deviceVendor == NULL);
			ASSERT_FALSE(devInfo.m_driverVersion == NULL);
			ASSERT_GT(devInfo.m_globalMemSize, 0);

			b3OpenCLUtils::printDeviceInfo(device);

			cl_command_queue q = clCreateCommandQueue(ctx, device, 0, &ciErrNum);
			ASSERT_FALSE(q == 0);

			clReleaseCommandQueue(q);
			q = 0;
		}
		clReleaseContext(ctx);
	}
}