Line | Count | Source |
1 | | /* GLIB - Library of useful routines for C programming |
2 | | * Copyright (C) 1991, 1992, 1996, 1997,1999,2004 Free Software Foundation, Inc. |
3 | | * Copyright (C) 2000 Eazel, Inc. |
4 | | * Copyright (C) 1995-1997 Peter Mattis, Spencer Kimball and Josh MacDonald |
5 | | * |
6 | | * SPDX-License-Identifier: LGPL-2.1-or-later |
7 | | * |
8 | | * This library is free software; you can redistribute it and/or |
9 | | * modify it under the terms of the GNU Lesser General Public |
10 | | * License as published by the Free Software Foundation; either |
11 | | * version 2.1 of the License, or (at your option) any later version. |
12 | | * |
13 | | * This library is distributed in the hope that it will be useful, |
14 | | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
15 | | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
16 | | * Lesser General Public License for more details. |
17 | | * |
18 | | * You should have received a copy of the GNU Lesser General Public |
19 | | * License along with this library; if not, see <http://www.gnu.org/licenses/>. |
20 | | */ |
21 | | |
22 | | #include "config.h" |
23 | | |
24 | | #include <limits.h> |
25 | | #include <stdlib.h> |
26 | | #include <string.h> |
27 | | #include "galloca.h" |
28 | | #include "gmem.h" |
29 | | |
30 | | #include "gqsort.h" |
31 | | |
32 | | #include "gtestutils.h" |
33 | | |
34 | | /* This file was originally from stdlib/msort.c in gnu libc, just changed |
35 | | to build inside glib and to not fall back to an unstable quicksort |
36 | | for large arrays. */ |
37 | | |
38 | | /* An alternative to qsort, with an identical interface. |
39 | | This file is part of the GNU C Library. |
40 | | Copyright (C) 1992,95-97,99,2000,01,02,04,07 Free Software Foundation, Inc. |
41 | | Written by Mike Haertel, September 1988. |
42 | | |
43 | | The GNU C Library is free software; you can redistribute it and/or |
44 | | modify it under the terms of the GNU Lesser General Public |
45 | | License as published by the Free Software Foundation; either |
46 | | version 2.1 of the License, or (at your option) any later version. |
47 | | |
48 | | The GNU C Library is distributed in the hope that it will be useful, |
49 | | but WITHOUT ANY WARRANTY; without even the implied warranty of |
50 | | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
51 | | Lesser General Public License for more details. |
52 | | |
53 | | You should have received a copy of the GNU Lesser General Public |
54 | | License along with the GNU C Library; if not, see |
55 | | <http://www.gnu.org/licenses/>. */ |
56 | | |
57 | | |
58 | | struct msort_param |
59 | | { |
60 | | size_t s; |
61 | | size_t var; |
62 | | GCompareDataFunc cmp; |
63 | | void *arg; |
64 | | char *t; |
65 | | }; |
66 | | |
67 | | static void msort_with_tmp (const struct msort_param *p, void *b, size_t n); |
68 | | |
69 | | static void |
70 | | msort_with_tmp (const struct msort_param *p, void *b, size_t n) |
71 | 0 | { |
72 | 0 | char *b1, *b2; |
73 | 0 | size_t n1, n2; |
74 | 0 | char *tmp = p->t; |
75 | 0 | const size_t s = p->s; |
76 | 0 | GCompareDataFunc cmp = p->cmp; |
77 | 0 | void *arg = p->arg; |
78 | |
|
79 | 0 | if (n <= 1) |
80 | 0 | return; |
81 | | |
82 | 0 | n1 = n / 2; |
83 | 0 | n2 = n - n1; |
84 | 0 | b1 = b; |
85 | 0 | b2 = (char *) b + (n1 * p->s); |
86 | |
|
87 | 0 | msort_with_tmp (p, b1, n1); |
88 | 0 | msort_with_tmp (p, b2, n2); |
89 | |
|
90 | 0 | switch (p->var) |
91 | 0 | { |
92 | 0 | case 0: |
93 | 0 | while (n1 > 0 && n2 > 0) |
94 | 0 | { |
95 | 0 | if ((*cmp) (b1, b2, arg) <= 0) |
96 | 0 | { |
97 | 0 | *(guint32 *) tmp = *(guint32 *) b1; |
98 | 0 | b1 += sizeof (guint32); |
99 | 0 | --n1; |
100 | 0 | } |
101 | 0 | else |
102 | 0 | { |
103 | 0 | *(guint32 *) tmp = *(guint32 *) b2; |
104 | 0 | b2 += sizeof (guint32); |
105 | 0 | --n2; |
106 | 0 | } |
107 | 0 | tmp += sizeof (guint32); |
108 | 0 | } |
109 | 0 | break; |
110 | 0 | case 1: |
111 | 0 | while (n1 > 0 && n2 > 0) |
112 | 0 | { |
113 | 0 | if ((*cmp) (b1, b2, arg) <= 0) |
114 | 0 | { |
115 | 0 | *(guint64 *) tmp = *(guint64 *) b1; |
116 | 0 | b1 += sizeof (guint64); |
117 | 0 | --n1; |
118 | 0 | } |
119 | 0 | else |
120 | 0 | { |
121 | 0 | *(guint64 *) tmp = *(guint64 *) b2; |
122 | 0 | b2 += sizeof (guint64); |
123 | 0 | --n2; |
124 | 0 | } |
125 | 0 | tmp += sizeof (guint64); |
126 | 0 | } |
127 | 0 | break; |
128 | 0 | case 2: |
129 | 0 | while (n1 > 0 && n2 > 0) |
130 | 0 | { |
131 | 0 | guintptr *tmpl = (guintptr *) tmp; |
132 | 0 | guintptr *bl; |
133 | |
|
134 | 0 | tmp += s; |
135 | 0 | if ((*cmp) (b1, b2, arg) <= 0) |
136 | 0 | { |
137 | 0 | bl = (guintptr *) b1; |
138 | 0 | b1 += s; |
139 | 0 | --n1; |
140 | 0 | } |
141 | 0 | else |
142 | 0 | { |
143 | 0 | bl = (guintptr *) b2; |
144 | 0 | b2 += s; |
145 | 0 | --n2; |
146 | 0 | } |
147 | 0 | while (tmpl < (guintptr *) tmp) |
148 | 0 | *tmpl++ = *bl++; |
149 | 0 | } |
150 | 0 | break; |
151 | 0 | case 3: |
152 | 0 | while (n1 > 0 && n2 > 0) |
153 | 0 | { |
154 | 0 | if ((*cmp) (*(const void **) b1, *(const void **) b2, arg) <= 0) |
155 | 0 | { |
156 | 0 | *(void **) tmp = *(void **) b1; |
157 | 0 | b1 += sizeof (void *); |
158 | 0 | --n1; |
159 | 0 | } |
160 | 0 | else |
161 | 0 | { |
162 | 0 | *(void **) tmp = *(void **) b2; |
163 | 0 | b2 += sizeof (void *); |
164 | 0 | --n2; |
165 | 0 | } |
166 | 0 | tmp += sizeof (void *); |
167 | 0 | } |
168 | 0 | break; |
169 | 0 | default: |
170 | 0 | while (n1 > 0 && n2 > 0) |
171 | 0 | { |
172 | 0 | if ((*cmp) (b1, b2, arg) <= 0) |
173 | 0 | { |
174 | 0 | memcpy (tmp, b1, s); |
175 | 0 | tmp += s; |
176 | 0 | b1 += s; |
177 | 0 | --n1; |
178 | 0 | } |
179 | 0 | else |
180 | 0 | { |
181 | 0 | memcpy (tmp, b2, s); |
182 | 0 | tmp += s; |
183 | 0 | b2 += s; |
184 | 0 | --n2; |
185 | 0 | } |
186 | 0 | } |
187 | 0 | break; |
188 | 0 | } |
189 | | |
190 | 0 | if (n1 > 0) |
191 | 0 | memcpy (tmp, b1, n1 * s); |
192 | 0 | memcpy (b, p->t, (n - n2) * s); |
193 | 0 | } |
194 | | |
195 | | |
196 | | static void |
197 | | msort_r (void *b, size_t n, size_t s, GCompareDataFunc cmp, void *arg) |
198 | 0 | { |
199 | 0 | size_t size = n * s; |
200 | 0 | char *tmp = NULL; |
201 | 0 | struct msort_param p; |
202 | | |
203 | | /* For large object sizes use indirect sorting. */ |
204 | 0 | if (s > 32) |
205 | 0 | size = 2 * n * sizeof (void *) + s; |
206 | |
|
207 | 0 | if (size < 1024) |
208 | | /* The temporary array is small, so put it on the stack. */ |
209 | 0 | p.t = g_alloca (size); |
210 | 0 | else |
211 | 0 | { |
212 | | /* It's large, so malloc it. */ |
213 | 0 | tmp = g_malloc (size); |
214 | 0 | p.t = tmp; |
215 | 0 | } |
216 | |
|
217 | 0 | p.s = s; |
218 | 0 | p.var = 4; |
219 | 0 | p.cmp = cmp; |
220 | 0 | p.arg = arg; |
221 | |
|
222 | 0 | if (s > 32) |
223 | 0 | { |
224 | | /* Indirect sorting. */ |
225 | 0 | char *ip = (char *) b; |
226 | 0 | void **tp = (void **) (p.t + n * sizeof (void *)); |
227 | 0 | void **t = tp; |
228 | 0 | void *tmp_storage = (void *) (tp + n); |
229 | 0 | char *kp; |
230 | 0 | size_t i; |
231 | |
|
232 | 0 | while ((void *) t < tmp_storage) |
233 | 0 | { |
234 | 0 | *t++ = ip; |
235 | 0 | ip += s; |
236 | 0 | } |
237 | 0 | p.s = sizeof (void *); |
238 | 0 | p.var = 3; |
239 | 0 | msort_with_tmp (&p, p.t + n * sizeof (void *), n); |
240 | | |
241 | | /* tp[0] .. tp[n - 1] is now sorted, copy around entries of |
242 | | the original array. Knuth vol. 3 (2nd ed.) exercise 5.2-10. */ |
243 | 0 | for (i = 0, ip = (char *) b; i < n; i++, ip += s) |
244 | 0 | if ((kp = tp[i]) != ip) |
245 | 0 | { |
246 | 0 | size_t j = i; |
247 | 0 | char *jp = ip; |
248 | 0 | memcpy (tmp_storage, ip, s); |
249 | |
|
250 | 0 | do |
251 | 0 | { |
252 | 0 | size_t k = (kp - (char *) b) / s; |
253 | 0 | tp[j] = jp; |
254 | 0 | memcpy (jp, kp, s); |
255 | 0 | j = k; |
256 | 0 | jp = kp; |
257 | 0 | kp = tp[k]; |
258 | 0 | } |
259 | 0 | while (kp != ip); |
260 | |
|
261 | 0 | tp[j] = jp; |
262 | 0 | memcpy (jp, tmp_storage, s); |
263 | 0 | } |
264 | 0 | } |
265 | 0 | else |
266 | 0 | { |
267 | 0 | if ((s & (sizeof (guint32) - 1)) == 0 |
268 | 0 | && (gsize) (guintptr) b % G_ALIGNOF(guint32) == 0) |
269 | 0 | { |
270 | 0 | if (s == sizeof (guint32)) |
271 | 0 | p.var = 0; |
272 | 0 | else if (s == sizeof (guint64) |
273 | 0 | && (gsize) (guintptr) b % G_ALIGNOF(guint64) == 0) |
274 | 0 | p.var = 1; |
275 | 0 | else if ((s & (sizeof (void *) - 1)) == 0 |
276 | 0 | && (gsize) (guintptr) b % G_ALIGNOF(void *) == 0) |
277 | 0 | p.var = 2; |
278 | 0 | } |
279 | 0 | msort_with_tmp (&p, b, n); |
280 | 0 | } |
281 | 0 | g_free (tmp); |
282 | 0 | } |
283 | | |
284 | | /** |
285 | | * g_qsort_with_data: |
286 | | * @pbase: (not nullable): start of array to sort |
287 | | * @total_elems: elements in the array |
288 | | * @size: size of each element |
289 | | * @compare_func: (scope call): function to compare elements |
290 | | * @user_data: data to pass to @compare_func |
291 | | * |
292 | | * This is just like the standard C [`qsort()`](man:qsort(3)) function, but |
293 | | * the comparison routine accepts a user data argument |
294 | | * (like [`qsort_r()`](man:qsort_r(3))). |
295 | | * |
296 | | * Unlike `qsort()`, this is guaranteed to be a stable sort (since GLib 2.32). |
297 | | * |
298 | | * Deprecated: 2.82: `total_elems` is too small to represent larger arrays; use |
299 | | * [func@GLib.sort_array] instead |
300 | | */ |
301 | | void |
302 | | g_qsort_with_data (gconstpointer pbase, |
303 | | gint total_elems, |
304 | | gsize size, |
305 | | GCompareDataFunc compare_func, |
306 | | gpointer user_data) |
307 | 0 | { |
308 | 0 | g_sort_array (pbase, total_elems, size, compare_func, user_data); |
309 | 0 | } |
310 | | |
311 | | /** |
312 | | * g_sort_array: |
313 | | * @array: (not nullable) (array length=n_elements): start of array to sort |
314 | | * @n_elements: number of elements in the array |
315 | | * @element_size: size of each element |
316 | | * @compare_func: (scope call): function to compare elements |
317 | | * @user_data: data to pass to @compare_func |
318 | | * |
319 | | * This is just like the standard C [`qsort()`](man:qsort(3)) function, but |
320 | | * the comparison routine accepts a user data argument |
321 | | * (like [`qsort_r()`](man:qsort_r(3))). |
322 | | * |
323 | | * Unlike `qsort()`, this is guaranteed to be a stable sort. |
324 | | * |
325 | | * Since: 2.82 |
326 | | */ |
327 | | void |
328 | | g_sort_array (const void *array, |
329 | | size_t n_elements, |
330 | | size_t element_size, |
331 | | GCompareDataFunc compare_func, |
332 | | void *user_data) |
333 | 0 | { |
334 | 0 | msort_r ((void *) array, n_elements, element_size, compare_func, user_data); |
335 | 0 | } |