-/* PSPP - computes sample statistics.
- Copyright (C) 2007 Free Software Foundation, Inc.
+/* PSPP - a program for statistical analysis.
+ Copyright (C) 2007, 2009, 2010, 2011 Free Software Foundation, Inc.
- This program is free software; you can redistribute it and/or
- modify it under the terms of the GNU General Public License as
- published by the Free Software Foundation; either version 2 of the
- License, or (at your option) any later version.
+ This program is free software: you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation, either version 3 of the License, or
+ (at your option) any later version.
- This program is distributed in the hope that it will be useful, but
- WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- General Public License for more details.
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
You should have received a copy of the GNU General Public License
- along with this program; if not, write to the Free Software
- Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
- 02110-1301, USA. */
+ along with this program. If not, see <http://www.gnu.org/licenses/>. */
/* Balanced tree (BT) data structure.
#include <config.h>
#endif
-#include <libpspp/bt.h>
+#include "libpspp/bt.h"
+#include <limits.h>
#include <stdbool.h>
#include <stdint.h>
+#include "libpspp/cast.h"
+
static void rebalance_subtree (struct bt *, struct bt_node *, size_t);
static struct bt_node **down_link (struct bt *, struct bt_node *);
/* Initializes BT as an empty BT that uses the given COMPARE
function, passing in AUX as auxiliary data. */
void
-bt_init (struct bt *bt,
- bt_compare_func *compare,
- const void *aux)
+bt_init (struct bt *bt, bt_compare_func *compare, const void *aux)
{
bt->root = NULL;
bt->compare = compare;
bt_insert (struct bt *bt, struct bt_node *node)
{
int depth = 0;
-
+
node->down[0] = NULL;
node->down[1] = NULL;
{
size += 1 + count_nodes_in_subtree (sibling (s));
s = s->up;
- if (i > calculate_h_alpha (size))
+ if (i > calculate_h_alpha (size))
{
rebalance_subtree (bt, s, size);
break;
}
}
- else
+ else
{
rebalance_subtree (bt, bt->root, bt->size);
bt->max_size = bt->size;
break;
}
}
-
+
return NULL;
}
{
struct bt_node **q = down_link (bt, p);
struct bt_node *r = p->down[1];
- if (r == NULL)
+ if (r == NULL)
{
*q = p->down[0];
if (*q)
will cause us to do a little more rebalancing than strictly
necessary to maintain the scapegoat tree's height
invariant. */
- if (bt->size < bt->max_size * 3 / 4 && bt->size > 0)
+ if (bt->size < bt->max_size * 3 / 4 && bt->size > 0)
{
rebalance_subtree (bt, bt->root, bt->size);
bt->max_size = bt->size;
{
cmp = bt->compare (target, p, bt->aux);
if (cmp == 0)
- return (struct bt_node *) p;
+ return CONST_CAST (struct bt_node *, p);
}
return NULL;
{
const struct bt_node *p = bt->root;
const struct bt_node *q = NULL;
- while (p != NULL)
+ while (p != NULL)
{
int cmp = bt->compare (target, p, bt->aux);
if (cmp > 0)
p = p->down[1];
- else
+ else
{
q = p;
if (cmp < 0)
break;
}
}
- return (struct bt_node *) q;
+ return CONST_CAST (struct bt_node *, q);
}
/* Searches BT for, and returns, the last node in in-order whose
{
const struct bt_node *p = bt->root;
const struct bt_node *q = NULL;
- while (p != NULL)
+ while (p != NULL)
{
int cmp = bt->compare (target, p, bt->aux);
if (cmp < 0)
p = p->down[0];
- else
+ else
{
q = p;
if (cmp > 0)
break;
}
}
- return (struct bt_node *) q;
+ return CONST_CAST (struct bt_node *, q);
}
/* Returns the node in BT following P in in-order.
p = p->down[1];
while (p->down[0] != NULL)
p = p->down[0];
- return (struct bt_node *) p;
+ return CONST_CAST (struct bt_node *, p);
}
}
p = p->down[0];
while (p->down[1] != NULL)
p = p->down[1];
- return (struct bt_node *) p;
+ return CONST_CAST (struct bt_node *, p);
}
}
vine_nodes /= 2;
compress (q, vine_nodes);
}
- while ((*q)->down[0] != NULL)
+ while ((*q)->down[0] != NULL)
{
(*q)->down[0]->up = *q;
q = &(*q)->down[0];
/* Returns node P's sibling; that is, the other child of its
parent. P must not be the root. */
static inline struct bt_node *
-sibling (struct bt_node *p)
+sibling (struct bt_node *p)
{
struct bt_node *q = p->up;
return q->down[q->down[0] == p];
/* Returns the number of nodes in the given SUBTREE. */
static size_t
-count_nodes_in_subtree (const struct bt_node *subtree)
+count_nodes_in_subtree (const struct bt_node *subtree)
{
/* This is an in-order traversal modified to iterate only the
nodes in SUBTREE. */
size_t count = 0;
- if (subtree != NULL)
+ if (subtree != NULL)
{
const struct bt_node *p = subtree;
while (p->down[0] != NULL)
p = p->down[0];
- for (;;)
+ for (;;)
{
count++;
- if (p->down[1] != NULL)
+ if (p->down[1] != NULL)
{
p = p->down[1];
while (p->down[0] != NULL)
p = p->down[0];
}
- else
+ else
{
- for (;;)
+ for (;;)
{
const struct bt_node *q;
if (p == subtree)
p = p->up;
if (p->down[0] == q)
break;
- }
+ }
}
}
}
/* Returns the number of high-order 0-bits in X.
Undefined if X is zero. */
static inline int
-count_leading_zeros (size_t x)
+count_leading_zeros (size_t x)
{
#if __GNUC__ >= 4
#if SIZE_MAX > ULONG_MAX
/* Returns floor(log2(x)).
Undefined if X is zero. */
static inline int
-floor_log2 (size_t x)
+floor_log2 (size_t x)
{
return sizeof (size_t) * CHAR_BIT - 1 - count_leading_zeros (x);
}
Defined for X from 0 up to the number of bits in size_t minus
1. */
static inline size_t
-pow_sqrt2 (int x)
+pow_sqrt2 (int x)
{
/* These constants are sqrt(2) multiplied by 2**63 or 2**31,
respectively, and then rounded to nearest. */
/* Returns floor(log(n)/log(sqrt(2))).
Undefined if N is 0. */
static inline int
-calculate_h_alpha (size_t n)
+calculate_h_alpha (size_t n)
{
int log2 = floor_log2 (n);
/* The correct answer is either 2 * log2 or one more. So we
- see if n >= pow(sqrt(2), 2 * log2 + 1) and if so, add 1. */
+ see if n >= pow(sqrt(2), 2 * log2 + 1) and if so, add 1. */
return (2 * log2) + (n >= pow_sqrt2 (log2));
}