StructParameterAnalyzer.hpp

Go to the documentation of this file.

//
//  CeresEngine - A game development framework
//
//  Created by Rogiel Sulzbach.
//  Copyright (c) 2018-2022 Rogiel Sulzbach. All rights reserved.
//
 
#pragma once
 
#include "VisitorTracker.hpp"
 
#include "CeresEngine/Foundation/Container/Set.hpp"
 
#include "CeresEngine/Material/ShaderCompiler/Targets.hpp"
 
namespace CeresEngine::ShaderCompiler {
 
    class StructParameterAnalyzer : private VisitorTracker {
    private:
        Program* mProgram = nullptr;
        ShaderTarget mShaderTarget = ShaderTarget::VertexShader;
 
        Set<const AST*> mVisitSet;
 
    public:
        // Marks all declarational AST nodes (i.e. function decl, structure decl etc.) that are reachable from the specififed entry point.
        void markStructsFromEntryPoint(Program& program, const ShaderTarget shaderTarget);
 
    private:
        // Returns true if the specified AST has not yet been visited.
        bool notVisited(const AST* ast);
 
        void visitStatementList(const Vector<StatementPtr>& statements);
 
        // Returns true if the specified variable is a paramter of the entry point.
        bool isVariableAnEntryPointParameter(VarDeclarationStatement* var) const;
 
        // Returns true if the active function declaration is the main entry point.
        [[nodiscard]] bool isActiveFunctionDeclarationEntryPoint() const;
 
    private: // Visitor implementation
        DECLARATION_VISIT_PROC(CodeBlock);
        DECLARATION_VISIT_PROC(SwitchCase);
        DECLARATION_VISIT_PROC(TypeSpecifier);
 
        DECLARATION_VISIT_PROC(VarDeclaration);
        DECLARATION_VISIT_PROC(StructDeclaration);
        DECLARATION_VISIT_PROC(BufferDeclaration);
 
        DECLARATION_VISIT_PROC(FunctionDeclaration);
        DECLARATION_VISIT_PROC(UniformBufferDeclaration);
        DECLARATION_VISIT_PROC(BufferDeclarationStatement);
 
        DECLARATION_VISIT_PROC(CallExpression);
        DECLARATION_VISIT_PROC(ObjectExpression);
    };
 
} // namespace CeresEngine::ShaderCompiler