Use sort_template.h for qsort() and qsort_arg().
authorThomas Munro <tmunro@postgresql.org>
Wed, 3 Mar 2021 03:57:26 +0000 (16:57 +1300)
committerThomas Munro <tmunro@postgresql.org>
Wed, 3 Mar 2021 04:02:32 +0000 (17:02 +1300)
Reduce duplication by using the new template.

Reviewed-by: Daniel Gustafsson <daniel@yesql.se>
Discussion: https://postgr.es/m/CA%2BhUKGJ2-eaDqAum5bxhpMNhvuJmRDZxB_Tow0n-gse%2BHG0Yig%40mail.gmail.com

src/port/qsort.c
src/port/qsort_arg.c

index fa992e2081df5d7756fef0a12c240e19d1f3184a..7879e6cd563c3f37a6a7f2140dcb06750e5f4871 100644 (file)
 /*
  *     qsort.c: standard quicksort algorithm
- *
- *     Modifications from vanilla NetBSD source:
- *       Add do ... while() macro fix
- *       Remove __inline, _DIAGASSERTs, __P
- *       Remove ill-considered "swap_cnt" switch to insertion sort,
- *       in favor of a simple check for presorted input.
- *       Take care to recurse on the smaller partition, to bound stack usage.
- *
- *     CAUTION: if you change this file, see also qsort_arg.c, gen_qsort_tuple.pl
- *
- *     src/port/qsort.c
- */
-
-/*     $NetBSD: qsort.c,v 1.13 2003/08/07 16:43:42 agc Exp $   */
-
-/*-
- * Copyright (c) 1992, 1993
- *     The Regents of the University of California.  All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- * 1. Redistributions of source code must retain the above copyright
- *       notice, this list of conditions and the following disclaimer.
- * 2. Redistributions in binary form must reproduce the above copyright
- *       notice, this list of conditions and the following disclaimer in the
- *       documentation and/or other materials provided with the distribution.
- * 3. Neither the name of the University nor the names of its contributors
- *       may be used to endorse or promote products derived from this software
- *       without specific prior written permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
- * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
- * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
- * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
- * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
- * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
- * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
- * SUCH DAMAGE.
  */
 
 #include "c.h"
 
