From: Jérémie Galarneau <jeremie.galarneau@efficios.com>
Date: Tue, 28 Apr 2015 19:35:07 +0000 (-0400)
Subject: ir: add bt_ctf_event_set_payload_field()
X-Git-Tag: v2.0.0-pre1~1255
X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=commitdiff_plain;h=e5e6eb3ab98b9595fa7e58cda439c1ac225a4c48

ir: add bt_ctf_event_set_payload_field()

Signed-off-by: Jérémie Galarneau <jeremie.galarneau@efficios.com>
---

diff --git a/formats/ctf/ir/event.c b/formats/ctf/ir/event.c
index d5c66c4d..7b8844be 100644
--- a/formats/ctf/ir/event.c
+++ b/formats/ctf/ir/event.c
@@ -718,6 +718,41 @@ end:
 	return payload;
 }
 
+int bt_ctf_event_set_payload_field(struct bt_ctf_event *event,
+		struct bt_ctf_field *payload)
+{
+	int ret = 0;
+	struct bt_ctf_field_type *payload_type = NULL;
+
+	if (!event || !payload) {
+		ret = -1;
+		goto end;
+	}
+
+	payload_type = bt_ctf_field_get_type(payload);
+	if (!payload_type) {
+		ret = -1;
+		goto end;
+	}
+
+	if (bt_ctf_field_type_get_type_id(payload_type) != CTF_TYPE_STRUCT) {
+		ret = -1;
+		goto end;
+	}
+
+	bt_ctf_field_get(payload);
+	if (event->fields_payload) {
+		bt_ctf_field_put(event->fields_payload);
+	}
+	event->fields_payload = payload;
+
+end:
+	if (payload_type) {
+		bt_ctf_field_type_put(payload_type);
+	}
+	return ret;
+}
+
 struct bt_ctf_field *bt_ctf_event_get_payload(struct bt_ctf_event *event,
 		const char *name)
 {
diff --git a/include/babeltrace/ctf-ir/event.h b/include/babeltrace/ctf-ir/event.h
index 2f1ca8c8..09ada907 100644
--- a/include/babeltrace/ctf-ir/event.h
+++ b/include/babeltrace/ctf-ir/event.h
@@ -377,6 +377,17 @@ extern struct bt_ctf_clock *bt_ctf_event_get_clock(
 extern struct bt_ctf_field *bt_ctf_event_get_payload_field(
 		struct bt_ctf_event *event);
 
+/*
+ * bt_ctf_event_set_payload_field: set an event's payload.
+ *
+ * @param event Event instance.
+ * @param payload Field instance (must be a structure).
+ *
+ * Returns 0 on success, a negative value on error.
+ */
+extern int bt_ctf_event_set_payload_field(struct bt_ctf_event *event,
+		struct bt_ctf_field *payload);
+
 /*
  * bt_ctf_event_get_payload: get an event's field.
  *