libcdsb/src/set/modify.c

/* This software is licensed by the MIT License, see LICENSE file */
/*                                Copyright © 2022 Gregory Lirent */

#include "../../include/set.h"
#include "../__internal/rbtree.h"
#include "../__internal/assert.h"

bool libcdsb_vset_insert(set_t* x, vtype_variable value) {
    int     cmp;
    rbnode_t* n;
    rbnode_t* p;

    if (!rbnode_is_empty(n = x->root)) {
        do {
            p   = n;
            cmp = vtype_compare(value.pointer, value.type, vnode_peek(&n->value, x->type), x->type);

            if (cmp == 0) return false;

            n = (cmp < 0) ? n->left : n->right;
        } while (!rbnode_is_empty(n));

        n = rbnode_create(nullptr, p, 1);

        if (cmp < 0) p->left  = n;
        else         p->right = n;

        if (!rbnode_is_root(p))
            rbnode_fixup(&x->root, n);

    } else n = x->root = rbnode_create(nullptr, rbnode_empty, 0);

    n->value = vnode_tcreate(x->type, value.pointer, value.type);

    return true;
}


bool libcdsb_vset_attach(set_t* x, vtype_variable value) {
    int     cmp;
    rbnode_t* n;
    rbnode_t* p;
    vnode_t  vn;

    vnode_tattach(&vn, x->type, value.pointer, value.type);
    n             = x->root;
    value.type    = x->type;
    value.pointer = vnode_peek(&vn, value.type);

    if (!rbnode_is_empty(n)) {
        do {
            p   = n;
            cmp = vtype_compare(value.pointer, value.type, vnode_peek(&n->value, x->type), x->type);

            if (cmp == 0) {
                vnode_free(&vn, value.type);
                return false;
            }

            n = (cmp < 0) ? n->left : n->right;
        } while (!rbnode_is_empty(n));

        n = rbnode_create(vn, p, 1);

        if (cmp < 0) p->left  = n;
        else         p->right = n;

        if (!rbnode_is_root(p))
            rbnode_fixup(&x->root, n);

    } else n = x->root = rbnode_create(vn, rbnode_empty, 0);

    return true;
}