mirror of
https://github.com/libretro/RetroArch
synced 2024-12-29 12:31:05 +00:00
291 lines
6.8 KiB
C++
291 lines
6.8 KiB
C++
/* RetroArch - A frontend for libretro.
|
|
* Copyright (C) 2010-2016 - Hans-Kristian Arntzen
|
|
*
|
|
* RetroArch is free software: you can redistribute it and/or modify it under the terms
|
|
* of the GNU General Public License as published by the Free Software Found-
|
|
* ation, either version 3 of the License, or (at your option) any later version.
|
|
*
|
|
* RetroArch is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
|
|
* without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
* PURPOSE. See the GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along with RetroArch.
|
|
* If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <string>
|
|
#include <sstream>
|
|
|
|
#include <streams/file_stream.h>
|
|
#include <lists/string_list.h>
|
|
|
|
#include "glslang_util.hpp"
|
|
#include "glslang.hpp"
|
|
|
|
#include "../../general.h"
|
|
#include "../../libretro-common/include/file/file_path.h"
|
|
|
|
using namespace std;
|
|
|
|
static bool read_shader_file(const char *path, vector<string> *output)
|
|
{
|
|
char *buf = nullptr;
|
|
ssize_t len = 0;
|
|
struct string_list *list = NULL;
|
|
char include_path[PATH_MAX];
|
|
|
|
if (filestream_read_file(path, (void**)&buf, &len) < 0)
|
|
{
|
|
RARCH_ERR("Failed to open shader file: \"%s\".\n", path);
|
|
return false;
|
|
}
|
|
|
|
list = string_split(buf, "\n");
|
|
|
|
if (!list)
|
|
{
|
|
free(buf);
|
|
return false;
|
|
}
|
|
|
|
if (list->size == 0)
|
|
{
|
|
free(buf);
|
|
string_list_free(list);
|
|
return false;
|
|
}
|
|
|
|
for (size_t i = 0; i < list->size; i++)
|
|
{
|
|
const char *line = list->elems[i].data;
|
|
if (strstr(line, "#include ") == line)
|
|
{
|
|
char *c = (char*)strchr(line, '"');
|
|
if (!c)
|
|
{
|
|
RARCH_ERR("Invalid include statement \"%s\".\n", line);
|
|
return false;
|
|
}
|
|
c++;
|
|
char *closing = (char*)strchr(c, '"');
|
|
if (!closing)
|
|
{
|
|
RARCH_ERR("Invalid include statement \"%s\".\n", line);
|
|
return false;
|
|
}
|
|
*closing = '\0';
|
|
fill_pathname_resolve_relative(include_path, path, c, sizeof(include_path));
|
|
|
|
if (!read_shader_file(include_path, output))
|
|
return false;
|
|
}
|
|
else
|
|
output->push_back(line);
|
|
}
|
|
|
|
string_list_free(list);
|
|
return true;
|
|
}
|
|
|
|
static string build_stage_source(const vector<string> &lines, const char *stage)
|
|
{
|
|
ostringstream str;
|
|
bool active = true;
|
|
|
|
// Version header.
|
|
str << lines.front();
|
|
str << '\n';
|
|
|
|
for (auto itr = begin(lines) + 1; itr != end(lines); ++itr)
|
|
{
|
|
if (itr->find("#pragma stage ") == 0)
|
|
{
|
|
if (stage)
|
|
{
|
|
auto expected = string("#pragma stage ") + stage;
|
|
active = itr->find(expected) != string::npos;
|
|
}
|
|
|
|
// Improve debuggability.
|
|
if (active)
|
|
{
|
|
str << "#line ";
|
|
str << (itr - begin(lines)) + 2;
|
|
str << '\n';
|
|
}
|
|
}
|
|
else if (itr->find("#pragma name ") == 0 ||
|
|
itr->find("#pragma format ") == 0)
|
|
{
|
|
// Ignore
|
|
}
|
|
else if (active)
|
|
str << *itr;
|
|
str << '\n';
|
|
}
|
|
|
|
return str.str();
|
|
}
|
|
|
|
static const char *glslang_formats[] = {
|
|
"R8_UNORM",
|
|
"R8_UINT",
|
|
"R8_SINT",
|
|
"R8G8_UNORM",
|
|
"R8G8_UINT",
|
|
"R8G8_SINT",
|
|
"R8G8B8A8_UNORM",
|
|
"R8G8B8A8_UINT",
|
|
"R8G8B8A8_SINT",
|
|
"R8G8B8A8_SRGB",
|
|
|
|
"A2B10G10R10_UNORM_PACK32",
|
|
"A2B10G10R10_UINT_PACK32",
|
|
|
|
"R16_UINT",
|
|
"R16_SINT",
|
|
"R16_SFLOAT",
|
|
"R16G16_UINT",
|
|
"R16G16_SINT",
|
|
"R16G16_SFLOAT",
|
|
"R16G16B16A16_UINT",
|
|
"R16G16B16A16_SINT",
|
|
"R16G16B16A16_SFLOAT",
|
|
|
|
"R32_UINT",
|
|
"R32_SINT",
|
|
"R32_SFLOAT",
|
|
"R32G32_UINT",
|
|
"R32G32_SINT",
|
|
"R32G32_SFLOAT",
|
|
"R32G32B32A32_UINT",
|
|
"R32G32B32A32_SINT",
|
|
"R32G32B32A32_SFLOAT",
|
|
|
|
"UNKNOWN",
|
|
};
|
|
|
|
const char *glslang_format_to_string(enum glslang_format fmt)
|
|
{
|
|
return glslang_formats[fmt];
|
|
}
|
|
|
|
static glslang_format glslang_find_format(const char *fmt)
|
|
{
|
|
#undef FMT
|
|
#define FMT(x) if (!strcmp(fmt, #x)) return SLANG_FORMAT_ ## x
|
|
FMT(R8_UNORM);
|
|
FMT(R8_UINT);
|
|
FMT(R8_SINT);
|
|
FMT(R8G8_UNORM);
|
|
FMT(R8G8_UINT);
|
|
FMT(R8G8_SINT);
|
|
FMT(R8G8B8A8_UNORM);
|
|
FMT(R8G8B8A8_UINT);
|
|
FMT(R8G8B8A8_SINT);
|
|
FMT(R8G8B8A8_SRGB);
|
|
|
|
FMT(A2B10G10R10_UNORM_PACK32);
|
|
FMT(A2B10G10R10_UINT_PACK32);
|
|
|
|
FMT(R16_UINT);
|
|
FMT(R16_SINT);
|
|
FMT(R16_SFLOAT);
|
|
FMT(R16G16_UINT);
|
|
FMT(R16G16_SINT);
|
|
FMT(R16G16_SFLOAT);
|
|
FMT(R16G16B16A16_UINT);
|
|
FMT(R16G16B16A16_SINT);
|
|
FMT(R16G16B16A16_SFLOAT);
|
|
|
|
FMT(R32_UINT);
|
|
FMT(R32_SINT);
|
|
FMT(R32_SFLOAT);
|
|
FMT(R32G32_UINT);
|
|
FMT(R32G32_SINT);
|
|
FMT(R32G32_SFLOAT);
|
|
FMT(R32G32B32A32_UINT);
|
|
FMT(R32G32B32A32_SINT);
|
|
FMT(R32G32B32A32_SFLOAT);
|
|
|
|
return SLANG_FORMAT_UNKNOWN;
|
|
}
|
|
|
|
static bool glslang_parse_meta(const vector<string> &lines, glslang_meta *meta)
|
|
{
|
|
*meta = glslang_meta{};
|
|
for (auto &line : lines)
|
|
{
|
|
if (line.find("#pragma name ") == 0)
|
|
{
|
|
if (!meta->name.empty())
|
|
{
|
|
RARCH_ERR("[slang]: Trying to declare multiple names for file.\n");
|
|
return false;
|
|
}
|
|
|
|
const char *str = line.c_str() + strlen("#pragma name ");
|
|
while (*str == ' ')
|
|
str++;
|
|
meta->name = str;
|
|
}
|
|
else if (line.find("#pragma format ") == 0)
|
|
{
|
|
if (meta->rt_format != SLANG_FORMAT_UNKNOWN)
|
|
{
|
|
RARCH_ERR("[slang]: Trying to declare format multiple times for file.\n");
|
|
return false;
|
|
}
|
|
|
|
const char *str = line.c_str() + strlen("#pragma format ");
|
|
while (*str == ' ')
|
|
str++;
|
|
|
|
meta->rt_format = glslang_find_format(str);
|
|
if (meta->rt_format == SLANG_FORMAT_UNKNOWN)
|
|
{
|
|
RARCH_ERR("[slang]: Failed to find format \"%s\".\n", str);
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool glslang_compile_shader(const char *shader_path, glslang_output *output)
|
|
{
|
|
vector<string> lines;
|
|
|
|
RARCH_LOG("[slang]: Compiling shader \"%s\".\n", shader_path);
|
|
if (!read_shader_file(shader_path, &lines))
|
|
return false;
|
|
|
|
if (!glslang_parse_meta(lines, &output->meta))
|
|
return false;
|
|
|
|
auto &header = lines.front();
|
|
if (header.find_first_of("#version ") != 0)
|
|
{
|
|
RARCH_ERR("First line of the shader must contain a valid #version string.\n");
|
|
return false;
|
|
}
|
|
|
|
if (!glslang::compile_spirv(build_stage_source(lines, "vertex"),
|
|
glslang::StageVertex, &output->vertex))
|
|
{
|
|
RARCH_ERR("Failed to compile vertex shader stage.\n");
|
|
return false;
|
|
}
|
|
|
|
if (!glslang::compile_spirv(build_stage_source(lines, "fragment"),
|
|
glslang::StageFragment, &output->fragment))
|
|
{
|
|
RARCH_ERR("Failed to compile fragment shader stage.\n");
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|