/* Copyright (c) 2021-2024 Barcelona Supercomputing Center (BSC)
 * SPDX-License-Identifier: GPL-3.0-or-later */

#include "thread.h"
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "bay.h"
#include "cpu.h"
#include "emu_prv.h"
#include "mux.h"
#include "path.h"
#include "pv/pcf.h"
#include "pv/prv.h"
#include "pv/pvt.h"
#include "recorder.h"
#include "stream.h"
#include "value.h"
struct proc;

static const char *chan_name[TH_CHAN_MAX] = {
	[TH_CHAN_CPU]   = "cpu_gindex",
	[TH_CHAN_TID]   = "tid_active",
	[TH_CHAN_STATE] = "state",
};

static const int chan_type[TH_CHAN_MAX] = {
	[TH_CHAN_CPU]   = PRV_THREAD_CPU,
	[TH_CHAN_TID]   = PRV_THREAD_TID,
	[TH_CHAN_STATE] = PRV_THREAD_STATE,
};

static const long prv_flags[TH_CHAN_MAX] = {
	/* Add one to the zero-based cpu gindex */
	[TH_CHAN_CPU] = PRV_NEXT,

	/* FIXME: Only needed for delayed connect, as the state channel is used
	 * as select in the muxes, which is set to dirty when connecting it. */
	[TH_CHAN_STATE] = PRV_SKIPDUP,
};

static const char *pvt_name[TH_CHAN_MAX] = {
	[TH_CHAN_CPU]   = "Thread: CPU affinity",
	[TH_CHAN_TID]   = "Thread: TID of the ACTIVE thread",
	[TH_CHAN_STATE] = "Thread: thread state",
};

static const struct pcf_value_label state_name[] = {
	{ TH_ST_UNKNOWN, "Unknown" },
	{ TH_ST_RUNNING, "Running" },
	{ TH_ST_PAUSED,  "Paused" },
	{ TH_ST_DEAD,    "Dead" },
	{ TH_ST_COOLING, "Cooling" },
	{ TH_ST_WARMING, "Warming" },
	{ -1, NULL },
};

static const struct pcf_value_label (*pcf_labels[TH_CHAN_MAX])[] = {
	[TH_CHAN_STATE] = &state_name,
};

int
thread_stream_get_tid(struct stream *s)
{
	JSON_Object *meta = stream_metadata(s);

	double tid = json_object_dotget_number(meta, "ovni.tid");

	/* Zero is used for errors, so forbidden for tid too */
	if (tid == 0) {
		err("cannot get attribute ovni.tid for stream: %s",
				s->relpath);
		return -1;
	}

	return (int) tid;
}

int
thread_init_begin(struct thread *thread, int tid)
{
	memset(thread, 0, sizeof(struct thread));

	thread->state = TH_ST_UNKNOWN;
	thread->gindex = -1;
	thread->tid = tid;

	if (snprintf(thread->id, PATH_MAX, "thread.%d", tid) >= PATH_MAX) {
		err("relpath too long");
		return -1;
	}

	return 0;
}

void
thread_set_gindex(struct thread *th, int64_t gindex)
{
	th->gindex = gindex;
}

void
thread_set_proc(struct thread *th, struct proc *proc)
{
	th->proc = proc;
}

int
thread_init_end(struct thread *th)
{
	if (th->gindex < 0) {
		err("gindex not set");
		return -1;
	}

	if (th->meta == NULL) {
		err("missing thread metadata for %s", th->id);
		return -1;
	}

	for (int i = 0; i < TH_CHAN_MAX; i++) {
		chan_init(&th->chan[i], CHAN_SINGLE,
				"thread%"PRIi64".%s", th->gindex, chan_name[i]);
	}

	/* The transition Running -> Cooling causes a duplicate (the thread is
	 * still active) */
	chan_prop_set(&th->chan[TH_CHAN_TID], CHAN_IGNORE_DUP, 1);

	th->is_init = 1;
	return 0;
}

static int
create_values(struct pcf_type *t, int i)
{
	const struct pcf_value_label(*q)[] = pcf_labels[i];

	if (q == NULL)
		return 0;

	for (const struct pcf_value_label *p = *q; p->label != NULL; p++) {
		if (pcf_add_value(t, p->value, p->label) == NULL) {
			err("pcf_add_value failed");
			return -1;
		}
	}

	return 0;
}

static int
create_type(struct pcf *pcf, int i)
{
	long type = chan_type[i];

	if (type == -1)
		return 0;

	const char *label = pvt_name[i];
	struct pcf_type *pcftype = pcf_add_type(pcf, (int) type, label);
	if (pcftype == NULL) {
		err("pcf_add_type failed");
		return -1;
	}

	if (create_values(pcftype, i) != 0) {
		err("create_values failed");
		return -1;
	}

	return 0;
}

int
thread_connect(struct thread *th, struct bay *bay, struct recorder *rec)
{
	if (!th->is_init) {
		err("thread is not initialized");
		return -1;
	}

	/* Get thread prv */
	struct pvt *pvt = recorder_find_pvt(rec, "thread");
	if (pvt == NULL) {
		err("cannot find thread pvt");
		return -1;
	}
	struct prv *prv = pvt_get_prv(pvt);

	for (int i = 0; i < TH_CHAN_MAX; i++) {
		struct chan *c = &th->chan[i];
		if (bay_register(bay, c) != 0) {
			err("bay_register failed");
			return -1;
		}

		long type = chan_type[i];
		long row = (long) th->gindex;
		long flags = prv_flags[i];

		if (prv_register(prv, row, type, bay, c, flags)) {
			err("prv_register failed");
			return -1;
		}
	}

	return 0;
}

