Subversion Repositories wimsdev

/*    Copyright (C) 1998-2003 XIAO, Gang of Universite de Nice - Sophia Antipolis

 *

 *  This program is free software; you can redistribute it and/or modify

 *  it under the terms of the GNU General Public License as published by

 *  the Free Software Foundation; either version 2 of the License, or

 *  (at your option) any later version.

 *

 *  This program is distributed in the hope that it will be useful,

 *  but WITHOUT ANY WARRANTY; without even the implied warranty of

 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 *  GNU General Public License for more details.

 *

 *  You should have received a copy of the GNU General Public License

 *  along with this program; if not, write to the Free Software

 *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

 */

        /* Interface gnuplot to wims */

/*************** Customization: change values hereafter ****************/

#define tmpdir "/tmp"

        /* limit of input/output file sizes */

#define fsizelim 131072

        /* limit of parameter string */

#define parmlim 131072

        /* number of versions */

#define versions 2

        /* gp prompt string */

#define gpprompt "\n? "

        /* string which defines the start of gp output

         * (in order to cut gp header) */

#define startstring "start line 49521845"

        /* This is the good bye string of gp, signaling end of output. */

#define goodbyestring "? Good bye!"

char header[]="Pi=pi\n\

PI=pi\n\

e=exp(1)\n\

E=exp(1)\n\

ln(x) = log(x)\n\

ch(x) = cosh(x)\n\

sh(x) = sinh(x)\n\

th(x) = tanh(x)\n\

arccos(x) = acos(x)\n\

arcsin(x) = asin(x)\n\

tg(x)     = tan(x)\n\

arctan(x) = atan(x)\n\

arctg(x)  = atan(x)\n\

Argch(x)  = acosh(x)\n\

Argsh(x)  = asinh(x)\n\

Argth(x)  = atanh(x)\n\

argch(x)  = acosh(x)\n\

argsh(x)  = asinh(x)\n\

argth(x)  = atanh(x)\n\

set samples 400\n\

";

/***************** Nothing should need change hereafter *****************/

#include <stdio.h>

#include <stdlib.h>

#include <stdarg.h>

#include <unistd.h>

#include <string.h>

#include <ctype.h>

#include <time.h>

#include <sys/time.h>

#define SETUP_NO (sizeof(setups)/sizeof(setups[0]))

#define MYNAME_NO (sizeof(name2version)/sizeof(name2version[0]))

        /* gpversion=0: gp1.39; gpversion=1: gp2.0. */

int gpversion, mypid, precision;

char inputfname[256], outputfname[256];

char *parm;

void *xmalloc(size_t n)

{

    void *p;

    p=malloc(n);

    if(p==NULL) {

        fprintf(stderr, "pari: Malloc failure.");

        exit(1);

    }

    return p;

}

        /* system(), but with variable parms */

int call_sh(char *s,...)

{

    va_list vp;

    char buf[1024];

    va_start(vp,s);

    vsnprintf(buf,sizeof(buf),s,vp);

    va_end(vp);

    return system(buf);

}

        /* check for security violations in command string */

void check_parm(void)

{

    char *p;

    for(p=parm;*p!=0;p++) {

          /* no espace commands. */

        if(*p!='\n') continue;

        while(*p!=0 && isspace(*p)) p++;

        if(*p=='\\' && *(p+1)!='v') *p='.';

    }

}

        /* patches the gnuplot integer division (mis)feature. */

void gnuplot_patch(char *p,int oneline)

{

    char *pp;

    for(pp=strchr(p,'/');pp!=NULL;pp=strchr(pp+1,'/')) {

        char *p1;

        if(pp<=p || !isdigit(*(pp-1)) || !isdigit(*(pp+1))) continue;

        for(p1=pp-2;p1>=p && isdigit(*p1);p1--);

        if(p1>=p && *p1=='.') continue;

        for(p1=pp+2;*p1 && isdigit(*p1);p1++);

        if(*p1=='.') continue;

        string_modify(p,p1,p1,".0");

    }

    for(pp=strchr(p,'^');pp!=NULL;pp=strchr(pp+1,'^'))

      string_modify(p,pp,pp+1,"**");

        /* disallow new lines and ';' */

    if(oneline)

      for(pp=p;*pp!=0;pp++) if(*pp==';' || *pp=='\n') *pp=' ';

}

        /* This is to disable pipe in the gnuplot plotting function.

         * We do not allow ' followed by < . */

