[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("    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;
	struct bt_value *value;

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

	value = bt_value_map_get(source_cfg->params, "begin");
	if (value) {
	        enum bt_value_status ret;

		ret = bt_value_map_insert(trimmer_params, "begin",
				value);
	        BT_PUT(value);
		if (ret) {
			goto end;
		}
	}
	value = bt_value_map_get(source_cfg->params, "end");
	if (value) {
		enum bt_value_status ret;

		ret = bt_value_map_insert(trimmer_params, "end",
				value);
		BT_PUT(value);
		if (ret) {
			goto end;
		}
	}
	value = bt_value_map_get(source_cfg->params, "clock-gmt");
	if (value) {
		enum bt_value_status ret;

		ret = bt_value_map_insert(trimmer_params, "clock-gmt",
				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 (bt_value_map_has_key(source_cfg->params, "begin")
			|| bt_value_map_has_key(source_cfg->params, "end")) {
		/* 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, const 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 PP	211	bt_config_component->component_name->str);
	212	printf(" params:\n");
	213	print_value(bt_config_component->params, 6, true);
	214	}
	215
	216	static
	217	void print_bt_config_components(GPtrArray *array)
	218	{
	219	size_t i;
	220
	221	for (i = 0; i < array->len; i++) {
	222	struct bt_config_component *cfg_component =
e5bc7f81	223	bt_config_get_component(array, i);
c42c79ea PP	224	print_bt_config_component(cfg_component);
	225	BT_PUT(cfg_component);
	226	}
	227	}
	228
	229	static
	230	void print_cfg(struct bt_config *cfg)
	231	{
	232	printf("debug: %d\n", cfg->debug);
	233	printf("verbose: %d\n", cfg->verbose);
	234	printf("do list: %d\n", cfg->do_list);
	235	printf("force correlate: %d\n", cfg->force_correlate);
	236	printf("plugin paths:\n");
	237	print_value(cfg->plugin_paths, 2, true);
	238	printf("sources:\n");
	239	print_bt_config_components(cfg->sources);
	240	printf("sinks:\n");
	241	print_bt_config_components(cfg->sinks);
	242	}
	243
6c2f3ee5 JG	244	static
	245	struct bt_component create_trimmer(struct bt_config_component source_cfg)
	246	{
	247	struct bt_component *trimmer = NULL;
	248	struct bt_component_class *trimmer_class = NULL;
	249	struct bt_value *trimmer_params = NULL;
528debdf	250	struct bt_value *value;
6c2f3ee5 JG	251
	252	assert(component_factory);
	253	trimmer_params = bt_value_map_create();
	254	if (!trimmer_params) {
	255	goto end;
	256	}
	257
528debdf MD	258	value = bt_value_map_get(source_cfg->params, "begin");
528debdf MD	259	if (value) {
6c2f3ee5	260	enum bt_value_status ret;
6c2f3ee5	261
528debdf	262	ret = bt_value_map_insert(trimmer_params, "begin",
6c2f3ee5 JG	263	value);
	264	BT_PUT(value);
	265	if (ret) {
	266	goto end;
	267	}
	268	}
528debdf MD	269	value = bt_value_map_get(source_cfg->params, "end");
528debdf MD	270	if (value) {
6c2f3ee5	271	enum bt_value_status ret;
6c2f3ee5	272
528debdf MD	273	ret = bt_value_map_insert(trimmer_params, "end",
	274	value);
	275	BT_PUT(value);
	276	if (ret) {
6c2f3ee5 JG	277	goto end;
6c2f3ee5 JG	278	}
528debdf MD	279	}
	280	value = bt_value_map_get(source_cfg->params, "clock-gmt");
	281	if (value) {
	282	enum bt_value_status ret;
6c2f3ee5	283
528debdf	284	ret = bt_value_map_insert(trimmer_params, "clock-gmt",
6c2f3ee5 JG	285	value);
	286	BT_PUT(value);
	287	if (ret) {
	288	goto end;
	289	}
	290	}
	291
	292	trimmer_class = bt_component_factory_get_component_class(
	293	component_factory, "utils", BT_COMPONENT_TYPE_FILTER,
	294	"trimmer");
	295	if (!trimmer_class) {
	296	fprintf(stderr, "Could not find trimmer component class. Aborting...\n");
	297	goto end;
	298	}
	299	trimmer = bt_component_create(trimmer_class, "source_trimmer",
	300	trimmer_params);
	301	if (!trimmer) {
	302	goto end;
	303	}
	304	end:
	305	bt_put(trimmer_params);
	306	bt_put(trimmer_class);
	307	return trimmer;
	308	}
	309
	310	static
	311	int connect_source_sink(struct bt_component *source,
	312	struct bt_config_component *source_cfg,
	313	struct bt_component *sink)
	314	{
	315	int ret = 0;
	316	enum bt_component_status sink_status;
	317	struct bt_component *trimmer = NULL;
	318	struct bt_notification_iterator *source_it = NULL;
	319	struct bt_notification_iterator *to_sink_it = NULL;
	320
	321	source_it = bt_component_source_create_iterator(source);
	322	if (!source_it) {
	323	fprintf(stderr, "Failed to instantiate source iterator. Aborting...\n");
	324	ret = -1;
	325	goto end;
	326	}
	327
528debdf MD	328	if (bt_value_map_has_key(source_cfg->params, "begin")
528debdf MD	329	\|\| bt_value_map_has_key(source_cfg->params, "end")) {
6c2f3ee5 JG	330	/* A trimmer must be inserted in the graph. */
	331	enum bt_component_status trimmer_status;
	332
	333	trimmer = create_trimmer(source_cfg);
	334	if (!trimmer) {
	335	fprintf(stderr, "Failed to create trimmer component. Aborting...\n");
	336	ret = -1;
	337	goto end;
	338	}
	339
	340	trimmer_status = bt_component_filter_add_iterator(trimmer,
	341	source_it);
	342	BT_PUT(source_it);
	343	if (trimmer_status != BT_COMPONENT_STATUS_OK) {
	344	fprintf(stderr, "Failed to connect source to trimmer. Aborting...\n");
	345	ret = -1;
	346	goto end;
	347	}
	348
	349	to_sink_it = bt_component_filter_create_iterator(trimmer);
	350	if (!to_sink_it) {
	351	fprintf(stderr, "Failed to instantiate trimmer iterator. Aborting...\n");
	352	ret = -1;
	353	goto end;
	354	}
	355	} else {
	356	BT_MOVE(to_sink_it, source_it);
	357	}
	358
	359	sink_status = bt_component_sink_add_iterator(sink, to_sink_it);
	360	if (sink_status != BT_COMPONENT_STATUS_OK) {
	361	fprintf(stderr, "Failed to connect to sink component. Aborting...\n");
	362	ret = -1;
	363	goto end;
	364	}
	365	end:
	366	bt_put(trimmer);
	367	bt_put(source_it);
	368	bt_put(to_sink_it);
	369	return ret;
	370	}
	371
528debdf	372	int main(int argc, const char **argv)
34ac0e6c	373	{
7f26a816	374	int ret;
c42c79ea PP	375	struct bt_component_class *source_class = NULL;
c42c79ea PP	376	struct bt_component_class *sink_class = NULL;
7c7c0433 JG	377	struct bt_component source = NULL, sink = NULL;
7c7c0433 JG	378	struct bt_value source_params = NULL, sink_params = NULL;
c42c79ea	379	struct bt_config *cfg;
fec2a9f2	380	enum bt_component_status sink_status;
c42c79ea PP	381	struct bt_value *first_plugin_path_value = NULL;
c42c79ea PP	382	const char *first_plugin_path;
e5bc7f81	383	struct bt_config_component source_cfg = NULL, sink_cfg = NULL;
c42c79ea PP	384
	385	cfg = bt_config_from_args(argc, argv, &ret);
	386	if (cfg) {
	387	print_cfg(cfg);
	388	} else {
56a1cced JG	389	goto end;
	390	}
	391
8b596b57 JG	392	babeltrace_verbose = cfg->verbose;
	393	babeltrace_debug = cfg->debug;
	394
c42c79ea PP	395	/* TODO handle more than 1 source and 1 sink. */
c42c79ea PP	396	if (cfg->sources->len != 1 \|\| cfg->sinks->len != 1) {
5f45c11f	397	fprintf(stderr, "Unexpected configuration, aborting...\n");
56a1cced JG	398	ret = -1;
	399	goto end;
	400	}
	401
34ac0e6c MD	402	printf_verbose("Verbose mode active.\n");
34ac0e6c MD	403	printf_debug("Debug mode active.\n");
33bceaf8 JG	404	component_factory = bt_component_factory_create();
	405	if (!component_factory) {
	406	fprintf(stderr, "Failed to create component factory.\n");
	407	ret = -1;
	408	goto end;
	409	}
	410
b86d17b1 JG	411	if (cfg->plugin_paths && !bt_value_array_is_empty(cfg->plugin_paths)) {
	412	first_plugin_path_value = bt_value_array_get(
	413	cfg->plugin_paths, 0);
	414	bt_value_string_get(first_plugin_path_value,
	415	&first_plugin_path);
	416	ret = bt_component_factory_load_recursive(component_factory,
	417	first_plugin_path);
	418	if (ret) {
	419	fprintf(stderr, "Failed to dynamically load plugins.\n");
	420	goto end;
	421	}
33bceaf8 JG	422	}
33bceaf8 JG	423
cba174d5 JG	424	ret = bt_component_factory_load_static(component_factory);
	425	if (ret) {
	426	fprintf(stderr, "Failed to load static plugins.\n");
	427	goto end;
	428	}
	429
c42c79ea	430	print_component_classes_found(component_factory);
e5bc7f81 JG	431
	432	source_cfg = bt_config_get_component(cfg->sources, 0);
	433	source_params = bt_get(source_cfg->params);
56a1cced	434	source_class = bt_component_factory_get_component_class(
e5bc7f81 JG	435	component_factory, source_cfg->plugin_name->str,
	436	BT_COMPONENT_TYPE_SOURCE,
	437	source_cfg->component_name->str);
56a1cced	438	if (!source_class) {
e5bc7f81 JG	439	fprintf(stderr, "Could not find %s.%s source component class. Aborting...\n",
	440	source_cfg->plugin_name->str,
	441	source_cfg->component_name->str);
56a1cced JG	442	ret = -1;
	443	goto end;
	444	}
7c7c0433	445
e5bc7f81 JG	446	sink_cfg = bt_config_get_component(cfg->sinks, 0);
e5bc7f81 JG	447	sink_params = bt_get(sink_cfg->params);
7c7c0433	448	sink_class = bt_component_factory_get_component_class(component_factory,
e5bc7f81 JG	449	sink_cfg->plugin_name->str, BT_COMPONENT_TYPE_SINK,
e5bc7f81 JG	450	sink_cfg->component_name->str);
7c7c0433	451	if (!sink_class) {
e5bc7f81 JG	452	fprintf(stderr, "Could not find %s.%s output component class. Aborting...\n",
	453	sink_cfg->plugin_name->str,
	454	sink_cfg->component_name->str);
7c7c0433 JG	455	ret = -1;
	456	goto end;
	457	}
	458
e5bc7f81	459	source = bt_component_create(source_class, "source", source_params);
56a1cced	460	if (!source) {
e5bc7f81	461	fprintf(stderr, "Failed to instantiate selected source component. Aborting...\n");
c42c79ea PP	462	ret = -1;
	463	goto end;
	464	}
56a1cced	465
e5bc7f81	466	sink = bt_component_create(sink_class, "sink", sink_params);
7c7c0433	467	if (!sink) {
e5bc7f81	468	fprintf(stderr, "Failed to instantiate selected output component. Aborting...\n");
7c7c0433	469	ret = -1;
2e339de1 JG	470	goto end;
	471	}
	472
6c2f3ee5 JG	473	ret = connect_source_sink(source, source_cfg, sink);
6c2f3ee5 JG	474	if (ret) {
fec2a9f2 JG	475	goto end;
fec2a9f2 JG	476	}
78586d8a	477
fec2a9f2 JG	478	while (true) {
fec2a9f2 JG	479	sink_status = bt_component_sink_consume(sink);
fec2a9f2 JG	480	switch (sink_status) {
	481	case BT_COMPONENT_STATUS_AGAIN:
	482	/* Wait for an arbitraty 500 ms. */
	483	usleep(500000);
78586d8a	484	break;
fec2a9f2 JG	485	case BT_COMPONENT_STATUS_OK:
	486	break;
	487	case BT_COMPONENT_STATUS_END:
	488	goto end;
	489	default:
	490	fprintf(stderr, "Sink component returned an error, aborting...\n");
	491	ret = -1;
	492	goto end;
78586d8a	493	}
fec2a9f2	494	}
11e1d048	495	end:
2e339de1	496	BT_PUT(component_factory);
7c7c0433 JG	497	BT_PUT(sink_class);
	498	BT_PUT(source_class);
	499	BT_PUT(source);
	500	BT_PUT(sink);
	501	BT_PUT(source_params);
	502	BT_PUT(sink_params);
c42c79ea PP	503	BT_PUT(cfg);
c42c79ea PP	504	BT_PUT(first_plugin_path_value);
e5bc7f81 JG	505	BT_PUT(sink_cfg);
e5bc7f81 JG	506	BT_PUT(source_cfg);
7f26a816	507	return ret ? 1 : 0;
4c8bfb7e	508	}