path: root/gramscii.c

/*
*
* This is `gramscii`, a simple tool to create ascii box graphs
*
*/

#include <stdio.h>
#include <stdlib.h>
#include <termios.h>
#include <unistd.h>
#include <signal.h>
#include <string.h>

#define MOVE   0x00
#define BOX    0x01
#define ARROW  0x02
#define TEXT   0x04


#define DIR_N  0x00
#define DIR_R  0x01
#define DIR_U  0x02
#define DIR_D  0x04
#define DIR_L  0x08


#define WIDTH  100
#define HEIGHT  25

#define NOFIX 0x0
#define FIX   0x1

#define BG        ' '
#define PTR       '+'
#define UND       '_'
#define ARR_L     '<'
#define ARR_R     '>'
#define ARR_U     '^'
#define ARR_D     'v'


#define MIN(x,y)  (x) < (y) ? (x) : (y)
#define MAX(x,y)  (x) > (y) ? (x) : (y)

char screen[HEIGHT][WIDTH+1];

int state;
int dir;
int x;
int y;
int step;
char cursor;
char corner;
char hlines[]  = {"-~=#@._ "};
char vlines[]  = {"|H#@:;i "};
char corners[] = {"+'H#@.\""};
int hlines_sz= sizeof(hlines) -1;
int vlines_sz= sizeof(vlines) -1;
int corners_sz = sizeof(corners) -1;
int cur_hl, cur_vl, cur_corn;
char line_h;
char line_v;

struct termios t1, t2;


void cleanup(int s){

	printf("\033[;H\033[2J");
	tcsetattr(0, TCSANOW, &t1);
	exit(0);
}

void show_cursor(){
	printf("\033[%d;%df", y+1, x+1);
}

void set(char c){
	screen[y][x] = c;
}

void set_xy(int x, int y, char c){
	/* FIXME: check if x and y are valid!!!! */
	screen[y][x] = c;
}

void draw_xy(int x, int y, char c){
	/* FIXME: check if x and y are valid!!!! */
	printf("\033[%d;%df",y+1,x+1);
	putchar(c);	
}

void init_screen(){
	int i;
	for(i=0; i<HEIGHT; i++){
		memset(screen[i], ' ', WIDTH);
		screen[i][WIDTH]='\0';
	}
	cur_corn = 0;
	corner = corners[0];
	cur_hl = cur_vl = 0;
	line_h = hlines[cur_hl];
	line_v = vlines[cur_vl];
}

char* state_str(){
	switch(state){
		case MOVE:
			return "mov";
		case TEXT:
			return "txt";
		case BOX:
			return "box";
		case ARROW:
			return "arr";
		default:
			return "ERR";
	}
	return "ERR";	
}


void status_bar(){
	
	printf("\033[%d;1f\033[7m", HEIGHT+1);
	printf(" x: %3d y: %3d -- mode: %4s hl: %c vl: %c cn: %c %10s", 
		x, y, state_str(), line_h, line_v, corner, "");

	printf("\033[0m");
}

void redraw(){
	int i;
	
	printf("\033[2J\033[1;1H");
	for (i=0;i<HEIGHT;i++){
		fprintf(stdout,"%s\n",screen[i]);
	}
	status_bar();
	show_cursor();
}

void update_current(){
	printf("\033[%d'%df",y+1,x+1);
	putchar(screen[y][x]);
}


void init(){

	signal(SIGHUP, cleanup);
	signal(SIGINT, cleanup);
	signal(SIGTERM, cleanup);
	signal(SIGQUIT, cleanup);
	
	tcgetattr(0, &t1);
	t2 = t1;
	t2.c_lflag &= ~(ICANON | ECHO);
	tcsetattr(0, TCSANOW, &t2);
	
	init_screen();
	x = WIDTH/2;
	y = HEIGHT/2;
	redraw();
}

void check_bound(){
	if (x<0) x=0;
	else if (x>=WIDTH) x = WIDTH-1;
	if (y<0) y=0;
	else if (y>=HEIGHT) y = HEIGHT -1; 
}


/*****  text, box, arrows  *****/

void get_text(){
	char c;
	int orig_x = x;
	
	redraw();
	while((c=getchar())!=EOF && c != 27){
		if(c=='\n'){
			set(BG);
			y += 1;
			x = orig_x;
		}
		else {
			set(c);
			update_current();
			x += 1;
			if (x >= WIDTH)
				x = orig_x;
		}
		check_bound();
		status_bar();
		show_cursor();
	}
	state=MOVE;
}


void draw_box(int x1, int y1, int fix){

	int xmin, ymin, xmax, ymax;
	int i;
	void (*f)(int, int, char);

	if (fix == FIX)
		f = set_xy;
	else
		f = draw_xy;
	
	xmin = MIN(x, x1);
	xmax = MAX(x, x1);
	ymin = MIN(y, y1);
	ymax = MAX(y, y1);

	for(i=xmin+1; i<=xmax; i++){
		f(i, ymin, line_h);
		f(i, ymax, line_h);
	}
	for(i=ymin+1; i<=ymax; i++){
		f(xmin, i, line_v);
		f(xmax, i, line_v);
	}
	f(xmin, ymin, corner);
	f(xmin, ymax, corner);
	f(xmax, ymin, corner);
	f(xmax, ymax, corner);
	show_cursor();
}


