77 lines
3.6 KiB
C++
77 lines
3.6 KiB
C++
// Copyright (C) 2022 Jérôme "Lynix" Leclercq (lynix680@gmail.com)
|
|
// This file is part of the "Nazara Engine - Shader module"
|
|
// For conditions of distribution and use, see copyright notice in Config.hpp
|
|
|
|
#pragma once
|
|
|
|
#ifndef NAZARA_SHADER_AST_ASTCONSTANTPROPAGATIONVISITOR_HPP
|
|
#define NAZARA_SHADER_AST_ASTCONSTANTPROPAGATIONVISITOR_HPP
|
|
|
|
#include <Nazara/Prerequisites.hpp>
|
|
#include <Nazara/Shader/Config.hpp>
|
|
#include <Nazara/Shader/Ast/AstCloner.hpp>
|
|
#include <Nazara/Shader/Ast/Module.hpp>
|
|
|
|
namespace Nz::ShaderAst
|
|
{
|
|
class NAZARA_SHADER_API AstConstantPropagationVisitor : public AstCloner
|
|
{
|
|
public:
|
|
struct Options;
|
|
|
|
AstConstantPropagationVisitor() = default;
|
|
AstConstantPropagationVisitor(const AstConstantPropagationVisitor&) = delete;
|
|
AstConstantPropagationVisitor(AstConstantPropagationVisitor&&) = delete;
|
|
~AstConstantPropagationVisitor() = default;
|
|
|
|
inline ExpressionPtr Process(Expression& expression);
|
|
inline ExpressionPtr Process(Expression& expression, const Options& options);
|
|
ModulePtr Process(const Module& shaderModule);
|
|
ModulePtr Process(const Module& shaderModule, const Options& options);
|
|
inline StatementPtr Process(Statement& statement);
|
|
inline StatementPtr Process(Statement& statement, const Options& options);
|
|
|
|
AstConstantPropagationVisitor& operator=(const AstConstantPropagationVisitor&) = delete;
|
|
AstConstantPropagationVisitor& operator=(AstConstantPropagationVisitor&&) = delete;
|
|
|
|
struct Options
|
|
{
|
|
std::function<const ConstantValue*(std::size_t constantId)> constantQueryCallback;
|
|
};
|
|
|
|
protected:
|
|
ExpressionPtr Clone(BinaryExpression& node) override;
|
|
ExpressionPtr Clone(CastExpression& node) override;
|
|
ExpressionPtr Clone(ConditionalExpression& node) override;
|
|
ExpressionPtr Clone(ConstantExpression& node) override;
|
|
ExpressionPtr Clone(SwizzleExpression& node) override;
|
|
ExpressionPtr Clone(UnaryExpression& node) override;
|
|
StatementPtr Clone(BranchStatement& node) override;
|
|
StatementPtr Clone(ConditionalStatement& node) override;
|
|
|
|
template<BinaryType Type> ExpressionPtr PropagateBinaryConstant(const ConstantValueExpression& lhs, const ConstantValueExpression& rhs);
|
|
template<typename TargetType> ExpressionPtr PropagateSingleValueCast(const ConstantValueExpression& operand);
|
|
template<std::size_t TargetComponentCount> ExpressionPtr PropagateConstantSwizzle(const std::array<UInt32, 4>& components, const ConstantValueExpression& operand);
|
|
template<UnaryType Type> ExpressionPtr PropagateUnaryConstant(const ConstantValueExpression& operand);
|
|
template<typename TargetType> ExpressionPtr PropagateVec2Cast(TargetType v1, TargetType v2);
|
|
template<typename TargetType> ExpressionPtr PropagateVec3Cast(TargetType v1, TargetType v2, TargetType v3);
|
|
template<typename TargetType> ExpressionPtr PropagateVec4Cast(TargetType v1, TargetType v2, TargetType v3, TargetType v4);
|
|
|
|
StatementPtr Unscope(StatementPtr node);
|
|
|
|
private:
|
|
Options m_options;
|
|
};
|
|
|
|
inline ExpressionPtr PropagateConstants(Expression& expr);
|
|
inline ExpressionPtr PropagateConstants(Expression& expr, const AstConstantPropagationVisitor::Options& options);
|
|
inline ModulePtr PropagateConstants(const Module& shaderModule);
|
|
inline ModulePtr PropagateConstants(const Module& shaderModule, const AstConstantPropagationVisitor::Options& options);
|
|
inline StatementPtr PropagateConstants(Statement& ast);
|
|
inline StatementPtr PropagateConstants(Statement& ast, const AstConstantPropagationVisitor::Options& options);
|
|
}
|
|
|
|
#include <Nazara/Shader/Ast/AstConstantPropagationVisitor.inl>
|
|
|
|
#endif // NAZARA_SHADER_AST_ASTCONSTANTPROPAGATIONVISITOR_HPP
|