#include <term.h>
#include <fcntl.h>
#include <stdio.h>
#include <string.h>
#include <termios.h>

static const char ttyfn[]="/dev/tty";

/* X^31+X^13+1 */
#define RPOLY 0x80002001UL
unsigned long rseed=1;

int rnd(int bits) {
	int i;
	for(i=0; i<bits; i++) {
		if(rseed&1) rseed^=RPOLY;
		rseed>>=1;
	}
	return rseed&((1<<bits)-1);
}

/* #define putp(s) write(1,s,strlen(s)); */
struct termios old_t, wrk_t;

static	char *t_cup;
static	int maxx,maxy;
char *t_dim,*t_bold,*t_setf;

term_init(char *t_type) {
	int t;
	if((t=open(ttyfn,O_RDWR,0))==-1 || dup2(t,1)==-1 || close(t)!=0)
		{ perror(ttyfn); return -1; }
	setupterm(NULL/*"vt100"*/,1,&t); if(t!=1)
		{ fprintf(stderr,"Nieznany typ terminala\n"); return -1; }
	/* zamiast &t mozna napisac NULL i nie sprawdzac -
	 * blad zakonczy program z odpowiednim komunikatem */
	maxx=tigetnum("cols");
	maxy=tigetnum("lines");
	if(maxx==-2 || maxy==-2)
		{ fprintf(stderr,"Nieznana geometria terminala\n"); return -1; }
	printf("Terminal ma %d linii po %d znakow\n",maxy,maxx);

	memset(&wrk_t,0,sizeof(wrk_t));
	if(tcgetattr(1,&old_t)==-1)
		{ perror("Odczyt atrybutow terminala"); return -1; }
	tcsetattr(1,TCSANOW,&wrk_t);

	t_cup=tigetstr("cup");
	t_dim=tigetstr("dim");
	t_bold=tigetstr("bold");
	t_setf=tigetstr("setf");
	return 0;
}

void term_gotoxy(int x, int y) {
	putp(tparm(t_cup,y,x));	
}

void clearscreen(void) {
	term_gotoxy(0,0);
	putp(tparm(tigetstr("ed")));
}

choinka_init(char *cfn, char *mig, int *mx, int *my, char *mc) {
  char buf[128]; FILE *cfd; int q,x,y,m;
  if((cfd=fopen(cfn,"r"))==NULL) { perror(cfn); return -1; }
  for(m=y=0; fgets(buf,sizeof(buf),cfd)!=NULL; y++) {
    for(x=0; buf[x]==' '; x++);
    for(q=x; buf[q]!=0; q++); while(buf[--q]==' '); q++;
    term_gotoxy(x,y); /* fputs(buf,stdout); */
    printf("\033[32m%.*s\033[0m",q-x,buf+x);
    for(x=0; buf[x]!=0; x++)
      if(strchr(mig,buf[x])!=NULL) {
	mx[m]=x; my[m]=y; mc[m++]=buf[x];
      }
  }
  fclose(cfd); return m;
}

choinka_mig(int m, int *mx, int *my, char *mc, char *mv) {
  int i,r; char c[]="\033[3xm";
  for(i=0; i<m; i++) {
    r=rnd(3); if(r>5*(mc[i]=='*')) continue;
    term_gotoxy(mx[i],my[i]);
    c[3]='0'+rnd(3); putp(c);
    putp((mv[i]^=1)==0 ? t_dim : t_bold);
    /*putp(tparm(t_setf,rnd(3)));	*/
    putchar(mc[i]);
  }
}

#define MXM 1000

main(void) {
/*	trzeba jakos opisac kierunki, moze tak: */
	static const int dt[]={0,1,1,1,0,-1,-1,-1,0,1};
/*	zmiana x: dt[d+2]; zmiana y: dt[d];	*/
	int i,x,y,ms,d,r,m,mx[MXM],my[MXM];
	char mv[MXM],mc[MXM];

/*	trzeba jakos generowac liczby pseudolosowe... */
	/* srand(12345); */ rseed=1|time(NULL);

/*	inicjalizacja obslugi terminala */
	if(term_init(NULL/*"vt100"*/)<0) return -1;

/*	wyczysc ekran */
	sleep(1); clearscreen();
	m=choinka_init("choinka.txt", "*+#", mx, my, mc);
	memset(mv,0,sizeof(mv)); ms=256;
	/*fprintf(stderr,"%d gwiazdek\n",m); fflush(stderr);*/

nast:	/* zaczekaj chwile */
	poll(NULL,0,ms);

	/* jesli sa dane z klawiatury to przeczytej */
	r=0; if(read(1,&r,1)==1) {
		if(r=='0') goto stop; /* '0' oznacza koniec */
		/* inne cyfry ustawiaja szybkosc weza */
		if((unsigned)(r-'0')<10) ms=2048>>(r-'1');
	}

	/*fprintf(stderr,"("); fflush(stderr);*/
	choinka_mig(m, mx, my, mc, mv);
	/*fprintf(stderr,")"); fflush(stderr);*/
	fflush(stdout); goto nast;

/*	konczac program odtworz ustawienia terminala */
stop:	clearscreen(); tcsetattr(1,TCSAFLUSH,&old_t); return 0;
}