rework display

t-bltg · Sep 2, 2022 · 054cd1e · 054cd1e
1 parent 1bc2281
commit 054cd1e
Showing 1 changed file with 8 additions and 13 deletions.
diff --git a/src/init.jl b/src/init.jl
@@ -119,20 +119,15 @@ function __init__()
                 InspectDRBackend,
                 GastonBackend,
             )
-                # showable(MIME("image/png"), Plot{be}) || continue  # will only work for currently loaded backends
-                @eval function _display(plt::Plot{$be})
-                    I = findfirst(
-                        d -> d isa ImageInTerminal.TerminalGraphicDisplay,
-                        Base.Multimedia.displays,
-                    )
-                    dsp = if I === nothing
-                        ImageInTerminal.TerminalGraphicDisplay(stdout)
-                    else
-                        Base.Multimedia.displays[I]
-                    end
-                    buf = IOBuffer()
+                @eval function Base.display(::PlotsDisplay, plt::Plot{$be})
+                    prepare_output(plt)
+                    buf = PipeBuffer()
                     show(buf, MIME("image/png"), plt)
-                    display(dsp, MIME("image/png"), take!(buf))
+                    display(
+                        ImageInTerminal.TerminalGraphicDisplay(stdout),
+                        MIME("image/png"),
+                        read(buf),
+                    )
                 end
             end
         end