#include <stdlib.h>
#include <time.h>
#include <signal.h>
#include <string.h>
#include <unistd.h>
#include "lanhep.h"


char *ModelName=NULL;
int  ModelNumber=0;

FILE *log_file = NULL;

char *OutputDirectory = NULL;
char *InputDirectory  = /*"/home/semenov/lanhep/mdl/"*/ NULL;
char *InputFile = NULL;
int  VerbMode = 0;
char *InitFile = "lhep.rc";
int  TexOutput = 0, FAOutput=0;
int  ForsedRedCol = 0;
int	 NoColors = 0;
int WriteColors=0;
int  No4Color = 0;
int  EvalPrm  = 0, EvalVrt=0;
int  ChepVersion=4;
int  opAbbrVrt=0;
int remove_rc=0;
int opRemDotWithFerm=1;
int MicroOmega=0;

int end_with_tty=0;

extern int opSplitCol1, opSplitCol2, opEvalVrt;
extern int verb_herm, err_cnt;
extern int write_all_vertices, opMaxiLegs;
extern SaveRules(char *);
extern void SetKeyFromArg(char *); 
extern int off_srefine, ch_sign, longest_lpline, longest_cfline, longest_pdline;
extern int eval_vrt_len, eval_vrt_more, kill_gamma_pm, opNoDummies,opTriHeu;
extern int SecondVaFu;
extern int inf_removed[10];

extern int C_F_WIDTH, L_P_WIDTH, P_D_WIDTH, TEX_max_pno, doinitfile;

static char nmbuf[128], nmbuf0[128];

extern void abbr_stat(void);

#ifdef __DJGPP__
static char *find_path(char *nm, char *env[])
	{
	char *pp;
	strcpy(nmbuf,nm);
    pp=strrchr(nmbuf,'/');
	if(pp==NULL)
		{
		puts("Fail locating lhep.rc file");
		exit(0);
		}
	pp[1]=0;
    strcat(nmbuf,"mdl/");
	return nmbuf;
	}
#else
static char *find_path(char *nm, char *env[])
	{
	int i;
	char *pp;
	strcpy(nmbuf0,nm);
bgn:
	for(i=0;i<128;nmbuf[i++]=0);
	if(readlink(nmbuf0,nmbuf,128)!=-1)
		{
		strcpy(nmbuf0,nmbuf);
		goto bgn;
		}
	if(access(nmbuf0,0)==0)
		{
		pp=strrchr(nmbuf0,'/');
		if(pp==NULL)
			return "./mdl/";
		else
			{
			pp[1]=0;
			strcat(nmbuf0,"mdl/");
			return nmbuf0;
			}
		}
	i=0;
	while(env[i] && strncmp(env[i],"PATH=",5))
		i++;
	if(env[i]==NULL)
		{
		puts("No PATH environment variable found.");
		exit(0);
		}
	pp=env[i];
	while(pp[0])
		{
		int ll;
		ll=0;
		while(pp[0] && pp[0]!=':')
			{
			nmbuf[ll]=pp[0];
			pp++;
			ll++;
			}
		if(pp[0]) pp++;
		nmbuf[ll]='/';
		ll++;
		strcpy(nmbuf+ll,nmbuf0);
		if(access(nmbuf,0)==0)
			{
			strcpy(nmbuf0,nmbuf);
			goto bgn;
			}
		}

	puts("Fail locating lhep.rc file");
	exit(0);
	return "???";
	}
#endif


void ProcessTerm(Term t)
	{
	Term res;
	Atom procc=0;
	char regbuf[80];

    if(is_atom(t))
		procc=t;
	if(is_compound(t))
		{
		procc=CompoundName(t);
		}
    if(procc==0)
		{
        if(!IsTermInput())
			printf("File \"%s\", line %d: ",CurrentInputFile(),
				CurrentInputLine());
		printf("Not a valid operator \'");
		WriteTerm(t);
		printf("\'.\n");
		FreeAtomic(t);
		return;
		}

	if(!IsTermInput())
		sprintf(regbuf, "ProcTerm: File %s, line %d: %s statement.",
			CurrentInputFile(), CurrentInputLine(), AtomValue(procc));
	else
		sprintf(regbuf, "(tty): %s statement.",AtomValue(procc));

	RegisterLine(regbuf);

	if(!is_function(t,NULL))
		{
        if(!IsTermInput())
			printf("File \"%s\", line %d: ",CurrentInputFile(),
				CurrentInputLine());
		printf("Unknown operator \'%s\' in expression\n\t",AtomValue(procc));
		WriteTerm(t);
		puts("");
		FreeAtomic(t);
		UnregisterLine();
		return;
		}

    if(is_compound(t) && ((VerbMode && CompoundName(t)==OPR_LTERM) ||
			VerbMode==3))
    	{
    	char sment[1024];
    	sWriteTerm(sment,t);
    	sment[55]='.';
    	sment[56]='.';
    	sment[57]='.';
    	sment[58]=0;
    	printf("%s:%3d: %s\n",CurrentInputFile(),CurrentInputLine(),sment);
    	}
    	
	res=CallFunction(t,0);
	if(res!=0 && (IsTermInput()))
		{
		if(is_compound(res) && CompoundArity(t)==2 && 
				CompoundName(res)==A_ALG1)
			res=Alg1ToExpr(res);
		if(is_list(res))
			DumpList(res);
		else
			{
			WriteTerm(res);
			puts("");
			}
		FreeAtomic(res);
		}
	UnregisterLine();
	}


