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


import os

import matplotlib.pyplot as plt

import WrightTools as wt


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


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


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


# create ax
fig, gs = wt.artists.create_figure(width='double')
ax = plt.subplot(gs[0, 0])

ax.set_xlim(0, 1.25)
ax.set_ylim(0, 1.25)


es = {}
es['MR-CMDS'] = (1, 0.3)
es['AFM'] = (0.5, 0.4)
es['GRAPES'] = (0.1, 1)
es['FTIR'] = (0.1, 0.1)
es['z-scan'] = (0.1, 0.2)
es['UV-VIS'] = (0.1, 0.3)
es['TRPL'] = (0.1, 0.4)
es['NMR'] = (0.1, 0.5)



for label, coordinates in es.items():
    ax.text(*coordinates, label, fontsize=20)

# decoration
xlabel = r'hardware complexity $\rightarrow$'
ylabel = r'measurement complexity $\rightarrow$'
wt.artists.set_ax_labels(xlabel=xlabel, ylabel=ylabel, xticks=False, yticks=False)

# save
p = os.path.join(here, 'quad.png')
wt.artists.savefig(p)