💿🐜 Antkeeper source code
https://antkeeper.com
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
216 lines
6.1 KiB
216 lines
6.1 KiB
/*
|
|
* Copyright (C) 2021 Christopher J. Howard
|
|
*
|
|
* This file is part of Antkeeper source code.
|
|
*
|
|
* Antkeeper source code 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 Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Antkeeper source code 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 Antkeeper source code. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "resources/resource-loader.hpp"
|
|
#include "resources/resource-manager.hpp"
|
|
#include "resources/text-file.hpp"
|
|
#include "gl/shader-stage.hpp"
|
|
#include "gl/shader-object.hpp"
|
|
#include "gl/shader-program.hpp"
|
|
#include <sstream>
|
|
#include <unordered_set>
|
|
#include <iterator>
|
|
|
|
/**
|
|
* Tokenizes a line of shader source code.
|
|
*/
|
|
static std::vector<std::string> tokenize(const std::string& line)
|
|
{
|
|
std::vector<std::string> tokens;
|
|
std::string token;
|
|
std::istringstream linestream(line);
|
|
while (linestream >> token)
|
|
tokens.push_back(token);
|
|
|
|
return tokens;
|
|
}
|
|
|
|
/**
|
|
* Handles `#pragma include` directives by loading the specified text files and inserting them in place.
|
|
*/
|
|
static void handle_includes(text_file* source, resource_manager* resource_manager)
|
|
{
|
|
// For each line in the source
|
|
for (std::size_t i = 0; i < source->size(); ++i)
|
|
{
|
|
// Tokenize line
|
|
std::vector<std::string> tokens = tokenize((*source)[i]);
|
|
|
|
// Look for `#pragma include` directives
|
|
if (tokens.size() == 3 && tokens[0] == "#pragma" && tokens[1] == "include")
|
|
{
|
|
// Get path to include file
|
|
std::string path = tokens[2].substr(1, tokens[2].length() - 2);
|
|
|
|
// Load include file
|
|
if (!resource_manager->load<text_file>(path))
|
|
{
|
|
std::string message = std::string("Failed to load shader include file \"") + path + std::string("\"");
|
|
throw std::runtime_error(message.c_str());
|
|
}
|
|
text_file include_file = *(resource_manager->load<text_file>(path));
|
|
|
|
// Handle `#pragma include` directives inside include file
|
|
handle_includes(&include_file, resource_manager);
|
|
|
|
// Replace #pragma include directive with include file contents
|
|
source->erase(source->begin() + i);
|
|
source->insert(source->begin() + i, include_file.begin(), include_file.end());
|
|
i += include_file.size() - 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Injects the `DEBUG` or `NDEBUG` macros after the `#version` directive.
|
|
*/
|
|
static void inject_debug_macro(text_file* source)
|
|
{
|
|
// For each line in the source
|
|
for (std::size_t i = 0; i < source->size(); ++i)
|
|
{
|
|
// Tokenize line
|
|
std::vector<std::string> tokens = tokenize((*source)[i]);
|
|
|
|
// Inject DEBUG and NDEBUG macros
|
|
if (!tokens.empty() && tokens[0] == "#version")
|
|
{
|
|
#if defined(NDEBUG)
|
|
source->insert(source->begin() + i + 1, "#define NDEBUG");
|
|
#else
|
|
source->insert(source->begin() + i + 1, "#define DEBUG");
|
|
#endif
|
|
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Injects a shader type macro definition after the `#version` directive.
|
|
*/
|
|
static void inject_shader_stage_macro(text_file* source, gl::shader_stage stage)
|
|
{
|
|
const char* vertex_macro = "#define _VERTEX";
|
|
const char* fragment_macro = "#define _FRAGMENT";
|
|
const char* geometry_macro = "#define _GEOMETRY";
|
|
|
|
const char* macro = nullptr;
|
|
if (stage == gl::shader_stage::vertex)
|
|
macro = vertex_macro;
|
|
else if (stage == gl::shader_stage::fragment)
|
|
macro = fragment_macro;
|
|
else if (stage == gl::shader_stage::geometry)
|
|
macro = geometry_macro;
|
|
|
|
// For each line in the source
|
|
for (std::size_t i = 0; i < source->size(); ++i)
|
|
{
|
|
// Tokenize line
|
|
std::vector<std::string> tokens = tokenize((*source)[i]);
|
|
|
|
// Inject shader stage macro
|
|
if (!tokens.empty() && tokens[0] == "#version")
|
|
{
|
|
source->insert(source->begin() + i + 1, macro);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
static std::unordered_set<gl::shader_stage> detect_shader_stages(text_file* source)
|
|
{
|
|
std::unordered_set<gl::shader_stage> types;
|
|
|
|
// For each line in the source
|
|
for (std::size_t i = 0; i < source->size(); ++i)
|
|
{
|
|
// Tokenize line
|
|
std::vector<std::string> tokens = tokenize((*source)[i]);
|
|
|
|
// Look for `#pragma include` directives
|
|
if (tokens.size() == 2 && tokens[0] == "#pragma")
|
|
{
|
|
if (tokens[1] == "vertex")
|
|
{
|
|
types.insert(gl::shader_stage::vertex);
|
|
}
|
|
else if (tokens[1] == "fragment")
|
|
{
|
|
types.insert(gl::shader_stage::fragment);
|
|
}
|
|
else if (tokens[1] == "geometry")
|
|
{
|
|
types.insert(gl::shader_stage::geometry);
|
|
}
|
|
}
|
|
}
|
|
|
|
return types;
|
|
}
|
|
|
|
static std::string generate_source_buffer(const std::vector<std::string>& source)
|
|
{
|
|
std::ostringstream stream;
|
|
std::copy(source.begin(), source.end(), std::ostream_iterator<std::string>(stream, "\n"));
|
|
return stream.str();
|
|
}
|
|
|
|
template <>
|
|
gl::shader_program* resource_loader<gl::shader_program>::load(resource_manager* resource_manager, PHYSFS_File* file)
|
|
{
|
|
// Load shader source
|
|
text_file* source = resource_loader<text_file>::load(resource_manager, file);
|
|
|
|
// Handle `#pragma include` directives
|
|
handle_includes(source, resource_manager);
|
|
|
|
// Inject `DEBUG` or `NDEBUG` macro definitions
|
|
inject_debug_macro(source);
|
|
|
|
// Detect declared shader types via the `#pragma vertex`, `#pragma fragment` and `#pragma geometry` directives
|
|
std::unordered_set<gl::shader_stage> shade_stages = detect_shader_stages(source);
|
|
|
|
// Load detected shaders
|
|
std::list<gl::shader_object*> shaders;
|
|
for (gl::shader_stage stage: shade_stages)
|
|
{
|
|
text_file stage_source = *source;
|
|
inject_shader_stage_macro(&stage_source, stage);
|
|
std::string source_buffer = generate_source_buffer(stage_source);
|
|
|
|
gl::shader_object* object = new gl::shader_object(stage);
|
|
object->source(source_buffer.c_str(), source_buffer.length());
|
|
object->compile();
|
|
|
|
shaders.push_back(object);
|
|
}
|
|
|
|
// Create shader program
|
|
gl::shader_program* program = new gl::shader_program(shaders);
|
|
|
|
// Delete shaders objects
|
|
for (gl::shader_object* shader: shaders)
|
|
{
|
|
delete shader;
|
|
}
|
|
|
|
return program;
|
|
}
|
|
|