mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-08-29 06:22:46 +08:00
95 lines
3.0 KiB
C++
95 lines
3.0 KiB
C++
/**
|
|
* The following unicode files:
|
|
* - unicode.h
|
|
* - unicode_data.cpp
|
|
* - unicode.cpp
|
|
* are copied from llama.cpp with minor modifications:
|
|
* https://github.com/ggerganov/llama.cpp/
|
|
* Commit hash: 8d59d911711b8f1ba9ec57c4b192ccd2628af033
|
|
*/
|
|
#pragma once
|
|
|
|
#include <cstdint>
|
|
#include <string>
|
|
#include <vector>
|
|
#include <unordered_map>
|
|
#include <unordered_set>
|
|
|
|
/* unicode-data.h */
|
|
struct range_nfd {
|
|
uint32_t first;
|
|
uint32_t last;
|
|
uint32_t nfd;
|
|
};
|
|
|
|
static const uint32_t MAX_CODEPOINTS = 0x110000;
|
|
|
|
extern const std::initializer_list<std::pair<uint32_t, uint16_t>> unicode_ranges_flags;
|
|
extern const std::unordered_set<uint32_t> unicode_set_whitespace;
|
|
extern const std::initializer_list<std::pair<uint32_t, uint32_t>> unicode_map_lowercase;
|
|
extern const std::initializer_list<std::pair<uint32_t, uint32_t>> unicode_map_uppercase;
|
|
extern const std::initializer_list<range_nfd> unicode_ranges_nfd;
|
|
|
|
/* original unicode.h */
|
|
|
|
struct unicode_cpt_flags {
|
|
enum {
|
|
UNDEFINED = 0x0001,
|
|
NUMBER = 0x0002, // regex: \p{N}
|
|
LETTER = 0x0004, // regex: \p{L}
|
|
SEPARATOR = 0x0008, // regex: \p{Z}
|
|
ACCENT_MARK = 0x0010, // regex: \p{M}
|
|
PUNCTUATION = 0x0020, // regex: \p{P}
|
|
SYMBOL = 0x0040, // regex: \p{S}
|
|
CONTROL = 0x0080, // regex: \p{C}
|
|
MASK_CATEGORIES = 0x00FF,
|
|
};
|
|
|
|
// codepoint type
|
|
uint16_t is_undefined : 1;
|
|
uint16_t is_number : 1; // regex: \p{N}
|
|
uint16_t is_letter : 1; // regex: \p{L}
|
|
uint16_t is_separator : 1; // regex: \p{Z}
|
|
uint16_t is_accent_mark : 1; // regex: \p{M}
|
|
uint16_t is_punctuation : 1; // regex: \p{P}
|
|
uint16_t is_symbol : 1; // regex: \p{S}
|
|
uint16_t is_control : 1; // regex: \p{C}
|
|
// helper flags
|
|
uint16_t is_whitespace : 1; // regex: \s
|
|
uint16_t is_lowercase : 1;
|
|
uint16_t is_uppercase : 1;
|
|
uint16_t is_nfd : 1;
|
|
|
|
// decode from uint16
|
|
inline unicode_cpt_flags(const uint16_t flags = 0) {
|
|
*reinterpret_cast<uint16_t*>(this) = flags;
|
|
}
|
|
|
|
inline uint16_t as_uint() const {
|
|
return *reinterpret_cast<const uint16_t*>(this);
|
|
}
|
|
|
|
inline uint16_t category_flag() const {
|
|
return this->as_uint() & MASK_CATEGORIES;
|
|
}
|
|
};
|
|
|
|
size_t unicode_len_utf8(char src);
|
|
|
|
std::string unicode_cpt_to_utf8 (uint32_t cpt);
|
|
uint32_t unicode_cpt_from_utf8(const std::string & utf8, size_t & offset);
|
|
|
|
std::vector<uint32_t> unicode_cpts_from_utf8(const std::string & utf8);
|
|
|
|
std::vector<uint32_t> unicode_cpts_normalize_nfd(const std::vector<uint32_t> & cpts);
|
|
|
|
unicode_cpt_flags unicode_cpt_flags_from_cpt (uint32_t cpt);
|
|
unicode_cpt_flags unicode_cpt_flags_from_utf8(const std::string & utf8);
|
|
|
|
std::string unicode_byte_to_utf8(uint8_t byte);
|
|
uint8_t unicode_utf8_to_byte(const std::string & utf8);
|
|
|
|
uint32_t unicode_tolower(uint32_t cpt);
|
|
|
|
std::vector<std::string> unicode_regex_split(const std::string & text, const std::vector<std::string> & regex_exprs);
|