-
-static char *med3(char *a, char *b, char *c,
-                                 int (*cmp) (const void *, const void *));
-static void swapfunc(char *, char *, size_t, int);
-
-/*
- * Qsort routine based on J. L. Bentley and M. D. McIlroy,
- * "Engineering a sort function",
- * Software--Practice and Experience 23 (1993) 1249-1265.
- *
- * We have modified their original by adding a check for already-sorted input,
- * which seems to be a win per discussions on pgsql-hackers around 2006-03-21.
- *
- * Also, we recurse on the smaller partition and iterate on the larger one,
- * which ensures we cannot recurse more than log(N) levels (since the
- * partition recursed to is surely no more than half of the input).  Bentley
- * and McIlroy explicitly rejected doing this on the grounds that it's "not
- * worth the effort", but we have seen crashes in the field due to stack
- * overrun, so that judgment seems wrong.
- */
-
-#define swapcode(TYPE, parmi, parmj, n) \
-do {           \
-       size_t i = (n) / sizeof (TYPE);                 \
-       TYPE *pi = (TYPE *)(void *)(parmi);                     \
-       TYPE *pj = (TYPE *)(void *)(parmj);                     \
-       do {                                            \
-               TYPE    t = *pi;                        \
-               *pi++ = *pj;                            \
-               *pj++ = t;                              \
-               } while (--i > 0);                              \
-} while (0)
-
-#define SWAPINIT(a, es) swaptype = ((char *)(a) - (char *)0) % sizeof(long) || \
-       (es) % sizeof(long) ? 2 : (es) == sizeof(long)? 0 : 1
-
-static void
-swapfunc(char *a, char *b, size_t n, int swaptype)
-{
-       if (swaptype <= 1)
-               swapcode(long, a, b, n);
-       else
-               swapcode(char, a, b, n);
-}
-
-#define swap(a, b)                                             \
-       if (swaptype == 0) {                                    \
-               long t = *(long *)(void *)(a);                  \
-               *(long *)(void *)(a) = *(long *)(void *)(b);    \
-               *(long *)(void *)(b) = t;                       \
-       } else                                                  \
-               swapfunc(a, b, es, swaptype)
-
-#define vecswap(a, b, n) if ((n) > 0) swapfunc(a, b, n, swaptype)
-
-static char *
-med3(char *a, char *b, char *c, int (*cmp) (const void *, const void *))
-{
-       return cmp(a, b) < 0 ?
-               (cmp(b, c) < 0 ? b : (cmp(a, c) < 0 ? c : a))
-               : (cmp(b, c) > 0 ? b : (cmp(a, c) < 0 ? a : c));
-}
-
-void
-pg_qsort(void *a, size_t n, size_t es, int (*cmp) (const void *, const void *))
-{
-       char       *pa,
-                          *pb,
-                          *pc,
-                          *pd,
-                          *pl,
-                          *pm,
-                          *pn;
-       size_t          d1,
-                               d2;
-       int                     r,
-                               swaptype,
-                               presorted;
-
-loop:SWAPINIT(a, es);
-       if (n < 7)
-       {
-               for (pm = (char *) a + es; pm < (char *) a + n * es; pm += es)
-                       for (pl = pm; pl > (char *) a && cmp(pl - es, pl) > 0;
-                                pl -= es)
-                               swap(pl, pl - es);
-               return;
-       }
-       presorted = 1;
-       for (pm = (char *) a + es; pm < (char *) a + n * es; pm += es)
-       {
-               if (cmp(pm - es, pm) > 0)
-               {
-                       presorted = 0;
-                       break;
-               }
-       }
-       if (presorted)
-               return;
-       pm = (char *) a + (n / 2) * es;
-       if (n > 7)
-       {
-               pl = (char *) a;
-               pn = (char *) a + (n - 1) * es;
-               if (n > 40)
-               {
-                       size_t          d = (n / 8) * es;
-
-                       pl = med3(pl, pl + d, pl + 2 * d, cmp);
-                       pm = med3(pm - d, pm, pm + d, cmp);
-                       pn = med3(pn - 2 * d, pn - d, pn, cmp);
-               }
-               pm = med3(pl, pm, pn, cmp);
-       }
-       swap(a, pm);
-       pa = pb = (char *) a + es;
-       pc = pd = (char *) a + (n - 1) * es;
-       for (;;)
-       {
-               while (pb <= pc && (r = cmp(pb, a)) <= 0)
-               {
-                       if (r == 0)
-                       {
-                               swap(pa, pb);
-                               pa += es;
-                       }
-                       pb += es;
-               }
-               while (pb <= pc && (r = cmp(pc, a)) >= 0)
-               {
-                       if (r == 0)
-                       {
-                               swap(pc, pd);
-                               pd -= es;
-                       }
-                       pc -= es;
-               }
-               if (pb > pc)
-                       break;
-               swap(pb, pc);
-               pb += es;
-               pc -= es;
-       }
-       pn = (char *) a + n * es;
-       d1 = Min(pa - (char *) a, pb - pa);
-       vecswap(a, pb - d1, d1);
-       d1 = Min(pd - pc, pn - pd - es);
-       vecswap(pb, pn - d1, d1);
-       d1 = pb - pa;
-       d2 = pd - pc;
-       if (d1 <= d2)
-       {
-               /* Recurse on left partition, then iterate on right partition */
-               if (d1 > es)
-                       pg_qsort(a, d1 / es, es, cmp);
-               if (d2 > es)
-               {
-                       /* Iterate rather than recurse to save stack space */
-                       /* pg_qsort(pn - d2, d2 / es, es, cmp); */
-                       a = pn - d2;
-                       n = d2 / es;
-                       goto loop;
-               }
-       }
-       else
-       {
-               /* Recurse on right partition, then iterate on left partition */
-               if (d2 > es)
-                       pg_qsort(pn - d2, d2 / es, es, cmp);
-               if (d1 > es)
-               {
-                       /* Iterate rather than recurse to save stack space */
-                       /* pg_qsort(a, d1 / es, es, cmp); */
-                       n = d1 / es;
-                       goto loop;
-               }
-       }
-}
+#define ST_SORT pg_qsort
+#define ST_ELEMENT_TYPE_VOID
+#define ST_COMPARE_RUNTIME_POINTER
+#define ST_SCOPE
+#define ST_DECLARE
+#define ST_DEFINE
+#include "lib/sort_template.h"
 
 /*
  * qsort comparator wrapper for strcmp.
index 6d54fbc2b42541e7f1d5b071ca283fec734720b6..fa7e11a3b8627aff4e7a61264a517eaadb3f246a 100644 (file)
 /*
  *     qsort_arg.c: qsort with a passthrough "void *" argument
- *
- *     Modifications from vanilla NetBSD source:
- *       Add do ... while() macro fix
- *       Remove __inline, _DIAGASSERTs, __P
- *       Remove ill-considered "swap_cnt" switch to insertion sort,
- *       in favor of a simple check for presorted input.
- *       Take care to recurse on the smaller partition, to bound stack usage.
- *
- *     CAUTION: if you change this file, see also qsort.c, gen_qsort_tuple.pl
- *
- *     src/port/qsort_arg.c
- */
-
-/*     $NetBSD: qsort.c,v 1.13 2003/08/07 16:43:42 agc Exp $   */
-
-/*-
- * Copyright (c) 1992, 1993
- *     The Regents of the University of California.  All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- * 1. Redistributions of source code must retain the above copyright
- *       notice, this list of conditions and the following disclaimer.
- * 2. Redistributions in binary form must reproduce the above copyright
- *       notice, this list of conditions and the following disclaimer in the
- *       documentation and/or other materials provided with the distribution.
- * 3. Neither the name of the University nor the names of its contributors
- *       may be used to endorse or promote products derived from this software
- *       without specific prior written permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
- * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
- * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
- * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
- * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
- * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
- * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
- * SUCH DAMAGE.
  */
 
 #include "c.h"
 
