mirror of
https://github.com/yanyiwu/cppjieba.git
synced 2025-07-18 00:00:12 +08:00
3c5ad24260
1. src/ -> include/cppjieba/ 2. src/limonp/ -> deps/limonp/ 3. server/husky -> deps/husky/ 4. test/unittest/gtest -> deps/gtest
91 lines
2.3 KiB
C++
91 lines
2.3 KiB
C++
#ifndef CPPJIEBA_MIXSEGMENT_H
|
|
#define CPPJIEBA_MIXSEGMENT_H
|
|
|
|
#include <cassert>
|
|
#include "MPSegment.hpp"
|
|
#include "HMMSegment.hpp"
|
|
#include "limonp/StringUtil.hpp"
|
|
|
|
namespace cppjieba {
|
|
class MixSegment: public SegmentBase {
|
|
public:
|
|
MixSegment(const string& mpSegDict, const string& hmmSegDict,
|
|
const string& userDict = "")
|
|
: mpSeg_(mpSegDict, userDict),
|
|
hmmSeg_(hmmSegDict) {
|
|
}
|
|
MixSegment(const DictTrie* dictTrie, const HMMModel* model)
|
|
: mpSeg_(dictTrie), hmmSeg_(model) {
|
|
}
|
|
~MixSegment() {
|
|
}
|
|
|
|
void Cut(const string& sentence, vector<string>& words, bool hmm = true) const {
|
|
PreFilter pre_filter(symbols_, sentence);
|
|
PreFilter::Range range;
|
|
vector<Unicode> uwords;
|
|
uwords.reserve(sentence.size());
|
|
while (pre_filter.HasNext()) {
|
|
range = pre_filter.Next();
|
|
Cut(range.begin, range.end, uwords, hmm);
|
|
}
|
|
TransCode::Encode(uwords, words);
|
|
}
|
|
|
|
void Cut(Unicode::const_iterator begin, Unicode::const_iterator end, vector<Unicode>& res, bool hmm) const {
|
|
if (!hmm) {
|
|
mpSeg_.Cut(begin, end, res);
|
|
return;
|
|
}
|
|
vector<Unicode> words;
|
|
words.reserve(end - begin);
|
|
mpSeg_.Cut(begin, end, words);
|
|
|
|
vector<Unicode> hmmRes;
|
|
hmmRes.reserve(end - begin);
|
|
Unicode piece;
|
|
piece.reserve(end - begin);
|
|
for (size_t i = 0, j = 0; i < words.size(); i++) {
|
|
//if mp Get a word, it's ok, put it into result
|
|
if (1 != words[i].size() || (words[i].size() == 1 && mpSeg_.IsUserDictSingleChineseWord(words[i][0]))) {
|
|
res.push_back(words[i]);
|
|
continue;
|
|
}
|
|
|
|
// if mp Get a single one and it is not in userdict, collect it in sequence
|
|
j = i;
|
|
while (j < words.size() && 1 == words[j].size() && !mpSeg_.IsUserDictSingleChineseWord(words[j][0])) {
|
|
piece.push_back(words[j][0]);
|
|
j++;
|
|
}
|
|
|
|
// Cut the sequence with hmm
|
|
hmmSeg_.Cut(piece.begin(), piece.end(), hmmRes);
|
|
|
|
//put hmm result to result
|
|
for (size_t k = 0; k < hmmRes.size(); k++) {
|
|
res.push_back(hmmRes[k]);
|
|
}
|
|
|
|
//clear tmp vars
|
|
piece.clear();
|
|
hmmRes.clear();
|
|
|
|
//let i jump over this piece
|
|
i = j - 1;
|
|
}
|
|
}
|
|
|
|
const DictTrie* GetDictTrie() const {
|
|
return mpSeg_.GetDictTrie();
|
|
}
|
|
private:
|
|
MPSegment mpSeg_;
|
|
HMMSegment hmmSeg_;
|
|
|
|
}; // class MixSegment
|
|
|
|
} // namespace cppjieba
|
|
|
|
#endif
|