Two-Phase-Flow-FSI/plotRestabilization_a.py at master · jbrillon/Two-Phase-Flow-FSI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
# Julien K.L. Brillon
# Purdue University
#-----------------------------------------------------
# Import libraries
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.lines import Line2D
from matplotlib import rc as matplotlibrc
matplotlibrc('text.latex', preamble='\usepackage{color}')
matplotlibrc('text',usetex=True)
matplotlibrc('font', family='serif')
#-----------------------------------------------------
from var import *
#-----------------------------------------------------
# Plot parameters
#-----------------------------------------------------
clr = ['b','r','g','tab:purple','c','k','y']
mrkr = ['s','o','^','v','>','<','d']
#-----------------------------------------------------
# Common
#-----------------------------------------------------
xlimits = [-1,145]
ylimits = [-5,30]
#-----------------------------------------------------
# Common
#-----------------------------------------------------
# u_mkr_values = [2.0,4.0,8.0,12.0]
u_mkr_values = []
u_mkr_values = np.array(u_mkr_values)
u_mkr_values = list(u_mkr_values)
#-----------------------------------------------------
modes = range(1,Nmodes+1)
names = [None]*Nmodes
for l in range(0,Nmodes):
    names[l] = "Mode %i" % modes[l]

subdirectories[0] += 'restabilization/'
#-----------------------------------------------------
print("---------------------------------------------")

figure_fileName = 'argand_restable_a'
figure_title = 'Argand Plot'
print('Plotting: ' + figure_title)
fig, ax = plt.subplots(figsize=(6,6))
plt.grid()
# ax.set_title(figure_title,fontsize=12,fontweight='bold')
ax.set_xlabel(r'Re$(\omega)$',fontsize=14)
ax.set_ylabel(r'Im$(\omega)$',rotation=90,fontsize=14)
ax.set_xlim(xlimits)
ax.set_ylim(ylimits)
ax.axhline(y=0, color='k', linestyle='--')
#-----------------------------------------------------
# Load data -- eps = 0.00
#-----------------------------------------------------
subsubdir = 'eps000/'
data_fileName = 'u_store'
u = np.loadtxt(subdirectories[0]+subsubdir+data_fileName+'.'+data_fileType, unpack=False)
# load complex
data_fileName = 'real_w_store'
real_omega = np.loadtxt(subdirectories[0]+subsubdir+data_fileName+'.'+data_fileType, unpack=False)
data_fileName = 'imag_w_store'
imag_omega = np.loadtxt(subdirectories[0]+subsubdir+data_fileName+'.'+data_fileType, unpack=False)
omega = real_omega + (1.0j)*imag_omega

nstop = [2999,2000,2000,2000]

for i in range(0,4):
    ax.plot(np.real(omega[:nstop[i],i]), np.imag(omega[:nstop[i],i]),color=clr[i])

#-----------------------------------------------------
# Load data -- eps = 0.01
#-----------------------------------------------------
subsubdir = 'eps010/'
data_fileName = 'u_store'
u = np.loadtxt(subdirectories[0]+subsubdir+data_fileName+'.'+data_fileType, unpack=False)
# load complex
data_fileName = 'real_w_store'
real_omega = np.loadtxt(subdirectories[0]+subsubdir+data_fileName+'.'+data_fileType, unpack=False)
data_fileName = 'imag_w_store'
imag_omega = np.loadtxt(subdirectories[0]+subsubdir+data_fileName+'.'+data_fileType, unpack=False)
omega = real_omega + (1.0j)*imag_omega

for i in range(0,4):
    ax.plot(np.real(omega[:nstop[i],i]), np.imag(omega[:nstop[i],i]),color=clr[i],linestyle='--')

# - - - - - - - - - - - - - - - - - - - - - - - - - - -
#               Annotate velocities
# - - - - - - - - - - - - - - - - - - - - - - - - - - -
data_fileName = 'u_crit'
u_crit = np.loadtxt(subdirectories[0]+subsubdir+data_fileName+'.'+data_fileType, unpack=False)
u_crit[u_crit==-1.0]=0.0
u_crit_flag = np.zeros((4,5),dtype='int64')
u_crit_flag[np.nonzero(u_crit)] = 1
u_crit = u_crit[np.nonzero(u_crit)]

u_mkrs = list(u_mkr_values)
u_mkrs.extend(list(u_crit))
u_mkrs.sort()
u_crit_index = np.argmin(abs(u_crit-u_mkrs))

