Commit c617c134 authored by Brice Videau's avatar Brice Videau

Merge branch 'tleaf' into 'master'

Tleaf

See merge request !6
parents e7610669 b9a6efec
......@@ -2,6 +2,6 @@ AM_CFLAGS = -Wall -Werror -pedantic
lib_LTLIBRARIES = libexcit.la
libexcit_la_SOURCES = excit.c slice.c prod.c cons.c repeat.c hilbert2d.c range.c tleaf.c
libexcit_la_SOURCES = excit.c composition.c prod.c cons.c repeat.c hilbert2d.c range.c tleaf.c
include_HEADERS = excit.h
#include "dev/excit.h"
#include "slice.h"
#include "composition.h"
static int slice_it_alloc(excit_t data)
static int composition_it_alloc(excit_t data)
{
struct slice_it_s *it = (struct slice_it_s *)data->data;
struct composition_it_s *it = (struct composition_it_s *)data->data;
it->src = NULL;
it->indexer = NULL;
return EXCIT_SUCCESS;
}
static void slice_it_free(excit_t data)
static void composition_it_free(excit_t data)
{
struct slice_it_s *it = (struct slice_it_s *)data->data;
struct composition_it_s *it = (struct composition_it_s *)data->data;
excit_free(it->src);
excit_free(it->indexer);
}
static int slice_it_copy(excit_t dst, const excit_t src)
static int composition_it_copy(excit_t dst, const excit_t src)
{
const struct slice_it_s *it = (const struct slice_it_s *)src->data;
struct slice_it_s *result = (struct slice_it_s *)dst->data;
const struct composition_it_s *it = (const struct composition_it_s *)src->data;
struct composition_it_s *result = (struct composition_it_s *)dst->data;
result->src = excit_dup(it->src);
if (!result->src)
......@@ -34,9 +34,9 @@ static int slice_it_copy(excit_t dst, const excit_t src)
return EXCIT_SUCCESS;
}
static int slice_it_next(excit_t data, ssize_t *indexes)
static int composition_it_next(excit_t data, ssize_t *indexes)
{
struct slice_it_s *it = (struct slice_it_s *)data->data;
struct composition_it_s *it = (struct composition_it_s *)data->data;
ssize_t n;
int err = excit_next(it->indexer, &n);
......@@ -45,9 +45,9 @@ static int slice_it_next(excit_t data, ssize_t *indexes)
return excit_nth(it->src, n, indexes);
}
static int slice_it_peek(const excit_t data, ssize_t *indexes)
static int composition_it_peek(const excit_t data, ssize_t *indexes)
{
const struct slice_it_s *it = (const struct slice_it_s *)data->data;
const struct composition_it_s *it = (const struct composition_it_s *)data->data;
ssize_t n;
int err = excit_peek(it->indexer, &n);
......@@ -56,23 +56,23 @@ static int slice_it_peek(const excit_t data, ssize_t *indexes)
return excit_nth(it->src, n, indexes);
}
static int slice_it_size(const excit_t data, ssize_t *size)
static int composition_it_size(const excit_t data, ssize_t *size)
{
const struct slice_it_s *it = (const struct slice_it_s *)data->data;
const struct composition_it_s *it = (const struct composition_it_s *)data->data;
return excit_size(it->indexer, size);
}
static int slice_it_rewind(excit_t data)
static int composition_it_rewind(excit_t data)
{
struct slice_it_s *it = (struct slice_it_s *)data->data;
struct composition_it_s *it = (struct composition_it_s *)data->data;
return excit_rewind(it->indexer);
}
static int slice_it_nth(const excit_t data, ssize_t n, ssize_t *indexes)
static int composition_it_nth(const excit_t data, ssize_t n, ssize_t *indexes)
{
const struct slice_it_s *it = (const struct slice_it_s *)data->data;
const struct composition_it_s *it = (const struct composition_it_s *)data->data;
ssize_t p;
int err = excit_nth(it->indexer, n, &p);
......@@ -81,10 +81,10 @@ static int slice_it_nth(const excit_t data, ssize_t n, ssize_t *indexes)
return excit_nth(it->src, p, indexes);
}
static int slice_it_rank(const excit_t data, const ssize_t *indexes,
static int composition_it_rank(const excit_t data, const ssize_t *indexes,
ssize_t *n)
{
const struct slice_it_s *it = (const struct slice_it_s *)data->data;
const struct composition_it_s *it = (const struct composition_it_s *)data->data;
ssize_t inner_n;
int err = excit_rank(it->src, indexes, &inner_n);
......@@ -93,16 +93,16 @@ static int slice_it_rank(const excit_t data, const ssize_t *indexes,
return excit_rank(it->indexer, &inner_n, n);
}
static int slice_it_pos(const excit_t data, ssize_t *n)
static int composition_it_pos(const excit_t data, ssize_t *n)
{
const struct slice_it_s *it = (const struct slice_it_s *)data->data;
const struct composition_it_s *it = (const struct composition_it_s *)data->data;
return excit_pos(it->indexer, n);
}
static int slice_it_split(const excit_t data, ssize_t n, excit_t *results)
static int composition_it_split(const excit_t data, ssize_t n, excit_t *results)
{
const struct slice_it_s *it = (const struct slice_it_s *)data->data;
const struct composition_it_s *it = (const struct composition_it_s *)data->data;
int err = excit_split(it->indexer, n, results);
if (err)
......@@ -114,7 +114,7 @@ static int slice_it_split(const excit_t data, ssize_t n, excit_t *results)
excit_t tmp2;
tmp = results[i];
results[i] = excit_alloc(EXCIT_SLICE);
results[i] = excit_alloc(EXCIT_COMPOSITION);
if (!results[i]) {
excit_free(tmp);
err = -EXCIT_ENOMEM;
......@@ -126,7 +126,7 @@ static int slice_it_split(const excit_t data, ssize_t n, excit_t *results)
err = -EXCIT_ENOMEM;
goto error;
}
err = excit_slice_init(results[i], tmp2, tmp);
err = excit_composition_init(results[i], tmp2, tmp);
if (err) {
excit_free(tmp);
excit_free(tmp2);
......@@ -140,26 +140,26 @@ error:
return err;
}
struct excit_func_table_s excit_slice_func_table = {
slice_it_alloc,
slice_it_free,
slice_it_copy,
slice_it_next,
slice_it_peek,
slice_it_size,
slice_it_rewind,
slice_it_split,
slice_it_nth,
slice_it_rank,
slice_it_pos
struct excit_func_table_s excit_composition_func_table = {
composition_it_alloc,
composition_it_free,
composition_it_copy,
composition_it_next,
composition_it_peek,
composition_it_size,
composition_it_rewind,
composition_it_split,
composition_it_nth,
composition_it_rank,
composition_it_pos
};
int excit_slice_init(excit_t it, excit_t src, excit_t indexer)
int excit_composition_init(excit_t it, excit_t src, excit_t indexer)
{
if (!it || it->type != EXCIT_SLICE || !src || !indexer
if (!it || it->type != EXCIT_COMPOSITION || !src || !indexer
|| indexer->dimension != 1)
return -EXCIT_EINVAL;
struct slice_it_s *slice_it = (struct slice_it_s *)it->data;
struct composition_it_s *composition_it = (struct composition_it_s *)it->data;
ssize_t size_src;
ssize_t size_indexer;
int err = excit_size(src, &size_src);
......@@ -171,8 +171,8 @@ int excit_slice_init(excit_t it, excit_t src, excit_t indexer)
return err;
if (size_indexer > size_src)
return -EXCIT_EDOM;
slice_it->src = src;
slice_it->indexer = indexer;
composition_it->src = src;
composition_it->indexer = indexer;
it->dimension = src->dimension;
return EXCIT_SUCCESS;
}
......
......@@ -4,12 +4,12 @@
#include "excit.h"
#include "dev/excit.h"
struct slice_it_s {
struct composition_it_s {
excit_t src;
excit_t indexer;
};
extern struct excit_func_table_s excit_slice_func_table;
extern struct excit_func_table_s excit_composition_func_table;
#endif //EXCIT_SLICE_H
#include <stdlib.h>
#include "excit.h"
#include "dev/excit.h"
#include "slice.h"
#include "composition.h"
#include "prod.h"
#include "cons.h"
#include "repeat.h"
......@@ -19,7 +19,7 @@ const char *excit_type_name(enum excit_type_e type)
CASE(EXCIT_REPEAT);
CASE(EXCIT_HILBERT2D);
CASE(EXCIT_PRODUCT);
CASE(EXCIT_SLICE);
CASE(EXCIT_COMPOSITION);
CASE(EXCIT_USER);
CASE(EXCIT_TYPE_MAX);
default:
......@@ -115,8 +115,8 @@ excit_t excit_alloc(enum excit_type_e type)
case EXCIT_PRODUCT:
ALLOC_EXCIT(prod);
break;
case EXCIT_SLICE:
ALLOC_EXCIT(slice);
case EXCIT_COMPOSITION:
ALLOC_EXCIT(composition);
break;
case EXCIT_TLEAF:
ALLOC_EXCIT(tleaf);
......@@ -270,12 +270,12 @@ int excit_split(const excit_t it, ssize_t n, excit_t *results)
if (!tmp)
goto error2;
tmp2 = results[i];
results[i] = excit_alloc(EXCIT_SLICE);
results[i] = excit_alloc(EXCIT_COMPOSITION);
if (!results[i]) {
excit_free(tmp2);
goto error2;
}
err = excit_slice_init(results[i], tmp, tmp2);
err = excit_composition_init(results[i], tmp, tmp2);
if (err) {
excit_free(tmp2);
goto error2;
......
......@@ -4,20 +4,87 @@
#include <stdlib.h>
/*
* The different types of iterator supported. All iterators use the same
* integer type (ssize_t) for values.
* excit library provides an interface to build multi-dimensional iterators over
* indexes.
* excit iterators walk an array of n elements indexed from 0 to n-1,
* and returns the aforementionned elements.
* excit return elements which type can only be an index of type ssize_t,
* or an array of indexes (a multi-dimensional index) if the defined iterator
* has several dimensions. ssize_t elements can fit pointers. Thus, it makes
* excit library the ideal tool box for indexing and ealking complex structures.
*
* excit implements its own interface with several iterators (see excit_type_e).
* For instance, excit implementation of product iterators enable to mix iterators
* to create more complex ones. The library balanced tree "tleaf" iterator is built
* on top of product iterator.
*
* excit library uses the concept of "ownership".
* An excit iterator has the ownership of its internal data, i.e it will free
* its owned data upon call to excit_free(). This ownership
* may be transferred to another iterator through a library function such as
* excit_cons_init() or excit_product_add().
* Thus ownership must be carefully watched to avoid memory leaks or double free.
*
* excit library provides a rank function to find the index given an element.
*/
enum excit_type_e {
EXCIT_INVALID, /*!< Tag for invalid iterators */
EXCIT_RANGE, /*!< Iterator over a range of values */
EXCIT_CONS, /*!< Sliding window iterator */
EXCIT_REPEAT, /*!< Ierator that stutters a certain amount of times */
EXCIT_HILBERT2D, /*!< Hilbert space filing curve */
EXCIT_PRODUCT, /*!< Iterator over the catesian product of iterators */
EXCIT_SLICE, /*!< Iterator using another iterator to index a third */
EXCIT_TLEAF, /*!< Iterator on tree leaves with all leaves at same depth */
EXCIT_USER, /*!< User-defined iterator */
EXCIT_TYPE_MAX /*!< Guard */
/*!< Tag for invalid iterators */
EXCIT_INVALID,
/*!<
* Iterator over a range of values.
* See function excit_range_init for further details on iterator
* behaviour.
*/
EXCIT_RANGE,
/*!<
* Sliding window iterator
* See function excit_cons_init for further details on iterator
* behaviour.
*/
EXCIT_CONS,
/*!<
* Iterator that stutters a certain amount of times.
* Builds an iterator on top of another iterator repeating the latter elements.
* See function excit_repeat_init() for further details on iterator
* behaviour.
*/
EXCIT_REPEAT,
/*!< Hilbert space filing curve */
EXCIT_HILBERT2D,
/*!<
* Iterator over the catesian product of iterators.
* The result iterator dimension is the sum of of input iterator dimensions.
*/
EXCIT_PRODUCT,
/*!<
* Iterator composing two iterators,
* i.e using an iterator to index another.
* It is possible to chain composition iterators as long as
* input and output sets are compatible.
* (Only dimension compatibility is not enforced by the library).
* It is straightforward to build a composition iterator by composing two range iterators.
*/
EXCIT_COMPOSITION,
/*!<
* Iterator on balanced tree leaves.
* The iterator walks an index of leaves according to a policy.
* tleaf iterator has a special tleaf_it_split() for splitting the
* tree at a specific level.
* See tleaf_it_policy_e and excit_tleaf_init() for further explaination.
*/
EXCIT_TLEAF,
/*!<
* User-defined iterator
* excit library allow users to define their own iterator.
* To do so, they need to populate the function table excit_func_table_s
* with the routines to manipulate the aforementionned iterator.
* The outcome is that they will enjoy the functionnalities of the library
* for mixing with other iterators.
*/
EXCIT_USER,
/*!< Guard */
EXCIT_TYPE_MAX
};
/*
......@@ -96,6 +163,8 @@ struct excit_func_table_s {
/*
* This funciton is called during excit_dup. It is responsible for
* duplicating the content of the inner data between src_it and dst_it.
* The internal state of the iterator must also be copied, i.e subsequent
* calls to excit_next() must return the same results for both iterators.
* Returns EXCIT_SUCCESS or an error code.
*/
int (*copy)(excit_t dst_it, const excit_t src_it);
......@@ -185,7 +254,7 @@ excit_t excit_alloc_user(struct excit_func_table_s *func_table,
size_t data_size);
/*
* Duplicates an iterator.
* Duplicates an iterator and keep its internal state.
* "it": iterator to duplicate.
* Returns an iterator that will need to be freed unless ownership is
* transfered or NULL if an error occured.
......@@ -262,7 +331,8 @@ int excit_size(const excit_t it, ssize_t *size);
int excit_split(const excit_t it, ssize_t n, excit_t *results);
/*
* Gets the nth element of an iterator.
* Gets the nth element of an iterator. If an iterator has k dimensions,
* then the nth element is an array of k nth elements along each dimension.
* "it": an iterator.
* "rank": rank of the element, comprised between 0 and the size of the
* iterator.
......@@ -273,14 +343,17 @@ int excit_split(const excit_t it, ssize_t n, excit_t *results);
int excit_nth(const excit_t it, ssize_t rank, ssize_t *indexes);
/*
* Gets the rank of an element of an iterator.
* Gets the rank of an element of an iterator. The rank of an element is its
* iteration index, i.e excit_nth(excit_rank(element)) should return element.
* If the iterator has k dimensions, element is an array of the k values
* composing element.
* "it": an iterator.
* "indexes": an array of indexes corresponding to the element of the iterator.
* "rank": a pointer to a variable where the result will be stored, no result is
* returned if NULL.
* Returns EXCIT_SUCCESS or an error code.
*/
int excit_rank(const excit_t it, const ssize_t *indexes, ssize_t *rank);
int excit_rank(const excit_t it, const ssize_t *element, ssize_t *rank);
/*
* Gets the position of the iterator.
......@@ -311,7 +384,7 @@ int excit_skip(excit_t it);
int excit_cyclic_next(excit_t it, ssize_t *indexes, int *looped);
/*
* Initializes a range iterator to iterate from first to last (included) by sep.
* Initializes a range iterator to iterate from first to last (included) by step.
* "it": a range iterator.
* "first": first value of the range.
* "last": last value of the range.
......@@ -398,13 +471,13 @@ int excit_product_split_dim(const excit_t it, ssize_t dim, ssize_t n,
excit_t *results);
/*
* Initializes a slice iterator by giving asrc iterator and an indexer iterator.
* "it": a slice iterator.
* Initializes a composition iterator by giving a src iterator and an indexer iterator.
* "it": a composition iterator.
* "src": the iterator whom elements are to be returned.
* "indexer": the iterator that will provide the rank of the elements to return.
* Returns EXCIT_SUCCESS or an error code.
*/
int excit_slice_init(excit_t it, excit_t src, excit_t indexer);
int excit_composition_init(excit_t it, excit_t src, excit_t indexer);
enum tleaf_it_policy_e {
TLEAF_POLICY_ROUND_ROBIN, /* Iterate on tree leaves in a round-robin fashion */
......@@ -414,28 +487,41 @@ enum tleaf_it_policy_e {
/*
* Initialize a tleaf iterator by giving its depth, levels arity and iteration policy.
* example building a user scatter policy:
* excit_tleaf_init(it, 4, {3, 2, 4}, TLEAF_POLICY_USER, {2, 1, 0});
* gives the output index:
* 0,6,12,18,3,9,15,21,1,7,13,19,4,10,16,22,2,8,14,20,5,11,17,23.
* "it": a tleaf iterator
* "depth": the total number of levels of the tree, including leaves
* "arity": An array of size (depth-1). For each level, the number of children attached to a node. Leaves have no children, hence last level arity is ignored. Arities are organized from root to leaves.
* "arity": An array of size (depth-1). For each level, the number of children attached to a node.
* Leaves have no children. Arities are organized from root to leaves.
* "index": NULL or an array of depth excit_t to re-index levels.
* It is intended to prune node of certain levels while keeping index of the initial structure.
* Ownership of index is not taken. The iterator allocates a copy of index and manage it internally.
* "policy": A policy for iteration on leaves.
* "user_policy": If policy is TLEAF_POLICY_USER, then this argument must be an array of size (depth-1) providing the order (from 0 to (depth-2)) in wich levels are walked.
* when resolving indexes. Underneath, a product iterator of range iterator returns indexes on last levels upon iterator queries. This set of indexes is then
* computed to a single leaf index. For instance TLEAF_POLICY_ROUND_ROBIN is obtained from walking from leaves to root whereas TLEAF_POLICY_SCATTER is
* "user_policy": If policy is TLEAF_POLICY_USER, then this argument must be an array of size (depth-1) providing the
* order (from 0 to (depth-2)) in which levels are walked.
* when resolving indexes. Underneath, a product iterator of range iterator returns indexes on last
* levels upon iterator queries. This set of indexes is then
* computed to a single leaf index. For instance TLEAF_POLICY_ROUND_ROBIN is obtained from walking
* from leaves to root whereas TLEAF_POLICY_SCATTER is
* obtained from walking from root to leaves.
*/
int excit_tleaf_init(excit_t it,
const ssize_t depth,
const ssize_t *arities,
const excit_t *index,
const enum tleaf_it_policy_e policy,
const ssize_t *user_policy);
/*
* Split a tree at a given level. The behaviour is different from the generic function excit_split for the split might be sparse.
* Split a tree at a given level. The behaviour is different from the generic function excit_split for the
* split might be sparse, depending on the tree level where the split occures and the number of parts.
* "it": a tleaf iterator.
* "level": The level to split.
* "n": The number of slices. n must divide the target level arity.
* "n": The number of parts. n must divide the target level arity.
* "out": an array of n allocated tleaf iterators.
*/
int tleaf_it_split(const excit_t it, const ssize_t level, const ssize_t n, excit_t *out);
#endif
......@@ -3,6 +3,14 @@
#include "dev/excit.h"
#include "tleaf.h"
static int tleaf_init_with_it(excit_t it,
const ssize_t depth,
const ssize_t *arities,
const excit_t *indexes,
const enum tleaf_it_policy_e policy,
const ssize_t *user_policy,
excit_t levels, excit_t levels_inverse);
static int tleaf_it_alloc(excit_t it)
{
if (it == NULL || it->data == NULL)
......@@ -12,10 +20,11 @@ static int tleaf_it_alloc(excit_t it)
data_it->depth = 0;
data_it->arities = NULL;
data_it->order = NULL;
data_it->order_inverse = NULL;
data_it->buf = NULL;
data_it->order = NULL;
data_it->levels = NULL;
data_it->order_inverse = NULL;
data_it->levels_inverse = NULL;
return EXCIT_SUCCESS;
}
......@@ -24,140 +33,21 @@ static void tleaf_it_free(excit_t it)
struct tleaf_it_s *data_it = it->data;
free(data_it->arities);
free(data_it->order);
free(data_it->order_inverse);
free(data_it->buf);
free(data_it->order);
excit_free(data_it->levels);
}
static int excit_tleaf_init_with_it(excit_t it,
const ssize_t depth,
const ssize_t *arities,
const enum tleaf_it_policy_e policy,
const ssize_t *user_policy, excit_t levels)
{
if (it == NULL || it->data == NULL)
return -EXCIT_EINVAL;
int err = EXCIT_SUCCESS;
struct tleaf_it_s *data_it = it->data;
ssize_t i;
data_it->depth = depth - 1;
/* Set order according to policy */
data_it->order = malloc(sizeof(*data_it->order) * data_it->depth);
if (data_it->order == NULL) {
err = -EXCIT_ENOMEM;
goto error;
}
switch (policy) {
case TLEAF_POLICY_ROUND_ROBIN:
for (i = 0; i < data_it->depth; i++)
data_it->order[i] = data_it->depth - i - 1;
break;
case TLEAF_POLICY_SCATTER:
for (i = 0; i < data_it->depth; i++)
data_it->order[i] = i;
break;
case TLEAF_POLICY_USER:
for (i = 0; i < data_it->depth; i++)
data_it->order[i] = user_policy[i];
break;
default:
err = -EXCIT_EINVAL;
goto error_with_levels;
}
/* Set order inverse */
data_it->order_inverse =
malloc(sizeof(*data_it->order_inverse) * data_it->depth);
if (data_it->order_inverse == NULL) {
err = -EXCIT_ENOMEM;
goto error_with_order;
}
for (i = 0; i < data_it->depth; i++)
data_it->order_inverse[data_it->order[i]] = i;
/* Set levels arity */
data_it->arities = malloc(sizeof(*data_it->arities) * data_it->depth);
if (data_it->arities == NULL) {
err = -EXCIT_ENOMEM;
goto error_with_order_inverse;
}
for (i = 0; i < data_it->depth; i++)
data_it->arities[i] = arities[i];
/* Set storage buffer for output of product iterator */
data_it->buf = malloc(sizeof(*data_it->buf) * data_it->depth);
if (data_it->buf == NULL) {
err = -EXCIT_ENOMEM;
goto error_with_arity;
}
/* Set product iterator if not provided */
if (levels == NULL) {
data_it->levels = excit_alloc(EXCIT_PRODUCT);
if (data_it->levels == NULL) {
err = -EXCIT_ENOMEM;
goto error_with_buf;
}
for (i = 0; i < data_it->depth; i++) {
excit_t level = excit_alloc(EXCIT_RANGE);
if (level == NULL) {
err = -EXCIT_ENOMEM;
goto error_with_levels;
}
err = excit_range_init(level, 0, arities[i] - 1, 1);
if (err != EXCIT_SUCCESS)
goto error_with_levels;
err = excit_product_add(data_it->levels, level);
if (err != EXCIT_SUCCESS)
goto error_with_levels;
}
} else {
data_it->levels = levels;
}
return EXCIT_SUCCESS;
error_with_levels:
excit_free(data_it->levels);
data_it->levels = NULL;
error_with_buf:
free(data_it->buf);
data_it->buf = NULL;
error_with_arity:
free(data_it->arities);
data_it->arities = NULL;
error_with_order_inverse:
free(data_it->order_inverse);
data_it->order_inverse = NULL;
error_with_order:
free(data_it->order);
data_it->order = NULL;
error:
return err;
}
int excit_tleaf_init(excit_t it,
const ssize_t depth,
const ssize_t *arities,
const enum tleaf_it_policy_e policy,
const ssize_t *user_policy)
{
return excit_tleaf_init_with_it(it, depth, arities, policy, user_policy,
NULL);
excit_free(data_it->levels_inverse);
}
static int tleaf_it_size(const excit_t it, ssize_t *size)
{
struct tleaf_it_s *data_it = it->data;
int err = excit_size(data_it->levels, size);
return excit_size(data_it->levels, size);
if (err != EXCIT_SUCCESS)
return err;
return EXCIT_SUCCESS;
}
static int tleaf_it_rewind(excit_t it)
......@@ -180,6 +70,7 @@ static int tleaf_it_copy(excit_t dst_it, const excit_t src_it)
free(dst->order);
free(dst->order_inverse);
excit_free(dst->levels);
excit_free(dst->levels_inverse);
}
/* dst is not initialized (anymore) */
......@@ -190,13 +81,23 @@ static int tleaf_it_copy(excit_t dst_it, const excit_t src_it)
goto error;
}
err = excit_tleaf_init_with_it(dst_it, src->depth + 1, src->arities,
TLEAF_POLICY_USER, src->order, levels);
if (err != EXCIT_SUCCESS)
excit_t levels_inverse = excit_dup(src->levels_inverse);
if (levels_inverse == NULL) {
err = -EXCIT_ENOMEM;
goto error_with_levels;
}
err = tleaf_init_with_it(dst_it, src->depth + 1, src->arities, NULL,
TLEAF_POLICY_USER, src->order, levels,
levels_inverse);
if (err != EXCIT_SUCCESS)
goto error_with_levels_inverse;
return EXCIT_SUCCESS;
error_with_levels_inverse:
excit_free(levels_inverse);
error_with_levels:
excit_free(levels);
error:
......@@ -210,14 +111,14 @@ static int tleaf_it_pos(const excit_t it, ssize_t *value)
return excit_pos(data_it->levels, value);
}
static ssize_t tleaf_it_value(struct tleaf_it_s *it, const int inverse)
static ssize_t tleaf_it_value(struct tleaf_it_s *it)
{
ssize_t i, acc = 1, val = 0;
ssize_t *order = inverse ? it->order_inverse : it->order;
for (i = 0; i < it->depth; i++) {
val += acc * it->buf[order[i]];
acc *= it->arities[order[i]];
/* levels are stacked following order, then decode result backward order_inverse */
val += acc * it->buf[it->order_inverse[it->depth - i - 1]];
acc *= it->arities[it->depth - i - 1];
}
return val;
}
......@@ -229,7 +130,8 @@ int tleaf_it_nth(const excit_t it, ssize_t n, ssize_t *indexes)
if (err != EXCIT_SUCCESS)
return err;
*indexes = tleaf_it_value(data_it, 0);
if (indexes != NULL)
*indexes = tleaf_it_value(data_it);
return EXCIT_SUCCESS;
}
......@@ -240,7 +142,8 @@ int tleaf_it_peek(const excit_t it, ssize_t *value)
if (err != EXCIT_SUCCESS)
return err;
*value = tleaf_it_value(data_it, 0);
if (value != NULL)
*value = tleaf_it_value(data_it);
return EXCIT_SUCCESS;
}
......@@ -251,81 +154,301 @@ int tleaf_it_next(excit_t it, ssize_t *indexes)
if (err != EXCIT_SUCCESS)
return err;
*indexes = tleaf_it_value(data_it, 0);
if (indexes != NULL)
*indexes = tleaf_it_value(data_it);
return EXCIT_SUCCESS;
}
int tleaf_it_rank(const excit_t it, const ssize_t *indexes, ssize_t *n)
{
ssize_t size;
int err;
err = tleaf_it_size(it, &size);
if (err != EXCIT_SUCCESS)
return err;
if (indexes == NULL || *indexes < 0 || *indexes >= size)