[babeltrace.git] / plugins / trimmer / iterator.c

/*
 * iterator.c
 *
 * Babeltrace Trace Trimmer Iterator
 *
 * Copyright 2016 Jérémie Galarneau <jeremie.galarneau@efficios.com>
 *
 * Author: Jérémie Galarneau <jeremie.galarneau@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 "trimmer.h"
#include "iterator.h"
#include <babeltrace/plugin/notification/iterator.h>
#include <babeltrace/plugin/notification/notification.h>
#include <babeltrace/plugin/notification/event.h>
#include <babeltrace/plugin/notification/stream.h>
#include <babeltrace/plugin/notification/packet.h>
#include <babeltrace/plugin/filter.h>
#include <babeltrace/ctf-ir/event.h>
#include <babeltrace/ctf-ir/stream.h>
#include <babeltrace/ctf-ir/stream-class.h>
#include <babeltrace/ctf-ir/clock.h>
#include <babeltrace/ctf-ir/packet.h>
#include <babeltrace/ctf-ir/trace.h>
#include <babeltrace/ctf-ir/fields.h>
#include <assert.h>

static
void trimmer_iterator_destroy(struct bt_notification_iterator *it)
{
	struct trimmer_iterator *it_data;

	it_data = bt_notification_iterator_get_private_data(it);
	assert(it_data);

	if (it_data->input_iterator_group) {
		g_ptr_array_free(it_data->input_iterator_group, TRUE);
	}
	bt_put(it_data->current_notification);
	g_free(it_data);
}

BT_HIDDEN
enum bt_component_status trimmer_iterator_init(struct bt_component *component,
		struct bt_notification_iterator *iterator)
{
	enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
	enum bt_notification_iterator_status it_ret;
	struct trimmer_iterator *it_data = g_new0(struct trimmer_iterator, 1);

	if (!it_data) {
		ret = BT_COMPONENT_STATUS_NOMEM;
		goto end;
	}

	/* FIXME init trimmer_iterator */
	it_ret = bt_notification_iterator_set_private_data(iterator, it_data);
	if (it_ret) {
		goto end;
	}

	it_ret = bt_notification_iterator_set_destroy_cb(iterator,
		        trimmer_iterator_destroy);
	if (it_ret) {
		ret = BT_COMPONENT_STATUS_ERROR;
		goto end;
	}

	it_ret = bt_notification_iterator_set_next_cb(iterator,
			trimmer_iterator_next);
	if (it_ret) {
		ret = BT_COMPONENT_STATUS_ERROR;
		goto end;
	}

	it_ret = bt_notification_iterator_set_get_cb(iterator,
			trimmer_iterator_get);
	if (it_ret) {
		ret = BT_COMPONENT_STATUS_ERROR;
		goto end;
	}

	it_ret = bt_notification_iterator_set_seek_time_cb(iterator,
			trimmer_iterator_seek_time);
	if (it_ret) {
		ret = BT_COMPONENT_STATUS_ERROR;
		goto end;
	}
end:
	return ret;
}

BT_HIDDEN
struct bt_notification *trimmer_iterator_get(
		struct bt_notification_iterator *iterator)
{
	struct trimmer_iterator *trim_it;

	trim_it = bt_notification_iterator_get_private_data(iterator);
	assert(trim_it);

	if (!trim_it->current_notification) {
		enum bt_notification_iterator_status it_ret;

		it_ret = trimmer_iterator_next(iterator);
		if (it_ret) {
			goto end;
		}
	}
end:
	return bt_get(trim_it->current_notification);
}

