bool initauxtbl(char *aux) {
char *filename;
FILE * fp ;
int pflen, nread, pfargc, targc, i, j ;
char *argstr, **targv ;
if( !getnpar(0,aux,"s",&filename) ) {
fprintf(stderr,"Can't get auxiliary input %s definition\n",aux) ;
return 0 ;
}
fp = fopen(filename, "r") ;
if( fp == NULL ) {
fprintf(stderr,"Can't open auxiliary input file %s=%s\n",aux,filename);
return 0 ;
}
/* Get the length */
efseek(fp, 0, SEEK_END);
pflen = eftell(fp);
rewind(fp);
argstr = (char *) ealloc1(1+pflen+1, 1);
/* Read the parfile */
nread = efread(argstr+1, 1, pflen, fp);
if (nread != pflen) {
err("%s: fread only %d bytes out of %d from %s",
__FILE__,nread,pflen,filename);
}
efclose(fp);
/* Zap whites in parfile to help in parsing */
argstr[0] = '\0' ;
pfargc = white2null(argstr, pflen);
targc = pfargc ;
/* Allocate space for total arg pointers */
targv = (char **) ealloc1(targc, sizeof(char*));
/* Parse the parfile. Skip over multiple NULLs */
for( j=1, i=0; j < pflen; j++) {
if( argstr[j] && !argstr[j-1] ) {
targv[i++] = argstr + j;
}
}
/* Allocate space for the pointer table */
argtbl = (ArgStruct*) ealloc1(targc, sizeof(ArgStruct));
/* Tabulate targv */
tabulate(targc, targv);
return 1 ;
}
/* Break up reflectors by duplicating interior (x,z) points */
void breakReflectors (int *nr, float **ar,
int **nu, float ***xu, float ***zu)
{
int nri,nro,*nui,*nuo,ir,jr,iu;
float *ari,*aro,**xui,**zui,**xuo,**zuo;
/* input reflectors */
nri = *nr;
ari = *ar;
nui = *nu;
xui = *xu;
zui = *zu;
/* number of output reflectors */
for (ir=0,nro=0; ir<nri; ++ir)
nro += nui[ir]-1;
/* make output reflectors and free space for input reflectors */
aro = ealloc1float(nro);
nuo = ealloc1int(nro);
xuo = ealloc1(nro,sizeof(float*));
zuo = ealloc1(nro,sizeof(float*));
for (ir=0,jr=0; ir<nri; ++ir) {
for (iu=0; iu<nui[ir]-1; ++iu,++jr) {
aro[jr] = ari[ir];
nuo[jr] = 2;
xuo[jr] = ealloc1float(2);
zuo[jr] = ealloc1float(2);
xuo[jr][0] = xui[ir][iu];
zuo[jr][0] = zui[ir][iu];
xuo[jr][1] = xui[ir][iu+1];
zuo[jr][1] = zui[ir][iu+1];
}
free1float(xui[ir]);
free1float(zui[ir]);
}
free1float(ari);
free1int(nui);
free1(xui);
free1(zui);
/* output reflectors */
*nr = nro;
*ar = aro;
*nu = nuo;
*xu = xuo;
*zu = zuo;
}
void makericker (float fpeak, float dt, Wavelet **w)
/*****************************************************************************
Make Ricker wavelet
******************************************************************************
Input:
fpeak peak frequency of wavelet
dt time sampling interval
Output:
w Ricker wavelet
*****************************************************************************/
{
int iw,lw,it,jt;
float t,x,*wv;
iw = -(1+1.0/(fpeak*dt));
lw = 1-2*iw;
wv = ealloc1float(lw);
for (it=iw,jt=0,t=it*dt; jt<lw; ++it,++jt,t+=dt) {
x = PI*fpeak*t;
x = x*x;
wv[jt] = exp(-x)*(1.0-2.0*x);
}
*w = ealloc1(1,sizeof(Wavelet));
(*w)->lw = lw;
(*w)->iw = iw;
(*w)->wv = wv;
}
float alpha_trim_w(float *a,float *w,int n,float p)
/******************************************************************
alpha_trim_w - Weighted Alpha trim of array
******************************************************************
Notes:
the value a[irp] is replaced by the mean of the
a values, but the largers and smallest p% of a
values are left out fo mean computation,array a is destroyed
******************************************************************
Author: Balasz Nemeth, given to CWP in 2008 by Potash Corporation
******************************************************************/
{
float mean=0;
unsigned int ist=NINT(n*(p/100.0));
unsigned int i,ic;
int *ind=NULL;
ind = ealloc1(n,sizeof(unsigned int));
for(i=0;i<n;++i) ind[i]=i;
qkisort(n,a,ind);
for(i=ist,ic=0;i<n-ist;i++,ic++)
mean += a[ind[i]]*w[ind[i]];
free1(ind);
return(mean/(float)ic);
}
/* Value getpar -- omitted string type for now */
void getparval(String name, String type, int n, Value *valp)
{
register int k;
short *h;
unsigned short *u;
long *l;
unsigned long *v;
int *i;
unsigned int *p;
float *f;
double *d;
switch(*type) {
case 'h':
h = (short*) ealloc1(n, sizeof(short));
getparshort(name, h);
for (k = 0; k < n; ++k) valp[k].h = h[k];
break;
case 'u':
u = (unsigned short*) ealloc1(n, sizeof(unsigned short));
getparushort(name, u);
for (k = 0; k < n; ++k) valp[k].u = u[k];
break;
case 'l':
l = (long*) ealloc1(n, sizeof(long));
getparlong(name, l);
for (k = 0; k < n; ++k) valp[k].l = l[k];
break;
case 'v':
v = (unsigned long*) ealloc1(n, sizeof(unsigned long));
getparulong(name, v);
for (k = 0; k < n; ++k) valp[k].v = v[k];
break;
case 'i':
i = (int*) ealloc1(n, sizeof(int));
getparint(name, i);
for (k = 0; k < n; ++k) valp[k].i = i[k];
break;
case 'p':
p = (unsigned int*) ealloc1(n, sizeof(unsigned int));
getparuint(name, p);
for (k = 0; k < n; ++k) valp[k].p = p[k];
break;
case 'f':
f = (float*) ealloc1(n, sizeof(float));
getparfloat(name, f);
for (k = 0; k < n; ++k) valp[k].f = f[k];
break;
case 'd':
d = (double*) ealloc1(n, sizeof(double));
getpardouble(name, d);
for (k = 0; k < n; ++k) valp[k].d = d[k];
break;
default:
err("getparval: %d: mysterious type %s", __LINE__, type);
}
}
void decodeReflectors (int *nrPtr,
float **aPtr, int **nxzPtr, float ***xPtr, float ***zPtr)
/*************************************************************************
decodeReflectors - parse reflectors parameter string
**************************************************************************
Output:
nrPtr pointer to nr an int specifying number of reflectors
aPtr pointer to a specifying reflector amplitudes
nxzPtr pointer to nxz specifying number of (x,z) pairs defining the
reflectors
xPtr pointer to array[x][nr] of x values for the entire model
zPtr array[z][nr] of z values for the entire model
***************************************************************************
Author: Dave Hale, Colorado School of Mines, 09/17/91
**************************************************************************/
{
int nr,*nxz,ir;
float *a,**x,**z;
char t[1024],*s;
/* count reflectors */
nr = countparname("ref");
if (nr==0) nr = 1;
/* allocate space */
a = ealloc1(nr,sizeof(float));
nxz = ealloc1(nr,sizeof(int));
x = ealloc1(nr,sizeof(float*));
z = ealloc1(nr,sizeof(float*));
/* get reflectors */
for (ir=0; ir<nr; ++ir) {
if (!getnparstring(ir+1,"ref",&s)) s = "1:1,2;4,2";
strcpy(t,s);
if (!decodeReflector(t,&a[ir],&nxz[ir],&x[ir],&z[ir]))
err("Reflector number %d specified "
"incorrectly!\n",ir+1);
}
/* set output parameters before returning */
*nrPtr = nr;
*aPtr = a;
*nxzPtr = nxz;
*xPtr = x;
*zPtr = z;
}
static unsigned char *makeBBytes (int nxa, int nya, unsigned char *abytes,
int nxb, int nyb, int ixb, int iyb)
/*****************************************************************************
Makes bytes (unsigned char) in zoom box, with bytes ordered
as nxb samples left-to-right, nyb scanlines top-to-bottom.
*****************************************************************************/
{
int ix,iy;
unsigned char *b,*bp,*ap;
bp = b = ealloc1(nxb*nyb+0*nya,sizeof(unsigned char));
for (iy=0; iy<nyb; ++iy)
for (ix=0,ap=abytes+(iyb+iy)*nxa+ixb; ix<nxb; ++ix)
*bp++ = *ap++;
return b;
}
XImage *xNewImage (Display *dpy, unsigned long pmin, unsigned long pmax,
int width, int height, unsigned char *bytes)
/*****************************************************************************
make a new image of pixels from bytes
******************************************************************************
Input:
dpy display pointer
pmin minimum pixel value (corresponding to byte=0)
pmax maximum pixel value (corresponding to byte=255)
width number of bytes in x dimension
height number of bytes in y dimension
bytes unsigned bytes to be mapped to an image
******************************************************************************
Author: Dave Hale, Colorado School of Mines, 06/08/90
*****************************************************************************/
{
int scr=DefaultScreen(dpy);
int i,j,k,line,iline,jline,widthpad;
float base,scale;
unsigned char map[256];
unsigned char *data;
/* build map for translating bytes to pixels */
base = pmin+0.499;
scale = (pmax-pmin)/255.0;
for (i=0; i<=255; ++i)
map[i] = base+i*scale;
/* allocate memory for image data */
widthpad = (1+(width-1)/(BitmapPad(dpy)/8))*BitmapPad(dpy)/8;
data = ealloc1(widthpad*height,sizeof(unsigned char));
/* translate bytes to pixels, padding scanlines as necessary */
for (line=0; line<height; line++) {
iline = line*width;
jline = line*widthpad;
for (i=iline,j=jline,k=0; k<width; ++i,++j,++k)
data[j] = map[bytes[i]];
for (j=jline+width,k=width; k<widthpad; ++j,++k)
data[j] = data[jline+width-1];
}
/* create and return image structure */
return XCreateImage(dpy,DefaultVisual(dpy,scr),
DefaultDepth(dpy,scr),ZPixmap,
0,data,widthpad,height,BitmapPad(dpy),widthpad);
}
static void dmooff (float offset, float fmax, float sdmo, int nx, float dx,
int nt, float dt, float ft, float *vrms, float **ptx)
/*****************************************************************************
perform dmo in f-k domain for one offset
******************************************************************************
Input:
offset source receiver offset
fmax maximum frequency
sdmo DMO stretch factor
nx number of midpoints
dx midpoint sampling interval
nt number of time samples
dt time sampling interval
ft first time
vrms array[nt] of rms velocities
ptx array[nx][nt] for p(t,x), zero-padded for fft (see notes)
Output:
ptx array[nx][nt] for dmo-corrected p(t,x)
******************************************************************************
Notes:
To avoid having to allocate a separate work space that is larger than the
array ptx[nx][nt], ptx must be zero-padded to enable Fourier transform from x
to k via prime factor FFT. nxpad (nx after zero-padding) can be estimated by
nxpad = 2+npfar(nx+(int)(0.5*ABS(offset/dx)));
where npfar() is a function that returns a valid length for real-to-complex
prime factor FFT. ptx[nx] to ptx[nxfft-1] must point to zeros.
******************************************************************************
Author: Dave Hale, Colorado School of Mines, 08/08/91
*****************************************************************************/
{
int nxfft,itmin,nu,nufft,nw,nk,ix,iu,iw,ik,it,iwn,
iwmin,iwmax,nupad,ikmax;
float dw,dk,tcon,wwscl,scale,scales,kmax,
amp,phase,fr,fi,pwr,pwi,
wmin,wmax,fftscl,du,fu,w,k,osdmo,*uoft,*tofu;
complex czero=cmplx(0.0,0.0),**ptk,*pu,*pw;
/* number of cdps after padding for fft */
nxfft = npfar(nx+(int)(0.5*ABS(offset/dx)));
/* get minimum time of first non-zero sample */
for (ix=0,itmin=nt; ix<nx; ++ix) {
for (it=0; it<itmin && ptx[ix][it]==0.0; ++it);
itmin = it;
}
/* if all zeros, simply return */
if (itmin>=nt) return;
/* make stretch and compress functions t(u) and u(t) */
maketu(offset,itmin,fmax,nt,dt,ft,vrms,&uoft,&nu,&du,&fu,&tofu,&tcon);
/* adjust DMO stretch factor for nominal error in log stretch; */
/* solve sdmo*(sqrt(1-a/sdmo)-1) = 0.5*log(1-a), where a=0.5 */
sdmo *= .62;
/* inverse of dmo stretch factor */
osdmo = 1.0/sdmo;
/* maximum DMO shift (in samples) for any wavenumber k */
nupad = 1.5*sdmo*tcon/du;
/* frequency sampling */
nufft = npfa(nu+nupad);
nw = nufft;
dw = 2.0*PI/(nufft*du);
/* allocate workspace */
pu = pw = ealloc1complex(nufft);
/* wavenumber sampling and maximum wavenumber to apply dmo */
nk = nxfft/2+1;
dk = 2.0*PI/ABS(nxfft*dx);
kmax = PI/ABS(dx);
ikmax = NINT(kmax/dk);
/* pointers to complex p(t,k) */
ptk = (complex**)ealloc1(nk,sizeof(complex*));
for (ik=0; ik<nk; ++ik)
ptk[ik] = (complex*)ptx[0]+ik*nt;
/* fft scale factor */
fftscl = (float)nk/(float)(ikmax+1)/(nufft*nxfft);
/* Fourier transform p(t,x) to p(t,k) */
pfa2rc(-1,2,nt,nxfft,ptx[0],ptk[0]);
/* loop over wavenumbers less than maximum */
for (ik=0,k=0.0; ik<=ikmax; ++ik,k+=dk) {
/* stretch p(t;k) to p(u) */
ints8c(nt,dt,ft,ptk[ik],czero,czero,nu,tofu,pu);
/* pad with zeros and Fourier transform p(u) to p(w) */
for (iu=nu; iu<nufft; ++iu)
pu[iu].r = pu[iu].i = 0.0;
pfacc(1,nufft,pu);
/* minimum and maximum frequencies to process */
wmin = ABS(0.5*vrms[0]*k);
wmax = ABS(PI/du);
iwmin = MAX(1,MIN((nw-1)/2,NINT(wmin/dw)));
iwmax = MAX(0,MIN((nw-1)/2,NINT(wmax/dw)));
/* constant independent of w */
wwscl = osdmo*pow(k*0.5*offset/tcon,2.0);
/* zero dc (should be zero anyway) */
pw[0].r = pw[0].i = 0.0;
/* zero frequencies below minimum */
for (iw=1,iwn=nw-iw; iw<iwmin; ++iw,--iwn)
pw[iw].r = pw[iw].i = pw[iwn].r = pw[iwn].i = 0.0;
/* do dmo between minimum and maximum frequencies */
for (iw=iwmin,iwn=nw-iwmin,w=iwmin*dw;
iw<=iwmax; ++iw,--iwn,w+=dw) {
scales = 1.0+wwscl/(w*w);
scale = sqrt(scales);
phase = sdmo*w*tcon*(scale-1.0);
amp = fftscl*(1.0-sdmo+sdmo/scale);
fr = amp*cos(phase);
fi = amp*sin(phase);
pwr = pw[iw].r;
pwi = pw[iw].i;
pw[iw].r = pwr*fr-pwi*fi;
pw[iw].i = pwr*fi+pwi*fr;
pwr = pw[iwn].r;
pwi = pw[iwn].i;
pw[iwn].r = pwr*fr+pwi*fi;
pw[iwn].i = pwi*fr-pwr*fi;
}
/* zero frequencies above maximum to Nyquist (if present) */
for (iw=iwmax+1,iwn=nw-iw; iw<=nw/2; ++iw,--iwn)
pw[iw].r = pw[iw].i = pw[iwn].r = pw[iwn].i = 0.0;
/* Fourier transform p(w) to p(u) */
pfacc(-1,nufft,pu);
/* compress p(u) to p(t;k) */
ints8c(nu,du,fu,pu,czero,czero,nt,uoft,ptk[ik]);
}
/* zero wavenumber between maximum and Nyquist */
for (; ik<nk; ++ik)
for (it=0; it<nt; ++it)
ptk[ik][it].r = ptk[ik][it].i = 0.0;
/* Fourier transform p(t,k) to p(t,x) */
pfa2cr(1,2,nt,nxfft,ptk[0],ptx[0]);
/* free workspace */
free1float(tofu);
free1float(uoft);
free1complex(pu);
free1(ptk);
}
static
int fgettr_internal(FILE *fp, segy *tp, cwp_Bool fixed_length)
{
int nread; /* bytes seen by fread calls */
off_t curoff;
if(fp != lastfp) /* search linked list for possible alternative */
searchlist(fp);
if (infoptr == ((struct insegyinfo *) NULL)) {
/* initialize new segy input stream */
unsigned int databytes; /* bytes from nsfirst */
/* allocate new segy input information table */
*oldinfoptr = (struct insegyinfo *)
malloc(sizeof(struct insegyinfo));
infoptr = *oldinfoptr;
infoptr->nextinfo = (struct insegyinfo *) NULL;
/* save FILE * ptr */
infoptr->infp = fp;
infoptr->itr = 0;
infoptr->ntr = -1;
/* allocate XDR struct and associate FILE * ptr */
infoptr->segy_xdr = (XDR *) malloc(sizeof(XDR));
switch (infoptr->ftype = filestat(fileno(fp))) {
case DIRECTORY:
err("%s: segy input can't be a directory", __FILE__);
case TTY:
err("%s: segy input can't be tty", __FILE__);
break;
default: /* the rest are ok */
break;
}
/* xdrstdio_create(infoptr->segy_xdr,fp,XDR_DECODE); */
infoptr->buf = ealloc1(sizeof(segy),sizeof(char));
xdrmem_create(infoptr->segy_xdr, infoptr->buf, sizeof(segy), XDR_DECODE);
infoptr->bufstart = xdr_getpos(infoptr->segy_xdr);
/* retrieve segy trace header */
nread = efread(infoptr->buf ,1 ,HDRBYTES ,infoptr->infp);
if(nread != HDRBYTES || FALSE == xdrhdrsub(infoptr->segy_xdr,tp))
err("%s: bad first header", __FILE__);
/* Have the header, now for the data */
infoptr->nsfirst = tp->ns;
if (infoptr->nsfirst > SU_NFLTS)
err("%s: unable to handle %d > %d samples per trace",
__FILE__, infoptr->nsfirst, SU_NFLTS);
switch(tp->trid) {
case CHARPACK:
infoptr->bytesper = sizeof(char); break;
case SHORTPACK:
infoptr->bytesper = 2*sizeof(char); break;
default:
infoptr->bytesper = BYTES_PER_XDR_UNIT; break;
}
databytes = infoptr->bytesper * tp->ns;
infoptr->nsegy = databytes + HDRBYTES;
nread = dataread(infoptr, tp, fixed_length);
switch (nread) {
case 0: err("%s: no data on first trace", __FILE__);
default: if (nread != databytes)
err("%s: first trace: tried to read %d bytes "
"read %d bytes",
__FILE__, databytes, nread);
else nread += HDRBYTES;
}
if(infoptr->ftype == DISK) { /* compute ntr */
curoff = eftello(fp);
efseeko(fp,(off_t) 0,SEEK_END);
infoptr->ntr = eftello(fp)/infoptr->nsegy;
efseeko(fp, curoff, SEEK_SET); /* restore location */
}
} else { /* Not first entry */
xdr_setpos(infoptr->segy_xdr, infoptr->bufstart);
nread = efread(infoptr->buf ,1 ,HDRBYTES ,infoptr->infp);
if ( nread != HDRBYTES || FALSE == xdrhdrsub(infoptr->segy_xdr,tp)) nread=0;
if(nread == HDRBYTES)
nread += dataread(infoptr, tp, fixed_length);
if (nread <= 0) {
lastfp = infoptr->infp;
return 0;
}
if (fixed_length && (tp->ns != infoptr->nsfirst))
err("%s: on trace #%ld, "
"number of samples in header (%d) "
"differs from number for first trace (%d)",
__FILE__, infoptr->itr, tp->ns, infoptr->nsfirst);
}
++(infoptr->itr);
lastfp = infoptr->infp;
return (nread);
}
int
main (int argc, char **argv)
{
int n1,n2,n1tic,n2tic,nfloats,bbox[4],
i1,i2,grid1,grid2,style,
i3,n3,notitle2=0,
n1c,n2c,n1s,n2s,i1beg,i1end,i2beg,i2end,i1c,i2c,
n1dsp=0,n2dsp=0,n3dsp=0,loopdsp,
nz,iz,i1step,i2step,verbose;
float labelsize,titlesize,perc,clip,bperc,wperc,bclip,wclip,
d1,f1,d2,f2,*z,*temp,zscale,zoffset,zi,
f3,d3,f3s,
xbox,ybox,wbox,hbox,
x1beg,x1end,x2beg,x2end,
x1min,x1max,x2min,x2max,
d1num,f1num,d2num,f2num,
p1beg,p1end,p2beg,p2end,matrix[6],
d1s,d2s;
unsigned char *cz,*czp,*sz;
char *label1="",*label2="",*title="",*title2="",
*labelfont="Helvetica",*titlefont="Helvetica-Bold",
*styles="seismic",*grid1s="none",*grid2s="none";
char title2s[80],c80[80];
FILE *infp=stdin;
/* initialize getpar */
initargs(argc,argv);
requestdoc(1);
/* get parameters describing 1st dimension sampling */
if (!getparint("n1",&n1)) err("must specify n1!\n");
if (!getparfloat("d1",&d1)) d1 = 1.0;
if (!getparfloat("f1",&f1)) f1 = 0.0;
/* get parameters describing 2nd dimension sampling */
if (!getparint("n2",&n2)) {
if (efseeko(infp,(off_t) 0,SEEK_END)!=0)
err("must specify n2 if in a pipe!\n");
nfloats = (int) (eftello(infp)/((off_t) sizeof(float)));
efseeko(infp,(off_t) 0,SEEK_SET);
n2 = nfloats/n1;
n3 = 1;
}
if (!getparfloat("d2",&d2)) d2 = 1.0;
if (!getparfloat("f2",&f2)) f2 = 0.0;
/* get parameters describing 3rd dimension sampling */
if (!getparint("n3",&n3)) {
if (efseeko(infp,(off_t) 0,SEEK_END)!=0)
err("must specify n3 if in a pipe!\n");
nfloats = (int) (eftello(infp)/((off_t) sizeof(float)));
efseeko(infp,(off_t) 0,SEEK_SET);
n3 = nfloats/n1/n2;
}
if (!getparfloat("d3",&d3)) d3 = 1.0;
if (!getparfloat("f3",&f3)) f3 = d3;
/* set up desired looping mode */
if (!getparint("loopdsp",&loopdsp)) loopdsp = 3;
if (loopdsp == 3) {
n1dsp = n1;
n2dsp=n2;
n3dsp=n3;
} else if (loopdsp == 1) {
n1dsp = n2;
n2dsp=n3;
n3dsp=n1;
} else if (loopdsp == 2) {
n1dsp = n1;
n2dsp=n3;
n3dsp=n2;
}
x1min = (d1>0.0)?f1:f1+(n1dsp-1)*d1;
x1max = (d1<0.0)?f1:f1+(n1dsp-1)*d1;
x2min = (d2>0.0)?f2:f2+(n2dsp-1)*d2;
x2max = (d2<0.0)?f2:f2+(n2dsp-1)*d2;
/* read binary data to be plotted */
nz = n1*n2*n3;
z = ealloc1float(nz);
if (fread(z,sizeof(float),nz,infp)!=nz)
err("error reading input file!\n");
/* if necessary, determine clips from percentiles */
if (getparfloat("clip",&clip)) {
bclip = clip;
wclip = -clip;
}
if ((!getparfloat("bclip",&bclip) || !getparfloat("wclip",&wclip)) &&
!getparfloat("clip",&clip)) {
perc = 100.0; getparfloat("perc",&perc);
temp = ealloc1float(nz);
for (iz=0; iz<nz; iz++)
temp[iz] = z[iz];
if (!getparfloat("bclip",&bclip)) {
bperc = perc; getparfloat("bperc",&bperc);
iz = (nz*bperc/100.0);
if (iz<0) iz = 0;
if (iz>nz-1) iz = nz-1;
qkfind(iz,nz,temp);
bclip = temp[iz];
}
if (!getparfloat("wclip",&wclip)) {
wperc = 100.0-perc; getparfloat("wperc",&wperc);
iz = (nz*wperc/100.0);
if (iz<0) iz = 0;
if (iz>nz-1) iz = nz-1;
qkfind(iz,nz,temp);
wclip = temp[iz];
}
free1float(temp);
}
/* transpose data if needed */
if (loopdsp == 1) {
temp = ealloc1float(nz);
for (iz=0;iz<nz;iz++) temp[iz] = z[iz];
for (i3=0;i3<n3;i3++) {
for (i2=0;i2<n2;i2++) {
for(i1=0;i1<n1;i1++) {
z[i2+i3*n2+i1*n2*n3]
= temp[i1+i2*n1+i3*n1*n2];
}
}
}
free1float(temp);
} else if (loopdsp == 2) {
temp = ealloc1float(nz);
for (iz=0;iz<nz;iz++) temp[iz] = z[iz];
for (i3=0;i3<n3;i3++) {
for (i2=0;i2<n2;i2++) {
for(i1=0;i1<n1;i1++) {
z[i1+i3*n1+i2*n1*n3]
= temp[i1+i2*n1+i3*n1*n2];
}
}
}
free1float(temp);
}
verbose = 1; getparint("verbose",&verbose);
if (verbose) warn("bclip=%g wclip=%g",bclip,wclip);
/* get scaled sampling intervals */
d1s = 1.0; getparfloat("d1s",&d1s);
d2s = 1.0; getparfloat("d2s",&d2s);
d1s = fabs(d1s); d1s *= d1;
d2s = fabs(d2s); d2s *= d2;
/* get axes parameters */
xbox = 1.0; getparfloat("xbox",&xbox);
ybox = 1.5; getparfloat("ybox",&ybox);
wbox = 6.0; getparfloat("wbox",&wbox);
hbox = 8.0; getparfloat("hbox",&hbox);
x1beg = x1min; getparfloat("x1beg",&x1beg);
x1end = x1max; getparfloat("x1end",&x1end);
d1num = 0.0; getparfloat("d1num",&d1num);
f1num = x1min; getparfloat("f1num",&f1num);
n1tic = 1; getparint("n1tic",&n1tic);
getparstring("grid1",&grid1s);
if (STREQ("dot",grid1s))
grid1 = DOT;
else if (STREQ("dash",grid1s))
grid1 = DASH;
else if (STREQ("solid",grid1s))
grid1 = SOLID;
else
grid1 = NONE;
getparstring("label1",&label1);
x2beg = x2min; getparfloat("x2beg",&x2beg);
x2end = x2max; getparfloat("x2end",&x2end);
d2num = 0.0; getparfloat("d2num",&d2num);
f2num = 0.0; getparfloat("f2num",&f2num);
n2tic = 1; getparint("n2tic",&n2tic);
getparstring("grid2",&grid2s);
if (STREQ("dot",grid2s))
grid2 = DOT;
else if (STREQ("dash",grid2s))
grid2 = DASH;
else if (STREQ("solid",grid2s))
grid2 = SOLID;
else
grid2 = NONE;
getparstring("label2",&label2);
getparstring("labelfont",&labelfont);
labelsize = 18.0; getparfloat("labelsize",&labelsize);
getparstring("title",&title);
if (!getparstring("title2",&title2)) notitle2=1;
getparstring("titlefont",&titlefont);
titlesize = 24.0; getparfloat("titlesize",&titlesize);
getparstring("style",&styles);
if (STREQ("normal",styles))
style = NORMAL;
else
style = SEISMIC;
/* adjust x1beg and x1end to fall on sampled values */
i1beg = NINT((x1beg-f1)/d1);
i1beg = MAX(0,MIN(n1dsp-1,i1beg));
x1beg = f1+i1beg*d1;
i1end = NINT((x1end-f1)/d1);
i1end = MAX(0,MIN(n1dsp-1,i1end));
x1end = f1+i1end*d1;
/* adjust x2beg and x2end to fall on sampled values */
i2beg = NINT((x2beg-f2)/d2);
i2beg = MAX(0,MIN(n2dsp-1,i2beg));
x2beg = f2+i2beg*d2;
i2end = NINT((x2end-f2)/d2);
i2end = MAX(0,MIN(n2dsp-1,i2end));
x2end = f2+i2end*d2;
/* allocate space for image bytes */
n1c = 1+abs(i1end-i1beg);
n2c = 1+abs(i2end-i2beg);
/* convert data to be imaged into unsigned characters */
zscale = (wclip!=bclip)?255.0/(wclip-bclip):1.0e10;
zoffset = -bclip*zscale;
i1step = (i1end>i1beg)?1:-1;
i2step = (i2end>i2beg)?1:-1;
/* determine sampling after scaling */
n1s = MAX(1,NINT(1+(n1c-1)*d1/d1s));
d1s = (n1s>1)?d1*(n1c-1)/(n1s-1):d1;
n2s = MAX(1,NINT(1+(n2c-1)*d2/d2s));
d2s = (n2s>1)?d2*(n2c-1)/(n2s-1):d2;
/* convert axes box parameters from inches to points */
xbox *= 72.0;
ybox *= 72.0;
wbox *= 72.0;
hbox *= 72.0;
/* set bounding box */
psAxesBBox3(
xbox,ybox,wbox,hbox,
labelfont,labelsize,
titlefont,titlesize,
style,bbox);
boundingbox(bbox[0],bbox[1],bbox[2],bbox[3]);
/* begin PostScript */
beginps();
/* loop over n3 */
for(i3=0;i3<n3dsp;i3++) {
cz = ealloc1(n1c*n2c,sizeof(char));
czp = cz;
for (i1c=0,i1=i1beg; i1c<n1c; i1c++,i1+=i1step) {
for (i2c=0,i2=i2beg; i2c<n2c; i2c++,i2+=i2step) {
zi = zoffset
+z[i1+i2*n1dsp+i3*n1dsp*n2dsp]*zscale;
if (zi<0.0) zi = 0.0;
if (zi>255.0) zi = 255.0;
*czp++ = (unsigned char)zi;
}
}
/* if necessary, interpolate to scaled sampling intervals */
if (n1s!=n1c || n2s!=n2c) {
sz = ealloc1(n1s*n2s,sizeof(char));
intl2b(n2c,d2,0.0,n1c,d1,0.0,cz,
n2s,d2s,0.0,n1s,d1s,0.0,sz);
free1(cz);
} else {
sz = cz;
}
/* determine axes pads */
p1beg = (x1end>x1beg)?-fabs(d1s)/2:fabs(d1s)/2;
p1end = (x1end>x1beg)?fabs(d1s)/2:-fabs(d1s)/2;
p2beg = (x2end>x2beg)?-fabs(d2s)/2:fabs(d2s)/2;
p2end = (x2end>x2beg)?fabs(d2s)/2:-fabs(d2s)/2;
newpage("1",i3+1);
/* save graphics state */
gsave();
/* translate coordinate system by box offset */
translate(xbox,ybox);
/* determine image matrix */
if (style==NORMAL) {
matrix[0] = 0; matrix[1] = n1s; matrix[2] = n2s;
matrix[3] = 0; matrix[4] = 0; matrix[5] = 0;
} else {
matrix[0] = n2s; matrix[1] = 0; matrix[2] = 0;
matrix[3] = -n1s; matrix[4] = 0; matrix[5] = n1s;
}
scale(wbox,hbox);
/* draw the image (before axes so grid lines are visible) */
image(n2s,n1s,8,matrix,sz);
/* restore graphics state */
grestore();
/* update title2 only if n3>1*/
strcpy(title2s,title2);
if ( notitle2 == 0 ) {
char *nullchar="";
f3s = f3 + i3 * d3;
sprintf(c80,"= %.4g %s",f3s,nullchar);
strcat(title2s,c80);
}
/* draw axes and title */
psAxesBox3(
xbox,ybox,wbox,hbox,
x1beg,x1end,p1beg,p1end,
d1num,f1num,n1tic,grid1,label1,
x2beg,x2end,p2beg,p2end,
d2num,f2num,n2tic,grid2,label2,
labelfont,labelsize,
title,titlefont,titlesize,
style, title2s);
showpage();
}
/* end PostScript */
endps();
return EXIT_SUCCESS;
}
int
main (int argc, char **argv)
{
int n1,n2,n3,nbpe,perm,i1,i2,i3,verbose;
char *v=NULL;
char *tempstem=NULL, **tempdirs=NULL;
int ntempdirs;
char tmpstem[] = "foo";
char tmpdir[] = "/tmp";
FILE *infp=stdin,*outfp=stdout;
VND *handle;
long N[3] ;
long key[3] ;
/* hook up getpar */
initargs(argc,argv);
requestdoc(1);
/* get parameters */
if (!getparint("n1",&n1)) err("must specify n1!");
if (!getparint("n2",&n2)) err("must specify n2!");
ntempdirs = countparval("scratchdir");
if(ntempdirs > 0) {
tempdirs = (char **) ealloc1(ntempdirs, sizeof(char *));
getparstringarray("scratchdir",tempdirs);
} else {
ntempdirs = 1;
tempdirs = (char **) ealloc1(ntempdirs, sizeof(char *));
strcpy(tempdirs[0],tmpdir);
}
if (!getparstring("scratchstem",&tempstem))
tempstem = &tmpstem[0];
if (!getparint("nbpe",&nbpe)) nbpe = sizeof(float);
if (!getparint("perm",&perm)) perm = 231; /* trace to time slice */
if (!getparint("n3",&n3)) {
if (efseeko(infp,(off_t) 0,SEEK_END)==-1)
err("input file size unknown; specify n3\n");
n3 = (int) (eftello(infp)/(((off_t)n1)*((off_t)n2)*((off_t)nbpe)));
efseeko(infp, (off_t) 0,SEEK_SET);
}
verbose = 0; getparint("verbose",&verbose);
if (verbose) warn("n1=%d n2=%d n3=%d nbpe=%d perm=%d\n",n1,n2,n3,nbpe,perm);
/* allocate space for a single vector in any dimension */
v = ealloc1((n1+n2+n3),nbpe);
/* allocate big matrix state */
N[0] = n1; N[1] = n2; N[2] = n3;
handle = VNDop(2,200000000,3,N,1,nbpe,tempstem,ntempdirs,tempdirs,1);
/* put vectors along 1st dimension to big matrix */
if (verbose) warn("Reading input file");
for (i3=0; i3<n3; i3++) {
for (i2=0; i2<n2; i2++) {
if (fread(v,1,nbpe*n1,infp)!=nbpe*n1)
err("Error reading input file: i2=%d\n",i2);
key[0] = 0; key[1] = i2; key[2] = i3;
VNDrw('w',0,handle,0,key,0,v,0,1,n1,21,NULL);
}
}
/* get vectors along 2nd dimension from big matrix */
if (verbose) warn("Writing output file");
switch(perm) {
case 123:
for (i3=0; i3<n3; i3++) {
for (i2=0; i2<n2; i2++) {
key[0] = 0; key[1] = i2; key[2] = i3;
VNDrw('r',0,handle,0,key,0,v,0,1,n1,123,NULL);
if (fwrite(v,1,nbpe*n1,outfp)!=nbpe*n1)
err("Error writing output file: i2=%d i3=%d",i2,i3);
}
}
break;
case 132:
for (i2=0; i2<n2; i2++) {
for (i3=0; i3<n3; i3++) {
key[0] = 0; key[1] = i2; key[2] = i3;
VNDrw('r',0,handle,0,key,0,v,0,1,n1,132,NULL);
if (fwrite(v,1,nbpe*n1,outfp)!=nbpe*n1)
err("Error writing output file: i2=%d i3=%d",i2,i3);
}
}
break;
case 213:
for (i3=0; i3<n3; i3++) {
for (i1=0; i1<n1; i1++) {
key[0] = i1; key[1] = 0; key[2] = i3;
VNDrw('r',0,handle,1,key,0,v,0,1,n2,213,NULL);
if (fwrite(v,1,nbpe*n2,outfp)!=nbpe*n2)
err("Error writing output file: i1=%d i3=%d",i1,i3);
}
}
break;
case 231:
for (i1=0; i1<n1; i1++) {
for (i3=0; i3<n3; i3++) {
key[0] = i1; key[1] = 0; key[2] = i3;
VNDrw('r',0,handle,1,key,0,v,0,1,n2,231,NULL);
if (fwrite(v,1,nbpe*n2,outfp)!=nbpe*n2)
err("Error writing output file: i1=%d i3=%d",i1,i3);
}
}
break;
case 312:
for (i2=0; i2<n2; i2++) {
for (i1=0; i1<n1; i1++) {
key[0] = i1; key[1] = i2; key[2] = 0;
VNDrw('r',0,handle,2,key,0,v,0,1,n3,312,NULL);
if (fwrite(v,1,nbpe*n3,outfp)!=nbpe*n3)
err("Error writing output file: i1=%d i2=%d",i1,i2);
}
}
break;
case 321:
for (i1=0; i1<n1; i1++) {
for (i2=0; i2<n2; i2++) {
key[0] = i1; key[1] = i2; key[2] = 0;
VNDrw('r',0,handle,2,key,0,v,0,1,n3,321,NULL);
if (fwrite(v,1,nbpe*n3,outfp)!=nbpe*n3)
err("Error writing output file: i1=%d i2=%d",i1,i2);
}
}
break;
default:
err("Unrecognized transpose permutation %d",perm);
}
/* free big matrix state */
VNDcl(handle,1);
if (verbose) warn("Transpose done!");
return(CWP_Exit());
}
static void drawimage(int hls, float colors[3][2],
int width, int height, int bps, float matrix[], unsigned char* z)
{
int row,col,rowbytes,hi,lo,zi,byte;
unsigned char *rtab,*gtab,*btab,*rgb,*rgbp,*crgbp,*zp;
/* handle gray scale as a special case */
if ( (hls &&
colors[L][0]==0.0 &&
colors[L][1]==1.0 &&
colors[S][0]==0.0 &&
colors[S][1]==0.0) ||
(colors[R][0]==0.0 && colors[R][1]==1.0 &&
colors[G][0]==0.0 && colors[G][1]==1.0 &&
colors[B][0]==0.0 && colors[B][1]==1.0) ) {
image(width,height,8,matrix,z);
return;
}
/* allocate space for rgb tables */
rtab = ealloc1(256,sizeof(unsigned char));
gtab = ealloc1(256,sizeof(unsigned char));
btab = ealloc1(256,sizeof(unsigned char));
/* if colors are hls values, map from hls to rgb tables */
if (hls) {
float hscale,lscale,sscale,hbase,lbase,sbase,h,l,s,r,g,b;
hbase = colors[H][0];
hscale = (colors[H][1]-colors[H][0])/255.0;
lbase = colors[L][0];
lscale = (colors[L][1]-colors[L][0])/255.0;
sbase = colors[S][0];
sscale = (colors[S][1]-colors[S][0])/255.0;
for (zi=0; zi<256; ++zi) {
h = hbase+zi*hscale;
l = lbase+zi*lscale;
s = sbase+zi*sscale;
hlsrgb(h*360.0,l,s,&r,&g,&b);
rtab[zi] = r*255.0;
gtab[zi] = g*255.0;
btab[zi] = b*255.0;
}
/* else colors are rgb values, map linearly to rgb tables */
} else {
float rscale,gscale,bscale,rbase,gbase,bbase;
rbase = colors[0][0]*255.0;
rscale = colors[0][1]-colors[0][0];
gbase = colors[1][0]*255.0;
gscale = colors[1][1]-colors[1][0];
bbase = colors[2][0]*255.0;
bscale = colors[2][1]-colors[2][0];
for (zi=0; zi<256; ++zi) {
rtab[zi] = rbase+zi*rscale;
gtab[zi] = gbase+zi*gscale;
btab[zi] = bbase+zi*bscale;
}
}
/* convert unsigned char to rgb unsigned char */
rgb = ealloc1(width*height*3,sizeof(unsigned char));
for (row=0,rgbp=rgb,zp=z; row<height; ++row) {
for (col=0; col<width; ++col) {
zi = *zp++;
*rgbp++ = (unsigned char)rtab[zi];
*rgbp++ = (unsigned char)gtab[zi];
*rgbp++ = (unsigned char)btab[zi];
}
}
/* free tables */
free1(rtab);
free1(gtab);
free1(btab);
/* if necessary, compress 24 bit color to 12 bit color */
if (bps==12) {
rowbytes = 1+(width*12-1)/8;
crgbp = rgbp = rgb;
for (row=0; row<height; ++row) {
for (byte=0; byte<rowbytes-1; ++byte) {
hi = (*rgbp++)>>4;
lo = (*rgbp++)>>4;
*crgbp++ = (hi<<4)|lo;
}
if (width%2) {
hi = (*rgbp++)>>4;
*crgbp++ = hi<<4;
} else {
int main (int argc, char **argv)
{
int n1,n2,n3,n1s,n2s,n3s,n1c,n2c,n3c,i1,i2,i3,i1c,i2c,i3c,
i1beg,i1end,i2beg,i2end,i3beg,i3end,i1step,i2step,i3step,
n1tic,n2tic,n3tic,grid1,grid2,grid3,nz,iz,
verbose,faces,hls,bps,style=SEISMIC,bbox[4],
legend,ugrid=SOLID,lstyle=VERTLEFT,lz,lbegsup=0,lendsup=0,ln=256,
lbbox[4];
float d1,d2,d3,d1s,d2s,d3s,f1,f2,f3,size1,size2,size3,xbox,ybox,angle,
x1min,x1max,x2min,x2max,x3min,x3max,
x1beg,x1end,x2beg,x2end,x3beg,x3end,
d1num,f1num,d2num,f2num,d3num,f3num,
p1beg,p1end,p2beg,p2end,p3beg,p3end,
clip,bclip,wclip,perc,bperc,wperc,
zscale,zoffset,zi,labelsize,titlesize,
*z,*zfront,*zside,*ztop,*temp,matrix[6],colors[3][2],
lwidth,lheight,lx,ly,lbeg,lend,
lmin=(float)FLT_MAX,lmax=(float)-FLT_MAX,
ldnum,lfnum,ld,lf=0,labmatrix[6];
unsigned char *czfront,*czside,*cztop,
*szfront,*szside,*sztop,*czp,
*data_legend=NULL;
char *label1="",*label2="",*label3="",*title="",
*labelfont="Helvetica",*titlefont="Helvetica-Bold",
*grid1s="none",*grid2s="none",*grid3s="none",
*titlecolor="black",*axescolor="black",*gridcolor="black",
*frontf,*sidef,*topf,
*units="", *legendfont="times_roman10", *lstyles="vertleft",*lgrids="none";
FILE *infp=stdin,*frontfp,*sidefp,*topfp;
/* initialize getpar */
initargs(argc,argv);
requestdoc(1);
/* get parameters describing 1st dimension sampling */
if (!getparint("n1",&n1)) err("must specify n1!\n");
if (!getparfloat("d1",&d1)) d1 = 1.0;
if (!getparfloat("f1",&f1)) f1 = 0.0;
/* get parameters describing 2nd dimension sampling */
if (!getparint("n2",&n2)) err("must specify n2!\n");
if (!getparfloat("d2",&d2)) d2 = 1.0;
if (!getparfloat("f2",&f2)) f2 = 0.0;
/* get parameters describing 3rd dimension sampling */
if (!getparint("n3",&n3)) err("must specify n3!\n");
if (!getparfloat("d3",&d3)) d3 = 1.0;
if (!getparfloat("f3",&f3)) f3 = 0.0;
/* determine input type */
if (!getparint("faces",&faces)) faces = 0;
/* read color parameters */
bps = 8;
hls = 0;
colors[R][0] = colors[G][0] = colors[B][0] = 0.0;
colors[R][1] = colors[G][1] = colors[B][1] = 1.0;
if (countparval("brgb") || countparval("wrgb")) {
float brgb[3],wrgb[3];
brgb[R] = brgb[G] = brgb[B] = 0.0;
wrgb[R] = wrgb[G] = wrgb[B] = 1.0;
getparfloat("brgb",&brgb[0]);
getparfloat("wrgb",&wrgb[0]);
brgb[R] = MAX(0.0,MIN(1.0,brgb[R]));
wrgb[R] = MAX(0.0,MIN(1.0,wrgb[R]));
brgb[G] = MAX(0.0,MIN(1.0,brgb[G]));
wrgb[G] = MAX(0.0,MIN(1.0,wrgb[G]));
brgb[B] = MAX(0.0,MIN(1.0,brgb[B]));
wrgb[B] = MAX(0.0,MIN(1.0,wrgb[B]));
colors[R][0] = brgb[R]; colors[R][1] = wrgb[R];
colors[G][0] = brgb[G]; colors[G][1] = wrgb[G];
colors[B][0] = brgb[B]; colors[B][1] = wrgb[B];
if (!getparint("bps",&bps)) bps = 12;
if (bps!=12 && bps!=24)
err("bps must equal 12 or 24 for color plots!\n");
} else if (countparval("bhls") || countparval("whls")) {
float bhls[3],whls[3];
hls = 1;
bhls[H] = whls[H] = 0.0;
bhls[L] = 0.0; whls[L] = 1.0;
bhls[S] = whls[S] = 0.0;
getparfloat("bhls",&bhls[0]);
getparfloat("whls",&whls[0]);
bhls[L] = MAX(0.0,MIN(1.0,bhls[L]));
whls[L] = MAX(0.0,MIN(1.0,whls[L]));
bhls[S] = MAX(0.0,MIN(1.0,bhls[S]));
whls[S] = MAX(0.0,MIN(1.0,whls[S]));
colors[H][0] = bhls[0]; colors[H][1] = whls[0];
colors[L][0] = bhls[1]; colors[L][1] = whls[1];
colors[S][0] = bhls[2]; colors[S][1] = whls[2];
if (!getparint("bps",&bps)) bps = 12;
if (bps!=12 && bps!=24)
err("bps must equal 12 or 24 for color plots!\n");
}
/* allocate space */
nz = n1*n2+n1*n3+n2*n3;
z = ealloc1float(nz);
zfront = z;
zside = zfront+n1*n2;
ztop = zside+n1*n3;
/* read data */
if (getparstring("front",&frontf)
&& getparstring("side",&sidef)
&& getparstring("top",&topf)) {
/* read face files */
if ((frontfp = fopen(frontf,"r")) == NULL)
err("error opening front file!\n");
if (fread(zfront,sizeof(float),n1*n2,frontfp)!=n1*n2)
err("error reading front file!\n");
if ((sidefp = fopen(sidef,"r")) == NULL)
err("error opening side file!\n");
if (fread(zside,sizeof(float),n1*n3,sidefp)!=n1*n3)
err("error reading side file!\n");
if ((topfp = fopen(topf,"r")) == NULL)
err("error opening top file!\n");
if (fread(ztop,sizeof(float),n2*n3,topfp)!=n2*n3)
err("error reading top file!\n");
} else if (getparstring("front",&frontf)
|| getparstring("side",&sidef)
|| getparstring("top",&topf)) {
err("must specify all or none of face, side, and top!\n");
} else if (faces) {
/* read faces from stdin */
if (fread(zfront,sizeof(float),n1*n2,infp)!=n1*n2)
err("error reading front from input!\n");
if (fread(zside,sizeof(float),n1*n3, infp)!=n1*n3)
err("error reading side from input!\n");
if (fread(ztop,sizeof(float),n2*n3, infp)!=n2*n3)
err("error reading top from input!\n");
} else {
/* read cube from stdin, pick off faces */
temp = ealloc1float(n1);
for (i3=0; i3<n3; i3++) {
for (i2=0; i2<n2; i2++) {
if (fread(temp,sizeof(float),n1,infp)!=n1)
err("error reading cube from input!\n");
if (i3==0)
for (i1=0; i1<n1; i1++)
zfront[i1+i2*n1] = temp[i1];
if (i2==n2-1)
for (i1=0; i1<n1; i1++)
zside[i1+i3*n1] = temp[i1];
ztop[i2+i3*n2] = temp[0];
}
}
free1float(temp);
}
/* if necessary, determine clips from percentiles */
if (getparfloat("clip",&clip)) {
bclip = clip;
wclip = -clip;
}
if ((!getparfloat("bclip",&bclip) || !getparfloat("wclip",&wclip)) &&
!getparfloat("clip",&clip)) {
perc = 100.0; getparfloat("perc",&perc);
temp = ealloc1float(nz);
for (iz=0; iz<nz; iz++)
temp[iz] = z[iz];
if (!getparfloat("bclip",&bclip)) {
bperc = perc; getparfloat("bperc",&bperc);
iz = (nz*bperc/100.0);
if (iz<0) iz = 0;
if (iz>nz-1) iz = nz-1;
qkfind(iz,nz,temp);
bclip = temp[iz];
}
if (!getparfloat("wclip",&wclip)) {
wperc = 100.0-perc; getparfloat("wperc",&wperc);
iz = (nz*wperc/100.0);
if (iz<0) iz = 0;
if (iz>nz-1) iz = nz-1;
qkfind(iz,nz,temp);
wclip = temp[iz];
}
free1float(temp);
}
if (!getparint("verbose",&verbose)) verbose = 1;
if (verbose) warn("bclip=%g wclip=%g",bclip,wclip);
/* get scaled sampling intervals */
if (!getparfloat("d1s",&d1s)) d1s = 1.0;
if (!getparfloat("d2s",&d2s)) d2s = 1.0;
if (!getparfloat("d3s",&d3s)) d3s = 1.0;
d1s = fabs(d1s); d1s *= d1;
d2s = fabs(d2s); d2s *= d2;
d3s = fabs(d3s); d3s *= d3;
/* get projection angle, convert to radians */
if(!getparfloat("angle",&angle)) angle = 45.0;
angle = MAX(angle,0.00001);
angle = MIN(angle,90.0);
angle *= PI/180.0;
/* get axes parameters */
if(!getparfloat("size1",&size1)) size1 = 4.0;
if(!getparfloat("size2",&size2)) size2 = 4.0;
if(!getparfloat("size3",&size3)) size3 = 3.0;
if (!getparfloat("xbox",&xbox)) xbox = 1.5;
if (!getparfloat("ybox",&ybox)) ybox = 1.5;
/* compute extreme values */
x1min = (d1>0.0)?f1:f1+(n1-1)*d1;
x1max = (d1<0.0)?f1:f1+(n1-1)*d1;
x2min = (d2>0.0)?f2:f2+(n2-1)*d2;
x2max = (d2<0.0)?f2:f2+(n2-1)*d2;
x3min = (d3>0.0)?f3:f3+(n3-1)*d3;
x3max = (d3<0.0)?f3:f3+(n3-1)*d3;
/* get axis1 parameters */
x1beg = x1min;
x1end = x1max; getparfloat("x1end",&x1end);
d1num = 0.0; getparfloat("d1num",&d1num);
f1num = x1min; getparfloat("f1num",&f1num);
n1tic = 1; getparint("n1tic",&n1tic);
getparstring("grid1",&grid1s);
if (STREQ("dot",grid1s)) grid1 = DOT;
else if (STREQ("dash",grid1s)) grid1 = DASH;
else if (STREQ("solid",grid1s)) grid1 = SOLID;
else grid1 = NONE;
getparstring("label1",&label1);
/* get axis2 parameters */
x2beg = x2min; getparfloat("x2beg",&x2beg);
x2end = x2max;
d2num = 0.0; getparfloat("d2num",&d2num);
f2num = x2min; getparfloat("f2num",&f2num);
n2tic = 1; getparint("n2tic",&n2tic);
getparstring("grid2",&grid2s);
if (STREQ("dot",grid2s)) grid2 = DOT;
else if (STREQ("dash",grid2s)) grid2 = DASH;
else if (STREQ("solid",grid2s)) grid2 = SOLID;
else grid2 = NONE;
getparstring("label2",&label2);
/* get axis3 parameters */
x3beg = x3min;
x3end = x3max; getparfloat("x3end",&x3end);
d3num = 0.0; getparfloat("d3num",&d3num);
f3num = x3min; getparfloat("f3num",&f3num);
n3tic = 1; getparint("n3tic",&n3tic);
getparstring("grid3",&grid3s);
if (STREQ("dot",grid3s)) grid3 = DOT;
else if (STREQ("dash",grid3s)) grid3 = DASH;
else if (STREQ("solid",grid3s)) grid3 = SOLID;
else grid3 = NONE;
getparstring("label3",&label3);
/* get additional font parameters */
getparstring("labelfont",&labelfont);
labelsize = 18.0; getparfloat("labelsize",&labelsize);
getparstring("title",&title);
getparstring("titlefont",&titlefont);
titlesize = 24.0; getparfloat("titlesize",&titlesize);
getparstring("titlecolor",&titlecolor);
getparstring("axescolor",&axescolor);
getparstring("gridcolor",&gridcolor);
style = SEISMIC;
/* adjust x1beg and x1end to fall on sampled values */
i1beg = NINT((x1beg-f1)/d1);
i1beg = MAX(0,MIN(n1,i1beg));
x1beg = f1+i1beg*d1;
i1end = NINT((x1end-f1)/d1);
i1end = MAX(0,MIN(n1-1,i1end));
x1end = f1+i1end*d1;
/* adjust x2beg and x2end to fall on sampled values */
i2beg = NINT((x2beg-f2)/d2);
i2beg = MAX(0,MIN(n2-1,i2beg));
x2beg = f2+i2beg*d2;
i2end = NINT((x2end-f2)/d2);
i2end = MAX(0,MIN(n2-1,i2end));
x2end = f2+i2end*d2;
/* adjust x3beg and x3end to fall on sampled values */
i3beg = NINT((x3beg-f3)/d3);
i3beg = MAX(0,MIN(n3-1,i3beg));
x3beg = f3+i3beg*d3;
i3end = NINT((x3end-f3)/d3);
i3end = MAX(0,MIN(n3-1,i3end));
x3end = f3+i3end*d3;
/* allocate space for image bytes */
n1c = 1+abs(i1end-i1beg);
n2c = 1+abs(i2end-i2beg);
n3c = 1+abs(i3end-i3beg);
czfront = ealloc1(n1c*n2c,sizeof(char));
czside = ealloc1(n1c*n3c,sizeof(char));
cztop = ealloc1(n2c*n3c,sizeof(char));
/* compute conversion constants */
zscale = (wclip!=bclip)?255.0/(wclip-bclip):1.0e10;
zoffset = -bclip*zscale;
i1step = (i1end>i1beg)?1:-1;
i2step = (i2end>i2beg)?1:-1;
i3step = (i3end>i3beg)?1:-1;
/* convert front data to be imaged into unsigned characters */
czp = czfront;
for (i2c=0,i2=i2beg; i2c<n2c; i2c++,i2+=i2step) {
for (i1c=0,i1=i1beg; i1c<n1c; i1c++,i1+=i1step) {
zi = zoffset+zfront[i1+i2*n1]*zscale;
if (zi<0.0) zi = 0.0;
if (zi>255.0) zi = 255.0;
*czp++ = (unsigned char)zi;
}
}
/* convert side data to be imaged into unsigned characters */
czp = czside;
for (i3c=0,i3=i3beg; i3c<n3c; i3c++,i3+=i3step) {
for (i1c=0,i1=i1beg; i1c<n1c; i1c++,i1+=i1step) {
zi = zoffset+zside[i1+i3*n1]*zscale;
if (zi<0.0) zi = 0.0;
if (zi>255.0) zi = 255.0;
*czp++ = (unsigned char)zi;
}
}
/* convert top data to be imaged into unsigned characters */
czp = cztop;
for (i3c=0,i3=i3beg; i3c<n3c; i3c++,i3+=i3step) {
for (i2c=0,i2=i2beg; i2c<n2c; i2c++,i2+=i2step) {
zi = zoffset+ztop[i2+i3*n2]*zscale;
if (zi<0.0) zi = 0.0;
if (zi>255.0) zi = 255.0;
*czp++ = (unsigned char)zi;
}
}
free1float(z);
/* determine sampling after scaling */
n1s = MAX(1,NINT(1+(n1c-1)*d1/d1s));
d1s = (n1s>1)?d1*(n1c-1)/(n1s-1):d1;
n2s = MAX(1,NINT(1+(n2c-1)*d2/d2s));
d2s = (n2s>1)?d2*(n2c-1)/(n2s-1):d2;
n3s = MAX(1,NINT(1+(n3c-1)*d3/d3s));
d3s = (n3s>1)?d3*(n3c-1)/(n3s-1):d3;
/* if necessary, interpolate front to scaled sampling intervals */
if (n1s!=n1c || n2s!=n2c) {
szfront = ealloc1(n1s*n2s,sizeof(char));
intl2b(n1c,d1,0.0,n2c,d2,0.0,czfront,
n1s,d1s,0.0,n2s,d2s,0.0,szfront);
free1(czfront);
} else {
szfront = czfront;
}
/* if necessary, interpolate side to scaled sampling intervals */
if (n1s!=n1c || n3s!=n3c) {
szside = ealloc1(n1s*n3s,sizeof(char));
intl2b(n1c,d1,0.0,n3c,d3,0.0,czside,
n1s,d1s,0.0,n3s,d3s,0.0,szside);
free1(czside);
} else {
szside = czside;
}
/* if necessary, interpolate top to scaled sampling intervals */
if (n2s!=n2c || n3s!=n3c) {
sztop = ealloc1(n2s*n3s,sizeof(char));
intl2b(n2c,d2,0.0,n3c,d3,0.0,cztop,
n2s,d2s,0.0,n3s,d3s,0.0,sztop);
free1(cztop);
} else {
sztop = cztop;
}
/* determine axes pads */
p1beg = (x1end>x1beg)?-fabs(d1s)/2:fabs(d1s)/2;
p1end = (x1end>x1beg)?fabs(d1s)/2:-fabs(d1s)/2;
p2beg = (x2end>x2beg)?-fabs(d2s)/2:fabs(d2s)/2;
p2end = (x2end>x2beg)?fabs(d2s)/2:-fabs(d2s)/2;
p3beg = (x3end>x3beg)?-fabs(d3s)/2:fabs(d3s)/2;
p3end = (x3end>x3beg)?fabs(d3s)/2:-fabs(d3s)/2;
/* get legend specs BEREND, Schoenfelder */
legend = 0; getparint("legend", &legend); /* BEREND, Schoenfelder */
getparstring("units", &units); /* BEREND, Schoenfelder */
getparstring("legendfont", &legendfont); /* BEREND, Schoenfelder */
/* Get or calc legend parameters */
/* Legend min and max: Calc from data read in */
if (legend) {
for (lz=0;lz<nz;lz++) {
lmin=FMIN(lmin,z[lz]);
lmax=FMAX(lmax,z[lz]);
}
if (verbose==2) warn("lmin=%g lmax=%g",lmin,lmax);
lbeg = lmin; if (getparfloat("lbeg",&lbeg)) lbegsup=1;
lend = lmax; if (getparfloat("lend",&lend)) lendsup=1;
/* Change wclip,bclip to be inside legend range */
wclip = FMAX(lbeg,wclip); /* [wclip,bclip] has to be in [lbeg,lend] */
bclip = FMIN(lend,bclip);
if (lbegsup!=1) { /* Add white and black areas to show possible clipping */
float rangeperc=(bclip-wclip)/20.;
lbeg=wclip-rangeperc;
}
if (lendsup!=1) {
float rangeperc=(bclip-wclip)/20.;
lend=bclip+rangeperc;
}
lfnum = lmin; getparfloat("lfnum",&lfnum);
getparstring("lstyle",&lstyles);
if (STREQ("vertright",lstyles))
lstyle = VERTRIGHT;
else if (STREQ("horibottom",lstyles))
lstyle = HORIBOTTOM;
/* legend dimensions (BEREND), Schoenfelder */
lwidth = 0.1 ;lheight = size1+sin(angle)*size3/2;
if (lstyle==HORIBOTTOM) {
lwidth=size2+cos(angle)*size3/1.2 ;lheight = 0.24;
}
getparfloat("lwidth",&lwidth);
getparfloat("lheight",&lheight);
lx=.8;ly = ybox+(size1+sin(angle)*size3-lheight)/2;
if (lstyle==VERTRIGHT) {
lx=xbox+size2+cos(angle)*size3+0.1;
} else if (lstyle==HORIBOTTOM) {
lx=xbox+(size2+cos(angle)*size3-lwidth)/2.0;ly = 1.0;
}
getparfloat("lx",&lx);
getparfloat("ly",&ly);
getparstring("lgrid",&lgrids);
if (STREQ("dot",lgrids))
ugrid = DOT;
else if (STREQ("dash",lgrids))
ugrid = DASH;
else if (STREQ("solid",lgrids))
ugrid = SOLID;
else
ugrid = NONE;
}
if (legend) {
/* Make legend color values */
int lll=0,lcount,perc5=13,ilbeg,ilend; /* color scale */
if (lbegsup!=1) {
ln+=perc5; /* white area */
}
if (lendsup!=1) {
ln+=perc5; /* black area */
}
data_legend = ealloc1(ln,sizeof(char));
if (lbegsup!=1) {
for (lll=0;lll<perc5;lll++) data_legend[lll]=(char) 255; /* white area */
}
for (lcount=255;lcount>=0;lcount--,lll++) data_legend[lll]=(char) lcount;
if (lendsup!=1) {
for (;lll<ln;lll++) data_legend[lll]=(char) 0; /* black area */
}
lf=lbeg;ld=(lend-lbeg)/(ln-1);
if (!(getparfloat("ldnum",&ldnum))) ldnum=0.0;
/* adjust lbeg and lend to fall on sampled values */
ilbeg = NINT((lbeg-lf)/ld);
ilbeg = MAX(0,MIN(ln-1,ilbeg));
lbeg = lf+ilbeg*ld;
ilend = NINT((lend-lf)/ld);
ilend = MAX(0,MIN(ln-1,ilend));
lend = lf+ilend*ld;
/* convert legend parameters to points */
lx *= 72.0; /* Schoenfelder */
ly *= 72.0; /* Schoenfelder */
lwidth *= 72.0; /* Schoenfelder */
lheight *= 72.0; /* Schoenfelder */
}
/* convert axes parameters to points */
size1 *= 72;
size2 *= 72;
size3 *= 72;
xbox *= 72;
ybox *= 72;
/* set bounding box */
psAxesBBox(xbox,ybox,size2+cos(angle)*size3,size1+sin(angle)*size3,
labelfont,labelsize,titlefont,titlesize,style,bbox);
if (legend) {
psLegendBBox( /* Space for legend Schoenfelder */
lx,ly,lwidth,lheight,
labelfont,labelsize,
lstyle,lbbox);
/* Include space for legend Schoenfelder */
bbox[0]=MIN(bbox[0],lbbox[0]);
bbox[1]=MIN(bbox[1],lbbox[1]);
bbox[2]=MAX(bbox[2],lbbox[2]);
bbox[3]=MAX(bbox[3],lbbox[3]);
}
boundingbox(bbox[0],bbox[1],bbox[2],bbox[3]);
/* begin PostScript */
begineps();
/* save graphics state */
gsave();
/* translate coordinate system by box offset */
translate(xbox,ybox);
/* begin front */
gsave();
/* transform coordinates */
translate(0,0);
scale(size2,size1);
/* determine image matrix */
matrix[0] = 0; matrix[1] = n2s; matrix[2] = -n1s;
matrix[3] = 0; matrix[4] = n1s; matrix[5] = 0;
/* draw the image */
drawimage(hls,colors,n1s,n2s,bps,matrix,szfront);
/* end front */
grestore();
/* begin side */
gsave();
/* transform and skew coordinates */
matrix[0] = 1; matrix[1] = tan(angle); matrix[2] = 0;
matrix[3] = 1; matrix[4] = 0; matrix[5] = 0;
translate(size2,0);
concat(matrix);
scale(size3*cos(angle),size1);
/* determine image matrix */
matrix[0] = 0; matrix[1] = n3s; matrix[2] = -n1s;
matrix[3] = 0; matrix[4] = n1s; matrix[5] = 0;
/* draw the image */
drawimage(hls,colors,n1s,n3s,bps,matrix,szside);
/* end side */
grestore();
/* begin top */
gsave();
/* transform and skew coordinates */
matrix[0] = 1; matrix[1] = 0; matrix[2] = 1.0/tan(angle);
matrix[3] = 1; matrix[4] = 0; matrix[5] = 0;
translate(0,size1);
concat(matrix);
scale(size2,size3*sin(angle));
/* determine image matrix */
matrix[0] = n2s; matrix[1] = 0; matrix[2] = 0;
matrix[3] = n3s; matrix[4] = 0; matrix[5] = 0;
/* draw the image */
drawimage(hls,colors,n2s,n3s,bps,matrix,sztop);
/* end top */
grestore();
if (legend) {
gsave();
/* translate coordinate system by box offset */
translate(-xbox,-ybox);
translate(lx,ly);
scale(lwidth,lheight);
if ((lstyle==VERTLEFT) || (lstyle==VERTRIGHT)) {
labmatrix[0] = 1; labmatrix[1] = 0; labmatrix[2] = 0;
labmatrix[3] = ln; labmatrix[4] = 0; labmatrix[5] = 0;
drawimage(hls,colors,1,ln,bps,labmatrix,data_legend);
} else {
labmatrix[0] = -1; labmatrix[1] = 0; labmatrix[2] = 0;
labmatrix[3] = ln; labmatrix[4] = 0; labmatrix[5] = 0;
rotate(-90);
drawimage(hls,colors,1,ln,bps,labmatrix,data_legend);
rotate(90);
}
grestore();
}
/* restore graphics state */
grestore();
psCubeAxesBox(xbox,ybox,size1,size2,size3,angle,
x1beg,x1end,p1beg,p1end,
d1num,f1num,n1tic,grid1,label1,
x2beg,x2end,p2beg,p2end,
d2num,f2num,n2tic,grid2,label2,
x3beg,x3end,p3beg,p3end,
d3num,f3num,n3tic,grid3,label3,
labelfont,labelsize,
title,titlefont,titlesize,
titlecolor,axescolor,gridcolor);
/* draw axes and title for legend Schoenfelder*/
if (legend) {
float lpbeg,lpend;
int lntic=1;
gsave();
lpbeg = 0.0; /*(lend>lbeg)?-fabs(d1s)/2:fabs(d1s)/2;*/
lpend = 0.0; /*(lend>lbeg)?fabs(d1s)/2:-fabs(d1s)/2;*/
psLegendBox(
lx,ly,lwidth,lheight,
lbeg,lend,lpbeg,lpend,
ldnum,lf,lntic,ugrid,units,
labelfont,labelsize,
axescolor,gridcolor,
lstyle);
grestore();
}
/* end PostScript */
showpage();
endeps();
return 0;
}
int main (int argc, char **argv)
{
int nt,it,np,ntau,itau,nx,ix,nk,nkmax,ik,
ntfft,nxfft,nv,trans,norm,conv,verbose;
float dt,dx,dpx,k,dk,kscl,t,*tt,*vt,*tmig,*vmig,
(*vsind)[4],**ptx,**divcor;
complex **ptk;
char *vfile="";
FILE *hfp,*tfp;
/* hook up getpar */
initargs(argc,argv);
requestdoc(1);
/* get information from the first header */
if (!gettr(&tr)) err("can't get first trace");
nt = tr.ns;
dt = tr.dt/1000000.0;
/* get parameters */
if (!getparfloat("dxcdp",&dx)) err("dxcdp required");
if (!getparint("np",&np)) np=50;
if (!getparint("trans",&trans)) trans=0;
if (!getparint("norm",&norm)) norm=1;
if (!getparint("conv",&conv)) conv=0;
if (!getparint("verbose",&verbose)) verbose=0;
/* get velocity function */
vt=ealloc1float(nt);
tt=ealloc1float(nt);
for (it=0; it<nt; it++)
tt[it]=it*dt;
if (!getparstring ("vfile",&vfile)){
ntau = countparval("tmig");
if (ntau==0) ntau=1;
tmig = ealloc1float(ntau);
if (!getparfloat("tmig",tmig)) tmig[0] = 0.0;
nv = countparval("vmig");
if (nv==0) nv=1;
if (nv!=ntau)
err("number of tmig and vmig must be equal");
vmig = ealloc1float(nv);
if (!getparfloat("vmig",vmig)) vmig[0] = 1500.0;
for (itau=1; itau<ntau; itau++)
if (tmig[itau]<=tmig[itau-1])
err("tmig must increase monotonically");
for (it=0,t=0.0; it<nt; ++it,t+=dt)
intlin(ntau,tmig,vmig,vmig[0],vmig[ntau-1],
1,&t,&vt[it]);
if (ntau!=nt){
vsind = (float (*)[4])ealloc1float(ntau*4);
cmonot(ntau,tmig,vmig,vsind);
intcub(0,ntau,tmig,vsind,nt,tt,vt);
}
} else{
if (fread(vt,sizeof(float),nt,fopen(vfile,"r"))!=nt)
err("Not %d velocities in file %s",nt,vfile);
}
/* copy traces and headers to temporary files */
tfp = tmpfile();
hfp = tmpfile();
nx = 0;
do {
nx++;
fwrite(&tr,HDRBYTES,1,hfp);
fwrite(tr.data,sizeof(float),nt,tfp);
} while(gettr(&tr));
fseek(hfp,0L,SEEK_SET);
fseek(tfp,0L,SEEK_SET);
if (verbose) fprintf(stderr,"\t%d traces input\n",nx);
/* determine wavenumber and frequency sampling */
nxfft = npfar(nx);
ntfft = npfa(nt);
nk = nxfft/2+1;
dx *= 0.001;
dk = 2.0*PI/(nxfft*dx);
/* allocate space for Fourier transform */
ptk = ealloc2complex(nt,nk);
ptx = ealloc1(nxfft,sizeof(float*));
for (ix=0; ix<nxfft; ++ix)
ptx[ix] = (float*)ptk[0]+ix*nt;
/* allocate space for divergence correction */
divcor=ealloc2float(nt,np);
/* build table of divergence corrections */
divcortable(nt,np,dt,tt,vt,divcor,trans,norm);
/* apply conventional correction if required */
if (conv==1){
for (ix=0; ix<nx; ++ix){
efread(ptx[ix],sizeof(float),nt,tfp);
for (it=0; it<nt; ++it)
ptx[ix][it] *= divcor[0][it];
}
} else {
/* read and apply fft scaling to traces */
kscl = 1.0/nxfft;
for (ix=0; ix<nx; ++ix) {
efread(ptx[ix],sizeof(float),nt,tfp);
for (it=0; it<nt; ++it)
ptx[ix][it] *= kscl;
}
/* pad with zeros */
for (ix=nx; ix<nxfft; ++ix)
for (it=0; it<nt; ++it)
ptx[ix][it] = 0.0;
/* Fourier transform ptx(t,x) to ptk(t,k) */
pfa2rc(-1,2,nt,nxfft,ptx[0],ptk[0]);
if (verbose) fprintf(stderr,"\tFourier transform done\n");
/* define relevant k range */
nkmax = MIN(nk,NINT(PI/dt/vt[0]/dk));
dpx = 1.0/(np-1)/vt[0];
fprintf(stderr,
"nkmax %d nk %d dk %f dpx %f \n",nkmax,nk,dk,dpx);
/* special case k=0 */
for (it=0; it<nt; it++){
ptk[0][it].r *= divcor[0][it];
ptk[0][it].i *= divcor[0][it];
}
/* loop over wavenumbers */
for (ik=1,k=dk; ik<nkmax; ++ik,k+=dk){
/* report */
if (verbose && ik%(nkmax/10>0?nkmax/10:1)==0)
fprintf(stderr,"\t%d of %d wavenumbers done\n",
ik,nkmax);
/* dip filter divergence correction */
dipfilt(k,dpx,dt,np,ntfft,nt,divcor,ptk[ik],ptk[ik]);
}
/* Fourier transform p(t,k) to p(t,x) */
pfa2cr(1,2,nt,nxfft,ptk[0],ptx[0]);
if (verbose)
fprintf(stderr,"\tinverse Fourier transform done\n");
} /* end else dipdivcor */
/* output migrated traces with headers */
for (ix=0; ix<nx; ++ix) {
efread(&tr,HDRBYTES,1,hfp);
memcpy((void *) tr.data,
(const void *) ptx[ix], nt*sizeof(float));
puttr(&tr);
}
return EXIT_SUCCESS;
}
static void dmooff (float offset, float fmax, int nx, float dx,
int nt, float dt, float ft, float *vrms, float **ptx, float gamma,
float *boh, float *zoh, int ntable, float s1, float s2, float sign)
/*****************************************************************************
perform dmo in f-k domain for one offset
******************************************************************************
Input:
offset source receiver offset
fmax maximum frequency
s1 DMO stretch factor
s2 DMO stretch factor
sign sign of shift
nx number of midpoints
dx midpoint sampling interval
nt number of time samples
dt time sampling interval
ft first time
vrms array[nt] of rms velocities
ptx array[nx][nt] for p(t,x), zero-padded for fft (see notes)
Output:
ptx array[nx][nt] for dmo-corrected p(t,x)
******************************************************************************
Notes:
To avoid having to allocate a separate work space that is larger than the
array ptx[nx][nt], ptx must be zero-padded to enable Fourier transform from x
to k via prime factor FFT. nxpad (nx after zero-padding) can be estimated by
nxpad = 2+npfar(nx+(int)(0.5*ABS(offset/dx)));
where npfar() is a function that returns a valid length for real-to-complex
prime factor FFT. ptx[nx] to ptx[nxfft-1] must point to zeros.
******************************************************************************
Author: Dave Hale, Colorado School of Mines, 08/08/91
*****************************************************************************/
{
int nxfft,itmin,nu,nufft,nw,nk,ix,iu,iw,ik,it,iwn,
iwmin,iwmax,nupad,ikmax,*itn,lnt;
float dw,dk,tcon,wwscl,wwscl2,scale,scales,kmax,lt,
amp,phase,fr,fi,pwr,pwi,cphase,sphase,os1,os2,
wmin,wmax,fftscl,du,fu,w,k,*uoft,*tofu,g1,h,vmin,hk,t,*bboh=NULL;
complex czero=cmplx(0.0,0.0),**ptk,*pu,*pw;
/* number of cdps after padding for fft */
nxfft = npfar(nx+(int)(0.5*ABS(offset/dx)));
/* get minimum time of first non-zero sample */
for (ix=0,itmin=nt; ix<nx; ++ix) {
for (it=0; it<itmin && ptx[ix][it]==0.0; ++it);
itmin = it;
}
/* if all zeros, simply return */
if (itmin>=nt) return;
/* make stretch and compress functions t(u) and u(t) */
maketu(offset,itmin,fmax,nt,dt,ft,vrms,&uoft,&nu,&du,&fu,&tofu,&tcon);
/* constants depending on gamma, offset, vrms[0], and nt */
g1 = 2.0*sqrt(gamma)/(1.0+gamma);
h = offset/2.0;
lnt = nt-1;
lt = (nt-1)*dt;
vmin = 0.5*MIN((1.0+1.0/gamma)*vrms[0],(1.0+gamma)*vrms[0]);
/* if gamma != 1, get bboh[] and itn[] for this offset */
if(gamma!=1.0)
getbbohitn (offset,itmin,nt,dt,vrms,ntable,boh,zoh, \
gamma,&bboh,&itn);
/* inverse of dmo stretch/squeeze factors */
os1 = 1.0/s1;
os2 = 1.0/s2;
/* maximum DMO shift (in samples) for any wavenumber k */
nupad = 1.5*((s1+s2)/2.0)*tcon/du;
/* frequency sampling */
nufft = npfa(nu+nupad);
nw = nufft;
dw = 2.0*PI/(nufft*du);
/* allocate workspace */
pu = pw = ealloc1complex(nufft);
/* wavenumber sampling and maximum wavenumber to apply dmo */
nk = nxfft/2+1;
dk = 2.0*PI/ABS(nxfft*dx);
kmax = PI/ABS(dx);
ikmax = NINT(kmax/dk);
/* pointers to complex p(t,k) */
ptk = (complex**)ealloc1(nk,sizeof(complex*));
for (ik=0; ik<nk; ++ik)
ptk[ik] = (complex*)ptx[0]+ik*nt;
/* fft scale factor */
fftscl = (float)nk/(float)(ikmax+1)/(nufft*nxfft);
/* Fourier transform p(t,x) to p(t,k) */
pfa2rc(-1,2,nt,nxfft,ptx[0],ptk[0]);
/* loop over wavenumbers less than maximum */
for (ik=0,k=0.0; ik<=ikmax; ++ik,k+=dk) {
/* apply time vriant linear phase shift */
hk=sign*h*k;
for (it=lnt,t=lt; it>=itmin; --it,t-=dt) {
/* calculate phase-shift=boh*h*k, h=offset/2 */
if(gamma != 1.0)
phase = bboh[it]*hk;
else
phase = 0.0;
/* phase shift p(t,k) */
cphase=cos(phase);
sphase=sin(phase);
fr = ptk[ik][it].r;
fi = ptk[ik][it].i;
ptk[ik][it].r = fr*cphase + fi*sphase;
ptk[ik][it].i = -fr*sphase + fi*cphase;
}
/* stretch p(t;k) to p(u) */
ints8c(nt,dt,ft,ptk[ik],czero,czero,nu,tofu,pu);
/* pad with zeros and Fourier transform p(u) to p(w) */
for (iu=nu; iu<nufft; ++iu)
pu[iu].r = pu[iu].i = 0.0;
pfacc(1,nufft,pu);
/* minimum and maximum frequencies to process */
wmin = ABS(0.5*vmin*k);
wmax = ABS(PI/du);
iwmin = MAX(1,MIN((nw-1)/2,NINT(wmin/dw)));
iwmax = MAX(0,MIN((nw-1)/2,NINT(wmax/dw)));
/* constant independent of w */
wwscl = os1*pow(g1*hk/tcon,2.0);
wwscl2 = wwscl*os2/os1;
/* zero dc (should be zero anyway) */
pw[0].r = pw[0].i = 0.0;
/* zero frequencies below minimum */
for (iw=1,iwn=nw-iw; iw<iwmin; ++iw,--iwn)
pw[iw].r = pw[iw].i = pw[iwn].r = pw[iwn].i = 0.0;
/* do dmo between minimum and maximum frequencies */
for (iw=iwmin,iwn=nw-iwmin,w=iwmin*dw;
iw<=iwmax; ++iw,--iwn,w+=dw) {
/* positive w */
scales = 1.0+wwscl/(w*w);
scale = sqrt(scales);
phase = s1*w*tcon*(scale-1.0);
amp = fftscl*(1.0-s1+s1/scale);
fr = amp*cos(phase);
fi = amp*sin(phase);
pwr = pw[iw].r;
pwi = pw[iw].i;
pw[iw].r = pwr*fr-pwi*fi;
pw[iw].i = pwr*fi+pwi*fr;
/* negative w */
scales = 1.0+wwscl2/(w*w);
scale = sqrt(scales);
phase = s2*w*tcon*(scale-1.0);
amp = fftscl*(1.0-s2+s2/scale);
fr = amp*cos(phase);
fi = amp*sin(phase);
pwr = pw[iwn].r;
pwi = pw[iwn].i;
pw[iwn].r = pwr*fr+pwi*fi;
pw[iwn].i = pwi*fr-pwr*fi;
}
/* zero frequencies above maximum to Nyquist (if present) */
for (iw=iwmax+1,iwn=nw-iw; iw<=nw/2; ++iw,--iwn)
pw[iw].r = pw[iw].i = pw[iwn].r = pw[iwn].i = 0.0;
/* Fourier transform p(w) to p(u) */
pfacc(-1,nufft,pu);
/* compress p(u) to p(t;k) */
ints8c(nu,du,fu,pu,czero,czero,nt,uoft,ptk[ik]);
}
/* zero wavenumber between maximum and Nyquist */
for (; ik<nk; ++ik)
for (it=0; it<nt; ++it)
ptk[ik][it].r = ptk[ik][it].i = 0.0;
/* Fourier transform p(t,k) to p(t,x) */
pfa2cr(1,2,nt,nxfft,ptk[0],ptx[0]);
/* free workspace */
free1float(tofu);
free1float(uoft);
free1complex(pu);
free1(ptk);
}
void makeref (float dsmax, int nr, float *ar,
int *nu, float **xu, float **zu, Reflector **r)
/*****************************************************************************
Make piecewise cubic reflectors
******************************************************************************
Input:
dsmax maximum length of reflector segment
nr number of reflectors
ar array[nr] of reflector amplitudes
nu array[nr] of numbers of (x,z) pairs; u = 0, 1, ..., nu[ir]
xu array[nr][nu[ir]] of reflector x coordinates x(u)
zu array[nr][nu[ir]] of reflector z coordinates z(u)
Output:
r array[nr] of reflectors
******************************************************************************
Notes:
Space for the ar, nu, xu, and zu arrays is freed by this function, since
they are only used to construct the reflectors.
This function is meant to be called only once, so it need not be very
efficient. Once made, the reflectors are likely to be used many times,
so the cost of making them is insignificant.
*****************************************************************************/
{
int ir,iu,nuu,iuu,ns,is;
float x,z,xlast,zlast,dx,dz,duu,uu,ds,fs,rsx,rsz,rsxd,rszd,
*u,*s,(*xud)[4],(*zud)[4],*us;
ReflectorSegment *rs;
Reflector *rr;
/* allocate space for reflectors */
*r = rr = ealloc1(nr,sizeof(Reflector));
/* loop over reflectors */
for (ir=0; ir<nr; ++ir) {
/* compute cubic spline coefficients for uniformly sampled u */
u = ealloc1float(nu[ir]);
for (iu=0; iu<nu[ir]; ++iu)
u[iu] = iu;
xud = (float(*)[4])ealloc1float(4*nu[ir]);
csplin(nu[ir],u,xu[ir],xud);
zud = (float(*)[4])ealloc1float(4*nu[ir]);
csplin(nu[ir],u,zu[ir],zud);
/* finely sample x(u) and z(u) and compute length s(u) */
nuu = 20*nu[ir];
duu = (u[nu[ir]-1]-u[0])/(nuu-1);
s = ealloc1float(nuu);
s[0] = 0.0;
xlast = xu[ir][0];
zlast = zu[ir][0];
for (iuu=1,uu=duu; iuu<nuu; ++iuu,uu+=duu) {
intcub(0,nu[ir],u,xud,1,&uu,&x);
intcub(0,nu[ir],u,zud,1,&uu,&z);
dx = x-xlast;
dz = z-zlast;
s[iuu] = s[iuu-1]+sqrt(dx*dx+dz*dz);
xlast = x;
zlast = z;
}
/* compute u(s) from s(u) */
ns = 1+s[nuu-1]/dsmax;
ds = s[nuu-1]/ns;
fs = 0.5*ds;
us = ealloc1float(ns);
yxtoxy(nuu,duu,0.0,s,ns,ds,fs,0.0,(float)(nu[ir]-1),us);
/* compute reflector segments uniformly sampled in s */
rs = ealloc1(ns,sizeof(ReflectorSegment));
for (is=0; is<ns; ++is) {
intcub(0,nu[ir],u,xud,1,&us[is],&rsx);
intcub(0,nu[ir],u,zud,1,&us[is],&rsz);
intcub(1,nu[ir],u,xud,1,&us[is],&rsxd);
intcub(1,nu[ir],u,zud,1,&us[is],&rszd);
rs[is].x = rsx;
rs[is].z = rsz;
rs[is].c = rsxd/sqrt(rsxd*rsxd+rszd*rszd);
rs[is].s = -rszd/sqrt(rsxd*rsxd+rszd*rszd);
}
/* fill in reflector structure */
rr[ir].ns = ns;
rr[ir].ds = ds;
rr[ir].a = ar[ir];
rr[ir].rs = rs;
/* free workspace */
free1float(us);
free1float(s);
free1float(u);
free1float((float*)xud);
free1float((float*)zud);
/* free space replaced by reflector segments */
free1(xu[ir]);
free1(zu[ir]);
}
/* free space replaced by reflector segments */
free1(nu);
free1(xu);
free1(zu);
}
int
main(int argc, char **argv)
{
int nline; /* number of (identical) lines */
int ntr; /* number of traces */
int nt; /* number of time samples */
int ninf; /* number of subsurface reflectors */
int itr; /* trace number index */
int iline; /* line number index */
int i, j; /* counters */
float dx; /* trace interval */
float dt; /* time interval */
float tdelay; /* recording time delay */
float x; /* zero offset position */
float xmin, xmax; /* min and max horizontal coordinates */
float *zint; /* reflector intercepts at x = 0 */
float *dip; /* reflector dips */
float *v; /* speeds in layers */
float *rho; /* densities in layers */
float *xl, *xr; /* min and max horiz coords in a layer */
float *meet; /* horizontal intercept for two adjacent
* layers, if layers not parallel */
float *trcoefs; /* transmission coefficients */
float *data; /* temp for holding synthetic trace */
float *tout; /* times[nt] for interpolation */
float **theta; /* angle at which the specular ray to
* interface j hits interface i */
float **m; /* slope for horiz travel calculation */
float **b; /* intercept for horiz travel calc */
float **d; /* distance traveled in a layer, if layers
* are parallel */
float **k; /* factor for finding horizontal distance
* in a layer */
float **w; /* for spreading factor calculation */
cwp_Bool *dorow; /* are adjacent reflectors parallel? */
/* hook up getpar */
initargs(argc, argv);
requestdoc(0);
/* get scalar parameters */
if (!getparint("nline",&nline)) nline=1;
if (!getparint("ntr",&ntr)) ntr=32; tr.ntr = ntr*nline;
if (!getparfloat("dx",&dx)) dx=10.0;
if (!getparint("nt",&nt)) nt=128;
CHECK_NT("nt",nt); tr.ns = nt;
if (!getparfloat("dt",&dt)) dt=0.004; tr.dt = dt*1000000;
if (!getparfloat("tdelay",&tdelay)) tdelay=0.0;
tr.delrt = tdelay*1000;
if (!getparint("ninf",&ninf)) ninf=4;
/* allocate space */
data = ealloc1float(nt);
v = ealloc1float(ninf+1);
rho = ealloc1float(ninf+1);
zint = ealloc1float(ninf+1);
dip = ealloc1float(ninf+1);
meet = ealloc1float(ninf+1);
xl = ealloc1float(ninf+1);
xr = ealloc1float(ninf+1);
trcoefs = ealloc1float(ninf+1);
theta = ealloc2float(ninf+1,ninf+1);
m = ealloc2float(ninf+1,ninf+1);
b = ealloc2float(ninf+1,ninf+1);
d = ealloc2float(ninf+1,ninf+1);
w = ealloc2float(ninf+1,ninf+1);
k = ealloc2float(ninf+1,ninf+1);
dorow = ealloc1(ninf+1,sizeof(cwp_Bool));
/* compute output times for interpolation */
tout = ealloc1float(nt);
for (i=0; i<nt; i++) tout[i]=i*dt;
/* getpar the reflector intercepts, dips, v's, and rho's */
zint[0] = 0.0; /* surface hard wired */
if (!getparfloat("zint", zint+1))
for (i = 1; i <= ninf; ++i) zint[i] = 100.0 * i;
dip[0] = 0.0; /* surface hard wired */
if (!getparfloat("dip", dip+1))
for (i = 1; i <= ninf; ++i) dip[i] = 5.0 * i;
if (!getparfloat("v", v))
for (i = 0; i <= ninf; ++i) v[i] = 1500.+ 500*i;
if (!getparfloat("rho", rho))
for (i = 0; i <= ninf; ++i) rho[i] = 1.0;
/* check dips and intercepts */
for (i = 1; i <= ninf; ++i) {
if (dip[i] < -90.0 || dip[i] > 90.0) err("dips1");
if (ABS(dip[i] - dip[i-1]) > 90.0) err("dips2");
dip[i] *= TORADS;
if (zint[i] <= zint[i-1]) err("intercept");
}
for (i = 0; i < ninf; ++i) {
if (v[i] <= 0.0) err("velocities");
if (rho[i] <= 0.0) err("density");
}
/* Table theta[j][i], the angle at which the specular to layer j
* leaves interface i. */
for (j = 1; j < ninf; ++j) {
theta[j][j-1] = -dip[j];
for (i = j-1; i > 0; --i) {
float temp;
temp = v[i-1]*sin(theta[j][i]+dip[i])/v[i];
if (ABS(temp) > 1.0) err("specular %d", j);
theta[j][i-1] = asin(temp) - dip[i];
}
}
/* Table m and b, which are used to find the x coordinate of
* the specular at the next interface from its x coordinate at
* the previous interface. */
for (j = 1; j <= ninf; ++j) {
for (i = 0; i < j; ++i) {
float s, temp;
s = sin(theta[j][i]);
temp = 1.0 - tan(dip[i+1])*s;
m[j][i] = (1.0 - tan(dip[i])*s)/temp;
b[j][i] = (zint[i+1]-zint[i])*s/temp;
}
}
/* Make the final checks on the substructure specification.
* The strategy is to check the x coordinates of the
* interface intercepts against x coordinates of
* specular-interface intercepts. Only the rays from the
* left and right endpoints need to be checked, since they define
* the area being "observed". */
xmin = 0.0;
xmax = dx * (ntr - 1);
for (j = 1; j <= ninf; ++j) {
xl[j] = xmin; /*** changed Brian's code here! */
xr[j] = xmax;
}
for (j=0; j<ninf; ++j) {
int j2;
for (j2=j+1; j2<=ninf; ++j2) {
if (dip[j2] != dip[j]) {
float intercept;
intercept = (zint[j2]-zint[j])/
(tan(dip[j])-tan(dip[j2]));
if (intercept > xmin && intercept < xmax) {
err("intercept");
}
}
}
for (j2=j+1; j<=ninf; ++j) {
xl[j2] = m[j2][j]*xl[j2] + b[j2][j];
xr[j2] = m[j2][j]*xr[j2] + b[j2][j];
if (xl[j2] < xmin) xmin = xl[j2];
if (xr[j2] > xmax) xmax = xr[j2];
}
}
/* Table the arrays meet and k if two adjacent interfaces
* intersect, otherwise d if they are parallel.
* Meet is the x coordinate of intersection if there is
* an intersection, k is a value which will be used to
* find the distance travelled in that layer. If there is no
* intersection d will be a constant for all j and can be
* stored now. */
for (i = 0; i < ninf; ++i) {
if (dip[i+1] == dip[i]) {
dorow[i] = cwp_false;
for (j = i + 1; j <= ninf; ++j) {
d[j][i] = (zint[i+1]-zint[i])*cos(dip[i]) /
cos(theta[j][i] + dip[i]);
}
} else {
dorow[i] = cwp_true;
meet[i] = (zint[i+1]-zint[i])/
(tan(dip[i])-tan(dip[i+1]));
for (j = i + 1; j <= ninf; ++j) {
k[j][i] = sin(dip[i]-dip[i+1]) /
(cos(theta[j][i]+dip[i+1])*cos(dip[i]));
}
}
}
/* Table t, the transmission coefficients. */
tabtrcoefs(ninf, rho, v, theta, dip, trcoefs);
/* Table w, the coefficients for the spreading factor calculation. */
for (j = 1; j <= ninf; ++j) {
w[j][0] = 1.0;
for (i = 1; i < j; ++i) {
float c1, c2;
c1 = cos(theta[j][i-1] + dip[i]);
c2 = cos(theta[j][i] + dip[i]);
w[j][i] = w[j][i-1]*c1*c1/(c2*c2);
}
}
/* Now ready to make synthetic traces. */
for (iline = 0; iline < nline; ++iline) {
x = 0.0;
for (itr = 0; itr < ntr; itr++) {
float ex, t0, time, tmp, p1, p2, dist, amp[1];
int itime, it;
memset( (void *) tr.data, 0, nt * FSIZE);
memset( (void *) data, 0, nt * FSIZE);
for (j = 1; j <= ninf; ++j) {
ex = x;
if (dorow[0]) {
dist = (meet[0]-ex)*k[j][0];
} else {
dist = d[j][0];
}
t0 = dist/v[0];
tmp = dist*v[0];
p1 = tmp;
p2 = tmp;
for (i = 1; i < j; ++i) {
ex = m[j][i-1]*ex + b[j][i-1];
if (dorow[i]) {
dist = (meet[i] - ex)*k[j][i];
} else {
dist = d[j][i];
}
t0 += dist/v[i];
tmp = dist*v[i];
p1 += tmp;
p2 += tmp*w[j][i];
}
/* set strength and time of spike response */
amp[0] = v[0]*trcoefs[j]/
(dt*FOURPI*2.0*sqrt(p1*p2));
time = 2.0*t0 - tdelay;
itime = NINT(time/dt);
if (itime >= 0 && itime < nt)
data[itime] = amp[0];
/* distribute spike response over full trace */
ints8r(1,dt,time,amp,0,0,nt,tout,data);
/* add distributed spike to segy trace */
for (it = 0; it < nt; ++it)
tr.data[it] += data[it];
}
/* fill tracl header and put trace out */
tr.tracl = itr + 1;
tr.gx = x;
tr.sx = x;
puttr(&tr);
/* set horizontal location of next trace */
x += dx;
}
}
/**
* Thats all folks.
**/
return(CWP_Exit());
}
/* parse one reflector specification; return 1 if valid, 0 otherwise */
int decodeReflector (char *string,
float *aPtr, int *nxzPtr, float **xPtr, float **zPtr)
/**************************************************************************
decodeReflector - parse one reflector specification
***************************************************************************
Input:
string string representing reflector
Output:
aPtr pointer to amplitude value
nxzPtr pointer to number of x,z pairs
xPtr array of x values for one reflector
zPtr array of z values for one reflector
***************************************************************************
Author: Dave Hale, Colorado School of Mines, 09/17/91
**************************************************************************/
{
int nxz,ixz;
float a,*x,*z;
char *s,*t;
/* if specified, get reflector amplitude; else, use default */
s = string;
if (strchr(s,':')==NULL) {
a = 1.0;
s = strtok(s,",;\0");
} else {
/* if (strcspn(s,":")>=strcspn(s,",;\0")) return 0;*/
if (((unsigned int) strcspn(s,":")) >=
((unsigned int) strcspn(s,",;\0"))) return 0;
a = atof(s=strtok(s,":"));
s = strtok(NULL,",;\0");
}
/* count x and z values, while splitting string into tokens */
for (t=s,nxz=0; t!=NULL; ++nxz)
t = strtok(NULL,",;\0");
/* total number of values must be even */
if (nxz%2) return 0;
/* number of (x,z) pairs */
nxz /= 2;
/* 2 or more (x,z) pairs are required */
if (nxz<2) return 0;
/* allocate space */
x = ealloc1(nxz,sizeof(float));
z = ealloc1(nxz,sizeof(float));
/* convert (x,z) values */
for (ixz=0; ixz<nxz; ++ixz) {
x[ixz] = atof(s);
s += strlen(s)+1;
z[ixz] = atof(s);
s += strlen(s)+1;
}
/* set output parameters before returning */
*aPtr = a;
*nxzPtr = nxz;
*xPtr = x;
*zPtr = z;
return 1;
}
void fputtr_internal(FILE *fp, segy *tp, cwp_Bool fixed_length)
{
unsigned int databytes; /* bytes from nsfirst */
int nwritten; /* bytes seen by fwrite calls */
/* search linked list for possible alternative */
if(fp != lastfp) searchlist(fp);
if (infoptr == ((struct outsegyinfo *) NULL)) {
/* initialize new segy output stream */
/* allocate new segy output information table */
*oldinfoptr = (struct outsegyinfo *)
malloc(sizeof(struct outsegyinfo));
infoptr = *oldinfoptr;
infoptr->nextinfo = (struct outsegyinfo *) NULL;
/* save FILE * ptr */
infoptr->outfp = fp;
infoptr->itr = 0;
/* allocate XDR struct and associate FILE * ptr */
infoptr->segy_xdr = (XDR *) malloc(sizeof(XDR));
switch (infoptr->ftype = filestat(fileno(fp))) {
case DIRECTORY:
err("%s: segy output can't be a directory", __FILE__);
case TTY:
err("%s: segy output can't be tty", __FILE__);
break;
default: /* the rest are ok */
break;
}
/* xdrstdio_create(infoptr->segy_xdr,fp,XDR_ENCODE);*/
infoptr->buf = ealloc1(sizeof(segy),sizeof(char));
xdrmem_create(infoptr->segy_xdr, infoptr->buf, sizeof(segy), XDR_ENCODE);
infoptr->bufstart = xdr_getpos(infoptr->segy_xdr);
/* Sanity check the segy header */
infoptr->nsfirst = tp->ns;
if (infoptr->nsfirst > SU_NFLTS)
err("%s: unable to handle %d > %d samples per trace",
__FILE__, infoptr->nsfirst, SU_NFLTS);
switch(tp->trid) {
case CHARPACK:
infoptr->bytesper = sizeof(char); break;
case SHORTPACK:
infoptr->bytesper = 2*sizeof(char); break;
default:
infoptr->bytesper = BYTES_PER_XDR_UNIT; break;
}
}
xdr_setpos(infoptr->segy_xdr, infoptr->bufstart);
if(FALSE == xdrhdrsub(infoptr->segy_xdr,tp))
err("%s: unable to write header on trace #%ld",
__FILE__, (infoptr->itr)+1);
nwritten = efwrite(infoptr->buf,1,HDRBYTES,infoptr->outfp);
if(nwritten != HDRBYTES)
err("%s: unable to write header on trace #%ld",
__FILE__, (infoptr->itr)+1);
databytes = infoptr->bytesper * (fixed_length?infoptr->nsfirst:tp->ns);
nwritten = datawrite(infoptr, tp, fixed_length);
if (nwritten != databytes)
err("%s: on trace #%ld, tried to write %d bytes, "
"wrote %d bytes",
__FILE__, (infoptr->itr)+1, databytes, nwritten);
++infoptr->itr;
lastfp = infoptr->outfp;
}
static unsigned char *makeABytes (int style,
float x1beg, float x1end, float x2beg, float x2end,
float fmin, float fmax, unsigned char bmin, unsigned char bmax,
int n1, float d1, float f1, int n2, float d2, float f2, float *floats,
int *nxa, int *nya)
/*****************************************************************************
Makes a frame of bytes from a frame of floats. This function
is responsible for making an initial (non-zoomed) frame of
bytes from a frame of floats. Floats outside the range
[fmin,fmax] are clipped in converting to bytes in the
corresponding range [bmin,bmax]. The output bytes are ordered
as nxa samples left-to-right, nya scanlines top-to-bottom, where
nxa and nya are computed according to the window specified
by x1beg, x1end, x2beg, and x2end. Only floats inside this
window are converted.
*****************************************************************************/
{
int i1,i2,ix,iy,i1beg,i1end,i1step,i2beg,i2end,i2step,nx,ny;
unsigned char *b,*bp;
float fbmin,fbmax,fscale=0.0,foffset,fi,*f=floats;
/* determine coefficients to convert from floats to bytes */
fbmin = bmin;
fbmax = bmax;
if(fbmax==0.0) fbmax=255.0;
/* if( fmax < fmin){float tmp=fmax; fmax=fmin; fmin=tmp;}*/
/* warn("fbmin=%f fbmax=%f fscale=%f",fbmin,fbmax,fscale); */
if(fbmax==0)fbmax=255;
fscale = (fmax!=fmin) ? (fbmax-fbmin)/(fmax-fmin) : 1.0e10;
foffset = fbmin-fmin*fscale;
/* warn("fbmin=%f fbmax=%f fscale=%f",fbmin,fbmax,fscale); */
/* adjust axes limits to nearest samples */
adjustAxesValues(n1,d1,f1,n2,d2,f2,&x1beg,&x1end,&x2beg,&x2end);
/* determine sample index limits and increments */
i1beg = NINT((x1beg-f1)/d1);
i1end = NINT((x1end-f1)/d1);
i1step = (i1end>i1beg) ? 1 : -1;
i2beg = NINT((x2beg-f2)/d2);
i2end = NINT((x2end-f2)/d2);
i2step = (i2end>i2beg) ? 1 : -1;
/* convert floats to bytes */
if (style==XtcwpNORMAL) {
nx = 1+ABS(i1end-i1beg);
ny = 1+ABS(i2end-i2beg);
b = ealloc1(nx*ny,sizeof(unsigned char));
for (iy=0,i2=i2beg; iy<ny; ++iy,i2+=i2step) {
bp = b+nx*ny-(iy+1)*nx;
for (ix=0,i1=i1beg; ix<nx; ++ix,i1+=i1step) {
fi = foffset+f[i1+i2*n1]*fscale;
if (fi<fbmin) fi = fbmin;
if (fi>fbmax) fi = fbmax;
*bp++ = (unsigned char)fi;
}
}
} else {
nx = 1+ABS(i2end-i2beg);
ny = 1+ABS(i1end-i1beg);
bp = b = ealloc1(nx*ny,sizeof(unsigned char));
for (iy=0,i1=i1beg; iy<ny; ++iy,i1+=i1step) {
for (ix=0,i2=i2beg; ix<nx; ++ix,i2+=i2step) {
fi = foffset+f[i1+i2*n1]*fscale;
if (fi<fbmin) fi = fbmin;
if (fi>fbmax) fi = fbmax;
*bp++ = (unsigned char)fi;
}
}
}
/* set output parameters and return */
*nxa = nx;
*nya = ny;
return b;
}
static XImage *makeImage (Display *dpy, int width, int height,
int nx, int ny, int interp, unsigned char *bytes)
/*****************************************************************************
Makes image from bytes ordered as left-to-right, top-to-bottom scanlines.
*****************************************************************************/
{
int widthpad;
unsigned char *pixels;
unsigned char *data;
Colormap wcmap;
XColor color;
int i,j,k;
int scr;
int depth;
unsigned long truecolor_pixel[256];
XImage *xim;
int byte_perpixel;
int line, iline,jline;
int ih,half;
Screen *scr1;
# define RGB_BLACK {0x00, 0x00, 0x00}
# define RGB_WHITE {0xff, 0xff, 0xff}
# define RGB_GRAY {0x80, 0x80, 0x80}
float c_rgb [3][3] =
{ RGB_BLACK, RGB_GRAY, RGB_WHITE };
half=128;
byte_perpixel=4;
scr1=XDefaultScreenOfDisplay(dpy);
xim=(XImage *) NULL;
scr=DefaultScreen(dpy);
wcmap=DefaultColormapOfScreen(scr1);
/* Build the 1st ramp */
for (ih = 0; ih < 128; ++ih) {
color.red = c_rgb[0][0] +
(c_rgb[1][0] - c_rgb[0][0]) * ((float) ih)/((float) half);
color.green = c_rgb[0][1] +
(c_rgb[1][1] - c_rgb[0][1]) * ((float) ih)/((float) half);
color.blue = c_rgb[0][2] +
(c_rgb[1][2] - c_rgb[0][2]) * ((float) ih)/((float) half);
color.red *= 257.0;
color.green *= 257.0;
color.blue *= 257.0;
color.flags = DoRed|DoGreen|DoBlue;
XAllocColor(dpy,wcmap,&color);
truecolor_pixel[ih]=(unsigned long )color.pixel;
}
/* Build the 2nd ramp */
for (ih=128; ih<256; ++ih) {
color.red = c_rgb[1][0] +
(c_rgb[2][0] - c_rgb[1][0]) * ((float) (ih-half))/((float) half);
color.green = c_rgb[1][1] +
(c_rgb[2][1] - c_rgb[1][1]) * ((float) (ih-half))/((float) half);
color.blue = c_rgb[1][2] +
(c_rgb[2][2] - c_rgb[1][2]) * ((float) (ih-half))/((float) half);
color.red *= 257.0;
color.green *= 257.0;
color.blue *= 257.0;
color.flags = DoRed|DoGreen|DoBlue;
XAllocColor(dpy,wcmap,&color);
truecolor_pixel[ih]=(unsigned long )color.pixel;
}
depth=(unsigned int)DefaultDepth(dpy,scr);
if(depth<=8) byte_perpixel=1;
else if(depth<=16) byte_perpixel=2;
/* determine scanline padding and allocate memory for pixels */
widthpad = (1+(width-1)/(BitmapPad(dpy)/8))*BitmapPad(dpy)/8;
pixels = ealloc1(widthpad*height,sizeof(unsigned char));
/* bilinearly interpolate bytes to pixels */
if (interp)
intl2b(nx,1.0,0.0,ny,1.0,0.0,bytes,
width,(float)(nx-1)/(float)(width-1),0.0,
height,(float)(ny-1)/(float)(height-1),0.0,pixels);
else
intn2b(nx,1.0,0.0,ny,1.0,0.0,bytes,
width,(float)(nx-1)/(float)(width-1),0.0,
height,(float)(ny-1)/(float)(height-1),0.0,pixels);
data = ealloc1(widthpad*height,byte_perpixel);
xim=XCreateImage( (Display *) dpy,
(Visual *) DefaultVisual(dpy,scr),
(unsigned int) DefaultDepth(dpy,scr),
(int) ZPixmap,
(int) 0,
(char *) data,
(unsigned int) widthpad,
(unsigned int) height,
(int) BitmapPad(dpy),
(int) widthpad*byte_perpixel);
byte_perpixel=xim->bits_per_pixel/8;
/*
fprintf(stderr,"\nbyte_perpixel = %d, depth= %d\n", byte_perpixel,depth);
*/
/* translate bytes to pixels, padding scanlines as necessary */
for (line=0; line<height; line++) {
iline = line*width;
jline = line*widthpad;
for (i=iline,j=jline,k=0; k<width; ++i,++j,++k)
{ if(byte_perpixel==1)
((unsigned char *)data)[j] =(unsigned char)pixels[i];
if(byte_perpixel==2)
{
int edn=xim->byte_order;
if(edn==LSBFirst){
((unsigned char *)data)[j*2+0] =(unsigned char)(truecolor_pixel[pixels[i]]);
((unsigned char *)data)[j*2+1] =(unsigned char)(truecolor_pixel[pixels[i]]>>8);
}else{
((unsigned char *)data)[j*2+0] =(unsigned char)(truecolor_pixel[pixels[i]]>>24);
((unsigned char *)data)[j*2+1] =(unsigned char)(truecolor_pixel[pixels[i]]>>16);
}
/*((unsigned short *)data)[j] =(unsigned short)(truecolor_pixel[pixels[i]]);*/
/*((unsigned short *)data)[j] =(unsigned short)(truecolor_pixel[pixels[i]]);*/
}
/* Initialize getpar */
static int getparinit (void)
{
static int targc; /* total number of args */
static char **targv; /* pointer to arg strings */
static char *pfname; /* name of parameter file */
FILE *pffd; /* file id of parameter file */
int pflen; /* length of parameter file in bytes */
static int pfargc; /* arg count from parameter file */
bool parfile; /* parfile existence flag */
int argstrlen;
char *argstr, *pargstr; /* storage for command line and
parameter file args */
int nread; /* bytes fread */
int i, j; /* counters */
char *getpfname(); /* return name of parameter file */
int white2null(); /* deliminate arg strings from parameter
file with (possibly multiple) NULLs
and return a count of the strings */
int tabulate(); /* install symbol table */
tabled = true; /* remember table is built */
/* Check if xargc was initiated */
if(!xargc)
err("%s: xargc=%d -- not initiated in main", __FILE__, xargc);
/* Space needed for command lines */
for (i = 1, argstrlen = 0; i < xargc; i++) {
argstrlen += strlen(xargv[i]) + 1;
}
/* Get parfile name if there is one */
/* parfile = (pfname = getpfname()) ? true : false; */
if (pfname = getpfname()) {
parfile = true;
} else {
parfile = false;
}
if (parfile) {
pffd = efopen(pfname, "r");
/* Get the length */
efseek(pffd, 0, SEEK_END);
pflen = eftell(pffd);
rewind(pffd);
argstrlen += pflen;
} else {
pflen = 0;
}
/* Allocate space for command line and parameter file
plus nulls at the ends to help with parsing. */
/* argstr = (char *) calloc((size_t) (1+argstrlen+1), 1); */
argstr = (char *) ealloc1(1+argstrlen+1, 1);
if (parfile) {
/* Read the parfile */
nread = efread(argstr + 1, 1, pflen, pffd);
if (nread != pflen) {
err("%s: fread only %d bytes out of %d from %s",
__FILE__, nread, pflen, pfname);
}
efclose(pffd);
/* Zap whites in parfile to help in parsing */
argstr[0] = '\0' ;
pfargc = white2null(argstr, pflen);
} else {
pfargc = 0;
}
/* Total arg count */
targc = pfargc + xargc - 1;
/* Allocate space for total arg pointers */
targv = (char **) ealloc1(targc, sizeof(char*));
if (parfile) {
/* Parse the parfile. Skip over multiple NULLs */
for (j = 1, i = 0; j < pflen; j++) {
if (argstr[j] && !argstr[j-1]) {
targv[i++] = argstr + j;
}
}
} else {
i = 0;
}
/* Copy command line arguments */
for (j = 1, pargstr = argstr + pflen + 2; j < xargc; j++) {
strcpy(pargstr,xargv[j]);
targv[i++] = pargstr;
pargstr += strlen(xargv[j]) + 1;
}
/* Allocate space for the pointer table */
argtbl = (ArgStruct*) ealloc1(targc, sizeof(ArgStruct));
/* Tabulate targv */
tabulate(targc, targv);
}
main(int argc, char **argv)
{
int nt; /* number of points on trace */
int databytes; /* ... in bytes */
int ntr; /* number of traces */
string stype; /* noise type (gauss, flat) as string */
int itype; /* ... as integer (for use in switch) */
float sn; /* signal to noise ratio */
unsigned int seed; /* random number seed */
FILE *hdrfp; /* fp for header storage file */
FILE *sigfp; /* fp for data ("signal") */
int nfloats; /* number of floats in "signal" */
float *noise; /* noise vector */
float noiscale; /* scale for noise */
float absmaxsig; /* absolute maximum in signal */
float noipow; /* a measure of noise power */
float f1; /* left lower corner frequency */
float f2; /* left upper corner frequency */
float f4; /* right lower corner frequency */
float f3; /* right upper corner frequency */
char * scrdir; /* scratch dir to put temporary data set */
/* Initialize */
initargs(argc, argv);
askdoc(1);
/* Get noise type */
if (!getparstring("noise", &stype)) stype = "gauss";
if (STREQ(stype, "gauss")) itype = GAUSS;
else if (STREQ(stype, "flat")) itype = FLAT;
else err("noise=\"%s\", must be gauss or flat", stype);
/* Get signal to noise ratio */
if (!getparfloat("sn", &sn)) sn = SN;
if (sn <= 0) err("sn=%d must be positive", sn);
/* Set seed */
if (!getparuint("seed", &seed)) { /* if not supplied, use clock */
if (-1 == (seed = (uint) time((time_t *) NULL))) {
err("time() failed to set seed");
}
}
(itype == GAUSS) ? srannor(seed) : sranuni(seed);
if( !getparstring("scrdir",&scrdir) ) {
scrdir = getenv("SU_SCRATCHDIR") ;
}
/* Prepare temporary files to hold headers and data */
/*
hdrfp = etmpfile();
sigfp = etmpfile();
*/
hdrfp = etempfile(NULL);
sigfp = etempfile(NULL);
/* Get info from first trace */
if (!gettr(&tr)) err("can't get first trace");
nt = tr.ns;
databytes = nt * FSIZE;
/* Loop over input traces & write headers and data to tmp files */
ntr = 0;
do {
++ntr;
efwrite(&tr, 1, HDRBYTES, hdrfp);
efwrite(tr.data, 1, databytes, sigfp);
} while (gettr(&tr));
nfloats = ntr * nt;
/* Compute absmax of signal over entire data set */
rewind(sigfp);
absmaxsig = 0.0;
{ register int i;
for (i = 0; i < nfloats; ++i) {
float sigval;
efread(&sigval, FSIZE, 1, sigfp);
absmaxsig = MAX(absmaxsig, ABS(sigval));
}
}
/* Compute noise vector elements in [-1, 1] */
noise = ealloc1float(nfloats);
switch (itype) {
register int i;
case GAUSS: /* frannor gives elements in N(0,1)--ie. pos & negs */
for (i = 0; i < nfloats; ++i) noise[i] = frannor();
break;
case FLAT: /* franuni gives elements in [0, 1] */
for (i = 0; i < nfloats; ++i) noise[i] = 2.0*franuni() - 1.0;
break;
default: /* defensive programming */
err("%d: mysterious itype = %d", __LINE__, itype);
}
/* Band limit noise traces if user getpars any of the f's */
if (getparfloat("f1", &f1) || getparfloat("f2", &f2) ||
getparfloat("f3", &f3) || getparfloat("f4", &f4) ) {
/* Set up call to suband */
char cmdbuf[BUFSIZ]; /* build suband command */
FILE *bandinfp; /* fp for input file */
FILE *fp; /* fp for pipe to suband */
int nsegy = HDRBYTES + databytes;
char *segybuf = ealloc1(nsegy, 1);
/* Trap signals so can remove tmpnam file */
signal(SIGINT, (void *) trapsig);
signal(SIGQUIT, (void *) trapsig);
signal(SIGHUP, (void *) trapsig);
signal(SIGTERM, (void *) trapsig);
/* Prepare temporary files to hold traces */
/*bandinfp = etmpfile();*/
bandinfp = etempfile(NULL);
/*bandoutfp = efopen(etmpnam(bandoutfile), "w+");*/
bandoutfile = etempnam(scrdir,NULL) ;
bandoutfp = efopen(bandoutfile, "w+");
/* Paste headers on noise traces and put in tmpfile */
rewind(hdrfp);
{ register int itr;
for (itr = 0; itr < ntr; ++itr) {
efread(&tr, 1, HDRBYTES, hdrfp);
memcpy(tr.data, noise + itr*nt, databytes);
fputtr(bandinfp, &tr);
}
}
/* Pipe to suband - suband handles the getpars */
sprintf(cmdbuf, "suband >%s", bandoutfile);
fp = epopen(cmdbuf, "w");
rewind (bandinfp);
{ register int itr;
for (itr = 0; itr < ntr; ++itr) {
efread(segybuf, 1, nsegy, bandinfp);
efwrite(segybuf, 1, nsegy, fp);
}
}
efclose(bandinfp);
epclose(fp);
/* Load bandlimited traces back into noise vector */
rewind(bandoutfp);
{ register int itr;
for (itr = 0; itr < ntr; ++itr) {
fgettr(bandoutfp, &tr);
memcpy(noise + itr*nt, tr.data, databytes);
}
}
efclose(bandoutfp);
eremove(bandoutfile);
} /* End optional bandlimiting */
/* Compute noise power */
noipow = 0.0;
{ register int i;
for (i = 0; i < nfloats; ++i) {
register float noiseval = noise[i];
noipow += noiseval * noiseval;
}
}
/* Compute noise scale for desired noise/signal ratio */
absmaxsig /= sqrt(2.0); /* make it look like a rmsq value */
noipow /= nfloats; /* make it the square of rmsq value */
noiscale = absmaxsig / (sn * sqrt(noipow));
/* Add scaled noise to trace and output sum */
rewind(hdrfp);
rewind(sigfp);
{ register int itr;
for (itr = 0; itr < ntr; ++itr) {
register int trshift = itr*nt;
register int i;
efread(&tr, 1, HDRBYTES, hdrfp);
efread(tr.data, 1, databytes, sigfp);
for (i = 0; i < nt; ++i)
tr.data[i] += noiscale * noise[trshift + i];
puttr(&tr);
}
}
return EXIT_SUCCESS;
}
int
main (int argc, char **argv)
{
int n1,n2,n1tic,n2tic,nfloats,bbox[4],
i1,i2,grid1,grid2,style,
n1c,n2c,n1s,n2s,i1beg,i1end,i2beg,i2end,i1c,i2c,
nz,iz,i1step,i2step,verbose,hls,bps,
legend,ugrid=SOLID,lstyle=VERTLEFT,lz,lbegsup=0,lendsup=0,ln=256,
lbbox[4], threecolor=0; /* BEREND, Schoenfelder */
int lnice; /* c liner */
float labelsize,titlesize,perc,clip,bperc,wperc,bclip,wclip,
d1,f1,d2,f2,*z,*temp,zscale,zoffset,zi,
xbox,ybox,width,height,
x1beg,x1end,x2beg,x2end,
x1min,x1max,x2min,x2max,
d1num,f1num,d2num,f2num,
p1beg,p1end,p2beg,p2end,matrix[6],colors[3][3], /* for 3 color mode */
d1s,d2s,
lwidth,lheight,lx,ly,lbeg,lend,lmin=(float) FLT_MAX,lmax=(float) -FLT_MAX,
ldnum,lfnum,ld,lf=0,labmatrix[6]; /* BEREND, Schoenfelder */
float axeswidth, ticwidth, gridwidth;
unsigned char *cz,*czp,*sz,*data_legend=NULL;
char *label1="",*label2="",*title="",*units="",
*legendfont="times_roman10",
*labelfont="Helvetica",*titlefont="Helvetica-Bold",
*styles="seismic",*grid1s="none",*grid2s="none",
*titlecolor="black",*axescolor="black",*gridcolor="black",
*lstyles="vertleft",*lgrids="none";
FILE *infp=stdin;
float **x1curve=NULL,**x2curve=NULL,*curvewidth=NULL;
int i,j,curve=0,*npair=NULL,ncurvecolor=0,ncurvewidth=0,ncurvedash=0,*curvedash=NULL;
char **curvecolor=NULL,**curvefile=NULL;
FILE *curvefp=NULL;
cwp_Bool is_curve = cwp_false;
/* initialize getpar */
initargs(argc,argv);
requestdoc(1);
/* get parameters describing 1st dimension sampling */
if (!getparint("n1",&n1)) err("must specify n1!\n");
d1 = 1.0; getparfloat("d1",&d1);
f1 = 0.0; getparfloat("f1",&f1);
x1min = (d1>0.0)?f1:f1+(n1-1)*d1;
x1max = (d1<0.0)?f1:f1+(n1-1)*d1;
/* get parameters describing 2nd dimension sampling */
if (!getparint("n2",&n2)) {
if (efseeko(infp,(off_t) 0,SEEK_END)!=0)
err("must specify n2 if in a pipe!\n");
nfloats = (int) (eftello(infp)/((off_t) sizeof(float)));
efseeko(infp,(off_t) 0,SEEK_SET);
n2 = nfloats/n1;
}
d2 = 1.0; getparfloat("d2",&d2);
f2 = 0.0; getparfloat("f2",&f2);
x2min = (d2>0.0)?f2:f2+(n2-1)*d2;
x2max = (d2<0.0)?f2:f2+(n2-1)*d2;
/* read color parameters */
if (!getparint("threecolor",&threecolor)) threecolor=1;
bps = 8;
hls = 0;
/* color[][0] is black, color[][2] is white in 2 color mode */
colors[R][0] = colors[G][0] = colors[B][0] = 0.0;
colors[R][1] = colors[G][1] = colors[B][1] = 0.5;
colors[R][2] = colors[G][2] = colors[B][2] = 1.0;
if (countparval("brgb") || countparval("wrgb")) {
float brgb[3],grgb[3],wrgb[3];
brgb[R] = brgb[G] = brgb[B] = 0.0;
wrgb[R] = wrgb[G] = wrgb[B] = 1.0;
getparfloat("brgb",&brgb[0]);
getparfloat("wrgb",&wrgb[0]);
grgb[R] = (brgb[R] + wrgb[R])/2.;
grgb[G] = (brgb[G] + wrgb[G])/2.;
grgb[B] = (brgb[B] + wrgb[B])/2.;
if (threecolor==1)
getparfloat("grgb",&grgb[0]);
brgb[R] = MAX(0.0,MIN(1.0,brgb[R]));
grgb[R] = MAX(0.0,MIN(1.0,grgb[R]));
wrgb[R] = MAX(0.0,MIN(1.0,wrgb[R]));
brgb[G] = MAX(0.0,MIN(1.0,brgb[G]));
grgb[G] = MAX(0.0,MIN(1.0,grgb[G]));
wrgb[G] = MAX(0.0,MIN(1.0,wrgb[G]));
brgb[B] = MAX(0.0,MIN(1.0,brgb[B]));
grgb[B] = MAX(0.0,MIN(1.0,grgb[B]));
wrgb[B] = MAX(0.0,MIN(1.0,wrgb[B]));
colors[R][0] = brgb[R]; colors[R][1] = grgb[R]; colors[R][2] = wrgb[R];
colors[G][0] = brgb[G]; colors[G][1] = grgb[G]; colors[G][2] = wrgb[G];
colors[B][0] = brgb[B]; colors[B][1] = grgb[B]; colors[B][2] = wrgb[B];
if (!getparint("bps",&bps)) bps = 12;
if (bps!=12 && bps!=24)
err("bps must equal 12 or 24 for color plots!\n");
} else if (countparval("bhls") || countparval("whls")) {
float bhls[3],ghls[3],whls[3];
hls = 1;
bhls[H] = ghls[H] = whls[H] = 0.0;
bhls[L] = 0.0; ghls[L] = 0.5; whls[L] = 1.0;
bhls[S] = ghls[S] = whls[S] = 0.0;
getparfloat("bhls",&bhls[0]);
getparfloat("whls",&whls[0]);
ghls[H] = (bhls[H] + whls[H])/2.;
ghls[L] = (bhls[L] + whls[L])/2.;
ghls[S] = (bhls[S] + whls[S])/2.;
if (threecolor==1)
getparfloat("ghls",&ghls[0]);
bhls[L] = MAX(0.0,MIN(1.0,bhls[L]));
ghls[L] = MAX(0.0,MIN(1.0,ghls[L]));
whls[L] = MAX(0.0,MIN(1.0,whls[L]));
bhls[S] = MAX(0.0,MIN(1.0,bhls[S]));
ghls[S] = MAX(0.0,MIN(1.0,ghls[S]));
whls[S] = MAX(0.0,MIN(1.0,whls[S]));
colors[H][0] = bhls[0]; colors[H][1] = ghls[0]; colors[H][2] = whls[0];
colors[L][0] = bhls[1]; colors[L][1] = ghls[1]; colors[L][2] = whls[1];
colors[S][0] = bhls[2]; colors[S][1] = ghls[2]; colors[S][2] = whls[2];
if (!getparint("bps",&bps)) bps = 12;
if (bps!=12 && bps!=24)
err("bps must equal 12 or 24 for color plots!\n");
}
/* get legend specs BEREND, Schoenfelder */
legend = 0; getparint("legend", &legend); /* BEREND, Schoenfelder */
getparstring("units", &units); /* BEREND, Schoenfelder */
getparstring("legendfont", &legendfont); /* BEREND, Schoenfelder */
/* set up curve plotting */
if ((curve=countparval("curve"))!=0) {
curvefile=(char**)ealloc1(curve,sizeof(void*));
getparstringarray("curve",curvefile);
if ((x1curve=(float**)malloc(curve*sizeof(void*)))==NULL)
err("Could not allocate x1curve pointers\n");
if ((x2curve=(float**)malloc(curve*sizeof(void*)))==NULL)
err("Could not allocate x2curve pointers\n");
npair=ealloc1int(curve);
getparint("npair",npair);
is_curve = cwp_true;
} else {
npair=(int *)NULL;
curvefile=(char **)NULL;
x1curve=(float **)NULL;
x2curve=(float **)NULL;
is_curve = cwp_false;
}
if (is_curve) {
if ((ncurvecolor=countparval("curvecolor"))<curve) {
curvecolor=(char**)ealloc1(curve,sizeof(void*));
if (!getparstringarray("curvecolor",curvecolor)) {
curvecolor[0]=(char *)cwp_strdup("black\0");
ncurvecolor=1;
}
for (i=ncurvecolor; i<curve; i++)
curvecolor[i]=(char *)cwp_strdup(curvecolor[ncurvecolor-1]);
} else if (ncurvecolor) {
curvecolor=(char**)ealloc1(ncurvecolor,sizeof(void*));
getparstringarray("curvecolor",curvecolor);
}
for (j=0; j<curve; j++) {
curvefp=efopen(curvefile[j],"r");
x1curve[j]=ealloc1float(npair[j]);
x2curve[j]=ealloc1float(npair[j]);
for (i=0; i<npair[j]; i++) {
fscanf(curvefp,"%f",&x1curve[j][i]);
fscanf(curvefp,"%f",&x2curve[j][i]);
}
efclose(curvefp);
}
}
/* read binary data to be plotted */
nz = n1*n2;
z = ealloc1float(nz);
if (fread(z,sizeof(float),nz,infp)!=nz)
err("error reading input file!\n");
/* if necessary, determine clips from percentiles */
if (getparfloat("clip",&clip)) {
bclip = clip;
wclip = -clip;
}
if ((!getparfloat("bclip",&bclip) || !getparfloat("wclip",&wclip)) &&
!getparfloat("clip",&clip)) {
perc = 100.0; getparfloat("perc",&perc);
temp = ealloc1float(nz);
for (iz=0; iz<nz; iz++)
temp[iz] = z[iz];
if (!getparfloat("bclip",&bclip)) {
bperc = perc; getparfloat("bperc",&bperc);
iz = (nz*bperc/100.0);
if (iz<0) iz = 0;
if (iz>nz-1) iz = nz-1;
qkfind(iz,nz,temp);
bclip = temp[iz];
}
if (!getparfloat("wclip",&wclip)) {
wperc = 100.0-perc; getparfloat("wperc",&wperc);
iz = (nz*wperc/100.0);
if (iz<0) iz = 0;
if (iz>nz-1) iz = nz-1;
qkfind(iz,nz,temp);
wclip = temp[iz];
}
free1float(temp);
}
verbose = 1; getparint("verbose",&verbose);
if (verbose) warn("bclip=%g wclip=%g",bclip,wclip);
/* get scaled sampling intervals */
d1s = 1.0; getparfloat("d1s",&d1s);
d2s = 1.0; getparfloat("d2s",&d2s);
d1s = fabs(d1s); d1s *= d1;
d2s = fabs(d2s); d2s *= d2;
/* get axes parameters */
xbox = 1.5; getparfloat("xbox",&xbox); /* if psimage is called by ximage, it */
ybox = 1.5; getparfloat("ybox",&ybox); /* will xbox=1.166 and ybox=1.167 */
width = 6.0; getparfloat("wbox",&width); getparfloat("width",&width);
height = 8.0;getparfloat("hbox",&height);getparfloat("height",&height);
/* begin c liner */
lnice = 0; getparint("lnice",&lnice);
if (lnice==1) {
ybox = 2.2;
/* lx=8 is set below, after getpar on lx ... c liner */
width = 5.4;
height = 7.2;
}
/* end c liner */
x1beg = x1min; getparfloat("x1beg",&x1beg);
x1end = x1max; getparfloat("x1end",&x1end);
d1num = 0.0; getparfloat("d1num",&d1num);
f1num = x1min; getparfloat("f1num",&f1num);
n1tic = 1; getparint("n1tic",&n1tic);
getparstring("grid1",&grid1s);
if (STREQ("dot",grid1s))
grid1 = DOT;
else if (STREQ("dash",grid1s))
grid1 = DASH;
else if (STREQ("solid",grid1s))
grid1 = SOLID;
else
grid1 = NONE;
getparstring("label1",&label1);
x2beg = x2min; getparfloat("x2beg",&x2beg);
x2end = x2max; getparfloat("x2end",&x2end);
d2num = 0.0; getparfloat("d2num",&d2num);
f2num = 0.0; getparfloat("f2num",&f2num);
n2tic = 1; getparint("n2tic",&n2tic);
getparstring("grid2",&grid2s);
if (STREQ("dot",grid2s))
grid2 = DOT;
else if (STREQ("dash",grid2s))
grid2 = DASH;
else if (STREQ("solid",grid2s))
grid2 = SOLID;
else
grid2 = NONE;
getparstring("label2",&label2);
getparstring("labelfont",&labelfont);
labelsize = 18.0; getparfloat("labelsize",&labelsize);
getparstring("title",&title);
getparstring("titlefont",&titlefont);
titlesize = 24.0; getparfloat("titlesize",&titlesize);
getparstring("titlecolor",&titlecolor);
getparstring("axescolor",&axescolor);
getparstring("gridcolor",&gridcolor);
/* axes and tic width */
if(!getparfloat("axeswidth",&axeswidth)) axeswidth=1;
if (!getparfloat("ticwidth",&ticwidth)) ticwidth=axeswidth;
if(!getparfloat("gridwidth",&gridwidth)) gridwidth =axeswidth;
if (is_curve) {
if ((ncurvewidth=countparval("curvewidth"))<curve) {
curvewidth=ealloc1float(curve);
if (!getparfloat("curvewidth",curvewidth)) {
curvewidth[0]=axeswidth;
ncurvewidth=1;
}
for (i=ncurvewidth; i<curve; i++)
curvewidth[i]=curvewidth[ncurvewidth-1];
} else {
curvewidth=ealloc1float(ncurvewidth);
getparfloat("curvewidth",curvewidth);
}
if ((ncurvedash=countparval("curvedash"))<curve) {
curvedash=ealloc1int(curve);
if (!getparint("curvedash",curvedash)) {
curvedash[0]=0;
ncurvedash=1;
}
for (i=ncurvedash; i<curve; i++)
curvedash[i]=curvedash[ncurvedash-1];
} else {
curvedash=ealloc1int(ncurvedash);
getparint("curvedash",curvedash);
}
}
getparstring("style",&styles);
if (STREQ("normal",styles))
style = NORMAL;
else
style = SEISMIC;
/* Get or calc legend parameters */
/* Legend min and max: Calc from data read in */
if (legend) {
for (lz=0;lz<nz;lz++) {
lmin=FMIN(lmin,z[lz]);
lmax=FMAX(lmax,z[lz]);
}
if (verbose==2) warn("lmin=%g lmax=%g",lmin,lmax);
}
if (legend) {
lbeg = lmin; if (getparfloat("lbeg",&lbeg)) lbegsup=1;
lend = lmax; if (getparfloat("lend",&lend)) lendsup=1;
/* Change wclip,bclip to be inside legend range */
wclip = FMAX(lbeg,wclip); /* [wclip,bclip] has to be in [lbeg,lend] */
bclip = FMIN(lend,bclip);
if (lbegsup!=1) { /* Add white and black areas to show possible clipping */
float rangeperc=(bclip-wclip)/20.;
lbeg=wclip-rangeperc;
}
if (lendsup!=1) {
float rangeperc=(bclip-wclip)/20.;
lend=bclip+rangeperc;
}
lfnum = lmin; getparfloat("lfnum",&lfnum);
getparstring("lstyle",&lstyles);
if (STREQ("vertright",lstyles))
lstyle = VERTRIGHT;
else if (STREQ("horibottom",lstyles))
lstyle = HORIBOTTOM;
/* legend dimensions (BEREND), Schoenfelder */
lwidth = 0.1 ;lheight = height/2;
if (lstyle==HORIBOTTOM) {
lwidth=width/1.2 ;lheight = 0.24;
}
getparfloat("lwidth",&lwidth);
getparfloat("lheight",&lheight);
lx=.8;ly = ybox+(height-lheight)/2;
if (lstyle==VERTRIGHT) {
lx=xbox+width+0.1;
} else if (lstyle==HORIBOTTOM) {
lx=xbox+(width-lwidth)/2.0;ly = 1.0;
}
getparfloat("lx",&lx);
if (lnice==1) lx = 8; /* c liner */
getparfloat("ly",&ly);
getparstring("lgrid",&lgrids);
if (STREQ("dot",lgrids))
ugrid = DOT;
else if (STREQ("dash",lgrids))
ugrid = DASH;
else if (STREQ("solid",lgrids))
ugrid = SOLID;
else
ugrid = NONE;
}
/* adjust x1beg and x1end to fall on sampled values */
/* This will not allow to display an area greater than the data supplied */
i1beg = NINT((x1beg-f1)/d1);
i1beg = MAX(0,MIN(n1-1,i1beg));
x1beg = f1+i1beg*d1;
i1end = NINT((x1end-f1)/d1);
i1end = MAX(0,MIN(n1-1,i1end));
x1end = f1+i1end*d1;
/* adjust x2beg and x2end to fall on sampled values */
i2beg = NINT((x2beg-f2)/d2);
i2beg = MAX(0,MIN(n2-1,i2beg));
x2beg = f2+i2beg*d2;
i2end = NINT((x2end-f2)/d2);
i2end = MAX(0,MIN(n2-1,i2end));
x2end = f2+i2end*d2;
if (legend) {
/* Make legend color values */
int lll=0,lcount,perc5=13,ilbeg,ilend; /* color scale */
if (lbegsup!=1) {
ln+=perc5; /* white area */
}
if (lendsup!=1) {
ln+=perc5; /* black area */
}
data_legend = ealloc1(ln,sizeof(char));
if (lbegsup!=1) {
for (lll=0;lll<perc5;lll++) data_legend[lll]=(char) 255; /* white area */
}
for (lcount=255;lcount>=0;lcount--,lll++) data_legend[lll]=(char) lcount;
if (lendsup!=1) {
for (;lll<ln;lll++) data_legend[lll]=(char) 0; /* black area */
}
lf=lbeg;ld=(lend-lbeg)/(ln-1);
if (!(getparfloat("ldnum",&ldnum))) ldnum=0.0;
/* adjust lbeg and lend to fall on sampled values */
ilbeg = NINT((lbeg-lf)/ld);
ilbeg = MAX(0,MIN(ln-1,ilbeg));
lbeg = lf+ilbeg*ld;
ilend = NINT((lend-lf)/ld);
ilend = MAX(0,MIN(ln-1,ilend));
lend = lf+ilend*ld;
}
/* allocate space for image bytes */
n1c = 1+abs(i1end-i1beg);
n2c = 1+abs(i2end-i2beg);
cz = ealloc1(n1c*n2c,sizeof(char));
/* convert data to be imaged into unsigned characters */
zscale = (wclip!=bclip)?255.0/(wclip-bclip):1.0e10;
zoffset = -bclip*zscale;
i1step = (i1end>i1beg)?1:-1;
i2step = (i2end>i2beg)?1:-1;
czp = cz;
for (i1c=0,i1=i1beg; i1c<n1c; i1c++,i1+=i1step) {
for (i2c=0,i2=i2beg; i2c<n2c; i2c++,i2+=i2step) {
zi = zoffset+z[i1+i2*n1]*zscale;
if (zi<0.0) zi = 0.0;
if (zi>255.0) zi = 255.0;
*czp++ = (unsigned char)zi;
}
}
free1float(z);
/* determine sampling after scaling */
n1s = MAX(1,NINT(1+(n1c-1)*d1/d1s));
d1s = (n1s>1)?d1*(n1c-1)/(n1s-1):d1;
n2s = MAX(1,NINT(1+(n2c-1)*d2/d2s));
d2s = (n2s>1)?d2*(n2c-1)/(n2s-1):d2;
/* if necessary, interpolate to scaled sampling intervals */
if (n1s!=n1c || n2s!=n2c) {
sz = ealloc1(n1s*n2s,sizeof(char));
intl2b(n2c,d2,0.0,n1c,d1,0.0,cz,n2s,d2s,0.0,n1s,d1s,0.0,sz); /* Interpol array */
free1(cz);
} else {
sz = cz;
}
/* determine axes pads */
p1beg = (x1end>x1beg)?-fabs(d1s)/2:fabs(d1s)/2;
p1end = (x1end>x1beg)?fabs(d1s)/2:-fabs(d1s)/2;
p2beg = (x2end>x2beg)?-fabs(d2s)/2:fabs(d2s)/2;
p2end = (x2end>x2beg)?fabs(d2s)/2:-fabs(d2s)/2;
/* convert axes box parameters from inches to points */
xbox *= 72.0;
ybox *= 72.0;
width *= 72.0;
height *= 72.0;
if (legend) {
lx *= 72.0; /* Schoenfelder */
ly *= 72.0; /* Schoenfelder */
lwidth *= 72.0; /* Schoenfelder */
lheight *= 72.0; /* Schoenfelder */
}
/* set bounding box */
psAxesBBox(
xbox,ybox,width,height,
labelfont,labelsize,
titlefont,titlesize,
style,bbox);
if (legend) {
psLegendBBox( /* Space for legend Schoenfelder */
lx,ly,lwidth,lheight,
labelfont,labelsize,
lstyle,lbbox);
/* Include space for legend Schoenfelder */
bbox[0]=MIN(bbox[0],lbbox[0]);
bbox[1]=MIN(bbox[1],lbbox[1]);
bbox[2]=MAX(bbox[2],lbbox[2]);
bbox[3]=MAX(bbox[3],lbbox[3]);
}
boundingbox(bbox[0],bbox[1],bbox[2],bbox[3]);
/* begin PostScript */
begineps();
/* save graphics state */
gsave();
/* translate coordinate system by box offset */
translate(xbox,ybox);
/* determine image matrix */
if (style==NORMAL) {
matrix[0] = 0; matrix[1] = n1s; matrix[2] = n2s;
matrix[3] = 0; matrix[4] = 0; matrix[5] = 0;
} else {
matrix[0] = n2s; matrix[1] = 0; matrix[2] = 0;
matrix[3] = -n1s; matrix[4] = 0; matrix[5] = n1s;
}
scale(width,height);
/* draw the image (before axes so grid lines are visible) */
drawimage(hls,colors,n2s,n1s,bps,matrix,sz);
/***************************/
/* main image has been drawn, restore graphics state */
grestore();
/* *********************************/
/* draw the colorbar (before axes so grid lines are visible) Schoenfelder*/
if (legend) {
gsave();
translate(lx,ly);
scale(lwidth,lheight);
if ((lstyle==VERTLEFT) || (lstyle==VERTRIGHT)) {
labmatrix[0] = 1; labmatrix[1] = 0; labmatrix[2] = 0;
labmatrix[3] = ln; labmatrix[4] = 0; labmatrix[5] = 0;
drawimage(hls,colors,1,ln,bps,labmatrix,data_legend);
} else {
labmatrix[0] = -1; labmatrix[1] = 0; labmatrix[2] = 0;
labmatrix[3] = ln; labmatrix[4] = 0; labmatrix[5] = 0;
rotate(-90);
drawimage(hls,colors,1,ln,bps,labmatrix,data_legend);
rotate(90);
}
grestore();
}
/* draw curve */
for (i=0; i<curve; i++) {
gsave();
psDrawCurve(
xbox,ybox,width,height,
x1beg,x1end,p1beg,p1end,
x2beg,x2end,p2beg,p2end,
x1curve[i],x2curve[i],npair[i],
curvecolor[i],curvewidth[i],curvedash[i],style);
grestore();
}
gsave();
/* draw axes and title */
psAxesBox(
xbox,ybox,width,height,
x1beg,x1end,p1beg,p1end,
d1num,f1num,n1tic,grid1,label1,
x2beg,x2end,p2beg,p2end,
d2num,f2num,n2tic,grid2,label2,
labelfont,labelsize,
title,titlefont,titlesize,
titlecolor,axescolor,gridcolor,
ticwidth,axeswidth,gridwidth,
style);
/* restore graphics state */
grestore();
/* draw axes and title for legend Schoenfelder*/
if (legend) {
float lpbeg,lpend;
int lntic=1;
gsave();
lpbeg = 0.0; /*(lend>lbeg)?-fabs(d1s)/2:fabs(d1s)/2;*/
lpend = 0.0; /*(lend>lbeg)?fabs(d1s)/2:-fabs(d1s)/2;*/
psLegendBox(
lx,ly,lwidth,lheight,
lbeg,lend,lpbeg,lpend,
ldnum,lf,lntic,ugrid,units,
labelfont,labelsize,
axescolor,gridcolor,
lstyle);
grestore();
}
/* end PostScript */
showpage();
endeps();
if (curve) {
free1int(npair);
for (i=0; i<curve; i++) {
free1float(x1curve[i]);
free1float(x2curve[i]);
}
free1float(curvewidth);
free1int(curvedash);
free((void**)x1curve);
free((void**)x2curve);
free((void**)curvefile);
free((void**)curvecolor);
}
return 0;
}
int
main(int argc ,char **argv)
{
char buf[1024];
char *ptr;
int i;
int j;
int k;
int total;
int t1;
int t2;
int t3;
int t4;
int t5;
float r;
float v[128];
float x;
float dx;
float min_x;
float max_x;
float rmin = 0.0;
float rmax = 0.0;
float y;
float dy;
float min_y;
float max_y;
int ix;
int iy;
int ir = -1;
int nx = 100;
int ny = 100;
float** bin;
int* xcnt;
float sigma;
/* intermediates for calculating standard deviation */
float* a;
float* b;
float sum;
float m;
int logx = 0;
int logy = 0;
/*--------------------------------------------------------------------*\
Get the commandline arguents in SU style
\*--------------------------------------------------------------------*/
initargs( argc, argv );
requestdoc( 1 );
if( !getparint( "ix" ,&ix ) ){
err( "Missing X column index, ix=" );
}
if( !getparint( "iy" ,&iy ) ){
err( "Missing Y column index, ix=" );
}
getparint( "ir" ,&ir );
if( ir > 0
&& !getparfloat( "rmax" ,&rmax )
&& !getparfloat( "rmin" ,&rmin ) ){
err( "Failed to specify rejection threshold" );
}
if( !getparfloat( "min_x" ,&min_x ) ){
err( "min_x= not specified" );
}
if( !getparfloat( "max_x" ,&max_x ) ){
err( "max_x= not specified" );
}
if( !getparfloat( "min_y" ,&min_y ) ){
err( "min_y= not specified" );
}
if( !getparfloat( "max_y" ,&max_y ) ){
err( "max_y= not specified" );
}
getparint( "nx" ,&nx );
getparint( "ny" ,&ny );
getparint( "logx" ,&logx );
getparint( "logy" ,&logy );
if( logx ){
if( min_x < 0 || max_x < 0 ){
err( "Cannot use logscale for X" );
}else{
min_x = log(min_x);
max_x = log(max_x);
}
}
if( logy ){
if( min_y < 0 || max_y < 0 ){
err( "Cannot use logscale for Y" );
}else{
min_y = log(min_y);
max_y = log(max_y);
}
}
dx = (max_x - min_x) / nx;
dy = (max_y - min_y) / ny;
/*--------------------------------------------------------------------*\
Allocate and clear the space for the work arrays.
\*--------------------------------------------------------------------*/
bin = (float**) ealloc2( ny ,nx ,sizeof(float) );
xcnt = (int*) ealloc1( nx ,sizeof(int) );
a = (float*) ealloc1( nx ,sizeof(float) );
b = (float*) ealloc1( nx ,sizeof(float) );
for( i=0; i<nx; i++ ){
memset( bin[i] ,0 ,ny*sizeof(float) );
}
memset( xcnt ,0 ,nx*sizeof(int) );
memset( a ,0 ,nx*sizeof(float) );
memset( b ,0 ,nx*sizeof(float) );
total = 0;
/*--------------------------------------------------------------------*\
Read the input data as space delimited ASCII text
\*--------------------------------------------------------------------*/
while (!feof(stdin) && fgets( buf ,sizeof(buf) ,stdin ) ){
/* tokenize and read input */
ptr = buf;
k = 0;
while( (ptr=strtok( ptr ," \t\n" )) ){
sscanf( ptr ,"%f" ,&v[k] );
k++;
ptr = 0;
}
x = v[ix-1];
y = v[iy-1];
if( logx && x > 0 ){
x = log(x);
}
if( logy && y > 0 ){
y = log(y);
}
i = (x - min_x ) /dx;
j = (y - min_y ) /dy;
if( ir > 0 ){
/* 3rd value threshold options */
r = v[ir-1];
if( rmin != 0.0
&& r > rmin
&& i > 0 && i < nx
&& j > 0 && j < ny ){
/* lower threshold */
bin[i][j]++;
xcnt[i]++;
a[i] += y*y;
b[i] += y;
total++;
}else if( rmax != 0.0
&& r < rmax
&& i > 0 && i < nx
&& j > 0 && j < ny ){
/* upper threshold */
bin[i][j]++;
xcnt[i]++;
a[i] += y*y;
b[i] += y;
total++;
}
}else{
/* no threshold value */
if( i > 0 && i < nx
&& j > 0 && j < ny ){
bin[i][j]++;
xcnt[i]++;
a[i] += y*y;
b[i] += y;
total++;
}
}
}
for( i=0; i<nx; i++ ){
if( xcnt[i] > 0.0001*total ){
sum = 0.0;
if( logx ){
printf( "%f " ,exp(min_x + i*dx) );
}else{
printf( "%f " ,min_x + i*dx );
}
printf( "%d " ,xcnt[i] );
t1 = 0;
t2 = 0;
t3 = 0;
t4 = 0;
t5 = 0;
for( j=0; j<ny; j++ ){
sum += bin[i][j];
m = (100.0*sum ) / xcnt[i];
if( !t1 && m >= 2.28 ){
t1 = 1;
if( logy ){
printf( "%f " ,exp(min_y + j*dy) );
}else{
printf( "%f " ,min_y + j*dy );
}
}else if( !t2 && m >= 15.87 ){
t2 = 1;
if( logy ){
printf( "%f " ,exp(min_y + j*dy) );
}else{
printf( "%f " ,min_y + j*dy );
}
}else if( !t3 && m >= 50.00 ){
t3 = 1;
if( logy ){
printf( "%f " ,exp(min_y + j*dy) );
}else{
printf( "%f " ,min_y + j*dy );
}
}else if( !t4 && m >= 84.13 ){
t4 = 1;
if( logy ){
printf( "%f " ,exp(min_y + j*dy) );
}else{
printf( "%f " ,min_y + j*dy );
}
}else if( !t5 && m >= 97.72 ){
t5 = 1;
if( logy ){
printf( "%f " ,exp(min_y + j*dy) );
}else{
printf( "%f " ,min_y + j*dy );
}
}
}
sigma = sqrt( (xcnt[i]*a[i] - b[i]*b[i])/(xcnt[i]*xcnt[i]-1) );
if( logy && sigma > 0 ){
printf( "%f " ,log(sigma) );
}else{
printf( "%f " ,sigma );
}
printf( "\n" );
}
}
return EXIT_SUCCESS;
}
