Fix compilation error, missing size_t declaration
[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 <stdlib.h>
34
35 #include "BLI_utildefines.h"
36
37 #include "BLI_sort.h"
38
39 /* **** qsort based on FreeBSD source (libkern\qsort.c) **** */
40 BLI_INLINE char *med3(char *, char *, char *, BLI_sort_cmp_t, void *);
41 BLI_INLINE void  swapfunc(char *, char *, int, int);
42
43 #define min(a, b)       (a) < (b) ? a : b
44 #define swapcode(TYPE, parmi, parmj, n)         \
45 {                                                                                       \
46         long i = (n) / sizeof(TYPE);                    \
47         TYPE *pi = (TYPE *) (parmi);                    \
48         TYPE *pj = (TYPE *) (parmj);                    \
49         do {                                                                    \
50                 TYPE    t = *pi;                                        \
51                 *pi++ = *pj;                                            \
52                 *pj++ = t;                                                      \
53         } while (--i > 0);                                              \
54 }
55 #define SWAPINIT(a, es) swaptype = ((char *)a - (char *)0) % sizeof(long) || \
56         es % sizeof(long) ? 2 : es == sizeof(long)? 0 : 1;
57
58 BLI_INLINE void swapfunc(char* a, char* b, int n, int swaptype)
59 {
60         if(swaptype <= 1)
61                 swapcode(long, a, b, n)
62         else
63         swapcode(char, a, b, n)
64 }
65
66 #define swap(a, b)                                      \
67         if (swaptype == 0) {                    \
68                 long t = *(long *)(a);          \
69                 *(long *)(a) = *(long *)(b);\
70                 *(long *)(b) = t;                       \
71         } else                                                  \
72                 swapfunc(a, b, es, swaptype)
73
74 #define vecswap(a, b, n)        if ((n) > 0) swapfunc(a, b, n, swaptype)
75 #define CMP(t, x, y) (cmp((t), (x), (y)))
76
77 BLI_INLINE char *med3(char *a, char *b, char *c, BLI_sort_cmp_t cmp, void *thunk)
78 {
79         return CMP(thunk, a, b) < 0 ?
80                 (CMP(thunk, b, c) < 0 ? b : (CMP(thunk, a, c) < 0 ? c : a ))
81                 :(CMP(thunk, b, c) > 0 ? b : (CMP(thunk, a, c) < 0 ? a : c ));
82 }
83
84 void BLI_qsort_r(void *a, size_t n, size_t es, void *thunk, BLI_sort_cmp_t cmp)
85 {
86         char *pa, *pb, *pc, *pd, *pl, *pm, *pn;
87         int d, r, swaptype, swap_cnt;
88
89 loop:
90         SWAPINIT(a, es);
91         swap_cnt = 0;
92         if (n < 7) {
93                 for (pm = (char *)a + es; pm < (char *)a + n * es; pm += es)
94                         for (pl = pm;
95                                 pl > (char *)a && CMP(thunk, pl - es, pl) > 0;
96                                 pl -= es)
97                                 swap(pl, pl - es);
98                 return;
99         }
100         pm = (char *)a + (n / 2) * es;
101         if (n > 7) {
102                 pl = (char *)a;
103                 pn = (char *)a + (n - 1) * es;
104                 if (n > 40) {
105                         d = (n / 8) * es;
106                         pl = med3(pl, pl + d, pl + 2 * d, cmp, thunk);
107                         pm = med3(pm - d, pm, pm + d, cmp, thunk);
108                         pn = med3(pn - 2 * d, pn - d, pn, cmp, thunk);
109                 }
110                 pm = med3(pl, pm, pn, cmp, thunk);
111         }
112         swap((char *)a, pm);
113         pa = pb = (char *)a + es;
114
115         pc = pd = (char *)a + (n - 1) * es;
116         for (;;) {
117                 while (pb <= pc && (r = CMP(thunk, pb, a)) <= 0) {
118                         if (r == 0) {
119                                 swap_cnt = 1;
120                                 swap(pa, pb);
121                                 pa += es;
122                         }
123                         pb += es;
124                 }
125                 while (pb <= pc && (r = CMP(thunk, pc, a)) >= 0) {
126                         if (r == 0) {
127                                 swap_cnt = 1;
128                                 swap(pc, pd);
129                                 pd -= es;
130                         }
131                         pc -= es;
132                 }
133                 if (pb > pc)
134                         break;
135                 swap(pb, pc);
136                 swap_cnt = 1;
137                 pb += es;
138                 pc -= es;
139         }
140         if (swap_cnt == 0) {  /* Switch to insertion sort */
141                 for (pm = (char *)a + es; pm < (char *)a + n * es; pm += es)
142                         for (pl = pm;
143                                 pl > (char *)a && CMP(thunk, pl - es, pl) > 0;
144                                 pl -= es)
145                                 swap(pl, pl - es);
146                 return;
147         }
148
149         pn = (char *)a + n * es;
150         r = min(pa - (char *)a, pb - pa);
151         vecswap((char *)a, pb - r, r);
152         r = min(pd - pc, pn - pd - es);
153         vecswap(pb, pn - r, r);
154         if ((r = pb - pa) > es)
155                 BLI_qsort_r(a, r / es, es, thunk, cmp);
156         if ((r = pd - pc) > es) {
157                 /* Iterate rather than recurse to save stack space */
158                 a = pn - r;
159                 n = r / es;
160                 goto loop;
161         }
162 }