Convert LTTngTop to C++ and state system

[lttngtop.git] / src / cputop.cpp

/*
 * 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;

}