Clara Holoscan v0.3.0

Program Listing for File io_spec.hpp

Return to documentation for file (include/holoscan/core/io_spec.hpp)

Copy
Copied!
            

/* * SPDX-FileCopyrightText: Copyright (c) 2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved. * SPDX-License-Identifier: Apache-2.0 * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #ifndef HOLOSCAN_CORE_IO_SPEC_HPP #define HOLOSCAN_CORE_IO_SPEC_HPP #include "./common.hpp" #include <iostream> #include <memory> #include <typeinfo> #include <utility> #include <vector> #include "./conditions/gxf/boolean.hpp" #include "./conditions/gxf/count.hpp" #include "./conditions/gxf/downstream_affordable.hpp" #include "./conditions/gxf/message_available.hpp" namespace holoscan { class IOSpec { public: enum class IOType { kInput, kOutput }; IOSpec(OperatorSpec* op_spec, const std::string& name, IOType io_type, const std::type_info* typeinfo) : op_spec_(op_spec), name_(name), io_type_(io_type), typeinfo_(typeinfo) {} OperatorSpec* op_spec() const { return op_spec_; } std::string& name() { return name_; } IOType io_type() const { return io_type_; } const std::type_info* typeinfo() const { return typeinfo_; } std::shared_ptr<Resource> resource() const { return resource_; } void resource(std::shared_ptr<Resource> resource) { resource_ = resource; } std::vector<std::pair<ConditionType, std::shared_ptr<Condition>>>& conditions() { return conditions_; } template <typename... ArgsT> IOSpec& condition(ConditionType type, ArgsT&&... args) { switch (type) { case ConditionType::kMessageAvailable: conditions_.emplace_back( type, std::make_shared<MessageAvailableCondition>(std::forward<ArgsT>(args)...)); break; case ConditionType::kDownstreamMessageAffordable: conditions_.emplace_back( type, std::make_shared<DownstreamMessageAffordableCondition>(std::forward<ArgsT>(args)...)); break; case ConditionType::kCount: conditions_.emplace_back(type, std::make_shared<CountCondition>(std::forward<ArgsT>(args)...)); break; case ConditionType::kBoolean: conditions_.emplace_back(type, std::make_shared<BooleanCondition>(std::forward<ArgsT>(args)...)); break; case ConditionType::kNone: HOLOSCAN_LOG_ERROR("Condition type kNone is not allowed in IOSpec::condition()"); break; } return *this; } private: OperatorSpec* op_spec_ = nullptr; std::string name_; IOType io_type_; const std::type_info* typeinfo_ = nullptr; std::shared_ptr<Resource> resource_; std::vector<std::pair<ConditionType, std::shared_ptr<Condition>>> conditions_; }; } // namespace holoscan #endif/* HOLOSCAN_CORE_IO_SPEC_HPP */

© Copyright 2022, NVIDIA. Last updated on Jun 28, 2023.