"""
Bruker cw ESR Data
==================

Load cw Bruker ESR data, both 1D and 2D.

Check out the
:ref:`Simple UV <sphx_glr_auto_examples_UV_Cary_simple.py>`
example to understand how
pySpecData locates the file here.
"""

import matplotlib.pyplot as plt
import pyspecdata as psd

# %%
# Load some 1D ESR data with harmonic + phase info.
# The data is initial organized into two dimensions -- `harmonic` and $B_0$.
#

d = psd.find_file(
    "S175R1a.*DHPC.*today.*200304.*",
    exp_type="francklab_esr/Sam",
    zenodo="21287247",
)
print(d.shape)
print(
    "here, we see the harmonic axis contains both harmonic and phase info",
    repr(d.getaxis("harmonic")),
)
d.chunk_auto("harmonic", "phase")

# %%
# `chunk_auto` breaks the `harmonic` dimensions since it was labeled with an
# axis that had 2 fields.

print(d.shape)

plt.figure(1)
psd.plot(d["phase", 0], alpha=0.5)
psd.plot(d["phase", 1], ":", alpha=0.5)
plt.title("1D Data with Multiple Harmonics")

# %%
# Next, let's load some power-saturation data

d = psd.find_file(
    ".*Power.*Sat_200303.*",
    exp_type="francklab_esr/Sam",
    zenodo="21287247",
)
d.chunk_auto("harmonic", "phase")
plt.figure(2)
psd.image(
    d["harmonic", 0]["phase", 0]
    .C.set_axis("Microwave Power", "#")
    .set_units("Microwave Power", "scan #")
)
plt.title("2D Power Saturation")
plt.gca().set_aspect("auto")
plt.show()