X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=lib%2Fiterator.c;h=0af151402b429d3fe95d9c92c42e9e07eb9436f5;hp=d595e7617a82b4998d3bf82c6cd14eb098711520;hb=48d5f76c4831624cb4332c7707c2615c2474284c;hpb=d6425aaf6026d168a695268c4aa7539956c0d0ca diff --git a/lib/iterator.c b/lib/iterator.c index d595e761..0af15140 100644 --- a/lib/iterator.c +++ b/lib/iterator.c @@ -20,13 +20,12 @@ #include #include -#include #include #include -#include #include #include #include +#include #include #include @@ -111,13 +110,10 @@ static int seek_file_stream_by_timestamp(struct ctf_file_stream *cfs, for (i = 0; i < stream_pos->packet_index->len; i++) { index = &g_array_index(stream_pos->packet_index, struct packet_index, i); - if (index->timestamp_begin >= timestamp || - index->timestamp_end <= timestamp) + if (index->timestamp_end <= timestamp) continue; - stream_pos->cur_index = i; - stream_pos->move_pos_slow(&stream_pos->parent, - index->offset, SEEK_SET); + stream_pos->packet_seek(&stream_pos->parent, i, SEEK_SET); while (cfs->parent.timestamp < timestamp) { ret = stream_read_event(cfs); if (ret < 0) @@ -180,7 +176,7 @@ int bt_iter_set_pos(struct bt_iter *iter, const struct bt_iter_pos *iter_pos) switch (iter_pos->type) { case BT_SEEK_RESTORE: if (!iter_pos->u.restore) - goto error_arg; + return -EINVAL; heap_free(iter->stream_heap); ret = heap_init(iter->stream_heap, 0, stream_compare); @@ -191,7 +187,6 @@ int bt_iter_set_pos(struct bt_iter *iter, const struct bt_iter_pos *iter_pos) i++) { struct stream_saved_pos *saved_pos; struct ctf_stream_pos *stream_pos; - struct packet_index *index; struct ctf_stream *stream; saved_pos = &g_array_index( @@ -199,17 +194,13 @@ int bt_iter_set_pos(struct bt_iter *iter, const struct bt_iter_pos *iter_pos) struct stream_saved_pos, i); stream = &saved_pos->file_stream->parent; stream_pos = &saved_pos->file_stream->pos; - index = &g_array_index(stream_pos->packet_index, - struct packet_index, - saved_pos->cur_index); - stream_pos->cur_index = saved_pos->cur_index; - stream_pos->move_pos_slow(&stream_pos->parent, - index->offset, SEEK_SET); + stream_pos->packet_seek(&stream_pos->parent, + saved_pos->cur_index, SEEK_SET); /* * the timestamp needs to be restored after - * move_pos_slow, because this function resets + * packet_seek, because this function resets * the timestamp to the beginning of the packet */ stream->timestamp = saved_pos->current_timestamp; @@ -236,9 +227,6 @@ int bt_iter_set_pos(struct bt_iter *iter, const struct bt_iter_pos *iter_pos) case BT_SEEK_TIME: tc = iter->ctx->tc; - if (!iter_pos->u.seek_time) - goto error_arg; - heap_free(iter->stream_heap); ret = heap_init(iter->stream_heap, 0, stream_compare); if (ret < 0) @@ -261,13 +249,11 @@ int bt_iter_set_pos(struct bt_iter *iter, const struct bt_iter_pos *iter_pos) return 0; default: /* not implemented */ - goto error_arg; + return -EINVAL; } return 0; -error_arg: - ret = -EINVAL; error: heap_free(iter->stream_heap); if (heap_init(iter->stream_heap, 0, stream_compare) < 0) { @@ -387,7 +373,7 @@ static int babeltrace_filestream_seek(struct ctf_file_stream *file_stream, */ break; case BT_SEEK_BEGIN: - file_stream->pos.move_pos_slow(&file_stream->pos.parent, + file_stream->pos.packet_seek(&file_stream->pos.parent, 0, SEEK_SET); ret = stream_read_event(file_stream); break; @@ -442,21 +428,16 @@ end: return ret; } -struct bt_iter *bt_iter_create(struct bt_context *ctx, +int bt_iter_init(struct bt_iter *iter, + struct bt_context *ctx, struct bt_iter_pos *begin_pos, struct bt_iter_pos *end_pos) { int i, stream_id; int ret = 0; - struct bt_iter *iter; - iter = g_new0(struct bt_iter, 1); iter->stream_heap = g_new(struct ptr_heap, 1); iter->end_pos = end_pos; - iter->callbacks = g_array_new(0, 1, sizeof(struct bt_stream_callbacks)); - iter->recalculate_dep_graph = 0; - iter->main_callbacks.callback = NULL; - iter->dep_gc = g_ptr_array_new(); bt_context_get(ctx); iter->ctx = ctx; @@ -505,49 +486,43 @@ struct bt_iter *bt_iter_create(struct bt_context *ctx, } } - return iter; + return 0; error: heap_free(iter->stream_heap); error_heap_init: g_free(iter->stream_heap); - g_free(iter); - return NULL; + return ret; } -void bt_iter_destroy(struct bt_iter *iter) +struct bt_iter *bt_iter_create(struct bt_context *ctx, + struct bt_iter_pos *begin_pos, + struct bt_iter_pos *end_pos) { - struct bt_stream_callbacks *bt_stream_cb; - struct bt_callback_chain *bt_chain; - int i, j; + struct bt_iter *iter; + int ret; + + iter = g_new0(struct bt_iter, 1); + ret = bt_iter_init(iter, ctx, begin_pos, end_pos); + if (ret) { + g_free(iter); + return NULL; + } + return iter; +} +void bt_iter_fini(struct bt_iter *iter) +{ if (iter->stream_heap) { heap_free(iter->stream_heap); g_free(iter->stream_heap); } - - /* free all events callbacks */ - if (iter->main_callbacks.callback) - g_array_free(iter->main_callbacks.callback, TRUE); - - /* free per-event callbacks */ - for (i = 0; i < iter->callbacks->len; i++) { - bt_stream_cb = &g_array_index(iter->callbacks, - struct bt_stream_callbacks, i); - if (!bt_stream_cb || !bt_stream_cb->per_id_callbacks) - continue; - for (j = 0; j < bt_stream_cb->per_id_callbacks->len; j++) { - bt_chain = &g_array_index(bt_stream_cb->per_id_callbacks, - struct bt_callback_chain, j); - if (bt_chain->callback) { - g_array_free(bt_chain->callback, TRUE); - } - } - g_array_free(bt_stream_cb->per_id_callbacks, TRUE); - } - bt_context_put(iter->ctx); +} +void bt_iter_destroy(struct bt_iter *iter) +{ + bt_iter_fini(iter); g_free(iter); } @@ -579,28 +554,3 @@ int bt_iter_next(struct bt_iter *iter) end: return ret; } - -struct bt_ctf_event *bt_iter_read_ctf_event(struct bt_iter *iter) -{ - struct ctf_file_stream *file_stream; - struct bt_ctf_event *ret = &iter->current_ctf_event; - - file_stream = heap_maximum(iter->stream_heap); - if (!file_stream) { - /* end of file for all streams */ - goto stop; - } - ret->stream = &file_stream->parent; - ret->event = g_ptr_array_index(ret->stream->events_by_id, - ret->stream->event_id); - - if (ret->stream->stream_id > iter->callbacks->len) - goto end; - - process_callbacks(iter, ret->stream); - -end: - return ret; -stop: - return NULL; -}