63 lines
1.5 KiB
C++
63 lines
1.5 KiB
C++
// Copyright (C) 2020 Jérôme Leclercq
|
|
// This file is part of the "Nazara Engine - Shader generator"
|
|
// For conditions of distribution and use, see copyright notice in Config.hpp
|
|
|
|
#include <Nazara/Shader/Ast/TransformVisitor.hpp>
|
|
#include <Nazara/Shader/Debug.hpp>
|
|
|
|
namespace Nz::ShaderAst
|
|
{
|
|
inline TransformVisitor::TransformVisitor() :
|
|
m_nextFuncIndex(0),
|
|
m_nextVarIndex(0)
|
|
{
|
|
}
|
|
|
|
inline auto TransformVisitor::FindIdentifier(const std::string_view& identifierName) const -> const Identifier*
|
|
{
|
|
auto it = std::find_if(m_identifiersInScope.rbegin(), m_identifiersInScope.rend(), [&](const Identifier& identifier) { return identifier.name == identifierName; });
|
|
if (it == m_identifiersInScope.rend())
|
|
return nullptr;
|
|
|
|
return &*it;
|
|
}
|
|
|
|
inline std::size_t TransformVisitor::RegisterFunction(std::string name)
|
|
{
|
|
std::size_t funcIndex = m_nextFuncIndex++;
|
|
return funcIndex;
|
|
}
|
|
|
|
|
|
inline std::size_t TransformVisitor::RegisterStruct(std::string name, StructDescription description)
|
|
{
|
|
std::size_t structIndex = m_structs.size();
|
|
m_structs.emplace_back(std::move(description));
|
|
|
|
m_identifiersInScope.push_back({
|
|
std::move(name),
|
|
Struct {
|
|
structIndex
|
|
}
|
|
});
|
|
|
|
return structIndex;
|
|
}
|
|
|
|
inline std::size_t TransformVisitor::RegisterVariable(std::string name)
|
|
{
|
|
std::size_t varIndex = m_nextVarIndex++;
|
|
|
|
m_identifiersInScope.push_back({
|
|
std::move(name),
|
|
Variable {
|
|
varIndex
|
|
}
|
|
});
|
|
|
|
return varIndex;
|
|
}
|
|
}
|
|
|
|
#include <Nazara/Shader/DebugOff.hpp>
|