static
int update_lazy_bound(struct trimmer_bound *bound, const char *name,
		int64_t ts, bool *lazy_update)
{
	struct tm tm;
	int64_t value;
	time_t timeval;

	*lazy_update = false;

	if (!bound->lazy) {
		return 0;
	}
	tm.tm_isdst = -1;
	timeval = ts / NSEC_PER_SEC;

	if (bound->lazy_values.gmt) {
		/* Get day, month, year. */
		if (!gmtime_r(&timeval, &tm)) {
			printf_error("Failure in gmtime_r()");
			goto error;
		}
		tm.tm_sec = bound->lazy_values.ss;
		tm.tm_min = bound->lazy_values.mm;
		tm.tm_hour = bound->lazy_values.hh;
		timeval = timegm(&tm);
		if (timeval < 0) {
			printf_error("Failure in timegm(), incorrectly formatted %s timestamp",
					name);
			goto error;
		}
	} else {
		/* Get day, month, year. */
		if (!localtime_r(&timeval, &tm)) {
			printf_error("Failure in localtime_r()");
			goto error;
		}
		tm.tm_sec = bound->lazy_values.ss;
		tm.tm_min = bound->lazy_values.mm;
		tm.tm_hour = bound->lazy_values.hh;
		timeval = mktime(&tm);
		if (timeval < 0) {
			printf_error("Failure in mktime(), incorrectly formatted %s timestamp",
				name);
			goto error;
		}
	}
	value = (int64_t) timeval;
	value *= NSEC_PER_SEC;
	value += bound->lazy_values.ns;
	bound->value = value;
	bound->set = true;
	bound->lazy = false;
	*lazy_update = true;
	return 0;

error:
	return -1;
}

static
enum bt_notification_iterator_status
evaluate_event_notification(struct bt_notification *notification,
		struct trimmer_bound *begin, struct trimmer_bound *end,
		bool *_event_in_range)
{
	int64_t ts;
	int clock_ret;
	struct bt_ctf_event *event = NULL;
	bool in_range = true;
	struct bt_ctf_clock *clock = NULL;
	struct bt_ctf_trace *trace = NULL;
	struct bt_ctf_stream *stream = NULL;
	struct bt_ctf_stream_class *stream_class = NULL;
	struct bt_ctf_clock_value *clock_value = NULL;
	enum bt_notification_iterator_status ret =
			BT_NOTIFICATION_ITERATOR_STATUS_OK;
	bool lazy_update = false;

	event = bt_notification_event_get_event(notification);
	assert(event);

	stream = bt_ctf_event_get_stream(event);
	assert(stream);

	stream_class = bt_ctf_stream_get_class(stream);
	assert(stream_class);

	trace = bt_ctf_stream_class_get_trace(stream_class);
	assert(trace);

	/* FIXME multi-clock? */
	clock = bt_ctf_trace_get_clock(trace, 0);
	if (!clock) {
		goto end;
	}

	clock_value = bt_ctf_event_get_clock_value(event, clock);
	if (!clock_value) {
		printf_error("Failed to retrieve clock value");
		ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
		goto end;
	}

	clock_ret = bt_ctf_clock_value_get_value_ns_from_epoch(
			clock_value, &ts);
	if (clock_ret) {
		printf_error("Failed to retrieve clock value timestamp");
		ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
		goto end;
	}
	if (update_lazy_bound(begin, "begin", ts, &lazy_update)) {
		ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
		goto end;
	}
	if (update_lazy_bound(end, "end", ts, &lazy_update)) {
		ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
		goto end;
	}
	if (lazy_update && begin->set && end->set) {
		if (begin->value > end->value) {
			printf_error("Unexpected: time range begin value is above end value");
			ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
			goto end;
		}
	}
	if (begin->set && ts < begin->value) {
		in_range = false;
	}
	if (end->set && ts > end->value) {
		in_range = false;
	}
end:
	bt_put(event);
	bt_put(clock);
	bt_put(trace);
	bt_put(stream);
	bt_put(stream_class);
	bt_put(clock_value);
	*_event_in_range = in_range;
	return ret;
}

static
int ns_from_integer_field(struct bt_ctf_field *integer, int64_t *ns)
{
	int ret = 0;
	int is_signed;
	uint64_t raw_clock_value;
	struct bt_ctf_field_type *integer_type = NULL;
	struct bt_ctf_clock *clock = NULL;
	struct bt_ctf_clock_value *clock_value = NULL;

	integer_type = bt_ctf_field_get_type(integer);
	assert(integer_type);
	clock = bt_ctf_field_type_integer_get_mapped_clock(integer_type);
	if (!clock) {
		ret = -1;
		goto end;
	}

	is_signed = bt_ctf_field_type_integer_get_signed(integer_type);
	if (!is_signed) {
		ret = bt_ctf_field_unsigned_integer_get_value(integer,
				&raw_clock_value);
		if (ret) {
			goto end;
		}
	} else {
		/* Signed clock values are unsupported. */
		goto end;
	}

	clock_value = bt_ctf_clock_value_create(clock, raw_clock_value);
        if (!clock_value) {
		goto end;
	}

	ret = bt_ctf_clock_value_get_value_ns_from_epoch(clock_value, ns);
end:
	bt_put(integer_type);
	bt_put(clock);
	bt_put(clock_value);
	return ret;
}

