[babeltrace.git] / converter / babeltrace.c

/*
 * babeltrace.c
 *
 * Babeltrace Trace Converter
 *
 * Copyright 2010-2011 EfficiOS Inc. and Linux Foundation
 *
 * Author: Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#include <babeltrace/babeltrace.h>
#include <babeltrace/plugin/component-factory.h>
#include <babeltrace/plugin/plugin.h>
#include <babeltrace/plugin/component-class.h>
#include <babeltrace/plugin/notification/iterator.h>
#include <babeltrace/ref.h>
#include <babeltrace/values.h>
#include <stdlib.h>
#include <babeltrace/ctf-ir/metadata.h>	/* for clocks */
#include <popt.h>
#include <string.h>
#include <stdio.h>
#include "babeltrace-cfg.h"

static struct bt_component_factory *component_factory;

static
const char *component_type_str(enum bt_component_type type)
{
	switch (type) {
	case BT_COMPONENT_TYPE_SOURCE:
		return "source";
	case BT_COMPONENT_TYPE_SINK:
		return "sink";
	case BT_COMPONENT_TYPE_FILTER:
		return "filter";
	case BT_COMPONENT_TYPE_UNKNOWN:
	default:
		return "unknown";
	}
}

static
void print_component_classes_found(struct bt_component_factory *factory)
{
	int count, i;

	if (!babeltrace_verbose) {
		return;
	}

	count = bt_component_factory_get_component_class_count(factory);
	if (count <= 0) {
		fprintf(stderr, "No component classes found. Please make sure your plug-in search path is set correctly.\n");
		return;
	}

	printf_verbose("Found %d component classes.\n", count);
	for (i = 0; i < count; i++) {
		struct bt_component_class *component_class =
				bt_component_factory_get_component_class_index(
				factory, i);
		struct bt_plugin *plugin = bt_component_class_get_plugin(
				component_class);
		const char *plugin_name = bt_plugin_get_name(plugin);
		const char *component_name = bt_component_class_get_name(
				component_class);
		const char *path = bt_plugin_get_path(plugin);
		const char *author = bt_plugin_get_author(plugin);
		const char *license = bt_plugin_get_license(plugin);
		const char *plugin_description = bt_plugin_get_description(
				plugin);
		const char *component_description =
				bt_component_class_get_description(
					component_class);
		enum bt_component_type type = bt_component_class_get_type(
				component_class);

		printf_verbose("[%s - %s (%s)]\n", plugin_name, component_name,
			       component_type_str(type));
		printf_verbose("\tpath: %s\n", path ? path : "None");
		printf_verbose("\tauthor: %s\n", author ? author : "Unknown");
		printf_verbose("\tlicense: %s\n", license ? license : "Unknown");
		printf_verbose("\tplugin description: %s\n",
				plugin_description ? plugin_description : "None");
		printf_verbose("\tcomponent description: %s\n",
				component_description ? component_description : "None");

		bt_put(plugin);
		bt_put(component_class);
	}
}

static
void print_indent(size_t indent)
{
	size_t i;

	for (i = 0; i < indent; i++) {
		printf(" ");
	}
}

static
void print_value(struct bt_value *, size_t, bool);

static
bool print_map_value(const char *key, struct bt_value *object, void *data)
{
	size_t indent = (size_t) data;

	print_indent(indent);
	printf("\"%s\": ", key);
	print_value(object, indent, false);

	return true;
}

static
void print_value(struct bt_value *value, size_t indent, bool do_indent)
{
	bool bool_val;
	int64_t int_val;
	double dbl_val;
	const char *str_val;
	int size;
	int i;

	if (!value) {
		return;
	}

	if (do_indent) {
		print_indent(indent);
	}

	switch (bt_value_get_type(value)) {
	case BT_VALUE_TYPE_NULL:
		printf("null\n");
		break;
	case BT_VALUE_TYPE_BOOL:
		bt_value_bool_get(value, &bool_val);
		printf("%s\n", bool_val ? "true" : "false");
		break;
	case BT_VALUE_TYPE_INTEGER:
		bt_value_integer_get(value, &int_val);
		printf("%" PRId64 "\n", int_val);
		break;
	case BT_VALUE_TYPE_FLOAT:
		bt_value_float_get(value, &dbl_val);
		printf("%lf\n", dbl_val);
		break;
	case BT_VALUE_TYPE_STRING:
		bt_value_string_get(value, &str_val);
		printf("\"%s\"\n", str_val);
		break;
	case BT_VALUE_TYPE_ARRAY:
		size = bt_value_array_size(value);
		printf("[\n");

		for (i = 0; i < size; i++) {
			struct bt_value *element =
					bt_value_array_get(value, i);

			print_value(element, indent + 2, true);
			BT_PUT(element);
		}

		print_indent(indent);
		printf("]\n");
		break;
	case BT_VALUE_TYPE_MAP:
		if (bt_value_map_is_empty(value)) {
			printf("{}\n");
			return;
		}

		printf("{\n");
		bt_value_map_foreach(value, print_map_value,
			(void *) (indent + 2));
		print_indent(indent);
		printf("}\n");
		break;
	default:
		assert(false);
	}
}

