Plotting helpers for TiledDataset

changelog/408.feature.rst

@@ -0,0 +1 @@
+Add `TiledDataset.plot()` quicklook method.

dkist/data/_sample.py

@@ -57,7 +57,7 @@
     for i, tarpath in enumerate(results):
         output_path = path / file_folder[Path(tarpath).name]
         with tarfile.open(tarpath, "r:*") as tar:
-            tar.extractall(path=output_path, filter="data")
+            tar.extractall(path=output_path, filter=tarfile.fully_trusted_filter)
         results[i] = output_path
 
     return results

dkist/dataset/tests/test_tiled_dataset.py

@@ -1,9 +1,11 @@
 import copy
 
+import matplotlib.pyplot as plt
 import numpy as np
 import pytest
 
-from dkist import Dataset, TiledDataset
+from dkist import Dataset, TiledDataset, load_dataset
+from dkist.tests.helpers import figure_test
 
 
 def test_tiled_dataset(simple_tiled_dataset, dataset):
@@ -59,3 +61,18 @@
         assert ds.files == fm
         assert ds.meta["inventory"] is inventory
         assert ds.meta["headers"] is headers
+
+
+# @pytest.mark.mpl_image_compare
+@figure_test
+@pytest.mark.parametrize("share_scale", [True, False])
+def test_plot(share_scale):
+    from dkist.data.sample import VBI_AJQWW
+    ds = load_dataset(VBI_AJQWW)
+    newtiles = []
+    for tile in ds.flat:
+        newtiles.append(tile.rebin((1, 8, 8), operation=np.sum))
+    ds = TiledDataset(np.array(newtiles).reshape(ds.shape), inventory=ds.inventory)
+    fig = plt.figure(figsize=(600, 800))
+    ds.plot(0, share_scale=share_scale)
+    return plt.gcf()

dkist/dataset/tiled_dataset.py

@@ -7,6 +7,7 @@
 """
 from collections.abc import Collection
 
+import matplotlib.pyplot as plt
 import numpy as np
 
 from astropy.table import vstack
@@ -124,5 +125,30 @@
         """
         return self._data.shape
 
-    # TODO: def plot()
+    def plot(self, slice_index: int, share_scale=False, **kwargs):
+        vmin, vmax = np.inf, 0
+        fig = plt.figure()
+        for i, tile in enumerate(self.flat):
+            ax = fig.add_subplot(self.shape[0], self.shape[1], i+1, projection=tile[0].wcs)
+            tile[slice_index].plot(axes=ax, **kwargs)
+            if i == 0:
+                xlabel = ax.coords[0].get_axislabel() or ax.coords[0]._get_default_axislabel()
+                ylabel = ax.coords[1].get_axislabel() or ax.coords[1]._get_default_axislabel()
+                for coord in ax.coords:
+                    if "b" in coord.axislabels.get_visible_axes():
+                        fig.supxlabel(xlabel, y=0.05)
+                    if "l" in coord.axislabels.get_visible_axes():
+                        fig.supylabel(ylabel, x=0.05)
+            axmin, axmax = ax.get_images()[0].get_clim()
+            vmin = axmin if axmin < vmin else vmin
+            vmax = axmax if axmax > vmax else vmax
+            ax.set_ylabel(" ")
+            ax.set_xlabel(" ")
+        if share_scale:
+            for ax in fig.get_axes():
+                ax.get_images()[0].set_clim(vmin, vmax)
+        timestamp = self[0, 0].axis_world_coords("time")[-1].iso[slice_index]
+        fig.suptitle(f"TiledDataset {self.inventory['datasetId']} at time {timestamp} (slice={slice_index})", y=0.95)
+        return fig
+
     # TODO: def regrid()

dkist/tests/figure_hashes_mpl_391_ft_261_astropy_611_animators_111_ndcube_222.json

@@ -5,5 +5,7 @@
   "dkist.dataset.tests.test_plotting.test_2d_plot[aslice0]": "b31423d5ec45941849564f4ec7e276f2c52a0fa5038ce0b3c8d4af1b7f848a1d",
   "dkist.dataset.tests.test_plotting.test_2d_plot[aslice1]": "cbb84fbae51d8238803f8f0d6820c575f024fe54b1656f1b181dc4ec645e9ff9",
   "dkist.dataset.tests.test_plotting.test_2d_plot[aslice2]": "132c5615832daff457dacb4cb770498f1fbb4460a5b90b5d4d01d224c70eeb28",
-  "dkist.dataset.tests.test_plotting.test_2d_plot2": "409b5a10ad8ccf005331261505e63ce8febdc38eb8b5a34f8863e567e3cccb9c"
+  "dkist.dataset.tests.test_plotting.test_2d_plot2": "409b5a10ad8ccf005331261505e63ce8febdc38eb8b5a34f8863e567e3cccb9c",
+  "dkist.dataset.tests.test_tiled_dataset.test_plot[True]": "1d7b4a34ff6af242460934b8ff4b7d41adcd626eca6993b8ce51d8215043bcdb",
+  "dkist.dataset.tests.test_tiled_dataset.test_plot[False]": "5ac0baf18c66b87107fde794b0b6fae4cf1d78b01d31a5fbe84c23a00df4e19b"
 }

pytest.ini

@@ -31,6 +31,7 @@ addopts =
     --doctest-rst
     -p no:unraisableexception
     -p no:threadexception
+    -m "not mpl_image_compare"
 mpl-results-path = figure_test_images
 mpl-deterministic = true
 filterwarnings =

tox.ini

@@ -58,6 +58,7 @@ extras =
 commands_pre =
     oldestdeps: minimum_dependencies dkist --filename requirements-min.txt
     oldestdeps: pip install -r requirements-min.txt cryptography<42 jsonschema==4.0.1
+    figure: python -c "from dkist.data.sample import download_all_sample_data; download_all_sample_data()"
     pip freeze --all --no-input
 commands =
     figure: /bin/sh -c "mkdir -p ./figure_test_images; python -c 'import matplotlib as mpl; print(mpl.ft2font.__file__, mpl.ft2font.__freetype_version__, mpl.ft2font.__freetype_build_type__)' > ./figure_test_images/figure_version_info.txt"