| Index: third_party/android_prediction/suggest/policyimpl/dictionary/structure/v2/bigram/bigram_list_policy.h
|
| diff --git a/third_party/android_prediction/suggest/policyimpl/dictionary/structure/v2/bigram/bigram_list_policy.h b/third_party/android_prediction/suggest/policyimpl/dictionary/structure/v2/bigram/bigram_list_policy.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..7e03322651c0e0a51b30b1d4c15095b94e214bbb
|
| --- /dev/null
|
| +++ b/third_party/android_prediction/suggest/policyimpl/dictionary/structure/v2/bigram/bigram_list_policy.h
|
| @@ -0,0 +1,60 @@
|
| +/*
|
| + * Copyright (C) 2013 The Android Open Source Project
|
| + *
|
| + * Licensed under the Apache License, Version 2.0 (the "License");
|
| + * you may not use this file except in compliance with the License.
|
| + * You may obtain a copy of the License at
|
| + *
|
| + * http://www.apache.org/licenses/LICENSE-2.0
|
| + *
|
| + * Unless required by applicable law or agreed to in writing, software
|
| + * distributed under the License is distributed on an "AS IS" BASIS,
|
| + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
| + * See the License for the specific language governing permissions and
|
| + * limitations under the License.
|
| + */
|
| +
|
| +#ifndef LATINIME_BIGRAM_LIST_POLICY_H
|
| +#define LATINIME_BIGRAM_LIST_POLICY_H
|
| +
|
| +#include <cstdint>
|
| +
|
| +#include "third_party/android_prediction/defines.h"
|
| +#include "third_party/android_prediction/suggest/core/policy/dictionary_bigrams_structure_policy.h"
|
| +#include "third_party/android_prediction/suggest/policyimpl/dictionary/structure/pt_common/bigram/bigram_list_read_write_utils.h"
|
| +
|
| +namespace latinime {
|
| +
|
| +class BigramListPolicy : public DictionaryBigramsStructurePolicy {
|
| + public:
|
| + BigramListPolicy(const uint8_t *const bigramsBuf, const int bufSize)
|
| + : mBigramsBuf(bigramsBuf), mBufSize(bufSize) {}
|
| +
|
| + ~BigramListPolicy() {}
|
| +
|
| + void getNextBigram(int *const outBigramPos, int *const outProbability, bool *const outHasNext,
|
| + int *const pos) const {
|
| + BigramListReadWriteUtils::BigramFlags flags;
|
| + if (!BigramListReadWriteUtils::getBigramEntryPropertiesAndAdvancePosition(mBigramsBuf,
|
| + mBufSize, &flags, outBigramPos, pos)) {
|
| + AKLOGE("Cannot read bigram entry. mBufSize: %d, pos: %d. ", mBufSize, *pos);
|
| + *outProbability = NOT_A_PROBABILITY;
|
| + *outHasNext = false;
|
| + return;
|
| + }
|
| + *outProbability = BigramListReadWriteUtils::getProbabilityFromFlags(flags);
|
| + *outHasNext = BigramListReadWriteUtils::hasNext(flags);
|
| + }
|
| +
|
| + bool skipAllBigrams(int *const pos) const {
|
| + return BigramListReadWriteUtils::skipExistingBigrams(mBigramsBuf, mBufSize, pos);
|
| + }
|
| +
|
| + private:
|
| + DISALLOW_IMPLICIT_CONSTRUCTORS(BigramListPolicy);
|
| +
|
| + const uint8_t *const mBigramsBuf;
|
| + const int mBufSize;
|
| +};
|
| +} // namespace latinime
|
| +#endif // LATINIME_BIGRAM_LIST_POLICY_H
|
|
|