Added the option to show latex in juopyter notebookks, and some extra code to make sure graphics is the not incorrectky mangled.

Author: fremling

src/display.jl

@@ -1,4 +1,4 @@
-const graphics = (W"Graphics", W"Graphics3D", W"Legended")
+const graphics = (W"Graphics",W"GeoGraphics", W"Graphics3D", W"Legended")
 
 Base.Multimedia.showable(::MIME"image/svg+xml", w::MathLink.WExpr) = in(w.head, graphics)
 Base.show(io::IO, ::MIME"image/svg+xml", w::MathLink.WExpr) = print(io, weval(W"ExportString"(w, "SVG")))
@@ -8,3 +8,73 @@ function Base.show(io::IO, ::MIME"image/png", w::MathLink.WExpr)
     x = weval(W"ExportByteArray"(w, "PNG"))
     write(io, x.args[1].args)
 end
+
+
+export HasGraphicsHead, HasRecursiveGraphicsHead, W2Tex
+
+
+#### Code to produce LaTex strings
+W2Tex(x::WTypes) = weval(W`ToString@TeXForm[#]&`(x))
+
+#### Allow latex string to be shown when supported. Relevant for the jupyter notebook.
+
+HasRecursiveGraphicsHead(w::MathLink.WSymbol) = false
+function HasRecursiveGraphicsHead(w::MathLink.WExpr)
+    if HasGraphicsHead(w)
+        return true
+    end
+    for arg in w.args
+        if typeof(arg) == MathLink.WExpr
+            ##Only check for MathLink Expressions
+            if HasRecursiveGraphicsHead(arg)
+                return true
+            end
+        end
+    end
+    return false
+end
+
+function HeadsEndsWith(HeadString,Target)
+    ###Check if name ends with $Target
+    if length(HeadString) >= length(Target)  
+        return  HeadString[end-(length(Target)-1):end] == Target
+    end
+    return false
+end
+
+
+const graphics_heads = (W"Graphics", W"Graphics3D", W"Legended")
+
+###Check if an expression has a grapics head
+HasGraphicsHead(w::MathLink.WSymbol) = false
+function HasGraphicsHead(w::MathLink.WExpr)
+    HeadString = w.head.name
+    ###Check for graphics related names not based on ending on Plot* or Chart*
+    if in(w.head, graphics)
+        return  true
+    end
+
+    HeadsEndsWith(HeadString,"Plot") && return true
+    HeadsEndsWith(HeadString,"Chart") && return true
+    HeadsEndsWith(HeadString,"Plot3D") && return true
+    HeadsEndsWith(HeadString,"Chart3D") && return true
+    return false
+end
+
+import Base.show
+Base.show(io,::MIME"text/latex",x::MathLink.WSymbol) = print(io,"\$"*W2Tex(x)*"\$")
+
+import Base.Multimedia.showable
+function Base.Multimedia.showable(::MIME"text/latex", w::MathLink.WExpr)
+    if HasRecursiveGraphicsHead(w)
+        return false
+    else
+        return true
+    end
+end
+function Base.show(io,::MIME"text/latex",x::MathLink.WExpr)
+    print(io,"\$"*W2Tex(x)*"\$")
+end
+
+
+

test/runtests.jl

@@ -182,3 +182,43 @@ end
     P14 * Mat
     P14 * Mat* P14
 end
+
+
+
+
+@testset  "Find Graphiscs" begin
+    @test !HasGraphicsHead(W"a")
+    @test HasGraphicsHead(W`Plot[x,{x,0,1}]`)
+    @test HasGraphicsHead(W`ListPlot[x,{x,0,1}]`)
+    @test HasGraphicsHead(W`ListLinePlot3D[x,{x,0,1}]`)
+    @test HasGraphicsHead(W`Plot3D[x,{x,0,1}]`)
+    @test !HasGraphicsHead(W"a"+W"b")
+    @test HasGraphicsHead(weval(W`Plot[x,{x,0,1}]`))
+    @test !HasGraphicsHead(W`{Plot[x,{x,0,1}],Plot[x^2,{x,0,1}]}`)
+    @test !HasGraphicsHead(weval(W`{Plot[x,{x,0,1}],Plot[x^2,{x,0,1}]}`))
+
+    @test !HasRecursiveGraphicsHead(W`{2,a+v,{4+d}}`)
+    @test HasRecursiveGraphicsHead(W`Plot[x,{x,0,1}]`)
+    @test HasRecursiveGraphicsHead(W`{2,Plot[x^2,{x,0,1}]}`)
+    @test HasRecursiveGraphicsHead(W`{a+b,Plot[x^2,{x,0,1}]}`)
+    @test HasRecursiveGraphicsHead(W`{Plot[x,{x,0,1}],Plot[x^2,{x,0,1}]}`)
+    @test HasRecursiveGraphicsHead(W`{1,{Plot[x,{x,0,1}],Plot[x^2,{x,0,1}]}}`)
+    @test HasRecursiveGraphicsHead(weval(W`{Plot[x,{x,0,1}],Plot[x^2,{x,0,1}]}`))
+end
+
+@testset "W2Tex - LaTex conversion" begin
+    @test W2Tex(W`(a+b)^(b+x)`) == "(a+b)^{b+x}"
+    @test W2Tex(W`a`) == "a"
+    @test W2Tex(W`ab`) == "\\text{ab}"
+    @test W2Tex(W`ab*cd`) == "\\text{ab} \\text{cd}"
+
+    ###Testing that MIME form exists for the text/latex option of show.
+    io = IOBuffer();
+    show(IOContext(io, :limit => true, :displaysize => (10, 10)),
+         "text/plain",W"a")
+    @test String(take!(io)) == "W\"a\""
+    show(IOContext(io, :limit => true, :displaysize => (10, 10)),
+         "text/plain",W"a"+W"b")
+    @test String(take!(io)) == "W\"Plus\"(W\"a\", W\"b\")"
+end
+