mirror of
https://github.com/microsoft/PowerToys.git
synced 2024-12-13 19:19:23 +08:00
154 lines
5.2 KiB
C#
154 lines
5.2 KiB
C#
using System;
|
|
using System.Collections.Concurrent;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using hyjiacan.util.p4n;
|
|
using hyjiacan.util.p4n.format;
|
|
using Wox.Infrastructure.Logger;
|
|
using Wox.Infrastructure.Storage;
|
|
using Wox.Infrastructure.UserSettings;
|
|
|
|
namespace Wox.Infrastructure
|
|
{
|
|
public static class Alphabet
|
|
{
|
|
private static readonly HanyuPinyinOutputFormat Format = new HanyuPinyinOutputFormat();
|
|
private static ConcurrentDictionary<string, string[][]> PinyinCache;
|
|
private static BinaryStorage<ConcurrentDictionary<string, string[][]>> _pinyinStorage;
|
|
private static Settings _settings;
|
|
|
|
public static void Initialize(Settings settings)
|
|
{
|
|
_settings = settings;
|
|
InitializePinyinHelpers();
|
|
}
|
|
|
|
private static void InitializePinyinHelpers()
|
|
{
|
|
Format.setToneType(HanyuPinyinToneType.WITHOUT_TONE);
|
|
|
|
Stopwatch.Normal("|Wox.Infrastructure.Alphabet.Initialize|Preload pinyin cache", () =>
|
|
{
|
|
_pinyinStorage = new BinaryStorage<ConcurrentDictionary<string, string[][]>>("Pinyin");
|
|
PinyinCache = _pinyinStorage.TryLoad(new ConcurrentDictionary<string, string[][]>());
|
|
|
|
// force pinyin library static constructor initialize
|
|
PinyinHelper.toHanyuPinyinStringArray('T', Format);
|
|
});
|
|
Log.Info($"|Wox.Infrastructure.Alphabet.Initialize|Number of preload pinyin combination<{PinyinCache.Count}>");
|
|
}
|
|
|
|
public static void Save()
|
|
{
|
|
if (!_settings.ShouldUsePinyin)
|
|
{
|
|
return;
|
|
}
|
|
_pinyinStorage.Save(PinyinCache);
|
|
}
|
|
|
|
private static string[] EmptyStringArray = new string[0];
|
|
private static string[][] Empty2DStringArray = new string[0][];
|
|
|
|
/// <summary>
|
|
/// replace chinese character with pinyin, non chinese character won't be modified
|
|
/// <param name="word"> should be word or sentence, instead of single character. e.g. 微软 </param>
|
|
/// </summary>
|
|
public static string[] Pinyin(string word)
|
|
{
|
|
if (!_settings.ShouldUsePinyin)
|
|
{
|
|
return EmptyStringArray;
|
|
}
|
|
|
|
var pinyin = word.Select(c =>
|
|
{
|
|
var pinyins = PinyinHelper.toHanyuPinyinStringArray(c);
|
|
var result = pinyins == null ? c.ToString() : pinyins[0];
|
|
return result;
|
|
}).ToArray();
|
|
return pinyin;
|
|
}
|
|
|
|
/// <summmary>
|
|
/// replace chinese character with pinyin, non chinese character won't be modified
|
|
/// Because we don't have words dictionary, so we can only return all possiblie pinyin combination
|
|
/// e.g. 音乐 will return yinyue and yinle
|
|
/// <param name="characters"> should be word or sentence, instead of single character. e.g. 微软 </param>
|
|
/// </summmary>
|
|
public static string[][] PinyinComination(string characters)
|
|
{
|
|
if (!_settings.ShouldUsePinyin || string.IsNullOrEmpty(characters))
|
|
{
|
|
return Empty2DStringArray;
|
|
}
|
|
|
|
if (!PinyinCache.ContainsKey(characters))
|
|
{
|
|
var allPinyins = new List<string[]>();
|
|
foreach (var c in characters)
|
|
{
|
|
var pinyins = PinyinHelper.toHanyuPinyinStringArray(c, Format);
|
|
if (pinyins != null)
|
|
{
|
|
var r = pinyins.Distinct().ToArray();
|
|
allPinyins.Add(r);
|
|
}
|
|
else
|
|
{
|
|
var r = new[] { c.ToString() };
|
|
allPinyins.Add(r);
|
|
}
|
|
}
|
|
|
|
var combination = allPinyins.Aggregate(Combination).Select(c => c.Split(';')).ToArray();
|
|
PinyinCache[characters] = combination;
|
|
return combination;
|
|
}
|
|
else
|
|
{
|
|
return PinyinCache[characters];
|
|
}
|
|
}
|
|
|
|
public static string Acronym(string[] pinyin)
|
|
{
|
|
var acronym = string.Join("", pinyin.Select(p => p[0]));
|
|
return acronym;
|
|
}
|
|
|
|
public static bool ContainsChinese(string word)
|
|
{
|
|
if (!_settings.ShouldUsePinyin)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (word.Length > 40)
|
|
{
|
|
Log.Debug($"|Wox.Infrastructure.StringMatcher.ScoreForPinyin|skip too long string: {word}");
|
|
return false;
|
|
}
|
|
|
|
var chinese = word.Select(PinyinHelper.toHanyuPinyinStringArray)
|
|
.Any(p => p != null);
|
|
return chinese;
|
|
}
|
|
|
|
private static string[] Combination(string[] array1, string[] array2)
|
|
{
|
|
if (!_settings.ShouldUsePinyin)
|
|
{
|
|
return EmptyStringArray;
|
|
}
|
|
|
|
var combination = (
|
|
from a1 in array1
|
|
from a2 in array2
|
|
select $"{a1};{a2}"
|
|
).ToArray();
|
|
return combination;
|
|
}
|
|
}
|
|
}
|