From ebe242230ebc80d6b3d90613f1dacd5d61c260f0 Mon Sep 17 00:00:00 2001
From: Blaise Thompson <blaise@untzag.com>
Date: Sat, 21 Apr 2018 22:05:11 -0500
Subject: 2018-04-21 22:05

---
 presentation/singh_czech.py | 62 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)
 create mode 100644 presentation/singh_czech.py

(limited to 'presentation/singh_czech.py')

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')
-- 
cgit v1.2.3