/**
Setup ray tracing operator HXF from xloc to aperture ploc along DM fiting directions*/
static void
setup_recon_HXF(RECON_T *recon, const PARMS_T *parms){
if(parms->load.HXF && zfexist(parms->load.HXF)){
warning("Loading saved HXF\n");
recon->HXF=dspcellread("%s",parms->load.HXF);
}else{
info2("Generating HXF");TIC;tic;
const int nfit=parms->fit.nfit;
const int npsr=recon->npsr;
recon->HXF=cellnew(nfit, npsr);
PDSPCELL(recon->HXF,HXF);
for(int ifit=0; ifit<nfit; ifit++){
double hs=parms->fit.hs->p[ifit];
for(int ips=0; ips<npsr; ips++){
const double ht = recon->ht->p[ips];
const double scale=1.-ht/hs;
double displace[2];
displace[0]=parms->fit.thetax->p[ifit]*ht;
displace[1]=parms->fit.thetay->p[ifit]*ht;
HXF[ips][ifit]=mkh(recon->xloc->p[ips], recon->floc,
displace[0], displace[1], scale);
}
}
if(parms->save.setup){
writebin(recon->HXF,"HXF");
}
toc2(" ");
}
}
/**
Setup the detector transfer functions. See maos/setup_powfs.c
*/
static void setup_powfs_dtf(POWFS_S *powfs, const PARMS_S* parms){
const int npowfs=parms->maos.npowfs;
for(int ipowfs=0; ipowfs<npowfs; ipowfs++){
const int ncomp=parms->maos.ncomp[ipowfs];
const int ncomp2=ncomp>>1;
const int embfac=parms->maos.embfac[ipowfs];
const int pixpsa=parms->skyc.pixpsa[ipowfs];
const double pixtheta=parms->skyc.pixtheta[ipowfs];
const double blur=parms->skyc.pixblur[ipowfs]*pixtheta;
const double e0=exp(-2*M_PI*M_PI*blur*blur);
const double dxsa=parms->maos.dxsa[ipowfs];
const double pixoffx=parms->skyc.pixoffx[ipowfs];
const double pixoffy=parms->skyc.pixoffy[ipowfs];
const double pxo=-(pixpsa/2-0.5+pixoffx)*pixtheta;
const double pyo=-(pixpsa/2-0.5+pixoffy)*pixtheta;
loc_t *loc_ccd=mksqloc(pixpsa, pixpsa, pixtheta, pixtheta, pxo, pyo);
powfs[ipowfs].dtf=mycalloc(parms->maos.nwvl,DTF_S);
for(int iwvl=0; iwvl<parms->maos.nwvl; iwvl++){
const double wvl=parms->maos.wvl[iwvl];
const double dtheta=wvl/(dxsa*embfac);
const double pdtheta=pixtheta*pixtheta/(dtheta*dtheta);
const double du=1./(dtheta*ncomp);
const double du2=du*du;
const double dupth=du*pixtheta;
cmat *nominal=cnew(ncomp,ncomp);
//cfft2plan(nominal,-1);
//cfft2plan(nominal,1);
cmat* pn=nominal;
const double theta=0;
const double ct=cos(theta);
const double st=sin(theta);
for(int iy=0; iy<ncomp; iy++){
int jy=iy-ncomp2;
for(int ix=0; ix<ncomp; ix++){
int jx=ix-ncomp2;
double ir=ct*jx+st*jy;
double ia=-st*jx+ct*jy;
IND(pn,ix,iy)=sinc(ir*dupth)*sinc(ia*dupth)
*pow(e0,ir*ir*du2)*pow(e0,ia*ia*du2)
*pdtheta;
}
}
if(parms->skyc.fnpsf1[ipowfs]){
warning("powfs %d has additional otf to be multiplied\n", ipowfs);
dcell *psf1c=dcellread("%s", parms->skyc.fnpsf1[ipowfs]);
dmat *psf1=NULL;
if(psf1c->nx == 1){
psf1=dref(psf1c->p[0]);
}else if(psf1c->nx==parms->maos.nwvl){
psf1=dref(psf1c->p[iwvl]);
}else{
error("skyc.fnpsf1 has wrong dimension\n");
}
dcellfree(psf1c);
if(psf1->ny!=2){
error("skyc.fnpsf1 has wrong dimension\n");
}
dmat *psf1x=dnew_ref(psf1->nx, 1, psf1->p);
dmat *psf1y=dnew_ref(psf1->nx, 1, psf1->p+psf1->nx);
dmat *psf2x=dnew(ncomp*ncomp, 1);
for(int iy=0; iy<ncomp; iy++){
int jy=iy-ncomp2;
for(int ix=0; ix<ncomp; ix++){
int jx=ix-ncomp2;
psf2x->p[ix+iy*ncomp]=sqrt(jx*jx+jy*jy)*dtheta;
}
}
info("powfs %d, iwvl=%d, dtheta=%g\n", ipowfs, iwvl, dtheta*206265000);
writebin(psf2x, "powfs%d_psf2x_%d", ipowfs,iwvl);
dmat *psf2=dinterp1(psf1x, psf1y, psf2x, 0);
normalize_sum(psf2->p, psf2->nx*psf2->ny, 1);
psf2->nx=ncomp; psf2->ny=ncomp;
writebin(psf2, "powfs%d_psf2_%d", ipowfs,iwvl);
cmat *otf2=cnew(ncomp, ncomp);
//cfft2plan(otf2, -1);
ccpd(&otf2, psf2);//peak in center
cfftshift(otf2);//peak in corner
cfft2(otf2, -1);
cfftshift(otf2);//peak in center
writebin(otf2, "powfs%d_otf2_%d", ipowfs, iwvl);
writebin(nominal, "powfs%d_dtf%d_nominal_0",ipowfs,iwvl);
for(int i=0; i<ncomp*ncomp; i++){
nominal->p[i]*=otf2->p[i];
}
writebin(nominal, "powfs%d_dtf%d_nominal_1",ipowfs,iwvl);
dfree(psf1x);
dfree(psf1y);
dfree(psf2x);
dfree(psf1);
cfree(otf2);
dfree(psf2);
}
cfftshift(nominal);//peak in corner
cfft2(nominal,-1);
cfftshift(nominal);//peak in center
cfft2i(nominal,1);
warning_once("double check nominal for off centered skyc.fnpsf1\n");
/*This nominal will multiply to OTF with peak in corner. But after
inverse fft, peak will be in center*/
ccp(&powfs[ipowfs].dtf[iwvl].nominal, nominal);
cfree(nominal);
loc_t *loc_psf=mksqloc(ncomp, ncomp, dtheta, dtheta, -ncomp2*dtheta, -ncomp2*dtheta);
powfs[ipowfs].dtf[iwvl].si=mkh(loc_psf,loc_ccd,0,0,1);
locfree(loc_psf);
if(parms->skyc.dbg){
writebin(powfs[ipowfs].dtf[iwvl].nominal,
"%s/powfs%d_dtf%d_nominal",dirsetup,ipowfs,iwvl);
writebin(powfs[ipowfs].dtf[iwvl].si,
"%s/powfs%d_dtf%d_si",dirsetup,ipowfs,iwvl);
}
powfs[ipowfs].dtf[iwvl].U=cnew(ncomp,1);
dcomplex *U=powfs[ipowfs].dtf[iwvl].U->p;
for(int ix=0; ix<ncomp; ix++){
int jx=ix<ncomp2?ix:(ix-ncomp);
U[ix]=COMPLEX(0, -2.*M_PI*jx*du);
}
}/*iwvl */
locfree(loc_ccd);
}/*ipowfs */
}
static void test_accuracy(int argc, char **argv){
double displacex=0.01;
double displacey=0.05;
double scale=1.01;/*.414065; */
int wrap=0;
double D=30;
double D2=32;
int save=1;
if(argc>1){
scale=strtod(argv[1], 0);
}
if(argc>2){
displacex=strtod(argv[2], 0);
}
if(argc>3){
displacey=strtod(argv[3], 0);
}
if(argc>4){
wrap=strtol(argv[4], 0, 10);
}
if(argc>5){
D=strtod(argv[5], 0);
}
if(argc>6){
D2=strtod(argv[6], 0);
}
if(argc>7){
save=strtol(argv[7], 0, 10);
}
double dx=1/64.;
double dsa=0.5;
map_t *screen=mapnew(D2/dx, D2/dx, dx, dx, 0);
dset((dmat*)screen, 1);
for(long iy=0; iy<screen->ny; iy++){
for(long ix=0; ix<screen->nx; ix++){
screen->p[ix+iy*screen->nx]=sin((double)ix/screen->nx*2*M_PI)*sin((double)iy/screen->ny*2*M_PI);
}
}
/*loc for the map */
loc_t *locin=mksqloc(screen->nx,screen->ny,dx,dx,screen->ox,screen->oy);
//pts_t *pts=realloc(mkannloc(D, 0, 1./2.,0), sizeof(pts_t));
pts_t *pts=realloc(mksqloc_auto(D/dsa, D/dsa, dsa, dsa), sizeof(pts_t));
pts->dx=dx;
pts->dy=dx;
pts->nx=dsa/dx;
pts->ny=pts->nx;
loc_t *loc=pts2loc(pts);
loc_create_map(locin);
loc_create_stat(loc);
loc_create_map(loc);
locstat_t *locstat=loc->stat;
map_t *screen2=mapnew2(locin->map);
dset((dmat*)screen2, NAN);
loc_embed(screen2, locin, screen->p);
double *phi_pts, *phi_loc, *phi_stat;
double *phi_loc2loc, *phi_h, *phi_cub, *phi_cub2, *phi_cubh;
double cubic=0.3;
int ii;
info("displacex=%g, displacey=%g, scale=%g, wrap=%d\n",
displacex, displacey,scale,wrap);
double diff1, diff2,diff3,diff14,diff15;
diff1=0;
diff2=0;
diff3=0;
diff14=0;
diff15=0;
phi_pts=calloc(loc->nloc, sizeof(double));
phi_loc=calloc(loc->nloc, sizeof(double));
phi_stat=calloc(loc->nloc, sizeof(double));
phi_loc2loc=calloc(loc->nloc, sizeof(double));
map_t *map1=mapnew2(loc->map);
prop_grid_map(screen, map1, -2, displacex, displacey, scale, wrap, 0,0);
tic;
prop_grid_map(screen, map1, 1, displacex, displacey, scale, wrap, 0,0);
toc("map\t\t");
prop_grid_pts(screen, pts, phi_pts, -2, displacex, displacey, scale, wrap, 0,0);
tic;
prop_grid_pts(screen, pts, phi_pts, 1, displacex, displacey, scale, wrap, 0,0);
toc("pts\t\t");
prop_grid_stat(screen, locstat, phi_stat, -2,displacex, displacey, scale, wrap, 0,0);
tic;
prop_grid_stat(screen, locstat, phi_stat , 1, displacex, displacey, scale, wrap, 0,0);
toc("stat\t");tic;
prop_grid(screen, loc, phi_loc, -2,displacex, displacey, scale, wrap, 0,0);
tic;
prop_grid(screen, loc, phi_loc, 1,displacex, displacey, scale, wrap, 0,0);
toc("loc\t\t");
prop_nongrid(locin, screen->p,loc, phi_loc2loc, -2,displacex, displacey, scale,0,0);
toc("nongrid\t"); tic;
prop_nongrid(locin, screen->p,loc, phi_loc2loc, 1,displacex, displacey, scale,0,0);
toc("nongrid\t");
phi_h=calloc(loc->nloc,sizeof(double));
tic;
dsp *hfor=mkh(locin, loc, displacex, displacey, scale);
toc("mkh\t\t\t");
dspmulvec(phi_h,hfor, screen->p, 'n', -2);
tic;
dspmulvec(phi_h,hfor, screen->p, 'n', 1);
toc("mul h\t\t");
phi_cub=calloc(loc->nloc,sizeof(double));
phi_cub2=calloc(loc->nloc,sizeof(double));
double *phi_cub3=calloc(loc->nloc,sizeof(double));
double *phi_cub4=calloc(loc->nloc,sizeof(double));
phi_cubh=calloc(loc->nloc, sizeof(double));
prop_nongrid_cubic(locin,screen->p,loc,phi_cub,-2, displacex, displacey, scale, cubic,0,0);
tic;
prop_nongrid_cubic(locin,screen->p,loc,phi_cub,1, displacex, displacey, scale, cubic,0,0);
toc("nongrid, cubic\t");
prop_grid_cubic(screen, loc, phi_cub2, -2,displacex, displacey, scale, cubic, 0,0);
tic;
prop_grid_cubic(screen, loc, phi_cub2, 1,displacex, displacey, scale, cubic, 0,0);
toc("grid, cubic\t");
prop_grid_cubic(screen2, loc,phi_cub3, -2,displacex, displacey, scale, cubic, 0,0);
tic;
prop_grid_cubic(screen2, loc,phi_cub3, 1,displacex, displacey, scale, cubic, 0,0);
toc("grid2, cubic\t");
prop_grid_stat_cubic(screen, locstat,phi_cub4, -2,displacex, displacey, scale, cubic, 0,0);
tic;
prop_grid_stat_cubic(screen, locstat,phi_cub4, 1,displacex, displacey, scale, cubic, 0,0);
toc("grid 2stat, cubic\t");
dsp *hforcubic;
tic;
hforcubic=mkh_cubic(locin, loc, displacex, displacey, scale, cubic);
toc("mkh cubic \t\t");
dspmulvec(phi_cubh, hforcubic,screen->p,'n',-2);
tic;
dspmulvec(phi_cubh, hforcubic,screen->p,'n',1);
toc("cubic mul h\t\t");
double diffc12=0,diff45=0,diff46=0,diff47=0;
for(ii=0; ii<loc->nloc; ii++){
diff1+=fabs(phi_loc[ii]-phi_pts[ii]);
diff2+=fabs(phi_stat[ii]-phi_loc[ii]);
diff3+=fabs(phi_stat[ii]-phi_pts[ii]);
diff14+=fabs(phi_loc2loc[ii]-phi_pts[ii]);
diff15+=fabs(phi_h[ii]-phi_pts[ii]);
diff45+=fabs(phi_loc2loc[ii]-phi_h[ii]);
diffc12+=fabs(phi_cub[ii]-phi_cubh[ii]);
diff46+=fabs(phi_cub[ii]-phi_cub2[ii]);
diff47+=fabs(phi_cub[ii]-phi_cub3[ii]);
}
info2("(pts-loc)=\t%g\n(loc-stat)=\t%g\n(stat-pts)=\t%g\n"
"(loc2loc-pts)=\t%g\n(h-pts)=\t%g\n"
"(loc2loc-h)=\t%g\n"
"(cub:h-loc2loc)=\t%g\n"
"(cub:map2loc-loc2loc)=\t%g\n"
"(cub:locmap2loc-loc2loc=\t%g\n"
,diff1, diff2,diff3,diff14,diff15,diff45,diffc12,
diff46, diff47);
// exit(0);
if(!save) return;
mapwrite(screen2,"accphi_screen2");
mapwrite(screen,"accphi_screen");
locwrite((loc_t*)pts,"accphi_pts");
locwrite(loc,"accphi_loc");
locwrite(locin, "accphi_locin");
loc_create_map_npad(locin, 0, 0, 0);
mapwrite(locin->map, "accphi_locin_map");
mapwrite(loc->map, "accphi_loc_map");
mapwrite(map1, "accphi_map2map.bin");
writedbl(phi_pts,loc->nloc,1,"accphi_pts1.bin");
writedbl(phi_loc,loc->nloc,1,"accphi_loc0.bin");
writedbl(phi_stat,loc->nloc,1,"accphi_stat.bin");
writedbl(phi_loc2loc,loc->nloc,1,"accphi_loc2loc.bin");
writedbl(phi_h,loc->nloc,1,"accphi_loc2h.bin");
writedbl(phi_cub,loc->nloc,1,"accphi_cub_loc2loc.bin");
writedbl(phi_cub2,loc->nloc,1,"accphi_cub_map2loc.bin");
writedbl(phi_cub3,loc->nloc,1,"accphi_cub_locmap2loc.bin");
writedbl(phi_cub4,loc->nloc,1,"accphi_cub_locmap2stat.bin");
writedbl(phi_cubh,loc->nloc,1,"accphi_cub_loc2h.bin");
info("saved\n");
writedbl(phi_pts,loc->nloc,1,"accphi_pts.bin");
writedbl(phi_cub,loc->nloc,1,"accphi_cub.bin");
writebin(hfor, "accphi_hfor");
writebin(hforcubic, "accphi_cub_hfor");
dspfree(hfor);
dspfree(hforcubic);
free(phi_loc); free(phi_stat);
free(phi_loc2loc);
free(phi_pts);
free(phi_h);
free(phi_cub);
free(phi_cubh);
cellfree(screen);
locfree(locin);
}
