def fft_test():
aux_mat = gp.prepare_aux_matrices(512)
#p_lg = gp.gen_lg_pattern(1,10,aux_mat,5)
#p_bg = gp.gen_blaze_grating(50,aux_mat)
#p_swirl = gp.gen_swirl_pattern(aux_mat, dz=10, twist=10)
#p_hg = gp.gen_hg_pattern(512, m=1, n=2)
p_z = exp(1j*gp.phasemap(p_hg+p_bg)*2*pi)
#imshow(gp.phasemap(p_swirl), cmap=get_cmap('binary'))
#figure(2)
imshow(fftpack.fftshift(abs(fftpack.fft2(p_z))), cmap=get_cmap('binary'))
show()
def fft_test():
aux_mat = gp.prepare_aux_matrices(512)
#p_lg = gp.gen_lg_pattern(1,10,aux_mat,5)
#p_bg = gp.gen_blaze_grating(50,aux_mat)
#p_swirl = gp.gen_swirl_pattern(aux_mat, dz=10, twist=10)
#p_hg = gp.gen_hg_pattern(512, m=1, n=2)
p_z = exp(1j * gp.phasemap(p_hg + p_bg) * 2 * pi)
#imshow(gp.phasemap(p_swirl), cmap=get_cmap('binary'))
#figure(2)
imshow(fftpack.fftshift(abs(fftpack.fft2(p_z))), cmap=get_cmap('binary'))
show()
layout.setMargin(0)
self.setLayout(layout)
self.setGeometry(1681, 1050-768, 768, 768)
self.fc.mpl_connect('button_press_event', self.close_on_click)
def close_on_click(self, event):
self.close()
def imshow(self, img):
axes = self.fig.add_axes([0,0,1,1])
axes.axis('off')
axes.imshow(img, cmap=get_cmap('binary'))
if __name__ == '__main__':
import gen_pattern as gp
qApp = QtGui.QApplication(sys.argv)
SLM = SLM_Dialog()
p_lg = gp.gen_lg_pattern(768,3,3,4)
p_g = gp.gen_blaze_grating(768,50, 0.9)
pm = gp.phasemap(p_lg + p_g)
#pm = gp.phasemap(p_lg)
SLM.imshow(pm)
SLM.show()
sys.exit(qApp.exec_())
