fanlusky
a7779d2b05
`vendor/MetasequoiaImeEngine` submodule 改为 `engine/` 目录，导入的是 product-lock.json 原本钉住的提交 c810d201，逐文件核对过 blob，因此这是一次 搬迁而不是引擎升级。引擎自己的嵌套 submodule（googlepinyinime-rev、utfcpp、 miniaudio）按各自 gitlink 一并展开，来源与提交见 engine/UPSTREAM.md。 Windows 专门化裁掉了不属于本产品的部分：whisper.cpp 与本地推理 provider、 macOS 示例宿主、整个 dictionary/ 建库流水线、上游 CI 与 release-please 配置。 词库仍按 product-lock.json 在构建时取发布产物。 引擎不再进 product-lock.json：没有 gitlink 就没有可比对的东西，锁里写一个上游 commit 等于宣称这棵树仍等于上游，而第一个 Windows 专门化改动落地时这句话就假了。 verify-contracts / verify-checkout 随之移除，verify-published 保留。 CI 与工具链：引擎源码并入根 scripts/format.sh（删掉引擎自己的那份，避免两套排除 列表分叉）；引擎 voice.yml 里的 silero_vad.onnx 摘要校验移到本仓 lock job； engine-bump-triage 工作流与其测试删除；gitsubmodule 的 dependabot 频率由 daily 改为 monthly。 README、README.en、AGENTS.md、CONTRIBUTING、docs/product-release.md 及各子目录 文档同步说明新的位置与边界。
350 lines
13 KiB
C++
350 lines
13 KiB
C++
#include "date_time_query.h"
|
||
|
||
#include <algorithm>
|
||
#include <array>
|
||
#include <cstdint>
|
||
#include <cstdio>
|
||
#include <ctime>
|
||
|
||
namespace metasequoia::local_modes
|
||
{
|
||
namespace
|
||
{
|
||
bool is_date_keyword(const std::string &keyword)
|
||
{
|
||
return keyword == "rq" || keyword == "riqi" || keyword == "date";
|
||
}
|
||
|
||
bool is_time_keyword(const std::string &keyword)
|
||
{
|
||
return keyword == "sj" || keyword == "shijian" || keyword == "time";
|
||
}
|
||
|
||
bool is_week_keyword(const std::string &keyword)
|
||
{
|
||
return keyword == "xq" || keyword == "xingqi" || keyword == "week";
|
||
}
|
||
|
||
constexpr const char *kWeekdays[] = {"星期日", "星期一", "星期二", "星期三", "星期四", "星期五", "星期六"};
|
||
constexpr const char *kShortWeekdays[] = {"周日", "周一", "周二", "周三", "周四", "周五", "周六"};
|
||
constexpr const char *kEnglishWeekdays[] = {"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"};
|
||
constexpr const char *kEnglishFullWeekdays[] = {"Sunday", "Monday", "Tuesday", "Wednesday",
|
||
"Thursday", "Friday", "Saturday"};
|
||
|
||
unsigned weekday_index(const LocalDateTime &now)
|
||
{
|
||
return std::min(now.weekday, 6U);
|
||
}
|
||
|
||
std::string format(const char *pattern, unsigned first, unsigned second = 0, unsigned third = 0)
|
||
{
|
||
char buffer[96] = {};
|
||
std::snprintf(buffer, sizeof(buffer), pattern, first, second, third);
|
||
return buffer;
|
||
}
|
||
|
||
std::string chinese_digits(unsigned value)
|
||
{
|
||
static constexpr const char *kDigits[] = {"〇", "一", "二", "三", "四", "五", "六", "七", "八", "九"};
|
||
std::string result;
|
||
for (const char digit : std::to_string(value))
|
||
{
|
||
result += kDigits[digit - '0'];
|
||
}
|
||
return result;
|
||
}
|
||
|
||
std::string chinese_number(unsigned value)
|
||
{
|
||
static constexpr const char *kDigits[] = {"", "一", "二", "三", "四", "五", "六", "七", "八", "九"};
|
||
if (value < 10)
|
||
{
|
||
return kDigits[value];
|
||
}
|
||
if (value == 10)
|
||
{
|
||
return "十";
|
||
}
|
||
if (value < 20)
|
||
{
|
||
return std::string("十") + kDigits[value - 10];
|
||
}
|
||
if (value % 10 == 0)
|
||
{
|
||
return std::string(kDigits[value / 10]) + "十";
|
||
}
|
||
return std::string(kDigits[value / 10]) + "十" + kDigits[value % 10];
|
||
}
|
||
|
||
std::string financial_digits(unsigned value, unsigned minimum_digits = 1)
|
||
{
|
||
static constexpr const char *kDigits[] = {"零", "壹", "贰", "叁", "肆", "伍", "陆", "柒", "捌", "玖"};
|
||
std::string digits = std::to_string(value);
|
||
if (digits.size() < minimum_digits)
|
||
{
|
||
digits.insert(0, minimum_digits - digits.size(), '0');
|
||
}
|
||
|
||
std::string result;
|
||
for (const char digit : digits)
|
||
{
|
||
result += kDigits[digit - '0'];
|
||
}
|
||
return result;
|
||
}
|
||
|
||
// Each entry encodes the month lengths and leap month of a Chinese lunar year.
|
||
// The table covers 1900-2100; 1900-01-31 is lunar 1900-01-01.
|
||
constexpr std::array<unsigned, 201> kLunarYearInfo = {
|
||
0x04bd8, 0x04ae0, 0x0a570, 0x054d5, 0x0d260, 0x0d950, 0x16554, 0x056a0, 0x09ad0, 0x055d2, 0x04ae0, 0x0a5b6, 0x0a4d0,
|
||
0x0d250, 0x1d255, 0x0b540, 0x0d6a0, 0x0ada2, 0x095b0, 0x14977, 0x04970, 0x0a4b0, 0x0b4b5, 0x06a50, 0x06d40, 0x1ab54,
|
||
0x02b60, 0x09570, 0x052f2, 0x04970, 0x06566, 0x0d4a0, 0x0ea50, 0x06e95, 0x05ad0, 0x02b60, 0x186e3, 0x092e0, 0x1c8d7,
|
||
0x0c950, 0x0d4a0, 0x1d8a6, 0x0b550, 0x056a0, 0x1a5b4, 0x025d0, 0x092d0, 0x0d2b2, 0x0a950, 0x0b557, 0x06ca0, 0x0b550,
|
||
0x15355, 0x04da0, 0x0a5b0, 0x14573, 0x052b0, 0x0a9a8, 0x0e950, 0x06aa0, 0x0aea6, 0x0ab50, 0x04b60, 0x0aae4, 0x0a570,
|
||
0x05260, 0x0f263, 0x0d950, 0x05b57, 0x056a0, 0x096d0, 0x04dd5, 0x04ad0, 0x0a4d0, 0x0d4d4, 0x0d250, 0x0d558, 0x0b540,
|
||
0x0b6a0, 0x195a6, 0x095b0, 0x049b0, 0x0a974, 0x0a4b0, 0x0b27a, 0x06a50, 0x06d40, 0x0af46, 0x0ab60, 0x09570, 0x04af5,
|
||
0x04970, 0x064b0, 0x074a3, 0x0ea50, 0x06b58, 0x055c0, 0x0ab60, 0x096d5, 0x092e0, 0x0c960, 0x0d954, 0x0d4a0, 0x0da50,
|
||
0x07552, 0x056a0, 0x0abb7, 0x025d0, 0x092d0, 0x0cab5, 0x0a950, 0x0b4a0, 0x0baa4, 0x0ad50, 0x055d9, 0x04ba0, 0x0a5b0,
|
||
0x15176, 0x052b0, 0x0a930, 0x07954, 0x06aa0, 0x0ad50, 0x05b52, 0x04b60, 0x0a6e6, 0x0a4e0, 0x0d260, 0x0ea65, 0x0d530,
|
||
0x05aa0, 0x076a3, 0x096d0, 0x04bd7, 0x04ad0, 0x0a4d0, 0x1d0b6, 0x0d250, 0x0d520, 0x0dd45, 0x0b5a0, 0x056d0, 0x055b2,
|
||
0x049b0, 0x0a577, 0x0a4b0, 0x0aa50, 0x1b255, 0x06d20, 0x0ada0, 0x14b63, 0x09370, 0x049f8, 0x04970, 0x064b0, 0x168a6,
|
||
0x0ea50, 0x06b20, 0x1a6c4, 0x0aae0, 0x092e0, 0x0d2e3, 0x0c960, 0x0d557, 0x0d4a0, 0x0da50, 0x05d55, 0x056a0, 0x0a6d0,
|
||
0x055d4, 0x052d0, 0x0a9b8, 0x0a950, 0x0b4a0, 0x0b6a6, 0x0ad50, 0x055a0, 0x0aba4, 0x0a5b0, 0x052b0, 0x0b273, 0x06930,
|
||
0x07337, 0x06aa0, 0x0ad50, 0x14b55, 0x04b60, 0x0a570, 0x054e4, 0x0d260, 0x0e968, 0x0d520, 0x0daa0, 0x16aa6, 0x056d0,
|
||
0x04ae0, 0x0a9d4, 0x0a4d0, 0x0d150, 0x0f252, 0x0d520,
|
||
};
|
||
|
||
unsigned lunar_month_days(unsigned year, unsigned month)
|
||
{
|
||
return (kLunarYearInfo[year - 1900] & (0x10000U >> month)) ? 30 : 29;
|
||
}
|
||
|
||
unsigned lunar_leap_month(unsigned year)
|
||
{
|
||
return kLunarYearInfo[year - 1900] & 0x0fU;
|
||
}
|
||
|
||
unsigned lunar_leap_days(unsigned year)
|
||
{
|
||
return lunar_leap_month(year) == 0 ? 0 : ((kLunarYearInfo[year - 1900] & 0x10000U) ? 30 : 29);
|
||
}
|
||
|
||
unsigned lunar_year_days(unsigned year)
|
||
{
|
||
unsigned days = 348 + lunar_leap_days(year);
|
||
for (unsigned mask = 0x8000; mask > 0x8; mask >>= 1)
|
||
{
|
||
if ((kLunarYearInfo[year - 1900] & mask) != 0)
|
||
{
|
||
++days;
|
||
}
|
||
}
|
||
return days;
|
||
}
|
||
|
||
bool is_leap_year(unsigned year)
|
||
{
|
||
return year % 4 == 0 && (year % 100 != 0 || year % 400 == 0);
|
||
}
|
||
|
||
unsigned days_in_month(unsigned year, unsigned month)
|
||
{
|
||
constexpr unsigned kDays[] = {0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
|
||
if (month == 0 || month > 12)
|
||
{
|
||
return 0;
|
||
}
|
||
return month == 2 && is_leap_year(year) ? 29 : kDays[month];
|
||
}
|
||
|
||
std::int64_t civil_day_number(int year, unsigned month, unsigned day)
|
||
{
|
||
year -= month <= 2;
|
||
const int era = (year >= 0 ? year : year - 399) / 400;
|
||
const unsigned year_of_era = static_cast<unsigned>(year - era * 400);
|
||
const unsigned day_of_year = (153 * (month + (month > 2 ? -3 : 9)) + 2) / 5 + day - 1;
|
||
const unsigned day_of_era = year_of_era * 365 + year_of_era / 4 - year_of_era / 100 + day_of_year;
|
||
return static_cast<std::int64_t>(era) * 146097 + day_of_era - 719468;
|
||
}
|
||
|
||
bool days_since_lunar_epoch(const LocalDateTime &date, unsigned &days)
|
||
{
|
||
if (date.year == 0 || date.month == 0 || date.month > 12 || date.day == 0 ||
|
||
date.day > days_in_month(date.year, date.month))
|
||
{
|
||
return false;
|
||
}
|
||
const std::int64_t epoch = civil_day_number(1900, 1, 31);
|
||
const std::int64_t current = civil_day_number(static_cast<int>(date.year), date.month, date.day);
|
||
if (current < epoch)
|
||
{
|
||
return false;
|
||
}
|
||
days = static_cast<unsigned>(current - epoch);
|
||
return true;
|
||
}
|
||
|
||
std::string lunar_date(const LocalDateTime &now)
|
||
{
|
||
unsigned remaining = 0;
|
||
if (!days_since_lunar_epoch(now, remaining))
|
||
{
|
||
return {};
|
||
}
|
||
|
||
unsigned year = 1900;
|
||
while (year <= 2100)
|
||
{
|
||
const unsigned year_days = lunar_year_days(year);
|
||
if (remaining < year_days)
|
||
{
|
||
break;
|
||
}
|
||
remaining -= year_days;
|
||
++year;
|
||
}
|
||
if (year > 2100)
|
||
{
|
||
return {};
|
||
}
|
||
|
||
unsigned month = 1;
|
||
bool is_leap = false;
|
||
for (; month <= 12; ++month)
|
||
{
|
||
const unsigned month_days = lunar_month_days(year, month);
|
||
if (remaining < month_days)
|
||
{
|
||
break;
|
||
}
|
||
remaining -= month_days;
|
||
|
||
if (lunar_leap_month(year) == month)
|
||
{
|
||
const unsigned leap_days = lunar_leap_days(year);
|
||
if (remaining < leap_days)
|
||
{
|
||
is_leap = true;
|
||
break;
|
||
}
|
||
remaining -= leap_days;
|
||
}
|
||
}
|
||
|
||
static constexpr const char *kStems[] = {"甲", "乙", "丙", "丁", "戊", "己", "庚", "辛", "壬", "癸"};
|
||
static constexpr const char *kBranches[] = {"子", "丑", "寅", "卯", "辰", "巳", "午", "未", "申", "酉", "戌", "亥"};
|
||
return std::string(kStems[(year - 4) % 10]) + kBranches[(year - 4) % 12] + "年" + (is_leap ? "闰" : "") +
|
||
chinese_number(month) + "月" + chinese_number(remaining + 1) + "日";
|
||
}
|
||
|
||
std::vector<std::string> date_candidates(const LocalDateTime &now)
|
||
{
|
||
const unsigned weekday = weekday_index(now);
|
||
return {
|
||
format("%u年%u月%u日", now.year, now.month, now.day),
|
||
format("%04u-%02u-%02u", now.year, now.month, now.day),
|
||
format("%04u/%02u/%02u", now.year, now.month, now.day),
|
||
format("%04u.%02u.%02u", now.year, now.month, now.day),
|
||
format("%04u%02u%02u", now.year, now.month, now.day),
|
||
format("%02u年%u月%u日", now.year % 100, now.month, now.day),
|
||
format("%u月%u日", now.month, now.day),
|
||
format("%02u-%02u", now.month, now.day),
|
||
format("%02u%02u", now.month, now.day),
|
||
format("%u年%u月%u日 ", now.year, now.month, now.day) + kWeekdays[weekday],
|
||
format("%u月%u日 ", now.month, now.day) + kShortWeekdays[weekday],
|
||
format("%04u-%02u-%02u ", now.year, now.month, now.day) + kEnglishWeekdays[weekday],
|
||
format("%04u-%02u-%02u ", now.year, now.month, now.day) + format("%02u:%02u", now.hour, now.minute),
|
||
format("%u月%u日 ", now.month, now.day) + format("%02u:%02u", now.hour, now.minute),
|
||
chinese_digits(now.year) + "年" + chinese_number(now.month) + "月" + chinese_number(now.day) + "日",
|
||
financial_digits(now.year) + "年" + financial_digits(now.month) + "月" + financial_digits(now.day, 2) + "日",
|
||
lunar_date(now),
|
||
};
|
||
}
|
||
|
||
std::vector<std::string> time_candidates(const LocalDateTime &now)
|
||
{
|
||
const unsigned hour12 = now.hour % 12 == 0 ? 12 : now.hour % 12;
|
||
const std::string period = now.hour < 12 ? "上午" : "下午";
|
||
const std::string meridiem_upper = now.hour < 12 ? "AM" : "PM";
|
||
const std::string meridiem_lower = now.hour < 12 ? "am" : "pm";
|
||
const std::string colloquial_hour = hour12 == 2 ? "两" : chinese_number(hour12);
|
||
const std::string colloquial_minutes =
|
||
now.minute == 0 ? "" : (now.minute == 30 ? "半" : chinese_number(now.minute) + "分");
|
||
return {
|
||
format("%02u:%02u", now.hour, now.minute),
|
||
format("%02u:%02u:%02u", now.hour, now.minute, now.second),
|
||
format("%02u%02u", now.hour, now.minute),
|
||
format("%02u%02u%02u", now.hour, now.minute, now.second),
|
||
period + format("%u:%02u", hour12, now.minute),
|
||
period + format("%u点%02u分", hour12, now.minute),
|
||
period + colloquial_hour + "点" + colloquial_minutes,
|
||
format("%u:%02u ", hour12, now.minute) + meridiem_upper,
|
||
format("%u:%02u", hour12, now.minute) + meridiem_lower,
|
||
format("%02u:%02u:%02u ", hour12, now.minute, now.second) + meridiem_upper,
|
||
format("%04u-%02u-%02u ", now.year, now.month, now.day) +
|
||
format("%02u:%02u:%02u", now.hour, now.minute, now.second),
|
||
format("%u年%u月%u日 ", now.year, now.month, now.day) + format("%02u:%02u", now.hour, now.minute),
|
||
format("%u月%u日 ", now.month, now.day) + period + format("%u:%02u", hour12, now.minute),
|
||
};
|
||
}
|
||
|
||
std::vector<std::string> week_candidates(const LocalDateTime &now)
|
||
{
|
||
const unsigned weekday = weekday_index(now);
|
||
std::vector<std::string> results = {kWeekdays[weekday]};
|
||
if (weekday == 0)
|
||
{
|
||
results.emplace_back("星期天");
|
||
}
|
||
results.emplace_back(kEnglishFullWeekdays[weekday]);
|
||
results.emplace_back(kEnglishWeekdays[weekday]);
|
||
return results;
|
||
}
|
||
} // namespace
|
||
|
||
LocalDateTime current_local_date_time()
|
||
{
|
||
const std::time_t current = std::time(nullptr);
|
||
std::tm local = {};
|
||
#ifdef _WIN32
|
||
if (current == static_cast<std::time_t>(-1) || localtime_s(&local, ¤t) != 0)
|
||
#else
|
||
if (current == static_cast<std::time_t>(-1) || localtime_r(¤t, &local) == nullptr)
|
||
#endif
|
||
{
|
||
return {};
|
||
}
|
||
return {static_cast<unsigned>(local.tm_year + 1900), static_cast<unsigned>(local.tm_mon + 1),
|
||
static_cast<unsigned>(local.tm_mday), static_cast<unsigned>(local.tm_wday),
|
||
static_cast<unsigned>(local.tm_hour), static_cast<unsigned>(local.tm_min),
|
||
static_cast<unsigned>(local.tm_sec)};
|
||
}
|
||
|
||
bool is_date_time_keyword(const std::string &keyword)
|
||
{
|
||
return is_date_keyword(keyword) || is_time_keyword(keyword) || is_week_keyword(keyword);
|
||
}
|
||
|
||
std::vector<WordItem> query_date_time(const std::string &keyword, const LocalDateTime *now, int limit)
|
||
{
|
||
if (limit <= 0 || !is_date_time_keyword(keyword))
|
||
{
|
||
return {};
|
||
}
|
||
|
||
const LocalDateTime current = now == nullptr ? current_local_date_time() : *now;
|
||
const std::vector<std::string> candidates =
|
||
is_date_keyword(keyword) ? date_candidates(current)
|
||
: (is_time_keyword(keyword) ? time_candidates(current) : week_candidates(current));
|
||
const std::size_t result_count = std::min(candidates.size(), static_cast<std::size_t>(limit));
|
||
std::vector<WordItem> results;
|
||
results.reserve(result_count);
|
||
for (std::size_t index = 0; index < result_count; ++index)
|
||
{
|
||
results.emplace_back("", candidates[index], static_cast<std::int64_t>(result_count - index),
|
||
CandidateSource::Generated);
|
||
}
|
||
return results;
|
||
}
|
||
} // namespace metasequoia::local_modes
|