static
enum bt_notification_iterator_status evaluate_packet_notification(
		struct bt_notification *notification,
		struct trimmer_bound *begin, struct trimmer_bound *end,
		bool *_packet_in_range)
{
        enum bt_notification_iterator_status ret =
			BT_NOTIFICATION_ITERATOR_STATUS_OK;
	int64_t begin_ns, pkt_begin_ns, end_ns, pkt_end_ns;
	bool in_range = true;
	struct bt_ctf_packet *packet = NULL;
	struct bt_ctf_field *packet_context = NULL,
			*timestamp_begin = NULL,
			*timestamp_end = NULL;

        switch (bt_notification_get_type(notification)) {
	case BT_NOTIFICATION_TYPE_PACKET_BEGIN:
		packet = bt_notification_packet_begin_get_packet(notification);
		break;
	case BT_NOTIFICATION_TYPE_PACKET_END:
		packet = bt_notification_packet_end_get_packet(notification);
		break;
	default:
		break;
	}
	assert(packet);

	packet_context = bt_ctf_packet_get_context(packet);
	if (!packet_context) {
		goto end;
	}

	if (!bt_ctf_field_is_structure(packet_context)) {
		goto end;
	}

	timestamp_begin = bt_ctf_field_structure_get_field(
			packet_context, "timestamp_begin");
	if (!timestamp_begin || !bt_ctf_field_is_integer(timestamp_begin)) {
		goto end;
	}
	timestamp_end = bt_ctf_field_structure_get_field(
			packet_context, "timestamp_end");
	if (!timestamp_end || !bt_ctf_field_is_integer(timestamp_end)) {
		goto end;
	}

	if (ns_from_integer_field(timestamp_begin, &pkt_begin_ns)) {
		goto end;
	}
	if (ns_from_integer_field(timestamp_end, &pkt_end_ns)) {
		goto end;
	}

	begin_ns = begin->set ? begin->value : INT64_MIN;
	end_ns = end->set ? end->value : INT64_MAX;

	/*
	 * Accept if there is any overlap between the selected region and the
	 * packet.
	 */
	in_range = (pkt_end_ns >= begin_ns) && (pkt_begin_ns <= end_ns);
end:
	*_packet_in_range = in_range;
	bt_put(packet);
	bt_put(packet_context);
	bt_put(timestamp_begin);
	bt_put(timestamp_end);
	return ret;
}

/* Return true if the notification should be forwarded. */
static
enum bt_notification_iterator_status evaluate_notification(
		struct bt_notification *notification,
		struct trimmer_bound *begin, struct trimmer_bound *end,
		bool *in_range)
{
	enum bt_notification_type type;
	enum bt_notification_iterator_status ret =
			BT_NOTIFICATION_ITERATOR_STATUS_OK;

	type = bt_notification_get_type(notification);
	switch (type) {
	case BT_NOTIFICATION_TYPE_EVENT:
	        ret = evaluate_event_notification(notification, begin,
				end, in_range);
		break;
	case BT_NOTIFICATION_TYPE_PACKET_BEGIN:
	case BT_NOTIFICATION_TYPE_PACKET_END:
	        ret = evaluate_packet_notification(notification, begin,
				end, in_range);
		break;
	default:
		/* Accept all other notifications. */
		break;
	}
	return ret;
}

BT_HIDDEN
enum bt_notification_iterator_status trimmer_iterator_next(
		struct bt_notification_iterator *iterator)
{
	struct trimmer_iterator *trim_it = NULL;
	struct bt_component *component = NULL;
	struct trimmer *trimmer = NULL;
	struct bt_notification_iterator *source_it = NULL;
	enum bt_notification_iterator_status ret =
			BT_NOTIFICATION_ITERATOR_STATUS_OK;
	enum bt_component_status component_ret;
	bool notification_in_range = false;

