/*****************************************************************************
*         ORTEP-II (Oak Ridge Thermal Ellipsoid Plot) for SGI                *
*                                                                            *
* ORTEP-II was originally written by C. K. Johnson (1976). Report ORNL-5138. *
* Oak Ridge National Lab., Oak Ridge, TN, USA.                               *
* ORTEP-II for OpenWindows was written by Norimasa Yamazaki, University of   *
* Electro-Communications, Tokyo, Japan (yam@crystal.pc.uec.ac.jp).           *
*                                                                            *
* ORTEP-II for SGI is based on the above two programs.                       *
* List of Files:                                                             *
*	Makefile: make file                                                  *
*       ort.f:    ORTEP-II main program written by C. K. Johnson             *
*       ortep.c:  interface to ORTEP-II for SGI                              *
* Compilation: Just type 'make'.                                             *
* Usage: ortep filename                                                      *
*        ortep -ps filename                                                  *
*                                                                            *
*        If -ps option is specified, PostScript output is printed on your    *
*        terminal (stdout). To save it in a file, redirect the standard      *
*        output, i.e., do 'ortep -ps filename > file.ps'                     *
* Features:                                                                  *
*	nothing fancy really.                                                *
*       It creates only either graphics window on SGI or PostScript file.    *
* Author: Dongchul Lim, Dept of Chemistry, Yale Univ., New Haven, CT, USA    *
*         (lim@rani.chem.yale.edu), April, 1994.                             *
******************************************************************************/

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <malloc.h>
#include <time.h>

#include <gl.h>
#include <device.h>
#include <fmclient.h>

#define ROUND(x)	(int)(((x)<0.0)?(x)-.5:(x)+.5)
#define MAX(x, y)	(((x) > (y)) ? (x) : (y))
#define MIN(x, y)	(((x) < (y)) ? (x) : (y))

#define MAXLINES	64000
#define MAXSYMBOLS	2000

typedef struct {
	float	x, y;
	float	height;
	char	str[72];
	float	angle;
	} Symbol;

typedef struct {
	float	x, y;
	int	id;
	} Lines;

static float	pensize = 0.5;
static float	xorigin = 0.0;
static float	yorigin = 0.0;
static Lines	lines[MAXLINES];
static int	nlines = 0;
static Symbol	symbols[MAXSYMBOLS];
static int	nsymbols = 0;
static fmfonthandle	fh = NULL;

/* dummy functions which are called by ortepmain */
void	gpsltm_(void) {}
void	ginit_(void) {}
void	vsini_(void) {}
void	device_(void) {}
void	rwind_(void) {}
void	charac_(void) {}
void	newpen_(void) {}
void	vsterm_(void) {}
void	tinput_(void) {}
void	pinit_(void) {}
void	drawtitle_(void) {}

void	setpensize_ (float *size)
{
	if (*size > 0 && *size < 4) pensize = *size;
	else pensize = 0.5;
}

void	plot_ (float *x, float *y, int *id)
{
	if (nlines >= MAXLINES) {
		fprintf(stderr, "Maximum # of lines reached.\n");
		return;
	}

	lines[nlines].x  = *x + xorigin;
	lines[nlines].y  = *y + yorigin;
	lines[nlines].id = *id;
	nlines++;

	if (*id < 0) {
		xorigin += *x;
		yorigin += *y;
	}
}

static int	trim_str (char *str, char *delim)
{
	trim_head(str, delim);
	return trim_tail(str, delim);
}

static int	trim_head (char *str, char *delim)
{
	char	*bgn, *p, *q;
	if (!str || !str[0]) return 0;
	bgn = str + strspn(str, delim);
	if (!*bgn) {
		str[0] = '\0';
		return 0;
	} else if (bgn == str) return 1;
	for(p=bgn,q=str;*p;p++,q++) *q = *p;
	*q = '\0';
	return 1;
}

