char_traits.hpp

浏览该文件的文档.

#ifndef NEFORCE_CORE_STRING_CHAR_TRAITS_HPP__
#define NEFORCE_CORE_STRING_CHAR_TRAITS_HPP__

 
#include "NeForce/core/algorithm/compare.hpp"
#include "NeForce/core/numeric/numeric_traits.hpp"
NEFORCE_BEGIN_NAMESPACE__


template <typename CharT, typename IntT>
struct base_char_traits {
    using char_type = CharT; 
    using int_type = IntT;   
 
    static_assert(sizeof(int_type) >= sizeof(char_type),
                  "int_type must be able to represent all char_type values plus EOF");

    static constexpr char_type* copy(char_type* dest, const char_type* srcs, const size_t count) noexcept {
        _NEFORCE memory_copy(dest, srcs, count * sizeof(char_type));
        return dest;
    }

    static constexpr char_type* move(char_type* dest, const char_type* srcs, const size_t count) noexcept {
        _NEFORCE memory_move(dest, srcs, count * sizeof(char_type));
        return dest;
    }

    NEFORCE_NODISCARD static constexpr int compare(const char_type* lhs, const char_type* rhs, size_t count) noexcept {
        return _NEFORCE string_compare(lhs, rhs, count);
    }

    NEFORCE_NODISCARD static constexpr size_t length(const char_type* str) noexcept {
        return _NEFORCE string_length(str);
    }

    NEFORCE_NODISCARD static constexpr const char_type* find(const char_type* str, const size_t count,
                                                             const char_type target) noexcept {
        return _NEFORCE string_find<char_type>(str, target, count);
    }

    static constexpr char_type* assign(char_type* const str, const size_t count, const char_type chr) noexcept {
        return _NEFORCE string_set<char_type>(str, chr, count);
    }

    static constexpr void assign(char_type& lhs, const char_type rhs) noexcept { lhs = rhs; }

    NEFORCE_NODISCARD static constexpr bool eq(const char_type lhs, const char_type rhs) noexcept { return lhs == rhs; }

    NEFORCE_NODISCARD static constexpr bool lt(const char_type lhs, const char_type rhs) noexcept { return lhs < rhs; }

    NEFORCE_NODISCARD static constexpr int_type not_eof(const int_type rsc) noexcept {
        return rsc == eof() ? static_cast<int_type>(0) : rsc;
    }

    NEFORCE_NODISCARD static constexpr int_type eof() noexcept { return static_cast<int_type>(-1); }
};

template <typename CharT, typename IntT>
struct narrow_char_traits : private base_char_traits<CharT, IntT> {
    static_assert(sizeof(CharT) == sizeof(byte_t), "size of CharT must be the same as byte type");
 
private:
    using base_type = base_char_traits<CharT, IntT>;
 
public:
    using char_type = CharT; 
    using int_type = IntT;   
 
    using base_type::copy;
    using base_type::eof;
    using base_type::eq;
    using base_type::length;
    using base_type::lt;
    using base_type::move;
    using base_type::not_eof;
 
public:
    NEFORCE_NODISCARD static constexpr int compare(const char_type* lhs, const char_type* rhs,
                                                   const size_t n) noexcept {
        return _NEFORCE memory_compare(lhs, rhs, n);
    }

    NEFORCE_NODISCARD static constexpr const char_type* find(const char_type* str, const size_t n,
                                                             const char_type chr) noexcept {
        return static_cast<const char_type*>(_NEFORCE memory_find(str, chr, n));
    }

    static constexpr char_type* assign(char_type* str, size_t n, const char_type chr) noexcept {
        return static_cast<char_type*>(_NEFORCE memory_set(str, chr, n));
    }

    static constexpr void assign(char_type& lhs, const char_type& rhs) noexcept { lhs = rhs; }
};
 

template <typename CharT>
struct char_traits : base_char_traits<CharT, int64_t> {};