	trim_it = bt_notification_iterator_get_private_data(iterator);
	assert(trim_it);

	component = bt_notification_iterator_get_component(iterator);
	assert(component);
	trimmer = bt_component_get_private_data(component);
	assert(trimmer);

	/* FIXME, should handle input iterator groups. */
	component_ret = bt_component_filter_get_input_iterator(component, 0,
			&source_it);
	assert((component_ret == BT_COMPONENT_STATUS_OK) && source_it);

	while (!notification_in_range) {
		struct bt_notification *notification;

		ret = bt_notification_iterator_next(source_it);
		if (ret != BT_NOTIFICATION_ITERATOR_STATUS_OK) {
			goto end;
		}

		notification = bt_notification_iterator_get_notification(
			        source_it);
		if (!notification) {
			ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
			goto end;
		}

	        ret = evaluate_notification(notification,
				&trimmer->begin, &trimmer->end,
				&notification_in_range);
		if (notification_in_range) {
			BT_MOVE(trim_it->current_notification, notification);
		} else {
			bt_put(notification);
		}
	}
end:
	bt_put(source_it);
	bt_put(component);
	return ret;
}

BT_HIDDEN
enum bt_notification_iterator_status trimmer_iterator_seek_time(
		struct bt_notification_iterator *iterator, int64_t time)
{
	enum bt_notification_iterator_status ret;

	ret = BT_NOTIFICATION_ITERATOR_STATUS_OK;
end:
	return ret;
}
Commit	Line	Data
cab3f160 JG	1	/*
	2	* iterator.c
	3	*
	4	* Babeltrace Trace Trimmer Iterator
	5	*
	6	* Copyright 2016 Jérémie Galarneau <jeremie.galarneau@efficios.com>
	7	*
	8	* Author: Jérémie Galarneau <jeremie.galarneau@efficios.com>
	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.
	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.
	27	*/
	28
	29	#include "trimmer.h"
	30	#include "iterator.h"
	31	#include <babeltrace/plugin/notification/iterator.h>
44d3cbf0 JG	32	#include <babeltrace/plugin/notification/notification.h>
	33	#include <babeltrace/plugin/notification/event.h>
	34	#include <babeltrace/plugin/notification/stream.h>
	35	#include <babeltrace/plugin/notification/packet.h>
	36	#include <babeltrace/plugin/filter.h>
	37	#include <babeltrace/ctf-ir/event.h>
	38	#include <babeltrace/ctf-ir/stream.h>
	39	#include <babeltrace/ctf-ir/stream-class.h>
	40	#include <babeltrace/ctf-ir/clock.h>
	41	#include <babeltrace/ctf-ir/packet.h>
	42	#include <babeltrace/ctf-ir/trace.h>