static int	trim_tail (char *str, char *delim)
{
	char	*p;
	int	i, len;
	if (!str || !str[0]) return 0;
	len = strlen(str);
	for(i=len-1;i>=0;i--) {
		if (!strchr(delim, str[i])) return 1;
		else str[i] = '\0';
	}
	return 0;
}

void	symbol_ (float *x, float *y, float *height, char *ch, float *angle, int *len)
{
	char	*str;

	if (nsymbols >= MAXSYMBOLS) {
		fprintf(stderr, "Maximum # of symbols reached.\n");
		return;
	}
	symbols[nsymbols].x = *x + xorigin;
	symbols[nsymbols].y = *y + yorigin;
	symbols[nsymbols].height = *height;
	symbols[nsymbols].angle  = *angle;
	str = symbols[nsymbols].str;
	sprintf(str, "%.*s\0", *len, ch);
	trim_str(str, " \t\n");
	nsymbols++;
}

void	number_ (float *x, float *y, float *height, float *val, float *angle, int *num)
{
	char	*str;

	*x += 260;
	*y += 320;

	symbols[nsymbols].x = *x + xorigin;
	symbols[nsymbols].y = *y + yorigin;
	symbols[nsymbols].height = *height;
	symbols[nsymbols].angle = *angle;
	str = symbols[nsymbols].str;
	sprintf(str, "%.*f\0", *num, *val);
	trim_str(str, " \t\n");

	nsymbols++;
}

int	main (int argc, char *argv[])
{
	short	val;
	long	dev;
	char	*filename, *prog;

	if ((prog = strrchr(argv[0], '/'))) prog++; else prog = argv[0];
	if (argc < 2) {
		fprintf(stderr, "Usage: %s filename\n", prog);
		fprintf(stderr, "       %s -ps filename\n", prog);
		exit(1);
	}
	filename = argv[1];
	if (argc > 2) {
		if (strncasecmp(argv[1], "-ps", 3) == 0) {
			filename = argv[2];
			ortepmain_(filename);
			saveas_postscript(filename);
			exit(0);
		}
	}

	prefsize(650, 550);
	maxsize(650, 550);
	minsize(650, 550);
	winopen(filename);
	RGBmode();
	doublebuffer(1);
	backbuffer(1);
	gconfig();

	fminit();
	if (!(fh = fmfindfont("Times-Roman"))) {
		fprintf(stderr, "Can't find Helvetica font.\n");
		exit();
	}

	qdevice(ESCKEY);
	qdevice(WINQUIT);
	qdevice(WINSHUT);
	qreset();
	ortepmain_(filename);
	redraw_scene();
	while (1) {
		if (qtest()) {
			dev = qread(&val);
			switch (dev) {
			case REDRAW:
				ortepmain_(filename);
				redraw_scene();
				break;
			case ESCKEY:
			case WINQUIT:
			case WINSHUT:
				exit(0);
				break;
			}
		}
		sginap(1);
	}
}

static int	redraw_scene (void)
{
	register int	i;
	int	lw;
	double	height;
	float	v1[2], v2[2];
	fmfonthandle	fnt;
	fmfontinfo	info;

	cpack(0xFFFFFF);
	clear();
	cpack(0x0);
	lw = MAX(ROUND(pensize), 1);
	linewidth(lw);

	if (nlines > 0) {
		for(i=0;i<nlines;i++) {
			v2[0] = lines[i].x;
			v2[1] = 550 - lines[i].y;
			if (lines[i].id == 2) {
				bgnline();
				v2f(v1);
				v2f(v2);
				endline();
			}
			v1[0] = v2[0];
			v1[1] = v2[1];
		}

	}
	if (nsymbols > 0) {
		height = 12.0;
		fnt = fmscalefont(fh, height);
		fmgetfontinfo(fnt, &info);
		fmsetfont(fnt);
		for(i=0;i<nsymbols;i++) {
			if (height != symbols[i].height) {
				height = symbols[i].height;
				fnt = fmscalefont(fh, height);
				fmgetfontinfo(fnt, &info);
				fmsetfont(fnt);
			}
			cmov2(symbols[i].x, 550.0-symbols[i].y - info.height/2);
			fmprstr(symbols[i].str);
		}
	}
	swapbuffers();
}

