58 lines
2.2 KiB
C++
58 lines
2.2 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
|
|
|
|
#pragma once
|
|
|
|
#ifndef NAZARA_SHADERASTOPTIMISER_HPP
|
|
#define NAZARA_SHADERASTOPTIMISER_HPP
|
|
|
|
#include <Nazara/Prerequisites.hpp>
|
|
#include <Nazara/Shader/Config.hpp>
|
|
#include <Nazara/Shader/Ast/AstCloner.hpp>
|
|
#include <optional>
|
|
#include <vector>
|
|
|
|
namespace Nz::ShaderAst
|
|
{
|
|
class NAZARA_SHADER_API AstOptimizer : public AstCloner
|
|
{
|
|
public:
|
|
AstOptimizer() = default;
|
|
AstOptimizer(const AstOptimizer&) = delete;
|
|
AstOptimizer(AstOptimizer&&) = delete;
|
|
~AstOptimizer() = default;
|
|
|
|
StatementPtr Optimise(Statement& statement);
|
|
StatementPtr Optimise(Statement& statement, UInt64 enabledConditions);
|
|
|
|
AstOptimizer& operator=(const AstOptimizer&) = delete;
|
|
AstOptimizer& operator=(AstOptimizer&&) = delete;
|
|
|
|
protected:
|
|
ExpressionPtr Clone(BinaryExpression& node) override;
|
|
ExpressionPtr Clone(CastExpression& node) override;
|
|
ExpressionPtr Clone(ConditionalExpression& node) override;
|
|
ExpressionPtr Clone(UnaryExpression& node) override;
|
|
StatementPtr Clone(BranchStatement& node) override;
|
|
StatementPtr Clone(ConditionalStatement& node) override;
|
|
|
|
template<BinaryType Type> ExpressionPtr PropagateBinaryConstant(std::unique_ptr<ConstantExpression>&& lhs, std::unique_ptr<ConstantExpression>&& rhs);
|
|
template<typename TargetType> ExpressionPtr PropagateSingleValueCast(std::unique_ptr<ConstantExpression>&& operand);
|
|
template<UnaryType Type> ExpressionPtr PropagateUnaryConstant(std::unique_ptr<ConstantExpression>&& 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);
|
|
|
|
private:
|
|
std::optional<UInt64> m_enabledOptions;
|
|
};
|
|
|
|
inline StatementPtr Optimize(Statement& ast);
|
|
inline StatementPtr Optimize(Statement& ast, UInt64 enabledConditions);
|
|
}
|
|
|
|
#include <Nazara/Shader/Ast/AstOptimizer.inl>
|
|
|
|
#endif
|