rte_member_sketch.h

/* SPDX-License-Identifier: BSD-3-Clause
 * Copyright(c) 2017 Intel Corporation
 */
 
#ifndef RTE_MEMBER_SKETCH_H
#define RTE_MEMBER_SKETCH_H
 
#ifdef __cplusplus
extern "C" {
#endif
 
#include <rte_vect.h>
#include <rte_ring_elem.h>
 
#define NUM_ROW_SCALAR 5
#define INTERVAL (1 << 15)
 
#if !RTE_IS_POWER_OF_2(INTERVAL)
#error sketch INTERVAL macro must be a power of 2
#endif
 
int
rte_member_create_sketch(struct rte_member_setsum *ss,
             const struct rte_member_parameters *params,
             struct rte_ring *r);
 
int
rte_member_lookup_sketch(const struct rte_member_setsum *setsum,
             const void *key, member_set_t *set_id);
 
int
rte_member_add_sketch(const struct rte_member_setsum *setsum,
              const void *key,
              member_set_t set_id);
 
int
rte_member_add_sketch_byte_count(const struct rte_member_setsum *ss,
                 const void *key, uint32_t byte_count);
 
void
sketch_update_scalar(const struct rte_member_setsum *ss,
             const void *key,
             uint32_t count);
 
uint64_t
sketch_lookup_scalar(const struct rte_member_setsum *ss,
             const void *key);
 
void
sketch_delete_scalar(const struct rte_member_setsum *ss,
             const void *key);
 
int
rte_member_delete_sketch(const struct rte_member_setsum *setsum,
             const void *key);
 
int
rte_member_query_sketch(const struct rte_member_setsum *setsum,
            const void *key, uint64_t *output);
 
void
rte_member_free_sketch(struct rte_member_setsum *ss);
 
void
rte_member_reset_sketch(const struct rte_member_setsum *setsum);
 
int
rte_member_report_heavyhitter_sketch(const struct rte_member_setsum *setsum,
                     void **key, uint64_t *count);
 
void
rte_member_update_heap(const struct rte_member_setsum *ss);
 
static __rte_always_inline uint64_t
count_min(const struct rte_member_setsum *ss, const uint32_t *hash_results)
{
    uint64_t *count_array = ss->table;
    uint64_t count;
    uint32_t cur_row;
    uint64_t min = UINT64_MAX;
 
    for (cur_row = 0; cur_row < ss->num_row; cur_row++) {
        uint64_t cnt = count_array[cur_row * ss->num_col + hash_results[cur_row]];
 
        if (cnt < min)
            min = cnt;
    }
    count = min;
 
    return count;
}
 
#ifdef __cplusplus
}
#endif
 
#endif /* RTE_MEMBER_SKETCH_H */