openscad/dotSCAD
1
0
Fork 0
mirror of https://github.com/JustinSDK/dotSCAD.git synced 2025-07-31 20:10:36 +02:00
Code Issues Releases Wiki Activity

support key param

Browse Source
This commit is contained in:
Justin Lin
2022-04-29 10:59:17 +08:00
parent 77f74c36ec
commit e4709ab049
3 changed files with 22 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/util/_impl/_sorted_impl.scad
View File

@@ -7,16 +7,24 @@ function before_after(lt, pivot, less, leng, before = [], after = [], j = 1) =
j + 1
);
function _sorted(lt, less) =
identity = function(elem) elem;
function elems(lt, elem) = [for(e = lt) elem(e)];
function _sorted(lt, less, elem = identity) =
let(leng = len(lt))
leng <= 1 ? lt :
leng == 2 ? !less(lt[1], lt[0]) ? lt : [lt[1], lt[0]] :
leng <= 1 ? elems(lt, elem) :
leng == 2 ? !less(lt[1], lt[0]) ? elems(lt, elem) : elems([lt[1], lt[0]], elem) :
let(
pivot = lt[0],
b_a = before_after(lt, pivot, less, leng)
)
[each _sorted(b_a[0], less), pivot, each _sorted(b_a[1], less)];
[each _sorted(b_a[0], less, elem), elem(pivot), each _sorted(b_a[1], less, elem)];
function _sorted_default(lt) = _sorted(lt, function(a, b) a < b);
function _sorted_cmp(lt, cmp) = _sorted(lt, function(a, b) cmp(a, b) < 0);
function _sorted_cmp(lt, cmp) = _sorted(lt, function(a, b) cmp(a, b) < 0);
function _sorted_key(lt, key) =
let(key_elem_lt = [for(elem = lt) [key(elem), elem]])
_sorted(key_elem_lt, function(a, b) a[0] < b[0], function(elem) elem[1]);

6
src/util/sorted.scad
View File

@@ -10,5 +10,7 @@
use <_impl/_sorted_impl.scad>;
function sorted(lt, cmp = undef) =
is_undef(cmp) ? _sorted_default(lt) : _sorted_cmp(lt, cmp);
function sorted(lt, cmp = undef, key = undef) =
!is_undef(cmp) ? _sorted_cmp(lt, cmp) :
!is_undef(key) ? _sorted_key(lt, key) :
_sorted_default(lt);

5
test/util/test_sorted.scad
View File

@@ -21,6 +21,11 @@ module test_sorted() {
descending = function(e1, e2) e2 - e1;
assert(sorted([2, 1, 3, 5, 4], ascending) == [1, 2, 3, 4, 5]);
assert(sorted([2, 1, 3, 5, 4], descending) == [5, 4, 3, 2, 1]);
assert(
[[2, 0, 0], [5, 0, 0], [7, 0, 0], [9, 0, 0], [10, 0, 0]] ==
sorted([[10, 0, 0], [5, 0, 0], [7, 0, 0], [2, 0, 0], [9, 0, 0]], key = function(elem) elem.x)
);
}
test_sorted();