fix plots

PharmCat · Jul 31, 2024 · 47a06b1 · 47a06b1
1 parent c5951ad
commit 47a06b1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/docs/src/examples.md b/docs/src/examples.md
@@ -76,7 +76,9 @@ png(p, "plot2.png")
 
 p = pkplot(ds; elim = true, ls = true)
 
-png(p[1][2], "plot3.png")
+png(p[1], "plot3.png")
+
+# If pagesort used - return pairs with `Page ID` => `Plot`
 
 p = pkplot(ds; typesort = :Subject, pagesort = :Formulation)
 

diff --git a/src/plots.jl b/src/plots.jl
@@ -391,6 +391,8 @@ PK plot for subject set.
 * `namepref` - name prefix for saving files.
 
 Use `pagesort = MetidaNCA.NoPageSort()` to prevent page plotting.
+
+If pagesort used - return vector of pairs: `Page ID` => `Plot`
 """
 function pkplot(data::DataSet{T};
     typesort::Union{Nothing, Symbol, AbstractVector{Symbol}} = nothing,
@@ -468,7 +470,11 @@ function pkplot(data::DataSet{T};
             end
             if isnothing(namepref) namepref = "plot" end
             for i = 1:length(p) 
-                savefig(p[i][2], joinpath(savepath, namepref*"_$(i).png"))
+                if isa(p[i], Pair)
+                    savefig(p[i][2], joinpath(savepath, namepref*"_$(i).png"))
+                else
+                    savefig(p[i], joinpath(savepath, namepref*"_$(i).png"))
+                end
             end
         else
             @warn "savefig not defined, install Plots.jl for plot writing... plots NOT saved..."