plot_overlay.py

#!/usr/bin/python

import code_tools
import copy
import numpy as np
import matplotlib.pyplot as plt
plt.ion()

config = code_tools.ConfigureDataSet()
config.load_config('/home/daniel/Downloads/test.cfg')

arraytools = code_tools.ArrayTools(config)
plottools = code_tools.PlotTools(config)

data_dict = arraytools.load_mat('/home/daniel/Downloads/test.mat')
real = data_dict['comp'].real
inten = data_dict['inten']
v_min, v_max = plottools.get_data_scale(data_dict,'real')

my_cmap = copy.copy(plt.cm.get_cmap('gray')) # get a copy of the gray color map
my_cmap.set_bad(alpha=0) # set how the colormap handles 'bad' values

lattice = plt.imread('/home/daniel/Downloads/epsilon.png')
im = plt.imshow(real[0,:,:],vmin=v_min,vmax=v_max,extent=(0,32,0,32),interpolation='bilinear',cmap='jet')

lattice_mod = lattice
#lattice_mod[lattice_mod < 1 ] = 0

lattice_mod[lattice_mod > 0] = np.nan # insert 'bad' values into your lattice (the white)
lattice_mod[lattice_mod == 0] = 255
 
im2 = plt.imshow(lattice_mod,extent=(0,32,0,32),cmap=my_cmap,vmin=0, vmax=255)

plt.show()

###
#print "\n*********testing*********\n"
#im = plt.imshow(inten[0,:,:])
#im.set_extent((0,32,0,32))
#fig = im.get_figure()
#fig.canvas.draw()