static void stop(int sig)
	{
	puts("\nInterrupt signal received, exiting...");
	DumpRegistered();
	AtomStat1();
	ListStat1();
	exit(0);
	}

static void sigdmp(int sig)
	{
	puts("\nCurrent state:");
	DumpRegistered();
	AtomStat1();
	ListStat1();
	puts("");
	fflush(stdout);
	signal(SIGUSR1, sigdmp);
	}


static void sigsegv(int sig)
	{
	puts("Internal error (segmentation violation), aborted...");
	DumpRegistered();
	AtomStatistics();
	ListStatistics();
	abort();
	}



static void exv(int, char **);

int main(int argc, char **argv, char **env)
	{
	int i;
	int save_flag=0;
	char *save_file=NULL;


	
	if(sizeof(Term)!=4)
		{
		puts("Compilation error");
		return -1;
		}

	if(argc>1 && strcmp(argv[1],"-exv")==0)
		{
		exv(argc-1,argv+1);
		return 0;
		}

	InitAtoms();
	InitFuncs();
	AlwaysBracets = 0;
	WideWriting = 0;
	signal(SIGINT, stop);
	signal(SIGSEGV, sigsegv);
	signal(SIGUSR1, sigdmp);
	

	for(i=1; i<argc; i++)
		{
		if(strcmp(argv[i],"-v")==0)
			{
			VerbMode=1;
			continue;
			}
		if(strcmp(argv[i],"-vv")==0)
			{
			VerbMode=2;
			continue;
			}
		if(strcmp(argv[i],"-vvv")==0)
			{
			VerbMode=3;
			continue;
			}
		if(strcmp(argv[i],"-c4")==0)
			{
			ChepVersion=4;
			continue;
			}	
		if(strcmp(argv[i],"-c3")==0)
			{
			ChepVersion=3;
			continue;
			}	
		if(strcmp(argv[i],"-mOmega")==0)
			{
			ChepVersion=4;
			MicroOmega=1;
			continue;
			}	
		if(strcmp(argv[i],"-OutDir")==0)
			{
			OutputDirectory=argv[++i];
			continue;
			}
		if(strcmp(argv[i],"-InDir")==0)
			{
			InputDirectory=argv[++i];
			continue;
			}
		if(strcmp(argv[i],"-allvrt")==0)
			{
			write_all_vertices=1;
			continue;
			}
		if(strcmp(argv[i],"-rc")==0)
			{
			InitFile=argv[++i];
			continue;
			}
		if(strcmp(argv[i],"-tex")==0)
			{
			TexOutput=1;
			opSplitCol1=0;
			continue;
			}
		if(strcmp(argv[i],"-feynarts")==0 || strcmp(argv[i],"-FeynArts")==0)
			{
			FAOutput=1;
			opSplitCol1=0;
			opSplitCol2=0;
			continue;
			}
		if(strcmp(argv[i],"-save")==0)
			{
			save_flag=1;
			save_file=argv[++i];
			printf("Feynman rules will be saved in '%s' file.\n",save_file);
			continue;
			}
		if(strcmp(argv[i],"-eval-prm")==0)
			{
			EvalPrm=1;
			continue;
			}
		if(strcmp(argv[i],"-eval-vrt")==0)
			{
			EvalVrt=1;
			continue;
			}
		if(strcmp(argv[i],"-frc")==0)
			{
			ForsedRedCol=1;
			continue;
			}
		if(strcmp(argv[i],"-nocolor")==0)
			{
			NoColors=1;
			continue;
			}
		if(strcmp(argv[i],"-colors")==0)
			{
			WriteColors=1;
			continue;
			}
        if(strcmp(argv[i],"-no4color")==0)
			{
            No4Color=1;
			continue;
            }
		if(strcmp(argv[i],"-nocdot")==0)
			{
			TEX_set_dot=0;
			continue;
			}
		if(strcmp(argv[i],"-v-charges")==0)
			{
			verb_charge=1;
			continue;
			}
		if(strcmp(argv[i],"-v-herm")==0)
			{
			verb_herm=1;
			continue;
			}
		if(strcmp(argv[i],"-v-imprt")==0)
			{
			verb_imprt=1;
			continue;
			}
		if(strcmp(argv[i],"-off-srefine")==0)
			{
			off_srefine=1;
			continue;
			}
		if(strcmp(argv[i],"-chep-srefine")==0)
			{
			ch_sign=1;
			continue;
			}
		if(strcmp(argv[i],"-texLines")==0)
			{
			sscanf(argv[++i],"%d",&TEX_lines);
			continue;
			}
		if(strcmp(argv[i],"-texLineLength")==0)
			{
			sscanf(argv[++i],"%d",&TEX_spec_in_line);
			continue;
			}
		if(strcmp(argv[i],"-texMaxPrtNo")==0)
			{
			sscanf(argv[++i],"%d",&TEX_max_pno);
			continue;
			}
		if(strncmp(argv[i],"-abbr",5)==0)
			{
			if(eval_vrt_len)
				{
				puts("Error: -evl and -abbr options are not compatible");
				continue;
				}
			opAbbrVrt=1;
			opEvalVrt=0;
			opTriHeu=0;
			if(argv[i][5]>'1' && argv[i][5]<'9')
				opAbbrVrt=argv[i][5]-'1'+1;
			opNoDummies=1;
			continue;
			}
		if(strcmp(argv[i],"-evl")==0)
			{
			if(opAbbrVrt)
			{
				puts("Error: -evl and -abbr options are not compatible");
				i++;
				continue;
			}
			sscanf(argv[++i],"%d",&eval_vrt_len);
			if(eval_vrt_len==2)
			{
				opNoDummies=1;
				eval_vrt_more=1;
				/*kill_gamma_pm=1;*/
			}
			continue;
			}
		if(strcmp(argv[i],"-sleep")==0)
			{
			int sec;
			sscanf(argv[++i],"%d",&sec);
			sleep(sec*60);
			continue;
			}
		if(strcmp(argv[i],"-key")==0)
			{
			SetKeyFromArg(argv[++i]);
			continue;
			}
		if(strcmp(argv[i],"-edbg")==0)
			{
			end_with_tty=1;
			continue;
			}
		if(strcmp(argv[i],"-norc")==0)
			{
			remove_rc=1;
			continue;
			}
		if(argv[i][0]=='-')
			{
			printf("Warning: unknown option %s.\n",argv[i]);
			continue;
			}
		if(InputFile)
			{
			printf("Warning: unknown option %s.\n",argv[i]);
			continue;
			}
		InputFile=argv[i];
		}

	if(!write_all_vertices && !TexOutput)
	{
		if(FAOutput)
			opMaxiLegs=5;
		else
			opMaxiLegs=4;
	}
		
	if(MicroOmega)
		SetKeyFromArg("MicrOmega=1");
	else
		SetKeyFromArg("MicrOmega=0");
	
	if(FAOutput)
		SetKeyFromArg("FeynArts=1");
	else
		SetKeyFromArg("FeynArts=0");
		
	if(InputDirectory==NULL)
		{
		InputDirectory=find_path(argv[0],env);
		if(VerbMode)
			printf("Input directory is '%s'\n",InputDirectory);
		}
	doinitfile=1;	
    ReadFile(InitFile);
	doinitfile=0;
	if(InputFile!=NULL)
		ReadFile(InputFile);
	else
		{
		printf(
"Welcome to LanHEP                                Version 3.0.0  (Apr 25 2008)\n");
		/*
		log_file=fopen("lhep.log","w");
		if(log_file==NULL)
			printf("Warning: can not open file lhep.log for writing.\n");
		*/
		ReadFile(NULL);
		if(log_file!=NULL)
			{
			fclose(log_file);
			log_file=0;
			}
		}

puts("");

/*	AtomStatistics();
	ListStatistics();
*/
	if(save_flag)
		{
		Term t;
		SaveRules(save_file);
		if(itrSetIn("q.sav")==0)
			return 0;

		while((t=itrIn())!=0)
			{
			WriteTerm(t);
			puts("");
			}

		itrCloseIn();

		return 0;
		}

	if(ModelNumber!=0 || InputFile!=NULL)
		{
		if(MicroOmega)
			ModelNumber=1;
		RegisterLine("MAIN: writing lagrangian.");
		WriteLagrFile(ModelNumber,ModelName);
		UnregisterLine();
		RegisterLine("MAIN: writing parameters and particles.");
		WriteParameters(ModelNumber,ModelName);
		WriteParticles(ModelNumber,ModelName);
		if(MicroOmega)
		{
			ModelNumber=2;
			SecondVaFu=1;
			WriteParameters(ModelNumber,ModelName);
		}
		UnregisterLine();
		
		if(!TexOutput && !FAOutput && C_F_WIDTH<longest_cfline)
		{
			printf("Error: 'Common Factor' field longer than maximum of %d symbols.\n",
					C_F_WIDTH);
			printf("Use the statement 'option chepCFWidth=%d.'\n",longest_cfline+1);
		}
		if(!TexOutput && !FAOutput && L_P_WIDTH<longest_lpline)
		{
			printf("Error: 'Lorentz part' field longer than maximum of %d symbols.\n",
					L_P_WIDTH);
			printf("Use the statement 'option chepLPWidth=%d.'\n",longest_lpline+1);
		}
		if(!TexOutput && !FAOutput && P_D_WIDTH<longest_pdline)
		{
			printf("Error: 'Parameter expression' field longer than maximum of %d symbols.\n",
					P_D_WIDTH);
			printf("Use the statement 'option chepPDWidth=%d.'\n",longest_pdline+1);
		}
		
		if(!TexOutput && !FAOutput && VerbMode)
			{
			printf("Longest Common factor line is %d symbols (max %d)\n",
											longest_cfline,C_F_WIDTH);
			printf("Longest Lorentz part  line is %d symbols (max %d)\n",
											longest_lpline,L_P_WIDTH);
			printf("Longest Parameter dep line is %d symbols (max %d)\n\n",
											longest_pdline,P_D_WIDTH);
			}
		}
/*
	else
		puts("Model # is zero");
*/
	if(VerbMode)
		{
		int i;
		printf("%5.1fMB of memory used.\n",
			(ListMemory()+TermMemory())*0.001);
		abbr_stat();
		/*AtomStatistics();
		ListStatistics();
		for(i=0;i<10;i++)
			if(inf_removed[i]) printf("%d pwr of inf: %d\n",i,inf_removed[i]);*/
		}

	if(err_cnt)
		puts("!!!!!!!!!!!!! THERE WERE ERRORS DURING PROCESSING !!!!!!!!");
		
		if(end_with_tty)
		{
			ReadFile(NULL);
			puts("");
		}
		
	return 0;
	}



