void freqfilt4pi_spec (const float* x /* input */, float** y /* spectrum */)
/*< compute 2-D spectrum >*/
{
int ik, iw;
for (ik=0; ik < m2; ik++) {
for (iw=0; iw < m1; iw++) {
trace[iw] = x[ik*m1+iw];
}
for (iw=m1; iw < nfft; iw++) {
trace[iw]=0.;
}
kiss_fftr (tfor,trace,ctrace);
for (iw=0; iw < nw; iw++) {
fft[ik][iw] = ik%2? sf_cneg(ctrace[iw]): ctrace[iw];
}
}
for (iw=0; iw < nw; iw++) {
kiss_fft_stride(xfor,fft[0]+iw,ctrace2,nw);
for (ik=0; ik < m2; ik++) {
y[iw][ik] = sf_cabsf(ctrace2[ik]); /* transpose */
}
}
}
kiss_fft_cpx sf_cpowf(kiss_fft_cpx a, kiss_fft_cpx b)
/*< complex power >*/
{
float i, r, rho, theta;
kiss_fft_cpx c;
r = sf_cabsf(a);
i = sf_cargf (a);
if (r == 0.0) {
c.r = 0.0;
c.i = 0.0;
} else {
theta = i * b.r;
if (b.i == 0.0) {
rho = powf (r,b.r);
} else {
r = logf(r);
theta += r * b.i;
rho = expf(r * b.r - i * b.i);
}
c.r = rho * cosf (theta);
c.i = rho * sinf (theta);
}
return c;
}
void sf_csharpinv(sf_coperator oper /* inverted operator */,
float scale /* extra operator scaling */,
int niter /* number of outer iterations */,
int ncycle /* number of iterations */,
float perc /* sharpening percentage */,
bool verb /* verbosity flag */,
int nq, int np /* model and data size */,
sf_complex *qq /* model */,
sf_complex *pp /* data */,
bool twhole /* thresholding flag */)
/*< sharp inversion for complex-valued operators >*/
{
int iter, i, i1, ti;
sf_complex *q0, *p0, *p1, *tp=NULL;
float qdif0=0., pdif0=0., qdif, pdif, pi;
if (!twhole) {
sf_sharpen_init(np,perc);
} else {
sf_sharpen_init(nq,perc);
}
q0 = sf_complexalloc(nq);
p0 = sf_complexalloc(np);
p1 = sf_complexalloc(np);
for (i1=0; i1 < np; i1++) {
p0[i1] = pp[i1];
}
for (iter=0; iter < niter; iter++) { /* outer iteration */
oper(true,false,nq,np,qq,p0);
for (i1=0; i1 < nq; i1++) {
q0[i1] = qq[i1];
}
for (i1=0; i1 < np; i1++) {
p1[i1] = p0[i1];
}
for (i=0; i < ncycle; i++) { /* inner iteration */
oper(false,false,nq,np,qq,p1);
for (i1=0; i1 < np; i1++) {
#ifdef SF_HAS_COMPLEX_H
p1[i1] *= (-scale);
#else
p1[i1] = sf_crmul(pp[i1],-scale);
#endif
}
oper(true,true,nq,np,qq,p1);
for (i1=0; i1 < nq; i1++) {
#ifdef SF_HAS_COMPLEX_H
qq[i1] += q0[i1];
#else
qq[i1] = sf_cadd(qq[i1],q0[i1]);
#endif
}
if (!twhole) {
tp = sf_complexalloc(np);
for (ti=0; ti < (nq/np); ti++) {
for (i1=0; i1 < np; i1++) {
tp[i1] = qq[ti*np+i1];
}
sf_csharpen(tp);
sf_cweight_apply(np,tp);
for (i1=0; i1 < np; i1++) {
qq[ti*np+i1] = tp[i1];
}
}
} else {
sf_csharpen(qq);
sf_cweight_apply(nq,qq);
}
if (verb) {
qdif = 0.;
for (i1=0; i1 < nq; i1++) {
qdif += cabsf(qq[i1]);
}
if (0==i) {
qdif0 = qdif;
qdif=1.;
} else {
qdif /= qdif0;
}
sf_warning("inner iteration %d mnorm: %f",i,qdif);
}
} /* inner iteration */
oper(false,false,nq,np,qq,p1);
for (i1=0; i1 < np; i1++) {
#ifdef SF_HAS_COMPLEX_H
p1[i1] *= (-scale);
#else
p1[i1] = sf_crmul(pp[i1],-scale);
#endif
}
for (i1=0; i1 < np; i1++) {
#ifdef SF_HAS_COMPLEX_H
p0[i1] += pp[i1] + p1[i1];
#else
p0[i1] = sf_cadd(p0[i1],sf_cadd(pp[i1],p1[i1]));
#endif
}
if (verb) {
pdif = 0.;
for (i1=0; i1 < np; i1++) {
#ifdef SF_HAS_COMPLEX_H
pi = cabs(pp[i1]+p1[i1]);
#else
pi = sf_cabsf(sf_cadd(pp[i1],p1[i1]));
#endif
pdif += pi*pi;
}
if (0==iter) {
pdif0 = pdif;
pdif=1.;
} else {
pdif /= pdif0;
}
sf_warning("outer iteration %d dres: %f",iter,pdif);
}
} /* outer iteration */
free(q0);
free(p0);
free(p1);
if (!twhole) free(tp);
sf_sharpen_close();
}
int main(int argc, char* argv[])
{
int nt, nx, it, ix, niter, iter, ntfft, nxfft,np, ip, ikt, ikx, iktn, ikxn, ifsnr; /* iktn, ikxn, iNyquist*/
float dt, dx, pmin, pmax, dp, p, cmax, scalar, sembpmax, num, den;
float *sembp, *mask, *gy, *fden, *fshift, *SNR;
float **fdata, **taup, **odata, **tdata, **odatat, **semb; /* tdata is the true data */
kiss_fft_cpx **cdata, **cdatat;
char *type;
sf_file inp, outp, m, spec1, spec2, trued, snr;
sf_init(argc,argv);
inp=sf_input("in");
m=sf_input("mask");
outp=sf_output("out");
if(!sf_histint(inp,"n1",&nt)) sf_warning("No n1 in input");
if(!sf_histint(inp,"n2",&nx)) sf_warning("No n2 in input");
if(!sf_histfloat(inp,"d1",&dt)) sf_warning("No n1 in input");
if(!sf_histfloat(inp,"d2",&dx)) sf_warning("No n2 in input");
ntfft = 2*kiss_fft_next_fast_size((nt+1)/2);
nxfft = 2*kiss_fft_next_fast_size((nx+1)/2);
scalar = 1./(ntfft*nxfft);
iktn=ntfft/2; ikxn=nxfft/2;
float dkt = 1.0/(ntfft*dt), fkt = 0.0,kt;
float dkx = 1.0/(nxfft*dx), fkx = 0.0,kx;
if (NULL == (type=sf_getstring("type"))) type="amplitude";
/* [amplitude, semblance] thresholding type, the default is amplitude thresholding */
if(!sf_getint("niter",&niter)) niter = 10;
/* Get the number of iterations */
if(!sf_getint("ifsnr",&ifsnr)) ifsnr = 0;
/* If compute SNR during iteration */
if(type[0]=='s')
{
if(!sf_getfloat("pmin",&pmin)) pmin=-2;
/* minimum p */
if(!sf_getfloat("pmax",&pmin)) pmax=2;
/* maximum p */
if(!sf_getint("np",&np)) np=nx;
/* number of p */
dp=(pmax-pmin)/(np-1);
sembp =sf_floatalloc(np);
semb =sf_floatalloc2(nt,np);
taup =sf_floatalloc2(nt,np);
}
/* output files */
if (NULL!=sf_getstring("spec2"))
{
spec2=sf_output("spec2");
sf_putint(spec2, "n1", ntfft);
sf_putint(spec2, "n2", nxfft);
}
if (NULL!=sf_getstring("spec1"))
{
spec1=sf_output("spec1");
sf_putint(spec1, "n1", ntfft);
sf_putint(spec1, "n2", nxfft);
}
if (ifsnr==1 && (NULL!=sf_getstring("true")))
{
snr=sf_output("snr");
trued=sf_input("true");
tdata=sf_floatalloc2(nt,nx);
SNR=sf_floatalloc(niter);
sf_floatread(tdata[0],nt*nx,trued);
sf_putint(snr,"n1",niter);
sf_putint(snr,"d1",1);
sf_putint(snr,"n2",1);
}
/* Allocate memory */
cdata =(kiss_fft_cpx**) sf_complexalloc2(ntfft,nxfft);
cdatat =(kiss_fft_cpx**) sf_complexalloc2(ntfft,nxfft); /* temporary file */
fshift= sf_floatalloc(ntfft);
fden = sf_floatalloc(ntfft);
gy = sf_floatalloc(nxfft);
mask =sf_floatalloc(nx);
odata =sf_floatalloc2(nt,nx);
odatat =sf_floatalloc2(ntfft,nxfft);
fdata =sf_floatalloc2(ntfft,nxfft);
memset(&odata[0][0],0,ntfft*nxfft*sizeof(float));
/* Read data */
sf_floatread(odata[0],nt*nx,inp);
sf_floatread(mask,nx,m);
if(type[0]=='s')
{
slant(dt,nt,nx,gy,pmin,dp,np,odata,taup,semb,sembp,&sembpmax,fshift,fden);
}
for (iter=niter-1; iter>=0; iter--) {
tfft(odata, cdatat, nx, ntfft);
xfft(cdatat, cdata, ntfft, nxfft);
cmax = findmax(nxfft,ntfft,cdata);
if (iter==0 || iter==niter-1) { // beginning and ending spectra
for (ix=0; ix<nxfft; ix++)
for (it=0; it<ntfft; it++)
fdata[ix][it] = sf_cabsf(cdata[ix][it]);
if (iter==0 && (NULL!=sf_getstring("spec2"))) sf_floatwrite(fdata[0],ntfft*nxfft,spec2);
if (iter==niter-1 && (NULL!=sf_getstring("spec1"))) sf_floatwrite(fdata[0],ntfft*nxfft,spec1);
}
if(type[0]=='a')
{
for (ix=0; ix<nxfft; ix++) // Abma Kabir FT amplitude thresholding
for (it=0; it<ntfft; it++)
if (sf_cabsf(cdata[ix][it])<iter*1./niter*cmax) cdata[ix][it] = cmplx(0.,0.);
}
else
{
for (ix=0; ix<nxfft; ix++) // Abma Kabir FT amplitude thresholding
for (it=0; it<ntfft; it++)
if (sf_cabsf(cdata[ix][it])<iter*1./niter*cmax) cdata[ix][it] = cmplx(0.,0.);
for (ikx=0,kx=fkx; ikx<=ikxn; ++ikx,kx+=dkx) {
for (ikt=0,kt=fkt; ikt<=iktn; ++ikt,kt+=dkt) {
if (kx==0) {
if (sf_cabsf(cdata[ikx][ikt])<iter*1./niter*cmax) cdata[ikx][ikt] = cmplx(0.,0.);
continue;
}
p = -kx/kt; ip = round((p-pmin)/dp);
//if (ip<0 || ip>=np) { cdata[ikx][ikt] = 0.;continue; }
if (ip<0 || ip>=np) { }
else if (sembp[ip] <iter*1./niter*sembpmax) cdata[ikx][ikt] = cmplx(0.,0.);
if (ikx>0 && ikx<(nxfft+1)/2) { // kx>=0, kx<0
p = kx/kt; ip = round((p-pmin)/dp);
//if (ip<0 || ip>=np) { cdata[nxfft-ikx][ikt] = 0.;continue; }
if (ip<0 || ip>=np) { }
else if (sembp[ip] <iter*1./niter*sembpmax) cdata[nxfft-ikx][ikt] = cmplx(0.,0.);
}
if (ikt>0 && ikt<(ntfft+1)/2) { // kt<0, kx>=0
p = kx/kt; ip = round((p-pmin)/dp);
//if (ip<0 || ip>=np) { cdata[ikx][ntfft-ikt] = 0.;continue; }
if (ip<0 || ip>=np) { }
else if (sembp[ip] <iter*1./niter*sembpmax) cdata[ikx][ntfft-ikt] = cmplx(0.,0.);
}
if (ikx>0 && ikx<(nxfft+1)/2 && ikt>0 && ikt<(ntfft+1)/2) { // kt<0, kx<0
p = -kx/kt; ip = round((p-pmin)/dp);
//if (ip<0 || ip>=np) { cdata[nxfft-ikx][ntfft-ikt] = 0.;continue; }
if (ip<0 || ip>=np) { }
else if (sembp[ip] <iter*1./niter*sembpmax) cdata[nxfft-ikx][ntfft-ikt] =cmplx(0.,0.);
}
}}
}
ixfft(cdata, cdatat, ntfft, nxfft);
itfft(cdatat, odatat, nxfft, ntfft);
for (ix=0; ix<nx; ix++) { // put in ORIGINAL KNOWN data
if (mask[ix]==1) continue;
for (it=0; it<nt; it++) odata[ix][it]=odatat[ix][it];
}
num=0;den=0;
/* If output the SNR file. */
if (ifsnr==1 && (NULL!=sf_getstring("true")))
{
for(ix=0;ix<nx;ix++)
for(it=0;it<nt;it++)
{
num+=tdata[ix][it]*tdata[ix][it];
den+=(tdata[ix][it]-odata[ix][it])*(tdata[ix][it]-odata[ix][it]);
}
SNR[niter-iter-1]=10*logf(num/den);
}
}
if (ifsnr==1 && (NULL!=sf_getstring("true")))
{ sf_floatwrite(SNR,niter,snr); }
sf_floatwrite(odata[0],nt*nx,outp);
exit (0);
}
int main (int argc, char* argv[])
{
bool opt, comp, phase, verb;
double n1;
float nw1, b, a, dw, pi;
int nt, nw, i, j, N, m1, M, k, f;
float *TDEx,*TDEy,*TDHx,*TDHy,*outp,*ExHyres,*EyHxres,*ExHypha,*EyHxpha;
float *bb, *pp, *qq, *dd;
sf_file in, out, Ey, Hx, Hy;
kiss_fft_cpx *FDEx=NULL,*FDEy=NULL,*FDHx=NULL,*FDHy=NULL;
kiss_fft_cpx *A=NULL,*B=NULL,*ExAs1=NULL,*HyBs1=NULL,*ExBs1=NULL;
kiss_fft_cpx *HyAs1=NULL,*HxAs1=NULL,*HxBs1=NULL;
kiss_fft_cpx *EyAs1=NULL,*EyBs1=NULL,*ExAs2=NULL;
kiss_fft_cpx *HyBs2=NULL,*ExBs2=NULL,*HyAs2=NULL,*HxAs2=NULL;
kiss_fft_cpx *HxBs2=NULL,*EyAs2=NULL,*EyBs2=NULL,*ExAs3=NULL;
kiss_fft_cpx *HyBs3=NULL,*ExBs3=NULL,*HyAs3=NULL,*HxAs3=NULL;
kiss_fft_cpx *HxBs3=NULL,*EyAs3=NULL,*EyBs3=NULL,*ExAs4=NULL;
kiss_fft_cpx *HyBs4=NULL,*ExBs4=NULL,*HyAs4=NULL,*HxAs4=NULL;
kiss_fft_cpx *HxBs4=NULL,*EyAs4=NULL,*EyBs4=NULL,*ExA1=NULL;
kiss_fft_cpx *HyB1=NULL,*ExB1=NULL,*HyA1=NULL,*HxA1=NULL,*HxB1=NULL;
kiss_fft_cpx *EyA1=NULL,*EyB1=NULL,*ExA2=NULL,*HyB2=NULL,*ExB2=NULL;
kiss_fft_cpx *HyA2=NULL,*HxA2=NULL,*HxB2=NULL,*EyA2=NULL,*EyB2=NULL;
kiss_fft_cpx *ExA3=NULL,*HyB3=NULL,*ExB3=NULL,*HyA3=NULL,*HxA3=NULL;
kiss_fft_cpx *HxB3=NULL,*EyA3=NULL,*EyB3=NULL,*ExA4=NULL,*HyB4=NULL;
kiss_fft_cpx *ExB4=NULL,*HyA4=NULL,*HxA4=NULL,*HxB4=NULL,*EyA4=NULL;
kiss_fft_cpx *EyB4=NULL,*Zxys1=NULL,*Zyxs1=NULL,*Zxys2=NULL,*Zyxs2=NULL;
kiss_fft_cpx *Zxys3=NULL,*Zyxs3=NULL,*Zxys4=NULL,*Zyxs4=NULL;
kiss_fft_cpx *Zxy=NULL,*Zyx=NULL;
kiss_fftr_cfg cfg;
sf_init (argc, argv);
in= sf_input("in");
Ey=sf_input("Ey");
Hx=sf_input("Hx");
Hy=sf_input("Hy");
out = sf_output("out");
if (!sf_getbool("opt",&opt)) opt=true;
/* if y, determine optimal size for efficiency */
if (!sf_getbool("comp",&comp)) comp=true;
/* component selection */
if (!sf_getbool("verb",&verb)) verb = false;
/* verbosity flag */
if (!sf_getbool("phase",&phase)) phase=false;
/* if y, calculate apparent resistivity, otherwise calculate phase */
if (!sf_histdouble(in,"n1",&n1)) sf_error("No n1= in input");
bb= sf_floatalloc(n1);
pp= sf_floatalloc(n1);
qq= sf_floatalloc(n1);
dd= sf_floatalloc(n1);
sf_floatread(bb,n1,in);
sf_floatread(pp,n1,Ey);
sf_floatread(qq,n1,Hx);
sf_floatread(dd,n1,Hy);
M=49;
outp =sf_floatalloc(M);
pi=3.14;
f=1;
b=-3;
for(k=0;k<M;k++) {
a=pow(10,b);
nw=(int)(128000./(a*1000));
nt = opt? 2*kiss_fft_next_fast_size((nw+1)/2): nw;
if (nt%2) nt++;
nw1 = nt/2+1;
dw = 128./nt;
m1=n1;
if(m1%nw) N=m1/nw+1;
else N=m1/nw;
if(verb) sf_warning("slice %d of %d, freq=%f, stack=%d;",k+1,M,a,N);
FDEx= (kiss_fft_cpx*)sf_complexalloc(nt);
FDEy= (kiss_fft_cpx*)sf_complexalloc(nt);
FDHx= (kiss_fft_cpx*)sf_complexalloc(nt);
FDHy= (kiss_fft_cpx*)sf_complexalloc(nt);
TDEx= sf_floatalloc(nt);
TDEy= sf_floatalloc(nt);
TDHx= sf_floatalloc(nt);
TDHy= sf_floatalloc(nt);
sf_putint(out,"n1",M);
sf_putfloat(out,"d1",0.1);
sf_putfloat(out,"o1",-3);
ExAs1=(kiss_fft_cpx*)sf_complexalloc(nt);
HyBs1=(kiss_fft_cpx*)sf_complexalloc(nt);
ExBs1=(kiss_fft_cpx*)sf_complexalloc(nt);
HyAs1=(kiss_fft_cpx*)sf_complexalloc(nt);
HxAs1=(kiss_fft_cpx*)sf_complexalloc(nt);
HxBs1=(kiss_fft_cpx*)sf_complexalloc(nt);
EyAs1=(kiss_fft_cpx*)sf_complexalloc(nt);
EyBs1=(kiss_fft_cpx*)sf_complexalloc(nt);
ExAs2=(kiss_fft_cpx*)sf_complexalloc(nt);
HyBs2=(kiss_fft_cpx*)sf_complexalloc(nt);
ExBs2=(kiss_fft_cpx*)sf_complexalloc(nt);
HyAs2=(kiss_fft_cpx*)sf_complexalloc(nt);
HxAs2=(kiss_fft_cpx*)sf_complexalloc(nt);
HxBs2=(kiss_fft_cpx*)sf_complexalloc(nt);
EyAs2=(kiss_fft_cpx*)sf_complexalloc(nt);
EyBs2=(kiss_fft_cpx*)sf_complexalloc(nt);
ExAs3=(kiss_fft_cpx*)sf_complexalloc(nt);
HyBs3=(kiss_fft_cpx*)sf_complexalloc(nt);
ExBs3=(kiss_fft_cpx*)sf_complexalloc(nt);
HyAs3=(kiss_fft_cpx*)sf_complexalloc(nt);
HxAs3=(kiss_fft_cpx*)sf_complexalloc(nt);
HxBs3=(kiss_fft_cpx*)sf_complexalloc(nt);
EyAs3=(kiss_fft_cpx*)sf_complexalloc(nt);
EyBs3=(kiss_fft_cpx*)sf_complexalloc(nt);
ExAs4=(kiss_fft_cpx*)sf_complexalloc(nt);
HyBs4=(kiss_fft_cpx*)sf_complexalloc(nt);
ExBs4=(kiss_fft_cpx*)sf_complexalloc(nt);
HyAs4=(kiss_fft_cpx*)sf_complexalloc(nt);
HxAs4=(kiss_fft_cpx*)sf_complexalloc(nt);
HxBs4=(kiss_fft_cpx*)sf_complexalloc(nt);
EyAs4=(kiss_fft_cpx*)sf_complexalloc(nt);
EyBs4=(kiss_fft_cpx*)sf_complexalloc(nt);
ExA1=(kiss_fft_cpx*)sf_complexalloc(nt);
HyB1=(kiss_fft_cpx*)sf_complexalloc(nt);
ExB1=(kiss_fft_cpx*)sf_complexalloc(nt);
HyA1=(kiss_fft_cpx*)sf_complexalloc(nt);
HxA1=(kiss_fft_cpx*)sf_complexalloc(nt);
HxB1=(kiss_fft_cpx*)sf_complexalloc(nt);
EyA1=(kiss_fft_cpx*)sf_complexalloc(nt);
EyB1=(kiss_fft_cpx*)sf_complexalloc(nt);
ExA2=(kiss_fft_cpx*)sf_complexalloc(nt);
HyB2=(kiss_fft_cpx*)sf_complexalloc(nt);
ExB2=(kiss_fft_cpx*)sf_complexalloc(nt);
HyA2=(kiss_fft_cpx*)sf_complexalloc(nt);
HxA2=(kiss_fft_cpx*)sf_complexalloc(nt);
HxB2=(kiss_fft_cpx*)sf_complexalloc(nt);
EyA2=(kiss_fft_cpx*)sf_complexalloc(nt);
EyB2=(kiss_fft_cpx*)sf_complexalloc(nt);
ExA3=(kiss_fft_cpx*)sf_complexalloc(nt);
HyB3=(kiss_fft_cpx*)sf_complexalloc(nt);
ExB3=(kiss_fft_cpx*)sf_complexalloc(nt);
HyA3=(kiss_fft_cpx*)sf_complexalloc(nt);
HxA3=(kiss_fft_cpx*)sf_complexalloc(nt);
HxB3=(kiss_fft_cpx*)sf_complexalloc(nt);
EyA3=(kiss_fft_cpx*)sf_complexalloc(nt);
EyB3=(kiss_fft_cpx*)sf_complexalloc(nt);
ExA4=(kiss_fft_cpx*)sf_complexalloc(nt);
HyB4=(kiss_fft_cpx*)sf_complexalloc(nt);
ExB4=(kiss_fft_cpx*)sf_complexalloc(nt);
HyA4=(kiss_fft_cpx*)sf_complexalloc(nt);
HxA4=(kiss_fft_cpx*)sf_complexalloc(nt);
HxB4=(kiss_fft_cpx*)sf_complexalloc(nt);
EyA4=(kiss_fft_cpx*)sf_complexalloc(nt);
EyB4=(kiss_fft_cpx*)sf_complexalloc(nt);
Zxys1=(kiss_fft_cpx*)sf_complexalloc(nt);
Zyxs1=(kiss_fft_cpx*)sf_complexalloc(nt);
Zxys2=(kiss_fft_cpx*)sf_complexalloc(nt);
Zyxs2=(kiss_fft_cpx*)sf_complexalloc(nt);
Zxys3=(kiss_fft_cpx*)sf_complexalloc(nt);
Zyxs3=(kiss_fft_cpx*)sf_complexalloc(nt);
Zxys4=(kiss_fft_cpx*)sf_complexalloc(nt);
Zyxs4=(kiss_fft_cpx*)sf_complexalloc(nt);
Zxy=(kiss_fft_cpx*)sf_complexalloc(nt);
Zyx=(kiss_fft_cpx*)sf_complexalloc(nt);
ExHyres=sf_floatalloc(nt);
EyHxres=sf_floatalloc(nt);
ExHypha=sf_floatalloc(nt);
EyHxpha=sf_floatalloc(nt);
cfg = kiss_fftr_alloc(nt,0,NULL,NULL);
for(j=0;j<N;j++) {
for(i=0;i<nt;i++){
if((i<nw)&&((j*nw+i)<n1)) {
TDEx[i]=bb[j*nw+i]*0.5*(1-cos(2*pi*i/nt));
}
else {
TDEx[i]=0.;
}
}
kiss_fftr(cfg,TDEx,FDEx);
for(i=0;i<nt;i++){
if((i<nw)&&((j*nw+i)<n1)) {
TDEy[i]=pp[j*nw+i]*0.5*(1-cos(2*pi*i/nt));
}
else{
TDEy[i]=0.;
}
}
kiss_fftr(cfg,TDEy,FDEy);
for(i=0;i<nt;i++){
if((i<nw)&&((j*nw+i)<n1)) {
TDHx[i]=qq[j*nw+i]*0.5*(1-cos(2*pi*i/nt));
}
else{TDHx[i]=0.;}
}
kiss_fftr(cfg,TDHx,FDHx);
for(i=0;i<nt;i++){
if((i<nw)&&((j*nw+i)<n1)) {
TDHy[i]=dd[j*nw+i]*0.5*(1-cos(2*pi*i/nt));
}
else{
TDHy[i]=0.;
}
}
kiss_fftr(cfg,TDHy,FDHy);
A=FDEx;B=FDEy;
ExAs1[f]=sf_cadd(ExAs1[f],sf_cmul(FDEx[f],sf_conjf(A[f])));
HyBs1[f]=sf_cadd(HyBs1[f],sf_cmul(FDHy[f],sf_conjf(B[f])));
ExBs1[f]=sf_cadd(ExBs1[f],sf_cmul(FDEx[f],sf_conjf(B[f])));
HyAs1[f]=sf_cadd(HyAs1[f],sf_cmul(FDHy[f],sf_conjf(A[f])));
HxAs1[f]=sf_cadd(HxAs1[f],sf_cmul(FDHx[f],sf_conjf(A[f])));
HxBs1[f]=sf_cadd(HxBs1[f],sf_cmul(FDHx[f],sf_conjf(B[f])));
EyAs1[f]=sf_cadd(EyAs1[f],sf_cmul(FDEy[f],sf_conjf(A[f])));
EyBs1[f]=sf_cadd(EyBs1[f],sf_cmul(FDEy[f],sf_conjf(B[f])));
A=FDEx;B=FDHx;
ExAs2[f]= sf_cadd(ExAs2[f],sf_cmul(FDEx[f],sf_conjf(A[f])));
HyBs2[f]= sf_cadd(HyBs2[f],sf_cmul(FDHy[f],sf_conjf(B[f])));
ExBs2[f]= sf_cadd(ExBs2[f],sf_cmul(FDEx[f],sf_conjf(B[f])));
HyAs2[f]= sf_cadd(HyAs2[f],sf_cmul(FDHy[f],sf_conjf(A[f])));
HxAs2[f]= sf_cadd(HxAs2[f],sf_cmul(FDHx[f],sf_conjf(A[f])));
HxBs2[f]= sf_cadd(HxBs2[f],sf_cmul(FDHx[f],sf_conjf(B[f])));
EyAs2[f]= sf_cadd(EyAs2[f],sf_cmul(FDEy[f],sf_conjf(A[f])));
EyBs2[f]= sf_cadd(EyBs2[f],sf_cmul(FDEy[f],sf_conjf(B[f])));
A=FDEy;B=FDHy;
ExAs3[f]= sf_cadd(ExAs3[f],sf_cmul(FDEx[f],sf_conjf(A[f])));
HyBs3[f]= sf_cadd(HyBs3[f],sf_cmul(FDHy[f],sf_conjf(B[f])));
ExBs3[f]= sf_cadd(ExBs3[f],sf_cmul(FDEx[f],sf_conjf(B[f])));
HyAs3[f]= sf_cadd(HyAs3[f],sf_cmul(FDHy[f],sf_conjf(A[f])));
HxAs3[f]= sf_cadd(HxAs3[f],sf_cmul(FDHx[f],sf_conjf(A[f])));
HxBs3[f]= sf_cadd(HxBs3[f],sf_cmul(FDHx[f],sf_conjf(B[f])));
EyAs3[f]= sf_cadd(EyAs3[f],sf_cmul(FDEy[f],sf_conjf(A[f])));
EyBs3[f]= sf_cadd(EyBs3[f],sf_cmul(FDEy[f],sf_conjf(B[f])));
A=FDHx;B=FDHy;
ExAs4[f]= sf_cadd(ExAs4[f],sf_cmul(FDEx[f],sf_conjf(A[f])));
HyBs4[f]= sf_cadd(HyBs4[f],sf_cmul(FDHy[f],sf_conjf(B[f])));
ExBs4[f]= sf_cadd(ExBs4[f],sf_cmul(FDEx[f],sf_conjf(B[f])));
HyAs4[f]= sf_cadd(HyAs4[f],sf_cmul(FDHy[f],sf_conjf(A[f])));
HxAs4[f]= sf_cadd(HxAs4[f],sf_cmul(FDHx[f],sf_conjf(A[f])));
HxBs4[f]= sf_cadd(HxBs4[f],sf_cmul(FDHx[f],sf_conjf(B[f])));
EyAs4[f]= sf_cadd(EyAs4[f],sf_cmul(FDEy[f],sf_conjf(A[f])));
EyBs4[f]= sf_cadd(EyBs4[f],sf_cmul(FDEy[f],sf_conjf(B[f])));
}
ExA1[f]=sf_crmul(ExAs1[f],1./N);
HyB1[f]=sf_crmul(HyBs1[f],1./N);
ExB1[f]=sf_crmul(ExBs1[f],1./N);
HyA1[f]=sf_crmul(HyAs1[f],1./N);
HxA1[f]=sf_crmul(HxAs1[f],1./N);
HxB1[f]=sf_crmul(HxBs1[f],1./N);
EyA1[f]=sf_crmul(EyAs1[f],1./N);
EyB1[f]=sf_crmul(EyBs1[f],1./N);
ExA2[f]=sf_crmul(ExAs2[f],1./N);
HyB2[f]=sf_crmul(HyBs2[f],1./N);
ExB2[f]=sf_crmul(ExBs2[f],1./N);
HyA2[f]=sf_crmul(HyAs2[f],1./N);
HxA2[f]=sf_crmul(HxAs2[f],1./N);
HxB2[f]=sf_crmul(HxBs2[f],1./N);
EyA2[f]=sf_crmul(EyAs2[f],1./N);
EyB2[f]=sf_crmul(EyBs2[f],1./N);
ExA3[f]=sf_crmul(ExAs3[f],1./N);
HyB3[f]=sf_crmul(HyBs3[f],1./N);
ExB3[f]=sf_crmul(ExBs3[f],1./N);
HyA3[f]=sf_crmul(HyAs3[f],1./N);
HxA3[f]=sf_crmul(HxAs3[f],1./N);
HxB3[f]=sf_crmul(HxBs3[f],1./N);
EyA3[f]=sf_crmul(EyAs3[f],1./N);
EyB3[f]=sf_crmul(EyBs3[f],1./N);
ExA4[f]=sf_crmul(ExAs4[f],1./N);
HyB4[f]=sf_crmul(HyBs4[f],1./N);
ExB4[f]=sf_crmul(ExBs4[f],1./N);
HyA4[f]=sf_crmul(HyAs4[f],1./N);
HxA4[f]=sf_crmul(HxAs4[f],1./N);
HxB4[f]=sf_crmul(HxBs4[f],1./N);
EyA4[f]=sf_crmul(EyAs4[f],1./N);
EyB4[f]=sf_crmul(EyBs4[f],1./N);
Zxys1[f]=sf_cdiv((sf_csub(sf_cmul(ExA1[f],HxB1[f]),
sf_cmul(ExB1[f],HxA1[f]))),
(sf_csub(sf_cmul(HyA1[f],HxB1[f]),
sf_cmul(HyB1[f],HxA1[f]))));
Zyxs1[f]=sf_cdiv((sf_csub(sf_cmul(EyA1[f],HyB1[f]),
sf_cmul(EyB1[f],HyA1[f]))),
(sf_csub(sf_cmul(HxA1[f],HyB1[f]),
sf_cmul(HxB1[f],HyA1[f]))));
Zxys2[f]=sf_cdiv((sf_csub(sf_cmul(ExA2[f],HxB2[f]),
sf_cmul(ExB2[f],HxA2[f]))),
(sf_csub(sf_cmul(HyA2[f],HxB2[f]),
sf_cmul(HyB2[f],HxA2[f]))));
Zyxs2[f]=sf_cdiv((sf_csub(sf_cmul(EyA2[f],HyB2[f]),
sf_cmul(EyB2[f],HyA2[f]))),
(sf_csub(sf_cmul(HxA2[f],HyB2[f]),
sf_cmul(HxB2[f],HyA2[f]))));
Zxys3[f]=sf_cdiv((sf_csub(sf_cmul(ExA3[f],HxB3[f]),
sf_cmul(ExB3[f],HxA3[f]))),
(sf_csub(sf_cmul(HyA3[f],HxB3[f]),
sf_cmul(HyB3[f],HxA3[f]))));
Zyxs3[f]=sf_cdiv((sf_csub(sf_cmul(EyA3[f],HyB3[f]),
sf_cmul(EyB3[f],HyA3[f]))),
(sf_csub(sf_cmul(HxA3[f],HyB3[f]),
sf_cmul(HxB3[f],HyA3[f]))));
Zxys4[f]=sf_cdiv((sf_csub(sf_cmul(ExA4[f],HxB4[f]),
sf_cmul(ExB4[f],HxA4[f]))),
(sf_csub(sf_cmul(HyA4[f],HxB4[f]),
sf_cmul(HyB4[f],HxA4[f]))));
Zyxs4[f]=sf_cdiv((sf_csub(sf_cmul(EyA4[f],HyB4[f]),
sf_cmul(EyB4[f],HyA4[f]))),
(sf_csub(sf_cmul(HxA4[f],HyB4[f]),
sf_cmul(HxB4[f],HyA4[f]))));
Zxy[f]=sf_crmul(sf_cadd(sf_cadd(Zxys1[f],Zxys2[f]),
sf_cadd(Zxys3[f],Zxys4[f])),0.25);
Zyx[f]=sf_crmul(sf_cadd(sf_cadd(Zyxs1[f],Zyxs2[f]),
sf_cadd(Zyxs3[f],Zyxs4[f])),0.25);
ExHyres[f]=0.2*sf_cabsf(Zxy[f])*sf_cabsf(Zxy[f])/a;
EyHxres[f]=0.2*sf_cabsf(Zyx[f])*sf_cabsf(Zyx[f])/a;
ExHypha[f]=sf_cargf(Zxy[f]);
EyHxpha[f]=sf_cargf(Zyx[f]);
if(phase) {
if(comp) {
outp[k]= ExHypha[f];
} else {
outp[k]= EyHxpha[f];
}
} else {
if(comp) {
outp[k]= ExHyres[f];
} else {
outp[k]= EyHxres[f];
}
}
b=b+0.1;
}
sf_floatwrite(outp,M,out);
exit(0);
}
