133 lines
3.5 KiB
PHP
Executable File
133 lines
3.5 KiB
PHP
Executable File
<?php
|
||
|
||
namespace Overtrue\Pinyin\Converters;
|
||
|
||
use Overtrue\Pinyin\Collection;
|
||
|
||
use function array_map;
|
||
use function mb_strlen;
|
||
use function mb_substr;
|
||
use function str_starts_with;
|
||
|
||
/**
|
||
* 缓存版本的转换器
|
||
*
|
||
* 特点：
|
||
* - 缓存所有词典数据
|
||
* - 更快的重复转换速度
|
||
* - 适合批处理和长时运行的进程
|
||
* - 内存占用较高（~4MB）
|
||
*/
|
||
class CachedConverter extends AbstractConverter
|
||
{
|
||
private static ?array $charsCache = null;
|
||
|
||
private static ?array $surnamesCache = null;
|
||
|
||
private static array $wordsCache = [];
|
||
|
||
private static ?array $fullDictionary = null;
|
||
|
||
public function convert(string $string): Collection
|
||
{
|
||
$string = $this->preprocessString($string);
|
||
|
||
return $this->determineConversionStrategy($string);
|
||
}
|
||
|
||
private function determineConversionStrategy(string $string): Collection
|
||
{
|
||
// 多音字处理
|
||
if ($this->heteronym) {
|
||
return $this->convertAsChars($string, true);
|
||
}
|
||
|
||
// 仅字符转换
|
||
if ($this->noWords) {
|
||
return $this->convertAsChars($string);
|
||
}
|
||
|
||
// 替换姓氏
|
||
if ($this->asSurname) {
|
||
$string = $this->convertSurname($string);
|
||
}
|
||
|
||
// 使用缓存的完整词典
|
||
$dictionary = $this->getFullDictionary();
|
||
$string = strtr($string, $dictionary);
|
||
|
||
return $this->split($string);
|
||
}
|
||
|
||
private function getFullDictionary(): array
|
||
{
|
||
if (self::$fullDictionary === null) {
|
||
self::$fullDictionary = [];
|
||
// 按顺序加载，保证长词优先
|
||
foreach ($this->wordSegmentPaths() as $path) {
|
||
self::$fullDictionary += $this->loadWordsSegment($path);
|
||
}
|
||
}
|
||
|
||
return self::$fullDictionary;
|
||
}
|
||
|
||
private function loadWordsSegment(string $path): array
|
||
{
|
||
if (! isset(self::$wordsCache[$path])) {
|
||
self::$wordsCache[$path] = require $path;
|
||
}
|
||
|
||
return self::$wordsCache[$path];
|
||
}
|
||
|
||
protected function convertAsChars(string $string, bool $polyphonic = false): Collection
|
||
{
|
||
self::$charsCache ??= require self::CHARS_PATH;
|
||
|
||
$chars = mb_str_split($string);
|
||
$items = [];
|
||
|
||
foreach ($chars as $char) {
|
||
if (isset(self::$charsCache[$char])) {
|
||
if ($polyphonic) {
|
||
$pinyin = array_map(fn ($pinyin) => $this->formatTone($pinyin, $this->toneStyle->value), self::$charsCache[$char]);
|
||
if ($this->heteronymAsList) {
|
||
$items[] = [$char => $pinyin];
|
||
} else {
|
||
$items[$char] = $pinyin;
|
||
}
|
||
} else {
|
||
$items[$char] = $this->formatTone(self::$charsCache[$char][0], $this->toneStyle->value);
|
||
}
|
||
}
|
||
}
|
||
|
||
return new Collection($items);
|
||
}
|
||
|
||
protected function convertSurname(string $name): string
|
||
{
|
||
self::$surnamesCache ??= require self::SURNAMES_PATH;
|
||
|
||
foreach (self::$surnamesCache as $surname => $pinyin) {
|
||
if (str_starts_with($name, $surname)) {
|
||
return $pinyin.mb_substr($name, mb_strlen($surname));
|
||
}
|
||
}
|
||
|
||
return $name;
|
||
}
|
||
|
||
/**
|
||
* 清理缓存（可选）
|
||
*/
|
||
public static function clearCache(): void
|
||
{
|
||
self::$charsCache = null;
|
||
self::$surnamesCache = null;
|
||
self::$wordsCache = [];
|
||
self::$fullDictionary = null;
|
||
}
|
||
}
|