116 lines
2.4 KiB
C++
116 lines
2.4 KiB
C++
#ifndef MARISA_GRIMOIRE_TRIE_RANGE_H_
|
|
#define MARISA_GRIMOIRE_TRIE_RANGE_H_
|
|
|
|
#include "marisa/base.h"
|
|
|
|
namespace marisa {
|
|
namespace grimoire {
|
|
namespace trie {
|
|
|
|
class Range {
|
|
public:
|
|
Range() : begin_(0), end_(0), key_pos_(0) {}
|
|
|
|
void set_begin(std::size_t begin) {
|
|
MARISA_DEBUG_IF(begin > MARISA_UINT32_MAX, MARISA_SIZE_ERROR);
|
|
begin_ = static_cast<UInt32>(begin);
|
|
}
|
|
void set_end(std::size_t end) {
|
|
MARISA_DEBUG_IF(end > MARISA_UINT32_MAX, MARISA_SIZE_ERROR);
|
|
end_ = static_cast<UInt32>(end);
|
|
}
|
|
void set_key_pos(std::size_t key_pos) {
|
|
MARISA_DEBUG_IF(key_pos > MARISA_UINT32_MAX, MARISA_SIZE_ERROR);
|
|
key_pos_ = static_cast<UInt32>(key_pos);
|
|
}
|
|
|
|
std::size_t begin() const {
|
|
return begin_;
|
|
}
|
|
std::size_t end() const {
|
|
return end_;
|
|
}
|
|
std::size_t key_pos() const {
|
|
return key_pos_;
|
|
}
|
|
|
|
private:
|
|
UInt32 begin_;
|
|
UInt32 end_;
|
|
UInt32 key_pos_;
|
|
};
|
|
|
|
inline Range make_range(std::size_t begin, std::size_t end,
|
|
std::size_t key_pos) {
|
|
Range range;
|
|
range.set_begin(begin);
|
|
range.set_end(end);
|
|
range.set_key_pos(key_pos);
|
|
return range;
|
|
}
|
|
|
|
class WeightedRange {
|
|
public:
|
|
WeightedRange() : range_(), weight_(0.0F) {}
|
|
|
|
void set_range(const Range &range) {
|
|
range_ = range;
|
|
}
|
|
void set_begin(std::size_t begin) {
|
|
range_.set_begin(begin);
|
|
}
|
|
void set_end(std::size_t end) {
|
|
range_.set_end(end);
|
|
}
|
|
void set_key_pos(std::size_t key_pos) {
|
|
range_.set_key_pos(key_pos);
|
|
}
|
|
void set_weight(float weight) {
|
|
weight_ = weight;
|
|
}
|
|
|
|
const Range &range() const {
|
|
return range_;
|
|
}
|
|
std::size_t begin() const {
|
|
return range_.begin();
|
|
}
|
|
std::size_t end() const {
|
|
return range_.end();
|
|
}
|
|
std::size_t key_pos() const {
|
|
return range_.key_pos();
|
|
}
|
|
float weight() const {
|
|
return weight_;
|
|
}
|
|
|
|
private:
|
|
Range range_;
|
|
float weight_;
|
|
};
|
|
|
|
inline bool operator<(const WeightedRange &lhs, const WeightedRange &rhs) {
|
|
return lhs.weight() < rhs.weight();
|
|
}
|
|
|
|
inline bool operator>(const WeightedRange &lhs, const WeightedRange &rhs) {
|
|
return lhs.weight() > rhs.weight();
|
|
}
|
|
|
|
inline WeightedRange make_weighted_range(std::size_t begin, std::size_t end,
|
|
std::size_t key_pos, float weight) {
|
|
WeightedRange range;
|
|
range.set_begin(begin);
|
|
range.set_end(end);
|
|
range.set_key_pos(key_pos);
|
|
range.set_weight(weight);
|
|
return range;
|
|
}
|
|
|
|
} // namespace trie
|
|
} // namespace grimoire
|
|
} // namespace marisa
|
|
|
|
#endif // MARISA_GRIMOIRE_TRIE_RANGE_H_
|