65 lines
2.1 KiB
C++
65 lines
2.1 KiB
C++
// Copyright (c) Wojciech Figat. All rights reserved.
|
|
|
|
#include "ImportShader.h"
|
|
|
|
#if COMPILE_WITH_ASSETS_IMPORTER
|
|
|
|
#include "Engine/Core/Log.h"
|
|
#include "Engine/Platform/File.h"
|
|
#include "Engine/Graphics/Shaders/Cache/ShaderStorage.h"
|
|
#include "Engine/Graphics/Shaders/Cache/ShaderCacheManager.h"
|
|
#include "Engine/Utilities/Encryption.h"
|
|
#include "Engine/Content/Assets/Shader.h"
|
|
|
|
CreateAssetResult ImportShader::Import(CreateAssetContext& context)
|
|
{
|
|
// Base
|
|
IMPORT_SETUP(Shader, 20);
|
|
const int32 SourceCodeChunk = 15;
|
|
context.SkipMetadata = true;
|
|
|
|
// Read text (handles any Unicode convert into ANSI)
|
|
StringAnsi sourceCodeText;
|
|
if (File::ReadAllText(context.InputPath, sourceCodeText))
|
|
return CreateAssetResult::InvalidPath;
|
|
|
|
// Load source code
|
|
if (context.AllocateChunk(SourceCodeChunk))
|
|
return CreateAssetResult::CannotAllocateChunk;
|
|
const auto sourceCodeSize = sourceCodeText.Length();
|
|
if (sourceCodeSize < 10)
|
|
{
|
|
LOG(Warning, "Empty shader source file.");
|
|
return CreateAssetResult::Error;
|
|
}
|
|
|
|
// Ensure the source code has an empty line at the end (expected by glslang)
|
|
auto sourceCodeChunkSize = sourceCodeSize + 1;
|
|
if (sourceCodeText[sourceCodeSize - 1] != '\n')
|
|
sourceCodeChunkSize++;
|
|
|
|
const auto& sourceCodeChunk = context.Data.Header.Chunks[SourceCodeChunk];
|
|
sourceCodeChunk->Data.Allocate(sourceCodeChunkSize);
|
|
const auto sourceCode = sourceCodeChunk->Get();
|
|
Platform::MemoryCopy(sourceCode, sourceCodeText.Get(), sourceCodeSize);
|
|
sourceCode[sourceCodeChunkSize - 2] = '\n';
|
|
|
|
// Encrypt source code
|
|
Encryption::EncryptBytes(sourceCode, sourceCodeChunkSize - 1);
|
|
sourceCode[sourceCodeChunkSize - 1] = 0;
|
|
|
|
// Set Custom Data with Header
|
|
ShaderStorage::Header20 shaderHeader;
|
|
Platform::MemoryClear(&shaderHeader, sizeof(shaderHeader));
|
|
context.Data.CustomData.Copy(&shaderHeader);
|
|
|
|
#if COMPILE_WITH_SHADER_CACHE_MANAGER
|
|
// Invalidate shader cache
|
|
ShaderCacheManager::RemoveCache(context.Data.Header.ID);
|
|
#endif
|
|
|
|
return CreateAssetResult::Ok;
|
|
}
|
|
|
|
#endif
|