"""
Created on Mon Mar 27 17:25:39 2023

@author: Xiang Xi
"""
import numpy as np
import matplotlib.pylab as plt
import h5py


#%% Figure S8
file_name='figure_S8_data'

#load data
tempt=h5py.File(file_name, "r")
subfig1='figS8_dev1'
plot_x_1=np.array(tempt[subfig1]['plot_x'])
plot_y_1=np.array(tempt[subfig1]['plot_y'])
amplitude_1=np.array(tempt[subfig1]['amplitude'])


subfig2='figS8_dev2'
plot_x_2=np.array(tempt[subfig2]['plot_x'])
plot_y_2=np.array(tempt[subfig2]['plot_y'])
amplitude_2=np.array(tempt[subfig2]['amplitude'])


subfig3='figS8_dev3'
plot_x_3=np.array(tempt[subfig3]['plot_x'])
plot_y_3=np.array(tempt[subfig3]['plot_y'])
amplitude_3=np.array(tempt[subfig3]['amplitude'])


subfig4='figS8_dev4'
plot_x_4=np.array(tempt[subfig4]['plot_x'])
plot_y_4=np.array(tempt[subfig4]['plot_y'])
amplitude_4=np.array(tempt[subfig4]['amplitude'])


subfig5='figS8_dev5'
plot_x_5=np.array(tempt[subfig5]['plot_x'])
plot_y_5=np.array(tempt[subfig5]['plot_y'])
amplitude_5=np.array(tempt[subfig5]['amplitude'])
tempt.close()
#plot
font = {'family': 'arial', 'color': 'black', 'size': 12}

plt.close(subfig1)
plt.figure(subfig1)
plt.scatter(plot_x_1,plot_y_1, c=amplitude_1,s=2,cmap='inferno')
plt.xlabel('x position (um)',**font)
plt.xlim([-1600,1600])
plt.ylabel('y position (um)',**font)
plt.ylim([-1600,1600])
plt.axis('equal')
plt.title(subfig1)
plt.xticks(**font)
plt.yticks(**font)
ax = plt.gca()
ax.spines['top'].set_linewidth(1)    # Top spine
ax.spines['right'].set_linewidth(1)  # Right spine
ax.spines['bottom'].set_linewidth(1) # Bottom spine
ax.spines['left'].set_linewidth(1)   # Left spin
cbar = plt.colorbar()
cbar.set_label('Normalized amplitude',**font)
cbar.outline.set_linewidth(1)
for cb in cbar.ax.get_yticklabels():
     cb.set_fontsize(12)
     cb.set_fontname('arial')   
plt.clim([0,1.7e-6])

plt.close(subfig2)
plt.figure(subfig2)
plt.scatter(plot_x_2,plot_y_2, c=amplitude_2,s=2,cmap='inferno')
plt.xlabel('x position (um)',**font)
plt.xlim([-1600,1600])
plt.ylabel('y position (um)',**font)
plt.ylim([-1600,1600])
plt.axis('equal')
plt.title(subfig2)
plt.xticks(**font)
plt.yticks(**font)
ax = plt.gca()
ax.spines['top'].set_linewidth(1)    # Top spine
ax.spines['right'].set_linewidth(1)  # Right spine
ax.spines['bottom'].set_linewidth(1) # Bottom spine
ax.spines['left'].set_linewidth(1)   # Left spin
cbar = plt.colorbar()
cbar.set_label('Normalized amplitude',**font)
cbar.outline.set_linewidth(1)
for cb in cbar.ax.get_yticklabels():
     cb.set_fontsize(12)
     cb.set_fontname('arial')   
plt.clim([0,0.6])


plt.close(subfig3)
plt.figure(subfig3)
plt.scatter(plot_x_3,plot_y_3, c=amplitude_3,s=2,cmap='inferno')
plt.xlabel('x position (um)',**font)
plt.xlim([-1600,1600])
plt.ylabel('y position (um)',**font)
plt.ylim([-1600,1600])
plt.axis('equal')
plt.title(subfig3)
plt.xticks(**font)
plt.yticks(**font)
ax = plt.gca()
ax.spines['top'].set_linewidth(1)    # Top spine
ax.spines['right'].set_linewidth(1)  # Right spine
ax.spines['bottom'].set_linewidth(1) # Bottom spine
ax.spines['left'].set_linewidth(1)   # Left spin
cbar = plt.colorbar()
cbar.set_label('Normalized amplitude',**font)
cbar.outline.set_linewidth(1)
for cb in cbar.ax.get_yticklabels():
     cb.set_fontsize(12)
     cb.set_fontname('arial')   
plt.clim([0,1])

plt.close(subfig4)
plt.figure(subfig4)
plt.scatter(plot_x_4,plot_y_4, c=amplitude_4,s=2,cmap='inferno')
plt.xlabel('x position (um)',**font)
plt.xlim([-1600,1600])
plt.ylabel('y position (um)',**font)
plt.ylim([-1600,1600])
plt.axis('equal')
plt.title(subfig4)
plt.xticks(**font)
plt.yticks(**font)
ax = plt.gca()
ax.spines['top'].set_linewidth(1)    # Top spine
ax.spines['right'].set_linewidth(1)  # Right spine
ax.spines['bottom'].set_linewidth(1) # Bottom spine
ax.spines['left'].set_linewidth(1)   # Left spin
cbar = plt.colorbar()
cbar.set_label('Normalized amplitude',**font)
cbar.outline.set_linewidth(1)
for cb in cbar.ax.get_yticklabels():
     cb.set_fontsize(12)
     cb.set_fontname('arial')   
plt.clim([0,0.9])

plt.close(subfig5)
plt.figure(subfig5)
plt.scatter(plot_x_5,plot_y_5, c=amplitude_5,s=2,cmap='inferno')
plt.xlabel('x position (um)',**font)
plt.xlim([-1600,1600])
plt.ylabel('y position (um)',**font)
plt.ylim([-1600,1600])
plt.axis('equal')
plt.title(subfig5)
plt.xticks(**font)
plt.yticks(**font)
ax = plt.gca()
ax.spines['top'].set_linewidth(1)    # Top spine
ax.spines['right'].set_linewidth(1)  # Right spine
ax.spines['bottom'].set_linewidth(1) # Bottom spine
ax.spines['left'].set_linewidth(1)   # Left spin
cbar = plt.colorbar()
cbar.set_label('Normalized amplitude',**font)
cbar.outline.set_linewidth(1)
for cb in cbar.ax.get_yticklabels():
     cb.set_fontsize(12)
     cb.set_fontname('arial')   
plt.clim([0,1.1])