Label delay space

Using WrightTools to label delay space.

/home/docs/checkouts/readthedocs.org/user_builds/wrighttools/envs/stable/lib/python3.12/site-packages/numpy/_core/numeric.py:2376: RuntimeWarning: One of rtol or atol is not valid, atol: inf, rtol: 0
  res = all(isclose(a, b, rtol=rtol, atol=atol, equal_nan=equal_nan))
Correction factor applied to d1
Correction factor applied to d2
Correction factor applied to d1_points
Correction factor applied to d2_points
data created at /tmp/ge7uv021.wt5::/
  axes: ('d1', 'd2')
  shape: (101, 101)
axis d1 converted from ps to fs
axis d2 converted from ps to fs
Correction factor applied to d1
Correction factor applied to d2
Correction factor applied to d1_points
Correction factor applied to d2_points
data created at /tmp/gg1vpo8k.wt5::/
  axes: ('d1', 'd2')
  shape: (101, 101)
axis d1 converted from ps to fs
axis d2 converted from ps to fs

<matplotlib.colorbar.Colorbar object at 0x7ac9f37e4b00>

import matplotlib.pyplot as plt

import WrightTools as wt
from WrightTools import datasets

fig, gs = wt.artists.create_figure(width="double", cols=[1, 1, "cbar"])


def set_lim(ax):
    ax.set_xlim(-175, 175)
    ax.set_ylim(-175, 175)


# traditional delay space
ax = plt.subplot(gs[0, 0])
p = datasets.PyCMDS.d1_d2_000
data = wt.data.from_PyCMDS(p)
data.convert("fs")
data.channels[0].symmetric_root(2)
data.channels[0].normalize()
data.channels[0].clip(min=0, replace="value")
ax.pcolor(data)
wt.diagrams.delay.label_sectors(ax=ax)  # using default labels
set_lim(ax)
ax.set_title(r"$\mathsf{\vec{k}_1 - \vec{k}_2 + \vec{k}_{2^\prime}}$", fontsize=20)

# conjugate delay space
ax = plt.subplot(gs[0, 1])
p = datasets.PyCMDS.d1_d2_001
data = wt.data.from_PyCMDS(p)
data.convert("fs")
data.channels[0].symmetric_root(2)
data.channels[0].normalize()
data.channels[0].clip(min=0, replace="value")
ax.pcolor(data)
labels = ["II", "I", "III", "V", "VI", "IV"]
wt.diagrams.delay.label_sectors(ax=ax, labels=labels)
set_lim(ax)
ax.set_title(r"$\mathsf{\vec{k}_1 + \vec{k}_2 - \vec{k}_{2^\prime}}$", fontsize=20)

# label
wt.artists.set_fig_labels(xlabel=data.d1.label, ylabel=data.d2.label)

# colorbar
cax = plt.subplot(gs[:, -1])
wt.artists.plot_colorbar(cax=cax, label="amplitude")