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

Added contourlines method accepting top fractions

Browse Source
This commit is contained in:
Giorgio Calderone 2020-04-27 18:37:59 +02:00
parent 8e140ee998
commit 4d1768e15e
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/Gnuplot.jl
View File

@ -2006,11 +2006,31 @@ end
``` ```
""" """
contourlines(h::Histogram2D, args...) = contourlines(h.bins1, h.bins2, h.counts, args...) contourlines(h::Histogram2D, args...) = contourlines(h.bins1, h.bins2, h.counts, args...)
function contourlines(x::AbstractVector{Float64}, y::AbstractVector{Float64}, z::AbstractMatrix{Float64},
fraction::Vector{Float64})
@assert minimum(fraction) > 0
@assert maximum(fraction) < 1
@assert length(fraction) >= 1
# The following is necessary since `countourlines` return levels
# sorted in increasing order, corresponding to decreasing order
# top fractions.
@assert issorted(fraction, rev=true) "`fraction` must be sorted in decreasing order"
i = sortperm(z[:], rev=true)
topfrac = cumsum(z[i]) ./ sum(z)
selection = Int[]
for f in fraction
push!(selection, minimum(findall(topfrac .>= f)))
end
levels = z[i[selection]]
clines = contourlines(x, y, z, "levels discrete " * join(string.(levels), ", "))
end
function contourlines(x::AbstractVector{Float64}, y::AbstractVector{Float64}, z::AbstractMatrix{Float64}, function contourlines(x::AbstractVector{Float64}, y::AbstractVector{Float64}, z::AbstractMatrix{Float64},
cntrparam="level auto 10") cntrparam="level auto 10")
lines = gp_write_table("set contour base", "unset surface", lines = gp_write_table("set contour base", "unset surface",
"set cntrparam $cntrparam", x, y, z, is3d=true) "set cntrparam $cntrparam", x, y, z, is3d=true)
level = NaN level = NaN
path = Path2d() path = Path2d()
paths = Vector{Path2d}() paths = Vector{Path2d}()