omni/graph/exec/unstable/INodeGraphDef.h

File members: omni/graph/exec/unstable/INodeGraphDef.h

// Copyright (c) 2022-2023, NVIDIA CORPORATION. All rights reserved.
//
// NVIDIA CORPORATION and its licensors retain all intellectual property
// and proprietary rights in and to this software, related documentation
// and any modifications thereto. Any use, reproduction, disclosure or
// distribution of this software and related documentation without an express
// license agreement from NVIDIA CORPORATION is strictly prohibited.
//

#pragma once

#include <omni/graph/exec/unstable/IDef.h>
#include <omni/graph/exec/unstable/INodeFactory.h>

namespace omni
{
namespace graph
{
namespace exec
{
namespace unstable
{

class INode;
class INodeGraphDef_abi;
class INodeGraphDef;
class ITopology;

class INodeGraphDef_abi : public omni::core::Inherits<IDef, OMNI_TYPE_ID("omni.graph.exec.unstable.INodeGraphDef")>
{
protected:
    virtual OMNI_ATTR("not_null, no_acquire") ITopology* getTopology_abi() noexcept = 0;

    virtual void initializeState_abi(OMNI_ATTR("in, out, not_null, ref") ExecutionTask* rootTask) noexcept = 0;

    virtual OMNI_ATTR("nodiscard") Status
        preExecute_abi(OMNI_ATTR("in, out, not_null, ref") ExecutionTask* info) noexcept = 0;

    virtual OMNI_ATTR("nodiscard") Status
        postExecute_abi(OMNI_ATTR("in, out, not_null, ref") ExecutionTask* info) noexcept = 0;

    virtual INodeFactory* getNodeFactory_abi() noexcept = 0;
};

using NodeGraphDefPtr = omni::core::ObjectPtr<INodeGraphDef>;

} // namespace unstable
} // namespace exec
} // namespace graph
} // namespace omni

#define OMNI_BIND_INCLUDE_INTERFACE_DECL
#include <omni/graph/exec/unstable/INodeGraphDef.gen.h>

class omni::graph::exec::unstable::INodeGraphDef
    : public omni::core::Generated<omni::graph::exec::unstable::INodeGraphDef_abi>
{
public:
    inline INode* getRoot() noexcept;
};

#include <omni/graph/exec/unstable/ITopology.h>

inline omni::graph::exec::unstable::INode* omni::graph::exec::unstable::INodeGraphDef::getRoot() noexcept
{
    return getTopology()->getRoot();
}

#define OMNI_BIND_INCLUDE_INTERFACE_IMPL
#include <omni/graph/exec/unstable/INodeGraphDef.gen.h>