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
|
### import ####################################################################
import os
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
matplotlib.rcParams['font.size'] = 14
matplotlib.rcParams['savefig.pad_inches'] = 1
import WrightTools as wt
### define ####################################################################
ddir = wt.kit.get_path_matching('Dissertation')
cmap = wt.artists.colormaps['default']
signed_cmap = wt.artists.colormaps['signed']
### TA single vs dual chopping comparison #####################################
if True:
# get data
single_p = os.path.join(ddir, 'data', 'method development', 'TA dual chopping', 'SCAN [wmw1, d2] 2016.05.30 11_50_06 single', 'SCAN [wmw1, d2] 2016.05.30 11_50_06 single.data')
single = wt.data.from_PyCMDS(single_p)
dual1_p = os.path.join(ddir, 'data', 'method development', 'TA dual chopping', 'SCAN [wmw1, d2] 2016.05.30 12_19_10 dual', 'SCAN [wmw1, d2] 2016.05.30 12_19_10 dual.data')
dual1 = wt.data.from_PyCMDS(dual1_p)
dual2_p = os.path.join(ddir, 'data', 'method development', 'TA dual chopping', 'SCAN [wmw1, d2] 2016.05.30 13_32_42 dual', 'SCAN [wmw1, d2] 2016.05.30 13_32_42 dual.data')
dual2 = wt.data.from_PyCMDS(dual2_p)
datas = [single, dual1, dual2]
pump = wt.units.converter(621.118012, 'nm', 'eV')
# process data
for data in datas:
data.convert('eV')
data.bring_to_front('signal_diff')
data.transpose()
# prepare figure
fig, gs = wt.artists.create_figure(nrows=3, cols=[1, 'cbar'],
default_aspect=0.5, width='single')
# plot methods
def get_vmin_vmax(zi):
extent = max(zi.max(), -zi.min())
return -extent, extent
def plot(data, index, name='name', x=False):
ax = plt.subplot(gs[index, 0])
xi = data.axes[1].points
yi = data.axes[0].points
zi = data.channels[0].values
X, Y, Z = wt.artists.pcolor_helper(xi, yi, zi)
vmin, vmax = get_vmin_vmax(zi)
ax.pcolor(xi, yi, zi, vmin=vmin, vmax=vmax, cmap=signed_cmap)
ax.axhline(0, lw=1, c='k')
ax.set_xlim(xi.min(), xi.max())
ax.set_ylim(yi.min(), yi.max())
wt.artists.corner_text(name, ax=ax, fontsize=18)
ax.grid()
ax.set_ylabel(single.axes[0].get_label(), fontsize=18)
ax.axvline(pump, c='k', lw=4, alpha=0.25)
if x:
single.axes[1].label_seed = ['probe', 'm']
ax.set_xlabel(single.axes[1].get_label(), fontsize=18)
#ax.xaxis.set_label_coords(0.5, 1)
#ax.xaxis.label._clipon = False
else:
plt.setp(ax.get_xticklabels(), visible=False)
# fill plots
plot(single, 0, '500 single')
plot(dual1, 1, '500 dual')
plot(dual2, 2, '1000 dual', x=True)
# colorbar
cax = plt.subplot(gs[:, -1])
ticks = np.linspace(-1, 1, 9)
cbar = wt.artists.plot_colorbar(cax, signed_cmap, ticks=ticks,
label='$\mathsf{dI}$')
# finish
plt.savefig('TA chopping comparison.png', dpi=300, transparent=True,
pad_inches=1)
plt.close(fig)
|
