diff options
author | Igor Pashev <pashev.igor@gmail.com> | 2012-06-24 22:28:35 +0000 |
---|---|---|
committer | Igor Pashev <pashev.igor@gmail.com> | 2012-06-24 22:28:35 +0000 |
commit | 3950ffe2a485479f6561c27364d3d7df5a21d124 (patch) | |
tree | 468c6e14449d1b1e279222ec32f676b0311917d2 /src/lib/libast/comp/tsearch.c | |
download | ksh-upstream.tar.gz |
Imported Upstream version 93u+upstream
Diffstat (limited to 'src/lib/libast/comp/tsearch.c')
-rw-r--r-- | src/lib/libast/comp/tsearch.c | 240 |
1 files changed, 240 insertions, 0 deletions
diff --git a/src/lib/libast/comp/tsearch.c b/src/lib/libast/comp/tsearch.c new file mode 100644 index 0000000..486f082 --- /dev/null +++ b/src/lib/libast/comp/tsearch.c @@ -0,0 +1,240 @@ +/*********************************************************************** +* * +* This software is part of the ast package * +* Copyright (c) 1985-2011 AT&T Intellectual Property * +* and is licensed under the * +* Eclipse Public License, Version 1.0 * +* by AT&T Intellectual Property * +* * +* A copy of the License is available at * +* http://www.eclipse.org/org/documents/epl-v10.html * +* (with md5 checksum b35adb5213ca9657e911e9befb180842) * +* * +* Information and Software Systems Research * +* AT&T Research * +* Florham Park NJ * +* * +* Glenn Fowler <gsf@research.att.com> * +* David Korn <dgk@research.att.com> * +* Phong Vo <kpv@research.att.com> * +* * +***********************************************************************/ +/* + * tsearch() for systems that have <search.h> but no tsearch() + * why would such a system provide the interface but not the + * implementation? that's what happens when one slimes their + * way through standards compliance + * + * NOTE: please excuse the crude feature test + */ + +#if !_UWIN + +void _STUB_tsearch(){} + +#else + +#if _PACKAGE_ast +#include <ast.h> +#endif + +#define tdelete ______tdelete +#define tfind ______tfind +#define tsearch ______tsearch +#define twalk ______twalk + +#include <search.h> + +#undef tdelete +#undef tfind +#undef tsearch +#undef twalk + +#include "dthdr.h" + +extern Void_t* dtfinger(Dt_t*); + +/* POSIX tsearch library based on libcdt +** Written by Kiem-Phong Vo (AT&T Research, 07/19/95) +*/ + +typedef struct _tree_s +{ Dtlink_t link; + Void_t* key; +} Tree_t; + +typedef struct _treedisc_s +{ Dtdisc_t disc; + int(* comparf)_ARG_((const Void_t*, const Void_t*)); +} Treedisc_t; + +#if defined(__EXPORT__) +#define extern __EXPORT__ +#endif + +/* compare function */ +#if __STD_C +static int treecompare(Dt_t* dt, char* one, char* two, Dtdisc_t* disc) +#else +static int treecompare(dt, one, two, disc) +Dt_t* dt; +char* one; +char* two; +Dtdisc_t* disc; +#endif +{ + return (*((Treedisc_t*)disc)->comparf)((Void_t*)one,(Void_t*)two); +} + +static Treedisc_t Treedisc = +{ { sizeof(Dtlink_t), -1, /* object is key */ + 0, + NIL(Dtmake_f), NIL(Dtfree_f), + treecompare, + NIL(Dthash_f), + NIL(Dtmemory_f), + NIL(Dtevent_f) + }, + 0 +}; + +extern +#if __STD_C +Void_t* tsearch(const Void_t* key, Void_t** rootp, + int(*comparf)(const Void_t*,const Void_t*) ) +#else +Void_t* tsearch(key, rootp, comparf) +Void_t* key; +Void_t** rootp; +int(* comparf)(); +#endif +{ + reg Dt_t* dt; + reg Tree_t* o; + + if(!rootp || + (!(dt = *((Dt_t**)rootp)) && !(dt = dtopen((Dtdisc_t*)(&Treedisc),Dtoset))) ) + return NIL(Void_t*); + + /* dangerous to set comparf on each call but that's tsearch */ + Treedisc.comparf = comparf; + + if(!(o = (Tree_t*)dtmatch(dt,key)) ) + { if(!(o = (Tree_t*)malloc(sizeof(Tree_t))) ) + return NIL(Void_t*); + o->key = (Void_t*)key; + dtinsert(dt,o); + } + + if(o) + *rootp = (Void_t*)dt; + else if(*rootp == NIL(Void_t*) ) + dtclose(dt); + + return (Void_t*)(&o->key); +} + +extern +#if __STD_C +Void_t* tfind(const Void_t* key, Void_t*const* rootp, + int(*comparf)(const Void_t*, const Void_t*) ) +#else +Void_t* tfind(key, rootp, comparf) +Void_t* key; +Void_t** rootp; +int(* comparf)(); +#endif +{ + reg Dt_t* dt; + reg Tree_t* o; + + if(!rootp || !(dt = *((Dt_t**)rootp)) ) + return NIL(Void_t*); + Treedisc.comparf = comparf; + + return (o = (Tree_t*)dtmatch(dt,key)) ? (Void_t*)(&o->key) : NIL(Void_t*); +} + +/* the original tdelete() specifies that it will return the parent pointer +** in the tree if there is one. Since we are using a splay tree, a deleted +** node is always rotated to the root first. So this implementation always +** returns the key of the new root. +*/ +extern +#if __STD_C +Void_t* tdelete(const Void_t* key, Void_t** rootp, + int(*comparf)(const Void_t*, const Void_t*) ) +#else +Void_t* tdelete(key, rootp, comparf) +Void_t* key; +Void_t** rootp; +int(* comparf)(); +#endif +{ + reg Dt_t* dt; + reg Tree_t* o; + Tree_t obj; + + if(!rootp || !(dt = *((Dt_t**)rootp)) ) + return NIL(Void_t*); + + Treedisc.comparf = comparf; + + obj.key = (Void_t*)key; + dtdelete(dt,&obj); + + if(!(o = dtfinger(dt)) ) + { dtclose(dt); + *rootp = NIL(Void_t*); + } + + return o ? (Void_t*)(&o->key) : NIL(Void_t*); +} + +/* the below routine assumes a particular layout of Dtlink_t. +** If this ever gets changed, this routine should be redone. +*/ +#define lchild link.lh.__left +#define rchild link.rh.__rght + +#if __STD_C +static void _twalk(Tree_t* obj, void(*action)(const Void_t*,VISIT,int), int level) +#else +static void _twalk(obj,action,level) +Tree_t* obj; +void(* action)(); +int level; +#endif +{ if(!obj->lchild && !obj->rchild) + (*action)((Void_t*)obj,leaf,level); + else + { (*action)((Void_t*)obj,preorder,level); + if(obj->lchild) + _twalk((Tree_t*)obj->lchild,action,level+1); + (*action)((Void_t*)obj,postorder,level); + if(obj->rchild) + _twalk((Tree_t*)obj->rchild,action,level+1); + (*action)((Void_t*)obj,endorder,level); + } +} + +/* the original twalk allows specifying arbitrary node to start traversal. +** Since our root is a dictionary structure, the search here will start +** at whichever node happens to be current root. +*/ +extern +#if __STD_C +void twalk(const Void_t* root, void(*action)(const Void_t*,VISIT,int) ) +#else +void twalk(root, action) +Void_t* root; +void(* action)(); +#endif +{ + reg Tree_t* o; + + if(root && (o = (Tree_t*)dtfinger((Dt_t*)root)) ) + _twalk(o,action,0); +} + +#endif |