int main(int argc, char* argv[])
{
bool adj, verb;
int n1, n2, n12, n3, i3, order, ns;
float *input, *smooth, **slope, eps;
sf_file in, out, dip;
sf_init(argc,argv);
in = sf_input("in");
dip = sf_input("dip");
out = sf_output("out");
if (!sf_histint(in,"n1",&n1)) sf_error("No n1= in input");
if (!sf_histint(in,"n2",&n2)) sf_error("No n2= in input");
n3 = sf_leftsize(in,2);
n12 = n1*n2;
if (!sf_getbool("verb",&verb)) verb=false;
/* verbosity flag */
if (!sf_getint("ns",&ns)) ns=0;
/* smoothing radius */
if (!sf_getbool("adj",&adj)) adj=false;
/* adjoint flag */
if (!sf_getfloat("eps",&eps)) eps=0.01;
/* regularization */
if (!sf_getint("order",&order)) order=1;
/* accuracy order */
input = sf_floatalloc(n12);
smooth = sf_floatalloc(n12);
slope = sf_floatalloc2(n1,n2);
pwsmooth_init(ns, n1, n2, order, eps);
for (i3=0; i3 < n3; i3++) {
if (verb) sf_warning("slice %d of %d;",i3+1,n3);
sf_floatread(input,n12,in);
sf_floatread(slope[0],n12,dip);
pwsmooth_set(slope);
if (adj) {
pwsmooth_lop(true,false,n12,n12,smooth,input);
} else {
pwsmooth_lop(false,false,n12,n12,input,smooth);
}
sf_floatwrite(smooth,n12,out);
}
if (verb) sf_warning(".");
exit(0);
}
int main(int argc, char* argv[])
{
bool hermite_false, hermite_true, precond;
int n1, n2, npml, pad1, pad2, ns, nw, radius;
float d1, d2, ds, os, dw, ow;
double omega;
sf_complex ***f, ***obs;
sf_file in, out, source, receiver, record, dip;
char *order;
int uts, mts, i, j, k, iw, niter, iter;
float **cur_grad, **pre_grad, **cur_dir, **pre_dir;
float *cur_grad_input, *cur_grad_smooth, *cur_grad_smooth2;
float misfit0, misfit1, misfit2, misfitold, beta, alpha;
float **v, **vnew, **slope, **recloc;
sf_init(argc, argv);
in = sf_input("in");
out = sf_output("out");
if (!sf_getint("niter",&niter)) niter=0;
if (!sf_getint("uts",&uts)) uts=0;
//#ifdef _OPENMP
// mts = omp_get_max_threads();
//#else
mts = 1;
//#endif
uts = (uts < 1)? mts: uts;
hermite_false=false;
hermite_true=true;
/* Hermite operator */
if (!sf_getint("npml",&npml)) npml=20;
/* PML width */
if (NULL == (order = sf_getstring("order"))) order="j";
/* discretization scheme (default optimal 9-point) */
fdprep_order(order);
/* read input dimension */
if (!sf_histint(in,"n1",&n1)) sf_error("No n1= in input.");
if (!sf_histint(in,"n2",&n2)) sf_error("No n2= in input.");
if (!sf_histfloat(in,"d1",&d1)) sf_error("No d1= in input.");
if (!sf_histfloat(in,"d2",&d2)) sf_error("No d2= in input.");
v = sf_floatalloc2(n1,n2);
vnew = sf_floatalloc2(n1,n2);
sf_floatread(v[0],n1*n2,in);
/* PML padding */
pad1 = n1+2*npml;
pad2 = n2+2*npml;
/* read receiver */
if (NULL == sf_getstring("receiver")) sf_error("Need receiver=");
receiver = sf_input("receiver");
recloc = sf_floatalloc2(n1,n2);
sf_floatread(recloc[0],n1*n2,receiver);
/* read source */
if (NULL == sf_getstring("source")) sf_error("Need source=");
source = sf_input("source");
if (!sf_histint(source,"n3",&ns)) sf_error("No ns=.");
if (!sf_histfloat(source,"d3",&ds)) ds=d2;
if (!sf_histfloat(source,"o3",&os)) os=0.;
f = sf_complexalloc3(n1,n2,ns);
if (NULL == sf_getstring("record")) sf_error("Need record=");
record = sf_input("record");
if (!sf_histint(record,"n4",&nw)) sf_error("No nw=.");
if (!sf_histfloat(record,"d4",&dw)) sf_error("No dw=.");
if (!sf_histfloat(record,"o4",&ow)) sf_error("No ow=.");
/* read in slope information */
if (!sf_getbool("precond",&precond)) precond=false;
if (precond) {
if (NULL == sf_getstring("dip")) sf_error("Need dip=");
dip = sf_input("dip");
slope = sf_floatalloc2(n1,n2);
sf_floatread(slope[0],n1*n2,dip);
if (!sf_getint("radius",&radius)) sf_error("Need radius=");
cur_grad_input = sf_floatalloc(n1*n2);
cur_grad_smooth = sf_floatalloc(n1*n2);
cur_grad_smooth2 = sf_floatalloc(n1*n2);
}
sf_putint(out,"n1",n1);
sf_putint(out,"n2",n2);
sf_putint(out,"n3",niter*nw);
obs = sf_complexalloc3(n1,n2,ns);
cur_grad = sf_floatalloc2(n1,n2);
cur_dir = sf_floatalloc2(n1,n2);
pre_grad = sf_floatalloc2(n1,n2);
pre_dir = sf_floatalloc2(n1,n2);
/* Loop over frequency */
for ( iw = 0; iw < nw; iw ++ ) {
omega=(double) 2.*SF_PI*(ow+iw*dw);
sf_warning("Calculating frequency %d out of %d for %f HZ.\n",iw+1,nw,ow+iw*dw);
sf_complexread(f[0][0],n1*n2*ns,source);
sf_complexread(obs[0][0],n1*n2*ns,record);
misfitold = 100000000.0;
iter = 0;
pwsmooth_init(radius, n1, n2, 5, 0.01);
while (iter < niter) {
sf_warning("Calculating %d out of %d iteration", iter+1, niter);
misfit0 = adjfwi_operator(uts, pad1, pad2, omega, n1, n2, d1, d2,
npml, ns, f, obs, hermite_false, hermite_true, recloc,
v, cur_grad);
/* Preconditioning or regularization */
if (precond) {
sf_warning("Preconditioning gradient.");
/* change from 2D array to 1D */
k=0;
for (j=0;j<n2;j++) {
for (i=0;i<n1;i++) {
cur_grad_input[k]=cur_grad[j][i];
k=k+1;
}
}
pwsmooth_set(slope);
pwsmooth_lop(true,false,n1*n2,n1*n2,cur_grad_smooth,cur_grad_input); /* adjoint of pwspray */
pwsmooth_lop(false,false,n1*n2,n1*n2,cur_grad_smooth,cur_grad_smooth2); /* forward of pwspray */
/* reset cur_grad to smooth version */
k=0;
for (j=0; j<n2;j++) {
for (i=0; i<n1; i++) {
cur_grad[j][i]=cur_grad_smooth2[k];
/* cur_grad[j][i]=cur_grad_smooth[k]; */
k=k+1;
}
}
}
/* calculate direction */
if (iter == 0) {
direction_sd(cur_grad,cur_dir,n1,n2);
beta = 0.0;
} else {
beta = direction_cg_polak(pre_grad, pre_dir, cur_grad, cur_dir, n1, n2);
}
sf_warning("Finish direction calculation.");
/* test model 1 */
update_model_fwi(v,vnew,cur_dir,0.1,n1,n2);
misfit1 = forward_operator(uts, pad1, pad2, omega, n1, n2, d1, d2,
npml, ns, f, obs, hermite_false, recloc,
vnew);
sf_warning("Finish test1 calculation.");
/* test model 2 */
update_model_fwi(v,vnew,cur_dir,0.2,n1,n2);
misfit2 = forward_operator(uts, pad1, pad2, omega, n1, n2, d1, d2,
npml, ns, f, obs, hermite_false, recloc,
vnew);
sf_warning("Finish test2 calculation.");
/* update model, quaduartic fit */
alpha = (misfit2-4.0*misfit1+3.0*misfit0)/(20.0*(misfit2-2.0*misfit1+misfit0));
if ( alpha < 0.0 ) {
alpha = 0.001;
sf_warning("alpha is smaller than 0.0");
}
sf_warning("In iteration %d, alpha = %f, beta = %f, misfit0 = %f.",iter+1,alpha,beta,misfit0);
sf_warning("Test model alpha=0.1, misfit1 = %f, alpha=0.2, misfit2 = %f.\n",misfit1,misfit2);
if ( misfit0 > misfitold ) {
sf_warning("Terminate at iteration %d.",iter);
break;
} else {
iter++;
update_model_fwi(v,vnew,cur_dir,alpha,n1,n2);
for (j=0; j<n2; j++ ) {
for (i=0; i<n1; i++ ) {
v[j][i]=vnew[j][i];
pre_grad[j][i]=cur_grad[j][i];
pre_dir[j][i]=cur_dir[j][i];
}
}
misfitold=misfit0;
}
sf_floatwrite(v[0],n1*n2,out);
} /* end iteration */
sf_warning("Ending frequency %d out of %d for %f HZ.\n\n",iw+1,nw,ow+iw*dw);
} /* end frequency */
exit(0);
}
int main(int argc, char* argv[])
{
int i, niter, n1, n2, n12, i3, n3, ns, order, np;
float *mm, *dd, *xx, **pp, **qq, lam, eps;
bool *known;
sf_file in, out, dip, mask;
sf_init (argc,argv);
in = sf_input("in");
out = sf_output("out");
dip = sf_input("dip");
if (!sf_histint(in,"n1",&n1)) sf_error("No n1= in input");
if (!sf_histint(in,"n2",&n2)) sf_error("No n2= in input");
n12 = n1*n2;
n3 = sf_leftsize(in,2);
if (!sf_getint("niter",&niter)) niter=100;
/* number of iterations */
if (!sf_getint("order",&order)) order=1;
/* accuracy order */
pp = sf_floatalloc2(n1,n2);
mm = sf_floatalloc(n12);
xx = sf_floatalloc(n12);
known = sf_boolalloc(n12);
np = sf_leftsize(dip,2);
if (np > n3) {
qq = sf_floatalloc2(n1,n2);
} else {
qq = NULL;
}
if (NULL != sf_getstring ("mask")) {
mask = sf_input("mask");
dd = sf_floatalloc(n12);
} else {
mask = NULL;
dd = NULL;
}
if (!sf_getint("ns",&ns)) ns=1;
/* smoothing radius */
if (!sf_getfloat("eps",&eps)) eps=0.01;
/* regularization */
sf_mask_init(known);
for (i3=0; i3 < n3; i3++) {
sf_warning("slice %d of %d",i3+1,n3);
sf_floatread(mm,n12,in);
for (i=0; i < n12; i++) {
xx[i] = mm[i];
}
if (NULL != mask) {
sf_floatread(dd,n12,mask);
} else {
dd = mm;
}
/* figure out scaling and make known data mask */
lam = 0.;
for (i=0; i < n12; i++) {
if (dd[i] != 0.) {
known[i] = true;
lam += 1.;
} else {
known[i] = false;
}
}
lam = sqrtf(lam/n12);
/* read dip */
sf_floatread(pp[0],n12,dip);
sf_conjgrad_init(n12, n12, n12, n12, lam, 10*FLT_EPSILON, true, true);
if (NULL != qq) {
sf_floatread(qq[0],n12,dip);
pwsmooth2_init(ns, n1, n2, order, eps, pp, qq);
sf_conjgrad(NULL,sf_mask_lop,pwsmooth2_lop,xx,mm,mm,niter);
pwsmooth2_close();
} else {
pwsmooth_init(ns, n1, n2, order, eps, pp);
sf_conjgrad(NULL,sf_mask_lop,pwsmooth_lop,xx,mm,mm,niter);
pwsmooth_close();
}
sf_conjgrad_close();
sf_floatwrite (mm,n12,out);
}
exit(0);
}
