mantao/Gnuplot.jl
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Added gp_write_table function

Browse Source
This commit is contained in:
Giorgio Calderone 2020-03-31 14:06:58 +02:00
parent a5dab66b31
commit 767a886e98
1 changed files with 23 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
src/Gnuplot.jl
View File

@ -8,7 +8,7 @@ import Base.write
export session_names, dataset_names, palette_names, linetypes, palette, export session_names, dataset_names, palette_names, linetypes, palette,
terminal, terminals, test_terminal, terminal, terminals, test_terminal,
stats, @gp, @gsp, save, stats, @gp, @gsp, save,
contourlines, hist boxxyerror, contourlines, hist
# ╭───────────────────────────────────────────────────────────────────╮ # ╭───────────────────────────────────────────────────────────────────╮
# │ TYPE DEFINITIONS │ # │ TYPE DEFINITIONS │
@ -370,6 +370,21 @@ function getsession(sid::Symbol=options.default)
end end
function gp_write_table(args...; kw...)
tmpfile = Base.Filesystem.tempname()
sid = Symbol("j", Base.Libc.getpid())
gp = getsession(sid)
reset(gp)
exec(sid, "set term unknown")
driver(sid, "set table '$tmpfile'", args...; kw...)
exec(sid, "unset table")
quit(sid)
out = readlines(tmpfile)
rm(tmpfile)
return out
end
# ╭───────────────────────────────────────────────────────────────────╮ # ╭───────────────────────────────────────────────────────────────────╮
# │ write() and writeread() │ # │ write() and writeread() │
# ╰───────────────────────────────────────────────────────────────────╯ # ╰───────────────────────────────────────────────────────────────────╯
@ -620,7 +635,7 @@ function add_dataset(gp::Session, name::String, args...)
return gpsource return gpsource
catch err catch err
if isa(err, MethodError) if isa(err, MethodError)
@warn "No method to write data as a binary file, resort to inline datablock..." # @warn "No method to write data as a binary file, resort to inline datablock..."
else else
rethrow() rethrow()
end end
@ -1487,6 +1502,9 @@ end
# -------------------------------------------------------------------- # --------------------------------------------------------------------
"""
boxxyerror(x, y; xmin=NaN, ymin=NaN, xmax=NaN, ymax=NaN, cartesian=false)
"""
function boxxyerror(x, y; xmin=NaN, ymin=NaN, xmax=NaN, ymax=NaN, cartesian=false) function boxxyerror(x, y; xmin=NaN, ymin=NaN, xmax=NaN, ymax=NaN, cartesian=false)
function box(v; vmin=NaN, vmax=NaN) function box(v; vmin=NaN, vmax=NaN)
vlow = Vector{Float64}(undef, length(v)) vlow = Vector{Float64}(undef, length(v))
@ -1583,25 +1601,14 @@ end
``` ```
""" """
function contourlines(args...; cntrparam="level auto 10") function contourlines(args...; cntrparam="level auto 10")
tmpfile = Base.Filesystem.tempname() lines = gp_write_table("set contour base", "unset surface",
sid = Symbol("j", Base.Libc.getpid()) "set cntrparam $cntrparam", args..., flag3d=true)
if !haskey(Gnuplot.sessions, sid)
gp = getsession(sid)
end
Gnuplot.exec(sid, "set term unknown")
@gsp sid "set contour base" "unset surface" :-
@gsp :- sid "set cntrparam $cntrparam" :-
@gsp :- sid "set table '$tmpfile'" :-
@gsp :- sid args...
Gnuplot.exec(sid, "unset table")
Gnuplot.exec(sid, "reset")
level = NaN level = NaN
path = Path2d() path = Path2d()
paths = Vector{Path2d}() paths = Vector{Path2d}()
levels = Vector{Float64}() levels = Vector{Float64}()
for l in readlines(tmpfile) for l in lines
l = strip(l) l = strip(l)
if (l == "") || if (l == "") ||
!isnothing(findfirst("# Contour ", l)) !isnothing(findfirst("# Contour ", l))
@ -1627,7 +1634,6 @@ function contourlines(args...; cntrparam="level auto 10")
push!(paths, path) push!(paths, path)
push!(levels, level) push!(levels, level)
end end
rm(tmpfile)
@assert length(paths) > 0 @assert length(paths) > 0
i = sortperm(levels) i = sortperm(levels)
paths = paths[ i] paths = paths[ i]