#pragma once
#include <iterator>
#include <sstream>
#include "Common.hpp"
namespace opencc {
class OPENCC_EXPORT Segments {
public:
Segments() {}
Segments(std::initializer_list<const char*> initList) {
for (const char* item : initList) {
AddSegment(item);
}
}
Segments(std::initializer_list<std::string> initList) {
for (const std::string& item : initList) {
AddSegment(item);
}
}
void AddSegment(const char* unmanagedString) {
indexes.push_back(std::make_pair(unmanaged.size(), false));
unmanaged.push_back(unmanagedString);
}
void AddSegment(const std::string& str) {
indexes.push_back(std::make_pair(managed.size(), true));
managed.push_back(str);
}
class iterator {
public:
using iterator_category = std::input_iterator_tag;
using value_type = const char*;
iterator(const Segments* const _segments, size_t _cursor)
: segments(_segments), cursor(_cursor) {}
iterator& operator++() {
cursor++;
return *this;
}
bool operator==(const iterator& that) const {
return cursor == that.cursor && segments == that.segments;
}
bool operator!=(const iterator& that) const {
return !this->operator==(that);
}
const char* operator*() const { return segments->At(cursor); }
private:
const Segments* const segments;
size_t cursor;
};
const char* At(size_t cursor) const {
const auto& index = indexes[cursor];
if (index.second) {
return managed[index.first].c_str();
} else {
return unmanaged[index.first];
}
}
size_t Length() const { return indexes.size(); }
iterator begin() const { return iterator(this, 0); }
iterator end() const { return iterator(this, indexes.size()); }
std::string ToString() const {
std::ostringstream buffer;
for (const char* segment : *this) {
buffer << segment;
}
return buffer.str();
}
private:
Segments(const Segments&) {}
std::vector<const char*> unmanaged;
std::vector<std::string> managed;
std::vector<std::pair<size_t, bool>> indexes;
};
}