leafNode.h

Go to the documentation of this file.

//
// Copyright 2025 Pixar
//
// Licensed under the terms set forth in the LICENSE.txt file available at
// https://openusd.org/license.
//
#ifndef PXR_EXEC_EF_LEAF_NODE_H
#define PXR_EXEC_EF_LEAF_NODE_H
 
 
#include "pxr/pxr.h"
 
#include "pxr/exec/ef/api.h"
 
#include "pxr/base/tf/staticTokens.h"
#include "pxr/exec/vdf/node.h"
#include "pxr/exec/vdf/connection.h"
 
PXR_NAMESPACE_OPEN_SCOPE
 
#define EF_LEAF_TOKENS \
    (in)
 
TF_DECLARE_PUBLIC_TOKENS(EfLeafTokens, EF_API, EF_LEAF_TOKENS);
 
class EF_API_TYPE EfLeafNode final : public VdfNode
{
public:
    static bool IsALeafNode(const VdfNode &node) {
        return node.GetNumOutputs() == 0 && node.IsA<EfLeafNode>();
    }
 
    static EfLeafNode* AsALeafNode(VdfNode *const node) {
        if (node && node->GetNumOutputs() == 0) {
            return dynamic_cast<EfLeafNode *>(node);
        }
        return nullptr;
    }
 
    static const EfLeafNode* AsALeafNode(const VdfNode *const node) {
        if (node && node->GetNumOutputs() == 0) {
            return dynamic_cast<const EfLeafNode *>(node);
        }
        return nullptr;
    }
 
    static const VdfOutput *GetSourceOutput(const VdfNode &node);
 
    static VdfMaskedOutput GetSourceMaskedOutput(const VdfNode &node);
 
    EF_API
    EfLeafNode(VdfNetwork *network, TfType inputType);
 
private:
 
    virtual void Compute(const VdfContext &context) const override;
 
    // Only a network is allowed to delete nodes.
    virtual ~EfLeafNode();
};
 
inline const VdfOutput *
EfLeafNode::GetSourceOutput(const VdfNode &node)
{
    const VdfInput *const input = node.GetInputsIterator().begin()->second;
    if (input && input->GetNumConnections() > 0) {
        return &(*input)[0].GetSourceOutput();
    }
    return nullptr;
}
 
inline VdfMaskedOutput
EfLeafNode::GetSourceMaskedOutput(const VdfNode &node)
{
    const VdfInput *const input = node.GetInputsIterator().begin()->second;
    if (input && input->GetNumConnections() > 0) {
        return (*input)[0].GetSourceMaskedOutput();
    }
    return VdfMaskedOutput();
}
 
PXR_NAMESPACE_CLOSE_SCOPE
 
#endif