static
void print_bt_config_component(struct bt_config_component *bt_config_component)
{
	printf("  %s.%s\n", bt_config_component->plugin_name->str,
		bt_config_component->component_name->str);
	printf("    begin timestamp: ");

	if (!bt_config_component->begin.set) {
		printf("not set\n");
	} else {
		printf("%" PRIu64 " ns\n", bt_config_component->begin.value_ns);
	}

	printf("    end timestamp: ");

	if (!bt_config_component->end.set) {
		printf("not set\n");
	} else {
		printf("%" PRIu64 " ns\n", bt_config_component->end.value_ns);
	}

	printf("    params:\n");
	print_value(bt_config_component->params, 6, true);
}

static
void print_bt_config_components(GPtrArray *array)
{
	size_t i;

	for (i = 0; i < array->len; i++) {
		struct bt_config_component *cfg_component =
				bt_config_get_component(array, i);
		print_bt_config_component(cfg_component);
		BT_PUT(cfg_component);
	}
}

static
void print_cfg(struct bt_config *cfg)
{
	printf("debug:           %d\n", cfg->debug);
	printf("verbose:         %d\n", cfg->verbose);
	printf("do list:         %d\n", cfg->do_list);
	printf("force correlate: %d\n", cfg->force_correlate);
	printf("plugin paths:\n");
	print_value(cfg->plugin_paths, 2, true);
	printf("sources:\n");
	print_bt_config_components(cfg->sources);
	printf("sinks:\n");
	print_bt_config_components(cfg->sinks);
}

static
struct bt_component *create_trimmer(struct bt_config_component *source_cfg)
{
	struct bt_component *trimmer = NULL;
	struct bt_component_class *trimmer_class = NULL;
	struct bt_value *trimmer_params = NULL;

	assert(component_factory);
	trimmer_params = bt_value_map_create();
	if (!trimmer_params) {
		goto end;
	}

	if (source_cfg->begin.set) {
	        enum bt_value_status ret;
		struct bt_value *value;

		value = bt_value_integer_create_init(
				source_cfg->begin.value_ns);
		if (!value) {
			goto end;
		}

		ret = bt_value_map_insert(trimmer_params, "begin_ns_epoch",
				value);
	        BT_PUT(value);
		if (ret) {
			goto end;
		}
	}
	if (source_cfg->end.set) {
		enum bt_value_status ret;
		struct bt_value *value;

		value = bt_value_integer_create_init(
				source_cfg->end.value_ns);
		if (!value) {
			goto end;
		}

		ret = bt_value_map_insert(trimmer_params, "end_ns_epoch",
				value);
		BT_PUT(value);
		if (ret) {
			goto end;
		}
	}

	trimmer_class = bt_component_factory_get_component_class(
			component_factory, "utils", BT_COMPONENT_TYPE_FILTER,
			"trimmer");
	if (!trimmer_class) {
		fprintf(stderr, "Could not find trimmer component class. Aborting...\n");
		goto end;
	}
	trimmer = bt_component_create(trimmer_class, "source_trimmer",
			trimmer_params);
	if (!trimmer) {
		goto end;
	}
end:
	bt_put(trimmer_params);
	bt_put(trimmer_class);
	return trimmer;
}

static
int connect_source_sink(struct bt_component *source,
		struct bt_config_component *source_cfg,
		struct bt_component *sink)
{
	int ret = 0;
	enum bt_component_status sink_status;
	struct bt_component *trimmer = NULL;
	struct bt_notification_iterator *source_it = NULL;
	struct bt_notification_iterator *to_sink_it = NULL;

