SlangShaderGenerator.h

Go to the documentation of this file.

//
// Copyright Contributors to the MaterialX Project
// SPDX-License-Identifier: Apache-2.0
//
 
#ifndef MATERIALX_SLANGSHADERGENERATOR_H
#define MATERIALX_SLANGSHADERGENERATOR_H

 
#include <MaterialXGenSlang/Export.h>
 
#include <MaterialXGenHw/HwShaderGenerator.h>
 
MATERIALX_NAMESPACE_BEGIN
 
using SlangShaderGeneratorPtr = shared_ptr<class SlangShaderGenerator>;

class MX_GENSLANG_API SlangShaderGenerator : public HwShaderGenerator
{
  public:
    SlangShaderGenerator(TypeSystemPtr typeSystem);

    static ShaderGeneratorPtr create(TypeSystemPtr typeSystem = nullptr)
    {
        return std::make_shared<SlangShaderGenerator>(typeSystem ? typeSystem : TypeSystem::create());
    }

    ShaderPtr generate(const string& name, ElementPtr element, GenContext& context) const override;

    const string& getTarget() const override { return TARGET; }

    virtual const string& getVersion() const { return VERSION; }

    void emitVariableDeclaration(const ShaderPort* variable, const string& qualifier, GenContext& context, ShaderStage& stage,
                                 bool assignValue = true) const override;

    string getVertexDataPrefix(const VariableBlock& vertexData) const override;
 
  public:
    static const string TARGET;

    static const string VERSION;
 
  protected:
    virtual void emitVertexStage(const ShaderGraph& graph, GenContext& context, ShaderStage& stage) const;
    virtual void emitPixelStage(const ShaderGraph& graph, GenContext& context, ShaderStage& stage) const;
 
    virtual void emitDirectives(GenContext& context, ShaderStage& stage) const;
    virtual void emitConstants(GenContext& context, ShaderStage& stage) const;
    virtual void emitUniforms(GenContext& context, ShaderStage& stage, bool emitLighting) const;
    virtual void emitInputs(GenContext& context, ShaderStage& stage) const;
    virtual void emitOutputs(GenContext& context, ShaderStage& stage) const;

    virtual bool requiresLighting(const ShaderGraph& graph) const override;

    virtual void emitSpecularEnvironment(GenContext& context, ShaderStage& stage) const;

    virtual void emitTransmissionRender(GenContext& context, ShaderStage& stage) const;

    virtual void emitLightFunctionDefinitions(const ShaderGraph& graph, GenContext& context, ShaderStage& stage) const;
 
    static void toVec4(TypeDesc type, string& variable);
    [[deprecated]] static void toVec4(const TypeDesc* type, string& variable) { toVec4(*type, variable); }
 
    void SlangSyntaxFromGlsl(ShaderStage& shaderStage) const;

    vector<ShaderNodePtr> _lightSamplingNodes;
};
 
MATERIALX_NAMESPACE_END
 
#endif