ShaderTranslator.h

//
// Copyright Contributors to the MaterialX Project
// SPDX-License-Identifier: Apache-2.0
//
 
#ifndef MATERIALX_SHADERTRANSLATOR_H
#define MATERIALX_SHADERTRANSLATOR_H
 
#include <MaterialXGenShader/Export.h>
 
#include <MaterialXGenShader/Shader.h>
#include <MaterialXGenShader/Util.h>
 
MATERIALX_NAMESPACE_BEGIN

using ShaderTranslatorPtr = shared_ptr<class ShaderTranslator>;

class MX_GENSHADER_API ShaderTranslator
{
  public:
    static ShaderTranslatorPtr create()
    {
        return ShaderTranslatorPtr(new ShaderTranslator());
    }

    void translateShader(NodePtr shader, const string& destCategory);

    void translateAllMaterials(DocumentPtr doc, const string& destShader);
 
  protected:
    ShaderTranslator() { }
 
    // Connect translation node inputs from the original shader
    void connectTranslationInputs(NodePtr shader, NodeDefPtr translationNodeDef);
 
    // Connect translation node outputs to finalize shader translation
    void connectTranslationOutputs(NodePtr shader);
 
  protected:
    NodeGraphPtr _graph;
    NodePtr _translationNode;
};
 
MATERIALX_NAMESPACE_END
 
#endif