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