Add sample_layout

Author: dlfivefifty

Project.toml

@@ -1,7 +1,7 @@
 name = "ClassicalOrthogonalPolynomials"
 uuid = "b30e2e7b-c4ee-47da-9d5f-2c5c27239acd"
 authors = ["Sheehan Olver <[email protected]>"]
-version = "0.15.7"
+version = "0.15.8"
 
 
 [deps]

src/roots.jl

@@ -51,11 +51,18 @@ function _searchsortedfirst(::ExpansionLayout{<:AbstractOPLayout}, f, x; iterati
 end
 searchsortedfirst(f::AbstractQuasiVector, x; kwds...) = _searchsortedfirst(MemoryLayout(f), f, x; kwds...)
 
-function sample(f::AbstractQuasiVector, n...)
+sample(f::AbstractQuasiArray, n...) = sample_layout(MemoryLayout(f), f, n...)
+
+function sample_layout(_, f::AbstractQuasiVector, n...)
     g = cumsum(f)
     searchsortedfirst.(Ref(g/last(g)), rand(n...))
 end
 
+function sample_layout(_, f::AbstractQuasiMatrix, n...)
+    @assert size(f,2) == 1 # TODO generalise 
+    sample(f[:,1], n...)
+end
+
 ####
 # min/max/extrema
 ####