void get_box(){
	char c;
	int orig_x=x, orig_y=y;
	redraw();
	step = 1;
	draw_box(x,y,NOFIX);
	while((c=getchar())!=EOF && c != 27 && c!= 'b'){
		switch(c){
			case 'H': step = 5;
			case 'h': 
				x -= step;
				break;	
			case 'J': step = 5;
			case 'j':
				y += step;
				break;
			case 'K': step = 5;
			case 'k':
				y -= step;
				break;
			case 'L': step = 5;
			case 'l':
				x += step;
				break; 	
		}
		check_bound();
		redraw();
		step = 1;
		draw_box(orig_x, orig_y, NOFIX);
		status_bar();
		show_cursor();
	}
	if (c == 'b')
		draw_box(orig_x, orig_y, FIX);
	redraw();
	state = MOVE;
}


int progr_x(int dir){
	switch(dir){
		case DIR_L:
			return -1;
		case DIR_R:
			return +1;
		default:
			return 0;
	}
	return 0;
}


int progr_y(int dir){
	switch(dir){
		case DIR_D:
			return +1;
		case DIR_U:
			return -1;
		default:
			return 0;
	}
	return 0;
}


/* FIXME: fix pointer position */
/* FIXME: set initial and final markers */
/* FIXME: draw "corner" as first char after change of dir */
void draw_arrow(int x, int y, char *a, int a_len, int fix){

	int i, j;
	char line;
	void (*f)(int, int, char);

		
	if (fix == FIX)
		f = set_xy;
	else
		f = draw_xy;

	if (!a_len){
		f(x,y,corner);
		return;
	}
	line = (a[0] & DIR_L) || (a[0] & DIR_R) ? line_h : line_v;
	f(x,y,line);
	for (i=0; i<a_len; i+=2){
		for(j=0; j<a[i+1]; j++){
			line = (a[i] & DIR_L) || (a[i] & DIR_R) ? line_h : line_v;
			x += progr_x(a[i]);
			y += progr_y(a[i]);
			f(x, y, line); 
		}
	}
	show_cursor();
}

void get_arrow(){

	char c;
	int orig_x=x, orig_y=y, arrow_len;
	static char *arrow = NULL;
	static int arrow_sz;

	if (!arrow){
		arrow_sz = 100;
		arrow = malloc(arrow_sz * sizeof(char));
	}
	arrow_len = 0;
	dir = DIR_N;
	
	redraw();
	step = 1;
	//draw_arrow(x,y,NOFIX);
	while((c=getchar())!=EOF && c != 27 && c!= 'a'){
		switch(c){
			case 'H': step = 5;
			case 'h': 
				dir = DIR_L;
				x -= step;
				break;	
			case 'J': step = 5;
			case 'j':
				dir = DIR_D;
				y += step;
				break;
			case 'K': step = 5;
			case 'k':
				dir = DIR_U;
				y -= step;
				break;
			case 'L': step = 5;
			case 'l':
				dir = DIR_R;
				x += step;
				break;
			default:
				continue;
		}
		check_bound();
		/* FIXME: if we are out of bound, do nothing? */
		if (arrow_len == arrow_sz){
			arrow_sz *=2;
			arrow = realloc(arrow, arrow_sz * sizeof(char));
		}
		arrow[arrow_len++] = dir;
		arrow[arrow_len++] = step;
		redraw();
		step = 1;
		draw_arrow(orig_x, orig_y, arrow, arrow_len, NOFIX);
		status_bar();
		show_cursor();
	}
	if (c == 'a')
		draw_arrow(orig_x, orig_y, arrow, arrow_len, FIX);
	redraw();
	state = MOVE;
}

void write_file(){
	char fname[256];
	FILE *f;
}

void toggle_hline(){

	cur_hl = (cur_hl + 1) % hlines_sz;
	line_h = hlines[cur_hl];
	
}

void toggle_corner(){
	
	cur_corn = (cur_corn + 1 ) % corners_sz;
	corner = corners[cur_corn];

}

void toggle_vline(){

	cur_vl = (cur_vl + 1) % vlines_sz;
	line_v = vlines[cur_vl];

}

void commands(){

	char c;
	while((c=getchar())!=EOF){
		//screen[y][x]=BG;
		switch(c){
			case 'H':
				step=5;
			case 'h':
				x-=step;
				break;
			case 'J':
				step=5;
			case 'j':
				y+=step;
				break;
			case 'K':
				step=5;
			case 'k':
				y-=step;
				break;
			case 'L':
				step=5;
			case 'l':
				x+=step;
				break;
			case 'i':
				state = TEXT;
				get_text();
				break;
			case 'r':
				redraw();
				break;
			case 'b':
				state = BOX;
				get_box();
				break;
			case 'a':
				state = ARROW;
				get_arrow();
				break;
			case 'w':
				write_file();
				break;
			case '-':
				toggle_hline();
				break;
			case '|':
				toggle_vline();
				break;
			case '+':
				toggle_corner();
				break;
			case 'Q':
			case 'q':
				cleanup(0);
				exit(0);
				break;
			default:;
				//statu("got: %d\n", c);
		}
		check_bound();
		status_bar();
		show_cursor();
		step = 1;
	}

}




int main(int argc, char *argv[]){

	init();

	commands();
	cleanup(0);
}