#include "cpu.h"
#include <iostream>
#include <iomanip>

using namespace std;

extern double	gtime();
extern string	elapsed(double,double);

void cpu_init()
{
	clog << "CPU Init" << endl;
}

void cpu_fini()
{
	clog << "CPU Fini" << endl;
}

void cpu_run(float* u,int n,int nit)
{
	extern void vinit(float* u,int n);
	double tt,ts,te;
	float*	tmp;

	cout << "CPU ==============================================" << endl;

	vinit(u,n);

	clog << "CPU Alloc:    ";
	ts = gtime();
	tmp = new float[n];
	te = gtime();
	cout << elapsed(ts,te) << endl;

	clog << "CPU Run:      ";
	ts = gtime();
	for (int it=0; it<nit; it++)
	{
#if 1
		for (int i=0; i<n; i++)
		{
		    int i0 = (i-1+n)%n;
		    int i1 = (i+1)%n;
			tmp[i] = 0.25f*(u[i0]+2.0f*u[i]+u[i1]);
		}
#else
		tmp[0] = u[0];
		for (int i=1; i<n-1; i++)
		{
			tmp[i] = 0.25f*(u[i-1]+2.0f*u[i]+u[i+1]);
		}
		tmp[n-1] = u[n-1];
#endif
		for (int i=0; i<n; i++)
			u[i] = tmp[i];
	}
	te = gtime();
	cout.setf(ios::fixed);
	cout << elapsed(ts,te) << endl;

	clog << "CPU Free:     ";
	ts = gtime();
	delete[] tmp;
	te = gtime();
	cout << elapsed(ts,te) << endl;
	cout << "CPU ==============================================" << endl;
	cout << endl;
}