-
-static char *med3(char *a, char *b, char *c,
-                                 qsort_arg_comparator cmp, void *arg);
-static void swapfunc(char *, char *, size_t, int);
-
-/*
- * Qsort routine based on J. L. Bentley and M. D. McIlroy,
- * "Engineering a sort function",
- * Software--Practice and Experience 23 (1993) 1249-1265.
- *
- * We have modified their original by adding a check for already-sorted input,
- * which seems to be a win per discussions on pgsql-hackers around 2006-03-21.
- *
- * Also, we recurse on the smaller partition and iterate on the larger one,
- * which ensures we cannot recurse more than log(N) levels (since the
- * partition recursed to is surely no more than half of the input).  Bentley
- * and McIlroy explicitly rejected doing this on the grounds that it's "not
- * worth the effort", but we have seen crashes in the field due to stack
- * overrun, so that judgment seems wrong.
- */
-
-#define swapcode(TYPE, parmi, parmj, n) \
-do {           \
-       size_t i = (n) / sizeof (TYPE);                 \
-       TYPE *pi = (TYPE *)(void *)(parmi);                     \
-       TYPE *pj = (TYPE *)(void *)(parmj);                     \
-       do {                                            \
-               TYPE    t = *pi;                        \
-               *pi++ = *pj;                            \
-               *pj++ = t;                              \
-               } while (--i > 0);                              \
-} while (0)
-
-#define SWAPINIT(a, es) swaptype = ((char *)(a) - (char *)0) % sizeof(long) || \
-       (es) % sizeof(long) ? 2 : (es) == sizeof(long)? 0 : 1
-
-static void
-swapfunc(char *a, char *b, size_t n, int swaptype)
-{
-       if (swaptype <= 1)
-               swapcode(long, a, b, n);
-       else
-               swapcode(char, a, b, n);
-}
-
-#define swap(a, b)                                             \
-       if (swaptype == 0) {                                    \
-               long t = *(long *)(void *)(a);                  \
-               *(long *)(void *)(a) = *(long *)(void *)(b);    \
-               *(long *)(void *)(b) = t;                       \
-       } else                                                  \
-               swapfunc(a, b, es, swaptype)
-
-#define vecswap(a, b, n) if ((n) > 0) swapfunc(a, b, n, swaptype)
-
-static char *
-med3(char *a, char *b, char *c, qsort_arg_comparator cmp, void *arg)
-{
-       return cmp(a, b, arg) < 0 ?
-               (cmp(b, c, arg) < 0 ? b : (cmp(a, c, arg) < 0 ? c : a))
-               : (cmp(b, c, arg) > 0 ? b : (cmp(a, c, arg) < 0 ? a : c));
-}
-
-void
-qsort_arg(void *a, size_t n, size_t es, qsort_arg_comparator cmp, void *arg)
-{
-       char       *pa,
-                          *pb,
-                          *pc,
-                          *pd,
-                          *pl,
-                          *pm,
-                          *pn;
-       size_t          d1,
-                               d2;
-       int                     r,
-                               swaptype,
-                               presorted;
-
-loop:SWAPINIT(a, es);
-       if (n < 7)
-       {
-               for (pm = (char *) a + es; pm < (char *) a + n * es; pm += es)
-                       for (pl = pm; pl > (char *) a && cmp(pl - es, pl, arg) > 0;
-                                pl -= es)
-                               swap(pl, pl - es);
-               return;
-       }
-       presorted = 1;
-       for (pm = (char *) a + es; pm < (char *) a + n * es; pm += es)
-       {
-               if (cmp(pm - es, pm, arg) > 0)
-               {
-                       presorted = 0;
-                       break;
-               }
-       }
-       if (presorted)
-               return;
-       pm = (char *) a + (n / 2) * es;
-       if (n > 7)
-       {
-               pl = (char *) a;
-               pn = (char *) a + (n - 1) * es;
-               if (n > 40)
-               {
-                       size_t          d = (n / 8) * es;
-
-                       pl = med3(pl, pl + d, pl + 2 * d, cmp, arg);
-                       pm = med3(pm - d, pm, pm + d, cmp, arg);
-                       pn = med3(pn - 2 * d, pn - d, pn, cmp, arg);
-               }
-               pm = med3(pl, pm, pn, cmp, arg);
-       }
-       swap(a, pm);
-       pa = pb = (char *) a + es;
-       pc = pd = (char *) a + (n - 1) * es;
-       for (;;)
-       {
-               while (pb <= pc && (r = cmp(pb, a, arg)) <= 0)
-               {
-                       if (r == 0)
-                       {
-                               swap(pa, pb);
-                               pa += es;
-                       }
-                       pb += es;
-               }
-               while (pb <= pc && (r = cmp(pc, a, arg)) >= 0)
-               {
-                       if (r == 0)
-                       {
-                               swap(pc, pd);
-                               pd -= es;
-                       }
-                       pc -= es;
-               }
-               if (pb > pc)
-                       break;
-               swap(pb, pc);
-               pb += es;
-               pc -= es;
-       }
-       pn = (char *) a + n * es;
-       d1 = Min(pa - (char *) a, pb - pa);
-       vecswap(a, pb - d1, d1);
-       d1 = Min(pd - pc, pn - pd - es);
-       vecswap(pb, pn - d1, d1);
-       d1 = pb - pa;
-       d2 = pd - pc;
-       if (d1 <= d2)
-       {
-               /* Recurse on left partition, then iterate on right partition */
-               if (d1 > es)
-                       qsort_arg(a, d1 / es, es, cmp, arg);
-               if (d2 > es)
-               {
-                       /* Iterate rather than recurse to save stack space */
-                       /* qsort_arg(pn - d2, d2 / es, es, cmp, arg); */
-                       a = pn - d2;
-                       n = d2 / es;
-                       goto loop;
-               }
-       }
-       else
-       {
-               /* Recurse on right partition, then iterate on left partition */
-               if (d2 > es)
-                       qsort_arg(pn - d2, d2 / es, es, cmp, arg);
-               if (d1 > es)
-               {
-                       /* Iterate rather than recurse to save stack space */
-                       /* qsort_arg(a, d1 / es, es, cmp, arg); */
-                       n = d1 / es;
-                       goto loop;
-               }
-       }
-}
+#define ST_SORT qsort_arg
+#define ST_ELEMENT_TYPE_VOID
+#define ST_COMPARATOR_TYPE_NAME qsort_arg_comparator
+#define ST_COMPARE_RUNTIME_POINTER
+#define ST_COMPARE_ARG_TYPE void
+#define ST_SCOPE
+#define ST_DEFINE
+#include "lib/sort_template.h"