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
