comparison: src/tree.c
src/tree.c
- changeset 1318
- 12fa1d37fe48
- parent 1314
- 25108c15e2d4
- child 1319
- aa1f580f8f59
equal
deleted
inserted
replaced
| 1317:eeb2fc3850e2 | 1318:12fa1d37fe48 |
|---|---|
| 324 if (iter->depth == 1) { | 324 if (iter->depth == 1) { |
| 325 // there is no parent - we have iterated the entire tree | 325 // there is no parent - we have iterated the entire tree |
| 326 // invalidate the iterator and free the node stack | 326 // invalidate the iterator and free the node stack |
| 327 iter->node = iter->node_next = NULL; | 327 iter->node = iter->node_next = NULL; |
| 328 iter->stack_capacity = iter->depth = 0; | 328 iter->stack_capacity = iter->depth = 0; |
| 329 free(iter->stack); | 329 cxFree(cxDefaultAllocator, iter->stack); |
| 330 iter->stack = NULL; | 330 iter->stack = NULL; |
| 331 } else { | 331 } else { |
| 332 // the parent node can be obtained from the top of stack | 332 // the parent node can be obtained from the top of stack |
| 333 // this way we can avoid the loc_parent in the iterator | 333 // this way we can avoid the loc_parent in the iterator |
| 334 iter->depth--; | 334 iter->depth--; |
| 384 | 384 |
| 385 // visit the root node | 385 // visit the root node |
| 386 iter.node = root; | 386 iter.node = root; |
| 387 if (root != NULL) { | 387 if (root != NULL) { |
| 388 iter.stack_capacity = 16; | 388 iter.stack_capacity = 16; |
| 389 iter.stack = malloc(sizeof(void *) * 16); | 389 iter.stack = cxMalloc(cxDefaultAllocator, sizeof(void *) * 16); |
| 390 iter.stack[0] = root; | 390 iter.stack[0] = root; |
| 391 iter.counter = 1; | 391 iter.counter = 1; |
| 392 iter.depth = 1; | 392 iter.depth = 1; |
| 393 } else { | 393 } else { |
| 394 iter.stack_capacity = 0; | 394 iter.stack_capacity = 0; |
| 414 static void cx_tree_visitor_enqueue_siblings( | 414 static void cx_tree_visitor_enqueue_siblings( |
| 415 struct cx_tree_visitor_s *iter, void *node, ptrdiff_t loc_next) { | 415 struct cx_tree_visitor_s *iter, void *node, ptrdiff_t loc_next) { |
| 416 node = tree_next(node); | 416 node = tree_next(node); |
| 417 while (node != NULL) { | 417 while (node != NULL) { |
| 418 struct cx_tree_visitor_queue_s *q; | 418 struct cx_tree_visitor_queue_s *q; |
| 419 q = malloc(sizeof(struct cx_tree_visitor_queue_s)); | 419 q = cxMalloc(cxDefaultAllocator, sizeof(struct cx_tree_visitor_queue_s)); |
| 420 q->depth = iter->queue_last->depth; | 420 q->depth = iter->queue_last->depth; |
| 421 q->node = node; | 421 q->node = node; |
| 422 iter->queue_last->next = q; | 422 iter->queue_last->next = q; |
| 423 iter->queue_last = q; | 423 iter->queue_last = q; |
| 424 node = tree_next(node); | 424 node = tree_next(node); |
| 443 } else { | 443 } else { |
| 444 children = tree_children(iter->node); | 444 children = tree_children(iter->node); |
| 445 } | 445 } |
| 446 if (children != NULL) { | 446 if (children != NULL) { |
| 447 struct cx_tree_visitor_queue_s *q; | 447 struct cx_tree_visitor_queue_s *q; |
| 448 q = malloc(sizeof(struct cx_tree_visitor_queue_s)); | 448 q = cxMalloc(cxDefaultAllocator, sizeof(struct cx_tree_visitor_queue_s)); |
| 449 q->depth = iter->depth + 1; | 449 q->depth = iter->depth + 1; |
| 450 q->node = children; | 450 q->node = children; |
| 451 if (iter->queue_last == NULL) { | 451 if (iter->queue_last == NULL) { |
| 452 assert(iter->queue_next == NULL); | 452 assert(iter->queue_next == NULL); |
| 453 iter->queue_next = q; | 453 iter->queue_next = q; |
| 472 iter->queue_next = q->next; | 472 iter->queue_next = q->next; |
| 473 if (iter->queue_next == NULL) { | 473 if (iter->queue_next == NULL) { |
| 474 assert(iter->queue_last == q); | 474 assert(iter->queue_last == q); |
| 475 iter->queue_last = NULL; | 475 iter->queue_last = NULL; |
| 476 } | 476 } |
| 477 free(q); | 477 cxFree(cxDefaultAllocator, q); |
| 478 } | 478 } |
| 479 | 479 |
| 480 // increment the node counter | 480 // increment the node counter |
| 481 iter->counter++; | 481 iter->counter++; |
| 482 } | 482 } |