72208f03	43	#include <babeltrace/ctf-ir/fields.h>
44d3cbf0 JG	44	#include <assert.h>
	45
	46	static
	47	void trimmer_iterator_destroy(struct bt_notification_iterator *it)
	48	{
	49	struct trimmer_iterator *it_data;
	50
	51	it_data = bt_notification_iterator_get_private_data(it);
	52	assert(it_data);
	53
	54	if (it_data->input_iterator_group) {
	55	g_ptr_array_free(it_data->input_iterator_group, TRUE);
	56	}
	57	bt_put(it_data->current_notification);
44d3cbf0 JG	58	g_free(it_data);
44d3cbf0 JG	59	}
cab3f160 JG	60
	61	BT_HIDDEN
	62	enum bt_component_status trimmer_iterator_init(struct bt_component *component,
	63	struct bt_notification_iterator *iterator)
	64	{
44d3cbf0	65	enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
cab3f160	66	enum bt_notification_iterator_status it_ret;
44d3cbf0 JG	67	struct trimmer_iterator *it_data = g_new0(struct trimmer_iterator, 1);
	68
	69	if (!it_data) {
	70	ret = BT_COMPONENT_STATUS_NOMEM;
	71	goto end;
	72	}
	73
	74	/* FIXME init trimmer_iterator */
	75	it_ret = bt_notification_iterator_set_private_data(iterator, it_data);
	76	if (it_ret) {
	77	goto end;
	78	}
	79
	80	it_ret = bt_notification_iterator_set_destroy_cb(iterator,
	81	trimmer_iterator_destroy);
	82	if (it_ret) {
	83	ret = BT_COMPONENT_STATUS_ERROR;
	84	goto end;
	85	}
cab3f160 JG	86
	87	it_ret = bt_notification_iterator_set_next_cb(iterator,
	88	trimmer_iterator_next);
	89	if (it_ret) {
	90	ret = BT_COMPONENT_STATUS_ERROR;
	91	goto end;
	92	}
	93
	94	it_ret = bt_notification_iterator_set_get_cb(iterator,
	95	trimmer_iterator_get);
	96	if (it_ret) {
	97	ret = BT_COMPONENT_STATUS_ERROR;
	98	goto end;
	99	}
	100
	101	it_ret = bt_notification_iterator_set_seek_time_cb(iterator,
	102	trimmer_iterator_seek_time);
	103	if (it_ret) {
	104	ret = BT_COMPONENT_STATUS_ERROR;
	105	goto end;
	106	}
	107	end:
	108	return ret;
	109	}
	110
	111	BT_HIDDEN
	112	struct bt_notification *trimmer_iterator_get(
	113	struct bt_notification_iterator *iterator)
	114	{
44d3cbf0 JG	115	struct trimmer_iterator *trim_it;
	116
	117	trim_it = bt_notification_iterator_get_private_data(iterator);
	118	assert(trim_it);
	119
	120	if (!trim_it->current_notification) {
	121	enum bt_notification_iterator_status it_ret;
	122
	123	it_ret = trimmer_iterator_next(iterator);
	124	if (it_ret) {
	125	goto end;
	126	}
	127	}
	128	end:
	129	return bt_get(trim_it->current_notification);
	130	}
	131
528debdf MD	132	static
528debdf MD	133	int update_lazy_bound(struct trimmer_bound bound, const char name,
55595636	134	int64_t ts, bool *lazy_update)
528debdf MD	135	{
	136	struct tm tm;
	137	int64_t value;
	138	time_t timeval;
	139
55595636 MD	140	*lazy_update = false;
55595636 MD	141
528debdf MD	142	if (!bound->lazy) {
	143	return 0;
	144	}
	145	tm.tm_isdst = -1;
	146	timeval = ts / NSEC_PER_SEC;
	147
	148	if (bound->lazy_values.gmt) {
	149	/* Get day, month, year. */
	150	if (!gmtime_r(&timeval, &tm)) {
55595636	151	printf_error("Failure in gmtime_r()");
528debdf MD	152	goto error;
	153	}
	154	tm.tm_sec = bound->lazy_values.ss;
	155	tm.tm_min = bound->lazy_values.mm;
	156	tm.tm_hour = bound->lazy_values.hh;
	157	timeval = timegm(&tm);
	158	if (timeval < 0) {
55595636 MD	159	printf_error("Failure in timegm(), incorrectly formatted %s timestamp",
55595636 MD	160	name);
528debdf MD	161	goto error;
	162	}
	163	} else {
	164	/* Get day, month, year. */
	165	if (!localtime_r(&timeval, &tm)) {
55595636	166	printf_error("Failure in localtime_r()");
528debdf MD	167	goto error;
	168	}
	169	tm.tm_sec = bound->lazy_values.ss;
	170	tm.tm_min = bound->lazy_values.mm;
	171	tm.tm_hour = bound->lazy_values.hh;
	172	timeval = mktime(&tm);
	173	if (timeval < 0) {
55595636	174	printf_error("Failure in mktime(), incorrectly formatted %s timestamp",
528debdf MD	175	name);
	176	goto error;
	177	}
	178	}
	179	value = (int64_t) timeval;
	180	value *= NSEC_PER_SEC;
	181	value += bound->lazy_values.ns;
	182	bound->value = value;
	183	bound->set = true;
	184	bound->lazy = false;
55595636	185	*lazy_update = true;
528debdf MD	186	return 0;
	187
	188	error:
528debdf MD	189	return -1;
	190	}
	191