	source_it = bt_component_source_create_iterator(source);
	if (!source_it) {
		fprintf(stderr, "Failed to instantiate source iterator. Aborting...\n");
                ret = -1;
                goto end;
        }

	if (source_cfg->begin.set || source_cfg->begin.set) {
		/* A trimmer must be inserted in the graph. */
		enum bt_component_status trimmer_status;

		trimmer = create_trimmer(source_cfg);
		if (!trimmer) {
			fprintf(stderr, "Failed to create trimmer component. Aborting...\n");
			ret = -1;
			goto end;
		}

		trimmer_status = bt_component_filter_add_iterator(trimmer,
				source_it);
		BT_PUT(source_it);
		if (trimmer_status != BT_COMPONENT_STATUS_OK) {
			fprintf(stderr, "Failed to connect source to trimmer. Aborting...\n");
			ret = -1;
			goto end;
		}

		to_sink_it = bt_component_filter_create_iterator(trimmer);
		if (!to_sink_it) {
			fprintf(stderr, "Failed to instantiate trimmer iterator. Aborting...\n");
			ret = -1;
			goto end;
		}
	} else {
		BT_MOVE(to_sink_it, source_it);
	}

	sink_status = bt_component_sink_add_iterator(sink, to_sink_it);
	if (sink_status != BT_COMPONENT_STATUS_OK) {
		fprintf(stderr, "Failed to connect to sink component. Aborting...\n");
		ret = -1;
		goto end;
	}
end:
	bt_put(trimmer);
	bt_put(source_it);
	bt_put(to_sink_it);
	return ret;
}

