external/vulkancts/framework/vulkan/vkSpirVAsm.cpp

/*-------------------------------------------------------------------------
 * Vulkan CTS Framework
 * --------------------
 *
 * Copyright (c) 2015 Google Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and/or associated documentation files (the
 * "Materials"), to deal in the Materials without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of the Materials, and to
 * permit persons to whom the Materials are furnished to do so, subject to
 * the following conditions:
 *
 * The above copyright notice(s) and this permission notice shall be
 * included in all copies or substantial portions of the Materials.
 *
 * THE MATERIALS ARE PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
 * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * MATERIALS OR THE USE OR OTHER DEALINGS IN THE MATERIALS.
 *
 *//*!
 * \file
 * \brief SPIR-V assembly to binary.
 *//*--------------------------------------------------------------------*/

#include "vkSpirVAsm.hpp"
#include "vkSpirVProgram.hpp"
#include "deArrayUtil.hpp"
#include "deMemory.h"
#include "deClock.h"
#include "qpDebugOut.h"

#if defined(DEQP_HAVE_SPIRV_TOOLS)
#	include "libspirv/libspirv.h"
#endif

namespace vk
{

using std::string;
using std::vector;

#if defined(DEQP_HAVE_SPIRV_TOOLS)


void assembleSpirV (const SpirVAsmSource* program, std::vector<deUint8>* dst, SpirVProgramInfo* buildInfo)
{
	spv_context context = spvContextCreate();

	const std::string&	spvSource			= program->program.str();
	spv_binary			binary				= DE_NULL;
	spv_diagnostic		diagnostic			= DE_NULL;
	const deUint64		compileStartTime	= deGetMicroseconds();
	const spv_result_t	compileOk			= spvTextToBinary(context, spvSource.c_str(), spvSource.size(), &binary, &diagnostic);

	{
		buildInfo->source			= program;
		buildInfo->infoLog			= diagnostic? diagnostic->error : ""; // \todo [2015-07-13 pyry] Include debug log?
		buildInfo->compileTimeUs	= deGetMicroseconds() - compileStartTime;
		buildInfo->compileOk		= (compileOk == SPV_SUCCESS);
	}

	if (compileOk != SPV_SUCCESS)
		TCU_FAIL("Failed to compile shader");

	dst->resize((int)binary->wordCount * sizeof(deUint32));
#if (DE_ENDIANNESS == DE_LITTLE_ENDIAN)
	deMemcpy(&(*dst)[0], &binary->code[0], dst->size());
#else
#	error "Big-endian not supported"
#endif
	spvBinaryDestroy(binary);
	spvDiagnosticDestroy(diagnostic);
	spvContextDestroy(context);
	return;
}

bool validateSpirV (const std::vector<deUint8>& spirv, std::string* infoLog)
{
	const size_t bytesPerWord = sizeof(uint32_t) / sizeof(deUint8);
	DE_ASSERT(spirv.size() % bytesPerWord == 0);
	std::vector<uint32_t> words(spirv.size() / bytesPerWord);
	deMemcpy(words.data(), spirv.data(), spirv.size());
	spv_const_binary_t	cbinary		= { words.data(), words.size() };
	spv_diagnostic		diagnostic	= DE_NULL;
	spv_context			context		= spvContextCreate();
	const spv_result_t	valid		= spvValidate(context, &cbinary, SPV_VALIDATE_ALL, &diagnostic);
	if (diagnostic)
		*infoLog += diagnostic->error;
	spvContextDestroy(context);
	spvDiagnosticDestroy(diagnostic);
	return valid == SPV_SUCCESS;
}

#else // defined(DEQP_HAVE_SPIRV_TOOLS)

void assembleSpirV (const SpirVAsmSource*, std::vector<deUint8>*, SpirVProgramInfo*)
{
	TCU_THROW(NotSupportedError, "SPIR-V assembly not supported (DEQP_HAVE_SPIRV_TOOLS not defined)");
}

bool validateSpirV (std::vector<deUint8>*, std::string*)
{
	TCU_THROW(NotSupportedError, "SPIR-V validation not supported (DEQP_HAVE_SPIRV_TOOLS not defined)");
}

#endif

} // vk