Add support to display PNG for arbitrary types

JuliaImages · May 20, 2024 · 126d2ce · 126d2ce
1 parent 340ebeb
commit 126d2ce
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/src/display.jl b/src/display.jl
@@ -4,7 +4,15 @@ struct TerminalGraphicDisplay{TC<:IO,TS<:IO} <: AbstractDisplay
 end
 TerminalGraphicDisplay(io::IO) = TerminalGraphicDisplay(io, io)
 
-Base.displayable(::TerminalGraphicDisplay, ::MIME"image/png") = true
+Base.displayable(::TerminalGraphicDisplay, ::MIME"image/png", x::Any) = showable("image/png", x)
+Base.displayable(::TerminalGraphicDisplay, ::MIME"image/png", ::Vector{UInt8}) = true
+Base.displayable(::TerminalGraphicDisplay, ::MIME"image/png", ::AbstractArray{<:Colorant}) = true
+
+function Base.display(d::TerminalGraphicDisplay, ::MIME"image/png", x::Any)
+    io = IOBuffer()
+    show(io, "image/png", x)
+    display(d, MIME("image/png"), FileIO.load(io))
+end
 
 function Base.display(d::TerminalGraphicDisplay, ::MIME"image/png", bytes::Vector{UInt8})
     # In this case, assume it to be png byte sequences, use FileIO to find a decoder for it.

diff --git a/test/tst_display.jl b/test/tst_display.jl
@@ -15,4 +15,10 @@
     dsp = ImageInTerminal.TerminalGraphicDisplay(io)
     display(dsp, MIME("image/png"), bytes)
     @test length(read(io, String)) > 5_000
+
+    struct Foo end
+    Base.show(io::IO, ::MIME"image/png", ::Foo) = FileIO.save(Stream{format"PNG"}(io), FileIO.load(fn))
+    dsp = ImageInTerminal.TerminalGraphicDisplay(io)
+    display(dsp, MIME("image/png"), Foo())
+    @test length(read(io, String)) > 5_000
 end