int main(int argc, char **argv)
{
	int ret;
	struct bt_component_class *source_class = NULL;
	struct bt_component_class *sink_class = NULL;
	struct bt_component *source = NULL, *sink = NULL;
	struct bt_value *source_params = NULL, *sink_params = NULL;
	struct bt_config *cfg;
	enum bt_component_status sink_status;
	struct bt_value *first_plugin_path_value = NULL;
	const char *first_plugin_path;
	struct bt_config_component *source_cfg = NULL, *sink_cfg = NULL;

	cfg = bt_config_from_args(argc, argv, &ret);
	if (cfg) {
		print_cfg(cfg);
	} else {
		goto end;
	}

	babeltrace_verbose = cfg->verbose;
	babeltrace_debug = cfg->debug;

	/* TODO handle more than 1 source and 1 sink. */
	if (cfg->sources->len != 1 || cfg->sinks->len != 1) {
		fprintf(stderr, "Unexpected configuration, aborting...\n");
		ret = -1;
		goto end;
	}

	printf_verbose("Verbose mode active.\n");
	printf_debug("Debug mode active.\n");
	component_factory = bt_component_factory_create();
	if (!component_factory) {
		fprintf(stderr, "Failed to create component factory.\n");
		ret = -1;
		goto end;
	}

	if (cfg->plugin_paths && !bt_value_array_is_empty(cfg->plugin_paths)) {
		first_plugin_path_value = bt_value_array_get(
				cfg->plugin_paths, 0);
		bt_value_string_get(first_plugin_path_value,
				&first_plugin_path);
		ret = bt_component_factory_load_recursive(component_factory,
				first_plugin_path);
		if (ret) {
			fprintf(stderr, "Failed to dynamically load plugins.\n");
			goto end;
		}
	}

	ret = bt_component_factory_load_static(component_factory);
	if (ret) {
		fprintf(stderr, "Failed to load static plugins.\n");
		goto end;
	}

	print_component_classes_found(component_factory);

	source_cfg = bt_config_get_component(cfg->sources, 0);
	source_params = bt_get(source_cfg->params);
	source_class = bt_component_factory_get_component_class(
			component_factory, source_cfg->plugin_name->str,
			BT_COMPONENT_TYPE_SOURCE,
			source_cfg->component_name->str);
	if (!source_class) {
		fprintf(stderr, "Could not find %s.%s source component class. Aborting...\n",
				source_cfg->plugin_name->str,
				source_cfg->component_name->str);
		ret = -1;
		goto end;
	}

	sink_cfg = bt_config_get_component(cfg->sinks, 0);
	sink_params = bt_get(sink_cfg->params);
	sink_class = bt_component_factory_get_component_class(component_factory,
			sink_cfg->plugin_name->str, BT_COMPONENT_TYPE_SINK,
			sink_cfg->component_name->str);
	if (!sink_class) {
		fprintf(stderr, "Could not find %s.%s output component class. Aborting...\n",
				sink_cfg->plugin_name->str,
				sink_cfg->component_name->str);
		ret = -1;
		goto end;
	}

	source = bt_component_create(source_class, "source", source_params);
	if (!source) {
		fprintf(stderr, "Failed to instantiate selected source component. Aborting...\n");
                ret = -1;
                goto end;
        }

	sink = bt_component_create(sink_class, "sink", sink_params);
	if (!sink) {
		fprintf(stderr, "Failed to instantiate selected output component. Aborting...\n");
		ret = -1;
		goto end;
	}

	ret = connect_source_sink(source, source_cfg, sink);
	if (ret) {
		goto end;
	}

	while (true) {
		sink_status = bt_component_sink_consume(sink);
		switch (sink_status) {
		case BT_COMPONENT_STATUS_AGAIN:
			/* Wait for an arbitraty 500 ms. */
			usleep(500000);
			break;
		case BT_COMPONENT_STATUS_OK:
			break;
		case BT_COMPONENT_STATUS_END:
			goto end;
		default:
			fprintf(stderr, "Sink component returned an error, aborting...\n");
			ret = -1;
			goto end;
		}
	}
end:
	BT_PUT(component_factory);
	BT_PUT(sink_class);
	BT_PUT(source_class);
	BT_PUT(source);
	BT_PUT(sink);
	BT_PUT(source_params);
	BT_PUT(sink_params);
	BT_PUT(cfg);
	BT_PUT(first_plugin_path_value);
	BT_PUT(sink_cfg);
	BT_PUT(source_cfg);
	return ret ? 1 : 0;
}
Commit	Line	Data
34ac0e6c MD	1	/*
	2	* babeltrace.c
	3	*
	4	* Babeltrace Trace Converter
	5	*
64fa3fec MD	6	* Copyright 2010-2011 EfficiOS Inc. and Linux Foundation
	7	*
	8	* Author: Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
34ac0e6c MD	9	*
	10	* Permission is hereby granted, free of charge, to any person obtaining a copy
	11	* of this software and associated documentation files (the "Software"), to deal
	12	* in the Software without restriction, including without limitation the rights
	13	* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
	14	* copies of the Software, and to permit persons to whom the Software is
	15	* furnished to do so, subject to the following conditions:
	16	*
	17	* The above copyright notice and this permission notice shall be included in
	18	* all copies or substantial portions of the Software.
c462e188 MD	19	*
	20	* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
	21	* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
	22	* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
	23	* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
	24	* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
	25	* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
	26	* SOFTWARE.
34ac0e6c	27	*/
4c8bfb7e	28
95d36295	29	#include <babeltrace/babeltrace.h>
33bceaf8	30	#include <babeltrace/plugin/component-factory.h>
7c7c0433 JG	31	#include <babeltrace/plugin/plugin.h>
7c7c0433 JG	32	#include <babeltrace/plugin/component-class.h>
90298357	33	#include <babeltrace/plugin/notification/iterator.h>
2e339de1 JG	34	#include <babeltrace/ref.h>
2e339de1 JG	35	#include <babeltrace/values.h>
34ac0e6c	36	#include <stdlib.h>
a44bc4c9	37	#include <babeltrace/ctf-ir/metadata.h> /* for clocks */
7f26a816 PP	38	#include <popt.h>
	39	#include <string.h>
	40	#include <stdio.h>
c42c79ea	41	#include "babeltrace-cfg.h"
34ac0e6c	42
6c2f3ee5 JG	43	static struct bt_component_factory *component_factory;
6c2f3ee5 JG	44
7c7c0433 JG	45	static
	46	const char *component_type_str(enum bt_component_type type)
	47	{
	48	switch (type) {
	49	case BT_COMPONENT_TYPE_SOURCE:
	50	return "source";
	51	case BT_COMPONENT_TYPE_SINK:
	52	return "sink";
	53	case BT_COMPONENT_TYPE_FILTER:
	54	return "filter";
	55	case BT_COMPONENT_TYPE_UNKNOWN:
	56	default:
	57	return "unknown";
	58	}
	59	}
	60
	61	static
