visualise/vmax_vs_aj_small.py at master · feltor-dev/visualise · 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
import os
import sys
import glob
import json
import numpy as np
from netCDF4 import Dataset
import matplotlib.pyplot as plt
import matplotlib as mpl


def pgf4a0line():
    rcParameter = {
        'pgf.texsystem': 'lualatex',
        'text.usetex': True,
        'pgf.rcfonts': False,
        'font.family': 'serif',
        'font.serif': [],
        'font.sans-serif': [],
        'font.monospace': [],
        'font.size': 6.22,
        'axes.labelsize': 6.22,
        'xtick.labelsize': 5.19,
        'ytick.labelsize': 5.19,
        'lines.linewidth' : .3,
        'axes.linewidth': .6,
        'xtick.major.size' : 3,
        'xtick.minor.size' : 1,
        'xtick.major.width' : 0.4,
        'xtick.minor.width' : 0.3,
        'ytick.major.size' : 3,
        'ytick.minor.size' : 1,
        'ytick.major.width' : 0.4,
        'ytick.minor.width' : 0.3,
        'legend.fontsize' : 6.22,
        'pgf.preamble': [
            '\\usepackage{amsmath}',
        ]
    }
    return rcParameter

rcParameter = pgf4a0line()
mpl.rcParams.update(rcParameter)

def wxh_cm(w):
    w = w/2.54
    return (w, w/1.618)

def save_pgf(filename):
    plt.savefig(filename, facecolor='none',
                edgecolor='none', bbox_inches='tight', dpi=600)

path_data = "/media/e3r/usb3hdd2_seagate2tb/phd_run/"
os.chdir(path_data)

prefices = ["muz2_az_", "muz2p5_az_", "muz5_az_", "muz10_az_", "muz20_az_"]
mu_j = np.array([2, 2.5, 5.0, 10.0, 20.0])
a_j = np.array([1.e-3, 2.e-3, 5.e-3, 1.e-2, 2.e-2, 5.e-2, 1.e-1, 2.e-1, 5.e-1])

vel_x__max = []
t_vel_x__max = []
for prefix in prefices:
    v_max, t_max = [], []
    for i, val in enumerate(a_j):
        filename = prefix+"{}_analysed.nc".format(i)
        with Dataset(filename, "r") as data:
            v_max.append(np.amax(data["electrons_analysed"]["velX"][:]))
            t_max.append(data["time"][np.argmax(data["electrons_analysed"]["velX"][:])])
    vel_x__max.append(np.array(v_max))
    t_vel_x__max.append(np.array(t_max))

marker = ["ko", "kD", "kv", "ks", "k^"]
# raw: a_j : vel_x__max(n_e)
fig, ax = plt.subplots(figsize=wxh_cm(7))
for i, val in enumerate(mu_j):
    ax.plot(a_j, vel_x__max[i], marker[i], markeredgewidth=0.0,
            label="$\\mu_j = {}$".format(mu_j[i]), zorder=10, markersize=3)
ax.set_xscale("log")

leg = ax.legend(loc="lower left", numpoints = 1)
leg.get_frame().set_linewidth(0.0)

ax.set_xlim(5.5E-4, 1.4E+0)
ax.set_ylim(0.008, 0.062)
dax = 0.03
ax.spines['bottom'].set_position(('axes', -dax*1.618))
ax.spines['left'].set_position(('axes', -dax))
ax.spines['right'].set_visible(False)
ax.spines['top'].set_visible(False)
# Ticks
major_xticks = [1.E-3, 1.E-2, 1.E-1, 1.E+0]
minor_xticks = [2.E-3, 5.E-3, 2.E-2, 5.E-2, 2.E-1, 5.E-1]
major_yticks = [0.01, 0.02, 0.03, 0.04, 0.05, 0.06]
ax.set_xticks(major_xticks)
ax.set_yticks(major_yticks)
ax.yaxis.set_ticks_position('left')
ax.xaxis.set_ticks_position('bottom')
for tick in minor_xticks:
    ax.axvline(tick, linestyle='--', dashes=(3,6), color="gray", linewidth=0.2) # vertical lines
ax.grid(which="major", linestyle='--', dashes=(3,3), color="gray", linewidth=0.2)
ax.set_xlabel("$a_z \; [\,]$", labelpad=3)
ax.set_ylabel("$v^x_{max} \; [c_s]$", labelpad=5)
save_pgf("/home/e3r/oepg_poster2016/handout/scale.pgf")

plt.show()