1/* 2 * Copyright (C) 2013 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17#ifndef LATINIME_BIGRAM_LIST_POLICY_H 18#define LATINIME_BIGRAM_LIST_POLICY_H 19 20#include <cstdint> 21 22#include "defines.h" 23#include "suggest/core/policy/dictionary_bigrams_structure_policy.h" 24#include "suggest/policyimpl/dictionary/structure/pt_common/bigram/bigram_list_read_write_utils.h" 25 26namespace latinime { 27 28class BigramListPolicy : public DictionaryBigramsStructurePolicy { 29 public: 30 BigramListPolicy(const uint8_t *const bigramsBuf, const int bufSize) 31 : mBigramsBuf(bigramsBuf), mBufSize(bufSize) {} 32 33 ~BigramListPolicy() {} 34 35 void getNextBigram(int *const outBigramPos, int *const outProbability, bool *const outHasNext, 36 int *const pos) const { 37 BigramListReadWriteUtils::BigramFlags flags; 38 if (!BigramListReadWriteUtils::getBigramEntryPropertiesAndAdvancePosition(mBigramsBuf, 39 mBufSize, &flags, outBigramPos, pos)) { 40 AKLOGE("Cannot read bigram entry. mBufSize: %d, pos: %d. ", mBufSize, *pos); 41 *outProbability = NOT_A_PROBABILITY; 42 *outHasNext = false; 43 return; 44 } 45 *outProbability = BigramListReadWriteUtils::getProbabilityFromFlags(flags); 46 *outHasNext = BigramListReadWriteUtils::hasNext(flags); 47 } 48 49 bool skipAllBigrams(int *const pos) const { 50 return BigramListReadWriteUtils::skipExistingBigrams(mBigramsBuf, mBufSize, pos); 51 } 52 53 private: 54 DISALLOW_IMPLICIT_CONSTRUCTORS(BigramListPolicy); 55 56 const uint8_t *const mBigramsBuf; 57 const int mBufSize; 58}; 59} // namespace latinime 60#endif // LATINIME_BIGRAM_LIST_POLICY_H 61