/*
 * Copyright 2010, Intel Corporation
 *
 * This file is part of PowerTOP
 *
 * This program file is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation; version 2 of the License.
 *
 * 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 in a file named COPYING; if not, write to the
 * Free Software Foundation, Inc,
 * 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301 USA
 * or just google for it.
 *
 * Authors:
 *	Arjan van de Ven <arjan@linux.intel.com>
 */



#include "process.h"
#include <stdio.h>
#include <unistd.h>
#include <errno.h>

#include <iostream>
#include <fstream>
#include <sstream>
#include <algorithm>
#include <iterator>
#include "../lib.h"


std::vector<std::unique_ptr<process>> all_processes;

void process::account_disk_dirty(void)
{
	disk_hits++;
}

void process::schedule_thread(uint64_t time, [[maybe_unused]] int thread_id)
{
	running_since = time;
	running = 1;
}


uint64_t process::deschedule_thread(uint64_t time, int thread_id)
{
	uint64_t delta = time - running_since;

	if (!running_since)
		return 0;

	if (time < running_since) {
		printf("%llu time    %llu since \n", (unsigned long long)time,
						     (unsigned long long)running_since);
		running_since = 0;
		return 0;
	}

	if (thread_id == 0) /* idle thread */
		delta = 0;
	accumulated_runtime += delta;
	running = 0;

	return delta;
}

static void cmdline_to_string(std::string& str)
{
	std::replace(str.begin(), str.end(), '\0', ' ');
}


process::process(const std::string &_comm, int _pid, int _tid) : power_consumer()
{
	comm = _comm;
	pid = _pid;
	is_idle = 0;
	running = 0;
	last_waker = nullptr;
	waker = nullptr;
	is_kernel = 0;
	tgid = _tid;

	if (_tid == 0) {
		const std::string content = read_file_content(std::format("/proc/{}/status", _pid));
		if (!content.empty()) {
			std::istringstream stream(content);
			std::string line;
			while (std::getline(stream, line)) {
				if (line.starts_with("Tgid:")) {
					const size_t pos = line.find(':');
					if (pos != std::string::npos) {
						try {
							tgid = std::stoi(line.substr(pos + 1));
						} catch (...) {}
						break;
					}
				}
			}
		}
	}

	if (comm.starts_with("kondemand/"))
		is_idle = 1;

	desc = pt_format(_("[PID {}] {}"), pid, comm);

	std::string cmdline = read_file_content(std::format("/proc/{}/cmdline", _pid));
	if (cmdline.empty()) {
		is_kernel = 1;
		desc = pt_format(_("[PID {}] [{}]"), pid, comm);
	} else {
		cmdline_to_string(cmdline);
		desc = pt_format(_("[PID {}] {}"), pid, cmdline);
	}
}

std::string process::description(void)
{

	if (child_runtime > accumulated_runtime)
		child_runtime = 0;

	return desc;
}

double process::usage_summary(void) const
{
	const double t = (accumulated_runtime - child_runtime) / 1000000.0 / measurement_time / 10;
	return t;
}

std::string process::usage_units_summary(void) const
{
	return "%";
}

class process * find_create_process(const std::string &comm, int pid)
{
	for (auto &p : all_processes)
		if (p->pid == pid && p->comm == comm)
			return p.get();

	all_processes.push_back(std::make_unique<process>(comm, pid));
	return all_processes.back().get();
}

static void merge_process(process &one, process &two)
{
	one.accumulated_runtime += two.accumulated_runtime;
	one.child_runtime += two.child_runtime;
	one.wake_ups += two.wake_ups;
	one.disk_hits += two.disk_hits;
	one.hard_disk_hits += two.hard_disk_hits;
	one.xwakes += two.xwakes;
	one.gpu_ops += two.gpu_ops;
	one.power_charge += two.power_charge;
}


void merge_processes(void)
{
	for (auto it1 = all_processes.begin(); it1 != all_processes.end(); ++it1) {
		for (auto it2 = it1 + 1; it2 != all_processes.end(); ) {
			/* fold threads */
			if ((*it1)->pid == (*it2)->tgid && (*it2)->tgid != 0) {
				merge_process(**it1, **it2);
				it2 = all_processes.erase(it2);
				continue;
			}
			/* find dupes and add up */
			if ((*it1)->desc == (*it2)->desc) {
				merge_process(**it1, **it2);
				it2 = all_processes.erase(it2);
				continue;
			}
			++it2;
		}
	}
}

void all_processes_to_all_power(void)
{
	for (auto &p : all_processes)
		if (p->accumulated_runtime || p->power_charge)
			all_power.push_back(p.get());
}

void clear_processes(void)
{
	all_processes.clear();
}

void process::collect_json_fields(std::string &_js) const
{
    power_consumer::collect_json_fields(_js);
    JSON_FIELD(running_since);
    JSON_FIELD(desc);
    JSON_FIELD(tgid);
    JSON_FIELD(comm);
    JSON_FIELD(pid);
    JSON_FIELD(is_idle);
    JSON_FIELD(running);
    JSON_FIELD(is_kernel);
}
