xref: /freebsd/contrib/libdivsufsort/lib/sssort.c (revision c08cbc64)
1*c08cbc64SXin LI /*
2*c08cbc64SXin LI  * sssort.c for libdivsufsort
3*c08cbc64SXin LI  * Copyright (c) 2003-2008 Yuta Mori All Rights Reserved.
4*c08cbc64SXin LI  *
5*c08cbc64SXin LI  * Permission is hereby granted, free of charge, to any person
6*c08cbc64SXin LI  * obtaining a copy of this software and associated documentation
7*c08cbc64SXin LI  * files (the "Software"), to deal in the Software without
8*c08cbc64SXin LI  * restriction, including without limitation the rights to use,
9*c08cbc64SXin LI  * copy, modify, merge, publish, distribute, sublicense, and/or sell
10*c08cbc64SXin LI  * copies of the Software, and to permit persons to whom the
11*c08cbc64SXin LI  * Software is furnished to do so, subject to the following
12*c08cbc64SXin LI  * conditions:
13*c08cbc64SXin LI  *
14*c08cbc64SXin LI  * The above copyright notice and this permission notice shall be
15*c08cbc64SXin LI  * included in all copies or substantial portions of the Software.
16*c08cbc64SXin LI  *
17*c08cbc64SXin LI  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
18*c08cbc64SXin LI  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
19*c08cbc64SXin LI  * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
20*c08cbc64SXin LI  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
21*c08cbc64SXin LI  * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
22*c08cbc64SXin LI  * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
23*c08cbc64SXin LI  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
24*c08cbc64SXin LI  * OTHER DEALINGS IN THE SOFTWARE.
25*c08cbc64SXin LI  */
26*c08cbc64SXin LI 
27*c08cbc64SXin LI #include "divsufsort_private.h"
28*c08cbc64SXin LI 
29*c08cbc64SXin LI 
30*c08cbc64SXin LI /*- Private Functions -*/
31*c08cbc64SXin LI 
32*c08cbc64SXin LI static const saint_t lg_table[256]= {
33*c08cbc64SXin LI  -1,0,1,1,2,2,2,2,3,3,3,3,3,3,3,3,4,4,4,4,4,4,4,4,4,4,4,4,4,4,4,4,
34*c08cbc64SXin LI   5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,5,
35*c08cbc64SXin LI   6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,
36*c08cbc64SXin LI   6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,6,
37*c08cbc64SXin LI   7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
38*c08cbc64SXin LI   7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
39*c08cbc64SXin LI   7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,
40*c08cbc64SXin LI   7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7,7
41*c08cbc64SXin LI };
42*c08cbc64SXin LI 
43*c08cbc64SXin LI #if (SS_BLOCKSIZE == 0) || (SS_INSERTIONSORT_THRESHOLD < SS_BLOCKSIZE)
44*c08cbc64SXin LI 
45*c08cbc64SXin LI static INLINE
46*c08cbc64SXin LI saint_t
ss_ilg(saidx_t n)47*c08cbc64SXin LI ss_ilg(saidx_t n) {
48*c08cbc64SXin LI #if SS_BLOCKSIZE == 0
49*c08cbc64SXin LI # if defined(BUILD_DIVSUFSORT64)
50*c08cbc64SXin LI   return (n >> 32) ?
51*c08cbc64SXin LI           ((n >> 48) ?
52*c08cbc64SXin LI             ((n >> 56) ?
53*c08cbc64SXin LI               56 + lg_table[(n >> 56) & 0xff] :
54*c08cbc64SXin LI               48 + lg_table[(n >> 48) & 0xff]) :
55*c08cbc64SXin LI             ((n >> 40) ?
56*c08cbc64SXin LI               40 + lg_table[(n >> 40) & 0xff] :
57*c08cbc64SXin LI               32 + lg_table[(n >> 32) & 0xff])) :
58*c08cbc64SXin LI           ((n & 0xffff0000) ?
59*c08cbc64SXin LI             ((n & 0xff000000) ?
60*c08cbc64SXin LI               24 + lg_table[(n >> 24) & 0xff] :
61*c08cbc64SXin LI               16 + lg_table[(n >> 16) & 0xff]) :
62*c08cbc64SXin LI             ((n & 0x0000ff00) ?
63*c08cbc64SXin LI                8 + lg_table[(n >>  8) & 0xff] :
64*c08cbc64SXin LI                0 + lg_table[(n >>  0) & 0xff]));
65*c08cbc64SXin LI # else
66*c08cbc64SXin LI   return (n & 0xffff0000) ?
67*c08cbc64SXin LI           ((n & 0xff000000) ?
68*c08cbc64SXin LI             24 + lg_table[(n >> 24) & 0xff] :
69*c08cbc64SXin LI             16 + lg_table[(n >> 16) & 0xff]) :
70*c08cbc64SXin LI           ((n & 0x0000ff00) ?
71*c08cbc64SXin LI              8 + lg_table[(n >>  8) & 0xff] :
72*c08cbc64SXin LI              0 + lg_table[(n >>  0) & 0xff]);
73*c08cbc64SXin LI # endif
74*c08cbc64SXin LI #elif SS_BLOCKSIZE < 256
75*c08cbc64SXin LI   return lg_table[n];
76*c08cbc64SXin LI #else
77*c08cbc64SXin LI   return (n & 0xff00) ?
78*c08cbc64SXin LI           8 + lg_table[(n >> 8) & 0xff] :
79*c08cbc64SXin LI           0 + lg_table[(n >> 0) & 0xff];
80*c08cbc64SXin LI #endif
81*c08cbc64SXin LI }
82*c08cbc64SXin LI 
83*c08cbc64SXin LI #endif /* (SS_BLOCKSIZE == 0) || (SS_INSERTIONSORT_THRESHOLD < SS_BLOCKSIZE) */
84*c08cbc64SXin LI 
85*c08cbc64SXin LI #if SS_BLOCKSIZE != 0
86*c08cbc64SXin LI 
87*c08cbc64SXin LI static const saint_t sqq_table[256] = {
88*c08cbc64SXin LI   0,  16,  22,  27,  32,  35,  39,  42,  45,  48,  50,  53,  55,  57,  59,  61,
89*c08cbc64SXin LI  64,  65,  67,  69,  71,  73,  75,  76,  78,  80,  81,  83,  84,  86,  87,  89,
90*c08cbc64SXin LI  90,  91,  93,  94,  96,  97,  98,  99, 101, 102, 103, 104, 106, 107, 108, 109,
91*c08cbc64SXin LI 110, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126,
92*c08cbc64SXin LI 128, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142,
93*c08cbc64SXin LI 143, 144, 144, 145, 146, 147, 148, 149, 150, 150, 151, 152, 153, 154, 155, 155,
94*c08cbc64SXin LI 156, 157, 158, 159, 160, 160, 161, 162, 163, 163, 164, 165, 166, 167, 167, 168,
95*c08cbc64SXin LI 169, 170, 170, 171, 172, 173, 173, 174, 175, 176, 176, 177, 178, 178, 179, 180,
96*c08cbc64SXin LI 181, 181, 182, 183, 183, 184, 185, 185, 186, 187, 187, 188, 189, 189, 190, 191,
97*c08cbc64SXin LI 192, 192, 193, 193, 194, 195, 195, 196, 197, 197, 198, 199, 199, 200, 201, 201,
98*c08cbc64SXin LI 202, 203, 203, 204, 204, 205, 206, 206, 207, 208, 208, 209, 209, 210, 211, 211,
99*c08cbc64SXin LI 212, 212, 213, 214, 214, 215, 215, 216, 217, 217, 218, 218, 219, 219, 220, 221,
100*c08cbc64SXin LI 221, 222, 222, 223, 224, 224, 225, 225, 226, 226, 227, 227, 228, 229, 229, 230,
101*c08cbc64SXin LI 230, 231, 231, 232, 232, 233, 234, 234, 235, 235, 236, 236, 237, 237, 238, 238,
102*c08cbc64SXin LI 239, 240, 240, 241, 241, 242, 242, 243, 243, 244, 244, 245, 245, 246, 246, 247,
103*c08cbc64SXin LI 247, 248, 248, 249, 249, 250, 250, 251, 251, 252, 252, 253, 253, 254, 254, 255
104*c08cbc64SXin LI };
105*c08cbc64SXin LI 
106*c08cbc64SXin LI static INLINE
107*c08cbc64SXin LI saidx_t
ss_isqrt(saidx_t x)108*c08cbc64SXin LI ss_isqrt(saidx_t x) {
109*c08cbc64SXin LI   saidx_t y, e;
110*c08cbc64SXin LI 
111*c08cbc64SXin LI   if(x >= (SS_BLOCKSIZE * SS_BLOCKSIZE)) { return SS_BLOCKSIZE; }
112*c08cbc64SXin LI   e = (x & 0xffff0000) ?
113*c08cbc64SXin LI         ((x & 0xff000000) ?
114*c08cbc64SXin LI           24 + lg_table[(x >> 24) & 0xff] :
115*c08cbc64SXin LI           16 + lg_table[(x >> 16) & 0xff]) :
116*c08cbc64SXin LI         ((x & 0x0000ff00) ?
117*c08cbc64SXin LI            8 + lg_table[(x >>  8) & 0xff] :
118*c08cbc64SXin LI            0 + lg_table[(x >>  0) & 0xff]);
119*c08cbc64SXin LI 
120*c08cbc64SXin LI   if(e >= 16) {
121*c08cbc64SXin LI     y = sqq_table[x >> ((e - 6) - (e & 1))] << ((e >> 1) - 7);
122*c08cbc64SXin LI     if(e >= 24) { y = (y + 1 + x / y) >> 1; }
123*c08cbc64SXin LI     y = (y + 1 + x / y) >> 1;
124*c08cbc64SXin LI   } else if(e >= 8) {
125*c08cbc64SXin LI     y = (sqq_table[x >> ((e - 6) - (e & 1))] >> (7 - (e >> 1))) + 1;
126*c08cbc64SXin LI   } else {
127*c08cbc64SXin LI     return sqq_table[x] >> 4;
128*c08cbc64SXin LI   }
129*c08cbc64SXin LI 
130*c08cbc64SXin LI   return (x < (y * y)) ? y - 1 : y;
131*c08cbc64SXin LI }
132*c08cbc64SXin LI 
133*c08cbc64SXin LI #endif /* SS_BLOCKSIZE != 0 */
134*c08cbc64SXin LI 
135*c08cbc64SXin LI 
136*c08cbc64SXin LI /*---------------------------------------------------------------------------*/
137*c08cbc64SXin LI 
138*c08cbc64SXin LI /* Compares two suffixes. */
139*c08cbc64SXin LI static INLINE
140*c08cbc64SXin LI saint_t
ss_compare(const sauchar_t * T,const saidx_t * p1,const saidx_t * p2,saidx_t depth)141*c08cbc64SXin LI ss_compare(const sauchar_t *T,
142*c08cbc64SXin LI            const saidx_t *p1, const saidx_t *p2,
143*c08cbc64SXin LI            saidx_t depth) {
144*c08cbc64SXin LI   const sauchar_t *U1, *U2, *U1n, *U2n;
145*c08cbc64SXin LI 
146*c08cbc64SXin LI   for(U1 = T + depth + *p1,
147*c08cbc64SXin LI       U2 = T + depth + *p2,
148*c08cbc64SXin LI       U1n = T + *(p1 + 1) + 2,
149*c08cbc64SXin LI       U2n = T + *(p2 + 1) + 2;
150*c08cbc64SXin LI       (U1 < U1n) && (U2 < U2n) && (*U1 == *U2);
151*c08cbc64SXin LI       ++U1, ++U2) {
152*c08cbc64SXin LI   }
153*c08cbc64SXin LI 
154*c08cbc64SXin LI   return U1 < U1n ?
155*c08cbc64SXin LI         (U2 < U2n ? *U1 - *U2 : 1) :
156*c08cbc64SXin LI         (U2 < U2n ? -1 : 0);
157*c08cbc64SXin LI }
158*c08cbc64SXin LI 
159*c08cbc64SXin LI 
160*c08cbc64SXin LI /*---------------------------------------------------------------------------*/
161*c08cbc64SXin LI 
162*c08cbc64SXin LI #if (SS_BLOCKSIZE != 1) && (SS_INSERTIONSORT_THRESHOLD != 1)
163*c08cbc64SXin LI 
164*c08cbc64SXin LI /* Insertionsort for small size groups */
165*c08cbc64SXin LI static
166*c08cbc64SXin LI void
ss_insertionsort(const sauchar_t * T,const saidx_t * PA,saidx_t * first,saidx_t * last,saidx_t depth)167*c08cbc64SXin LI ss_insertionsort(const sauchar_t *T, const saidx_t *PA,
168*c08cbc64SXin LI                  saidx_t *first, saidx_t *last, saidx_t depth) {
169*c08cbc64SXin LI   saidx_t *i, *j;
170*c08cbc64SXin LI   saidx_t t;
171*c08cbc64SXin LI   saint_t r;
172*c08cbc64SXin LI 
173*c08cbc64SXin LI   for(i = last - 2; first <= i; --i) {
174*c08cbc64SXin LI     for(t = *i, j = i + 1; 0 < (r = ss_compare(T, PA + t, PA + *j, depth));) {
175*c08cbc64SXin LI       do { *(j - 1) = *j; } while((++j < last) && (*j < 0));
176*c08cbc64SXin LI       if(last <= j) { break; }
177*c08cbc64SXin LI     }
178*c08cbc64SXin LI     if(r == 0) { *j = ~*j; }
179*c08cbc64SXin LI     *(j - 1) = t;
180*c08cbc64SXin LI   }
181*c08cbc64SXin LI }
182*c08cbc64SXin LI 
183*c08cbc64SXin LI #endif /* (SS_BLOCKSIZE != 1) && (SS_INSERTIONSORT_THRESHOLD != 1) */
184*c08cbc64SXin LI 
185*c08cbc64SXin LI 
186*c08cbc64SXin LI /*---------------------------------------------------------------------------*/
187*c08cbc64SXin LI 
188*c08cbc64SXin LI #if (SS_BLOCKSIZE == 0) || (SS_INSERTIONSORT_THRESHOLD < SS_BLOCKSIZE)
189*c08cbc64SXin LI 
190*c08cbc64SXin LI static INLINE
191*c08cbc64SXin LI void
ss_fixdown(const sauchar_t * Td,const saidx_t * PA,saidx_t * SA,saidx_t i,saidx_t size)192*c08cbc64SXin LI ss_fixdown(const sauchar_t *Td, const saidx_t *PA,
193*c08cbc64SXin LI            saidx_t *SA, saidx_t i, saidx_t size) {
194*c08cbc64SXin LI   saidx_t j, k;
195*c08cbc64SXin LI   saidx_t v;
196*c08cbc64SXin LI   saint_t c, d, e;
197*c08cbc64SXin LI 
198*c08cbc64SXin LI   for(v = SA[i], c = Td[PA[v]]; (j = 2 * i + 1) < size; SA[i] = SA[k], i = k) {
199*c08cbc64SXin LI     d = Td[PA[SA[k = j++]]];
200*c08cbc64SXin LI     if(d < (e = Td[PA[SA[j]]])) { k = j; d = e; }
201*c08cbc64SXin LI     if(d <= c) { break; }
202*c08cbc64SXin LI   }
203*c08cbc64SXin LI   SA[i] = v;
204*c08cbc64SXin LI }
205*c08cbc64SXin LI 
206*c08cbc64SXin LI /* Simple top-down heapsort. */
207*c08cbc64SXin LI static
208*c08cbc64SXin LI void
ss_heapsort(const sauchar_t * Td,const saidx_t * PA,saidx_t * SA,saidx_t size)209*c08cbc64SXin LI ss_heapsort(const sauchar_t *Td, const saidx_t *PA, saidx_t *SA, saidx_t size) {
210*c08cbc64SXin LI   saidx_t i, m;
211*c08cbc64SXin LI   saidx_t t;
212*c08cbc64SXin LI 
213*c08cbc64SXin LI   m = size;
214*c08cbc64SXin LI   if((size % 2) == 0) {
215*c08cbc64SXin LI     m--;
216*c08cbc64SXin LI     if(Td[PA[SA[m / 2]]] < Td[PA[SA[m]]]) { SWAP(SA[m], SA[m / 2]); }
217*c08cbc64SXin LI   }
218*c08cbc64SXin LI 
219*c08cbc64SXin LI   for(i = m / 2 - 1; 0 <= i; --i) { ss_fixdown(Td, PA, SA, i, m); }
220*c08cbc64SXin LI   if((size % 2) == 0) { SWAP(SA[0], SA[m]); ss_fixdown(Td, PA, SA, 0, m); }
221*c08cbc64SXin LI   for(i = m - 1; 0 < i; --i) {
222*c08cbc64SXin LI     t = SA[0], SA[0] = SA[i];
223*c08cbc64SXin LI     ss_fixdown(Td, PA, SA, 0, i);
224*c08cbc64SXin LI     SA[i] = t;
225*c08cbc64SXin LI   }
226*c08cbc64SXin LI }
227*c08cbc64SXin LI 
228*c08cbc64SXin LI 
229*c08cbc64SXin LI /*---------------------------------------------------------------------------*/
230*c08cbc64SXin LI 
231*c08cbc64SXin LI /* Returns the median of three elements. */
232*c08cbc64SXin LI static INLINE
233*c08cbc64SXin LI saidx_t *
ss_median3(const sauchar_t * Td,const saidx_t * PA,saidx_t * v1,saidx_t * v2,saidx_t * v3)234*c08cbc64SXin LI ss_median3(const sauchar_t *Td, const saidx_t *PA,
235*c08cbc64SXin LI            saidx_t *v1, saidx_t *v2, saidx_t *v3) {
236*c08cbc64SXin LI   saidx_t *t;
237*c08cbc64SXin LI   if(Td[PA[*v1]] > Td[PA[*v2]]) { SWAP(v1, v2); }
238*c08cbc64SXin LI   if(Td[PA[*v2]] > Td[PA[*v3]]) {
239*c08cbc64SXin LI     if(Td[PA[*v1]] > Td[PA[*v3]]) { return v1; }
240*c08cbc64SXin LI     else { return v3; }
241*c08cbc64SXin LI   }
242*c08cbc64SXin LI   return v2;
243*c08cbc64SXin LI }
244*c08cbc64SXin LI 
245*c08cbc64SXin LI /* Returns the median of five elements. */
246*c08cbc64SXin LI static INLINE
247*c08cbc64SXin LI saidx_t *
ss_median5(const sauchar_t * Td,const saidx_t * PA,saidx_t * v1,saidx_t * v2,saidx_t * v3,saidx_t * v4,saidx_t * v5)248*c08cbc64SXin LI ss_median5(const sauchar_t *Td, const saidx_t *PA,
249*c08cbc64SXin LI            saidx_t *v1, saidx_t *v2, saidx_t *v3, saidx_t *v4, saidx_t *v5) {
250*c08cbc64SXin LI   saidx_t *t;
251*c08cbc64SXin LI   if(Td[PA[*v2]] > Td[PA[*v3]]) { SWAP(v2, v3); }
252*c08cbc64SXin LI   if(Td[PA[*v4]] > Td[PA[*v5]]) { SWAP(v4, v5); }
253*c08cbc64SXin LI   if(Td[PA[*v2]] > Td[PA[*v4]]) { SWAP(v2, v4); SWAP(v3, v5); }
254*c08cbc64SXin LI   if(Td[PA[*v1]] > Td[PA[*v3]]) { SWAP(v1, v3); }
255*c08cbc64SXin LI   if(Td[PA[*v1]] > Td[PA[*v4]]) { SWAP(v1, v4); SWAP(v3, v5); }
256*c08cbc64SXin LI   if(Td[PA[*v3]] > Td[PA[*v4]]) { return v4; }
257*c08cbc64SXin LI   return v3;
258*c08cbc64SXin LI }
259*c08cbc64SXin LI 
260*c08cbc64SXin LI /* Returns the pivot element. */
261*c08cbc64SXin LI static INLINE
262*c08cbc64SXin LI saidx_t *
ss_pivot(const sauchar_t * Td,const saidx_t * PA,saidx_t * first,saidx_t * last)263*c08cbc64SXin LI ss_pivot(const sauchar_t *Td, const saidx_t *PA, saidx_t *first, saidx_t *last) {
264*c08cbc64SXin LI   saidx_t *middle;
265*c08cbc64SXin LI   saidx_t t;
266*c08cbc64SXin LI 
267*c08cbc64SXin LI   t = last - first;
268*c08cbc64SXin LI   middle = first + t / 2;
269*c08cbc64SXin LI 
270*c08cbc64SXin LI   if(t <= 512) {
271*c08cbc64SXin LI     if(t <= 32) {
272*c08cbc64SXin LI       return ss_median3(Td, PA, first, middle, last - 1);
273*c08cbc64SXin LI     } else {
274*c08cbc64SXin LI       t >>= 2;
275*c08cbc64SXin LI       return ss_median5(Td, PA, first, first + t, middle, last - 1 - t, last - 1);
276*c08cbc64SXin LI     }
277*c08cbc64SXin LI   }
278*c08cbc64SXin LI   t >>= 3;
279*c08cbc64SXin LI   first  = ss_median3(Td, PA, first, first + t, first + (t << 1));
280*c08cbc64SXin LI   middle = ss_median3(Td, PA, middle - t, middle, middle + t);
281*c08cbc64SXin LI   last   = ss_median3(Td, PA, last - 1 - (t << 1), last - 1 - t, last - 1);
282*c08cbc64SXin LI   return ss_median3(Td, PA, first, middle, last);
283*c08cbc64SXin LI }
284*c08cbc64SXin LI 
285*c08cbc64SXin LI 
286*c08cbc64SXin LI /*---------------------------------------------------------------------------*/
287*c08cbc64SXin LI 
288*c08cbc64SXin LI /* Binary partition for substrings. */
289*c08cbc64SXin LI static INLINE
290*c08cbc64SXin LI saidx_t *
ss_partition(const saidx_t * PA,saidx_t * first,saidx_t * last,saidx_t depth)291*c08cbc64SXin LI ss_partition(const saidx_t *PA,
292*c08cbc64SXin LI                     saidx_t *first, saidx_t *last, saidx_t depth) {
293*c08cbc64SXin LI   saidx_t *a, *b;
294*c08cbc64SXin LI   saidx_t t;
295*c08cbc64SXin LI   for(a = first - 1, b = last;;) {
296*c08cbc64SXin LI     for(; (++a < b) && ((PA[*a] + depth) >= (PA[*a + 1] + 1));) { *a = ~*a; }
297*c08cbc64SXin LI     for(; (a < --b) && ((PA[*b] + depth) <  (PA[*b + 1] + 1));) { }
298*c08cbc64SXin LI     if(b <= a) { break; }
299*c08cbc64SXin LI     t = ~*b;
300*c08cbc64SXin LI     *b = *a;
301*c08cbc64SXin LI     *a = t;
302*c08cbc64SXin LI   }
303*c08cbc64SXin LI   if(first < a) { *first = ~*first; }
304*c08cbc64SXin LI   return a;
305*c08cbc64SXin LI }
306*c08cbc64SXin LI 
307*c08cbc64SXin LI /* Multikey introsort for medium size groups. */
308*c08cbc64SXin LI static
309*c08cbc64SXin LI void
ss_mintrosort(const sauchar_t * T,const saidx_t * PA,saidx_t * first,saidx_t * last,saidx_t depth)310*c08cbc64SXin LI ss_mintrosort(const sauchar_t *T, const saidx_t *PA,
311*c08cbc64SXin LI               saidx_t *first, saidx_t *last,
312*c08cbc64SXin LI               saidx_t depth) {
313*c08cbc64SXin LI #define STACK_SIZE SS_MISORT_STACKSIZE
314*c08cbc64SXin LI   struct { saidx_t *a, *b, c; saint_t d; } stack[STACK_SIZE];
315*c08cbc64SXin LI   const sauchar_t *Td;
316*c08cbc64SXin LI   saidx_t *a, *b, *c, *d, *e, *f;
317*c08cbc64SXin LI   saidx_t s, t;
318*c08cbc64SXin LI   saint_t ssize;
319*c08cbc64SXin LI   saint_t limit;
320*c08cbc64SXin LI   saint_t v, x = 0;
321*c08cbc64SXin LI 
322*c08cbc64SXin LI   for(ssize = 0, limit = ss_ilg(last - first);;) {
323*c08cbc64SXin LI 
324*c08cbc64SXin LI     if((last - first) <= SS_INSERTIONSORT_THRESHOLD) {
325*c08cbc64SXin LI #if 1 < SS_INSERTIONSORT_THRESHOLD
326*c08cbc64SXin LI       if(1 < (last - first)) { ss_insertionsort(T, PA, first, last, depth); }
327*c08cbc64SXin LI #endif
328*c08cbc64SXin LI       STACK_POP(first, last, depth, limit);
329*c08cbc64SXin LI       continue;
330*c08cbc64SXin LI     }
331*c08cbc64SXin LI 
332*c08cbc64SXin LI     Td = T + depth;
333*c08cbc64SXin LI     if(limit-- == 0) { ss_heapsort(Td, PA, first, last - first); }
334*c08cbc64SXin LI     if(limit < 0) {
335*c08cbc64SXin LI       for(a = first + 1, v = Td[PA[*first]]; a < last; ++a) {
336*c08cbc64SXin LI         if((x = Td[PA[*a]]) != v) {
337*c08cbc64SXin LI           if(1 < (a - first)) { break; }
338*c08cbc64SXin LI           v = x;
339*c08cbc64SXin LI           first = a;
340*c08cbc64SXin LI         }
341*c08cbc64SXin LI       }
342*c08cbc64SXin LI       if(Td[PA[*first] - 1] < v) {
343*c08cbc64SXin LI         first = ss_partition(PA, first, a, depth);
344*c08cbc64SXin LI       }
345*c08cbc64SXin LI       if((a - first) <= (last - a)) {
346*c08cbc64SXin LI         if(1 < (a - first)) {
347*c08cbc64SXin LI           STACK_PUSH(a, last, depth, -1);
348*c08cbc64SXin LI           last = a, depth += 1, limit = ss_ilg(a - first);
349*c08cbc64SXin LI         } else {
350*c08cbc64SXin LI           first = a, limit = -1;
351*c08cbc64SXin LI         }
352*c08cbc64SXin LI       } else {
353*c08cbc64SXin LI         if(1 < (last - a)) {
354*c08cbc64SXin LI           STACK_PUSH(first, a, depth + 1, ss_ilg(a - first));
355*c08cbc64SXin LI           first = a, limit = -1;
356*c08cbc64SXin LI         } else {
357*c08cbc64SXin LI           last = a, depth += 1, limit = ss_ilg(a - first);
358*c08cbc64SXin LI         }
359*c08cbc64SXin LI       }
360*c08cbc64SXin LI       continue;
361*c08cbc64SXin LI     }
362*c08cbc64SXin LI 
363*c08cbc64SXin LI     /* choose pivot */
364*c08cbc64SXin LI     a = ss_pivot(Td, PA, first, last);
365*c08cbc64SXin LI     v = Td[PA[*a]];
366*c08cbc64SXin LI     SWAP(*first, *a);
367*c08cbc64SXin LI 
368*c08cbc64SXin LI     /* partition */
369*c08cbc64SXin LI     for(b = first; (++b < last) && ((x = Td[PA[*b]]) == v);) { }
370*c08cbc64SXin LI     if(((a = b) < last) && (x < v)) {
371*c08cbc64SXin LI       for(; (++b < last) && ((x = Td[PA[*b]]) <= v);) {
372*c08cbc64SXin LI         if(x == v) { SWAP(*b, *a); ++a; }
373*c08cbc64SXin LI       }
374*c08cbc64SXin LI     }
375*c08cbc64SXin LI     for(c = last; (b < --c) && ((x = Td[PA[*c]]) == v);) { }
376*c08cbc64SXin LI     if((b < (d = c)) && (x > v)) {
377*c08cbc64SXin LI       for(; (b < --c) && ((x = Td[PA[*c]]) >= v);) {
378*c08cbc64SXin LI         if(x == v) { SWAP(*c, *d); --d; }
379*c08cbc64SXin LI       }
380*c08cbc64SXin LI     }
381*c08cbc64SXin LI     for(; b < c;) {
382*c08cbc64SXin LI       SWAP(*b, *c);
383*c08cbc64SXin LI       for(; (++b < c) && ((x = Td[PA[*b]]) <= v);) {
384*c08cbc64SXin LI         if(x == v) { SWAP(*b, *a); ++a; }
385*c08cbc64SXin LI       }
386*c08cbc64SXin LI       for(; (b < --c) && ((x = Td[PA[*c]]) >= v);) {
387*c08cbc64SXin LI         if(x == v) { SWAP(*c, *d); --d; }
388*c08cbc64SXin LI       }
389*c08cbc64SXin LI     }
390*c08cbc64SXin LI 
391*c08cbc64SXin LI     if(a <= d) {
392*c08cbc64SXin LI       c = b - 1;
393*c08cbc64SXin LI 
394*c08cbc64SXin LI       if((s = a - first) > (t = b - a)) { s = t; }
395*c08cbc64SXin LI       for(e = first, f = b - s; 0 < s; --s, ++e, ++f) { SWAP(*e, *f); }
396*c08cbc64SXin LI       if((s = d - c) > (t = last - d - 1)) { s = t; }
397*c08cbc64SXin LI       for(e = b, f = last - s; 0 < s; --s, ++e, ++f) { SWAP(*e, *f); }
398*c08cbc64SXin LI 
399*c08cbc64SXin LI       a = first + (b - a), c = last - (d - c);
400*c08cbc64SXin LI       b = (v <= Td[PA[*a] - 1]) ? a : ss_partition(PA, a, c, depth);
401*c08cbc64SXin LI 
402*c08cbc64SXin LI       if((a - first) <= (last - c)) {
403*c08cbc64SXin LI         if((last - c) <= (c - b)) {
404*c08cbc64SXin LI           STACK_PUSH(b, c, depth + 1, ss_ilg(c - b));
405*c08cbc64SXin LI           STACK_PUSH(c, last, depth, limit);
406*c08cbc64SXin LI           last = a;
407*c08cbc64SXin LI         } else if((a - first) <= (c - b)) {
408*c08cbc64SXin LI           STACK_PUSH(c, last, depth, limit);
409*c08cbc64SXin LI           STACK_PUSH(b, c, depth + 1, ss_ilg(c - b));
410*c08cbc64SXin LI           last = a;
411*c08cbc64SXin LI         } else {
412*c08cbc64SXin LI           STACK_PUSH(c, last, depth, limit);
413*c08cbc64SXin LI           STACK_PUSH(first, a, depth, limit);
414*c08cbc64SXin LI           first = b, last = c, depth += 1, limit = ss_ilg(c - b);
415*c08cbc64SXin LI         }
416*c08cbc64SXin LI       } else {
417*c08cbc64SXin LI         if((a - first) <= (c - b)) {
418*c08cbc64SXin LI           STACK_PUSH(b, c, depth + 1, ss_ilg(c - b));
419*c08cbc64SXin LI           STACK_PUSH(first, a, depth, limit);
420*c08cbc64SXin LI           first = c;
421*c08cbc64SXin LI         } else if((last - c) <= (c - b)) {
422*c08cbc64SXin LI           STACK_PUSH(first, a, depth, limit);
423*c08cbc64SXin LI           STACK_PUSH(b, c, depth + 1, ss_ilg(c - b));
424*c08cbc64SXin LI           first = c;
425*c08cbc64SXin LI         } else {
426*c08cbc64SXin LI           STACK_PUSH(first, a, depth, limit);
427*c08cbc64SXin LI           STACK_PUSH(c, last, depth, limit);
428*c08cbc64SXin LI           first = b, last = c, depth += 1, limit = ss_ilg(c - b);
429*c08cbc64SXin LI         }
430*c08cbc64SXin LI       }
431*c08cbc64SXin LI     } else {
432*c08cbc64SXin LI       limit += 1;
433*c08cbc64SXin LI       if(Td[PA[*first] - 1] < v) {
434*c08cbc64SXin LI         first = ss_partition(PA, first, last, depth);
435*c08cbc64SXin LI         limit = ss_ilg(last - first);
436*c08cbc64SXin LI       }
437*c08cbc64SXin LI       depth += 1;
438*c08cbc64SXin LI     }
439*c08cbc64SXin LI   }
440*c08cbc64SXin LI #undef STACK_SIZE
441*c08cbc64SXin LI }
442*c08cbc64SXin LI 
443*c08cbc64SXin LI #endif /* (SS_BLOCKSIZE == 0) || (SS_INSERTIONSORT_THRESHOLD < SS_BLOCKSIZE) */
444*c08cbc64SXin LI 
445*c08cbc64SXin LI 
446*c08cbc64SXin LI /*---------------------------------------------------------------------------*/
447*c08cbc64SXin LI 
448*c08cbc64SXin LI #if SS_BLOCKSIZE != 0
449*c08cbc64SXin LI 
450*c08cbc64SXin LI static INLINE
451*c08cbc64SXin LI void
ss_blockswap(saidx_t * a,saidx_t * b,saidx_t n)452*c08cbc64SXin LI ss_blockswap(saidx_t *a, saidx_t *b, saidx_t n) {
453*c08cbc64SXin LI   saidx_t t;
454*c08cbc64SXin LI   for(; 0 < n; --n, ++a, ++b) {
455*c08cbc64SXin LI     t = *a, *a = *b, *b = t;
456*c08cbc64SXin LI   }
457*c08cbc64SXin LI }
458*c08cbc64SXin LI 
459*c08cbc64SXin LI static INLINE
460*c08cbc64SXin LI void
ss_rotate(saidx_t * first,saidx_t * middle,saidx_t * last)461*c08cbc64SXin LI ss_rotate(saidx_t *first, saidx_t *middle, saidx_t *last) {
462*c08cbc64SXin LI   saidx_t *a, *b, t;
463*c08cbc64SXin LI   saidx_t l, r;
464*c08cbc64SXin LI   l = middle - first, r = last - middle;
465*c08cbc64SXin LI   for(; (0 < l) && (0 < r);) {
466*c08cbc64SXin LI     if(l == r) { ss_blockswap(first, middle, l); break; }
467*c08cbc64SXin LI     if(l < r) {
468*c08cbc64SXin LI       a = last - 1, b = middle - 1;
469*c08cbc64SXin LI       t = *a;
470*c08cbc64SXin LI       do {
471*c08cbc64SXin LI         *a-- = *b, *b-- = *a;
472*c08cbc64SXin LI         if(b < first) {
473*c08cbc64SXin LI           *a = t;
474*c08cbc64SXin LI           last = a;
475*c08cbc64SXin LI           if((r -= l + 1) <= l) { break; }
476*c08cbc64SXin LI           a -= 1, b = middle - 1;
477*c08cbc64SXin LI           t = *a;
478*c08cbc64SXin LI         }
479*c08cbc64SXin LI       } while(1);
480*c08cbc64SXin LI     } else {
481*c08cbc64SXin LI       a = first, b = middle;
482*c08cbc64SXin LI       t = *a;
483*c08cbc64SXin LI       do {
484*c08cbc64SXin LI         *a++ = *b, *b++ = *a;
485*c08cbc64SXin LI         if(last <= b) {
486*c08cbc64SXin LI           *a = t;
487*c08cbc64SXin LI           first = a + 1;
488*c08cbc64SXin LI           if((l -= r + 1) <= r) { break; }
489*c08cbc64SXin LI           a += 1, b = middle;
490*c08cbc64SXin LI           t = *a;
491*c08cbc64SXin LI         }
492*c08cbc64SXin LI       } while(1);
493*c08cbc64SXin LI     }
494*c08cbc64SXin LI   }
495*c08cbc64SXin LI }
496*c08cbc64SXin LI 
497*c08cbc64SXin LI 
498*c08cbc64SXin LI /*---------------------------------------------------------------------------*/
499*c08cbc64SXin LI 
500*c08cbc64SXin LI static
501*c08cbc64SXin LI void
ss_inplacemerge(const sauchar_t * T,const saidx_t * PA,saidx_t * first,saidx_t * middle,saidx_t * last,saidx_t depth)502*c08cbc64SXin LI ss_inplacemerge(const sauchar_t *T, const saidx_t *PA,
503*c08cbc64SXin LI                 saidx_t *first, saidx_t *middle, saidx_t *last,
504*c08cbc64SXin LI                 saidx_t depth) {
505*c08cbc64SXin LI   const saidx_t *p;
506*c08cbc64SXin LI   saidx_t *a, *b;
507*c08cbc64SXin LI   saidx_t len, half;
508*c08cbc64SXin LI   saint_t q, r;
509*c08cbc64SXin LI   saint_t x;
510*c08cbc64SXin LI 
511*c08cbc64SXin LI   for(;;) {
512*c08cbc64SXin LI     if(*(last - 1) < 0) { x = 1; p = PA + ~*(last - 1); }
513*c08cbc64SXin LI     else                { x = 0; p = PA +  *(last - 1); }
514*c08cbc64SXin LI     for(a = first, len = middle - first, half = len >> 1, r = -1;
515*c08cbc64SXin LI         0 < len;
516*c08cbc64SXin LI         len = half, half >>= 1) {
517*c08cbc64SXin LI       b = a + half;
518*c08cbc64SXin LI       q = ss_compare(T, PA + ((0 <= *b) ? *b : ~*b), p, depth);
519*c08cbc64SXin LI       if(q < 0) {
520*c08cbc64SXin LI         a = b + 1;
521*c08cbc64SXin LI         half -= (len & 1) ^ 1;
522*c08cbc64SXin LI       } else {
523*c08cbc64SXin LI         r = q;
524*c08cbc64SXin LI       }
525*c08cbc64SXin LI     }
526*c08cbc64SXin LI     if(a < middle) {
527*c08cbc64SXin LI       if(r == 0) { *a = ~*a; }
528*c08cbc64SXin LI       ss_rotate(a, middle, last);
529*c08cbc64SXin LI       last -= middle - a;
530*c08cbc64SXin LI       middle = a;
531*c08cbc64SXin LI       if(first == middle) { break; }
532*c08cbc64SXin LI     }
533*c08cbc64SXin LI     --last;
534*c08cbc64SXin LI     if(x != 0) { while(*--last < 0) { } }
535*c08cbc64SXin LI     if(middle == last) { break; }
536*c08cbc64SXin LI   }
537*c08cbc64SXin LI }
538*c08cbc64SXin LI 
539*c08cbc64SXin LI 
540*c08cbc64SXin LI /*---------------------------------------------------------------------------*/
541*c08cbc64SXin LI 
542*c08cbc64SXin LI /* Merge-forward with internal buffer. */
543*c08cbc64SXin LI static
544*c08cbc64SXin LI void
ss_mergeforward(const sauchar_t * T,const saidx_t * PA,saidx_t * first,saidx_t * middle,saidx_t * last,saidx_t * buf,saidx_t depth)545*c08cbc64SXin LI ss_mergeforward(const sauchar_t *T, const saidx_t *PA,
546*c08cbc64SXin LI                 saidx_t *first, saidx_t *middle, saidx_t *last,
547*c08cbc64SXin LI                 saidx_t *buf, saidx_t depth) {
548*c08cbc64SXin LI   saidx_t *a, *b, *c, *bufend;
549*c08cbc64SXin LI   saidx_t t;
550*c08cbc64SXin LI   saint_t r;
551*c08cbc64SXin LI 
552*c08cbc64SXin LI   bufend = buf + (middle - first) - 1;
553*c08cbc64SXin LI   ss_blockswap(buf, first, middle - first);
554*c08cbc64SXin LI 
555*c08cbc64SXin LI   for(t = *(a = first), b = buf, c = middle;;) {
556*c08cbc64SXin LI     r = ss_compare(T, PA + *b, PA + *c, depth);
557*c08cbc64SXin LI     if(r < 0) {
558*c08cbc64SXin LI       do {
559*c08cbc64SXin LI         *a++ = *b;
560*c08cbc64SXin LI         if(bufend <= b) { *bufend = t; return; }
561*c08cbc64SXin LI         *b++ = *a;
562*c08cbc64SXin LI       } while(*b < 0);
563*c08cbc64SXin LI     } else if(r > 0) {
564*c08cbc64SXin LI       do {
565*c08cbc64SXin LI         *a++ = *c, *c++ = *a;
566*c08cbc64SXin LI         if(last <= c) {
567*c08cbc64SXin LI           while(b < bufend) { *a++ = *b, *b++ = *a; }
568*c08cbc64SXin LI           *a = *b, *b = t;
569*c08cbc64SXin LI           return;
570*c08cbc64SXin LI         }
571*c08cbc64SXin LI       } while(*c < 0);
572*c08cbc64SXin LI     } else {
573*c08cbc64SXin LI       *c = ~*c;
574*c08cbc64SXin LI       do {
575*c08cbc64SXin LI         *a++ = *b;
576*c08cbc64SXin LI         if(bufend <= b) { *bufend = t; return; }
577*c08cbc64SXin LI         *b++ = *a;
578*c08cbc64SXin LI       } while(*b < 0);
579*c08cbc64SXin LI 
580*c08cbc64SXin LI       do {
581*c08cbc64SXin LI         *a++ = *c, *c++ = *a;
582*c08cbc64SXin LI         if(last <= c) {
583*c08cbc64SXin LI           while(b < bufend) { *a++ = *b, *b++ = *a; }
584*c08cbc64SXin LI           *a = *b, *b = t;
585*c08cbc64SXin LI           return;
586*c08cbc64SXin LI         }
587*c08cbc64SXin LI       } while(*c < 0);
588*c08cbc64SXin LI     }
589*c08cbc64SXin LI   }
590*c08cbc64SXin LI }
591*c08cbc64SXin LI 
592*c08cbc64SXin LI /* Merge-backward with internal buffer. */
593*c08cbc64SXin LI static
594*c08cbc64SXin LI void
ss_mergebackward(const sauchar_t * T,const saidx_t * PA,saidx_t * first,saidx_t * middle,saidx_t * last,saidx_t * buf,saidx_t depth)595*c08cbc64SXin LI ss_mergebackward(const sauchar_t *T, const saidx_t *PA,
596*c08cbc64SXin LI                  saidx_t *first, saidx_t *middle, saidx_t *last,
597*c08cbc64SXin LI                  saidx_t *buf, saidx_t depth) {
598*c08cbc64SXin LI   const saidx_t *p1, *p2;
599*c08cbc64SXin LI   saidx_t *a, *b, *c, *bufend;
600*c08cbc64SXin LI   saidx_t t;
601*c08cbc64SXin LI   saint_t r;
602*c08cbc64SXin LI   saint_t x;
603*c08cbc64SXin LI 
604*c08cbc64SXin LI   bufend = buf + (last - middle) - 1;
605*c08cbc64SXin LI   ss_blockswap(buf, middle, last - middle);
606*c08cbc64SXin LI 
607*c08cbc64SXin LI   x = 0;
608*c08cbc64SXin LI   if(*bufend < 0)       { p1 = PA + ~*bufend; x |= 1; }
609*c08cbc64SXin LI   else                  { p1 = PA +  *bufend; }
610*c08cbc64SXin LI   if(*(middle - 1) < 0) { p2 = PA + ~*(middle - 1); x |= 2; }
611*c08cbc64SXin LI   else                  { p2 = PA +  *(middle - 1); }
612*c08cbc64SXin LI   for(t = *(a = last - 1), b = bufend, c = middle - 1;;) {
613*c08cbc64SXin LI     r = ss_compare(T, p1, p2, depth);
614*c08cbc64SXin LI     if(0 < r) {
615*c08cbc64SXin LI       if(x & 1) { do { *a-- = *b, *b-- = *a; } while(*b < 0); x ^= 1; }
616*c08cbc64SXin LI       *a-- = *b;
617*c08cbc64SXin LI       if(b <= buf) { *buf = t; break; }
618*c08cbc64SXin LI       *b-- = *a;
619*c08cbc64SXin LI       if(*b < 0) { p1 = PA + ~*b; x |= 1; }
620*c08cbc64SXin LI       else       { p1 = PA +  *b; }
621*c08cbc64SXin LI     } else if(r < 0) {
622*c08cbc64SXin LI       if(x & 2) { do { *a-- = *c, *c-- = *a; } while(*c < 0); x ^= 2; }
623*c08cbc64SXin LI       *a-- = *c, *c-- = *a;
624*c08cbc64SXin LI       if(c < first) {
625*c08cbc64SXin LI         while(buf < b) { *a-- = *b, *b-- = *a; }
626*c08cbc64SXin LI         *a = *b, *b = t;
627*c08cbc64SXin LI         break;
628*c08cbc64SXin LI       }
629*c08cbc64SXin LI       if(*c < 0) { p2 = PA + ~*c; x |= 2; }
630*c08cbc64SXin LI       else       { p2 = PA +  *c; }
631*c08cbc64SXin LI     } else {
632*c08cbc64SXin LI       if(x & 1) { do { *a-- = *b, *b-- = *a; } while(*b < 0); x ^= 1; }
633*c08cbc64SXin LI       *a-- = ~*b;
634*c08cbc64SXin LI       if(b <= buf) { *buf = t; break; }
635*c08cbc64SXin LI       *b-- = *a;
636*c08cbc64SXin LI       if(x & 2) { do { *a-- = *c, *c-- = *a; } while(*c < 0); x ^= 2; }
637*c08cbc64SXin LI       *a-- = *c, *c-- = *a;
638*c08cbc64SXin LI       if(c < first) {
639*c08cbc64SXin LI         while(buf < b) { *a-- = *b, *b-- = *a; }
640*c08cbc64SXin LI         *a = *b, *b = t;
641*c08cbc64SXin LI         break;
642*c08cbc64SXin LI       }
643*c08cbc64SXin LI       if(*b < 0) { p1 = PA + ~*b; x |= 1; }
644*c08cbc64SXin LI       else       { p1 = PA +  *b; }
645*c08cbc64SXin LI       if(*c < 0) { p2 = PA + ~*c; x |= 2; }
646*c08cbc64SXin LI       else       { p2 = PA +  *c; }
647*c08cbc64SXin LI     }
648*c08cbc64SXin LI   }
649*c08cbc64SXin LI }
650*c08cbc64SXin LI 
651*c08cbc64SXin LI /* D&C based merge. */
652*c08cbc64SXin LI static
653*c08cbc64SXin LI void
ss_swapmerge(const sauchar_t * T,const saidx_t * PA,saidx_t * first,saidx_t * middle,saidx_t * last,saidx_t * buf,saidx_t bufsize,saidx_t depth)654*c08cbc64SXin LI ss_swapmerge(const sauchar_t *T, const saidx_t *PA,
655*c08cbc64SXin LI              saidx_t *first, saidx_t *middle, saidx_t *last,
656*c08cbc64SXin LI              saidx_t *buf, saidx_t bufsize, saidx_t depth) {
657*c08cbc64SXin LI #define STACK_SIZE SS_SMERGE_STACKSIZE
658*c08cbc64SXin LI #define GETIDX(a) ((0 <= (a)) ? (a) : (~(a)))
659*c08cbc64SXin LI #define MERGE_CHECK(a, b, c)\
660*c08cbc64SXin LI   do {\
661*c08cbc64SXin LI     if(((c) & 1) ||\
662*c08cbc64SXin LI        (((c) & 2) && (ss_compare(T, PA + GETIDX(*((a) - 1)), PA + *(a), depth) == 0))) {\
663*c08cbc64SXin LI       *(a) = ~*(a);\
664*c08cbc64SXin LI     }\
665*c08cbc64SXin LI     if(((c) & 4) && ((ss_compare(T, PA + GETIDX(*((b) - 1)), PA + *(b), depth) == 0))) {\
666*c08cbc64SXin LI       *(b) = ~*(b);\
667*c08cbc64SXin LI     }\
668*c08cbc64SXin LI   } while(0)
669*c08cbc64SXin LI   struct { saidx_t *a, *b, *c; saint_t d; } stack[STACK_SIZE];
670*c08cbc64SXin LI   saidx_t *l, *r, *lm, *rm;
671*c08cbc64SXin LI   saidx_t m, len, half;
672*c08cbc64SXin LI   saint_t ssize;
673*c08cbc64SXin LI   saint_t check, next;
674*c08cbc64SXin LI 
675*c08cbc64SXin LI   for(check = 0, ssize = 0;;) {
676*c08cbc64SXin LI     if((last - middle) <= bufsize) {
677*c08cbc64SXin LI       if((first < middle) && (middle < last)) {
678*c08cbc64SXin LI         ss_mergebackward(T, PA, first, middle, last, buf, depth);
679*c08cbc64SXin LI       }
680*c08cbc64SXin LI       MERGE_CHECK(first, last, check);
681*c08cbc64SXin LI       STACK_POP(first, middle, last, check);
682*c08cbc64SXin LI       continue;
683*c08cbc64SXin LI     }
684*c08cbc64SXin LI 
685*c08cbc64SXin LI     if((middle - first) <= bufsize) {
686*c08cbc64SXin LI       if(first < middle) {
687*c08cbc64SXin LI         ss_mergeforward(T, PA, first, middle, last, buf, depth);
688*c08cbc64SXin LI       }
689*c08cbc64SXin LI       MERGE_CHECK(first, last, check);
690*c08cbc64SXin LI       STACK_POP(first, middle, last, check);
691*c08cbc64SXin LI       continue;
692*c08cbc64SXin LI     }
693*c08cbc64SXin LI 
694*c08cbc64SXin LI     for(m = 0, len = MIN(middle - first, last - middle), half = len >> 1;
695*c08cbc64SXin LI         0 < len;
696*c08cbc64SXin LI         len = half, half >>= 1) {
697*c08cbc64SXin LI       if(ss_compare(T, PA + GETIDX(*(middle + m + half)),
698*c08cbc64SXin LI                        PA + GETIDX(*(middle - m - half - 1)), depth) < 0) {
699*c08cbc64SXin LI         m += half + 1;
700*c08cbc64SXin LI         half -= (len & 1) ^ 1;
701*c08cbc64SXin LI       }
702*c08cbc64SXin LI     }
703*c08cbc64SXin LI 
704*c08cbc64SXin LI     if(0 < m) {
705*c08cbc64SXin LI       lm = middle - m, rm = middle + m;
706*c08cbc64SXin LI       ss_blockswap(lm, middle, m);
707*c08cbc64SXin LI       l = r = middle, next = 0;
708*c08cbc64SXin LI       if(rm < last) {
709*c08cbc64SXin LI         if(*rm < 0) {
710*c08cbc64SXin LI           *rm = ~*rm;
711*c08cbc64SXin LI           if(first < lm) { for(; *--l < 0;) { } next |= 4; }
712*c08cbc64SXin LI           next |= 1;
713*c08cbc64SXin LI         } else if(first < lm) {
714*c08cbc64SXin LI           for(; *r < 0; ++r) { }
715*c08cbc64SXin LI           next |= 2;
716*c08cbc64SXin LI         }
717*c08cbc64SXin LI       }
718*c08cbc64SXin LI 
719*c08cbc64SXin LI       if((l - first) <= (last - r)) {
720*c08cbc64SXin LI         STACK_PUSH(r, rm, last, (next & 3) | (check & 4));
721*c08cbc64SXin LI         middle = lm, last = l, check = (check & 3) | (next & 4);
722*c08cbc64SXin LI       } else {
723*c08cbc64SXin LI         if((next & 2) && (r == middle)) { next ^= 6; }
724*c08cbc64SXin LI         STACK_PUSH(first, lm, l, (check & 3) | (next & 4));
725*c08cbc64SXin LI         first = r, middle = rm, check = (next & 3) | (check & 4);
726*c08cbc64SXin LI       }
727*c08cbc64SXin LI     } else {
728*c08cbc64SXin LI       if(ss_compare(T, PA + GETIDX(*(middle - 1)), PA + *middle, depth) == 0) {
729*c08cbc64SXin LI         *middle = ~*middle;
730*c08cbc64SXin LI       }
731*c08cbc64SXin LI       MERGE_CHECK(first, last, check);
732*c08cbc64SXin LI       STACK_POP(first, middle, last, check);
733*c08cbc64SXin LI     }
734*c08cbc64SXin LI   }
735*c08cbc64SXin LI #undef STACK_SIZE
736*c08cbc64SXin LI }
737*c08cbc64SXin LI 
738*c08cbc64SXin LI #endif /* SS_BLOCKSIZE != 0 */
739*c08cbc64SXin LI 
740*c08cbc64SXin LI 
741*c08cbc64SXin LI /*---------------------------------------------------------------------------*/
742*c08cbc64SXin LI 
743*c08cbc64SXin LI /*- Function -*/
744*c08cbc64SXin LI 
745*c08cbc64SXin LI /* Substring sort */
746*c08cbc64SXin LI void
sssort(const sauchar_t * T,const saidx_t * PA,saidx_t * first,saidx_t * last,saidx_t * buf,saidx_t bufsize,saidx_t depth,saidx_t n,saint_t lastsuffix)747*c08cbc64SXin LI sssort(const sauchar_t *T, const saidx_t *PA,
748*c08cbc64SXin LI        saidx_t *first, saidx_t *last,
749*c08cbc64SXin LI        saidx_t *buf, saidx_t bufsize,
750*c08cbc64SXin LI        saidx_t depth, saidx_t n, saint_t lastsuffix) {
751*c08cbc64SXin LI   saidx_t *a;
752*c08cbc64SXin LI #if SS_BLOCKSIZE != 0
753*c08cbc64SXin LI   saidx_t *b, *middle, *curbuf;
754*c08cbc64SXin LI   saidx_t j, k, curbufsize, limit;
755*c08cbc64SXin LI #endif
756*c08cbc64SXin LI   saidx_t i;
757*c08cbc64SXin LI 
758*c08cbc64SXin LI   if(lastsuffix != 0) { ++first; }
759*c08cbc64SXin LI 
760*c08cbc64SXin LI #if SS_BLOCKSIZE == 0
761*c08cbc64SXin LI   ss_mintrosort(T, PA, first, last, depth);
762*c08cbc64SXin LI #else
763*c08cbc64SXin LI   if((bufsize < SS_BLOCKSIZE) &&
764*c08cbc64SXin LI       (bufsize < (last - first)) &&
765*c08cbc64SXin LI       (bufsize < (limit = ss_isqrt(last - first)))) {
766*c08cbc64SXin LI     if(SS_BLOCKSIZE < limit) { limit = SS_BLOCKSIZE; }
767*c08cbc64SXin LI     buf = middle = last - limit, bufsize = limit;
768*c08cbc64SXin LI   } else {
769*c08cbc64SXin LI     middle = last, limit = 0;
770*c08cbc64SXin LI   }
771*c08cbc64SXin LI   for(a = first, i = 0; SS_BLOCKSIZE < (middle - a); a += SS_BLOCKSIZE, ++i) {
772*c08cbc64SXin LI #if SS_INSERTIONSORT_THRESHOLD < SS_BLOCKSIZE
773*c08cbc64SXin LI     ss_mintrosort(T, PA, a, a + SS_BLOCKSIZE, depth);
774*c08cbc64SXin LI #elif 1 < SS_BLOCKSIZE
775*c08cbc64SXin LI     ss_insertionsort(T, PA, a, a + SS_BLOCKSIZE, depth);
776*c08cbc64SXin LI #endif
777*c08cbc64SXin LI     curbufsize = last - (a + SS_BLOCKSIZE);
778*c08cbc64SXin LI     curbuf = a + SS_BLOCKSIZE;
779*c08cbc64SXin LI     if(curbufsize <= bufsize) { curbufsize = bufsize, curbuf = buf; }
780*c08cbc64SXin LI     for(b = a, k = SS_BLOCKSIZE, j = i; j & 1; b -= k, k <<= 1, j >>= 1) {
781*c08cbc64SXin LI       ss_swapmerge(T, PA, b - k, b, b + k, curbuf, curbufsize, depth);
782*c08cbc64SXin LI     }
783*c08cbc64SXin LI   }
784*c08cbc64SXin LI #if SS_INSERTIONSORT_THRESHOLD < SS_BLOCKSIZE
785*c08cbc64SXin LI   ss_mintrosort(T, PA, a, middle, depth);
786*c08cbc64SXin LI #elif 1 < SS_BLOCKSIZE
787*c08cbc64SXin LI   ss_insertionsort(T, PA, a, middle, depth);
788*c08cbc64SXin LI #endif
789*c08cbc64SXin LI   for(k = SS_BLOCKSIZE; i != 0; k <<= 1, i >>= 1) {
790*c08cbc64SXin LI     if(i & 1) {
791*c08cbc64SXin LI       ss_swapmerge(T, PA, a - k, a, middle, buf, bufsize, depth);
792*c08cbc64SXin LI       a -= k;
793*c08cbc64SXin LI     }
794*c08cbc64SXin LI   }
795*c08cbc64SXin LI   if(limit != 0) {
796*c08cbc64SXin LI #if SS_INSERTIONSORT_THRESHOLD < SS_BLOCKSIZE
797*c08cbc64SXin LI     ss_mintrosort(T, PA, middle, last, depth);
798*c08cbc64SXin LI #elif 1 < SS_BLOCKSIZE
799*c08cbc64SXin LI     ss_insertionsort(T, PA, middle, last, depth);
800*c08cbc64SXin LI #endif
801*c08cbc64SXin LI     ss_inplacemerge(T, PA, first, middle, last, depth);
802*c08cbc64SXin LI   }
803*c08cbc64SXin LI #endif
804*c08cbc64SXin LI 
805*c08cbc64SXin LI   if(lastsuffix != 0) {
806*c08cbc64SXin LI     /* Insert last type B* suffix. */
807*c08cbc64SXin LI     saidx_t PAi[2]; PAi[0] = PA[*(first - 1)], PAi[1] = n - 2;
808*c08cbc64SXin LI     for(a = first, i = *(first - 1);
809*c08cbc64SXin LI         (a < last) && ((*a < 0) || (0 < ss_compare(T, &(PAi[0]), PA + *a, depth)));
810*c08cbc64SXin LI         ++a) {
811*c08cbc64SXin LI       *(a - 1) = *a;
812*c08cbc64SXin LI     }
813*c08cbc64SXin LI     *(a - 1) = i;
814*c08cbc64SXin LI   }
815*c08cbc64SXin LI }
816