Term ProcessQuit(Term t, Term ind)
	{
	if(is_compound(t))
	{
	    AtomStatistics();
		ListStatistics();
	}
	if(log_file!=NULL)
		fclose(log_file);
	/*
	if(ModelNumber!=0)
		{
		WriteParameters(ModelNumber,ModelName);
		WriteLagrFile(ModelNumber,ModelName);
		WriteParticles(ModelNumber,ModelName);
		}
	*/
	exit(0);
	return 0;
	}


static char cbuf[5000];

static void exv(int ac, char *av[])
	{
	FILE *f;
	int pno;
	pno=ac-2;
	if(pno<1) return;
	sprintf(cbuf,"lgrng%s.mdl",av[1]);
	f=fopen(cbuf,"r");
	if(f==NULL)
		{
		printf("can not open file '%s'\n",cbuf);
		return;
		}
	
	fgets(cbuf,5000,f);
	fgets(cbuf,5000,f);
	fgets(cbuf,5000,f);
	
	while(fgets(cbuf,5000,f)==cbuf)
		{
		int ppos=0;
		int sflag=1;
		int i,j;
		for(i=0;i<4;i++)
			{
			int clen=0;
			char sav;
			int ffo=0;
			if(cbuf[ppos]==' ')
				break;
			while(cbuf[ppos+clen]!=' ' && cbuf[ppos+clen]!='|')
				clen++;
			sav=cbuf[ppos+clen];
			cbuf[ppos+clen]=0;
			for(j=0;j<pno;j++)
				if(strcmp(cbuf+ppos,av[j+2])==0)
					ffo=1;
			if(ffo==0)
				sflag=0;
			cbuf[ppos+clen]=sav;
			while(cbuf[ppos]!='|')
				ppos++;
			ppos++;
			}
		if(sflag)
			printf(cbuf);
		}
	fclose(f);
	}