Program Listing for File Subnode.hpp

↰ Return to documentation for file (include/depthai/pipeline/Subnode.hpp)

#pragma once

#include <depthai/pipeline/DeviceNode.hpp>
#include <depthai/pipeline/Node.hpp>
#include <memory>
#include <string>

namespace dai {

template <typename T>
class Subnode {
    std::shared_ptr<Node> node;

   public:
    Subnode(Node& parent, std::string alias) {
        if(!parent.configureMode) {
            // Create node as well
            node = std::make_shared<T>();
            node->setAlias(alias);

            // Add node to parents map
            parent.nodeMap.push_back(node);

            // If node is DeviceNode, copy device from parent
            if(std::dynamic_pointer_cast<DeviceNode>(node) != nullptr) {
                // Check if the parent is DeviceNode
                if(dynamic_cast<DeviceNode*>(&parent) == nullptr) {
                    if(std::dynamic_pointer_cast<HostRunnable>(node) == nullptr) {
                        throw std::runtime_error("The parent node of a non HostRunnable DeviceNode must be a DeviceNode");
                    }
                } else {
                    auto device = dynamic_cast<DeviceNode*>(&parent)->getDevice();
                    std::dynamic_pointer_cast<DeviceNode>(node)->setDevice(device);
                }
            }

            // Now that node is created, call buildInternal
            node->buildInternal();
        }

        // Add reference
        parent.nodeRefs.push_back(&node);
    }
    T& operator*() const noexcept {
        return *std::static_pointer_cast<T>(node).get();
    }
    T* operator->() const noexcept {
        return std::static_pointer_cast<T>(node).get();
    }
};

}  // namespace dai