78 lines
2.2 KiB
C++
78 lines
2.2 KiB
C++
//
|
|
// Copyright Contributors to the MaterialX Project
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#include <MaterialXGenMsl/Nodes/LightShaderNodeMsl.h>
|
|
|
|
#include <MaterialXGenShader/Shader.h>
|
|
|
|
MATERIALX_NAMESPACE_BEGIN
|
|
|
|
LightShaderNodeMsl::LightShaderNodeMsl() :
|
|
_lightUniforms(HW::LIGHT_DATA, EMPTY_STRING)
|
|
{
|
|
}
|
|
|
|
ShaderNodeImplPtr LightShaderNodeMsl::create()
|
|
{
|
|
return std::make_shared<LightShaderNodeMsl>();
|
|
}
|
|
|
|
const string& LightShaderNodeMsl::getTarget() const
|
|
{
|
|
return MslShaderGenerator::TARGET;
|
|
}
|
|
|
|
void LightShaderNodeMsl::initialize(const InterfaceElement& element, GenContext& context)
|
|
{
|
|
SourceCodeNode::initialize(element, context);
|
|
|
|
if (_inlined)
|
|
{
|
|
throw ExceptionShaderGenError("Light shaders doesn't support inlined implementations'");
|
|
}
|
|
|
|
if (!element.isA<Implementation>())
|
|
{
|
|
throw ExceptionShaderGenError("Element '" + element.getName() + "' is not an Implementation element");
|
|
}
|
|
|
|
const Implementation& impl = static_cast<const Implementation&>(element);
|
|
|
|
// Store light uniforms for all inputs on the interface
|
|
NodeDefPtr nodeDef = impl.getNodeDef();
|
|
for (InputPtr input : nodeDef->getActiveInputs())
|
|
{
|
|
const TypeDesc type = context.getTypeDesc(input->getType());
|
|
_lightUniforms.add(type, input->getName(), input->getValue());
|
|
}
|
|
}
|
|
|
|
void LightShaderNodeMsl::createVariables(const ShaderNode&, GenContext& context, Shader& shader) const
|
|
{
|
|
ShaderStage& ps = shader.getStage(Stage::PIXEL);
|
|
|
|
// Create all light uniforms
|
|
VariableBlock& lightData = ps.getUniformBlock(HW::LIGHT_DATA);
|
|
for (size_t i = 0; i < _lightUniforms.size(); ++i)
|
|
{
|
|
const ShaderPort* u = _lightUniforms[i];
|
|
lightData.add(u->getType(), u->getName());
|
|
}
|
|
|
|
const MslShaderGenerator& shadergen = static_cast<const MslShaderGenerator&>(context.getShaderGenerator());
|
|
shadergen.addStageLightingUniforms(context, ps);
|
|
}
|
|
|
|
void LightShaderNodeMsl::emitFunctionCall(const ShaderNode&, GenContext& context, ShaderStage& stage) const
|
|
{
|
|
DEFINE_SHADER_STAGE(stage, Stage::PIXEL)
|
|
{
|
|
const ShaderGenerator& shadergen = context.getShaderGenerator();
|
|
shadergen.emitLine(_functionName + "(light, position, result)", stage);
|
|
}
|
|
}
|
|
|
|
MATERIALX_NAMESPACE_END
|