template <>
struct char_traits<char> : narrow_char_traits<char, int32_t> {};

template <>
struct char_traits<wchar_t> : base_char_traits<wchar_t, uint32_t> {};
 
#ifdef NEFORCE_STANDARD_20
template <>
struct char_traits<char8_t> : narrow_char_traits<char8_t, uint32_t> {};
#endif

template <>
struct char_traits<char16_t> : base_char_traits<char16_t, uint32_t> {};

template <>
struct char_traits<char32_t> : base_char_traits<char32_t, uint32_t> {};
 

template <typename Traits>
using char_traits_char_t = typename Traits::char_type;

template <typename Traits>
using char_traits_ptr_t = const typename Traits::char_type*;
 

NEFORCE_BEGIN_INNER__

template <typename CharT, bool IsChar = is_character_v<CharT>>
class __string_bitmap {
private:
    bool matches_[numeric_traits<byte_t>::max() + 1] = {}; 
 
public:
    constexpr __string_bitmap() = default;

    constexpr bool mark(const CharT* first, const CharT* const last) noexcept {
        for (; first != last; ++first) {
            matches_[static_cast<byte_t>(*first)] = true;
        }
        return true;
    }

    constexpr bool match(const CharT chr) const noexcept { return matches_[static_cast<byte_t>(chr)]; }
};
 
template <typename CharT>
class __string_bitmap<CharT, false> {};
 
NEFORCE_END_INNER__

template <typename Traits>
constexpr bool char_traits_equal(const char_traits_ptr_t<Traits> lhs, const size_t lh_size,
                                 const char_traits_ptr_t<Traits> rhs, const size_t rh_size) noexcept {
    if (lh_size != rh_size) {
        return false;
    }
    if (lh_size == 0U) {
        return true;
    }
 
    return Traits::compare(lhs, rhs, lh_size) == 0;
}

template <typename Traits>
constexpr int char_traits_compare(const char_traits_ptr_t<Traits> lhs, const size_t lh_size,
                                  const char_traits_ptr_t<Traits> rhs, const size_t rh_size) noexcept {
    const int state = Traits::compare(lhs, rhs, _NEFORCE min(lh_size, rh_size));
    if (state != 0) {
        return state;
    }
 
    if (lh_size < rh_size) {
        return -1;
    }
    if (lh_size > rh_size) {
        return 1;
    }
    return 0;
}

template <typename Traits>
constexpr size_t char_traits_find(const char_traits_ptr_t<Traits> dest, const size_t dest_size, const size_t start,
                                  const char_traits_ptr_t<Traits> rsc, const size_t rsc_size) noexcept {
    if (rsc_size > dest_size || start > dest_size - rsc_size) {
        return static_cast<size_t>(-1);
    }
    if (rsc_size == 0) {
        return start;
    }
 
    const auto may_match_end = dest + (dest_size - rsc_size) + 1;
    for (auto if_match = dest + start;; ++if_match) {
        if_match = Traits::find(if_match, static_cast<size_t>(may_match_end - if_match), *rsc);
        if (!if_match) {
            return static_cast<size_t>(-1);
        }
 
        if (Traits::compare(if_match, rsc, rsc_size) == 0) {
            return static_cast<size_t>(if_match - dest);
        }
    }
}

template <typename Traits>
constexpr size_t char_traits_find_char(const char_traits_ptr_t<Traits> dest, const size_t dest_size, const size_t start,
                                       const char_traits_char_t<Traits> chr) noexcept {
    if (start < dest_size) {
        const auto found = Traits::find(dest + start, dest_size - start, chr);
        if (found) {
            return static_cast<size_t>(found - dest);
        }
    }
    return static_cast<size_t>(-1);
}

