def plot2D(image,window,figTitle=None): N = numpy.shape(image)[0] transform = gft.gft2d(image,window) pylab.figure() pylab.subplot(121) if figTitle: pylab.title(figTitle) pylab.imshow(image.real) pylab.subplot(122) pylab.imshow(abs(transform)) partitions = gft.partitions(N) for i in partitions[0:len(partitions)/2]: pylab.axvline((N/2+i),color='r',alpha=0.2) pylab.axvline((N/2-i),color='r',alpha=0.2) pylab.axhline((N/2+i),color='r',alpha=0.2) pylab.axhline((N/2-i),color='r',alpha=0.2) pylab.axis((0,N,0,N))