#include "BsGpuProgramImporter.h"
#include "BsPath.h"
#include "BsDataStream.h"
#include "BsFileSystem.h"
#include "BsGpuProgramImportOptions.h"
#include "BsGpuProgram.h"
#include "BsDebug.h"

namespace BansheeEngine
{
	bool GpuProgramImporter::isExtensionSupported(const WString& ext) const
	{
		if(ext == L"gpuprog")
			return true;

		return false;
	}

	bool GpuProgramImporter::isMagicNumberSupported(const UINT8* magicNumPtr, UINT32 numBytes) const
	{
		// Gpu program files are plain-text so no magic number
		return true;
	}

	ResourcePtr GpuProgramImporter::import(const Path& filePath, ConstImportOptionsPtr importOptions)
	{
		WString ext = filePath.getWExtension();
		ext = ext.substr(1, ext.size() - 1); // Remove the .

		DataStreamPtr stream = FileSystem::openFile(filePath);
		String shaderSource = stream->getAsString();

		const GpuProgramImportOptions* gpuProgImportOptions = static_cast<const GpuProgramImportOptions*>(importOptions.get());

		String entryPoint = gpuProgImportOptions->getEntryPoint();
		GpuProgramProfile profile = gpuProgImportOptions->getProfile();
		String language = gpuProgImportOptions->getLanguage();
		GpuProgramType gptype = gpuProgImportOptions->getType();
		Vector<HGpuProgInclude> includes = gpuProgImportOptions->getIncludes();

		GpuProgramPtr gpuProgram = GpuProgram::_createPtr(shaderSource, entryPoint, language, gptype, profile, &includes);
		gpuProgram->synchronize();

		if (!gpuProgram->isCompiled())
		{
			LOGERR("Failed compiling GPU program: " + filePath.toString() + ". Error: " + gpuProgram->getCompileErrorMessage());
			// TODO - Return some dummy program here?
		}

		WString fileName = filePath.getWFilename(false);
		gpuProgram->setName(toString(fileName));

		return gpuProgram;
	}

	ImportOptionsPtr GpuProgramImporter::createImportOptions() const
	{
		return bs_shared_ptr<GpuProgramImportOptions, PoolAlloc>();
	}
}