freebsd-dev/usr.bin/tr/cset.c

291 lines
6.6 KiB
C

/*-
* Copyright (c) 2004 Tim J. Robbins.
* 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.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR 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 AUTHOR 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.
*/
/*
* "Set of characters" ADT implemented as a splay tree of extents, with
* a lookup table cache to simplify looking up the first bunch of
* characters (which are presumably more common than others).
*/
#include <sys/cdefs.h>
__FBSDID("$FreeBSD$");
#include <assert.h>
#include <stdbool.h>
#include <stdlib.h>
#include <wchar.h>
#include <wctype.h>
#include "cset.h"
static struct csnode * cset_delete(struct csnode *, wchar_t);
static __inline int cset_rangecmp(struct csnode *, wchar_t);
static struct csnode * cset_splay(struct csnode *, wchar_t);
/*
* cset_alloc --
* Allocate a set of characters.
*/
struct cset *
cset_alloc(void)
{
struct cset *cs;
if ((cs = malloc(sizeof(*cs))) == NULL)
return (NULL);
cs->cs_root = NULL;
cs->cs_classes = NULL;
cs->cs_havecache = false;
cs->cs_invert = false;
return (cs);
}
/*
* cset_add --
* Add a character to the set.
*/
bool
cset_add(struct cset *cs, wchar_t ch)
{
struct csnode *csn, *ncsn;
wchar_t oval;
cs->cs_havecache = false;
/*
* Inserting into empty tree; new item becomes the root.
*/
if (cs->cs_root == NULL) {
csn = malloc(sizeof(*cs->cs_root));
if (csn == NULL)
return (false);
csn->csn_left = csn->csn_right = NULL;
csn->csn_min = csn->csn_max = ch;
cs->cs_root = csn;
return (true);
}
/*
* Splay to check whether the item already exists, and otherwise,
* where we should put it.
*/
csn = cs->cs_root = cset_splay(cs->cs_root, ch);
/*
* Avoid adding duplicate nodes.
*/
if (cset_rangecmp(csn, ch) == 0)
return (true);
/*
* Allocate a new node and make it the new root.
*/
ncsn = malloc(sizeof(*ncsn));
if (ncsn == NULL)
return (false);
ncsn->csn_min = ncsn->csn_max = ch;
if (cset_rangecmp(csn, ch) < 0) {
ncsn->csn_left = csn->csn_left;
ncsn->csn_right = csn;
csn->csn_left = NULL;
} else {
ncsn->csn_right = csn->csn_right;
ncsn->csn_left = csn;
csn->csn_right = NULL;
}
cs->cs_root = ncsn;
/*
* Coalesce with left and right neighbours if possible.
*/
if (ncsn->csn_left != NULL) {
ncsn->csn_left = cset_splay(ncsn->csn_left, ncsn->csn_min - 1);
if (ncsn->csn_left->csn_max == ncsn->csn_min - 1) {
oval = ncsn->csn_left->csn_min;
ncsn->csn_left = cset_delete(ncsn->csn_left,
ncsn->csn_left->csn_min);
ncsn->csn_min = oval;
}
}
if (ncsn->csn_right != NULL) {
ncsn->csn_right = cset_splay(ncsn->csn_right,
ncsn->csn_max + 1);
if (ncsn->csn_right->csn_min == ncsn->csn_max + 1) {
oval = ncsn->csn_right->csn_max;
ncsn->csn_right = cset_delete(ncsn->csn_right,
ncsn->csn_right->csn_min);
ncsn->csn_max = oval;
}
}
return (true);
}
/*
* cset_in_hard --
* Determine whether a character is in the set without using
* the cache.
*/
bool
cset_in_hard(struct cset *cs, wchar_t ch)
{
struct csclass *csc;
for (csc = cs->cs_classes; csc != NULL; csc = csc->csc_next)
if (csc->csc_invert ^ iswctype(ch, csc->csc_type) != 0)
return (cs->cs_invert ^ true);
if (cs->cs_root != NULL) {
cs->cs_root = cset_splay(cs->cs_root, ch);
return (cs->cs_invert ^ cset_rangecmp(cs->cs_root, ch) == 0);
}
return (cs->cs_invert ^ false);
}
/*
* cset_cache --
* Update the cache.
*/
void
cset_cache(struct cset *cs)
{
wchar_t i;
for (i = 0; i < CS_CACHE_SIZE; i++)
cs->cs_cache[i] = cset_in_hard(cs, i);
cs->cs_havecache = true;
}
/*
* cset_invert --
* Invert the character set.
*/
void
cset_invert(struct cset *cs)
{
cs->cs_invert ^= true;
cs->cs_havecache = false;
}
/*
* cset_addclass --
* Add a wctype()-style character class to the set, optionally
* inverting it.
*/
bool
cset_addclass(struct cset *cs, wctype_t type, bool invert)
{
struct csclass *csc;
csc = malloc(sizeof(*csc));
if (csc == NULL)
return (false);
csc->csc_type = type;
csc->csc_invert = invert;
csc->csc_next = cs->cs_classes;
cs->cs_classes = csc;
cs->cs_havecache = false;
return (true);
}
static __inline int
cset_rangecmp(struct csnode *t, wchar_t ch)
{
if (ch < t->csn_min)
return (-1);
if (ch > t->csn_max)
return (1);
return (0);
}
static struct csnode *
cset_splay(struct csnode *t, wchar_t ch)
{
struct csnode N, *l, *r, *y;
/*
* Based on public domain code from Sleator.
*/
assert(t != NULL);
N.csn_left = N.csn_right = NULL;
l = r = &N;
for (;;) {
if (cset_rangecmp(t, ch) < 0) {
if (t->csn_left != NULL &&
cset_rangecmp(t->csn_left, ch) < 0) {
y = t->csn_left;
t->csn_left = y->csn_right;
y->csn_right = t;
t = y;
}
if (t->csn_left == NULL)
break;
r->csn_left = t;
r = t;
t = t->csn_left;
} else if (cset_rangecmp(t, ch) > 0) {
if (t->csn_right != NULL &&
cset_rangecmp(t->csn_right, ch) > 0) {
y = t->csn_right;
t->csn_right = y->csn_left;
y->csn_left = t;
t = y;
}
if (t->csn_right == NULL)
break;
l->csn_right = t;
l = t;
t = t->csn_right;
} else
break;
}
l->csn_right = t->csn_left;
r->csn_left = t->csn_right;
t->csn_left = N.csn_right;
t->csn_right = N.csn_left;
return (t);
}
static struct csnode *
cset_delete(struct csnode *t, wchar_t ch)
{
struct csnode *x;
assert(t != NULL);
t = cset_splay(t, ch);
assert(cset_rangecmp(t, ch) == 0);
if (t->csn_left == NULL)
x = t->csn_right;
else {
x = cset_splay(t->csn_left, ch);
x->csn_right = t->csn_right;
}
free(t);
return x;
}