"""Join figure."""


# --- import --------------------------------------------------------------------------------------


import os

import matplotlib.pyplot as plt

import WrightTools as wt


# --- define --------------------------------------------------------------------------------------


here = os.path.abspath(os.path.dirname(__file__))


# --- workspace -----------------------------------------------------------------------------------


wt.artists.apply_rcparams('publication')

fig, gs = wt.artists.create_figure(width='double', cols=[1, 0, 1, 1, 1, 'cbar'], nrows=4)

root = wt.open(os.path.join(here, 'MX2_TSF.wt5'))

def plot(ax, d):
    d.transform('w1', 'w2')
    d.sig.clip(0, 4)
    ax.pcolor(d)

# II
ax = plt.subplot(gs[0, 0])
plot(ax, root.processed_tsf.divided_tsf.II)

# IS
ax = plt.subplot(gs[1, 0])
plot(ax, root.processed_tsf.divided_tsf.IS)

# SS
ax = plt.subplot(gs[2, 0])
plot(ax, root.processed_tsf.divided_tsf.SS)


# savefig
p = os.path.join(here, 'join.png')
plt.show()