OslShaderGenerator.h

Go to the documentation of this file.

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

 
#include <MaterialXGenOsl/Export.h>
 
#include <MaterialXGenShader/ShaderGenerator.h>
 
MATERIALX_NAMESPACE_BEGIN
 
using OslShaderGeneratorPtr = shared_ptr<class OslShaderGenerator>;

class MX_GENOSL_API OslShaderGenerator : public ShaderGenerator
{
  public:
    OslShaderGenerator(TypeSystemPtr typeSystem);

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

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

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

    void emitFunctionCalls(const ShaderGraph& graph, GenContext& context, ShaderStage& stage, uint32_t classification = 0u) const override;

    void emitFunctionBodyBegin(const ShaderNode& node, GenContext& context, ShaderStage& stage, Syntax::Punctuation punc = Syntax::CURLY_BRACKETS) const override;

    static const string TARGET;

    void registerShaderMetadata(const DocumentPtr& doc, GenContext& context) const override;
 
  protected:
    virtual ShaderPtr createShader(const string& name, ElementPtr element, GenContext& context) const;

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

    virtual void emitShaderInputs(const VariableBlock& inputs, ShaderStage& stage) const;

    virtual void emitShaderOutputs(const VariableBlock& inputs, ShaderStage& stage) const;

    virtual void emitMetadata(const ShaderPort* port, ShaderStage& stage) const;
};
 
namespace OSL
{

extern MX_GENOSL_API const string UNIFORMS;
extern MX_GENOSL_API const string INPUTS;
extern MX_GENOSL_API const string OUTPUTS;
 
} // namespace OSL
 
MATERIALX_NAMESPACE_END
 
#endif