fix bug of using combinations(::Set, n)

guo-yong-zhi · Jan 19, 2021 · 0d84e57 · 0d84e57
1 parent f601894
commit 0d84e57
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/qtreetools.jl b/src/qtreetools.jl
@@ -124,6 +124,10 @@ function listcollision_native(qtrees::AbstractVector, mask::AbstractStackedQtree
    indpairs = combinations(inds, 2) |> collect |> shuffle!
    listcollision_native(qtrees, mask, indpairs, collist=collist, at=at)
 end
+function listcollision_native(qtrees::AbstractVector, mask::AbstractStackedQtree, 
+    inds::AbstractSet; kargs...)
+   listcollision_native(qtrees, mask, inds|>collect; kargs...)
+end
 
 function findroom(ground, q=[(levelnum(ground), 1, 1)])
     if isempty(q)
@@ -366,4 +370,4 @@ function listcollision(qtrees::AbstractVector, mask::AbstractStackedQtree, args.
     else
         return listcollision_native(qtrees, mask, args...; kargs...)
     end
-end
+end