support config in node_or

[protos/libecoli.git] / lib / ecoli_node_subset.c

diff --git a/lib/ecoli_node_subset.c b/lib/ecoli_node_subset.c
index 1afbde7..e3184ef 100644 (file)
--- a/lib/ecoli_node_subset.c
+++ b/lib/ecoli_node_subset.c
@@ -234,7 +234,10 @@ ec_node_subset_complete(const struct ec_node *gen_node,
 static void ec_node_subset_free_priv(struct ec_node *gen_node)
 {
        struct ec_node_subset *node = (struct ec_node_subset *)gen_node;
+       size_t i;
 
+       for (i = 0; i < node->len; i++)
+               ec_node_free(node->table[i]);
        ec_free(node->table);
 }
 
@@ -245,15 +248,18 @@ ec_node_subset_get_children_count(const struct ec_node *gen_node)
        return node->len;
 }
 
-static struct ec_node *
-ec_node_subset_get_child(const struct ec_node *gen_node, size_t i)
+static int
+ec_node_subset_get_child(const struct ec_node *gen_node, size_t i,
+                       struct ec_node **child, unsigned int *refs)
 {
        struct ec_node_subset *node = (struct ec_node_subset *)gen_node;
 
        if (i >= node->len)
-               return NULL;
+               return -1;
 
-       return node->table[i];
+       *child = node->table[i];
+       *refs = 1;
+       return 0;
 }
 
 static struct ec_node_type ec_node_subset_type = {
@@ -310,7 +316,7 @@ struct ec_node *__ec_node_subset(const char *id, ...)
 
        va_start(ap, id);
 
-       gen_node = __ec_node(&ec_node_subset_type, id);
+       gen_node = ec_node_from_type(&ec_node_subset_type, id);
        node = (struct ec_node_subset *)gen_node;
        if (node == NULL)
                fail = 1;;