44d3cbf0	192	static
55595636 MD	193	enum bt_notification_iterator_status
	194	evaluate_event_notification(struct bt_notification *notification,
	195	struct trimmer_bound begin, struct trimmer_bound end,
	196	bool *_event_in_range)
44d3cbf0	197	{
72208f03 JG	198	int64_t ts;
	199	int clock_ret;
	200	struct bt_ctf_event *event = NULL;
	201	bool in_range = true;
	202	struct bt_ctf_clock *clock = NULL;
	203	struct bt_ctf_trace *trace = NULL;
44d3cbf0	204	struct bt_ctf_stream *stream = NULL;
72208f03 JG	205	struct bt_ctf_stream_class *stream_class = NULL;
72208f03 JG	206	struct bt_ctf_clock_value *clock_value = NULL;
55595636 MD	207	enum bt_notification_iterator_status ret =
	208	BT_NOTIFICATION_ITERATOR_STATUS_OK;
	209	bool lazy_update = false;
44d3cbf0	210
72208f03 JG	211	event = bt_notification_event_get_event(notification);
72208f03 JG	212	assert(event);
44d3cbf0	213
72208f03 JG	214	stream = bt_ctf_event_get_stream(event);
72208f03 JG	215	assert(stream);
44d3cbf0	216
72208f03 JG	217	stream_class = bt_ctf_stream_get_class(stream);
	218	assert(stream_class);
	219
	220	trace = bt_ctf_stream_class_get_trace(stream_class);
	221	assert(trace);
	222
	223	/* FIXME multi-clock? */
	224	clock = bt_ctf_trace_get_clock(trace, 0);
	225	if (!clock) {
	226	goto end;
44d3cbf0	227	}
44d3cbf0	228
72208f03 JG	229	clock_value = bt_ctf_event_get_clock_value(event, clock);
72208f03 JG	230	if (!clock_value) {
55595636 MD	231	printf_error("Failed to retrieve clock value");
55595636 MD	232	ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
72208f03	233	goto end;
44d3cbf0	234	}
72208f03 JG	235
	236	clock_ret = bt_ctf_clock_value_get_value_ns_from_epoch(
	237	clock_value, &ts);
	238	if (clock_ret) {
55595636 MD	239	printf_error("Failed to retrieve clock value timestamp");
55595636 MD	240	ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
44d3cbf0 JG	241	goto end;
44d3cbf0 JG	242	}
55595636 MD	243	if (update_lazy_bound(begin, "begin", ts, &lazy_update)) {
55595636 MD	244	ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
528debdf MD	245	goto end;
528debdf MD	246	}
55595636 MD	247	if (update_lazy_bound(end, "end", ts, &lazy_update)) {
55595636 MD	248	ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
528debdf MD	249	goto end;
528debdf MD	250	}
55595636 MD	251	if (lazy_update && begin->set && end->set) {
	252	if (begin->value > end->value) {
	253	printf_error("Unexpected: time range begin value is above end value");
	254	ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
	255	goto end;
	256	}
	257	}
72208f03 JG	258	if (begin->set && ts < begin->value) {
	259	in_range = false;
	260	}
	261	if (end->set && ts > end->value) {
	262	in_range = false;
	263	}
44d3cbf0	264	end:
72208f03 JG	265	bt_put(event);
	266	bt_put(clock);
	267	bt_put(trace);
	268	bt_put(stream);
	269	bt_put(stream_class);
	270	bt_put(clock_value);
55595636 MD	271	*_event_in_range = in_range;
55595636 MD	272	return ret;
44d3cbf0 JG	273	}
44d3cbf0 JG	274
44d3cbf0	275	static
72208f03	276	int ns_from_integer_field(struct bt_ctf_field integer, int64_t ns)
44d3cbf0	277	{
72208f03 JG	278	int ret = 0;
	279	int is_signed;
	280	uint64_t raw_clock_value;
	281	struct bt_ctf_field_type *integer_type = NULL;
44d3cbf0	282	struct bt_ctf_clock *clock = NULL;
44d3cbf0 JG	283	struct bt_ctf_clock_value *clock_value = NULL;
44d3cbf0 JG	284
72208f03 JG	285	integer_type = bt_ctf_field_get_type(integer);
	286	assert(integer_type);
	287	clock = bt_ctf_field_type_integer_get_mapped_clock(integer_type);
	288	if (!clock) {
	289	ret = -1;
44d3cbf0 JG	290	goto end;
	291	}
	292
72208f03 JG	293	is_signed = bt_ctf_field_type_integer_get_signed(integer_type);
	294	if (!is_signed) {
	295	ret = bt_ctf_field_unsigned_integer_get_value(integer,
	296	&raw_clock_value);
	297	if (ret) {
44d3cbf0 JG	298	goto end;
44d3cbf0 JG	299	}
72208f03 JG	300	} else {
	301	/* Signed clock values are unsupported. */
	302	goto end;
	303	}
44d3cbf0	304
72208f03 JG	305	clock_value = bt_ctf_clock_value_create(clock, raw_clock_value);
	306	if (!clock_value) {
	307	goto end;
	308	}
44d3cbf0	309
72208f03 JG	310	ret = bt_ctf_clock_value_get_value_ns_from_epoch(clock_value, ns);
	311	end:
	312	bt_put(integer_type);
	313	bt_put(clock);
	314	bt_put(clock_value);
	315	return ret;
	316	}
