1 files changed, 27 insertions, 0 deletions
diff --git a/processing/fringes_transform.py b/processing/fringes_transform.py
new file mode 100644
index 0000000..1f80a86
--- /dev/null
+++ b/processing/fringes_transform.py
@@ -0,0 +1,27 @@
+import os
+import matplotlib.pyplot as plt
+import WrightTools as wt; from WrightTools import datasets
+
+here = os.path.abspath(os.path.dirname(__file__))
+
+data = wt.data.from_PyCMDS(datasets.PyCMDS.w2_w1_000)
+data.signal_mean.symmetric_root(0.5)  # to amplitude level
+data.convert('wn')
+
+fig, gs = wt.artists.create_figure(width='double', cols=[1, 1, 'cbar'])
+# as taken
+ax = plt.subplot(gs[0, 0])
+ax.pcolor(data)
+wt.artists.set_ax_labels(xlabel=data.w2.label, ylabel=data.w1.label)
+ax.grid(); ax.set_title('as taken', fontsize=20)
+# transformed
+ax = plt.subplot(gs[0, 1])
+data.transform('wm', 'w1')
+data.convert('wn')
+ax.pcolor(data)
+wt.artists.set_ax_labels(xlabel=data.wm.label, yticks=False)
+ax.grid(); ax.set_title('transformed', fontsize=20)
+# colorbar
+wt.artists.plot_colorbar(plt.subplot(gs[0, -1]), label='amplitude')
+# save
+wt.artists.savefig(os.path.join(here, 'fringes_transform.png'))