pythonStagYY/06.test.py at main · chingchenn/pythonStagYY · 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
#!/opt/anaconda3/bin/python3
# -*- coding: utf-8 -*-
"""
Created on Wed Mar 15 15:54:16 2023

@author: chingchen
"""

import stagpy
import numpy as np
from stagpy import stagyydata
import matplotlib.pyplot as plt
model = 'test04'
path = '/Users/chingchen/Desktop/data/'
figpath = '/Users/chingchen/Desktop/figure/'
plt.rcParams["font.family"] = "Times New Roman"
shot_list = [1,3,5,7,9,11,13,15,17,19,21,23,25,27,29,31,33,35,37,39]
shot_list = [101]
gif = 0
mp4 = 0
plotting = 0


# for shot in shot_list:
if plotting:
    shot = 101
    kk1 = np.loadtxt(path+model+'_get_meshes_x_of_'+str(shot)+'.txt')
    kk2 = np.loadtxt(path+model+'_get_meshes_z_of_'+str(shot)+'.txt')
    kk3 = np.loadtxt(path+model+'_get_meshes_T_of_'+str(shot)+'.txt')
    eta3 = np.loadtxt(path+model+'_get_meshes_eta_of_'+str(shot)+'.txt')
    rho3 = np.loadtxt(path+model+'_get_meshes_rho_of_'+str(shot)+'.txt')

    fig,(ax,ax2,ax3) = plt.subplots(1,3,figsize=(12,4))

    ax.set_aspect('equal')
    cmap = plt.cm.get_cmap('RdBu_r')
    colorbar = ax.scatter(kk1,kk2,c= kk3,cmap = cmap, vmin = 0,vmax = 1)
    ax.axis('off')
    cax = plt.axes([0.165, 0.05, 0.15, 0.05])
    cc1=fig.colorbar(colorbar, ax=ax,cax=cax,orientation='horizontal')
    cc1.ax.tick_params(labelsize=20)
    cc1.set_label(label='Temperature', size=25)
    cc1.ax.yaxis.set_label_position('left')
    ax.set_title(model+' at snapshot '+str(shot),fontsize = 26)
    ax2.set_aspect('equal')
    cmap = plt.cm.get_cmap('rainbow')
    colorbar = ax2.scatter(kk1,kk2,c= np.log10(eta3),cmap = cmap, vmin = -2.5,vmax = 2.5)
    ax2.axis('off')
    cax = plt.axes([0.445, 0.05, 0.15, 0.05])
    cc2=fig.colorbar(colorbar, ax=ax2,cax=cax,orientation='horizontal')
    cc2.ax.tick_params(labelsize=20)
    cc2.set_label(label='Viscosity', size=25)
    ax3.set_aspect('equal')
    cmap = plt.cm.get_cmap('afmhot_r')
    colorbar = ax3.scatter(kk1,kk2,c= rho3,cmap = cmap, vmin = 0.9,vmax = 1.1)
    ax3.axis('off')
    cax = plt.axes([0.72, 0.05, 0.15, 0.05])
    cc3=fig.colorbar(colorbar, ax=ax3,cax=cax,orientation='horizontal')
    cc3.set_label(label='Density', size=25)
    cc3.ax.tick_params(labelsize=20)
    #fig.savefig(figpath+model+'_'+'snapshot_'+str(shot)+'_field.png')
    #fig.gca()
    #plt.close(fig)
for shot in shot_list:
# if plotting:
    kk1 = np.loadtxt(path+model+'_get_meshes_x_of_'+str(shot)+'.txt')
    kk2 = np.loadtxt(path+model+'_get_meshes_z_of_'+str(shot)+'.txt')
    kk3 = np.loadtxt(path+model+'_get_meshes_T_of_'+str(shot)+'.txt')
    eta3 = np.loadtxt(path+model+'_get_meshes_eta_of_'+str(shot)+'.txt')
    # rho3 = np.loadtxt(path+model+'_get_meshes_rho_of_'+str(shot)+'.txt')

    fig,(ax,ax2) = plt.subplots(1,2,figsize=(8,5))

    ax.set_aspect('equal')
    cmap = plt.cm.get_cmap('RdBu_r')
    colorbar = ax.scatter(kk1,kk2,c= kk3,cmap = cmap, vmin = 0,vmax = 1)
    ax.axis('off')
    cax = plt.axes([0.157, 0.15, 0.3, 0.05])
    cc1=fig.colorbar(colorbar, ax=ax,cax=cax,orientation='horizontal')
    cc1.ax.tick_params(labelsize=15)
    cc1.set_label(label='Temperature', size=15)
    cc1.ax.yaxis.set_label_position('left')
    ax.set_title(model+' at time '+str(shot/1000),fontsize = 26)
    ax2.set_aspect('equal')
    cmap = plt.cm.get_cmap('rainbow')
    colorbar = ax2.scatter(kk1,kk2,c= np.log10(eta3),cmap = cmap, vmin = -2.5,vmax = 2.5)
    ax2.axis('off')
    cax = plt.axes([0.585, 0.15, 0.30, 0.05])
    cc2=fig.colorbar(colorbar, ax=ax2,cax=cax,orientation='horizontal')
    cc2.ax.tick_params(labelsize=15)
    cc2.set_label(label='Viscosity', size=15)
    #fig.savefig(figpath+model+'_'+'snapshot_'+str(shot)+'_field.png')
    #fig.gca()
    #plt.close(fig)
#-----------------------------creat GIF-----------------------------------------
if gif:
    from PIL import Image

    # Create the frames
    frames = []
    for shot in  range(1,39+1,2):
        img=figpath+model+'_'+'snapshot_'+str(shot)+'_field.png'
        new_frame = Image.open(img)
        frames.append(new_frame)

    # Save into a GIF file that loops forever
    frames[0].save(figpath+'png_to_gif.gif', format='GIF', append_images=frames[1:],
                   save_all=True, duration=40, loop=0)

#-----------------------------creat mp4-----------------------------------------
if mp4:
    import moviepy.editor as mp
    clip = mp.VideoFileClip(figpath+'png_to_gif.gif')
    clip.write_videofile(figpath+'fieldStagYY'+model+".mp4")