ref: 1b09060f468f530af2f4ef75441a40084e912e54
dir: /sys/src/cmd/graph/graph.c/
#include <u.h>
#include <libc.h>
#include <stdio.h>
#include "iplot.h"
#define INF 1.e+37
#define F .25
struct xy {
int xlbf; /*flag:explicit lower bound*/
int xubf; /*flag:explicit upper bound*/
int xqf; /*flag:explicit quantum*/
double (*xf)(double); /*transform function, e.g. log*/
float xa,xb; /*scaling coefficients*/
float xlb,xub; /*lower and upper bound*/
float xquant; /*quantum*/
float xoff; /*screen offset fraction*/
float xsize; /*screen fraction*/
int xbot,xtop; /*screen coords of border*/
float xmult; /*scaling constant*/
} xd,yd;
struct val {
float xv;
float yv;
int lblptr;
} *xx;
char *labels;
int labelsiz;
int tick = 50;
int top = 4000;
int bot = 200;
float absbot;
int n;
int erasf = 1;
int gridf = 2;
int symbf = 0;
int absf = 0;
int transf;
int equf;
int brkf;
int ovlay = 1;
float dx;
char *plotsymb;
#define BSIZ 80
char labbuf[BSIZ];
char titlebuf[BSIZ];
char *modes[] = {
"disconnected",
"solid",
"dotted",
"dotdashed",
"shortdashed",
"longdashed"
};
int mode = 1;
double ident(double x){
return(x);
}
struct z {
float lb,ub,mult,quant;
};
struct {
char *name;
int next;
} palette[] = {
['b'] { "blue", 'b' },
['c'] { "cyan", 'c' },
['g'] { "green", 'g' },
['k'] { "kblack", 'k' },
['m'] { "magenta", 'm' },
['r'] { "red", 'r' },
['w'] { "white", 'w' },
['y'] { "yellow", 'y' }
};
int pencolor = 'k';
void init(struct xy *);
void setopt(int, char *[]);
void readin(void);
void transpose(void);
void getlim(struct xy *, struct val *);
void equilibrate(struct xy *, struct xy *);
void scale(struct xy *);
void limread(struct xy *, int *, char ***);
int numb(float *, int *, char ***);
void colread(int *, char ***);
int copystring(int);
struct z setloglim(int, int, float, float);
struct z setlinlim(int, int, float, float);
void axes(void);
int setmark(int *, struct xy *);
void submark(int *, int *, float, struct xy *);
void plot(void);
int getfloat(float *);
int getstring(void);
void title(void);
void badarg(void);
int conv(float, struct xy *, int *);
int symbol(int, int, int);
void axlab(char, struct xy *, char *);
void main(int argc,char *argv[]){
openpl();
range(0,0,4096,4096);
init(&xd);
init(&yd);
xd.xsize = yd.xsize = 1.;
xx = (struct val *)malloc((unsigned)sizeof(struct val));
labels = malloc(1);
labels[labelsiz++] = 0;
setopt(argc,argv);
if(erasf)
erase();
readin();
transpose();
getlim(&xd,(struct val *)&xx->xv);
getlim(&yd,(struct val *)&xx->yv);
if(equf) {
equilibrate(&xd,&yd);
equilibrate(&yd,&xd);
}
scale(&xd);
scale(&yd);
axes();
title();
plot();
closepl();
exits(0);
}
void init(struct xy *p){
p->xf = ident;
p->xmult = 1;
}
void setopt(int argc, char *argv[]){
char *p1, *p2;
float temp;
xd.xlb = yd.xlb = INF;
xd.xub = yd.xub = -INF;
while(--argc > 0) {
argv++;
again: switch(argv[0][0]) {
case '-':
argv[0]++;
goto again;
case 'l': /* label for plot */
p1 = titlebuf;
if (argc>=2) {
argv++;
argc--;
p2 = argv[0];
while (*p1++ = *p2++);
}
break;
case 'd': /*disconnected,obsolete option*/
case 'm': /*line mode*/
mode = 0;
if(!numb(&temp,&argc,&argv))
break;
if(temp>=sizeof(modes)/sizeof(*modes))
mode = 1;
else if(temp>=-1)
mode = temp;
break;
case 'o':
if(numb(&temp,&argc,&argv) && temp>=1)
ovlay = temp;
break;
case 'a': /*automatic abscissas*/
absf = 1;
dx = 1;
if(!numb(&dx,&argc,&argv))
break;
if(numb(&absbot,&argc,&argv))
absf = 2;
break;
case 's': /*save screen, overlay plot*/
erasf = 0;
break;
case 'g': /*grid style 0 none, 1 ticks, 2 full*/
gridf = 0;
if(!numb(&temp,&argc,&argv))
temp = argv[0][1]-'0'; /*for caompatibility*/
if(temp>=0&&temp<=2)
gridf = temp;
break;
case 'c': /*character(s) for plotting*/
if(argc >= 2) {
symbf = 1;
plotsymb = argv[1];
argv++;
argc--;
}
break;
case 't': /*transpose*/
transf = 1;
break;
case 'e': /*equal scales*/
equf = 1;
break;
case 'b': /*breaks*/
brkf = 1;
break;
case 'x': /*x limits */
limread(&xd,&argc,&argv);
break;
case 'y':
limread(&yd,&argc,&argv);
break;
case 'h': /*set height of plot */
if(!numb(&yd.xsize, &argc,&argv))
badarg();
break;
case 'w': /*set width of plot */
if(!numb(&xd.xsize, &argc, &argv))
badarg();
break;
case 'r': /* set offset to right */
if(!numb(&xd.xoff, &argc, &argv))
badarg();
break;
case 'u': /*set offset up the screen*/
if(!numb(&yd.xoff,&argc,&argv))
badarg();
break;
case 'p': /*pen color*/
colread(&argc, &argv);
break;
default:
badarg();
}
}
}
void limread(struct xy *p, int *argcp, char ***argvp){
if(*argcp>1 && (*argvp)[1][0]=='l') {
(*argcp)--;
(*argvp)++;
p->xf = log10;
}
if(!numb(&p->xlb,argcp,argvp))
return;
p->xlbf = 1;
if(!numb(&p->xub,argcp,argvp))
return;
p->xubf = 1;
if(!numb(&p->xquant,argcp,argvp))
return;
p->xqf = 1;
}
isdigit(char c){
return '0'<=c && c<='9';
}
numb(float *np, int *argcp, char ***argvp){
char c;
if(*argcp <= 1)
return(0);
while((c=(*argvp)[1][0]) == '+')
(*argvp)[1]++;
if(!(isdigit(c) || c=='-'&&(*argvp)[1][1]<'A' || c=='.'))
return(0);
*np = atof((*argvp)[1]);
(*argcp)--;
(*argvp)++;
return(1);
}
void colread(int *argcp, char ***argvp){
int c, cnext;
int i, n;
if(*argcp<=1)
return;
n = strlen((*argvp)[1]);
if(strspn((*argvp)[1], "bcgkmrwy")!=n)
return;
pencolor = cnext = (*argvp)[1][0];
for(i=0; i<n-1; i++){
c = (unsigned char)(*argvp)[1][i];
cnext = (unsigned char)(*argvp)[1][i+1];
palette[c].next = cnext;
}
palette[cnext].next = pencolor;
(*argcp)--;
(*argvp)++;
}
void readin(void){
int i, t;
struct val *temp;
if(absf==1) {
if(xd.xlbf)
absbot = xd.xlb;
else if(xd.xf==log10)
absbot = 1;
}
for(;;) {
temp = (struct val *)realloc((char*)xx,
(unsigned)(n+ovlay)*sizeof(struct val));
if(temp==0)
return;
xx = temp;
if(absf)
xx[n].xv = n*dx/ovlay + absbot;
else
if(!getfloat(&xx[n].xv))
return;
t = 0; /* silence compiler */
for(i=0;i<ovlay;i++) {
xx[n+i].xv = xx[n].xv;
if(!getfloat(&xx[n+i].yv))
return;
xx[n+i].lblptr = -1;
t = getstring();
if(t>0)
xx[n+i].lblptr = copystring(t);
if(t<0 && i+1<ovlay)
return;
}
n += ovlay;
if(t<0)
return;
}
}
void transpose(void){
int i;
float f;
struct xy t;
if(!transf)
return;
t = xd; xd = yd; yd = t;
for(i= 0;i<n;i++) {
f = xx[i].xv; xx[i].xv = xx[i].yv; xx[i].yv = f;
}
}
int copystring(int k){
char *temp;
int i;
int q;
temp = realloc(labels,(unsigned)(labelsiz+1+k));
if(temp==0)
return(0);
labels = temp;
q = labelsiz;
for(i=0;i<=k;i++)
labels[labelsiz++] = labbuf[i];
return(q);
}
float modceil(float f, float t){
t = fabs(t);
return(ceil(f/t)*t);
}
float
modfloor(float f, float t){
t = fabs(t);
return(floor(f/t)*t);
}
void getlim(struct xy *p, struct val *v){
int i;
i = 0;
do {
if(!p->xlbf && p->xlb>v[i].xv)
p->xlb = v[i].xv;
if(!p->xubf && p->xub<v[i].xv)
p->xub = v[i].xv;
i++;
} while(i < n);
}
void setlim(struct xy *p){
float t,delta,sign;
struct z z;
int mark[50];
float lb,ub;
int lbf,ubf;
lb = p->xlb;
ub = p->xub;
delta = ub-lb;
if(p->xqf) {
if(delta*p->xquant <=0 )
badarg();
return;
}
sign = 1;
lbf = p->xlbf;
ubf = p->xubf;
if(delta < 0) {
sign = -1;
t = lb;
lb = ub;
ub = t;
t = lbf;
lbf = ubf;
ubf = t;
}
else if(delta == 0) {
if(ub > 0) {
ub = 2*ub;
lb = 0;
}
else
if(lb < 0) {
lb = 2*lb;
ub = 0;
}
else {
ub = 1;
lb = -1;
}
}
if(p->xf==log10 && lb>0 && ub>lb) {
z = setloglim(lbf,ubf,lb,ub);
p->xlb = z.lb;
p->xub = z.ub;
p->xmult *= z.mult;
p->xquant = z.quant;
if(setmark(mark,p)<2) {
p->xqf = lbf = ubf = 1;
lb = z.lb; ub = z.ub;
} else
return;
}
z = setlinlim(lbf,ubf,lb,ub);
if(sign > 0) {
p->xlb = z.lb;
p->xub = z.ub;
} else {
p->xlb = z.ub;
p->xub = z.lb;
}
p->xmult *= z.mult;
p->xquant = sign*z.quant;
}
struct z
setloglim(int lbf, int ubf, float lb, float ub){
float r,s,t;
struct z z;
for(s=1; lb*s<1; s*=10) ;
lb *= s;
ub *= s;
for(r=1; 10*r<=lb; r*=10) ;
for(t=1; t<ub; t*=10) ;
z.lb = !lbf ? r : lb;
z.ub = !ubf ? t : ub;
if(ub/lb<100) {
if(!lbf) {
if(lb >= 5*z.lb)
z.lb *= 5;
else if(lb >= 2*z.lb)
z.lb *= 2;
}
if(!ubf) {
if(ub*5 <= z.ub)
z.ub /= 5;
else if(ub*2 <= z.ub)
z.ub /= 2;
}
}
z.mult = s;
z.quant = r;
return(z);
}
struct z
setlinlim(int lbf, int ubf, float xlb, float xub){
struct z z;
float r,s,delta;
float ub,lb;
loop:
ub = xub;
lb = xlb;
delta = ub - lb;
/*scale up by s, a power of 10, so range (delta) exceeds 1*/
/*find power of 10 quantum, r, such that delta/10<=r<delta*/
r = s = 1;
while(delta*s < 10)
s *= 10;
delta *= s;
while(10*r < delta)
r *= 10;
lb *= s;
ub *= s;
/*set r=(1,2,5)*10**n so that 3-5 quanta cover range*/
if(r>=delta/2)
r /= 2;
else if(r<delta/5)
r *= 2;
z.ub = ubf? ub: modceil(ub,r);
z.lb = lbf? lb: modfloor(lb,r);
if(!lbf && z.lb<=r && z.lb>0) {
xlb = 0;
goto loop;
}
else if(!ubf && z.ub>=-r && z.ub<0) {
xub = 0;
goto loop;
}
z.quant = r;
z.mult = s;
return(z);
}
void scale(struct xy *p){
float edge;
setlim(p);
edge = top-bot;
p->xa = p->xsize*edge/((*p->xf)(p->xub) - (*p->xf)(p->xlb));
p->xbot = bot + edge*p->xoff;
p->xtop = p->xbot + (top-bot)*p->xsize;
p->xb = p->xbot - (*p->xf)(p->xlb)*p->xa + .5;
}
void equilibrate(struct xy *p, struct xy *q){
if(p->xlbf|| /* needn't test xubf; it implies xlbf*/
q->xubf&&q->xlb>q->xub)
return;
if(p->xlb>q->xlb) {
p->xlb = q->xlb;
p->xlbf = q->xlbf;
}
if(p->xub<q->xub) {
p->xub = q->xub;
p->xubf = q->xubf;
}
}
void axes(void){
int i;
int mark[50];
int xn, yn;
if(gridf==0)
return;
line(xd.xbot,yd.xbot,xd.xtop,yd.xbot);
vec(xd.xtop,yd.xtop);
vec(xd.xbot,yd.xtop);
vec(xd.xbot,yd.xbot);
xn = setmark(mark,&xd);
for(i=0; i<xn; i++) {
if(gridf==2)
line(mark[i],yd.xbot,mark[i],yd.xtop);
if(gridf==1) {
line(mark[i],yd.xbot,mark[i],yd.xbot+tick);
line(mark[i],yd.xtop-tick,mark[i],yd.xtop);
}
}
yn = setmark(mark,&yd);
for(i=0; i<yn; i++) {
if(gridf==2)
line(xd.xbot,mark[i],xd.xtop,mark[i]);
if(gridf==1) {
line(xd.xbot,mark[i],xd.xbot+tick,mark[i]);
line(xd.xtop-tick,mark[i],xd.xtop,mark[i]);
}
}
}
int
setmark(int *xmark, struct xy *p){
int xn = 0;
float x,xl,xu;
float q;
if(p->xf==log10&&!p->xqf) {
for(x=p->xquant; x<p->xub; x*=10) {
submark(xmark,&xn,x,p);
if(p->xub/p->xlb<=100) {
submark(xmark,&xn,2*x,p);
submark(xmark,&xn,5*x,p);
}
}
} else {
xn = 0;
q = p->xquant;
if(q>0) {
xl = modceil(p->xlb+q/6,q);
xu = modfloor(p->xub-q/6,q)+q/2;
} else {
xl = modceil(p->xub-q/6,q);
xu = modfloor(p->xlb+q/6,q)-q/2;
}
for(x=xl; x<=xu; x+=fabs(p->xquant))
xmark[xn++] = (*p->xf)(x)*p->xa + p->xb;
}
return(xn);
}
void submark(int *xmark, int *pxn, float x, struct xy *p){
if(1.001*p->xlb < x && .999*p->xub > x)
xmark[(*pxn)++] = log10(x)*p->xa + p->xb;
}
void plot(void){
int ix,iy;
int i,j;
int conn;
for(j=0;j<ovlay;j++) {
switch(mode) {
case -1:
pen(modes[j%(sizeof modes/sizeof *modes-1)+1]);
break;
case 0:
break;
default:
pen(modes[mode]);
}
color(palette[pencolor].name);
conn = 0;
for(i=j; i<n; i+=ovlay) {
if(!conv(xx[i].xv,&xd,&ix) ||
!conv(xx[i].yv,&yd,&iy)) {
conn = 0;
continue;
}
if(mode!=0) {
if(conn != 0)
vec(ix,iy);
else
move(ix,iy);
conn = 1;
}
conn &= symbol(ix,iy,xx[i].lblptr);
}
pencolor = palette[pencolor].next;
}
pen(modes[1]);
}
int
conv(float xv, struct xy *p, int *ip){
long ix;
ix = p->xa*(*p->xf)(xv*p->xmult) + p->xb;
if(ix<p->xbot || ix>p->xtop)
return(0);
*ip = ix;
return(1);
}
int
getfloat(float *p){
int i;
i = scanf("%f",p);
return(i==1);
}
int
getstring(void){
int i;
char junk[20];
i = scanf("%1s",labbuf);
if(i==-1)
return(-1);
switch(*labbuf) {
default:
if(!isdigit(*labbuf)) {
ungetc(*labbuf,stdin);
i = scanf("%s",labbuf);
break;
}
case '.':
case '+':
case '-':
ungetc(*labbuf,stdin);
return(0);
case '"':
i = scanf("%[^\"\n]",labbuf);
scanf("%[\"]",junk);
break;
}
if(i==-1)
return(-1);
return(strlen(labbuf));
}
int
symbol(int ix, int iy, int k){
if(symbf==0&&k<0) {
if(mode==0)
point(ix,iy);
return(1);
}
else {
move(ix,iy);
text(k>=0?labels+k:plotsymb);
move(ix,iy);
return(!brkf|k<0);
}
}
void title(void){
char buf[BSIZ+100];
buf[0] = ' ';
buf[1] = ' ';
buf[2] = ' ';
strcpy(buf+3,titlebuf);
if(erasf&&gridf) {
axlab('x',&xd,buf);
strcat(buf,",");
axlab('y',&yd,buf);
}
move(xd.xbot,yd.xbot-60);
text(buf);
}
void axlab(char c, struct xy *p, char *b){
char *dir;
dir = p->xlb<p->xub? "<=": ">=";
sprintf(b+strlen(b), " %g %s %c%s %s %g", p->xlb/p->xmult,
dir, c, p->xf==log10?" (log)":"", dir, p->xub/p->xmult);
}
void badarg(void){
fprintf(stderr,"graph: error in arguments\n");
closepl();
exits("bad arg");
}