ColorManagementSystem.h

Go to the documentation of this file.

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

 
#include <MaterialXGenShader/Export.h>
 
#include <MaterialXGenShader/ShaderNode.h>
#include <MaterialXGenShader/ShaderNodeImpl.h>
#include <MaterialXGenShader/TypeDesc.h>
 
#include <MaterialXCore/Document.h>
 
MATERIALX_NAMESPACE_BEGIN
 
class ShaderGenerator;

using ColorManagementSystemPtr = shared_ptr<class ColorManagementSystem>;

struct MX_GENSHADER_API ColorSpaceTransform
{
    ColorSpaceTransform(const string& ss, const string& ts, TypeDesc t);
 
    string sourceSpace;
    string targetSpace;
    TypeDesc type;

    bool operator==(const ColorSpaceTransform& other) const
    {
        return sourceSpace == other.sourceSpace &&
               targetSpace == other.targetSpace &&
               type == other.type;
    }
};

class MX_GENSHADER_API ColorManagementSystem
{
  public:
    virtual ~ColorManagementSystem() { }

    virtual const string& getName() const = 0;

    virtual void loadLibrary(DocumentPtr document);

    bool supportsTransform(const ColorSpaceTransform& transform) const;

    ShaderNodePtr createNode(const ShaderGraph* parent, const ColorSpaceTransform& transform, const string& name,
                             GenContext& context) const;

    virtual bool hasImplementation(const string& /*implName*/) const { return false; }

    virtual ShaderNodeImplPtr createImplementation(const string& /*implName*/) const { return {}; }
 
  protected:
    ColorManagementSystem();

    virtual NodeDefPtr getNodeDef(const ColorSpaceTransform& transform) const = 0;
 
  protected:
    DocumentPtr _document;
};
 
MATERIALX_NAMESPACE_END
 
#endif