Convert LTTngTop to C++ and state system

[lttngtop.git] / src / cputop.cpp

diff --git a/src/cputop.cpp b/src/cputop.cpp
new file mode 100644 (file)
index 0000000..4fde9a8
--- /dev/null
+++ b/src/cputop.cpp
@@ -0,0 +1,169 @@
+/*
+ * Copyright (C) 2011-2012 Julien Desfossez
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License Version 2 as
+ * published by the Free Software Foundation;
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+extern "C" {
+#include <babeltrace/babeltrace.h>
+}
+
+#include "lttngtoptypes.h"
+#include "common.h"
+#include "cputop.h"
+
+void update_cputop_data(unsigned long timestamp, int64_t cpu,
+               int prev_pid, int next_pid, char *prev_comm, char *next_comm)
+{
+       Quark cpu_quark;
+       Quark current_task_quark;
+       bool current_task_found;
+       int current_task_pid;
+       int current_task_tid;
+       unsigned long elapsed;
+       unsigned long task_start;
+       bool current_task_threadparent_found;
+       Quark current_task_threadparent;
+
+       cpu_quark = get_cpu(cpu, timestamp);
+       current_task_found = get_current_attribute_value_quark(&cpu_quark,
+               "current_task", current_task_quark);
+
+       if (current_task_found) {
+               get_current_attribute_value_int(&current_task_quark, "pid",
+                       current_task_pid);
+               if (current_task_pid == prev_pid) {
+                       get_current_attribute_value_ulong(&cpu_quark,
+                               "task_start", task_start);
+                       elapsed = timestamp - task_start;
+                       increase_attribute(timestamp, &current_task_quark,
+                               "totalcpunsec", elapsed);
+                       increase_attribute(timestamp, &current_task_quark,
+                               "threadstotalcpunsec", elapsed);
+                       get_current_attribute_value_int(&current_task_quark,
+                               "tid", current_task_tid);
+                       if (current_task_tid != current_task_pid) {
+                               current_task_threadparent_found =
+                                       get_current_attribute_value_quark(
+                                               &current_task_quark,
+                                               "threadparent",
+                                               current_task_threadparent);
+                               if (current_task_threadparent_found)
+                                       increase_attribute(timestamp,
+                                               &current_task_threadparent,
+                                               "threadstotalcpuns", elapsed);
+                       }
+               }
+       }
+
+       if (next_pid != 0)
+               modify_attribute(timestamp, &cpu_quark, "current_task",
+                                get_proc(next_pid, next_comm, timestamp));
+       else
+               nullify_attribute(timestamp, &cpu_quark, "current_task");
+
+       modify_attribute(timestamp, &cpu_quark, "task_start", timestamp);
+}
+
+enum bt_cb_ret handle_sched_switch(struct bt_ctf_event *call_data,
+               void *private_data)
+{
+       const struct definition *scope;
+       unsigned long timestamp;
+       uint64_t cpu_id;
+       char *prev_comm, *next_comm;
+       int prev_tid, next_tid;
+
+       timestamp = bt_ctf_get_timestamp(call_data);
+       if (timestamp == -1ULL)
+               goto error;
+
+       scope = bt_ctf_get_top_level_scope(call_data,
+                       BT_EVENT_FIELDS);
+       prev_comm = bt_ctf_get_char_array(bt_ctf_get_field(call_data,
+                               scope, "_prev_comm"));
+       if (bt_ctf_field_get_error()) {
+               fprintf(stderr, "Missing prev_comm context info\n");
+               goto error;
+       }
+
+       next_comm = bt_ctf_get_char_array(bt_ctf_get_field(call_data,
+                               scope, "_next_comm"));
+       if (bt_ctf_field_get_error()) {
+               fprintf(stderr, "Missing next_comm context info\n");
+               goto error;
+       }
+
+       prev_tid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
+                               scope, "_prev_tid"));
+       if (bt_ctf_field_get_error()) {
+               fprintf(stderr, "Missing prev_tid context info\n");
+               goto error;
+       }
+
+       next_tid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
+                               scope, "_next_tid"));
+       if (bt_ctf_field_get_error()) {
+               fprintf(stderr, "Missing next_tid context info\n");
+               goto error;
+       }
+
+       cpu_id = get_cpu_id(call_data);
+
+       update_cputop_data(timestamp, cpu_id, prev_tid, next_tid,
+                       prev_comm, next_comm);
+
+       return BT_CB_OK;
+
+error:
+       return BT_CB_ERROR_STOP;
+}
+
+enum bt_cb_ret handle_sched_process_free(struct bt_ctf_event *call_data,
+               void *private_data)
+{
+       const struct definition *scope;
+       unsigned long timestamp;
+       char *comm;
+       int tid;
+
+       timestamp = bt_ctf_get_timestamp(call_data);
+       if (timestamp == -1ULL)
+               goto error;
+
+       scope = bt_ctf_get_top_level_scope(call_data,
+                       BT_EVENT_FIELDS);
+       comm = bt_ctf_get_char_array(bt_ctf_get_field(call_data,
+                               scope, "_comm"));
+       if (bt_ctf_field_get_error()) {
+               fprintf(stderr, "Missing procname context info\n");
+               goto error;
+       }
+
+       tid = bt_ctf_get_int64(bt_ctf_get_field(call_data,
+                               scope, "_tid"));
+       if (bt_ctf_field_get_error()) {
+               fprintf(stderr, "Missing tid field\n");
+               goto error;
+       }
+
+       death_proc(tid, comm, timestamp);
+
+       return BT_CB_OK;
+
+error:
+       return BT_CB_ERROR_STOP;
+
+}
+