c42c79ea	62	void print_component_classes_found(struct bt_component_factory *factory)
7c7c0433 JG	63	{
	64	int count, i;
	65
	66	if (!babeltrace_verbose) {
	67	return;
	68	}
	69
	70	count = bt_component_factory_get_component_class_count(factory);
	71	if (count <= 0) {
5e86a071	72	fprintf(stderr, "No component classes found. Please make sure your plug-in search path is set correctly.\n");
7c7c0433 JG	73	return;
	74	}
	75
	76	printf_verbose("Found %d component classes.\n", count);
	77	for (i = 0; i < count; i++) {
	78	struct bt_component_class *component_class =
	79	bt_component_factory_get_component_class_index(
	80	factory, i);
	81	struct bt_plugin *plugin = bt_component_class_get_plugin(
	82	component_class);
	83	const char *plugin_name = bt_plugin_get_name(plugin);
	84	const char *component_name = bt_component_class_get_name(
	85	component_class);
	86	const char *path = bt_plugin_get_path(plugin);
	87	const char *author = bt_plugin_get_author(plugin);
	88	const char *license = bt_plugin_get_license(plugin);
	89	const char *plugin_description = bt_plugin_get_description(
	90	plugin);
	91	const char *component_description =
	92	bt_component_class_get_description(
	93	component_class);
	94	enum bt_component_type type = bt_component_class_get_type(
	95	component_class);
	96
	97	printf_verbose("[%s - %s (%s)]\n", plugin_name, component_name,
	98	component_type_str(type));
85411a39 JG	99	printf_verbose("\tpath: %s\n", path ? path : "None");
	100	printf_verbose("\tauthor: %s\n", author ? author : "Unknown");
	101	printf_verbose("\tlicense: %s\n", license ? license : "Unknown");
7c7c0433 JG	102	printf_verbose("\tplugin description: %s\n",
	103	plugin_description ? plugin_description : "None");
	104	printf_verbose("\tcomponent description: %s\n",
	105	component_description ? component_description : "None");
0cc9e945 JG	106
	107	bt_put(plugin);
	108	bt_put(component_class);
7c7c0433 JG	109	}
	110	}
	111
c42c79ea PP	112	static
	113	void print_indent(size_t indent)
	114	{
	115	size_t i;
	116
	117	for (i = 0; i < indent; i++) {
	118	printf(" ");
	119	}
	120	}
	121
	122	static
	123	void print_value(struct bt_value *, size_t, bool);
	124
	125	static
	126	bool print_map_value(const char key, struct bt_value object, void *data)
	127	{
	128	size_t indent = (size_t) data;
	129
	130	print_indent(indent);
	131	printf("\"%s\": ", key);
	132	print_value(object, indent, false);
	133
	134	return true;
	135	}
	136
	137	static
	138	void print_value(struct bt_value *value, size_t indent, bool do_indent)
	139	{
	140	bool bool_val;
	141	int64_t int_val;
	142	double dbl_val;
	143	const char *str_val;
	144	int size;
	145	int i;
	146
	147	if (!value) {
	148	return;
	149	}
	150
	151	if (do_indent) {
	152	print_indent(indent);
	153	}
	154
	155	switch (bt_value_get_type(value)) {
	156	case BT_VALUE_TYPE_NULL:
	157	printf("null\n");
	158	break;
	159	case BT_VALUE_TYPE_BOOL:
	160	bt_value_bool_get(value, &bool_val);
	161	printf("%s\n", bool_val ? "true" : "false");
	162	break;
	163	case BT_VALUE_TYPE_INTEGER:
	164	bt_value_integer_get(value, &int_val);
	165	printf("%" PRId64 "\n", int_val);
	166	break;
	167	case BT_VALUE_TYPE_FLOAT:
	168	bt_value_float_get(value, &dbl_val);
	169	printf("%lf\n", dbl_val);
	170	break;
	171	case BT_VALUE_TYPE_STRING:
	172	bt_value_string_get(value, &str_val);
	173	printf("\"%s\"\n", str_val);
	174	break;
	175	case BT_VALUE_TYPE_ARRAY:
176	size = bt_value_array_size(value);
177	printf("[\n");
178
179	for (i = 0; i < size; i++) {
180	struct bt_value *element =
181	bt_value_array_get(value, i);
182
183	print_value(element, indent + 2, true);
184	BT_PUT(element);
185	}
186
187	print_indent(indent);
188	printf("]\n");
189	break;
190	case BT_VALUE_TYPE_MAP:
191	if (bt_value_map_is_empty(value)) {
192	printf("{}\n");
193	return;
194	}
195
196	printf("{\n");
197	bt_value_map_foreach(value, print_map_value,
198	(void *) (indent + 2));
199	print_indent(indent);
200	printf("}\n");
201	break;
202	default:
203	assert(false);
204	}
205	}
206
207	static
208	void print_bt_config_component(struct bt_config_component *bt_config_component)
209	{
5f45c11f	210	printf(" %s.%s\n", bt_config_component->plugin_name->str,
c42c79ea	211	bt_config_component->component_name->str);
e442b41e PP	212	printf(" begin timestamp: ");
e442b41e PP	213
6c2f3ee5	214	if (!bt_config_component->begin.set) {
e442b41e PP	215	printf("not set\n");
e442b41e PP	216	} else {
6c2f3ee5	217	printf("%" PRIu64 " ns\n", bt_config_component->begin.value_ns);
e442b41e PP	218	}
	219
	220	printf(" end timestamp: ");
	221
6c2f3ee5	222	if (!bt_config_component->end.set) {
e442b41e PP	223	printf("not set\n");
e442b41e PP	224	} else {
6c2f3ee5	225	printf("%" PRIu64 " ns\n", bt_config_component->end.value_ns);
e442b41e PP	226	}
e442b41e PP	227
c42c79ea PP	228	printf(" params:\n");
	229	print_value(bt_config_component->params, 6, true);
	230	}
	231
	232	static
	233	void print_bt_config_components(GPtrArray *array)
	234	{
	235	size_t i;
	236
	237	for (i = 0; i < array->len; i++) {
	238	struct bt_config_component *cfg_component =
e5bc7f81	239	bt_config_get_component(array, i);
c42c79ea PP	240	print_bt_config_component(cfg_component);
	241	BT_PUT(cfg_component);
	242	}
	243	}
	244
	245	static
	246	void print_cfg(struct bt_config *cfg)
	247	{
	248	printf("debug: %d\n", cfg->debug);
	249	printf("verbose: %d\n", cfg->verbose);
	250	printf("do list: %d\n", cfg->do_list);
	251	printf("force correlate: %d\n", cfg->force_correlate);
	252	printf("plugin paths:\n");
	253	print_value(cfg->plugin_paths, 2, true);
	254	printf("sources:\n");
	255	print_bt_config_components(cfg->sources);
	256	printf("sinks:\n");
	257	print_bt_config_components(cfg->sinks);
	258	}
	259
