/*************************************************************************
* Global Variables and Main Loop for the Bailey-Makeham model            *
*   Commissioned by: R. Clifton Bailey, HCFA                             *
*                    Henry Krakauer, HCFA HSQB                           *
*   Based on code written by Jim Summe, NIH                              *
*   Author, Bill Rogers: Rand, Computing Resource Center                 *
*************************************************************************/
#include "bailey.h"

#if defined(BAILEY)
/* Model Dependent Parameters */
char *spname[NSP] = {"alpha","gamma","delta"};
char model_name[] = "Bailey-Makeham Survival Model 0.4";
Int2 modelid = 1;
#endif
#if defined(LOGIST)
/* Model Dependent Parameters */
char *spname[NSP] = {"beta","gamma"};
char model_name[] = "3-outcome Logistic Regression 0.2";
Int2 modelid = 2;
#endif

/* Program Options */
DOUBLE autofix=0.0;
DOUBLE interval=1.0;
int input_type=(-1);
int nocons=0;
int rescale=0;
int has_interval=0;
int optiondvs=0;
int ndv=0;
int insight = 0;
int onestep = 0;
Int2 n_extra_vals=0;
char quotechar = '\"';
DOUBLE *ltab;
char *ident[255];
int dvlist[NDV];
int weightvar=0;              /* variable number for the weight */
char dsn[64];
char outdsn[64];
char logfile[64]="";
char format[255]="";
DOUBLE extra_vals[5];

int parm_number[256];
int var_number[256];
Int2 vars=0;
FILE *logf=NULL;
unsigned char fixed[256];
int trace=0;
Int2 labheads=0;
int nread=0;

DOUBLE theta[256];
int result;
int MaxID;

int main(argc,argv) int argc; char **argv; {
	int i;
	ltab = (DOUBLE *) malloc(LTAB*sizeof(DOUBLE));
	if (!lexopen(argv[1],ltab,(DOUBLE *)&ltab[LTAB])) exit(1);

	/* Rest of main procedure */
	KeywordInstall();
	for (i=0; i<MAX_PARMS; ++i) {
		var_number[i]=-1;
		fixed[i]=0;
		theta[i]=0.0;
	}
	logf = stdout;
	result = yyparse();
	if (result) {
		printf("Parse was rejected\n");
		out_err(outdsn);
		return 1;
	}
	if (yydebug) printf("Parse was accepted\n");
	if (debug) printf("The dataset is %s\n", dsn);
	if (input_type==STATAIN) {
		result=stata_setup(dsn);
		if (result) mxerror(result,dsn);
	} else if (labheads){
		result = raw_setup(dsn,format);
		if (result) mxerror(10,dsn);
		headread();
	}
	variable_setup();
	setup_stop();
	msetup();
	mxlkhd();
	quit_stop();
	if (*outdsn) outparm(outdsn);
	fclose(logf);
	return 0;
}

void variable_setup() {
	int i, j, k;
	char *p;
	for (k=j=0; !k && j<ndv; ++j) {
		i = LookupID(p=ident[dvlist[j]]);
		k = set_variable(p,&d[i]);
	}
	if (k) mxerror(20,p);
	for (j=0; !k && j<vars; ++j) {
		i = LookupID(p=ident[parm_number[j]]);
		if (!strcmp("_cons",p)) continue;
		k = set_variable(p,&d[i]);
	}
	if (k) mxerror(18,p);
	if (weightvar) {
		i = LookupID(p=ident[weightvar]);
		k = set_variable(p,&d[i]);
		if (k) mxerror(19,p);
	}
}

/***********************************************************************
* Operating system utilities                                           *
***********************************************************************/

#include <signal.h>
#include <float.h>

static int itype=0;

void setstop(i)
int i;
{
	stopflag=1;
	itype = i;
}

setup_stop() {
	signal(SIGINT,setstop);
	signal(SIGABRT,setstop);
	signal(SIGTERM,setstop);
	signal(SIGFPE,setstop);
}

quit_stop () {
	printf("Stopping with argument %d\n", itype);
}

getid() {
	return itype;
}