/*------------------------------------------------------------------------------
 *
 * Copyright (c) 2011-2025, EURid vzw. All rights reserved.
 * The YADIFA TM software product is provided under the BSD 3-clause license:
 *
 * 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.
 *        * Neither the name of EURid nor the names of its contributors may be
 *          used to endorse or promote products derived from this software
 *          without specific prior written permission.
 *
 * 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.
 *
 *----------------------------------------------------------------------------*/

/**-----------------------------------------------------------------------------
 * @defgroup collections Generic collections functions
 * @ingroup dnscore
 * @brief A dictionary uint64_t => ptr based on the AVL code.
 *
 * A dictionary uint64_t => ptr based on the AVL code.
 * Mostly used for accessing hash => value
 *
 * @{
 *----------------------------------------------------------------------------*/

#ifndef _U64_TREEMAP_H
#define _U64_TREEMAP_H

#include <dnscore/sys_types.h>

#ifdef __cplusplus
extern "C"
{
#endif

/*
 * A digest is stored prefixed with its length ([1;255])
 */

/*
 * A structure to hold both children with direct access
 */

struct u64_treemap_node_s;

typedef struct u64_treemap_node_s u64_treemap_node_t;

struct u64_treemap_node_children_s
{
    u64_treemap_node_t *left;
    u64_treemap_node_t *right;
};

/*
 * A union to have access to the children with direct or indexed access
 */

union u64_treemap_children_u
{
    struct u64_treemap_node_children_s lr;
    u64_treemap_node_t                *child[2];
};

/*
 * The node structure CANNOT have a varying size on a given collection
 * This means that the digest size is a constant in the whole tree
 */

struct u64_treemap_node_s
{
    union u64_treemap_children_u children; /* 2 ptrs */
    union
    {
        void    *value; /* 1 ptr */
        intptr_t value_intptr;
        void (*void_function_void)();
        int64_t  value_s64;
        uint64_t value_u64;
        int32_t  value_s32;
        uint32_t value_u32;
    };
    uint64_t key;     /* 4 b */
    int8_t   balance; /* 1 b */
}; /* 29 OR 17 bytes (64/32) */

/*
 * AVL definition part begins here
 */

/*
 * The maximum depth of a tree.
 * 40 is enough for storing 433494436 items (worst case)
 *
 * Depth 0 is one node.
 *
 * Worst case : N is enough for sum[n = 0,N](Fn) where Fn is Fibonacci(n+1)
 * Best case : N is enough for (2^(N+1))-1
 */
#define AVL_DEPTH_MAX 52 // 139*10^9 items max (worst case)64

/*
 * The previx that will be put in front of each function name
 */
#define AVL_PREFIX    u64_treemap_

/*
 * The type that hold the node
 */
#define AVL_NODE_TYPE u64_treemap_node_t

/*
 * The type that hold the tree (should be AVL_NODE_TYPE*)
 */
#define AVL_TREE_TYPE AVL_NODE_TYPE *

typedef AVL_TREE_TYPE u64_treemap_t;

/*
 * The type that hold the tree (should be AVL_NODE_TYPE*)
 */
#define AVL_CONST_TREE_TYPE      AVL_NODE_TYPE *const

/*
 * How to find the root in the tree
 */
#define AVL_TREE_ROOT(__tree__)  (*(__tree__))

/*
 * Self explanatory
 */

/*
 * The type used for comparing the nodes.
 */

#define AVL_REFERENCE_TYPE       uint64_t
#define AVL_REFERENCE_IS_POINTER false
#define AVL_REFERENCE_IS_CONST   false

/*
 * The node has got a pointer to its parent
 *
 * 0   : disable
 * !=0 : enable
 */
#define AVL_HAS_PARENT_POINTER   0

#ifdef __cplusplus
}
#endif

#include <dnscore/avl.h.inc>

#ifdef __cplusplus
extern "C"
{
#endif

/*
 * I recommend setting a define to identify the C part of the template
 * So it can be used to undefine what is not required anymore for every
 * C file but that one.
 *
 */

#ifndef _U64_TREEMAP_C

#undef AVL_DEPTH_MAX
#undef AVL_PREFIX
#undef AVL_NODE_TYPE
#undef AVL_TREE_TYPE
#undef AVL_CONST_TREE_TYPE
#undef AVL_TREE_ROOT
#undef AVL_REFERENCE_TYPE
#undef AVL_HAS_PARENT_POINTER
#undef AVL_REFERENCE_IS_POINTER
#undef AVL_REFERENCE_IS_CONST

#undef _AVL_H_INC

#endif /* _U64_SET_C */

#define U64_TREEMAP_EMPTY NULL

void *u64_treemap_iterator_hasnext_next_value(u64_treemap_iterator_t *iterp);

#define FOREACH_U64_SET(cast__, var__, uint64_t_treemap__)                                                                                                                                                                                     \
    u64_treemap_iterator_t PREPROCESSOR_CONCAT_EVAL(foreach_u64_treemap_iter, __LINE__);                                                                                                                                                       \
    u64_treemap_iterator_init((u64_treemap__), &PREPROCESSOR_CONCAT_EVAL(foreach_u64_treemap_iter, __LINE__));                                                                                                                                 \
    for(cast__ var__; ((var__) = (cast__)u64_treemap_iterator_hasnext_next_value(&PREPROCESSOR_CONCAT_EVAL(foreach_u64_treemap_iter, __LINE__))) != NULL;)

#ifdef __cplusplus
}
#endif

#endif /* _U64_SET_H */
/** @} */