template <typename Traits>
constexpr size_t char_traits_rfind(const char_traits_ptr_t<Traits> dest, const size_t dest_size, const size_t start,
                                   const char_traits_ptr_t<Traits> rsc, const size_t rsc_size) noexcept {
    if (rsc_size == 0) {
        return _NEFORCE min(start, dest_size);
    }
 
    if (rsc_size <= dest_size) {
        for (auto if_match = dest + _NEFORCE min(start, dest_size - rsc_size);; --if_match) {
            if (Traits::eq(*if_match, *rsc) && Traits::compare(if_match, rsc, rsc_size) == 0) {
                return static_cast<size_t>(if_match - dest);
            }
 
            if (if_match == dest) {
                break;
            }
        }
    }
    return static_cast<size_t>(-1);
}

template <typename Traits>
constexpr size_t char_traits_rfind_char(const char_traits_ptr_t<Traits> dest, const size_t dest_size,
                                        const size_t start, const char_traits_char_t<Traits> chr) noexcept {
    if (dest_size != 0) {
        for (auto if_match = dest + _NEFORCE min(start, dest_size - 1);; --if_match) {
            if (Traits::eq(*if_match, chr)) {
                return static_cast<size_t>(if_match - dest);
            }
 
            if (if_match == dest) {
                break;
            }
        }
    }
    return static_cast<size_t>(-1);
}

template <typename Traits, enable_if_t<
#ifdef NEFORCE_STANDARD_17
                                   is_specialization_v<Traits, char_traits>
#else
                                   is_specialization_v<Traits, char_traits>()
#endif
                                   ,
                                   int> = 0>
constexpr size_t char_traits_find_first_of(const char_traits_ptr_t<Traits> dest, const size_t dest_size,
                                           const size_t start, const char_traits_ptr_t<Traits> rsc,
                                           const size_t rsc_size) noexcept {
    if (rsc_size != 0 && start < dest_size) {
        inner::__string_bitmap<char_traits_char_t<Traits>> match;
        if (!match.mark(rsc, rsc + rsc_size)) {
            return (char_traits_find_first_of<Traits, false>) (dest, dest_size, start, rsc, rsc_size);
        }
        const auto end = dest + dest_size;
        for (auto if_match = dest + start; if_match < end; ++if_match) {
            if (match.match(*if_match)) {
                return static_cast<size_t>(if_match - dest);
            }
        }
    }
    return static_cast<size_t>(-1);
}

template <typename Traits, enable_if_t<
#ifdef NEFORCE_STANDARD_17
                                   !is_specialization_v<Traits, char_traits>
#else
                                   !is_specialization_v<Traits, char_traits>()
#endif
                                   ,
                                   int> = 0>
constexpr size_t char_traits_find_first_of(const char_traits_ptr_t<Traits> dest, const size_t dest_size,
                                           const size_t start, const char_traits_ptr_t<Traits> rsc,
                                           const size_t rsc_size) noexcept {
    if (rsc_size != 0 && start < dest_size) {
        const auto end = dest + dest_size;
        for (auto if_match = dest + start; if_match < end; ++if_match) {
            if (Traits::find(rsc, rsc_size, *if_match)) {
                return static_cast<size_t>(if_match - dest);
            }
        }
    }
    return static_cast<size_t>(-1);
}

template <typename Traits, enable_if_t<
#ifdef NEFORCE_STANDARD_17
                                   is_specialization_v<Traits, char_traits>
#else
                                   is_specialization_v<Traits, char_traits>()
#endif
                                   ,
                                   int> = 0>
constexpr size_t char_traits_find_last_of(const char_traits_ptr_t<Traits> dest, const size_t dest_size,
                                          const size_t start, const char_traits_ptr_t<Traits> rsc,
                                          const size_t rsc_size) noexcept {
    if (rsc_size != 0 && dest_size != 0) {
        inner::__string_bitmap<char_traits_char_t<Traits>> match;
        if (!match.mark(rsc, rsc + rsc_size)) {
            return (char_traits_find_last_of<Traits, false>) (dest, dest_size, start, rsc, rsc_size);
        }
 
        for (auto if_match = dest + _NEFORCE min(start, dest_size - 1);; --if_match) {
            if (match.match(*if_match)) {
                return static_cast<size_t>(if_match - dest);
            }
 
            if (if_match == dest) {
                break;
            }
        }
    }
    return static_cast<size_t>(-1);
}

