add method to Base.show for plot pane output (StepTree)

hyrodium · Aug 14, 2023 · 353edd4 · 353edd4
1 parent e0b75e0
commit 353edd4
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/_io.jl b/src/_io.jl
@@ -234,3 +234,14 @@ function export_pinned_steps(
     end
     return paths_output
 end
+
+function Base.show(io::IO, mime::MIME"image/png", steptree::StepTree)
+    dir = mktempdir()
+    index = length(steptree.steps)
+    M = steptree.steps[index].manifold
+    export_one_step(dir, M, index)
+    img = RGB.(load(joinpath(dir, "combined", "combined-$(index).png")))
+    Base.show(io, mime, img)
+end
+
+Base.showable(::MIME"image/png", steptree::StepTree) = !iszero(length(steptree.steps))