src/map.c

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright 2023 Mike Becker, Olaf Wintermann All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *   1. Redistributions of source code must retain the above copyright
 *      notice, this list of conditions and the following disclaimer.
 *
 *   2. 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "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 COPYRIGHT HOLDER OR CONTRIBUTORS 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.
 */

#include "cx/map.h"
#include <string.h>

#include "cx/list.h"

// <editor-fold desc="empty map implementation">

static void cx_empty_map_noop(cx_attr_unused CxMap *map) {
    // this is a noop, but MUST be implemented
}

static void *cx_empty_map_get(
        cx_attr_unused const CxMap *map,
        cx_attr_unused CxHashKey key
) {
    return NULL;
}

static bool cx_empty_map_iter_valid(cx_attr_unused const void *iter) {
    return false;
}

static CxMapIterator cx_empty_map_iterator(
        const struct cx_map_s *map,
        cx_attr_unused enum cx_map_iterator_type type
) {
    CxMapIterator iter = {0};
    iter.map = (CxMap*) map;
    iter.base.valid = cx_empty_map_iter_valid;
    return iter;
}

static struct cx_map_class_s cx_empty_map_class = {
        cx_empty_map_noop,
        cx_empty_map_noop,
        NULL,
        cx_empty_map_get,
        NULL,
        cx_empty_map_iterator
};

CxMap cx_empty_map = {
    {
        NULL,
        NULL,
        0,
        0,
        NULL,
        NULL,
        NULL,
        false,
        true
    },
    &cx_empty_map_class
};

CxMap *const cxEmptyMap = &cx_empty_map;

// </editor-fold>

void cxMapClear(CxMap *map) {
    map->cl->clear(map);
}

size_t cxMapSize(const CxMap *map) {
    return map->collection.size;
}

CxMapIterator cxMapIteratorValues(const CxMap *map) {
    if (map == NULL) map = cxEmptyMap;
    return map->cl->iterator(map, CX_MAP_ITERATOR_VALUES);
}

CxMapIterator cxMapIteratorKeys(const CxMap *map) {
    if (map == NULL) map = cxEmptyMap;
    return map->cl->iterator(map, CX_MAP_ITERATOR_KEYS);
}

CxMapIterator cxMapIterator(const CxMap *map) {
    if (map == NULL) map = cxEmptyMap;
    return map->cl->iterator(map, CX_MAP_ITERATOR_PAIRS);
}

int cx_map_put(CxMap *map, CxHashKey key, void *value) {
    return map->cl->put(map, key, value) == NULL;
}

void *cx_map_emplace(CxMap *map, CxHashKey key) {
    return map->cl->put(map, key, NULL);
}

void *cx_map_get(const CxMap *map, CxHashKey key) {
    return map->cl->get(map, key);
}

int cx_map_remove(CxMap *map, CxHashKey key, void *targetbuf) {
    return map->cl->remove(map, key, targetbuf);
}

void cxMapFree(CxMap *map) {
    if (map == NULL) return;
    map->cl->deallocate(map);
}

static void cx_map_remove_uninitialized_entry(CxMap *map, CxHashKey key) {
    cx_destructor_func destr_bak = map->collection.simple_destructor;
    cx_destructor_func2 destr2_bak = map->collection.advanced_destructor;
    map->collection.simple_destructor = NULL;
    map->collection.advanced_destructor = NULL;
    cxMapRemove(map, key);
    map->collection.simple_destructor = destr_bak;
    map->collection.advanced_destructor = destr2_bak;
}