template <typename Traits, enable_if_t<
#ifdef NEFORCE_STANDARD_17
                                   !is_specialization_v<Traits, char_traits>
#else
                                   !is_specialization_v<Traits, char_traits>()
#endif
                                   ,
                                   int> = 0>
constexpr size_t char_traits_find_last_of(const char_traits_ptr_t<Traits> dest, const size_t dest_size,
                                          const size_t start, const char_traits_ptr_t<Traits> rsc,
                                          const size_t rsc_size) noexcept {
    if (rsc_size != 0 && dest_size != 0) {
        for (auto if_match = dest + _NEFORCE min(start, dest_size - 1);; --if_match) {
            if (Traits::find(rsc, rsc_size, *if_match)) {
                return static_cast<size_t>(if_match - dest);
            }
 
            if (if_match == dest) {
                break;
            }
        }
    }
    return static_cast<size_t>(-1);
}

template <typename Traits, enable_if_t<
#ifdef NEFORCE_STANDARD_17
                                   is_specialization_v<Traits, char_traits>
#else
                                   is_specialization_v<Traits, char_traits>()
#endif
                                   ,
                                   int> = 0>
constexpr size_t char_traits_find_first_not_of(const char_traits_ptr_t<Traits> dest, const size_t dest_size,
                                               const size_t start, const char_traits_ptr_t<Traits> rsc,
                                               const size_t rsc_size) noexcept {
    if (start < dest_size) {
        inner::__string_bitmap<char_traits_char_t<Traits>> match;
        if (!match.mark(rsc, rsc + rsc_size)) {
            return (char_traits_find_first_not_of<Traits, false>) (dest, dest_size, start, rsc, rsc_size);
        }
 
        const auto end = dest + dest_size;
        for (auto if_match = dest + start; if_match < end; ++if_match) {
            if (!match.match(*if_match)) {
                return static_cast<size_t>(if_match - dest);
            }
        }
    }
    return static_cast<size_t>(-1);
}

template <typename Traits, enable_if_t<
#ifdef NEFORCE_STANDARD_17
                                   !is_specialization_v<Traits, char_traits>
#else
                                   !is_specialization_v<Traits, char_traits>()
#endif
                                   ,
                                   int> = 0>
constexpr size_t char_traits_find_first_not_of(const char_traits_ptr_t<Traits> dest, const size_t dest_size,
                                               const size_t start, const char_traits_ptr_t<Traits> rsc,
                                               const size_t rsc_size) noexcept {
    if (start < dest_size) {
        const auto end = dest + dest_size;
        for (auto if_match = dest + start; if_match < end; ++if_match) {
            if (!Traits::find(rsc, rsc_size, *if_match)) {
                return static_cast<size_t>(if_match - dest);
            }
        }
    }
    return static_cast<size_t>(-1);
}

template <typename Traits>
constexpr size_t char_traits_find_not_char(const char_traits_ptr_t<Traits> dest, const size_t dest_size,
                                           const size_t start, const char_traits_char_t<Traits> chr) noexcept {
    if (start < dest_size) {
        const auto end = dest + dest_size;
        for (auto if_match = dest + start; if_match < end; ++if_match) {
            if (!Traits::eq(*if_match, chr)) {
                return static_cast<size_t>(if_match - dest);
            }
        }
    }
    return static_cast<size_t>(-1);
}

template <typename Traits, enable_if_t<
#ifdef NEFORCE_STANDARD_17
                                   is_specialization_v<Traits, char_traits>
#else
                                   is_specialization_v<Traits, char_traits>()
#endif
                                   ,
                                   int> = 0>