static int	saveas_postscript (char *filename)
{
	int	x1, y1, x2, y2;
	int	tx, ty, i;
	float	scalefactor;
	FILE	*fp = stdout;
	time_t	curtime;

	tx = 0;
	ty = 0;
	scalefactor = 1.0;

	bbox_ortep(&x1, &y1, &x2, &y2);
	x1 += tx;
	x2 += tx;
	y1 += ty;
	y2 += ty;
	x1 *= scalefactor;
	x2 *= scalefactor;
	y1 *= scalefactor;
	y2 *= scalefactor;

	fprintf(fp, "%%!PS-Adobe-2.0 EPSF-2.0\n");
	fprintf(fp, "%%%%BoundingBox:%d %d %d %d\n", x1, y1, x2, y2);
	fprintf(fp, "%%%%Title: %s\n", filename);
	fprintf(fp, "%%%%Creator: ORTEP-II FOR SGI\n");
	curtime = time(NULL);
	fprintf(fp, "%%%%CreationDate: %s", ctime(&curtime));
	fprintf(fp, "%%%%DocumentFonts: Times-Roman\n");
	fprintf(fp, "%%%%EndComments\n");
	fprintf(fp, "/m /moveto load def\n");
	fprintf(fp, "/l /lineto load def\n");
	fprintf(fp, "%%%%EndProlog\n\n");
	fprintf(fp, "%d %d translate\n", tx, ty);
	fprintf(fp, "%.2f %.2f scale\n", scalefactor, scalefactor);
	fprintf(fp, "%.2f setlinewidth 0 0 0 setrgbcolor\n", pensize);
	fprintf(fp, "\n");

	for(i=0;i<nlines;i++) {
		if (lines[i].id == 2) {
			fprintf(fp, "%.3f %.3f l\n", lines[i].x, lines[i].y);
		} else {
			if (i) fprintf(fp, "stroke\n");
			fprintf(fp, "%.3f %.3f m\n", lines[i].x, lines[i].y);
		}
	}
	if (i) fprintf(fp, "stroke\n");
	for(i=0;i<nsymbols;i++) {
		fprintf(fp, "gsave\n");
		fprintf(fp, "/Times-Roman findfont %.1f scalefont setfont\n", symbols[i].height);
		fprintf(fp, "%.3f %.3f m\n", symbols[i].x, symbols[i].y);
		if (symbols[i].angle != 0) fprintf(fp, "%.3f rotate\n", symbols[i].angle);
		fprintf(fp, "(%s)\n", symbols[i].str);
		fprintf(fp, "show\ngrestore\n");
	}
	fprintf(fp, "showpage\n");
	fprintf(fp, "%%Trailer\n");
}

/* get the bounding box of the plot */
static int	bbox_ortep (int *x1, int *y1, int *x2, int *y2)
{
	register int	i;
	int	lw;

	*x1 = *y1 =  1000;
	*x2 = *y2 = -1000;

	for(i=0;i<nlines;i++) {
		*x1 = MIN(*x1, ROUND(lines[i].x));
		*y1 = MIN(*y1, ROUND(lines[i].y));
		*x2 = MAX(*x2, ROUND(lines[i].x));
		*y2 = MAX(*y2, ROUND(lines[i].y));
	}

	for(i=0;i<nsymbols;i++) {
		*x1 = MIN(*x1, ROUND(symbols[i].x));
		*y1 = MIN(*y1, ROUND(symbols[i].y));
		*x2 = MAX(*x2, ROUND(symbols[i].x));
		*y2 = MAX(*y2, ROUND(symbols[i].y + symbols[i].height));
	}

	lw = MAX(pensize, 1);
	*x1 -= lw;
	*y1 -= lw;
	*x2 += lw;
	*y2 += lw;
}