void prepare_insplot_parm(char *p)

{

    int i,j; char *pp, *s;

    double d;

    char setbuf[MAX_LINELEN+10],buf[MAX_LINELEN+1];

    j=strlen(p);

      /* pipe in plot command */

    for(i=0;i<j;i++) {

        if(*(p+i)!='\'' && *(p+i)!='"') continue;

        pp=find_word_start(p+i+1); if(*pp=='<') {

            module_error("illegal_plot_cmd"); exit(1);

        }

    }

    gnuplot_patch(p,1);

        /* multiplot */

    pp=getvar("insplot_split");i=evalue(pp);

        /* arbitrary limit: 16 multiplots */

    if(i>0 && i<=16) {

        char tbuf[MAX_LINELEN+1];

        snprintf(buf,sizeof(buf),"%d",i); setenv("multiplot",buf,1);

        for(j=1;j<=i;j++) {

            snprintf(buf,sizeof(buf),"insplot_parm_%d",j);

            pp=getvar(buf);

            snprintf(tbuf,sizeof(tbuf),"%s",pp);

            gnuplot_patch(tbuf,1);

            setenv(buf,tbuf,1);

        }

    }

        /* no illegal chaining */

    pp=getvar("insplot_font"); if(pp!=NULL) {

        for(s=pp;s<pp+MAX_LINELEN && *s;s++)

          if(*s==';' || *s=='\n' || *s==' ') *s=0;

        if(s>=pp+MAX_LINELEN) *s=0;

        setvar("insplot_font",pp);

    }

    pp=getvar("insplot_set"); if(pp!=NULL) {

        char tbuf[MAX_LINELEN+1];

        snprintf(tbuf,sizeof(tbuf),"%s",pp);

        i=strlen(tbuf)-1;

        while(i>0 && isspace(tbuf[i])) i--;

        if(tbuf[i]==';') tbuf[i]=0;

        gnuplot_patch(tbuf,0);pp=tbuf;

        ovlstrcpy(setbuf,"set "); j=strlen("set ");

        for(i=0; *(pp+i)!=0 && j<MAX_LINELEN; i++) {

            if(*(pp+i)=='\n') {setbuf[j++]=' '; continue;}

            if(*(pp+i)!=';') {setbuf[j++]=*(pp+i); continue;}

            ovlstrcpy(setbuf+j,"\nset "); j+=strlen("\nset ");

        }

        setbuf[j]=0;

        setenv("insplot_set",setbuf,1);

    }

    else setenv("insplot_set","",1);

      /* frames of animation */

    pp=getvar("ins_anim_frames");

    if(pp!=NULL) i=evalue(pp); else i=1;

    if(i>=INS_LIMIT) i=INS_LIMIT-1; if(i<1) i=1;

    if(strstr(setbuf,"step")==NULL && strstr(p,"step")==NULL

       && varchr(setbuf,"s")==NULL && varchr(p,"s")==NULL) i=1;

    snprintf(buf,sizeof(buf),"%d",i);

    setenv("ins_anim_frames",buf,1);

    setvar("ins_anim_frames","");

      /* delay of animation */

    pp=getvar("ins_anim_delay");

    if(pp!=NULL) d=evalue(pp); else d=0;

    if(d>=10) d=10; if(d<0) d=0;

    i=d*100;

    snprintf(buf,sizeof(buf),"%d",i);

    setenv("ins_anim_delay",buf,1);

}

        /* strip trailing zeros */

void strip_zeros(char *p)

{

    char *pp, *p2, *numend, *ee;

    int i;

    for(pp=p;*pp!=0;pp++) {

        if(!isdigit(*pp)) continue;

        i=0;

        for(numend=pp;isdigit(*numend) || *numend=='.';numend++)

          if(*numend=='.') i=1;

        if(i==0) {

            pp=numend-1;continue;

        }

        for(p2=numend;p2>pp && *(p2-1)=='0';p2--);

        for(ee=numend;isspace(*ee);ee++);

        if(*(pp+1)=='.' && (*ee=='E' || *ee=='e')

           && *(ee+1)=='-') {

            int k=0;

            char *pt=ee+2;

            while(isdigit(*pt)) {

                k*=10;k+=*pt-'0';pt++;

            }

            if(k>precision*2 || (k>precision && *pp=='0')) {

                sprintf(pp,"0.0%s",pt);

                pp+=strlen("0.0")-1;

                continue;

            }

        }

        if(*(p2-1)=='.' && p2<numend) p2++;

        if(p2<numend) {

            ovlstrcpy(p2,numend);numend=p2;

        }

        pp=numend-1;

    }

}

        /* process and print gp output */

