#include <time.h>
#include <iostream>
#include "simulation.h"
#include "random.h"

#define NUM_PORTS 4
#define MIN_FRAME_SIZE 64
#define MAX_FRAME_SIZE 32768
#define FRAME_SPACE 960

#define t_frame (double)80*size
#define t_interval t_frame+FRAME_SPACE+negexp(1/(double)lambda)
#define t_sim 10*1E9

using namespace std;
 
class TFrame : public CProcess {
	private:
		unsigned in_port, out_port, size;
		double lambda, *Fx;
		virtual void Run();
	public:
		TFrame(unsigned ip, double l, double *F) {
			in_port = ip; lambda = l; Fx = F;
			out_port = discrete(Fx);
			size = (unsigned) uniform(MIN_FRAME_SIZE, MAX_FRAME_SIZE);
		};
		unsigned getInPort() {return in_port;};
		unsigned getOutPort() {return out_port;};
		unsigned getFrameSize() {return size;};
};

class TPort : public CProcess {
	private:
		int port;
		virtual void Run();
	public:
		TPort(int p) {port = p;};
};

template<class T>
class Stat {
	private:
		T suma;
		T max;
		int pocet;
	public:
		Stat() {suma=0; max=0; pocet=0;};
		void addItem(T item) {pocet++; suma+=item; if(item > max) max=item;};
		T getMax() {return max;};
		double getAvg() {return suma/(double) pocet;};
};


CHead	*PortFIFO[NUM_PORTS];	// Vystupni fronty portu
TPort	*Port[NUM_PORTS];		// Obsluhy portu

Stat<TIME>	*TimeStat[NUM_PORTS];		// Casove statistiky portu
Stat<TIME>	*TimeStatTotal;				// Casova statistika celeho switche
Stat<unsigned>	*FifoStat[NUM_PORTS];	// Statistiky front portu


void TFrame::Run() {
	TIME start;
	TFrame *P;
	unsigned fifo_size=0;

	(new TFrame(in_port, lambda, Fx))->ActivateDelay(t_interval);

	cout << Time() << ": P" << getInPort() << " -> P" << getOutPort() << " ("
	     << getFrameSize() << "B) INPUT start" << endl;
	
	Hold(t_frame);
	
	cout << Time() << ": P" << getInPort() << " -> P" << getOutPort() << " ("
	     << getFrameSize() << "B) INPUT end" << endl;
	
	start = Time();
	
	if (PortFIFO[out_port]->Empty()) {
		Into(PortFIFO[out_port]);
		Port[out_port]->ActivateAt(Time());
	} else
		Into(PortFIFO[out_port]);

	/* Statistiky front */
	P = (TFrame*) PortFIFO[out_port]->First();
	while (P != NULL) { 
		fifo_size += P->getFrameSize();
		P = (TFrame *)P->Suc();
	}
	FifoStat[out_port]->addItem(fifo_size);

	Passivate();

	cout << Time() << ": P" << getInPort() << " -> P" << getOutPort() << " ("
	<< getFrameSize() << "B) OUTPUT end" << endl;
	
	/* Casove statistiky */
	TimeStat[out_port]->addItem(Time()-start);
	TimeStatTotal->addItem(Time()-start);
}

void TPort::Run() {
	TFrame *P;
	unsigned size;

	while(true) {
		P = (TFrame*) PortFIFO[port]->First();
		size = P->getFrameSize();
		
		cout << Time() << ": P" << P->getInPort() << " -> P" << P->getOutPort()
		     << " (" << size << "B) OUTPUT start" << endl;

		P->Out();
		Hold(t_frame+FRAME_SPACE);
		P->ActivateAt(Time());
		if (PortFIFO[port]->Empty())
			Passivate();
	}
}

void CSimulation::Run() {
	srand((unsigned) time( NULL ));

	for (int i=0; i < NUM_PORTS ; i++) {
		PortFIFO[i] = new CHead;
		Port[i] = new TPort(i);
		TimeStat[i] = new Stat<TIME>();
		FifoStat[i] = new Stat<unsigned>();		
	}
	TimeStatTotal = new Stat<TIME>();
	
	/* Distribucni fce pro cesty mezi porty */
	double Fx0[]={0.00, 0.20, 0.70, 1.00};
	double Fx1[]={0.20, 0.20, 0.60, 1.00};
	double Fx2[]={0.45, 0.75, 0.75, 1.00};
	double Fx3[]={0.30, 0.70, 1.00, 1.00};

	(new TFrame(0, 5E6, Fx0))->ActivateAt(Time());
	(new TFrame(1, 20E6, Fx1))->ActivateAt(Time());
	(new TFrame(2, 12E6, Fx2))->ActivateAt(Time());
	(new TFrame(3, 8E6, Fx3))->ActivateAt(Time());
	
	Hold(t_sim);
	

	cout << endl << "---------- STATISTIKY ----------" << endl;
	for (int i=0; i<NUM_PORTS; i++) {
		cout << "AVG/MAX doba zpracovani [ms] - PORT " << i+1 << ": ";
		cout << TimeStat[i]->getAvg()/1E6 << " / " << TimeStat[i]->getMax()/1E6;
		cout << endl;
	}
	cout << "AVG/MAX doba zpracovani [ms] celkem: ";
	cout << TimeStatTotal->getAvg()/1E6 << " / " << TimeStatTotal->getMax()/1E6;
	cout << endl << endl;

	for (int i=0; i<NUM_PORTS; i++) {
		cout << "MAX delka fronty [B] - PORT " << i+1 << ": ";
		cout << FifoStat[i]->getMax() << endl;
	}
	cout << endl;

	exit(0);
};