source: mainline/uspace/app/bithenge/tree.c@ 11b9ad7

/*
 * Copyright (c) 2012 Sean Bartell
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/** @addtogroup bithenge
 * @{
 */
/**
 * @file
 * Trees and nodes.
 */

#include <errno.h>
#include <stdlib.h>
#include "tree.h"

int bithenge_node_destroy(bithenge_node_t *node)
{
        bithenge_string_node_t *string_node;
        switch (bithenge_node_type(node)) {
        case BITHENGE_NODE_STRING:
                string_node = bithenge_as_string_node(node);
                if (string_node->needs_free)
                        free(string_node->value);
                break;
        case BITHENGE_NODE_INTERNAL:
                /* TODO */
                break;
        case BITHENGE_NODE_BOOLEAN:
                return EOK;
        case BITHENGE_NODE_INTEGER: /* pass-through */
        case BITHENGE_NODE_NONE:
                break;
        }
        free(node);
        return EOK;
}

typedef struct
{
        bithenge_internal_node_t base;
        bithenge_node_t **nodes;
        bithenge_int_t len;
        bool needs_free;
} simple_internal_node_t;

static simple_internal_node_t *internal_as_simple(bithenge_internal_node_t *node)
{
        return (simple_internal_node_t *)node;
}

static int simple_internal_node_for_each(bithenge_internal_node_t *base, bithenge_for_each_func_t func, void *data)
{
        int rc;
        simple_internal_node_t *node = internal_as_simple(base);
        for (bithenge_int_t i = 0; i < node->len; i++) {
                rc = func(node->nodes[2*i+0], node->nodes[2*i+1], data);
                if (rc != EOK)
                        return rc;
        }
        return EOK;
}

static bithenge_internal_node_ops_t simple_internal_node_ops = {
        .for_each = simple_internal_node_for_each
};

static bithenge_node_t *simple_internal_as_node(simple_internal_node_t *node)
{
        return bithenge_internal_as_node(&node->base);
}

int bithenge_new_simple_internal_node(bithenge_node_t **out, bithenge_node_t **nodes, bithenge_int_t len, bool needs_free)
{
        assert(out);
        simple_internal_node_t *node = malloc(sizeof(*node));
        if (!node)
                return ENOMEM;
        node->base.base.type = BITHENGE_NODE_INTERNAL;
        node->base.ops = &simple_internal_node_ops;
        node->nodes = nodes;
        node->len = len;
        node->needs_free = needs_free;
        *out = simple_internal_as_node(node);
        return EOK;
}

static bithenge_boolean_node_t false_node = { {BITHENGE_NODE_BOOLEAN}, false };
static bithenge_boolean_node_t true_node = { {BITHENGE_NODE_BOOLEAN}, true };

int bithenge_new_boolean_node(bithenge_node_t **out, bool value)
{
        assert(out);
        *out = bithenge_boolean_as_node(value ? &true_node : &false_node);
        return EOK;
}

int bithenge_new_integer_node(bithenge_node_t **out, bithenge_int_t value)
{
        assert(out);
        bithenge_integer_node_t *node = malloc(sizeof(*node));
        if (!node)
                return ENOMEM;
        node->base.type = BITHENGE_NODE_INTEGER;
        node->value = value;
        *out = bithenge_integer_as_node(node);
        return EOK;
}

int bithenge_new_string_node(bithenge_node_t **out, const char *value, bool needs_free)
{
        assert(out);
        bithenge_string_node_t *node = malloc(sizeof(*node));
        if (!node)
                return ENOMEM;
        node->base.type = BITHENGE_NODE_STRING;
        node->value = value;
        node->needs_free = needs_free;
        *out = bithenge_string_as_node(node);
        return EOK;
}