Shader.h

Go to the documentation of this file.

//
// Copyright Contributors to the MaterialX Project
// SPDX-License-Identifier: Apache-2.0
//
 
#ifndef MATERIALX_SHADER_H
#define MATERIALX_SHADER_H
 
 
#include <MaterialXGenShader/Export.h>
 
#include <MaterialXGenShader/GenOptions.h>
#include <MaterialXGenShader/ShaderGraph.h>
#include <MaterialXGenShader/ShaderStage.h>
 
MATERIALX_NAMESPACE_BEGIN
 
class ShaderGenerator;
class Shader;
 
class MX_GENSHADER_API Shader
{
  public:
    Shader(const string& name, ShaderGraphPtr graph);
 
    virtual ~Shader() { }
 
    const string& getName() const { return _name; }
 
    size_t numStages() const { return _stages.size(); }
 
    ShaderStage& getStage(size_t index);
 
    const ShaderStage& getStage(size_t index) const;
 
    bool hasStage(const string& name);
 
    ShaderStage& getStage(const string& name);
 
    const ShaderStage& getStage(const string& name) const;
 
    bool hasAttribute(const string& attrib) const
    {
        return _attributeMap.count(attrib) != 0;
    }
 
    ValuePtr getAttribute(const string& attrib) const
    {
        auto it = _attributeMap.find(attrib);
        return it != _attributeMap.end() ? it->second : nullptr;
    }
 
    void setAttribute(const string& attrib, ValuePtr value)
    {
        _attributeMap[attrib] = value;
    }
 
    void setAttribute(const string& attrib)
    {
        _attributeMap[attrib] = Value::createValue<bool>(true);
    }
 
    const ShaderGraph& getGraph() const { return *_graph; }
 
    ShaderGraph& getGraph() { return *_graph; }
 
    bool hasClassification(unsigned int c) const { return _graph->hasClassification(c); }
 
    void setSourceCode(const string& code, const string& stage = Stage::PIXEL) { getStage(stage).setSourceCode(code); }
 
    const string& getSourceCode(const string& stage = Stage::PIXEL) const { return getStage(stage).getSourceCode(); }
 
  protected:
    ShaderStagePtr createStage(const string& name, ConstSyntaxPtr syntax);
 
    string _name;
    ShaderGraphPtr _graph;
    std::unordered_map<string, ShaderStagePtr> _stagesMap;
    vector<ShaderStage*> _stages;
    std::unordered_map<string, ValuePtr> _attributeMap;
 
    friend class ShaderGenerator;
};
 
MATERIALX_NAMESPACE_END
 
#endif