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_())