5ba608e5ec2a74fc79b9505de564b5fdeaff2ccd
[blender.git] / source / blender / blenlib / intern / sort.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2013 Blender Foundation
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): Benoit Bolsee,
24  *                 Sergey Sharybin.
25  *
26  * ***** END GPL LICENSE BLOCK *****
27  */
28
29 /** \file blender/blenlib/intern/sort.c
30  *  \ingroup bli
31  */
32
33 #include "BLI_utildefines.h"
34
35 #include "BLI_sort.h"
36
37 /* **** qsort based on FreeBSD source (libkern\qsort.c) **** */
38 BLI_INLINE char *med3(char *, char *, char *, BLI_sort_cmp_t, void *);
39 BLI_INLINE void  swapfunc(char *, char *, int, int);
40
41 #define min(a, b)       (a) < (b) ? a : b
42 #define swapcode(TYPE, parmi, parmj, n)         \
43 {                                                                                       \
44         long i = (n) / sizeof(TYPE);                    \
45         TYPE *pi = (TYPE *) (parmi);                    \
46         TYPE *pj = (TYPE *) (parmj);                    \
47         do {                                                                    \
48                 TYPE    t = *pi;                                        \
49                 *pi++ = *pj;                                            \
50                 *pj++ = t;                                                      \
51         } while (--i > 0);                                              \
52 }
53 #define SWAPINIT(a, es) swaptype = ((char *)a - (char *)0) % sizeof(long) || \
54         es % sizeof(long) ? 2 : es == sizeof(long)? 0 : 1;
55
56 BLI_INLINE void swapfunc(char* a, char* b, int n, int swaptype)
57 {
58         if(swaptype <= 1)
59                 swapcode(long, a, b, n)
60         else
61         swapcode(char, a, b, n)
62 }
63
64 #define swap(a, b)                                      \
65         if (swaptype == 0) {                    \
66                 long t = *(long *)(a);          \
67                 *(long *)(a) = *(long *)(b);\
68                 *(long *)(b) = t;                       \
69         } else                                                  \
70                 swapfunc(a, b, es, swaptype)
71
72 #define vecswap(a, b, n)        if ((n) > 0) swapfunc(a, b, n, swaptype)
73 #define CMP(t, x, y) (cmp((t), (x), (y)))
74
75 BLI_INLINE char *med3(char *a, char *b, char *c, BLI_sort_cmp_t cmp, void *thunk)
76 {
77         return CMP(thunk, a, b) < 0 ?
78                 (CMP(thunk, b, c) < 0 ? b : (CMP(thunk, a, c) < 0 ? c : a ))
79                 :(CMP(thunk, b, c) > 0 ? b : (CMP(thunk, a, c) < 0 ? a : c ));
80 }
81
82 void BLI_qsort_r(void *a, size_t n, size_t es, void *thunk, BLI_sort_cmp_t cmp)
83 {
84         char *pa, *pb, *pc, *pd, *pl, *pm, *pn;
85         int d, r, swaptype, swap_cnt;
86
87 loop:
88         SWAPINIT(a, es);
89         swap_cnt = 0;
90         if (n < 7) {
91                 for (pm = (char *)a + es; pm < (char *)a + n * es; pm += es)
92                         for (pl = pm;
93                                 pl > (char *)a && CMP(thunk, pl - es, pl) > 0;
94                                 pl -= es)
95                                 swap(pl, pl - es);
96                 return;
97         }
98         pm = (char *)a + (n / 2) * es;
99         if (n > 7) {
100                 pl = (char *)a;
101                 pn = (char *)a + (n - 1) * es;
102                 if (n > 40) {
103                         d = (n / 8) * es;
104                         pl = med3(pl, pl + d, pl + 2 * d, cmp, thunk);
105                         pm = med3(pm - d, pm, pm + d, cmp, thunk);
106                         pn = med3(pn - 2 * d, pn - d, pn, cmp, thunk);
107                 }
108                 pm = med3(pl, pm, pn, cmp, thunk);
109         }
110         swap((char *)a, pm);
111         pa = pb = (char *)a + es;
112
113         pc = pd = (char *)a + (n - 1) * es;
114         for (;;) {
115                 while (pb <= pc && (r = CMP(thunk, pb, a)) <= 0) {
116                         if (r == 0) {
117                                 swap_cnt = 1;
118                                 swap(pa, pb);
119                                 pa += es;
120                         }
121                         pb += es;
122                 }
123                 while (pb <= pc && (r = CMP(thunk, pc, a)) >= 0) {
124                         if (r == 0) {
125                                 swap_cnt = 1;
126                                 swap(pc, pd);
127                                 pd -= es;
128                         }
129                         pc -= es;
130                 }
131                 if (pb > pc)
132                         break;
133                 swap(pb, pc);
134                 swap_cnt = 1;
135                 pb += es;
136                 pc -= es;
137         }
138         if (swap_cnt == 0) {  /* Switch to insertion sort */
139                 for (pm = (char *)a + es; pm < (char *)a + n * es; pm += es)
140                         for (pl = pm;
141                                 pl > (char *)a && CMP(thunk, pl - es, pl) > 0;
142                                 pl -= es)
143                                 swap(pl, pl - es);
144                 return;
145         }
146
147         pn = (char *)a + n * es;
148         r = min(pa - (char *)a, pb - pa);
149         vecswap((char *)a, pb - r, r);
150         r = min(pd - pc, pn - pd - es);
151         vecswap(pb, pn - r, r);
152         if ((r = pb - pa) > es)
153                 BLI_qsort_r(a, r / es, es, thunk, cmp);
154         if ((r = pd - pc) > es) {
155                 /* Iterate rather than recurse to save stack space */
156                 a = pn - r;
157                 n = r / es;
158                 goto loop;
159         }
160 }