6c2f3ee5 JG	260	static
	261	struct bt_component create_trimmer(struct bt_config_component source_cfg)
	262	{
	263	struct bt_component *trimmer = NULL;
	264	struct bt_component_class *trimmer_class = NULL;
	265	struct bt_value *trimmer_params = NULL;
	266
	267	assert(component_factory);
	268	trimmer_params = bt_value_map_create();
	269	if (!trimmer_params) {
	270	goto end;
	271	}
	272
	273	if (source_cfg->begin.set) {
	274	enum bt_value_status ret;
	275	struct bt_value *value;
	276
	277	value = bt_value_integer_create_init(
	278	source_cfg->begin.value_ns);
	279	if (!value) {
	280	goto end;
	281	}
	282
	283	ret = bt_value_map_insert(trimmer_params, "begin_ns_epoch",
	284	value);
	285	BT_PUT(value);
	286	if (ret) {
	287	goto end;
	288	}
	289	}
	290	if (source_cfg->end.set) {
	291	enum bt_value_status ret;
	292	struct bt_value *value;
	293
	294	value = bt_value_integer_create_init(
	295	source_cfg->end.value_ns);
	296	if (!value) {
	297	goto end;
	298	}
	299
	300	ret = bt_value_map_insert(trimmer_params, "end_ns_epoch",
	301	value);
	302	BT_PUT(value);
	303	if (ret) {
	304	goto end;
	305	}
	306	}
	307
	308	trimmer_class = bt_component_factory_get_component_class(
	309	component_factory, "utils", BT_COMPONENT_TYPE_FILTER,
	310	"trimmer");
	311	if (!trimmer_class) {
	312	fprintf(stderr, "Could not find trimmer component class. Aborting...\n");
	313	goto end;
	314	}
	315	trimmer = bt_component_create(trimmer_class, "source_trimmer",
	316	trimmer_params);
	317	if (!trimmer) {
	318	goto end;
	319	}
	320	end:
	321	bt_put(trimmer_params);
	322	bt_put(trimmer_class);
	323	return trimmer;
324	}
325
326	static
327	int connect_source_sink(struct bt_component *source,
328	struct bt_config_component *source_cfg,
329	struct bt_component *sink)
330	{
331	int ret = 0;
332	enum bt_component_status sink_status;
333	struct bt_component *trimmer = NULL;
334	struct bt_notification_iterator *source_it = NULL;
335	struct bt_notification_iterator *to_sink_it = NULL;
336
337	source_it = bt_component_source_create_iterator(source);
338	if (!source_it) {
339	fprintf(stderr, "Failed to instantiate source iterator. Aborting...\n");
340	ret = -1;
341	goto end;
342	}
343
344	if (source_cfg->begin.set \|\| source_cfg->begin.set) {
345	/* A trimmer must be inserted in the graph. */
346	enum bt_component_status trimmer_status;
347
348	trimmer = create_trimmer(source_cfg);
349	if (!trimmer) {
350	fprintf(stderr, "Failed to create trimmer component. Aborting...\n");
351	ret = -1;
352	goto end;
353	}
354
355	trimmer_status = bt_component_filter_add_iterator(trimmer,
356	source_it);
357	BT_PUT(source_it);
358	if (trimmer_status != BT_COMPONENT_STATUS_OK) {
359	fprintf(stderr, "Failed to connect source to trimmer. Aborting...\n");
360	ret = -1;
361	goto end;
362	}
363
364	to_sink_it = bt_component_filter_create_iterator(trimmer);
365	if (!to_sink_it) {
366	fprintf(stderr, "Failed to instantiate trimmer iterator. Aborting...\n");
367	ret = -1;
368	goto end;
369	}
370	} else {
371	BT_MOVE(to_sink_it, source_it);
372	}
373
374	sink_status = bt_component_sink_add_iterator(sink, to_sink_it);
375	if (sink_status != BT_COMPONENT_STATUS_OK) {
376	fprintf(stderr, "Failed to connect to sink component. Aborting...\n");
377	ret = -1;
378	goto end;
379	}
380	end:
381	bt_put(trimmer);
382	bt_put(source_it);
383	bt_put(to_sink_it);
384	return ret;
385	}
386
bbefb8dd	387	int main(int argc, char **argv)
34ac0e6c	388	{
7f26a816	389	int ret;
c42c79ea PP	390	struct bt_component_class *source_class = NULL;
c42c79ea PP	391	struct bt_component_class *sink_class = NULL;
7c7c0433 JG	392	struct bt_component source = NULL, sink = NULL;
7c7c0433 JG	393	struct bt_value source_params = NULL, sink_params = NULL;
c42c79ea	394	struct bt_config *cfg;
fec2a9f2	395	enum bt_component_status sink_status;
c42c79ea PP	396	struct bt_value *first_plugin_path_value = NULL;
c42c79ea PP	397	const char *first_plugin_path;
e5bc7f81	398	struct bt_config_component source_cfg = NULL, sink_cfg = NULL;
c42c79ea PP	399
	400	cfg = bt_config_from_args(argc, argv, &ret);
	401	if (cfg) {
	402	print_cfg(cfg);
	403	} else {
56a1cced JG	404	goto end;
	405	}
	406
8b596b57 JG	407	babeltrace_verbose = cfg->verbose;
	408	babeltrace_debug = cfg->debug;
	409
c42c79ea PP	410	/* TODO handle more than 1 source and 1 sink. */
c42c79ea PP	411	if (cfg->sources->len != 1 \|\| cfg->sinks->len != 1) {
5f45c11f	412	fprintf(stderr, "Unexpected configuration, aborting...\n");
56a1cced JG	413	ret = -1;
	414	goto end;
	415	}
	416
34ac0e6c MD	417	printf_verbose("Verbose mode active.\n");
34ac0e6c MD	418	printf_debug("Debug mode active.\n");
33bceaf8 JG	419	component_factory = bt_component_factory_create();
	420	if (!component_factory) {
	421	fprintf(stderr, "Failed to create component factory.\n");
	422	ret = -1;
	423	goto end;
	424	}
	425
b86d17b1 JG	426	if (cfg->plugin_paths && !bt_value_array_is_empty(cfg->plugin_paths)) {
	427	first_plugin_path_value = bt_value_array_get(
	428	cfg->plugin_paths, 0);
	429	bt_value_string_get(first_plugin_path_value,
	430	&first_plugin_path);
	431	ret = bt_component_factory_load_recursive(component_factory,
	432	first_plugin_path);
	433	if (ret) {
	434	fprintf(stderr, "Failed to dynamically load plugins.\n");
	435	goto end;
	436	}
33bceaf8 JG	437	}
33bceaf8 JG	438
cba174d5 JG	439	ret = bt_component_factory_load_static(component_factory);
	440	if (ret) {
	441	fprintf(stderr, "Failed to load static plugins.\n");
	442	goto end;
	443	}
	444
c42c79ea	445	print_component_classes_found(component_factory);
e5bc7f81 JG	446
	447	source_cfg = bt_config_get_component(cfg->sources, 0);
	448	source_params = bt_get(source_cfg->params);
56a1cced	449	source_class = bt_component_factory_get_component_class(
e5bc7f81 JG	450	component_factory, source_cfg->plugin_name->str,
	451	BT_COMPONENT_TYPE_SOURCE,
	452	source_cfg->component_name->str);
56a1cced	453	if (!source_class) {
e5bc7f81 JG	454	fprintf(stderr, "Could not find %s.%s source component class. Aborting...\n",
	455	source_cfg->plugin_name->str,
	456	source_cfg->component_name->str);
56a1cced JG	457	ret = -1;
	458	goto end;
	459	}
7c7c0433	460
e5bc7f81 JG	461	sink_cfg = bt_config_get_component(cfg->sinks, 0);
e5bc7f81 JG	462	sink_params = bt_get(sink_cfg->params);
7c7c0433	463	sink_class = bt_component_factory_get_component_class(component_factory,
e5bc7f81 JG	464	sink_cfg->plugin_name->str, BT_COMPONENT_TYPE_SINK,
e5bc7f81 JG	465	sink_cfg->component_name->str);
7c7c0433	466	if (!sink_class) {
e5bc7f81 JG	467	fprintf(stderr, "Could not find %s.%s output component class. Aborting...\n",
	468	sink_cfg->plugin_name->str,
	469	sink_cfg->component_name->str);
7c7c0433 JG	470	ret = -1;
	471	goto end;
	472	}
	473
e5bc7f81	474	source = bt_component_create(source_class, "source", source_params);
56a1cced	475	if (!source) {
e5bc7f81	476	fprintf(stderr, "Failed to instantiate selected source component. Aborting...\n");
c42c79ea PP	477	ret = -1;
	478	goto end;
	479	}
56a1cced	480
e5bc7f81	481	sink = bt_component_create(sink_class, "sink", sink_params);
7c7c0433	482	if (!sink) {
e5bc7f81	483	fprintf(stderr, "Failed to instantiate selected output component. Aborting...\n");
7c7c0433	484	ret = -1;
2e339de1 JG	485	goto end;
	486	}
	487
6c2f3ee5 JG	488	ret = connect_source_sink(source, source_cfg, sink);
6c2f3ee5 JG	489	if (ret) {
fec2a9f2 JG	490	goto end;
fec2a9f2 JG	491	}
78586d8a	492
fec2a9f2 JG	493	while (true) {
fec2a9f2 JG	494	sink_status = bt_component_sink_consume(sink);
fec2a9f2 JG	495	switch (sink_status) {
	496	case BT_COMPONENT_STATUS_AGAIN:
	497	/* Wait for an arbitraty 500 ms. */
	498	usleep(500000);
78586d8a	499	break;
fec2a9f2 JG	500	case BT_COMPONENT_STATUS_OK:
	501	break;
	502	case BT_COMPONENT_STATUS_END:
	503	goto end;
	504	default:
	505	fprintf(stderr, "Sink component returned an error, aborting...\n");
	506	ret = -1;
	507	goto end;
78586d8a	508	}
fec2a9f2	509	}
11e1d048	510	end:
2e339de1	511	BT_PUT(component_factory);
7c7c0433 JG	512	BT_PUT(sink_class);
	513	BT_PUT(source_class);
	514	BT_PUT(source);
	515	BT_PUT(sink);
	516	BT_PUT(source_params);
	517	BT_PUT(sink_params);
c42c79ea PP	518	BT_PUT(cfg);
c42c79ea PP	519	BT_PUT(first_plugin_path_value);
e5bc7f81 JG	520	BT_PUT(sink_cfg);
e5bc7f81 JG	521	BT_PUT(source_cfg);
7f26a816	522	return ret ? 1 : 0;
4c8bfb7e	523	}