44d3cbf0	317
72208f03	318	static
55595636 MD	319	enum bt_notification_iterator_status evaluate_packet_notification(
	320	struct bt_notification *notification,
	321	struct trimmer_bound begin, struct trimmer_bound end,
	322	bool *_packet_in_range)
72208f03	323	{
55595636 MD	324	enum bt_notification_iterator_status ret =
55595636 MD	325	BT_NOTIFICATION_ITERATOR_STATUS_OK;
72208f03 JG	326	int64_t begin_ns, pkt_begin_ns, end_ns, pkt_end_ns;
	327	bool in_range = true;
	328	struct bt_ctf_packet *packet = NULL;
	329	struct bt_ctf_field *packet_context = NULL,
	330	*timestamp_begin = NULL,
	331	*timestamp_end = NULL;
	332
	333	switch (bt_notification_get_type(notification)) {
	334	case BT_NOTIFICATION_TYPE_PACKET_BEGIN:
	335	packet = bt_notification_packet_begin_get_packet(notification);
44d3cbf0	336	break;
72208f03 JG	337	case BT_NOTIFICATION_TYPE_PACKET_END:
	338	packet = bt_notification_packet_end_get_packet(notification);
	339	break;
	340	default:
55595636	341	break;
44d3cbf0	342	}
72208f03 JG	343	assert(packet);
	344
	345	packet_context = bt_ctf_packet_get_context(packet);
	346	if (!packet_context) {
	347	goto end;
	348	}
	349
	350	if (!bt_ctf_field_is_structure(packet_context)) {
	351	goto end;
	352	}
	353
	354	timestamp_begin = bt_ctf_field_structure_get_field(
	355	packet_context, "timestamp_begin");
	356	if (!timestamp_begin \|\| !bt_ctf_field_is_integer(timestamp_begin)) {
	357	goto end;
	358	}
	359	timestamp_end = bt_ctf_field_structure_get_field(
	360	packet_context, "timestamp_end");
	361	if (!timestamp_end \|\| !bt_ctf_field_is_integer(timestamp_end)) {
	362	goto end;
	363	}
	364
55595636	365	if (ns_from_integer_field(timestamp_begin, &pkt_begin_ns)) {
72208f03 JG	366	goto end;
72208f03 JG	367	}
55595636	368	if (ns_from_integer_field(timestamp_end, &pkt_end_ns)) {
72208f03 JG	369	goto end;
	370	}
	371
	372	begin_ns = begin->set ? begin->value : INT64_MIN;
	373	end_ns = end->set ? end->value : INT64_MAX;
	374
	375	/*
	376	* Accept if there is any overlap between the selected region and the
	377	* packet.
	378	*/
	379	in_range = (pkt_end_ns >= begin_ns) && (pkt_begin_ns <= end_ns);
	380	end:
55595636	381	*_packet_in_range = in_range;
72208f03 JG	382	bt_put(packet);
	383	bt_put(packet_context);
	384	bt_put(timestamp_begin);
	385	bt_put(timestamp_end);
55595636	386	return ret;
72208f03 JG	387	}
	388
	389	/* Return true if the notification should be forwarded. */
	390	static