u_mkrs_index = []
for i,u_val in enumerate(u_mkrs):
    index = np.argmin(abs(np.float64(u_val)-u))
    u_mkrs_index.append(index)

u_mkrs_xy = np.zeros((len(u_mkrs_index),2,4),dtype='float64')
for i in range(0,4):
    u_mkrs_xy[:,0,i] = np.real(omega[u_mkrs_index,i])
    u_mkrs_xy[:,1,i] = np.imag(omega[u_mkrs_index,i])

u_mkrs = np.array(u_mkrs,dtype='float64')

xshifts = 1.03*np.ones((np.size(u_mkrs),4),dtype='float64')
yshifts = 0.9*np.ones((np.size(u_mkrs),4),dtype='float64')

xshifts[0] = -10.0
xshifts[1] = -8.0
xshifts[2] = -2.0

yshifts[0] = 0.5
yshifts[1] = 0.5
yshifts[2] = 0.7

plot_flag = 1
for i in range(3,4):
    for k,u_val in enumerate(u_mkrs):
        if(k==u_crit_index):
            if(u_crit_flag[i,0]!=0):
                plot_flag = 1
            else:
                plot_flag = 0
        else:
            plot_flag = 1

        if(plot_flag != 0):
            xloc = xshifts[k,i]+u_mkrs_xy[k,0,i]
            yloc = yshifts[k,i]+u_mkrs_xy[k,1,i]
            if(((xloc>xlimits[0]) and (xloc<xlimits[1])) and ((yloc>ylimits[0]) and (yloc<ylimits[1]))):
                ax.plot(u_mkrs_xy[k,0,i],u_mkrs_xy[k,1,i],color=clr[i],marker='o',markersize=4, mfc=clr[i], linestyle='None')
                vel_name = '%.1f' % u_val
                t = ax.text(xloc, yloc, vel_name, fontsize=11, fontweight='bold',color=clr[i])
                t.set_bbox(dict(boxstyle='square,pad=0',facecolor='white', alpha=0.7 , edgecolor='none'))
# - - - - - - - - - - - - - - - - - - - - - - - - - - -

plt.text(1, -1.5, r'\textbf{Mode 1}', {'color': clr[0], 'fontsize': 13})
plt.text(31, -1.5, r'\textbf{Mode 2}', {'color': clr[1], 'fontsize': 13})
plt.text(70, 11, r'\textbf{Mode 3}', {'color': clr[2], 'fontsize': 13})
plt.text(121, -1.5, r'\textbf{Mode 4}', {'color': clr[3], 'fontsize': 13})

legend_elements = [Line2D([0],[0], color='k', label=r'Paidoussis (1970) $(\epsilon=0)$'),
                        Line2D([0],[0], color='k', linestyle='--', label=r'Two-phase $(\epsilon=0.1)$')]
leg = plt.legend(handles=legend_elements, loc='best', ncol=1, shadow=True, fancybox=True, fontsize=13)
plt.tight_layout()

print('\t ... Saving figure ...')
plt.savefig(subdirectories[1] + figure_fileName + '.' + figure_fileType,bbox_inches='tight',format=figure_fileType,dpi=500)
plt.close()

# print("---------------------------------------------")
# x_lim = 30
# x_axis_plot = np.linspace(0,x_lim,npoints)
# figure_fileName = 'velocity_imag'
# figure_title = 'Velocity Plot'
# print('Plotting: ' + figure_title)
# fig, ax = plt.subplots(figsize=(7,7))
# plt.grid()
# # ax.set_title(figure_title,fontsize=12,fontweight='bold')
# ax.set_xlabel(r'$u$',fontsize=14)
# ax.set_ylabel(r'$Im(\omega)$',rotation=90,fontsize=14)
# ax.set_xlim([0, x_lim])
# # ax.set_ylim([-3, 6])
# # ax.plot(x_axis_plot,np.zeros(npoints),'k')
# ax.axhline(y=0, color='k', linestyle='--')
# for i in range(0,Nmodes):
#   ax.plot(u, np.imag(omega[:,i]),label=('%i'%i))
# leg = plt.legend(loc='best', ncol=3, shadow=True, fancybox=True, fontsize=13)
# print('\t ... Saving figure ...')
# plt.savefig(subdirectories[1] + figure_fileName + '.' + figure_fileType,bbox_inches='tight',format=figure_fileType,dpi=500)
# plt.close()

print('-----------------------------------------------------')