import math
import matplotlib.pyplot as plt
import numpy as np
import matplotlib.image as mpimg
from scipy.interpolate import griddata
from matplotlib import rcParams
from scipy import interpolate
import scipy.interpolate as spi

fileName          = 'EigenPlotDifferDens'
readDir           = r'./'
saveDir2          = r'./'

###Font info
font_eqs = {
    "font.family":'serif',
    "font.size": 65,
    "mathtext.fontset":'stix',
}
ticksize = 65
rcParams.update(font_eqs)
fig, ax = plt.subplots(figsize=(68, 10))
plt.subplots_adjust(left=None, bottom=None, right=None, top=None, wspace=None, hspace=0.5)
###########################################################################
plt.subplot(2,2,1)
f=open(r'We04_rho1_Lam13/EigenVec_rho1_1','r')
x=[]#创建三个列表用于存放点的数据
y=[]
h1=[]
h2=[]
for line in f:
    p=line[:]
    a=p.split(' ')
    x.append(float(a[1]))
    y.append(float(a[2]))
    h1.append(float(a[3]))
    h2.append(float(a[4]))
f.close()

points=[]
for i in range(len(x)):
    point=[]
    point.append(x[i])
    point.append(y[i])   
    points.append(point)
points=np.array(points)

xi=np.linspace(min(x),max(x),500)
yi=np.linspace(min(y),max(y),500)

xi,yi=np.meshgrid(xi,yi)#网格化
zi=griddata(points,h1,(xi,yi),method='cubic')#内插
v = np.linspace(-max(map(max,zi)), max(map(max,zi)), 80, endpoint=True)
Power0 = plt.contourf(xi, yi, zi, v,cmap = 'RdBu_r', extend="both")#plt.cm.coolwarm

## 读取data.txt文件中的x和y数据
#with open('We12.5/interface1.dat', 'r') as f:
    #lines = f.readlines()
    #x = [float(line.split()[0]) for line in lines]
    #y = [float(line.split()[1]) for line in lines]
#plt.plot(x,y,linewidth =4.0,color='#1f77b4')

#with open('We12.5/interface2.dat', 'r') as f:
    #lines = f.readlines()
    #x = [float(line.split()[0]) for line in lines]
    #y = [float(line.split()[1]) for line in lines]
#plt.plot(x,y,linewidth =4.0,color='#1f77b4')

plt.xlim([0,32])
plt.xticks( size=ticksize)
plt.yticks( size=ticksize)
plt.ylabel(r'y', rotation=360, labelpad=10)
plt.title(label=r'$r=1$')

###########################################################################
plt.subplot(2,2,3)
f=open(r'We04_rho1_Lam13/EigenVec_rho1_2','r')
x=[]#创建三个列表用于存放点的数据
y=[]
h1=[]
h2=[]
for line in f:
    p=line[:]
    a=p.split(' ')
    x.append(float(a[1]))
    y.append(float(a[2]))
    h1.append(float(a[3]))
    h2.append(float(a[4]))
f.close()

points=[]
for i in range(len(x)):
    point=[]
    point.append(x[i])
    point.append(y[i])
    points.append(point)
points=np.array(points)

xi=np.linspace(min(x),max(x),500)
yi=np.linspace(min(y),max(y),500)

xi,yi=np.meshgrid(xi,yi)#网格化
zi=griddata(points,h1,(xi,yi),method='cubic')#内插
v = np.linspace(-max(map(max,zi)), max(map(max,zi)), 80, endpoint=True)
Power0 = plt.contourf(xi, yi, zi, v,cmap = 'RdBu_r', extend="both")#plt.cm.coolwarm

## 读取data.txt文件中的x和y数据
#with open('We12.5/interface1.dat', 'r') as f:
    #lines = f.readlines()
    #x = [float(line.split()[0]) for line in lines]
    #y = [float(line.split()[1]) for line in lines]
#plt.plot(x,y,linewidth =4.0,color='#1f77b4')

#with open('We12.5/interface2.dat', 'r') as f:
    #lines = f.readlines()
    #x = [float(line.split()[0]) for line in lines]
    #y = [float(line.split()[1]) for line in lines]
