X-Git-Url: http://git.tuebingen.mpg.de/?a=blobdiff_plain;f=buffer_tree.c;h=1046460d1d1c56dd424b0cc9a7a3e33c76857188;hb=0b101563e67c91bc660f6a16b5889901d9e8eeff;hp=b3aea0e4143cbed3ecaff1ca56055106b6108e5a;hpb=34fee5954159a9598110316ae79b64f0e408b423;p=paraslash.git diff --git a/buffer_tree.c b/buffer_tree.c index b3aea0e4..1046460d 100644 --- a/buffer_tree.c +++ b/buffer_tree.c @@ -44,8 +44,6 @@ struct btr_node { #define FOR_EACH_BUFFER_REF_SAFE(_br, _tmp, _btrn) \ list_for_each_entry_safe((_br), (_tmp), &(_btrn)->input_queue, node) -INIT_STDERR_LOGGING(0); - struct btr_node *btr_new_node(char *name, struct btr_node *parent) { struct btr_node *btrn = para_malloc(sizeof(*btrn)); @@ -143,14 +141,6 @@ bool btr_inplace_ok(struct btr_node *btrn) return list_is_singular(&btrn->parent->children); } -struct btr_buffer_reference *btr_next_br(struct btr_node *btrn) -{ - if (list_empty(&btrn->input_queue)) - return NULL; - return list_first_entry(&btrn->input_queue, struct btr_buffer_reference, node); -} - - static inline size_t br_available_bytes(struct btr_buffer_reference *br) { return br->btrb->size - br->consumed; @@ -162,9 +152,26 @@ size_t btr_get_buffer_by_reference(struct btr_buffer_reference *br, char **buf) return br_available_bytes(br); } -void btr_increase_used_bytes(struct btr_buffer_reference *br, size_t consumed) +size_t btr_next_buffer(struct btr_node *btrn, char **bufp) { - br->consumed += consumed; + struct btr_buffer_reference *br; + + if (list_empty(&btrn->input_queue)) { + *bufp = NULL; + return 0; + } + br = list_first_entry(&btrn->input_queue, struct btr_buffer_reference, node); + return btr_get_buffer_by_reference(br, bufp); +} + +void btr_consume(struct btr_node *btrn, size_t numbytes) +{ + struct btr_buffer_reference *br; + + assert(!list_empty(&btrn->input_queue)); + br = list_first_entry(&btrn->input_queue, struct btr_buffer_reference, node); + assert(br->consumed + numbytes <= br->btrb->size); + br->consumed += numbytes; if (br->consumed == br->btrb->size) btr_drop_buffer_reference(br); } @@ -180,6 +187,9 @@ void btr_del_node(struct btr_node *btrn) { struct btr_node *ch; + if (!btrn) + return; + PARA_NOTICE_LOG("deleting %s\n", btrn->name); FOR_EACH_CHILD(ch, btrn) ch->parent = NULL; flush_input_queue(btrn); @@ -199,7 +209,19 @@ size_t btr_get_input_queue_size(struct btr_node *btrn) return size; } -int main(void) +/** + * Return the size of the largest input queue. + * + * Iterates over all children of the given node. + */ +size_t btr_bytes_pending(struct btr_node *btrn) { - return 1; + size_t max_size = 0; + struct btr_node *ch; + + FOR_EACH_CHILD(ch, btrn) { + size_t size = btr_get_input_queue_size(ch); + max_size = PARA_MAX(max_size, size); + } + return max_size; }