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


#%% Figure S8
file_name='figure_S7_data'

#load data
tempt=h5py.File(file_name, "r")
subfig1='Figure_S7_left_corner_mode1'
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='Figure_S7_left_corner_mode2'
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='Figure_S7_right_corner_mode1'
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='Figure_S7_right_corner_mode2'
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='Figure_S7_up_corner_mode1'
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'])


subfig6='Figure_S7_up_corner_mode2'
plot_x_6=np.array(tempt[subfig6]['plot_x'])
plot_y_6=np.array(tempt[subfig6]['plot_y'])
amplitude_6=np.array(tempt[subfig6]['amplitude'])
tempt.close()


#plot
font = {'family': 'arial', 'color': 'black', 'size': 12}

plt.close(subfig1)
plt.figure(subfig1,figsize=(4,3))
plt.scatter(plot_x_1,plot_y_1, c=amplitude_1,s=12,cmap='inferno')

plt.xlabel('x position (um)')
plt.xlim([-600,600])
plt.ylabel('y position (um)')
plt.ylim([-600,600])
plt.axis('equal')
plt.colorbar()
plt.clim([0.0e-6,2e-6])
plt.show()

plt.close(subfig2)
plt.figure(subfig2,figsize=(4,3))
plt.scatter(plot_x_2,plot_y_2, c=amplitude_2,s=12,cmap='inferno')
plt.xlabel('x position (um)')
plt.xlim([-600,600])
plt.ylabel('y position (um)')
plt.ylim([-600,600])
plt.axis('equal')
plt.colorbar()
plt.clim([0.0e-6,2e-6])
plt.show()


plt.close(subfig3)
plt.figure(subfig3,figsize=(4,3))
plt.scatter(plot_x_3,plot_y_3, c=amplitude_3,s=12,cmap='inferno')
plt.xlabel('x position (um)')
plt.xlim([-600,600])
plt.ylabel('y position (um)')
plt.ylim([-600,600])
plt.axis('equal')
plt.colorbar()
plt.clim([0.0e-6,2e-6])
plt.show()

plt.close(subfig4)
plt.figure(subfig4,figsize=(4,3))
plt.scatter(plot_x_4,plot_y_4, c=amplitude_4,s=12,cmap='inferno')
plt.xlabel('x position (um)')
plt.xlim([-600,600])
plt.ylabel('y position (um)')
plt.ylim([-600,600])
plt.axis('equal')
plt.colorbar()
plt.clim([0.0e-6,2e-6])
plt.show()

plt.close(subfig5)
plt.figure(subfig5,figsize=(4,3))
plt.scatter(plot_x_5,plot_y_5, c=amplitude_5,s=12,cmap='inferno')
plt.xlabel('x position (um)')
plt.xlim([-600,600])
plt.ylabel('y position (um)')
plt.ylim([-600,600])
plt.axis('equal')
plt.colorbar()
plt.clim([0.0e-6,2e-6])
plt.show()

plt.close(subfig6)
plt.figure(subfig6,figsize=(4,3))
plt.scatter(plot_x_6,plot_y_6, c=amplitude_6,s=12,cmap='inferno')
plt.xlabel('x position (um)')
plt.xlim([-600,600])
plt.ylabel('y position (um)')
plt.ylim([-600,600])
plt.axis('equal')
plt.colorbar()
plt.clim([0.0e-6,2e-6])
plt.show()