#plt.plot(x,y,linewidth =4.0,color='#1f77b4')

plt.xlim([0,32])
plt.xticks(size=ticksize)
plt.yticks(size=ticksize)
plt.ylabel(r'y', rotation=360, labelpad=10)

###########################################################################
plt.subplot(2,2,2)
f=open(r'We04_rho2_Lam13/EigenVec_rho2_1','r')
x=[]#创建三个列表用于存放点的数据
y=[]
h1=[]
h2=[]
for line in f:
    p=line[:]
    a=p.split(' ')
    x.append(float(a[1]))
    y.append(float(a[2]))
    h1.append(float(a[3]))
    h2.append(float(a[4]))
f.close()

points=[]
for i in range(len(x)):
    point=[]
    point.append(x[i])
    point.append(y[i])
    points.append(point)
points=np.array(points)

xi=np.linspace(min(x),max(x),500)
yi=np.linspace(min(y),max(y),500)

xi,yi=np.meshgrid(xi,yi)#网格化
zi=griddata(points,h1,(xi,yi),method='cubic')#内插
v = np.linspace(-max(map(max,zi)), max(map(max,zi)), 80, endpoint=True)
Power0 = plt.contourf(xi, yi, zi, v,cmap = 'RdBu_r', extend="both")#plt.cm.coolwarm

## 读取data.txt文件中的x和y数据
#with open('We12.5/interface1.dat', 'r') as f:
    #lines = f.readlines()
    #x = [float(line.split()[0]) for line in lines]
    #y = [float(line.split()[1]) for line in lines]
#plt.plot(x,y,linewidth =4.0,color='#1f77b4')

#with open('We12.5/interface2.dat', 'r') as f:
    #lines = f.readlines()
    #x = [float(line.split()[0]) for line in lines]
    #y = [float(line.split()[1]) for line in lines]
#plt.plot(x,y,linewidth =4.0,color='#1f77b4')

plt.xlim([0,32])
plt.xticks(size=ticksize)
plt.yticks(size=ticksize)
plt.ylabel(r'y', rotation=360, labelpad=10)
plt.title(label=r'$r=2$')

###########################################################################
plt.subplot(2,2,4)
f=open(r'We04_rho2_Lam13/EigenVec_rho2_2','r')
x=[]#创建三个列表用于存放点的数据
y=[]
h1=[]
h2=[]
for line in f:
    p=line[:]
    a=p.split(' ')
    x.append(float(a[1]))
    y.append(float(a[2]))
    h1.append(float(a[3]))
    h2.append(float(a[4]))
f.close()

points=[]
for i in range(len(x)):
    point=[]
    point.append(x[i])
    point.append(y[i])
    points.append(point)
points=np.array(points)

xi=np.linspace(min(x),max(x),500)
yi=np.linspace(min(y),max(y),500)

xi,yi=np.meshgrid(xi,yi)#网格化
zi=griddata(points,h1,(xi,yi),method='cubic')#内插
v = np.linspace(-max(map(max,zi)), max(map(max,zi)), 80, endpoint=True)
Power0 = plt.contourf(xi, yi, zi, v,cmap = 'RdBu_r', extend="both")#plt.cm.coolwarm

## 读取data.txt文件中的x和y数据
#with open('We12.5/interface1.dat', 'r') as f:
    #lines = f.readlines()
    #x = [float(line.split()[0]) for line in lines]
    #y = [float(line.split()[1]) for line in lines]
#plt.plot(x,y,linewidth =4.0,color='#1f77b4')

#with open('We12.5/interface2.dat', 'r') as f:
    #lines = f.readlines()
    #x = [float(line.split()[0]) for line in lines]
    #y = [float(line.split()[1]) for line in lines]
#plt.plot(x,y,linewidth =4.0,color='#1f77b4')

plt.xlim([0,32])
plt.xticks(size=ticksize)
plt.yticks(size=ticksize)
plt.ylabel(r'y', rotation=360, labelpad=10)

# plt.rcParams['savefig.dpi'] = 300 #图片像素
# plt.rcParams['figure.dpi'] = 300 #分辨率
# plt.savefig(saveDir2 + fileName + '.eps', format='eps', dpi=500, bbox_inches='tight')