urbit/shrub
1
1
Fork 0
mirror of https://github.com/urbit/shrub.git synced 2024-12-21 18:01:32 +03:00
Code Issues Packages Projects Releases Wiki Activity

ur: adds ur_walk_fore - higher-ordered pre-order noun traversal

Browse Source
This commit is contained in:
Joe Bryan 2020-08-10 12:56:54 -07:00
parent d0e5c658a9
commit 039ef019b9
2 changed files with 58 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pkg/urbit/include/ur/hashcons.h
View File

@ -163,3 +163,10 @@ ur_nvec_free(ur_nvec_t *v);
void void
ur_nvec_init(ur_nvec_t *v, uint64_t size); ur_nvec_init(ur_nvec_t *v, uint64_t size);
void
ur_walk_fore(ur_root_t *r,
ur_nref ref,
void *v,
void (*atom)(ur_root_t*, ur_nref, void*),
ur_bool_t (*cell)(ur_root_t*, ur_nref, void*));

51
pkg/urbit/ur/hashcons.c
View File

@ -821,3 +821,54 @@ ur_nvec_init(ur_nvec_t *v, uint64_t size)
v->fill = 0; v->fill = 0;
v->refs = calloc(size, sizeof(ur_nref)); v->refs = calloc(size, sizeof(ur_nref));
} }
void
ur_walk_fore(ur_root_t *r,
ur_nref ref,
void *v,
void (*atom)(ur_root_t*, ur_nref, void*),
ur_bool_t (*cell)(ur_root_t*, ur_nref, void*))
{
uint64_t prev = 89, size = 144, fill = 0;
ur_nref *top, *don;
don = malloc(size * sizeof(*don));
top = don + ++fill;
*top = ref;
while ( top != don ) {
// visit atom, pop stack
//
if ( !ur_deep(ref) ) {
atom(r, ref, v);
top--; fill--;
}
// visit cell, pop stack if false
//
else if ( !cell(r, ref, v) ) {
top--; fill--;
}
// push the tail, continue into the head
//
else {
*top = ur_tail(r, ref);
// reallocate "stack" if full
//
if ( size == fill ) {
uint64_t next = prev + size;
don = realloc(don, next * sizeof(*don));
top = don + fill;
prev = size;
size = next;
}
top++; fill++;
*top = ur_head(r, ref);
}
ref = *top;
}
free(don);
}