aboutsummaryrefslogtreecommitdiffstats
path: root/searchlib/src/vespa/searchlib/diskindex/zcposocciterators.h
blob: 2eb2f12985ad0a42b7138f9513fd338e30ad3e7b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#pragma once

#include "zcpostingiterators.h"
#include <vespa/searchlib/bitcompression/posocccompression.h>

namespace search::diskindex {

struct Zc4PostingParams;

template <bool bigEndian, bool dynamic_k>
class ZcRareWordPosOccIterator : public ZcRareWordPostingIterator<bigEndian, dynamic_k>
{
private:
    using ParentClass = ZcRareWordPostingIterator<bigEndian, dynamic_k>;
    using ParentClass::_decodeContext;

    using DecodeContextReal = std::conditional_t<dynamic_k, bitcompression::EGPosOccDecodeContextCooked<bigEndian>, bitcompression::EG2PosOccDecodeContextCooked<bigEndian>>;
    DecodeContextReal _decodeContextReal;
public:
    ZcRareWordPosOccIterator(Position start, uint64_t bitLength, uint32_t docIdLimit,
                             bool decode_normal_features, bool decode_interleaved_features,
                             bool unpack_normal_features, bool unpack_interleaved_features,
                             const bitcompression::PosOccFieldsParams *fieldsParams,
                             fef::TermFieldMatchDataArray matchData);
};


template <bool bigEndian, bool dynamic_k>
class ZcPosOccIterator : public ZcPostingIterator<bigEndian>
{
private:
    using ParentClass = ZcPostingIterator<bigEndian>;
    using ParentClass::_decodeContext;

    using DecodeContext = std::conditional_t<dynamic_k, bitcompression::EGPosOccDecodeContextCooked<bigEndian>, bitcompression::EG2PosOccDecodeContextCooked<bigEndian>>;
    DecodeContext _decodeContextReal;
public:
    ZcPosOccIterator(Position start, uint64_t bitLength, uint32_t docIdLimit,
                     bool decode_normal_features, bool decode_interleaved_features,
                     bool unpack_normal_features, bool unpack_interleaved_features,
                     uint32_t minChunkDocs, const index::PostingListCounts &counts,
                     const bitcompression::PosOccFieldsParams *fieldsParams,
                     fef::TermFieldMatchDataArray matchData);
};

std::unique_ptr<search::queryeval::SearchIterator>
create_zc_posocc_iterator(bool bigEndian, const index::PostingListCounts &counts, bitcompression::Position start, uint64_t bit_length, const Zc4PostingParams &posting_params, const bitcompression::PosOccFieldsParams &fields_params, fef::TermFieldMatchDataArray match_data);

extern template class ZcRareWordPosOccIterator<false, false>;
extern template class ZcRareWordPosOccIterator<false, true>;
extern template class ZcRareWordPosOccIterator<true, false>;
extern template class ZcRareWordPosOccIterator<true, true>;

extern template class ZcPosOccIterator<false, false>;
extern template class ZcPosOccIterator<false, true>;
extern template class ZcPosOccIterator<true, false>;
extern template class ZcPosOccIterator<true, true>;

}