82 lines
2.6 KiB
C++
82 lines
2.6 KiB
C++
/*!
|
|
* Copyright 2021 by XGBoost Contributors
|
|
*/
|
|
#ifndef XGBOOST_STRING_VIEW_H_
|
|
#define XGBOOST_STRING_VIEW_H_
|
|
#include <xgboost/logging.h>
|
|
|
|
#include <algorithm>
|
|
#include <iterator>
|
|
#include <ostream>
|
|
#include <string>
|
|
|
|
namespace xgboost {
|
|
struct StringView {
|
|
private:
|
|
using CharT = char; // unsigned char
|
|
using Traits = std::char_traits<CharT>;
|
|
CharT const* str_{nullptr};
|
|
size_t size_{0};
|
|
|
|
public:
|
|
using iterator = const CharT*; // NOLINT
|
|
using const_iterator = iterator; // NOLINT
|
|
using reverse_iterator = std::reverse_iterator<const_iterator>; // NOLINT
|
|
using const_reverse_iterator = reverse_iterator; // NOLINT
|
|
|
|
public:
|
|
constexpr StringView() = default;
|
|
constexpr StringView(CharT const* str, size_t size) : str_{str}, size_{size} {}
|
|
explicit StringView(std::string const& str) : str_{str.c_str()}, size_{str.size()} {}
|
|
StringView(CharT const* str) : str_{str}, size_{Traits::length(str)} {} // NOLINT
|
|
|
|
CharT const& operator[](size_t p) const { return str_[p]; }
|
|
CharT const& at(size_t p) const { // NOLINT
|
|
CHECK_LT(p, size_);
|
|
return str_[p];
|
|
}
|
|
constexpr size_t size() const { return size_; } // NOLINT
|
|
StringView substr(size_t beg, size_t n) const { // NOLINT
|
|
CHECK_LE(beg, size_);
|
|
size_t len = std::min(n, size_ - beg);
|
|
return {str_ + beg, len};
|
|
}
|
|
CharT const* c_str() const { return str_; } // NOLINT
|
|
|
|
constexpr CharT const* cbegin() const { return str_; } // NOLINT
|
|
constexpr CharT const* cend() const { return str_ + size(); } // NOLINT
|
|
constexpr CharT const* begin() const { return str_; } // NOLINT
|
|
constexpr CharT const* end() const { return str_ + size(); } // NOLINT
|
|
|
|
const_reverse_iterator rbegin() const noexcept { // NOLINT
|
|
return const_reverse_iterator(this->end());
|
|
}
|
|
const_reverse_iterator crbegin() const noexcept { // NOLINT
|
|
return const_reverse_iterator(this->end());
|
|
}
|
|
const_reverse_iterator rend() const noexcept { // NOLINT
|
|
return const_reverse_iterator(this->begin());
|
|
}
|
|
const_reverse_iterator crend() const noexcept { // NOLINT
|
|
return const_reverse_iterator(this->begin());
|
|
}
|
|
};
|
|
|
|
inline std::ostream& operator<<(std::ostream& os, StringView const v) {
|
|
for (auto c : v) {
|
|
os.put(c);
|
|
}
|
|
return os;
|
|
}
|
|
|
|
inline bool operator==(StringView l, StringView r) {
|
|
if (l.size() != r.size()) {
|
|
return false;
|
|
}
|
|
return std::equal(l.cbegin(), l.cend(), r.cbegin());
|
|
}
|
|
|
|
inline bool operator!=(StringView l, StringView r) { return !(l == r); }
|
|
} // namespace xgboost
|
|
#endif // XGBOOST_STRING_VIEW_H_
|