void output(char *p)

{

    int i,n;

    char *pp, *pe, *pt;

    pp=strstr(p,startstring);

    if(pp==NULL) return;

    pp=strstr(pp,gpprompt); if(pp==NULL) return;

    pe=strstr(pp,goodbyestring);

    if(pe>=pp) *pe=0;

    while(pp!=NULL) {

        pp++;

        pe=strstr(pp,gpprompt);

        if(pe>=pp) *pe=0;

        pp=strchr(pp,'\n');

        if(pp==NULL) {

            emptyline:

            puts(""); pp=pe; continue;

        }

        pp++; n=strlen(pp);

        if(n==0) goto emptyline;

                /* make every output one-line */

        for(i=0;i<n;i++) {

            if(*(pp+i)=='\n') {

                if(*(pp+i+1)!='%') *(pp+i)=' ';

                else {*(pp+i)=0; break;}

            }

        }

          /* strip leading and trailing spaces */

        while(isspace(*pp) && pp<pe) pp++;

        pt=pp+strlen(pp)-1;

        while(isspace(*pt) && pt>pp) pt--;

          /* remove parentheses of matrix output */

        if(memcmp(pp,"Mat(",4)==0 && *pt==')') {

            *(pt--)=0; pp+=4;

        }

        if(*pp=='[' && *pt==']') {

            *(pt--)=0; pp++;

        }

        strip_zeros(pp);

        puts(pp); pp=pe;

    }

}

void call_gnuplot(void)

{

    int i,r;

    char *p;

    FILE *ff;

    struct timeval t;

    setenv(gprcenv,gprc[gpversion],1);

    mypid=getpid();

    snprintf(inputfname,sizeof(inputfname),"%s/gp_input.%d",tmpdir,mypid);

    snprintf(outputfname,sizeof(outputfname),"%s/gp_output.%d",tmpdir,mypid);

    ff=fopen(inputfname,"w");

    if(ff==NULL) {

        fprintf(stderr,"insplot: cannot open file %s.\n",inputfname); exit(1);

    }

    gettimeofday(&t,NULL);

    r=t.tv_usec*t.tv_sec;

    fprintf(ff,"\nsetrand(%d)\n",r);

    for(i=0;i<SETUP_NO;i++) {

        p=getenv(setups[i].wname);

        if(p==NULL || *p==0) p=setups[i].defaultval;

        fprintf(ff,"%s%s\n",setups[i].gpset[gpversion],p);

        if(strstr(setups[i].wname,"pari_precision")!=NULL)

          precision=atoi(p);

        if(precision<0) precision=-precision;

    }

    fputs(header,ff);

    fprintf(ff,"print(\"%s\")\n",startstring);

    check_parm();

    fwrite(parm,1,strlen(parm),ff);

    fclose(ff);

    call_sh("%s <%s >%s",gpcmd[gpversion],inputfname,outputfname);

    ff=fopen(outputfname,"r");

    if(ff!=NULL) {

        long int l;

        char *obuf;

        fseek(ff,0,SEEK_END); l=ftell(ff); fseek(ff,0,SEEK_SET);

        if(l<0) l=0; if(l>fsizelim) l=fsizelim;

        obuf=xmalloc(l+1);

        l=fread(obuf,1,l,ff); fclose(ff);

        if(l>0) obuf[l]=0; else obuf[0]=0;

        output(obuf);  

    }

    unlink(inputfname); unlink(outputfname);

}

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

{

    char *p;

    int i;

        /* Must have at least 2 parameters. */

    if(argc<3) return 0;

    parm=getenv("ins_source");

        /* nothing to do if no parameter */

    if(parm==NULL || *parm==0) return 0;

    i=strlen(parm); if(i<0 || i>parmlim) {

        fprintf(stderr,"insplot..processor: parameter too long. \n"); exit(1);

    }

    call_gnuplot();

    return 0;    

}