int cxMapClone(CxMap *dst, const CxMap *src, cx_clone_func clone_func,
        const CxAllocator *clone_allocator, void *data) {
    if (clone_allocator == NULL) clone_allocator = cxDefaultAllocator;
    CxMapIterator src_iter = cxMapIterator(src);
    if (cxCollectionStoresPointers(dst)) {
        for (size_t i = 0; i < cxMapSize(src); i++) {
            const CxMapEntry *entry = cxIteratorCurrent(src_iter);
            void **dst_mem = cxMapEmplace(dst, *(entry->key));
            if (dst_mem == NULL) {
                return 1;
            }
            void *dst_ptr = clone_func(NULL, entry->value, clone_allocator, data);
            if (dst_ptr == NULL) {
                cx_map_remove_uninitialized_entry(dst, *(entry->key));
                return 1;
            }
            *dst_mem = dst_ptr;
            cxIteratorNext(src_iter);
        }
    } else {
        for (size_t i = 0; i < cxMapSize(src); i++) {
            const CxMapEntry *entry = cxIteratorCurrent(src_iter);
            void *dst_mem = cxMapEmplace(dst, *(entry->key));
            if (dst_mem == NULL) {
                return 1;
            }
            if (clone_func(dst_mem, entry->value, clone_allocator, data) == NULL) {
                cx_map_remove_uninitialized_entry(dst, *(entry->key));
                return 1;
            }
            cxIteratorNext(src_iter);
        }
    }
    return 0;
}

int cxMapDifference(CxMap *dst, const CxMap *minuend, const CxMap *subtrahend,
        cx_clone_func clone_func, const CxAllocator *clone_allocator, void *data) {
    if (clone_allocator == NULL) clone_allocator = cxDefaultAllocator;

    const bool map_was_not_empty = cxMapSize(dst) > 0;
    CxMapIterator src_iter = cxMapIterator(minuend);
    if (cxCollectionStoresPointers(dst)) {
        cx_foreach(const CxMapEntry *, entry, src_iter) {
            if (cxMapContains(subtrahend, *entry->key)) {
                if (map_was_not_empty) {
                    cxMapRemove(dst, *entry->key);
                }
            } else {
                void** dst_mem = cxMapEmplace(dst, *entry->key);
                if (dst_mem == NULL) {
                    return 1;
                }

                void* dst_ptr = clone_func(NULL, entry->value, clone_allocator, data);
                if (dst_ptr == NULL) {
                    cx_map_remove_uninitialized_entry(dst, *(entry->key));
                    return 1;
                }
                *dst_mem = dst_ptr;
            }
        }
    } else {
        cx_foreach(const CxMapEntry *, entry, src_iter) {
            if (cxMapContains(subtrahend, *entry->key)) {
                if (map_was_not_empty) {
                    cxMapRemove(dst, *entry->key);
                }
            } else {
                void* dst_mem = cxMapEmplace(dst, *entry->key);
                if (dst_mem == NULL) {
                    return 1;
                }
                if (NULL == clone_func(dst_mem, entry->value, clone_allocator, data)) {
                    cx_map_remove_uninitialized_entry(dst, *entry->key);
                    return 1;
                }
            }
        }
    }
    return 0;
}

int cxMapListDifference(CxMap *dst, const CxMap *src, const CxList *keys,
        cx_clone_func clone_func, const CxAllocator *clone_allocator, void *data) {
    const bool map_was_not_empty = cxMapSize(dst) > 0;
    CxMapIterator src_iter = cxMapIterator(src);
    if (cxCollectionStoresPointers(dst)) {
        cx_foreach(const CxMapEntry *, entry, src_iter) {
            if (cxListContains(keys, entry->key)) {
                if (map_was_not_empty) {
                    cxMapRemove(dst, *entry->key);
                }
            } else {
                void** dst_mem = cxMapEmplace(dst, *entry->key);
                if (dst_mem == NULL) {
                    return 1;
                }

                void* dst_ptr = clone_func(NULL, entry->value, clone_allocator, data);
                if (dst_ptr == NULL) {
                    cx_map_remove_uninitialized_entry(dst, *(entry->key));
                    return 1;
                }
                *dst_mem = dst_ptr;
            }
        }
    } else {
        cx_foreach(const CxMapEntry *, entry, src_iter) {
            if (cxListContains(keys, entry->key)) {
                if (map_was_not_empty) {
                    cxMapRemove(dst, *entry->key);
                }
            } else {
                void* dst_mem = cxMapEmplace(dst, *entry->key);
                if (dst_mem == NULL) {
                    return 1;
                }
                if (NULL == clone_func(dst_mem, entry->value, clone_allocator, data)) {
                    cx_map_remove_uninitialized_entry(dst, *entry->key);
                    return 1;
                }
            }
        }
    }
    return 0;
}