2018-04-21 22:05

1 files changed, 62 insertions, 0 deletions

diff --git a/presentation/singh_czech.py b/presentation/singh_czech.py
new file mode 100644
index 0000000..398e22e
--- /dev/null
+++ b/presentation/singh_czech.py
@@ -0,0 +1,62 @@
+"""Bandwidth comparison between Singh & Czech."""
+
+
+# --- import --------------------------------------------------------------------------------------
+
+
+import os
+
+import numpy as np
+
+import matplotlib.pyplot as plt
+import matplotlib.patches as patches
+
+import WrightTools as wt
+from WrightTools import datasets
+
+
+# --- define --------------------------------------------------------------------------------------
+
+
+here = os.path.abspath(os.path.dirname(__file__))
+
+
+# --- workspace -----------------------------------------------------------------------------------
+
+
+fig, gs = wt.artists.create_figure(cols=[1, 'cbar'])
+ax = plt.subplot(gs[0, 0])
+
+# get data
+ps = datasets.COLORS.v2p1_MoS2_TrEE_movie
+data = wt.data.from_COLORS(ps)
+data.convert('eV', convert_variables=True)
+data.ai0.signed = False
+data.ai0.clip(min=0, replace='value')
+data = data.chop('w1=wm', 'w2', at={'d2': [-80, 'fs']})[0]
+
+# czech
+ax.pcolor(data)
+
+# singh
+edges = np.array([1660, 1610], dtype=np.float64)  # meV
+edges = wt.units.converter(edges, 'meV', 'eV')
+corner = min(edges)
+width = abs(edges[1] - edges[0])
+rectangle = patches.Rectangle((corner, corner), width, width, fc='m')
+ax.add_patch(rectangle)
+
+# decoration
+ax.set_xlim(corner, data.w1.max())
+ax.set_ylim(corner, data.w2.max())
+wt.artists.set_ax_labels(ax=ax, xlabel=data.w1__e__wm.label, ylabel=data.w2.label)
+ax.grid()
+wt.artists.diagonal_line()
+
+# colorbar
+cax = plt.subplot(gs[0, -1])
+wt.artists.plot_colorbar(cax=cax, label='intensity')
+
+# save
+p = os.path.join(here, 'singh_czech.png')
+plt.savefig(p, bbox_inches='tight')