accelerInt/plotter.py at master · andrewalferman/accelerInt · 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
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Wed Nov 15 2017

@author: andrewalferman
"""

import numpy as np
import csv as csv
from mpl_toolkits.axes_grid1 import host_subplot
import mpl_toolkits.axisartist as AA
#import matplotlib
import matplotlib.pyplot as plt
#matplotlib.use('Agg')
from StiffnessFuncs import *
from CSPfuncs import *
import sys as sys
from matplotlib.ticker import NullFormatter

#plt.ioff()

problem = 'GRIMech'

[ts, ts_timing, Ms, comptimes, CSPstiffness, Y1s, Y2s, Y3s, Y4s, sol, ratios,
    indicators, CEMs] = [[] for i in range(13)]

with open(problem + '.csv', newline='') as csvfile:
    reader = csv.reader(csvfile, delimiter=',')
    for row in reader:
        try:
            ts.append(float(row[0]))
            comptimes.append(float(row[1]))
            Ms.append(float(row[2]))
            CSPstiffness.append(float(row[3]))
            ratios.append(float(row[4]))
            indicators.append(float(row[5]))
            CEMs.append(float(row[6]))
            sol.append([float(row[i]) for i in range(7,len(row))])
        except ValueError:
            ts.append(complex(row[0]).real)
            comptimes.append(complex(row[1]).real)
            Ms.append(complex(row[2]).real)
            CSPstiffness.append(complex(row[3]).real)
            ratios.append(complex(row[4]).real)
            indicators.append(complex(row[5]).real)
            CEMs.append(complex(row[6]).real)
            sol.append([complex(row[i]).real for i in range(7,len(row))])

ts = np.array(ts)
ts_timing = np.array(ts_timing)
Ms = np.array(Ms)
comptimes = np.array(comptimes)
CSPstiffness = np.array(CSPstiffness)
sol = np.array(sol)
ratios = np.array(ratios)
indicators = np.array(indicators)
CEMs = np.array(CEMs)

for i in range(10):
    plt.figure(i)
    plt.clf()
plt.close('all')

figformat = 'png'

posCEM = [i for i in CEMs if i > 0]
if posCEM:
    CEMmin = min(posCEM)
    CEMmax = max(posCEM)

f, axarr = plt.subplots(4, 2, sharex='col', figsize=(9.0, 6.0))
#plt.gca().yaxis.set_minor_formatter(NullFormatter())
plt.suptitle('Plots for ' + problem)
f.add_subplot(111, frameon=False)
plt.tick_params(labelcolor='none', top='off', bottom='off', left='off',
                right='off')
if problem == 'CSPtest':
    showsols = 4
    axarr[0,0].set_xscale('log')
    axarr[0,1].set_xscale('log')
    plt.xlabel('x Value')
elif problem == 'Oregonator':
    showsols = 3
    axarr[0,0].set_yscale('log')
    plt.xlabel('Time')
else:
    showsols = 1
    plt.xlabel('x Value')
for i in range(showsols):
    lab = 'Y' + str(i+1)
    axarr[0,0].plot(ts, sol[:, i], label=lab)
if problem == 'CSPtest' or problem == 'Oregonator':
    axarr[0,0].legend(loc='upper left', bbox_to_anchor=(1,1),
                      fontsize='x-small')
axarr[0,0].set_title('Solution')
axarr[0,1].plot(ts, Ms)
axarr[0,1].set_title('CSP Fast Modes')
axarr[1,0].plot(ts, CSPstiffness)
axarr[1,0].set_title('CSP Stiffness')
axarr[1,0].set_yscale('log')
if max(CSPstiffness) > 1.0:
    axarr[1,0].set_ylim(min(CSPstiffness)*0.6,3.0)
axarr[1,1].plot(ts, indicators)
axarr[1,1].set_title('Stiffness Indicator')
axarr[2,0].plot(ts[:-5], indexes[:-5])
axarr[2,0].set_title('Stiffness Index')
axarr[2,0].set_yscale('log')
axarr[2,1].plot(ts, ratios)
axarr[2,1].set_title('Stiffness Ratio')
axarr[2,1].set_yscale('log')
axarr[3,0].plot(ts, CEMs)
axarr[3,0].set_title('Chemical Explosive Mode')
if posCEM:
    axarr[3,0].set_ylim(CEMmin*0.5, CEMmax*10.0)
    axarr[3,0].set_yscale('log')

# Fine-tune figure; make subplots farther from each other.
f.subplots_adjust(hspace=0.3)
f.subplots_adjust(wspace=0.3)

plt.savefig(problem + '.png', dpi=600)

plt.show()