void yaxis(
real x0,
real y0, /* axis starting point */
real yl, /* axis length */
real tick[], /* tick values or limits */
int nticks, /* number of ticks */
axis_proc ytrans, /* plotting transformation */
string label /* label for axis */
)
{
int i;
real t, y;
char val[32];
plmove(x0, y0);
plline(x0, y0 + yl);
for (i = 0; i < abs(nticks); i++) {
if (nticks > 0)
t = tick[i];
else
t = tick[0] + (i + 1) * (tick[1] - tick[0]) / (1.0 - nticks);
y = (*ytrans)(t);
if (label != NULL) {
plmove(x0 + ytikrt, y);
plline(x0 - ytiklf, y);
sprintf(val, "%-.*f", nydig, t);
trimval(val);
if (! formalaxis) {
pljust(0);
pltext(val, x0 - ynumlf, y, ysznum, 90.0);
} else {
pljust(1);
pltext(val, x0 - ynumlf, y, ysznum, 0.0);
}
} else {
plmove(x0 - ytikrt, y);
plline(x0 + ytiklf, y);
}
}
if (label != NULL && *label != 0) {
if (! formalaxis) {
pljust(0);
pltext(label, x0 - ylablf, y0 + yl / 2, yszlab, 90.0);
} else {
pljust(1);
#ifdef FUDGEAXIS
pltext(label, x0 - ylablf, y0 + yl - yszlab/4.0, yszlab, 0.0);
#else
pltext(label, x0 - ylablf, y0 + yl - yszlab/2.0, yszlab, 0.0);
#endif
}
}
pljust(-1);
}
void xaxis(
real x0, real y0, /* axis starting point */
real xl, /* axis length */
real tick[], /* tick values or limits */
int nticks, /* number of ticks */
axis_proc xtrans, /* plotting transformation */
string label /* label for axis */
)
{
int i;
real t, x;
char val[32];
plmove(x0, y0);
plline(x0 + xl, y0);
for (i = 0; i < abs(nticks); i++) {
if (nticks > 0)
t = tick[i];
else
t = tick[0] + (i + 1) * (tick[1] - tick[0]) / (1.0 - nticks);
x = (*xtrans)(t);
if (label != NULL) {
plmove(x, y0 + xtikup);
plline(x, y0 - xtikdn);
sprintf(val, "%-.*f", nxdig, t);
trimval(val);
pljust(0);
pltext(val, x, y0 - xnumdn, xsznum, 0.0);
} else {
plmove(x, y0 - xtikup);
plline(x, y0 + xtikdn);
}
}
if (label != NULL && *label != 0) {
if (! formalaxis) {
pljust(0);
pltext(label, x0 + xl / 2, y0 - xlabdn, xszlab, 0.0);
} else {
pljust(1);
#ifdef FUDGEAXIS
pltext(label, x0 + xl, y0 - xlabdn + xszlab/4.0, xszlab, 0.0);
#else
pltext(label, x0 + xl, y0 - xlabdn, xszlab, 0.0);
#endif
}
}
pljust(-1);
}
void plot_path(orbitptr o, real tstart, real tend, int n)
{
int i, plotfirst = 1;
real x,y;
for (i=0; i<(Nsteps(o)-n); i += n) {
if ((tstart<Torb(optr,i)) && (Torb(optr,i)<tend)) {
x = xtrans(Posorb(o,i,xvar_idx));
y = ytrans(Posorb(o,i,yvar_idx));
if (plotfirst) {
plpoint (x,y); /* mark first point */
plotfirst = 0;
}
plmove (x,y);
x = xtrans(Posorb(o,i+n,xvar_idx));
y = ytrans(Posorb(o,i+n,yvar_idx));
plline (x,y);
}
}
}
int pl_hitpopup(Panel *g, Mouse *m){
Panel *p;
Point d;
Popup *pp;
pp=g->data;
if(g->state==UP){
switch(m->buttons&7){
case 0: p=g->child; break;
case 1: p=pp->pop[0]; g->state=DOWN1; break;
case 2: p=pp->pop[1]; g->state=DOWN2; break;
case 4: p=pp->pop[2]; g->state=DOWN3; break;
default: p=0; break;
}
if(p==0){
p=g->child;
g->state=DOWN;
}
else if(g->state!=UP){
plpack(p, screen->clipr);
if(p->lastmouse)
d=subpt(m->xy, divpt(addpt(p->lastmouse->r.min,
p->lastmouse->r.max), 2));
else
d=subpt(m->xy, divpt(addpt(p->r.min, p->r.max), 2));
if(p->r.min.x+d.x<g->r.min.x) d.x=g->r.min.x-p->r.min.x;
if(p->r.max.x+d.x>g->r.max.x) d.x=g->r.max.x-p->r.max.x;
if(p->r.min.y+d.y<g->r.min.y) d.y=g->r.min.y-p->r.min.y;
if(p->r.max.y+d.y>g->r.max.y) d.y=g->r.max.y-p->r.max.y;
plmove(p, d);
pp->save=allocimage(display, p->r, g->b->chan, 0, DNofill);
if(pp->save!=0) draw(pp->save, p->r, g->b, 0, p->r.min);
pl_invis(p, 0);
pldraw(p, g->b);
}
}
else{
switch(g->state){
default: SET(p); break; /* can't happen! */
case DOWN1: p=pp->pop[0]; break;
case DOWN2: p=pp->pop[1]; break;
case DOWN3: p=pp->pop[2]; break;
case DOWN: p=g->child; break;
}
if((m->buttons&7)==0){
if(g->state!=DOWN){
if(pp->save!=0){
draw(g->b, p->r, pp->save, 0, p->r.min);
flushimage(display, 1);
freeimage(pp->save);
pp->save=0;
}
pl_invis(p, 1);
}
g->state=UP;
}
}
plmouse(p, m);
if((m->buttons&7)==0)
g->state=UP;
return (m->buttons&7)!=0;
}
out_slit()
{
real xsky, ysky, vrad, inv_surden, sigma, mass;
real xslit, yslit, xplt, yplt, sinpa, cospa;
real m_max, v_min, v_max, s_max; /* local min/max */
int i, islit;
Body *bp;
for (islit=0; islit<nslit; islit++) /* reset local variables */
v0star[islit] = v1star[islit] = v2star[islit] = 0.0;
m_max = v_min = v_max = s_max = 0.0;
inv_surden = 1.0 / (slit_width*slit_cell);
sinpa = sin(pa); cospa = cos(pa);
for(bp=btab, i=0; i<nobj; bp++, i++) { /* loop over all particles */
xsky = xvar(bp,tsnap,i);
ysky = yvar(bp,tsnap,i);
vrad = zvar(bp,tsnap,i);
mass = evar(bp,tsnap,i) * inv_surden;
xsky -= origin[0]; /* translate to slit origin */
ysky -= origin[1];
xslit = -cospa*ysky + sinpa*xsky; /* and rotate to slit frame */
yslit = sinpa*ysky + cospa*xsky; /* !!! check signs !!! */
if (fabs(yslit) > 0.5*slit_width)
continue; /* not in slit */
islit = (xslit+0.5*slit_length)/slit_cell;
if (islit<0 || islit>=nslit)
continue; /* not in slit */
v0star[islit] += mass;
v1star[islit] += vrad * mass;
v2star[islit] += sqr(vrad) * mass;
} /*-- end particles loop --*/
while (nsmooth-- > 0) { /* convolution */
dprintf (0,"Convolving with %d-length beam: ",lsmooth);
for (i=0; i<lsmooth; i++)
dprintf (0,"%f ",smooth[i]);
convolve (v0star, nslit, smooth, lsmooth);
convolve (v1star, nslit, smooth, lsmooth);
convolve (v2star, nslit, smooth, lsmooth);
dprintf (0,"\n");
}
for (islit=0; islit<nslit; islit++) { /* moment analysis: M, MV and MV^2 */
if (v0star[islit]==0.0)
continue; /* no data - skip to next pixel */
v1star[islit] /= v0star[islit];
sigma = v2star[islit]/v0star[islit] - sqr(v1star[islit]);
if (sigma<0.0) { /* should never happen */
warning("islit=%d sigma^2=%e < 0 !!!\n",islit,sigma);
v2star[islit] = 0.0;
continue; /* something really wrong */
}
v2star[islit] = sqrt(sigma);
if (v0star[islit] > m_max) m_max = v0star[islit];
if (v1star[islit] < v_min) v_min = v1star[islit];
if (v1star[islit] > v_max) v_max = v1star[islit];
if (v2star[islit] > s_max) s_max = v2star[islit];
if (Qtab) {
xslit = islit*slit_cell;
printf ("%g %g %g %g\n",
xslit,v0star[islit], v1star[islit], v2star[islit]);
}
} /* for(islit) */
if (Qtab)
return(0);
plinit ("***", 0.0, 20.0, 0.0, 20.0);
/* reset default autoscales to user supplied if necessary */
if (mmax==0.0) mmax=m_max;
if (vmin==0.0) vmin=v_min;
if (vmax==0.0) vmax=v_max;
if (smax==0.0) smax=s_max;
dprintf (0,"mmax=%f vmin=%f vmax=%f smax=%f reset to:\n",m_max,v_min,v_max,s_max);
if (mmax==0) mmax=1;
if (vmin==0 && vmax==0) vmax=1;
if (smax==0) smax=1;
dprintf (0,"mmax=%f vmin=%f vmax=%f smax=%f \n",mmax, vmin, vmax, smax);
/* general plot header */
sprintf (plabel,"File: %s; var{%s,%s,%s,%s} slit{%s %s %s %s}",
infile,getparam("xvar"),getparam("yvar"),
getparam("zvar"),getparam("evar"),
getparam("origin"),getparam("pa"),getparam("width"),
getparam("length"),getparam("cell"));
pltext (plabel,2.0,18.4, 0.32, 0.0);
#if 0
if (*headline!=NULL) /* identification */
pltext (headline,2.0,19.0,0.25,0.0);
#endif
xplot[0] = -0.5*slit_length; /* PLOT1: upper panel */
xplot[1] = 0.5*slit_length;
sprintf(xlabel,"slit: {x=%s,y=%s}",getparam("xvar"),getparam("yvar"));
yplot[0]=0.0;
yplot[1]=mmax;
strcpy (ylabel,"mass surface density");
xaxis ( 2.0,12.0, 16.0, xplot, -7, xtrans, NULL);
xaxis ( 2.0,17.0, 16.0, xplot, -7, xtrans, NULL);
yaxis ( 2.0,12.0, 5.0, yplot, -3, ytransm, ylabel);
yaxis (18.0,12.0, 5.0, yplot, -3, ytransm, NULL);
for (islit=0; islit<nslit; islit++) {
xplt = xtrans (-0.5*slit_length + (islit+0.5)*slit_cell);
yplt = ytransm (v0star[islit]);
plbox (xplt, yplt, SYMBOLSIZE);
}
yplot[0]=vmin; /* PLOT2: middle panel */
yplot[1]=vmax;
strcpy (ylabel,"velocity");
xaxis (2.0, 7.0, 16.0, xplot, -7, xtrans, NULL); /* line ?? */
yaxis (2.0, 7.0, 5.0, yplot, -3, ytransv1, ylabel);
yaxis (18.0,7.0, 5.0, yplot, -3, ytransv1, NULL);
for (islit=0; islit<nslit; islit++) {
xplt = xtrans (-0.5*slit_length + (islit+0.5)*slit_cell);
yplt = ytransv1 (v1star[islit]);
plcross (xplt, yplt, SYMBOLSIZE);
}
if (vmin<0.0 || vmax>0.0) {
plltype (1,2); /* dashed line at v=0 */
plmove (xtrans(xplot[0]), ytransv1(0.0));
plline (xtrans(xplot[1]), ytransv1(0.0));
plltype (1,1);
}
yplot[0]=0.0; /* PLOT3: bottom panel */
yplot[1]=smax;
strcpy (ylabel,"velocity dispersion");
xaxis (2.0, 2.0, 16.0, xplot, -7, xtrans, xlabel);
yaxis (2.0, 2.0, 5.0, yplot, -3, ytransv2, ylabel);
yaxis (18.0,2.0, 5.0, yplot, -3, ytransv2, NULL);
for (islit=0; islit<nslit; islit++) {
xplt = xtrans (-0.5*slit_length + (islit+0.5)*slit_cell);
yplt = ytransv2 (v2star[islit]);
plcross (xplt, yplt, -SYMBOLSIZE);
}
plstop();
}
nemo_main()
{
int i, j, np;
string name, dumpfile;
name = getparam("name");
dumpfile = getparam("screendump");
np = getiparam("pages");
printf("Testing wth pages=%d\n",np);
plinit(name, 0.0, 20.0, 0.0, 20.0); /* open device */
x_init_plobj();
plmove(0.0, 0.0);
plcolor(0);
plline(20.0, 0.0);
plcolor(1);
plline(20.0, 20.0);
plcolor(2);
plline(0.0, 20.0);
plcolor(3);
plline(0.0, 0.0);
plcolor(4);
plline(20.0, 20.0);
plmove(20.0, 0.0);
plcolor(5);
plline(0.0, 20.0);
plltype(12, 0);
plmove(4.0, 18.0);
plcolor(6);
plline(16.0, 18.0);
plltype(-6, 0);
plmove(6.0, 18.0);
plcolor(7);
plline(14.0, 18.0);
for (i = 1; i <= 4; i++) {
plcolor(8+i);
plltype(i, 1);
plmove(1.0, 13.0 - i);
plline(3.0, 13.0 - i);
plpoint(3.5, 13.0 - i);
plltype(1, i);
for (j = 1; j <= 4; j++) {
plmove(1.5, 13.0 - i - 0.2*j);
plline(1.5 + j, 13.0 - i - 0.2*j);
}
}
plcolor(12);
plltype(1, 1);
plcircle(15.0, 9.0, -0.5);
plcolor(13);
plcircle(16.0, 9.0, 0.25);
plcolor(14);
plcircle(17.0, 9.0, 0.125);
plcolor(15);
plcircle(18.0, 9.0, 0.0625);
plbox(16.0, 8.0, 0.4);
plbox(17.0, 8.0, 0.2);
plbox(18.0, 8.0, -0.2);
plcross(16.0, 7.0, 0.4);
plcross(17.0, 7.0, 0.2);
plcross(18.0, 7.0, -0.2);
plcolor(4);
pltext("Foo Bar!", 8.0, 5.0, 0.5, 0.0);
plcolor(5);
pltext("Fum Bar!", 8.0, 3.0, 0.25, 0.0);
plcolor(6);
for (i = 0; i <= 4; i++)
pltext(" testing angles", 16.0, 10.0, 0.2, 45.0*i);
plmove(10.0, 8.5);
plline(10.0, 11.5);
pljust(-1);
plcolor(3);
pltext("left justified", 10.0, 9.0, 0.25, 0.0);
plcolor(2);
pljust(0);
pltext("centered", 10.0, 10.0, 0.25, 0.0);
plcolor(1);
pljust(1);
pltext("right justified", 10.0, 11.0, 0.25, 0.0);
pljust(0);
plcolor(7);
pltext(getparam("headline"),10.0, 19.0, 0.5, 0.0);
plcolor(1);
plflush();
if (*dumpfile)
pl_screendump(dumpfile);
if (np>1) {
plflush();
plframe();
plmove(0.0, 0.0);
plline(20.0, 0.0);
plline(20.0, 20.0);
plline(0.0, 20.0);
plline(0.0, 0.0);
pljust(0);
pltext("This is page 2", 10.0,10.0,0.25,0.0);
#define IMAX 100
#define ISTEP 20.0/IMAX
plmove (0.0,0.0);
for (i=0; i<IMAX; i++)
plline(i*ISTEP, i*ISTEP);
}
plstop();
}
local void histogram(void)
{
int i,j,k, l, kmin, kmax, lcount = 0;
real count[MAXHIST], under, over;
real xdat,ydat,xplt,yplt,dx,r,sum,sigma2, q, qmax;
real mean, sigma, mad, skew, kurt, h3, h4, lmin, lmax, median;
real rmean, rsigma, rrange[2];
Moment m;
dprintf (0,"read %d values\n",npt);
dprintf (0,"min and max value in column(s) %s: %g %g\n",getparam("xcol"),xmin,xmax);
if (!Qauto) {
xmin = xrange[0];
xmax = xrange[1];
dprintf (0,"min and max value reset to : %g %g\n",xmin,xmax);
lmin = xmax;
lmax = xmin;
for (i=0; i<npt; i++) {
if (x[i]>xmin && x[i]<=xmax) {
lmin = MIN(lmin, x[i]);
lmax = MAX(lmax, x[i]);
}
}
dprintf (0,"min and max value in range : %g %g\n",lmin,lmax);
}
for (k=0; k<nsteps; k++)
count[k] = 0; /* init histogram */
under = over = 0;
ini_moment(&m, 4, Qrobust||Qmad ? npt : 0);
for (i=0; i<npt; i++) {
if (Qbin) {
k=ring_index(nsteps,bins,x[i]);
} else {
if (xmax != xmin)
k = (int) floor((x[i]-xmin)/(xmax-xmin)*nsteps);
else
k = 0;
dprintf(2,"%d k=%d %g\n",i,k,x[i]);
}
if (k==nsteps && x[i]==xmax) k--; /* include upper edge */
if (k<0) { under++; continue; }
if (k>=nsteps) { over++; continue; }
count[k] = count[k] + 1;
dprintf (4,"%d : %f %d\n",i,x[i],k);
accum_moment(&m,x[i],1.0);
}
if (under > 0) error("bug: under = %d",under);
if (over > 0) error("bug: over = %d",over);
under = Nunder;
over = Nover;
mean = mean_moment(&m);
sigma = sigma_moment(&m);
skew = skewness_moment(&m);
kurt = kurtosis_moment(&m);
h3 = h3_moment(&m);
h4 = h4_moment(&m);
if (Qmad) mad = mad_moment(&m);
if (nsigma > 0) { /* remove outliers iteratively, starting from the largest */
iq = (int *) allocate(npt*sizeof(int));
for (i=0; i<npt; i++) {
#if 1
iq[i] = x[i] < xmin || x[i] > xmax;
#else
iq[i] = 0;
#endif
}
lcount = 0;
do { /* loop to remove outliers one by one */
qmax = -1.0;
for (i=0, l=-1; i<npt; i++) { /* find largest deviation from current mean */
if (iq[i]) continue; /* but skip previously flagged points */
q = (x[i]-mean)/sigma;
q = ABS(q);
if (q > qmax) {
qmax = q;
l = i;
}
}
if (qmax > nsigma) {
lcount++;
iq[l] = 1;
decr_moment(&m,x[l],1.0);
mean = mean_moment(&m);
sigma = sigma_moment(&m);
skew = skewness_moment(&m);
kurt = kurtosis_moment(&m);
h3 = h3_moment(&m);
h4 = h4_moment(&m);
if (Qmad) mad = mad_moment(&m);
dprintf(1,"%d/%d: removing point %d, m/s=%g %g qmax=%g\n",
lcount,npt,l,mean,sigma,qmax);
if (sigma <= 0) {
/* RELATED TO presetting MINMAX */
warning("BUG");
accum_moment(&m,x[l],1.0);
mean = mean_moment(&m);
sigma = sigma_moment(&m);
skew = skewness_moment(&m);
kurt = kurtosis_moment(&m);
h3 = h3_moment(&m);
h4 = h4_moment(&m);
dprintf(1,"%d/%d: LAST removing point %d, m/s=%g %g qmax=%g\n",
lcount,npt,l,mean,sigma,qmax);
break;
}
} else
dprintf(1,"%d/%d: keeping point %d, m/s=%g %g qmax=%g\n",
lcount,npt,l,mean,sigma,qmax);
/* if (lcount > npt/2) break; */
} while (qmax > nsigma);
dprintf(0,"Removed %d/%d points for nsigma=%g\n",lcount,npt,nsigma);
/* @algorithm left shift array values from mask array */
/* now shift all points into the array, decreasing npt */
/* otherwise the median is not correctly computed */
for (i=0, k=0; i<npt; i++) {
dprintf(1,"iq->%d\n",iq[i]);
if (iq[i]) k++;
if (k==0) continue; /* ?? */
if (i-k < 0) continue;
dprintf(1,"SHIFT: %d <= %d\n",i-k,i);
x[i-k] = x[i];
}
npt -= lcount; /* correct for outliers */
free(iq);
} /* nsigma > 0 */
if (npt != n_moment(&m))
error("Counting error, probably in removing outliers...");
dprintf (0,"Number of points : %d\n",npt);
if (npt>1)
dprintf (0,"Mean and dispersion : %g %g %g\n",mean,sigma,sigma/sqrt(npt-1.0));
else
dprintf (0,"Mean and dispersion : %g %g 0.0\n",mean,sigma);
if (Qmad) dprintf (0,"MAD : %g\n",mad);
dprintf (0,"Skewness and kurtosis: %g %g\n",skew,kurt);
dprintf (0,"h3 and h4 : %g %g\n", h3, h4);
if (Qmedian) {
if (npt % 2)
median = x[(npt-1)/2];
else
median = 0.5 * (x[npt/2] + x[npt/2-1]);
dprintf (0,"Median : %g\n",median);
} else if (Qtorben) {
median = median_torben(npt,x,xmin,xmax);
dprintf (0,"Median_torben : %g\n",median);
}
dprintf (0,"Sum : %g\n",show_moment(&m,1));
if (Qrobust) {
compute_robust_moment(&m);
rmean = mean_robust_moment(&m);
rsigma = sigma_robust_moment(&m);
robust_range(&m, rrange);
dprintf (0,"Robust N : %d\n",n_robust_moment(&m));
dprintf (0,"Robust Mean Disp : %g %g\n",rmean,rsigma);
dprintf (0,"Robust Range : %g %g\n",rrange[0],rrange[1]);
if (outstr) {
for (i=0; i<npt; i++) {
if (x[i]<rrange[0] || x[i]>rrange[1]) continue;
fprintf(outstr,"%g %d\n",x[i],i+1);
}
}
}
if (lcount > 0) {
warning("Recompute histogram because of outlier removals");
/* recompute histogram if we've lost some outliers */
for (k=0; k<nsteps; k++)
count[k] = 0; /* init histogram */
under = over = 0;
for (i=0; i<npt; i++) {
if (xmax != xmin)
k = (int) floor((x[i]-xmin)/(xmax-xmin)*nsteps);
else
k = 0;
if (k==nsteps && x[i]==xmax) k--; /* include upper edge */
if (k<0) { under++; continue; }
if (k>=nsteps) { over++; continue; }
count[k] = count[k] + 1;
dprintf (4,"%d : %f %d\n",i,x[i],k);
}
if (under > 0 || over > 0) error("under=%d over=%d in recomputed histo",under,over);
}
dprintf (3,"Histogram values : \n");
dx=(xmax-xmin)/nsteps;
kmax=0;
sum=0.0;
for (k=0; k<nsteps; k++) {
sum = sum + dx*count[k];
if (ylog) {
if (count[k]>0.0)
count[k] = log10(count[k]);
else
count[k] = -1.0;
}
if (count[k]>kmax)
kmax=count[k];
dprintf (3,"%f ",count[k]);
if (Qcumul) {
if (k==0)
count[k] += under;
else
count[k] += count[k-1];
}
}
dprintf (3,"\n");
sigma2 = 2.0 * sigma * sigma; /* gaussian */
sum /= sigma * sqrt(2*PI); /* scaling factor for equal area gauss */
if (ylog && over>0) over = log10(over);
if (ylog && under>0) under = log10(under);
kmax *= 1.1; /* add 10% */
if (Qcumul) kmax = npt;
if (maxcount>0) /* force scaling by user ? */
kmax=maxcount;
if (Qtab) {
maxcount = 0;
for (k=0; k<nsteps; k++)
maxcount = MAX(maxcount,count[k]);
if (maxcount>0)
r = 29.0/maxcount;
else
r = 1.0;
printf(" Bin Value Number\n");
printf(" Underflow %d\n",Nunder);
for (k=0; k<nsteps; k++) {
j = (int) (r*count[k]) + 1;
if (ylog) printf("%3d %13.6g %13.6g ",
k+1, xmin+(k+0.5)*dx, count[k]);
else printf("%3d %13.6g %8d ",
k+1, xmin+(k+0.5)*dx, (int)count[k]);
while (j-- > 0) printf("*");
printf("\n");
}
printf(" Overflow %d\n",Nover);
stop(0);
}
#ifdef YAPP
/* PLOTTING */
plinit("***",0.0,20.0,0.0,20.0);
xplot[0] = xmin;
xplot[1] = xmax;
yplot[0] = 0.0;
yplot[1] = (real) kmax;
xaxis (2.0, 2.0, 16.0, xplot, -7, xtrans, xlab);
xaxis (2.0, 18.0,16.0, xplot, -7, xtrans, NULL);
yaxis (2.0, 2.0, 16.0, yplot, -7, ytrans, ylab);
yaxis (18.0, 2.0, 16.0, yplot, -7, ytrans, NULL);
pljust(-1); /* set to left just */
pltext(input,2.0,18.2,0.32,0.0); /* filename */
pljust(1);
pltext(headline,18.0,18.2,0.24,0.0); /* headline */
pljust(-1); /* return to left just */
xdat=xmin;
dx=(xmax-xmin)/nsteps;
plmove(xtrans(xmin),ytrans(0.0));
for (k=0; k<nsteps; k++) { /* nsteps= */
xplt = xtrans(xdat);
yplt = ytrans((real)count[k]);
plline (xplt,yplt);
xdat += dx;
xplt = xtrans(xdat);
plline (xplt,yplt);
}
plline(xplt,ytrans(0.0));
for (i=0; i<nxcoord; i++) {
plmove(xtrans(xcoord[i]),ytrans(yplot[0]));
plline(xtrans(xcoord[i]),ytrans(yplot[1]));
}
if (Qgauss) { /* plot model and residuals */
if (ylog)
plmove(xtrans(xmin),ytrans(-1.0));
else
plmove(xtrans(xmin),ytrans(0.0));
for (k=0; k<100; k++) {
xdat = xmin + (k+0.5)*(xmax-xmin)/100.0;
ydat = sum * exp( -sqr(xdat-mean)/sigma2);
if (ylog) ydat = log10(ydat);
plline(xtrans(xdat), ytrans(ydat));
}
}
if (Qresid) {
plltype(0,2); /* dotted residuals */
xdat = xmin+0.5*dx;
dprintf(1,"# residuals from gauss\n");
for (k=0; k<nsteps; k++, xdat +=dx) {
ydat = sum * exp( -sqr(xdat-mean)/sigma2);
dprintf(1,"%g %g %g\n",xdat,count[k],ydat);
if (ylog) ydat = log10(ydat);
ydat = count[k] - ydat;
if (k==0)
plmove(xtrans(xdat),ytrans(ydat));
else
plline(xtrans(xdat),ytrans(ydat));
}
plltype(0,1); /* back to normal line type */
}
plstop();
#endif
}
void nemo_main()
{
int i, dir, nrad, npots=0, ltype, ndim = NDIM, nx, ny, ns, ndat, nret;
int cols[4], n, idx, idx_max;
real pmax, symsize, rr, omk_max = 0.0, omk_rmax;
real rad[MAXPT], *vel, *vel1, *vel2, *vel3, *vel4, *curve;
real *ome, *kap, *opk, *omk, r0l[MAXPT+2], omega, *f;
real inrad[MAXPT], invel[MAXPT], inrade[MAXPT], invele[MAXPT];
double pos[3], acc[3], pot, time = 0.0;
/* char *fmt, s[20], pfmt[256]; */
char headline[256], fmt1[80];
string axis, mode, infile, plotlabel;
stream instr;
bool Qtab, Qplot, Qome, Qvel, Qlv, Qin, QoILR;
mode = getparam("mode");
n = getiparam("n");
plotlabel = getparam("headline");
sprintf(fmt1,"%s ",getparam("format"));
Qome = (*mode == 'o'); /* options are: velocity|omega|lv */
Qlv = (*mode == 'l');
Qvel = (*mode == 'v');
Qtab = getbparam("tab");
Qplot = getbparam("plot");
infile = getparam("in");
Qin = (*infile != 0);
if (Qin) {
nret = nemoinpi(getparam("cols"),cols,4);
if (nret<0 || nret > 4) error("cols= requires 4 numbers");
for (i=nret; i<4; i++)
cols[i] = 0;
instr = stropen(infile,"r");
ndat = read_table(instr,MAXPT,inrad,invel,inrade,invele,cols);
strclose(instr);
}
mypot1 = get_potential(getparam("name1"),getparam("pars1"),getparam("file1"));
omega = get_pattern();
dprintf(0,"Pattern speed: %f\n",omega);
mypot2 = get_potential(getparam("name2"),getparam("pars2"),getparam("file2"));
mypot3 = get_potential(getparam("name3"),getparam("pars3"),getparam("file3"));
mypot4 = get_potential(getparam("name4"),getparam("pars4"),getparam("file4"));
headline[0] = '\0'; /* accumulate headline */
if (mypot1) {
strcat(headline,getparam("name1"));
strcat(headline,"(");
strcat(headline,getparam("pars1"));
strcat(headline,")");
npots++;
}
if (mypot2) {
strcat(headline,getparam("name2"));
strcat(headline,"(");
strcat(headline,getparam("pars2"));
strcat(headline,") ");
npots++;
}
if (mypot3) {
strcat(headline,getparam("name3"));
strcat(headline,"(");
strcat(headline,getparam("pars3"));
strcat(headline,") ");
npots++;
}
if (mypot4) {
strcat(headline,getparam("name4"));
strcat(headline,"(");
strcat(headline,getparam("pars4"));
strcat(headline,")");
npots++;
}
nrad = nemoinpr(getparam("radii"),rad,MAXPT); /* get radii */
if (nrad <= 0)
warning("Using %d radii is not very productive",nrad);
vel = (real *) allocate(sizeof(real) * nrad); /* allocate stuff */
vel1 = (real *) allocate(sizeof(real) * nrad);
vel2 = (real *) allocate(sizeof(real) * nrad);
vel3 = (real *) allocate(sizeof(real) * nrad);
vel4 = (real *) allocate(sizeof(real) * nrad);
if (Qome) {
ome = (real *) allocate(4 * sizeof(real) * nrad); /* plus spline */
kap = (real *) allocate(sizeof(real) * nrad);
opk = (real *) allocate(sizeof(real) * nrad);
omk = (real *) allocate(sizeof(real) * nrad);
}
axis = getparam("axis");
dir = 0;
if (*axis == 'x') dir=0;
if (*axis == 'y') dir=1;
if (*axis == 'z') dir=2;
if (dir>NDIM) error("Axis %s not supported in NDIM=%d",axis,NDIM);
pmax = 0.0;
for (i=0; i<nrad; i++) { /* loop to compute */
CLRV(pos); /* clear positions */
pos[dir] = rad[i]; /* set the right axis */
vel[i] = 0.0;
if (mypot1) {
CLRV(acc);
(*mypot1) (&ndim,pos,acc,&pot,&time);
vel1[i] = -rad[i] * acc[dir];
vel[i] += vel1[i];
vel1[i] = sqrt(vel1[i]);
}
if (mypot2) {
CLRV(acc);
(*mypot2) (&ndim,pos,acc,&pot,&time);
vel2[i] = -rad[i] * acc[dir];
vel[i] += vel2[i];
vel2[i] = sqrt(vel2[i]);
}
if (mypot3) {
CLRV(acc);
(*mypot3) (&ndim,pos,acc,&pot,&time);
vel3[i] = -rad[i] * acc[dir];
vel[i] += vel3[i];
vel3[i] = sqrt(vel3[i]);
}
if (mypot4) {
CLRV(acc);
(*mypot4) (&ndim,pos,acc,&pot,&time);
vel4[i] = -rad[i] * acc[dir];
vel[i] += vel4[i];
vel4[i] = sqrt(vel4[i]);
}
vel[i] = sqrt(vel[i]);
}
if (Qome) {
lindblad(nrad,rad,vel,ome,kap,opk,omk,n);
if (omega> 0.0) { /* compute resonances */
f = opk;
idx = nrad-1;
if (omega < f[idx]) {
warning("Radii not far enough out for OLR: %g",f[idx]);
f = ome;
if (omega < f[idx]) {
warning("Radii not far enough out for CR: %g",f[idx]);
f = omk;
}
}
QoILR = FALSE;
for(; idx>0; idx--) {
if (omk[idx] > omk_max) {
idx_max = idx;
omk_max = omk[idx];
}
if (f==omk) {
if (QoILR) {
if (omega < f[idx]) continue;
} else {
if (omega > f[idx]) continue;
}
} else {
if (omega > f[idx]) continue;
}
/* found a resonance: */
rr = rad[idx] + (rad[idx+1]-rad[idx])*
(omega-f[idx])/(f[idx+1]-f[idx]);
if (f == omk) {
#if 0
if (QoILR) {
dprintf(0,"iILR: %g\n",rr);
break;
} else {
dprintf(0,"oILR: %g\n",rr);
QoILR = TRUE;
}
#endif
} else if (f == ome) {
dprintf(0,"CR: %g\n",rr);
f = omk;
} else if (f == opk) {
dprintf(0,"OLR: %g\n",rr);
f = ome;
} else
error("impossble resonance");
}
peak(nrad,rad,omk,idx_max,1, &omk_rmax, &omk_max);
dprintf(0,"OMK_max: %g\n",omk_max);
dprintf(0,"OMK_rmax: %g\n",omk_rmax);
if (omega < omk_max) { /* search for ILR */
for (idx=idx_max; idx<nrad; idx++) {
if (omega > omk[idx]) {
rr = rad[idx-1] + (rad[idx]-rad[idx-1])*
(omega-f[idx-1])/(f[idx]-f[idx-1]);
dprintf(0,"oILR: %g\n",rr);
break;
}
}
for (idx=idx_max; idx>0; idx--) {
if (omega > omk[idx]) {
rr = rad[idx] + (rad[idx+1]-rad[idx])*
(omega-f[idx])/(f[idx+1]-f[idx]);
dprintf(0,"iILR: %g\n",rr);
break;
}
}
}
}
}
for (i=0; i<nrad; i++) { /* loop to print */
if (Qtab) {
printf(fmt1,rad[i]);
printf(fmt1,vel[i]);
}
if (Qtab && npots>1 && !Qome) {
if (mypot1) printf(fmt1,vel1[i]);
if (mypot2) printf(fmt1,vel2[i]);
if (mypot3) printf(fmt1,vel3[i]);
if (mypot4) printf(fmt1,vel4[i]);
}
if (Qtab && Qome) {
printf(fmt1,ome[i]);
printf(fmt1,kap[i]);
printf(fmt1,opk[i]);
printf(fmt1,omk[i]);
}
if (Qtab) printf("\n");
if (Qome)
pmax = MAX(pmax,opk[i]);
else
pmax = MAX(pmax,vel[i]);
}
if (Qin && Qvel)
goodness(nrad,rad,vel,ndat,inrad,invel,(cols[3]>0?invele:NULL));
if (Qplot) {
plinit("***",0.0,20.0,0.0,20.0); /* open device */
nx = nemoinpr(getparam("xrange"),xplot,2); /* get xrange in plot */
switch(nx) {
case 0:
xplot[0] = rad[0];
case 1:
xplot[1] = rad[nrad-1];
break;
case 2:
break;
default:
warning("xrange= only accepts two values");
break;
}
ny = nemoinpr(getparam("yrange"),yplot,2); /* get yrange in plot */
switch(ny) {
case 0:
yplot[0] = 0.0;
yplot[1] = 1.1 * pmax; /* extra 10% for egde */
break;
case 1:
yplot[1] = 1.1 * pmax; /* extra 10% for egde */
break;
case 2:
break;
default:
warning("yrange= only accepts two values");
break;
}
xaxis ( 2.0, 2.0, 16.0, xplot, -7, xtrans, "R"); /* plot axes */
xaxis ( 2.0,18.0, 16.0, xplot, -7, xtrans, NULL);
if (Qome)
yaxis ( 2.0, 2.0, 16.0, yplot, -7, ytrans, "[V/R]");
else
yaxis ( 2.0, 2.0, 16.0, yplot, -7, ytrans, "V");
yaxis (18.0, 2.0, 16.0, yplot, -7, ytrans, NULL);
if (*plotlabel)
pltext(plotlabel,2.0,18.5,0.5,0.0);
else
pltext(headline,2.0,18.5,0.35,0.0);
if (*plotmsg)
pltext(plotmsg,8.0,2.5,0.25,0.0);
curve = (Qome ? ome : vel); /* assign first curve */
plltype(3,1); /* thick solid line */
plmove(xtrans(rad[0]),ytrans(curve[0]));
for (i=1; i<nrad; i++)
plline(xtrans(rad[i]),ytrans(curve[i]));
if (Qome) { /* if Lindblad - plot omk, opk */
plltype(1,1); /* all regular solid lines */
plmove(xtrans(rad[0]), ytrans(omk[0]));
for (i=1; i<nrad; i++)
plline(xtrans(rad[i]),ytrans(omk[i]));
plmove(xtrans(rad[0]), ytrans(opk[0]));
for (i=1; i<nrad; i++)
plline(xtrans(rad[i]),ytrans(opk[i]));
} else if (npots>1) { /* if velocity and > 1 component */
ltype = 1;
if (mypot1) {
plltype(1,++ltype);
plmove(xtrans(rad[0]),ytrans(vel1[0]));
for (i=1; i<nrad; i++)
plline(xtrans(rad[i]),ytrans(vel1[i]));
}
if (mypot2) {
plltype(1,++ltype);
plmove(xtrans(rad[0]),ytrans(vel2[0]));
for (i=1; i<nrad; i++)
plline(xtrans(rad[i]),ytrans(vel2[i]));
}
if (mypot3) {
plltype(1,++ltype);
plmove(xtrans(rad[0]),ytrans(vel2[0]));
for (i=1; i<nrad; i++)
plline(xtrans(rad[i]),ytrans(vel3[i]));
}
if (mypot4) {
plltype(1,++ltype);
plmove(xtrans(rad[0]),ytrans(vel2[0]));
for (i=1; i<nrad; i++)
plline(xtrans(rad[i]),ytrans(vel4[i]));
}
}
plltype(1,1);
symsize = 0.1;
if (Qin && Qvel) { /* if input file with velocities */
for (i=0; i<ndat; i++)
plbox(xtrans(inrad[i]),ytrans(invel[i]),symsize);
if (cols[3]>0) { /* if error bars in radius */
for (i=0; i<ndat; i++) {
plmove(xtrans(inrad[i]-inrade[i]),ytrans(invel[i]));
plline(xtrans(inrad[i]+inrade[i]),ytrans(invel[i]));
}
}
if (cols[4]>0) { /* if error bars in velocity */
for (i=0; i<ndat; i++) {
plmove(xtrans(inrad[i]),ytrans(invel[i]-invele[i]));
plline(xtrans(inrad[i]),ytrans(invel[i]+invele[i]));
}
}
} else if (Qin && Qome) { /* if input file with omega */
for (i=0; i<ndat; i++)
plbox(xtrans(inrad[i]),ytrans(invel[i]/inrad[i]),symsize);
}
plstop();
} /* if plot vel/ome */
if (Qlv) {
ns = nemoinpr(getparam("r0l"),r0l,MAXPT+2) - 2;
if (ns < 0)
error("r0l= needs at least two values: r0 and l");
else if (ns==0)
warning("r0l= no lv-radii array supplied");
lv(nrad,rad,vel,r0l[0],r0l[1],ns,&r0l[2]);
}
}