55595636 MD	391	enum bt_notification_iterator_status evaluate_notification(
	392	struct bt_notification *notification,
	393	struct trimmer_bound begin, struct trimmer_bound end,
	394	bool *in_range)
72208f03	395	{
72208f03	396	enum bt_notification_type type;
55595636 MD	397	enum bt_notification_iterator_status ret =
55595636 MD	398	BT_NOTIFICATION_ITERATOR_STATUS_OK;
72208f03 JG	399
	400	type = bt_notification_get_type(notification);
	401	switch (type) {
	402	case BT_NOTIFICATION_TYPE_EVENT:
55595636 MD	403	ret = evaluate_event_notification(notification, begin,
55595636 MD	404	end, in_range);
72208f03 JG	405	break;
	406	case BT_NOTIFICATION_TYPE_PACKET_BEGIN:
	407	case BT_NOTIFICATION_TYPE_PACKET_END:
55595636 MD	408	ret = evaluate_packet_notification(notification, begin,
55595636 MD	409	end, in_range);
72208f03	410	break;
44d3cbf0	411	default:
72208f03	412	/* Accept all other notifications. */
44d3cbf0 JG	413	break;
44d3cbf0 JG	414	}
55595636	415	return ret;
cab3f160 JG	416	}
	417
	418	BT_HIDDEN
	419	enum bt_notification_iterator_status trimmer_iterator_next(
	420	struct bt_notification_iterator *iterator)
	421	{
44d3cbf0 JG	422	struct trimmer_iterator *trim_it = NULL;
	423	struct bt_component *component = NULL;
	424	struct trimmer *trimmer = NULL;
	425	struct bt_notification_iterator *source_it = NULL;
	426	enum bt_notification_iterator_status ret =
55595636	427	BT_NOTIFICATION_ITERATOR_STATUS_OK;
44d3cbf0	428	enum bt_component_status component_ret;
72208f03	429	bool notification_in_range = false;
cab3f160	430
44d3cbf0 JG	431	trim_it = bt_notification_iterator_get_private_data(iterator);
	432	assert(trim_it);
	433
	434	component = bt_notification_iterator_get_component(iterator);
	435	assert(component);
	436	trimmer = bt_component_get_private_data(component);
	437	assert(trimmer);
	438
	439	/* FIXME, should handle input iterator groups. */
	440	component_ret = bt_component_filter_get_input_iterator(component, 0,
	441	&source_it);
	442	assert((component_ret == BT_COMPONENT_STATUS_OK) && source_it);
	443
72208f03	444	while (!notification_in_range) {
44d3cbf0 JG	445	struct bt_notification *notification;
	446
	447	ret = bt_notification_iterator_next(source_it);
	448	if (ret != BT_NOTIFICATION_ITERATOR_STATUS_OK) {
	449	goto end;
	450	}
	451
	452	notification = bt_notification_iterator_get_notification(
	453	source_it);
	454	if (!notification) {
	455	ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
	456	goto end;
	457	}
	458
55595636 MD	459	ret = evaluate_notification(notification,
	460	&trimmer->begin, &trimmer->end,
	461	&notification_in_range);
72208f03	462	if (notification_in_range) {
44d3cbf0 JG	463	BT_MOVE(trim_it->current_notification, notification);
	464	} else {
	465	bt_put(notification);
	466	}
	467	}
cab3f160	468	end:
44d3cbf0 JG	469	bt_put(source_it);
44d3cbf0 JG	470	bt_put(component);
cab3f160 JG	471	return ret;
	472	}
	473
	474	BT_HIDDEN
	475	enum bt_notification_iterator_status trimmer_iterator_seek_time(
	476	struct bt_notification_iterator *iterator, int64_t time)
	477	{
	478	enum bt_notification_iterator_status ret;
	479
	480	ret = BT_NOTIFICATION_ITERATOR_STATUS_OK;
	481	end:
	482	return ret;
	483	}