#include <stdio.h>
#include <stdlib.h>
#include<math.h>

//#include "c:\nr\nrutil.c"
#include "c:\juan\networks\progs\ODE.c"

#define m 3 // numero de ecuaciones
	
double gamma,beta; // Parametros

FILE *archivo;
char salida[40];

void imprimir(double, double []);

void sir(double t, double v[], double dv[])
{
double S,I,R;
S=v[0]; I=v[1]; R=v[2];

dv[0]=-beta*S*I;
dv[1]=beta*S*I-gamma*I;
dv[2]=gamma*I;
return;
}


void main()
{
double v[m],dv[m],t=0,tau,dt;

gamma=1.;
beta=1./50.;

fprintf(stderr,"Nombre del archivo de salida: ");
scanf("%s",salida);

fprintf(stderr,"Condiciones iniciales S0=? I0=? R0=?\n");
scanf("%lf %lf %lf",&v[0],&v[1],&v[2]);
printf("%4.2f %4.2f %4.2f\n",v[0],v[1],v[2]);

fprintf(stderr,"dt, t_iteracion \n");
scanf("%lf %lf",&dt,&tau);
fprintf(stderr,"%f %f\n",dt,tau);

/* salida */
while (t <tau)
	{
	imprimir(t,v);
	sir(t,v,dv);
    rk4(v,dv,m,t,dt,v,sir);
	t+=dt;
    }
exit(0);
}

void imprimir(double t,double v[])
{
archivo=fopen(salida,"a");
fprintf(archivo,"%2.3f\t%2.3f\t%2.3f\t%2.3f\n",t,v[0],v[1],v[2]);
fclose(archivo);
}