4 * Babeltrace Copy Trace Structure
6 * Copyright 2017 Julien Desfossez <jdesfossez@efficios.com>
8 * Author: Julien Desfossez <jdesfossez@efficios.com>
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:
17 * The above copyright notice and this permission notice shall be included in
18 * all copies or substantial portions of the Software.
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
29 #define BT_LOG_TAG "PLUGIN-UTILS-TRIMMER-FLT-COPY"
33 #include <babeltrace/ctf-ir/event.h>
34 #include <babeltrace/ctf-ir/packet.h>
35 #include <babeltrace/ctf-ir/event-class.h>
36 #include <babeltrace/ctf-ir/stream.h>
37 #include <babeltrace/ctf-ir/stream-class.h>
38 #include <babeltrace/ctf-ir/clock-class.h>
39 #include <babeltrace/ctf-ir/fields.h>
40 #include <babeltrace/ctf-writer/stream-class.h>
41 #include <babeltrace/ctf-writer/stream.h>
43 #include <ctfcopytrace.h>
47 struct bt_ctf_packet
*lookup_packet(struct trimmer_iterator
*trim_it
,
48 struct bt_ctf_packet
*packet
)
50 return (struct bt_ctf_packet
*) g_hash_table_lookup(
56 struct bt_ctf_packet
*insert_new_packet(struct trimmer_iterator
*trim_it
,
57 struct bt_ctf_packet
*packet
,
58 struct bt_ctf_stream
*stream
)
60 struct bt_ctf_packet
*writer_packet
= NULL
;
63 BT_LOGD_STR("Inserting a new packet.");
64 writer_packet
= bt_ctf_packet_create(stream
);
66 BT_LOGE_STR("Failed to create a new packet.");
70 ret
= ctf_packet_copy_header(trim_it
->err
, packet
, writer_packet
);
72 BT_LOGE_STR("Failed to copy packet header.");
76 g_hash_table_insert(trim_it
->packet_map
, (gpointer
) packet
,
81 BT_PUT(writer_packet
);
87 enum bt_component_status
update_packet_context_field(FILE *err
,
88 struct bt_ctf_packet
*writer_packet
,
89 const char *name
, int64_t value
)
91 enum bt_component_status ret
;
92 struct bt_ctf_field
*packet_context
= NULL
, *writer_packet_context
= NULL
;
93 struct bt_ctf_field_type
*struct_type
= NULL
, *field_type
= NULL
;
94 struct bt_ctf_field
*field
= NULL
, *writer_field
= NULL
;
95 int nr_fields
, i
, int_ret
;
97 BT_LOGD("Updating packet context field: name=%s", name
);
98 packet_context
= bt_ctf_packet_get_context(writer_packet
);
99 assert(packet_context
);
101 struct_type
= bt_ctf_field_get_type(packet_context
);
104 writer_packet_context
= bt_ctf_packet_get_context(writer_packet
);
105 assert(writer_packet_context
);
107 nr_fields
= bt_ctf_field_type_structure_get_field_count(struct_type
);
108 for (i
= 0; i
< nr_fields
; i
++) {
109 const char *field_name
;
111 field
= bt_ctf_field_structure_get_field_by_index(
114 BT_LOGE("Failed to get field in packet-context: field-name=\"%s\"",
118 if (bt_ctf_field_type_structure_get_field(struct_type
,
119 &field_name
, &field_type
, i
) < 0) {
120 BT_LOGE("Failed to get field: field-name=\"%s\"",
124 if (strcmp(field_name
, name
)) {
129 if (bt_ctf_field_type_get_type_id(field_type
) !=
130 BT_CTF_FIELD_TYPE_ID_INTEGER
) {
131 BT_LOGE("Expecting an integer for this field: field-name=\"%s\"",
136 writer_field
= bt_ctf_field_structure_get_field(writer_packet_context
,
138 assert(writer_field
);
140 int_ret
= bt_ctf_field_unsigned_integer_set_value(writer_field
, value
);
141 assert(int_ret
== 0);
143 BT_PUT(writer_field
);
148 ret
= BT_COMPONENT_STATUS_OK
;
152 bt_put(writer_field
);
155 ret
= BT_COMPONENT_STATUS_ERROR
;
158 bt_put(packet_context
);
163 struct bt_ctf_packet
*trimmer_new_packet(
164 struct trimmer_iterator
*trim_it
,
165 struct bt_ctf_packet
*packet
)
167 struct bt_ctf_stream
*stream
= NULL
;
168 struct bt_ctf_packet
*writer_packet
= NULL
;
171 stream
= bt_ctf_packet_get_stream(packet
);
175 * If a packet was already opened, close it and remove it from
178 writer_packet
= lookup_packet(trim_it
, packet
);
180 g_hash_table_remove(trim_it
->packet_map
, packet
);
181 BT_PUT(writer_packet
);
184 writer_packet
= insert_new_packet(trim_it
, packet
, stream
);
185 if (!writer_packet
) {
186 BT_LOGE_STR("Failed to insert new packet.");
189 bt_get(writer_packet
);
191 int_ret
= ctf_packet_copy_context(trim_it
->err
, packet
,
192 stream
, writer_packet
);
194 BT_LOGE_STR("Failed to copy packet context.");
201 BT_PUT(writer_packet
);
204 return writer_packet
;
208 struct bt_ctf_packet
*trimmer_close_packet(
209 struct trimmer_iterator
*trim_it
,
210 struct bt_ctf_packet
*packet
)
212 struct bt_ctf_packet
*writer_packet
= NULL
;
214 writer_packet
= lookup_packet(trim_it
, packet
);
215 if (!writer_packet
) {
216 BT_LOGE_STR("Failed to find existing packet.");
220 g_hash_table_remove(trim_it
->packet_map
, packet
);
223 return writer_packet
;
227 struct bt_ctf_event
*trimmer_output_event(
228 struct trimmer_iterator
*trim_it
,
229 struct bt_ctf_event
*event
)
231 struct bt_ctf_event_class
*event_class
= NULL
;
232 struct bt_ctf_event
*writer_event
= NULL
;
233 struct bt_ctf_packet
*packet
= NULL
, *writer_packet
= NULL
;
234 const char *event_name
;
237 event_class
= bt_ctf_event_get_class(event
);
240 event_name
= bt_ctf_event_class_get_name(event_class
);
242 writer_event
= ctf_copy_event(trim_it
->err
, event
, event_class
, false);
244 BT_LOGE("Failed to copy event: event-class-name=\"%s\", event-name=\"%s\"",
245 bt_ctf_event_class_get_name(event_class
),
250 packet
= bt_ctf_event_get_packet(event
);
253 writer_packet
= lookup_packet(trim_it
, packet
);
254 if (!writer_packet
) {
255 BT_LOGE_STR("Failed to find existing packet.");
258 bt_get(writer_packet
);
260 int_ret
= bt_ctf_event_set_packet(writer_event
, writer_packet
);
262 BT_LOGE("Failed to append event: event-class-name=\"%s\", event-name=\"%s\"",
263 bt_ctf_event_class_get_name(event_class
),
268 /* We keep the reference on the writer_event to create a notification. */
272 BT_PUT(writer_event
);
274 bt_put(writer_packet
);