/*
 * Copyright (c) 2021 Barcelona Supercomputing Center (BSC)
 *
 * This program 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, either version 3 of the License, or
 * (at your option) any later version.
 *
 * 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, see <https://www.gnu.org/licenses/>.
 */

#include <assert.h>
#include "uthash.h"

#include "ovni.h"
#include "ovni_trace.h"
#include "emu.h"
#include "prv.h"
#include "chan.h"

/* --------------------------- init ------------------------------- */

void
hook_init_tampi(struct ovni_emu *emu)
{
	struct ovni_ethread *th;
	struct ovni_cpu *cpu;
	size_t i;
	int row;
	FILE *prv_th, *prv_cpu;
	int64_t *clock;
	struct ovni_chan **uth, **ucpu;

	clock = &emu->delta_time;
	prv_th = emu->prv_thread;
	prv_cpu = emu->prv_cpu;

	/* Init the channels in all threads */
	for(i=0; i<emu->total_nthreads; i++)
	{
		th = emu->global_thread[i];
		row = th->gindex + 1;
		uth = &emu->th_chan;

		chan_th_init(th, uth, CHAN_TAMPI_MODE, CHAN_TRACK_TH_RUNNING, 0, 0, 1, row, prv_th, clock);
	}

	/* Init the channels in all cpus */
	for(i=0; i<emu->total_ncpus; i++)
	{
		cpu = emu->global_cpu[i];
		row = cpu->gindex + 1;
		ucpu = &emu->cpu_chan;

		chan_cpu_init(cpu, ucpu, CHAN_TAMPI_MODE, CHAN_TRACK_TH_RUNNING, 0, 0, 1, row, prv_cpu, clock);
	}
}

/* --------------------------- pre ------------------------------- */


static void
pre_send(struct ovni_emu *emu)
{
	struct ovni_ethread *th;

	th = emu->cur_thread;

	switch(emu->cur_ev->header.value)
	{
		case '[':
			chan_push(&th->chan[CHAN_TAMPI_MODE], ST_TAMPI_SEND);
			break;
		case ']':
			chan_pop(&th->chan[CHAN_TAMPI_MODE], ST_TAMPI_SEND);
			break;
		default:
			  abort();
	}
}

static void
pre_recv(struct ovni_emu *emu)
{
	struct ovni_ethread *th;

	th = emu->cur_thread;

	switch(emu->cur_ev->header.value)
	{
		case '[':
			chan_push(&th->chan[CHAN_TAMPI_MODE], ST_TAMPI_RECV);
			break;
		case ']':
			chan_pop(&th->chan[CHAN_TAMPI_MODE], ST_TAMPI_RECV);
			break;
		default:
			  abort();
	}
}

static void
pre_isend(struct ovni_emu *emu)
{
	struct ovni_ethread *th;

	th = emu->cur_thread;

	switch(emu->cur_ev->header.value)
	{
		case '[':
			chan_push(&th->chan[CHAN_TAMPI_MODE], ST_TAMPI_ISEND);
			break;
		case ']':
			chan_pop(&th->chan[CHAN_TAMPI_MODE], ST_TAMPI_ISEND);
			break;
		default:
			  abort();
	}
}

static void
pre_irecv(struct ovni_emu *emu)
{
	struct ovni_ethread *th;

	th = emu->cur_thread;

	switch(emu->cur_ev->header.value)
	{
		case '[':
			chan_push(&th->chan[CHAN_TAMPI_MODE], ST_TAMPI_IRECV);
			break;
		case ']':
			chan_pop(&th->chan[CHAN_TAMPI_MODE], ST_TAMPI_IRECV);
			break;
		default:
			  abort();
	}
}

static void
pre_wait(struct ovni_emu *emu)
{
	struct ovni_ethread *th;

	th = emu->cur_thread;

	switch(emu->cur_ev->header.value)
	{
		case '[':
			chan_push(&th->chan[CHAN_TAMPI_MODE], ST_TAMPI_WAIT);
			break;
		case ']':
			chan_pop(&th->chan[CHAN_TAMPI_MODE], ST_TAMPI_WAIT);
			break;
		default:
			  abort();
	}
}

static void
pre_waitall(struct ovni_emu *emu)
{
	struct ovni_ethread *th;

	th = emu->cur_thread;

	switch(emu->cur_ev->header.value)
	{
		case '[':
			chan_push(&th->chan[CHAN_TAMPI_MODE], ST_TAMPI_WAITALL);
			break;
		case ']':
			chan_pop(&th->chan[CHAN_TAMPI_MODE], ST_TAMPI_WAITALL);
			break;
		default:
			  abort();
	}
}

void
hook_pre_tampi(struct ovni_emu *emu)
{
	assert(emu->cur_ev->header.model == 'T');

	switch(emu->cur_ev->header.category)
	{
		case 'S': pre_send(emu); break;
		case 'R': pre_recv(emu); break;
		case 's': pre_isend(emu); break;
		case 'r': pre_irecv(emu); break;
		case 'V': pre_wait(emu); break;
		case 'W': pre_waitall(emu); break;
		default:
			break;
	}
}