from sosepy import sose from numpy import * from scipy.interpolate import NearestNDInterpolator as npi import popy from pylab import * hfac = sose.loadparameter('hFacC') y = arange(320) x = arange(-10, 2160 + 10) xx, yy = meshgrid(x, y) xs = xx.flatten() ys = yy.flatten() nz, ny, nx = hfac.shape newx = zeros((nz, ny, nx + 20)) newy = zeros((nz, ny, nx + 20)) for k in arange(nz): print k hfc = hfac[k, ...] hf = c_[hfc[:, -10:], hfc, hfc[:, :10]] ip = (hf.flatten() > 0) px = npi((xs[ip], ys[ip]), xs[ip]) py = npi((xs[ip], ys[ip]), ys[ip]) newx[k, ...] = (px((xx, yy)).reshape(320, -1) - xx) #[:,1:-1] newy[k, ...] = (py((xx, yy)).reshape(320, -1) - yy) #[:,1:-1] del hfc, hf, ip, py, px newx[..., 8:-8].astype('>f4').tofile('reflect_x.bin') newy[..., 8:-8].astype('>f4').tofile('reflect_y.bin')
from sosepy import sose from numpy import * from scipy.interpolate import NearestNDInterpolator as npi import popy from pylab import * hfac=sose.loadparameter('hFacC') y=arange(320) x=arange(-10,2160+10) xx,yy=meshgrid(x,y) xs=xx.flatten() ys=yy.flatten() nz,ny,nx=hfac.shape newx=zeros((nz,ny,nx+20)) newy=zeros((nz,ny,nx+20)) for k in arange(nz): print k hfc=hfac[k,...] hf=c_[hfc[:,-10:],hfc,hfc[:,:10]] ip=(hf.flatten()>0) px = npi((xs[ip],ys[ip]),xs[ip]) py = npi((xs[ip],ys[ip]),ys[ip]) newx[k,...]=(px((xx,yy)).reshape(320,-1)-xx)#[:,1:-1] newy[k,...]=(py((xx,yy)).reshape(320,-1)-yy)#[:,1:-1] del hfc, hf, ip, py, px newx[...,8:-8].astype('>f4').tofile('reflect_x.bin')
from sosepy import sose from scipy.interpolate import interp1d from numpy import * from pylab import * z=sose.loadparameter('RF').flatten() z=abs(z) z[0]=0 print z print z.shape ff=interp1d(linspace(0,42,43),z,'cubic') print ff(r_[10,10.1]) newz = ff(linspace(0,42,421)) newz.astype('>f4').tofile('k_to_z_lookup_table.bin') plot(linspace(0,42,43),z,'o') plot(linspace(0,42,421),newz,'-') show() print newz.shape print newz