constexpr size_t char_traits_find_last_not_of(const char_traits_ptr_t<Traits> dest, const size_t dest_size,
                                              const size_t start, const char_traits_ptr_t<Traits> rsc,
                                              const size_t rsc_size) noexcept {
    if (dest_size != 0) {
        inner::__string_bitmap<char_traits_char_t<Traits>> match;
        if (!match.mark(rsc, rsc + rsc_size)) {
            return (char_traits_find_last_not_of<Traits, false>) (dest, dest_size, start, rsc, rsc_size);
        }
 
        for (auto if_match = dest + _NEFORCE min(start, dest_size - 1);; --if_match) {
            if (!match.match(*if_match)) {
                return static_cast<size_t>(if_match - dest);
            }
 
            if (if_match == dest) {
                break;
            }
        }
    }
    return static_cast<size_t>(-1);
}

template <typename Traits, enable_if_t<
#ifdef NEFORCE_STANDARD_17
                                   !is_specialization_v<Traits, char_traits>
#else
                                   !is_specialization_v<Traits, char_traits>()
#endif
                                   ,
                                   int> = 0>
constexpr size_t char_traits_find_last_not_of(const char_traits_ptr_t<Traits> dest, const size_t dest_size,
                                              const size_t start, const char_traits_ptr_t<Traits> rsc,
                                              const size_t rsc_size) noexcept {
    if (dest_size != 0) {
        for (auto if_match = dest + _NEFORCE min(start, dest_size - 1);; --if_match) {
            if (!Traits::find(rsc, rsc_size, *if_match)) {
                return static_cast<size_t>(if_match - dest);
            }
 
            if (if_match == dest) {
                break;
            }
        }
    }
    return static_cast<size_t>(-1);
}

template <typename Traits>
constexpr size_t char_traits_rfind_not_char(const char_traits_ptr_t<Traits> dest, const size_t dest_size,
                                            const size_t start, const char_traits_char_t<Traits> chr) noexcept {
    if (dest_size != 0) {
        for (auto if_match = dest + _NEFORCE min(start, dest_size - 1);; --if_match) {
            if (!Traits::eq(*if_match, chr)) {
                return static_cast<size_t>(if_match - dest);
            }
 
            if (if_match == dest) {
                break;
            }
        }
    }
    return static_cast<size_t>(-1);
}
 // CharTraits
 
#define __NEFORCE_BUILD_CHAR_PTR_HASH(OPT)                                                  \
    template <>                                                                             \
    struct hash<OPT*> {                                                                     \
        NEFORCE_NODISCARD constexpr size_t operator()(const OPT* str) const noexcept {      \
            return FNV_hash_string(str, char_traits<OPT>::length(str));                     \
        }                                                                                   \
    };                                                                                      \
    template <>                                                                             \
    struct hash<const OPT*> {                                                               \
        NEFORCE_NODISCARD constexpr size_t operator()(const OPT* str) const noexcept {      \
            return FNV_hash_string(str, char_traits<OPT>::length(str));                     \
        }                                                                                   \
    };                                                                                      \
    template <size_t N>                                                                     \
    struct hash<OPT[N]> {                                                                   \
        NEFORCE_NODISCARD constexpr size_t operator()(const OPT (&str)[N]) const noexcept { \
            return FNV_hash_string(str, N - 1);                                             \
        }                                                                                   \
    };                                                                                      \
    template <size_t N>                                                                     \
    struct hash<const OPT[N]> {                                                             \
        NEFORCE_NODISCARD constexpr size_t operator()(const OPT (&str)[N]) const noexcept { \
            return FNV_hash_string(str, N - 1);                                             \
        }                                                                                   \
    };
 
NEFORCE_MACRO_RANGE_CHARS(__NEFORCE_BUILD_CHAR_PTR_HASH)
#undef __NEFORCE_BUILD_CHAR_PTR_HASH
 
NEFORCE_END_NAMESPACE__
#endif // NEFORCE_CORE_STRING_CHAR_TRAITS_HPP__