Option for no sorting in caterpillarplot! (#12)

* Zenodo badge

* orderby nothing

* patch bump

* remove extra newline

* use the correct index

* drop support for julia-1.5

* swap to PNG due to SVG bug

* sorting note

* Tighten the code a bit.

Co-authored-by: Douglas Bates <[email protected]>

Author: palday

.github/workflows/ci.yml

@@ -21,7 +21,7 @@ jobs:
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
-        version: [1.5, 1]
+        version: [1]
         arch: [x64]
         os: [ubuntu-18.04] # macos-10.15, windows-2019
     steps:

Project.toml

@@ -1,7 +1,7 @@
 name = "MixedModelsMakie"
 uuid = "b12ae82c-6730-437f-aff9-d2c38332a376"
-authors = ["Phillip Alday <[email protected]> and contributors"]
-version = "0.1.0"
+authors = ["Phillip Alday <[email protected]>, Douglas Bates <[email protected]> and contributors"]
+version = "0.1.1"
 
 [deps]
 AbstractPlotting = "537997a7-5e4e-5d89-9595-2241ea00577e"
@@ -13,7 +13,7 @@ MixedModels = "ff71e718-51f3-5ec2-a782-8ffcbfa3c316"
 AbstractPlotting = "0.18"
 DataFrames = "1"
 MixedModels = "3.6, 4"
-julia = "1.5"
+julia = "1.6"
 
 [extras]
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"

README.md

@@ -4,6 +4,8 @@
 [![Stable Docs][docs-stable-img]][docs-stable-url]
 [![Dev Docs][docs-dev-img]][docs-dev-url]
 [![Codecov](https://codecov.io/gh/palday/MixedModelsMakie.jl/branch/master/graph/badge.svg)](https://codecov.io/gh/palday/MixedModelsMakie.jl)
+[![DOI](https://zenodo.org/badge/337082315.svg)](https://zenodo.org/badge/latestdoi/337082315)
+
 
 [docs-dev-img]: https://img.shields.io/badge/docs-dev-blue.svg
 [docs-dev-url]: https://palday.github.io/MixedModelsMakie.jl/dev

docs/src/api.md

@@ -28,7 +28,7 @@ caterpillar!
 
 ```@example Caterpillar
 using CairoMakie
-CairoMakie.activate!(type = "svg")
+CairoMakie.activate!(type = "png")
 using MixedModels
 using MixedModelsMakie
 sleepstudy = MixedModels.dataset(:sleepstudy)
@@ -43,6 +43,10 @@ caterpillar!(Figure(; resolution=(800,600)), subjre)
 caterpillar!(Figure(; resolution=(800,600)), subjre; orderby=2)
 ```
 
+```@example Caterpillar
+caterpillar!(Figure(; resolution=(800,600)), subjre; orderby=nothing)
+```
+
 ## Shrinkage Plots
 
 ```@docs

src/MixedModelsMakie.jl

@@ -20,4 +20,4 @@ module MixedModelsMakie
     include("caterpillar.jl")
     include("xyplot.jl")
 
-end
+end # module

src/caterpillar.jl

@@ -44,12 +44,19 @@ Add Axes of a caterpillar plot from `r` to `f`.
 
 The order of the levels on the vertical axes is increasing `orderby` column
 of `r.ranef`, usually the `(Intercept)` random effects.
+Setting `orderby=nothing` will disable sorting, i.e. return the levels in the
+order they are stored in.
+
+!!! note
+    Even when not sorting the levels, they might have already been sorted during
+    model matrix construction. If you want impose a particular ordering on the
+    levels, then you must sort the relevant fields in the `RanefInfo` object before
+    calling `caterpillar!`.
 """
 function caterpillar!(f::Figure, r::RanefInfo; orderby=1)
     rr = r.ranef
-    vv = view(rr, :, orderby)
-    ord = sortperm(vv)
     y = axes(rr, 1)
+    ord = isnothing(orderby) ? y : sortperm(view(rr, :, orderby))
     cn = r.cnames
     axs = [Axis(f[1, j]) for j in axes(rr, 2)]
     linkyaxes!(axs...)