query.h

Go to the documentation of this file.

#ifndef GPUGenie_query_h
#define GPUGenie_query_h

#include <genie/table/inv_list.h>
#include <genie/table/inv_table.h>
#include <genie/configure.h>

#include <vector>

namespace genie
{
namespace query
{

typedef unsigned int u32;
typedef unsigned long long u64;


class Query
{
public:
    struct range
    {
        int query;
        int order;
        int dim;
        int low;
        int up;
        float weight;
        int low_offset;
        int up_offset;
        float selectivity;
    };
    struct dim
    {
        int query;
        int order;
        int start_pos;
        int end_pos;
        float weight;
        dim() = default;
        dim(int query, int order, int start_pos, int end_pos, float weight);
    };
private:
    genie::table::inv_table* _ref_table;

    std::map<int, std::vector<range> > _attr_map;

    std::map<int, std::vector<dim> > _dim_map;

    int _topk;

    float _selectivity;

    int _index;

    int _count;

public:
    bool is_load_balanced;
    bool use_load_balance;

    std::vector<dim> _dims;

    Query(genie::table::inv_table* ref, int index);

    Query(genie::table::inv_table& ref, int index);

    Query();

    genie::table::inv_table*
    ref_table();

    void
    attr(int index, int low, int up, float weight, int order);
    void
    attr(int index, int value, float weight, float selectivity, int order);
    void
    attr(int index, int low, int up, float weight, float selectivity, int order);

    void
    clear_dim(int index);

    void
    selectivity(float s);

    float
    selectivity();

    void
    apply_adaptive_query_range();

    void
    build_and_apply_load_balance(int max_load);

    #ifdef GENIE_COMPR
    void
    build_compressed(int max_load);
    #endif

    void
    topk(int k);

    int
    topk();

    void
    build();

    void
    build(std::vector<dim> &);


    int
    dump(std::vector<dim>& vout);

    int
    dump(std::vector<range>& ranges);

    void
    print(int limit);

    int
    index();

    int
    count_ranges();

    void
    build_sequence();
};

} // namespace query
} // namespace genie

#endif