* readline.scm: moved to ./ice-9/
[bpt/guile.git] / libguile / sort.c
1 /* Copyright (C) 1999,2000,2001,2002 Free Software Foundation, Inc.
2 * This library is free software; you can redistribute it and/or
3 * modify it under the terms of the GNU Lesser General Public
4 * License as published by the Free Software Foundation; either
5 * version 2.1 of the License, or (at your option) any later version.
6 *
7 * This library is distributed in the hope that it will be useful,
8 * but WITHOUT ANY WARRANTY; without even the implied warranty of
9 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
10 * Lesser General Public License for more details.
11 *
12 * You should have received a copy of the GNU Lesser General Public
13 * License along with this library; if not, write to the Free Software
14 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
15 */
16
17
18
19 /* Written in December 1998 by Roland Orre <orre@nada.kth.se>
20 * This implements the same sort interface as slib/sort.scm
21 * for lists and vectors where slib defines:
22 * sorted?, merge, merge!, sort, sort!
23 * For scsh compatibility sort-list and sort-list! are also defined.
24 * In cases where a stable-sort is required use stable-sort or
25 * stable-sort!. An additional feature is
26 * (restricted-vector-sort! vector less? startpos endpos)
27 * which allows you to sort part of a vector.
28 * Thanks to Aubrey Jaffer for the slib/sort.scm library.
29 * Thanks to Richard A. O'Keefe (based on Prolog code by D.H.D.Warren)
30 * for the merge sort inspiration.
31 * Thanks to Douglas C. Schmidt (schmidt@ics.uci.edu) for the
32 * quicksort code.
33 */
34
35 #include "libguile/_scm.h"
36 #include "libguile/eval.h"
37 #include "libguile/unif.h"
38 #include "libguile/ramap.h"
39 #include "libguile/feature.h"
40 #include "libguile/vectors.h"
41 #include "libguile/lang.h"
42
43 #include "libguile/validate.h"
44 #include "libguile/sort.h"
45
46
47 /* The routine quicksort was extracted from the GNU C Library qsort.c
48 written by Douglas C. Schmidt (schmidt@ics.uci.edu)
49 and adapted to guile by adding an extra pointer less
50 to quicksort by Roland Orre <orre@nada.kth.se>.
51
52 The reason to do this instead of using the library function qsort
53 was to avoid dependency of the ANSI-C extensions for local functions
54 and also to avoid obscure pool based solutions.
55
56 This sorting routine is not much more efficient than the stable
57 version but doesn't consume extra memory.
58 */
59
60 #define SWAP(a, b) do { const SCM _tmp = a; a = b; b = _tmp; } while (0)
61
62
63 /* Order size using quicksort. This implementation incorporates
64 four optimizations discussed in Sedgewick:
65
66 1. Non-recursive, using an explicit stack of pointer that store the next
67 array partition to sort. To save time, this maximum amount of space
68 required to store an array of MAX_SIZE_T is allocated on the stack.
69 Assuming a bit width of 32 bits for size_t, this needs only
70 32 * sizeof (stack_node) == 128 bytes. Pretty cheap, actually.
71
72 2. Chose the pivot element using a median-of-three decision tree. This
73 reduces the probability of selecting a bad pivot value and eliminates
74 certain extraneous comparisons.
75
76 3. Only quicksorts NR_ELEMS / MAX_THRESH partitions, leaving insertion sort
77 to order the MAX_THRESH items within each partition. This is a big win,
78 since insertion sort is faster for small, mostly sorted array segments.
79
80 4. The larger of the two sub-partitions is always pushed onto the
81 stack first, with the algorithm then concentrating on the
82 smaller partition. This *guarantees* no more than log (n)
83 stack size is needed (actually O(1) in this case)! */
84
85
86 /* Discontinue quicksort algorithm when partition gets below this size.
87 * This particular magic number was chosen to work best on a Sun 4/260. */
88 #define MAX_THRESH 4
89
90
91 /* Inline stack abstraction: The stack size for quicksorting at most as many
92 * elements as can be given by a value of type size_t is, as described above,
93 * log (MAX_SIZE_T), which is the number of bits of size_t. More accurately,
94 * we would only need ceil (log (MAX_SIZE_T / MAX_THRESH)), but this is
95 * ignored below. */
96
97 /* Stack node declarations used to store unfulfilled partition obligations. */
98 typedef struct {
99 size_t lo;
100 size_t hi;
101 } stack_node;
102
103 #define STACK_SIZE (8 * sizeof (size_t)) /* assume 8 bit char */
104 #define PUSH(low, high) ((void) ((top->lo = (low)), (top->hi = (high)), ++top))
105 #define POP(low, high) ((void) (--top, (low = top->lo), (high = top->hi)))
106 #define STACK_NOT_EMPTY (stack < top)
107
108
109 static void
110 quicksort (SCM *const base_ptr, size_t nr_elems, scm_t_trampoline_2 cmp, SCM less)
111 {
112 static const char s_buggy_less[] = "buggy less predicate used when sorting";
113
114 if (nr_elems == 0)
115 /* Avoid lossage with unsigned arithmetic below. */
116 return;
117
118 if (nr_elems > MAX_THRESH)
119 {
120 size_t lo = 0;
121 size_t hi = nr_elems - 1;
122
123 stack_node stack[STACK_SIZE];
124 stack_node *top = stack + 1;
125
126 while (STACK_NOT_EMPTY)
127 {
128 size_t left;
129 size_t right;
130
131 /* Select median value from among LO, MID, and HI. Rearrange
132 LO and HI so the three values are sorted. This lowers the
133 probability of picking a pathological pivot value and
134 skips a comparison for both the left and right. */
135
136 size_t mid = lo + (hi - lo) / 2;
137
138 if (!SCM_FALSEP ((*cmp) (less, base_ptr[mid], base_ptr[lo])))
139 SWAP (base_ptr[mid], base_ptr[lo]);
140 if (!SCM_FALSEP ((*cmp) (less, base_ptr[hi], base_ptr[mid])))
141 SWAP (base_ptr[mid], base_ptr[hi]);
142 else
143 goto jump_over;
144 if (!SCM_FALSEP ((*cmp) (less, base_ptr[mid], base_ptr[lo])))
145 SWAP (base_ptr[mid], base_ptr[lo]);
146 jump_over:;
147
148 left = lo + 1;
149 right = hi - 1;
150
151 /* Here's the famous ``collapse the walls'' section of quicksort.
152 Gotta like those tight inner loops! They are the main reason
153 that this algorithm runs much faster than others. */
154 do
155 {
156 while (!SCM_FALSEP ((*cmp) (less, base_ptr[left], base_ptr[mid])))
157 {
158 left++;
159 /* The comparison predicate may be buggy */
160 if (left > hi)
161 scm_misc_error (NULL, s_buggy_less, SCM_EOL);
162 }
163
164 while (!SCM_FALSEP ((*cmp) (less, base_ptr[mid], base_ptr[right])))
165 {
166 right--;
167 /* The comparison predicate may be buggy */
168 if (right < lo)
169 scm_misc_error (NULL, s_buggy_less, SCM_EOL);
170 }
171
172 if (left < right)
173 {
174 SWAP (base_ptr[left], base_ptr[right]);
175 left++;
176 right--;
177 }
178 else if (left == right)
179 {
180 left++;
181 right--;
182 break;
183 }
184 }
185 while (left <= right);
186
187 /* Set up pointers for next iteration. First determine whether
188 left and right partitions are below the threshold size. If so,
189 ignore one or both. Otherwise, push the larger partition's
190 bounds on the stack and continue sorting the smaller one. */
191
192 if ((size_t) (right - lo) <= MAX_THRESH)
193 {
194 if ((size_t) (hi - left) <= MAX_THRESH)
195 /* Ignore both small partitions. */
196 POP (lo, hi);
197 else
198 /* Ignore small left partition. */
199 lo = left;
200 }
201 else if ((size_t) (hi - left) <= MAX_THRESH)
202 /* Ignore small right partition. */
203 hi = right;
204 else if ((right - lo) > (hi - left))
205 {
206 /* Push larger left partition indices. */
207 PUSH (lo, right);
208 lo = left;
209 }
210 else
211 {
212 /* Push larger right partition indices. */
213 PUSH (left, hi);
214 hi = right;
215 }
216 }
217 }
218
219 /* Once the BASE_PTR array is partially sorted by quicksort the rest is
220 completely sorted using insertion sort, since this is efficient for
221 partitions below MAX_THRESH size. BASE_PTR points to the beginning of the
222 array to sort, and END idexes the very last element in the array (*not*
223 one beyond it!). */
224
225 {
226 size_t tmp = 0;
227 size_t end = nr_elems - 1;
228 size_t thresh = min (end, MAX_THRESH);
229 size_t run;
230
231 /* Find smallest element in first threshold and place it at the
232 array's beginning. This is the smallest array element,
233 and the operation speeds up insertion sort's inner loop. */
234
235 for (run = tmp + 1; run <= thresh; run++)
236 if (!SCM_FALSEP ((*cmp) (less, base_ptr[run], base_ptr[tmp])))
237 tmp = run;
238
239 if (tmp != 0)
240 SWAP (base_ptr[tmp], base_ptr[0]);
241
242 /* Insertion sort, running from left-hand-side up to right-hand-side. */
243
244 run = 1;
245 while (++run <= end)
246 {
247 tmp = run - 1;
248 while (!SCM_FALSEP ((*cmp) (less, base_ptr[run], base_ptr[tmp])))
249 {
250 /* The comparison predicate may be buggy */
251 if (tmp == 0)
252 scm_misc_error (NULL, s_buggy_less, SCM_EOL);
253
254 tmp--;
255 }
256
257 tmp++;
258 if (tmp != run)
259 {
260 SCM to_insert = base_ptr[run];
261 size_t hi, lo;
262
263 for (hi = lo = run; --lo >= tmp; hi = lo)
264 base_ptr[hi] = base_ptr[lo];
265 base_ptr[hi] = to_insert;
266 }
267 }
268 }
269 }
270
271
272 static scm_t_trampoline_2
273 compare_function (SCM less, unsigned int arg_nr, const char* fname)
274 {
275 const scm_t_trampoline_2 cmp = scm_trampoline_2 (less);
276 SCM_ASSERT_TYPE (cmp != NULL, less, arg_nr, fname, "less predicate");
277 return cmp;
278 }
279
280
281 /* Question: Is there any need to make this a more general array sort?
282 It is probably enough to manage the vector type. */
283 /* endpos equal as for substring, i.e. endpos is not included. */
284 /* More natural with length? */
285
286 SCM_DEFINE (scm_restricted_vector_sort_x, "restricted-vector-sort!", 4, 0, 0,
287 (SCM vec, SCM less, SCM startpos, SCM endpos),
288 "Sort the vector @var{vec}, using @var{less} for comparing\n"
289 "the vector elements. @var{startpos} and @var{endpos} delimit\n"
290 "the range of the vector which gets sorted. The return value\n"
291 "is not specified.")
292 #define FUNC_NAME s_scm_restricted_vector_sort_x
293 {
294 const scm_t_trampoline_2 cmp = compare_function (less, 2, FUNC_NAME);
295 size_t vlen, spos, len;
296 SCM *vp;
297
298 SCM_VALIDATE_VECTOR (1, vec);
299 vp = SCM_WRITABLE_VELTS (vec); /* vector pointer */
300 vlen = SCM_VECTOR_LENGTH (vec);
301
302 SCM_VALIDATE_INUM_MIN_COPY (3, startpos, 0, spos);
303 SCM_ASSERT_RANGE (3, startpos, spos <= vlen);
304 SCM_VALIDATE_INUM_RANGE (4, endpos,0, vlen+1);
305 len = SCM_INUM (endpos) - spos;
306
307 quicksort (&vp[spos], len, cmp, less);
308 scm_remember_upto_here_1 (vec);
309
310 return SCM_UNSPECIFIED;
311 }
312 #undef FUNC_NAME
313
314
315 /* (sorted? sequence less?)
316 * is true when sequence is a list (x0 x1 ... xm) or a vector #(x0 ... xm)
317 * such that for all 1 <= i <= m,
318 * (not (less? (list-ref list i) (list-ref list (- i 1)))). */
319 SCM_DEFINE (scm_sorted_p, "sorted?", 2, 0, 0,
320 (SCM items, SCM less),
321 "Return @code{#t} iff @var{items} is a list or a vector such that\n"
322 "for all 1 <= i <= m, the predicate @var{less} returns true when\n"
323 "applied to all elements i - 1 and i")
324 #define FUNC_NAME s_scm_sorted_p
325 {
326 const scm_t_trampoline_2 cmp = compare_function (less, 2, FUNC_NAME);
327 long len, j; /* list/vector length, temp j */
328 SCM item, rest; /* rest of items loop variable */
329 SCM const *vp;
330
331 if (SCM_NULL_OR_NIL_P (items))
332 return SCM_BOOL_T;
333
334 if (SCM_CONSP (items))
335 {
336 len = scm_ilength (items); /* also checks that it's a pure list */
337 SCM_ASSERT_RANGE (1, items, len >= 0);
338 if (len <= 1)
339 return SCM_BOOL_T;
340
341 item = SCM_CAR (items);
342 rest = SCM_CDR (items);
343 j = len - 1;
344 while (j > 0)
345 {
346 if (!SCM_FALSEP ((*cmp) (less, SCM_CAR (rest), item)))
347 return SCM_BOOL_F;
348 else
349 {
350 item = SCM_CAR (rest);
351 rest = SCM_CDR (rest);
352 j--;
353 }
354 }
355 return SCM_BOOL_T;
356 }
357 else
358 {
359 SCM_VALIDATE_VECTOR (1, items);
360
361 vp = SCM_VELTS (items); /* vector pointer */
362 len = SCM_VECTOR_LENGTH (items);
363 j = len - 1;
364 while (j > 0)
365 {
366 if (!SCM_FALSEP ((*cmp) (less, vp[1], vp[0])))
367 return SCM_BOOL_F;
368 else
369 {
370 vp++;
371 j--;
372 }
373 }
374 return SCM_BOOL_T;
375 }
376
377 return SCM_BOOL_F;
378 }
379 #undef FUNC_NAME
380
381
382 /* (merge a b less?)
383 takes two lists a and b such that (sorted? a less?) and (sorted? b less?)
384 and returns a new list in which the elements of a and b have been stably
385 interleaved so that (sorted? (merge a b less?) less?).
386 Note: this does _not_ accept vectors. */
387 SCM_DEFINE (scm_merge, "merge", 3, 0, 0,
388 (SCM alist, SCM blist, SCM less),
389 "Merge two already sorted lists into one.\n"
390 "Given two lists @var{alist} and @var{blist}, such that\n"
391 "@code{(sorted? alist less?)} and @code{(sorted? blist less?)},\n"
392 "return a new list in which the elements of @var{alist} and\n"
393 "@var{blist} have been stably interleaved so that\n"
394 "@code{(sorted? (merge alist blist less?) less?)}.\n"
395 "Note: this does _not_ accept vectors.")
396 #define FUNC_NAME s_scm_merge
397 {
398 SCM build;
399
400 if (SCM_NULL_OR_NIL_P (alist))
401 return blist;
402 else if (SCM_NULL_OR_NIL_P (blist))
403 return alist;
404 else
405 {
406 const scm_t_trampoline_2 cmp = compare_function (less, 3, FUNC_NAME);
407 long alen, blen; /* list lengths */
408 SCM last;
409
410 SCM_VALIDATE_NONEMPTYLIST_COPYLEN (1, alist, alen);
411 SCM_VALIDATE_NONEMPTYLIST_COPYLEN (2, blist, blen);
412 if (!SCM_FALSEP ((*cmp) (less, SCM_CAR (blist), SCM_CAR (alist))))
413 {
414 build = scm_cons (SCM_CAR (blist), SCM_EOL);
415 blist = SCM_CDR (blist);
416 blen--;
417 }
418 else
419 {
420 build = scm_cons (SCM_CAR (alist), SCM_EOL);
421 alist = SCM_CDR (alist);
422 alen--;
423 }
424 last = build;
425 while ((alen > 0) && (blen > 0))
426 {
427 if (!SCM_FALSEP ((*cmp) (less, SCM_CAR (blist), SCM_CAR (alist))))
428 {
429 SCM_SETCDR (last, scm_cons (SCM_CAR (blist), SCM_EOL));
430 blist = SCM_CDR (blist);
431 blen--;
432 }
433 else
434 {
435 SCM_SETCDR (last, scm_cons (SCM_CAR (alist), SCM_EOL));
436 alist = SCM_CDR (alist);
437 alen--;
438 }
439 last = SCM_CDR (last);
440 }
441 if ((alen > 0) && (blen == 0))
442 SCM_SETCDR (last, alist);
443 else if ((alen == 0) && (blen > 0))
444 SCM_SETCDR (last, blist);
445 }
446 return build;
447 }
448 #undef FUNC_NAME
449
450
451 static SCM
452 scm_merge_list_x (SCM alist, SCM blist,
453 long alen, long blen,
454 scm_t_trampoline_2 cmp, SCM less)
455 {
456 SCM build, last;
457
458 if (SCM_NULL_OR_NIL_P (alist))
459 return blist;
460 else if (SCM_NULL_OR_NIL_P (blist))
461 return alist;
462 else
463 {
464 if (!SCM_FALSEP ((*cmp) (less, SCM_CAR (blist), SCM_CAR (alist))))
465 {
466 build = blist;
467 blist = SCM_CDR (blist);
468 blen--;
469 }
470 else
471 {
472 build = alist;
473 alist = SCM_CDR (alist);
474 alen--;
475 }
476 last = build;
477 while ((alen > 0) && (blen > 0))
478 {
479 if (!SCM_FALSEP ((*cmp) (less, SCM_CAR (blist), SCM_CAR (alist))))
480 {
481 SCM_SETCDR (last, blist);
482 blist = SCM_CDR (blist);
483 blen--;
484 }
485 else
486 {
487 SCM_SETCDR (last, alist);
488 alist = SCM_CDR (alist);
489 alen--;
490 }
491 last = SCM_CDR (last);
492 }
493 if ((alen > 0) && (blen == 0))
494 SCM_SETCDR (last, alist);
495 else if ((alen == 0) && (blen > 0))
496 SCM_SETCDR (last, blist);
497 }
498 return build;
499 } /* scm_merge_list_x */
500
501
502 SCM_DEFINE (scm_merge_x, "merge!", 3, 0, 0,
503 (SCM alist, SCM blist, SCM less),
504 "Takes two lists @var{alist} and @var{blist} such that\n"
505 "@code{(sorted? alist less?)} and @code{(sorted? blist less?)} and\n"
506 "returns a new list in which the elements of @var{alist} and\n"
507 "@var{blist} have been stably interleaved so that\n"
508 " @code{(sorted? (merge alist blist less?) less?)}.\n"
509 "This is the destructive variant of @code{merge}\n"
510 "Note: this does _not_ accept vectors.")
511 #define FUNC_NAME s_scm_merge_x
512 {
513 if (SCM_NULL_OR_NIL_P (alist))
514 return blist;
515 else if (SCM_NULL_OR_NIL_P (blist))
516 return alist;
517 else
518 {
519 const scm_t_trampoline_2 cmp = compare_function (less, 3, FUNC_NAME);
520 long alen, blen; /* list lengths */
521 SCM_VALIDATE_NONEMPTYLIST_COPYLEN (1, alist, alen);
522 SCM_VALIDATE_NONEMPTYLIST_COPYLEN (2, blist, blen);
523 return scm_merge_list_x (alist, blist, alen, blen, cmp, less);
524 }
525 }
526 #undef FUNC_NAME
527
528
529 /* This merge sort algorithm is same as slib's by Richard A. O'Keefe.
530 The algorithm is stable. We also tried to use the algorithm used by
531 scsh's merge-sort but that algorithm showed to not be stable, even
532 though it claimed to be.
533 */
534 static SCM
535 scm_merge_list_step (SCM * seq, scm_t_trampoline_2 cmp, SCM less, long n)
536 {
537 SCM a, b;
538
539 if (n > 2)
540 {
541 long mid = n / 2;
542 a = scm_merge_list_step (seq, cmp, less, mid);
543 b = scm_merge_list_step (seq, cmp, less, n - mid);
544 return scm_merge_list_x (a, b, mid, n - mid, cmp, less);
545 }
546 else if (n == 2)
547 {
548 SCM p = *seq;
549 SCM rest = SCM_CDR (*seq);
550 SCM x = SCM_CAR (*seq);
551 SCM y = SCM_CAR (SCM_CDR (*seq));
552 *seq = SCM_CDR (rest);
553 SCM_SETCDR (rest, SCM_EOL);
554 if (!SCM_FALSEP ((*cmp) (less, y, x)))
555 {
556 SCM_SETCAR (p, y);
557 SCM_SETCAR (rest, x);
558 }
559 return p;
560 }
561 else if (n == 1)
562 {
563 SCM p = *seq;
564 *seq = SCM_CDR (p);
565 SCM_SETCDR (p, SCM_EOL);
566 return p;
567 }
568 else
569 return SCM_EOL;
570 } /* scm_merge_list_step */
571
572
573 SCM_DEFINE (scm_sort_x, "sort!", 2, 0, 0,
574 (SCM items, SCM less),
575 "Sort the sequence @var{items}, which may be a list or a\n"
576 "vector. @var{less} is used for comparing the sequence\n"
577 "elements. The sorting is destructive, that means that the\n"
578 "input sequence is modified to produce the sorted result.\n"
579 "This is not a stable sort.")
580 #define FUNC_NAME s_scm_sort_x
581 {
582 long len; /* list/vector length */
583 if (SCM_NULL_OR_NIL_P (items))
584 return items;
585
586 if (SCM_CONSP (items))
587 {
588 const scm_t_trampoline_2 cmp = compare_function (less, 2, FUNC_NAME);
589 SCM_VALIDATE_LIST_COPYLEN (1, items, len);
590 return scm_merge_list_step (&items, cmp, less, len);
591 }
592 else if (SCM_VECTORP (items))
593 {
594 len = SCM_VECTOR_LENGTH (items);
595 scm_restricted_vector_sort_x (items,
596 less,
597 SCM_MAKINUM (0L),
598 SCM_MAKINUM (len));
599 return items;
600 }
601 else
602 SCM_WRONG_TYPE_ARG (1, items);
603 }
604 #undef FUNC_NAME
605
606
607 SCM_DEFINE (scm_sort, "sort", 2, 0, 0,
608 (SCM items, SCM less),
609 "Sort the sequence @var{items}, which may be a list or a\n"
610 "vector. @var{less} is used for comparing the sequence\n"
611 "elements. This is not a stable sort.")
612 #define FUNC_NAME s_scm_sort
613 {
614 if (SCM_NULL_OR_NIL_P (items))
615 return items;
616
617 if (SCM_CONSP (items))
618 {
619 const scm_t_trampoline_2 cmp = compare_function (less, 2, FUNC_NAME);
620 long len;
621
622 SCM_VALIDATE_LIST_COPYLEN (1, items, len);
623 items = scm_list_copy (items);
624 return scm_merge_list_step (&items, cmp, less, len);
625 }
626 #if SCM_HAVE_ARRAYS
627 /* support ordinary vectors even if arrays not available? */
628 else if (SCM_VECTORP (items))
629 {
630 long len = SCM_VECTOR_LENGTH (items);
631 SCM sortvec = scm_make_uve (len, scm_array_prototype (items));
632
633 scm_array_copy_x (items, sortvec);
634 scm_restricted_vector_sort_x (sortvec,
635 less,
636 SCM_MAKINUM (0L),
637 SCM_MAKINUM (len));
638 return sortvec;
639 }
640 #endif
641 else
642 SCM_WRONG_TYPE_ARG (1, items);
643 }
644 #undef FUNC_NAME
645
646
647 static void
648 scm_merge_vector_x (SCM vec,
649 SCM * temp,
650 scm_t_trampoline_2 cmp,
651 SCM less,
652 long low,
653 long mid,
654 long high)
655 {
656 long it; /* Index for temp vector */
657 long i1 = low; /* Index for lower vector segment */
658 long i2 = mid + 1; /* Index for upper vector segment */
659
660 /* Copy while both segments contain more characters */
661 for (it = low; (i1 <= mid) && (i2 <= high); ++it)
662 {
663 /*
664 Every call of LESS might invoke GC. For full correctness, we
665 should reset the generation of vecbase and tempbase between
666 every call of less.
667
668 */
669 register SCM *vp = SCM_WRITABLE_VELTS(vec);
670
671 if (!SCM_FALSEP ((*cmp) (less, vp[i2], vp[i1])))
672 temp[it] = vp[i2++];
673 else
674 temp[it] = vp[i1++];
675 }
676
677 {
678 register SCM *vp = SCM_WRITABLE_VELTS(vec);
679
680 /* Copy while first segment contains more characters */
681 while (i1 <= mid)
682 temp[it++] = vp[i1++];
683
684 /* Copy while second segment contains more characters */
685 while (i2 <= high)
686 temp[it++] = vp[i2++];
687
688 /* Copy back from temp to vp */
689 for (it = low; it <= high; ++it)
690 vp[it] = temp[it];
691 }
692 } /* scm_merge_vector_x */
693
694
695 static void
696 scm_merge_vector_step (SCM vp,
697 SCM * temp,
698 scm_t_trampoline_2 cmp,
699 SCM less,
700 long low,
701 long high)
702 {
703 if (high > low)
704 {
705 long mid = (low + high) / 2;
706 scm_merge_vector_step (vp, temp, cmp, less, low, mid);
707 scm_merge_vector_step (vp, temp, cmp, less, mid+1, high);
708 scm_merge_vector_x (vp, temp, cmp, less, low, mid, high);
709 }
710 } /* scm_merge_vector_step */
711
712
713 SCM_DEFINE (scm_stable_sort_x, "stable-sort!", 2, 0, 0,
714 (SCM items, SCM less),
715 "Sort the sequence @var{items}, which may be a list or a\n"
716 "vector. @var{less} is used for comparing the sequence elements.\n"
717 "The sorting is destructive, that means that the input sequence\n"
718 "is modified to produce the sorted result.\n"
719 "This is a stable sort.")
720 #define FUNC_NAME s_scm_stable_sort_x
721 {
722 const scm_t_trampoline_2 cmp = compare_function (less, 2, FUNC_NAME);
723 long len; /* list/vector length */
724
725 if (SCM_NULL_OR_NIL_P (items))
726 return items;
727
728 if (SCM_CONSP (items))
729 {
730 SCM_VALIDATE_LIST_COPYLEN (1, items, len);
731 return scm_merge_list_step (&items, cmp, less, len);
732 }
733 else if (SCM_VECTORP (items))
734 {
735 SCM *temp;
736 len = SCM_VECTOR_LENGTH (items);
737
738 /*
739 the following array does not contain any new references to
740 SCM objects, so we can get away with allocing it on the heap.
741 */
742 temp = scm_malloc (len * sizeof(SCM));
743
744 scm_merge_vector_step (items, temp, cmp, less, 0, len - 1);
745 free(temp);
746 return items;
747 }
748 else
749 SCM_WRONG_TYPE_ARG (1, items);
750 }
751 #undef FUNC_NAME
752
753
754 SCM_DEFINE (scm_stable_sort, "stable-sort", 2, 0, 0,
755 (SCM items, SCM less),
756 "Sort the sequence @var{items}, which may be a list or a\n"
757 "vector. @var{less} is used for comparing the sequence elements.\n"
758 "This is a stable sort.")
759 #define FUNC_NAME s_scm_stable_sort
760 {
761 const scm_t_trampoline_2 cmp = compare_function (less, 2, FUNC_NAME);
762
763 if (SCM_NULL_OR_NIL_P (items))
764 return items;
765
766 if (SCM_CONSP (items))
767 {
768 long len; /* list/vector length */
769
770 SCM_VALIDATE_LIST_COPYLEN (1, items, len);
771 items = scm_list_copy (items);
772 return scm_merge_list_step (&items, cmp, less, len);
773 }
774 #if SCM_HAVE_ARRAYS
775 /* support ordinary vectors even if arrays not available? */
776 else if (SCM_VECTORP (items))
777 {
778 long len = SCM_VECTOR_LENGTH (items);
779 SCM *temp = scm_malloc (len * sizeof (SCM));
780 SCM retvec = scm_make_uve (len, scm_array_prototype (items));
781 scm_array_copy_x (items, retvec);
782
783 scm_merge_vector_step (retvec, temp, cmp, less, 0, len - 1);
784 free (temp);
785 return retvec;
786 }
787 #endif
788 else
789 SCM_WRONG_TYPE_ARG (1, items);
790 }
791 #undef FUNC_NAME
792
793
794 SCM_DEFINE (scm_sort_list_x, "sort-list!", 2, 0, 0,
795 (SCM items, SCM less),
796 "Sort the list @var{items}, using @var{less} for comparing the\n"
797 "list elements. The sorting is destructive, that means that the\n"
798 "input list is modified to produce the sorted result.\n"
799 "This is a stable sort.")
800 #define FUNC_NAME s_scm_sort_list_x
801 {
802 const scm_t_trampoline_2 cmp = compare_function (less, 2, FUNC_NAME);
803 long len;
804
805 SCM_VALIDATE_LIST_COPYLEN (1, items, len);
806 return scm_merge_list_step (&items, cmp, less, len);
807 }
808 #undef FUNC_NAME
809
810
811 SCM_DEFINE (scm_sort_list, "sort-list", 2, 0, 0,
812 (SCM items, SCM less),
813 "Sort the list @var{items}, using @var{less} for comparing the\n"
814 "list elements. This is a stable sort.")
815 #define FUNC_NAME s_scm_sort_list
816 {
817 const scm_t_trampoline_2 cmp = compare_function (less, 2, FUNC_NAME);
818 long len;
819
820 SCM_VALIDATE_LIST_COPYLEN (1, items, len);
821 items = scm_list_copy (items);
822 return scm_merge_list_step (&items, cmp, less, len);
823 }
824 #undef FUNC_NAME
825
826
827 void
828 scm_init_sort ()
829 {
830 #include "libguile/sort.x"
831
832 scm_add_feature ("sort");
833 }
834
835 /*
836 Local Variables:
837 c-file-style: "gnu"
838 End:
839 */