#include <utility>
-#include "run-in.hpp"
#include "common/assert.h"
+#include "cpp-common/bt2/component-class-dev.hpp"
+#include "cpp-common/bt2/component-class.hpp"
+
+#include "run-in.hpp"
+
+namespace {
struct RunInData final
{
RunInMsgIterClsInitFunc msgIterCtxFunc;
};
-static const RunInData& runInDataFromMethodData(void * const methodData)
-{
- return *static_cast<const RunInData *>(methodData);
-}
+class RunInSource;
-static bt_component_class_initialize_method_status
-compClsInit(bt_self_component_source * const selfComp, bt_self_component_source_configuration *,
- const bt_value *, void * const initMethodData)
+class RunInSourceMsgIter final : public bt2::UserMessageIterator<RunInSourceMsgIter, RunInSource>
{
- 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);
+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& data = port.data<const RunInData>();
- if (data.compCtxFunc) {
- data.compCtxFunc(bt_self_component_source_as_self_component(selfComp));
+ if (data.msgIterCtxFunc) {
+ data.msgIterCtxFunc(self);
+ }
}
- return BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_OK;
-}
+ void _next(bt2::ConstMessageArray&)
+ {
+ }
+};
-static 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)
+class RunInSource final :
+ public bt2::UserSourceComponent<RunInSource, RunInSourceMsgIter, const RunInData,
+ const RunInData>
{
- auto& data = runInDataFromMethodData(methodData);
+public:
+ static constexpr auto name = "run-in-src";
+
+ explicit RunInSource(const bt2::SelfSourceComponent self, bt2::ConstMapValue,
+ const RunInData * const runInData) :
+ bt2::UserSourceComponent<RunInSource, RunInSourceMsgIter, const RunInData,
+ const RunInData> {self, "RUN-IN-SRC"},
+ _mRunInData {runInData}
+ {
+ this->_addOutputPort("out", *runInData);
- if (data.compClsCtxFunc) {
- data.compClsCtxFunc(bt_self_component_class_source_as_self_component_class(selfCompCls));
+ if (_mRunInData->compCtxFunc) {
+ _mRunInData->compCtxFunc(self);
+ }
}
- *result = bt_value_null;
- return BT_COMPONENT_CLASS_QUERY_METHOD_STATUS_OK;
-}
-
-static 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)
-{
- auto& data = runInDataFromMethodData(bt_self_component_port_get_data(
- bt_self_component_port_output_as_self_component_port(port)));
+ static bt2::Value::Shared _query(const bt2::SelfComponentClass self, bt2::PrivateQueryExecutor,
+ bt2c::CStringView, bt2::ConstValue,
+ const RunInData * const data)
+ {
+ if (data->compClsCtxFunc) {
+ data->compClsCtxFunc(self);
+ }
- if (data.msgIterCtxFunc) {
- data.msgIterCtxFunc(selfMsgIter);
+ return bt2::NullValue {}.shared();
}
- return BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_OK;
-}
-
-static 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;
-}
+private:
+ const RunInData *_mRunInData;
+};
-void runIn(RunInCompClsQueryFunc compClsCtxFunc, RunInCompClsInitFunc compCtxFunc,
- RunInMsgIterClsInitFunc msgIterCtxFunc)
+class DummySink : public bt2::UserSinkComponent<DummySink>
{
- 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:
+ static constexpr auto name = "dummy";
+ explicit DummySink(const bt2::SelfSinkComponent self, bt2::ConstMapValue, void *) :
+ bt2::UserSinkComponent<DummySink>(self, "DUMMY-SINK")
{
- const auto status =
- bt_message_iterator_class_set_initialize_method(msgIterCls, msgIterClsInit);
-
- BT_ASSERT(status == BT_MESSAGE_ITERATOR_CLASS_SET_METHOD_STATUS_OK);
+ this->_addInputPort("in");
}
- const auto srcCompCls = bt_component_class_source_create("yo", msgIterCls);
-
- BT_ASSERT(srcCompCls);
-
+ void _graphIsConfigured()
{
- const auto status =
- bt_component_class_source_set_initialize_method(srcCompCls, compClsInit);
-
- BT_ASSERT(status == BT_COMPONENT_CLASS_SET_METHOD_STATUS_OK);
+ _mMsgIter = this->_createMessageIterator(this->_inputPorts()["in"]);
}
+ bool _consume()
{
- const auto status = bt_component_class_source_set_query_method(srcCompCls, compClsQuery);
-
- BT_ASSERT(status == BT_COMPONENT_CLASS_SET_METHOD_STATUS_OK);
+ return _mMsgIter->next().has_value();
}
+private:
+ bt2::MessageIterator::Shared _mMsgIter;
+};
+
+} /* namespace */
+
+void runIn(RunInCompClsQueryFunc compClsCtxFunc, RunInCompClsInitFunc compCtxFunc,
+ RunInMsgIterClsInitFunc msgIterCtxFunc)
+{
+ RunInData data {std::move(compClsCtxFunc), std::move(compCtxFunc), std::move(msgIterCtxFunc)};
+ const auto srcCompCls = bt2::SourceComponentClass::create<RunInSource>();
+
/* Execute a query (executes `compClsCtxFunc`) */
{
const auto queryExec = bt_query_executor_create_with_method_data(
- bt_component_class_source_as_component_class(srcCompCls), "", nullptr, &data);
+ bt_component_class_source_as_component_class(srcCompCls->libObjPtr()), "", nullptr,
+ &data);
BT_ASSERT(queryExec);
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);
}
{
const auto status = bt_graph_add_source_component_with_initialize_method_data(
- graph, srcCompCls, "whatever", NULL, &data, BT_LOGGING_LEVEL_NONE, &srcComp);
+ graph, srcCompCls->libObjPtr(), "the-source", NULL, &data, BT_LOGGING_LEVEL_NONE,
+ &srcComp);
BT_ASSERT(status == BT_GRAPH_ADD_COMPONENT_STATUS_OK);
}
- /* Find `sink.utils.dummy` component class */
- const bt_plugin *utilsPlugin = nullptr;
-
- {
- const auto status =
- bt_plugin_find("utils", BT_TRUE, BT_TRUE, BT_TRUE, BT_TRUE, BT_TRUE, &utilsPlugin);
-
- BT_ASSERT(status == BT_PLUGIN_FIND_STATUS_OK);
- }
-
- /* Add `sink.utils.dummy` component */
+ /* Add dummy sink component */
const bt_component_sink *sinkComp;
{
- const auto dummyCompCls =
- bt_plugin_borrow_sink_component_class_by_name_const(utilsPlugin, "dummy");
-
- BT_ASSERT(dummyCompCls);
-
- const auto status = bt_graph_add_sink_component(graph, dummyCompCls, "the-sink", nullptr,
- BT_LOGGING_LEVEL_NONE, &sinkComp);
+ const auto sinkCompCls = bt2::SinkComponentClass::create<DummySink>();
+ const auto status = bt_graph_add_sink_component_with_initialize_method_data(
+ graph, sinkCompCls->libObjPtr(), "the-sink", nullptr, nullptr, BT_LOGGING_LEVEL_NONE,
+ &sinkComp);
BT_ASSERT(status == BT_GRAPH_ADD_COMPONENT_STATUS_OK);
}
}
/* Run graph (executes `msgIterCtxFunc`) */
- const auto status = bt_graph_run(graph);
+ {
+ const auto status = bt_graph_run(graph);
- BT_ASSERT(status == BT_GRAPH_RUN_STATUS_OK);
+ BT_ASSERT(status == BT_GRAPH_RUN_STATUS_OK);
+ }
- /* Discard plugin and graph */
- bt_plugin_put_ref(utilsPlugin);
+ /* Discard owned objects */
bt_graph_put_ref(graph);
- bt_component_class_source_put_ref(srcCompCls);
- bt_message_iterator_class_put_ref(msgIterCls);
}
void runInCompClsQuery(RunInCompClsQueryFunc func)
projects / babeltrace.git / blobdiff