From 4d1768e15e87430d5f24ce56ade7b1c078f44816 Mon Sep 17 00:00:00 2001
From: Giorgio Calderone <giorgio.calderone@gmail.com>
Date: Mon, 27 Apr 2020 18:37:59 +0200
Subject: [PATCH] Added contourlines method accepting top fractions

---
 src/Gnuplot.jl | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/src/Gnuplot.jl b/src/Gnuplot.jl
index 10a6c51..ced8932 100644
--- a/src/Gnuplot.jl
+++ b/src/Gnuplot.jl
@@ -2006,11 +2006,31 @@ end
 ```
 """
 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},
                       cntrparam="level auto 10")
     lines = gp_write_table("set contour base", "unset surface",
                            "set cntrparam $cntrparam", x, y, z, is3d=true)
-
     level = NaN
     path = Path2d()
     paths = Vector{Path2d}()