int
thread_create_pcf_types(struct pcf *pcf)
{
	for (int i = 0; i < TH_CHAN_MAX; i++) {
		if (create_type(pcf, i) != 0) {
			err("create_type failed");
			return -1;
		}
	}
	return 0;
}

struct pcf_type *
thread_get_affinity_pcf_type(struct pcf *pcf)
{
	int type_id = chan_type[TH_CHAN_CPU];
	return pcf_find_type(pcf, type_id);
}

int
thread_get_tid(struct thread *thread)
{
	return thread->tid;
}

/* Sets the state of the thread and updates the thread tracking channels */
int
thread_set_state(struct thread *th, enum thread_state state)
{
	/* The state must be updated when a cpu is set */
	if (th->cpu == NULL) {
		err("thread %d doesn't have a CPU", th->tid);
		return -1;
	}

	th->state = state;
	th->is_running = (state == TH_ST_RUNNING) ? 1 : 0;
	th->is_active = (state == TH_ST_RUNNING
					|| state == TH_ST_COOLING
					|| state == TH_ST_WARMING)
					? 1
					: 0;

	struct chan *st = &th->chan[TH_CHAN_STATE];
	if (chan_set(st, value_int64(th->state)) != 0) {
		err("chan_set() failed");
		return -1;
	}

	struct value tid_active = value_null();

	if (th->is_active)
		tid_active = value_int64(th->tid);

	if (chan_set(&th->chan[TH_CHAN_TID], tid_active) != 0) {
		err("chan_set() failed");
		return -1;
	}

	return 0;
}

int
thread_select_active(struct mux *mux,
		struct value value,
		struct mux_input **input)
{
	if (value.type == VALUE_NULL) {
		*input = NULL;
		return 0;
	}

	if (value.type != VALUE_INT64) {
		err("expecting NULL or INT64 channel value");
		return -1;
	}

	enum thread_state state = (enum thread_state) value.i;

	if (mux->ninputs != 1) {
		err("mux doesn't have one input but %"PRIi64, mux->ninputs);
		return -1;
	}

	switch (state) {
		case TH_ST_RUNNING:
		case TH_ST_COOLING:
		case TH_ST_WARMING:
			*input = mux_get_input(mux, 0);
			break;
		default:
			*input = NULL;
			break;
	}

	return 0;
}

int
thread_select_running(struct mux *mux,
		struct value value,
		struct mux_input **input)
{
	if (value.type == VALUE_NULL) {
		*input = NULL;
		return 0;
	}

	if (value.type != VALUE_INT64) {
		err("expecting NULL or INT64 channel value");
		return -1;
	}

	enum thread_state state = (enum thread_state) value.i;

	if (mux->ninputs != 1) {
		err("mux doesn't have one input but %"PRIi64, mux->ninputs);
		return -1;
	}

	switch (state) {
		case TH_ST_RUNNING:
			*input = mux_get_input(mux, 0);
			break;
		default:
			*input = NULL;
			break;
	}

	return 0;
}

int
thread_set_cpu(struct thread *th, struct cpu *cpu)
{
	if (cpu == NULL) {
		err("CPU is NULL");
		return -1;
	}

	if (th->cpu != NULL) {
		err("thread %d already has a CPU", th->tid);
		return -1;
	}

	dbg("thread%"PRIi64" sets cpu%"PRIi64, th->gindex, cpu->gindex);
	th->cpu = cpu;

	/* Update cpu channel */
	struct chan *c = &th->chan[TH_CHAN_CPU];
	if (chan_set(c, value_int64(cpu->gindex)) != 0) {
		err("chan_set failed");
		return -1;
	}

	return 0;
}

int
thread_unset_cpu(struct thread *th)
{
	if (th->cpu == NULL) {
		err("thread %d doesn't have a CPU", th->tid);
		return -1;
	}

	dbg("thread%"PRIi64" unsets cpu", th->gindex);
	th->cpu = NULL;

	struct chan *c = &th->chan[TH_CHAN_CPU];
	if (chan_set(c, value_null()) != 0) {
		err("chan_set failed");
		return -1;
	}

	return 0;
}

int
thread_migrate_cpu(struct thread *th, struct cpu *cpu)
{
	if (th->cpu == NULL) {
		err("thread %d doesn't have a CPU", th->tid);
		return -1;
	}

	dbg("thread%"PRIi64" migrates to cpu%"PRIi64, th->gindex, cpu->gindex);
	th->cpu = cpu;

	struct chan *c = &th->chan[TH_CHAN_CPU];
	if (chan_set(c, value_int64(cpu->gindex)) != 0) {
		err("chan_set failed");
		return -1;
	}

	return 0;
}

int
thread_load_metadata(struct thread *thread, struct stream *s)
{
	JSON_Object *meta = stream_metadata(s);

	if (thread->meta != NULL) {
		err("thread %s already loaded metadata", thread->id);
		return -1;
	}

	if (json_object_dotget_number(meta, "ovni.finished") != 1) {
		err("missing ovni.finished: %s", s->relpath);
		return -1;
	}

	thread->meta = meta;

	return 0;
}