* Copyright (C) 2020-2023 EfficiOS, inc.
*/
-#include <utility>
-
#include "common/assert.h"
-#include "cpp-common/bt2/wrap.hpp"
+#include "cpp-common/bt2/component-class-dev.hpp"
+#include "cpp-common/bt2/component-class.hpp"
+#include "cpp-common/bt2/graph.hpp"
+#include "cpp-common/bt2/plugin-load.hpp"
+#include "cpp-common/bt2/plugin.hpp"
+#include "cpp-common/bt2/query-executor.hpp"
+#include "cpp-common/bt2c/call.hpp"
#include "run-in.hpp"
-namespace {
-
-struct RunInData final
-{
- RunInCompClsQueryFunc compClsCtxFunc;
- RunInCompClsInitFunc compCtxFunc;
- RunInMsgIterClsInitFunc msgIterCtxFunc;
-};
-
-const RunInData& runInDataFromMethodData(void * const methodData)
-{
- return *static_cast<const RunInData *>(methodData);
-}
-
-bt_component_class_initialize_method_status compClsInit(bt_self_component_source * const selfComp,
- bt_self_component_source_configuration *,
- const bt_value *,
- void * const initMethodData)
-{
- const auto status =
- bt_self_component_source_add_output_port(selfComp, "out", initMethodData, nullptr);
-
- BT_ASSERT(status == BT_SELF_COMPONENT_ADD_PORT_STATUS_OK);
-
- auto& data = runInDataFromMethodData(initMethodData);
-
- if (data.compCtxFunc) {
- data.compCtxFunc(bt2::wrap(bt_self_component_source_as_self_component(selfComp)));
- }
-
- return BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_OK;
-}
-
-bt_component_class_query_method_status
-compClsQuery(bt_self_component_class_source * const selfCompCls, bt_private_query_executor *,
- const char *, const bt_value *, void * const methodData,
- const bt_value ** const result)
-{
- auto& data = runInDataFromMethodData(methodData);
-
- if (data.compClsCtxFunc) {
- data.compClsCtxFunc(
- bt2::wrap(bt_self_component_class_source_as_self_component_class(selfCompCls)));
- }
-
- *result = bt_value_null;
- return BT_COMPONENT_CLASS_QUERY_METHOD_STATUS_OK;
-}
-
-bt_message_iterator_class_initialize_method_status
-msgIterClsInit(bt_self_message_iterator * const selfMsgIter,
- bt_self_message_iterator_configuration *, bt_self_component_port_output * const port)
+void RunIn::onQuery(bt2::SelfComponentClass)
{
- auto& data = runInDataFromMethodData(bt_self_component_port_get_data(
- bt_self_component_port_output_as_self_component_port(port)));
-
- if (data.msgIterCtxFunc) {
- data.msgIterCtxFunc(bt2::wrap(selfMsgIter));
- }
-
- return BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_OK;
-}
-
-bt_message_iterator_class_next_method_status
-msgIterClsNext(bt_self_message_iterator *, bt_message_array_const, uint64_t, uint64_t *)
-{
- return BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_END;
}
-struct DummySinkData
+void RunIn::onCompInit(bt2::SelfComponent)
{
- bt_message_iterator *msgIter;
-};
-
-bt_component_class_initialize_method_status dummySinkInit(bt_self_component_sink * const self,
- bt_self_component_sink_configuration *,
- const bt_value *,
- void * const initMethodData)
-{
- const auto status = bt_self_component_sink_add_input_port(self, "in", NULL, nullptr);
-
- BT_ASSERT(status == BT_SELF_COMPONENT_ADD_PORT_STATUS_OK);
- bt_self_component_set_data(bt_self_component_sink_as_self_component(self), initMethodData);
- return BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_OK;
}
-DummySinkData& dummySinkDataFromSelfCompSink(bt_self_component_sink * const self)
+void RunIn::onMsgIterInit(bt2::SelfMessageIterator)
{
- return *static_cast<DummySinkData *>(
- bt_self_component_get_data(bt_self_component_sink_as_self_component(self)));
}
-bt_component_class_sink_graph_is_configured_method_status
-dummySinkGraphIsConfigured(bt_self_component_sink * const self)
-{
- const auto port = bt_self_component_sink_borrow_input_port_by_name(self, "in");
-
- BT_ASSERT(port);
-
- const auto status = bt_message_iterator_create_from_sink_component(
- self, port, &dummySinkDataFromSelfCompSink(self).msgIter);
-
- BT_ASSERT(status == BT_MESSAGE_ITERATOR_CREATE_FROM_SINK_COMPONENT_STATUS_OK);
- return BT_COMPONENT_CLASS_SINK_GRAPH_IS_CONFIGURED_METHOD_STATUS_OK;
-}
-
-bt_component_class_sink_consume_method_status dummySinkConsume(bt_self_component_sink * const self)
-{
- bt_message_array_const msgs;
- uint64_t msgCount;
- const auto status =
- bt_message_iterator_next(dummySinkDataFromSelfCompSink(self).msgIter, &msgs, &msgCount);
-
- BT_ASSERT(status == BT_MESSAGE_ITERATOR_NEXT_STATUS_END);
- return BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_END;
-}
+namespace {
-} /* namespace */
+class RunInSource;
-void runIn(RunInCompClsQueryFunc compClsCtxFunc, RunInCompClsInitFunc compCtxFunc,
- RunInMsgIterClsInitFunc msgIterCtxFunc)
+class RunInSourceMsgIter final : public bt2::UserMessageIterator<RunInSourceMsgIter, RunInSource>
{
- RunInData data {std::move(compClsCtxFunc), std::move(compCtxFunc), std::move(msgIterCtxFunc)};
-
- /* Create and configure custom source component class */
- const auto msgIterCls = bt_message_iterator_class_create(msgIterClsNext);
-
- BT_ASSERT(msgIterCls);
-
+public:
+ explicit RunInSourceMsgIter(const bt2::SelfMessageIterator self,
+ bt2::SelfMessageIteratorConfiguration,
+ const bt2::SelfComponentOutputPort port) :
+ bt2::UserMessageIterator<RunInSourceMsgIter, RunInSource> {self, "RUN-IN-SRC-MSG-ITER"}
{
- const auto status =
- bt_message_iterator_class_set_initialize_method(msgIterCls, msgIterClsInit);
-
- BT_ASSERT(status == BT_MESSAGE_ITERATOR_CLASS_SET_METHOD_STATUS_OK);
+ port.data<RunIn>().onMsgIterInit(self);
}
- const auto srcCompCls = bt_component_class_source_create("yo", msgIterCls);
-
- BT_ASSERT(srcCompCls);
-
+ void _next(bt2::ConstMessageArray&)
{
- const auto status =
- bt_component_class_source_set_initialize_method(srcCompCls, compClsInit);
-
- BT_ASSERT(status == BT_COMPONENT_CLASS_SET_METHOD_STATUS_OK);
}
+};
+class RunInSource final :
+ public bt2::UserSourceComponent<RunInSource, RunInSourceMsgIter, RunIn, RunIn>
+{
+public:
+ static constexpr auto name = "run-in-src";
+
+ explicit RunInSource(const bt2::SelfSourceComponent self, bt2::ConstMapValue,
+ RunIn * const runIn) :
+ bt2::UserSourceComponent<RunInSource, RunInSourceMsgIter, RunIn, RunIn> {self,
+ "RUN-IN-SRC"},
+ _mRunIn {runIn}
{
- const auto status = bt_component_class_source_set_query_method(srcCompCls, compClsQuery);
-
- BT_ASSERT(status == BT_COMPONENT_CLASS_SET_METHOD_STATUS_OK);
+ this->_addOutputPort("out", *runIn);
+ _mRunIn->onCompInit(self);
}
- /* Execute a query (executes `compClsCtxFunc`) */
+ static bt2::Value::Shared _query(const bt2::SelfComponentClass self, bt2::PrivateQueryExecutor,
+ bt2c::CStringView, bt2::ConstValue, RunIn *data)
{
- const auto queryExec = bt_query_executor_create_with_method_data(
- bt_component_class_source_as_component_class(srcCompCls), "", nullptr, &data);
-
- BT_ASSERT(queryExec);
-
- const bt_value *queryRes;
- const auto status = bt_query_executor_query(queryExec, &queryRes);
-
- BT_ASSERT(status == BT_QUERY_EXECUTOR_QUERY_STATUS_OK);
- bt_value_put_ref(queryRes);
- bt_query_executor_put_ref(queryExec);
+ data->onQuery(self);
+ return bt2::NullValue {}.shared();
}
- /* Create a dummy sink component */
- const auto sinkCompCls = bt_component_class_sink_create("dummy", dummySinkConsume);
-
- BT_ASSERT(sinkCompCls);
-
- {
- const auto status =
- bt_component_class_sink_set_initialize_method(sinkCompCls, dummySinkInit);
+private:
+ RunIn *_mRunIn;
+};
- BT_ASSERT(status == BT_COMPONENT_CLASS_SET_METHOD_STATUS_OK);
- }
+} /* namespace */
- {
- const auto status = bt_component_class_sink_set_graph_is_configured_method(
- sinkCompCls, dummySinkGraphIsConfigured);
+void runIn(RunIn& runIn)
+{
+ const auto srcCompCls = bt2::SourceComponentClass::create<RunInSource>();
- BT_ASSERT(status == BT_COMPONENT_CLASS_SET_METHOD_STATUS_OK);
- }
+ /* Execute a query */
+ bt2::QueryExecutor::create(*srcCompCls, "object-name", runIn)->query();
/* Create graph */
- const auto graph = bt_graph_create(0);
-
- BT_ASSERT(graph);
+ const auto graph = bt2::Graph::create(0);
/* Add custom source component (executes `compCtxFunc`) */
- const bt_component_source *srcComp;
-
- {
- const auto status = bt_graph_add_source_component_with_initialize_method_data(
- graph, srcCompCls, "the-source", NULL, &data, BT_LOGGING_LEVEL_NONE, &srcComp);
-
- BT_ASSERT(status == BT_GRAPH_ADD_COMPONENT_STATUS_OK);
- }
+ const auto srcComp = graph->addComponent(*srcCompCls, "the-source", runIn);
/* Add dummy sink component */
- const bt_component_sink *sinkComp;
- DummySinkData dummySinkData;
+ const auto sinkComp = bt2c::call([&] {
+ const auto utilsPlugin = bt2::findPlugin("utils");
- {
- const auto status = bt_graph_add_sink_component_with_initialize_method_data(
- graph, sinkCompCls, "the-sink", NULL, &dummySinkData, BT_LOGGING_LEVEL_NONE, &sinkComp);
+ BT_ASSERT(utilsPlugin);
- BT_ASSERT(status == BT_GRAPH_ADD_COMPONENT_STATUS_OK);
- }
+ const auto dummySinkCompCls = utilsPlugin->sinkComponentClasses()["dummy"];
- /* Connect ports */
- {
- const auto outPort = bt_component_source_borrow_output_port_by_name_const(srcComp, "out");
+ BT_ASSERT(dummySinkCompCls);
- BT_ASSERT(outPort);
+ return graph->addComponent(*dummySinkCompCls, "the-sink");
+ });
- const auto inPort = bt_component_sink_borrow_input_port_by_name_const(sinkComp, "in");
-
- BT_ASSERT(inPort);
+ /* Connect ports */
+ const auto outPort = srcComp.outputPorts()["out"];
+ BT_ASSERT(outPort);
- const auto status = bt_graph_connect_ports(graph, outPort, inPort, nullptr);
+ const auto inPort = sinkComp.inputPorts()["in"];
+ BT_ASSERT(inPort);
- BT_ASSERT(status == BT_GRAPH_CONNECT_PORTS_STATUS_OK);
- }
+ graph->connectPorts(*outPort, *inPort);
/* Run graph (executes `msgIterCtxFunc`) */
- {
- const auto status = bt_graph_run(graph);
-
- BT_ASSERT(status == BT_GRAPH_RUN_STATUS_OK);
- }
-
- /* Discard owned objects */
- bt_graph_put_ref(graph);
- bt_component_class_source_put_ref(srcCompCls);
- bt_component_class_sink_put_ref(sinkCompCls);
- bt_message_iterator_class_put_ref(msgIterCls);
-}
-
-void runInCompClsQuery(RunInCompClsQueryFunc func)
-{
- runIn(std::move(func), nullptr, nullptr);
-}
-
-void runInCompClsInit(RunInCompClsInitFunc func)
-{
- runIn(nullptr, std::move(func), nullptr);
-}
-
-void runInMsgIterClsInit(RunInMsgIterClsInitFunc func)
-{
- runIn(nullptr, nullptr, std::move(func));
+ graph->run();
}