diff options
Diffstat (limited to 'processing/fringes_transform.py')
-rw-r--r-- | processing/fringes_transform.py | 27 |
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')) |