GenerateUnicodeData.cpp 41 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066
  1. /*
  2. * Copyright (c) 2021, Tim Flynn <trflynn89@pm.me>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include <AK/AllOf.h>
  7. #include <AK/Array.h>
  8. #include <AK/CharacterTypes.h>
  9. #include <AK/Find.h>
  10. #include <AK/HashMap.h>
  11. #include <AK/Optional.h>
  12. #include <AK/QuickSort.h>
  13. #include <AK/SourceGenerator.h>
  14. #include <AK/String.h>
  15. #include <AK/StringUtils.h>
  16. #include <AK/Types.h>
  17. #include <AK/Vector.h>
  18. #include <LibCore/ArgsParser.h>
  19. #include <LibCore/File.h>
  20. // Some code points are excluded from UnicodeData.txt, and instead are part of a "range" of code
  21. // points, as indicated by the "name" field. For example:
  22. // 3400;<CJK Ideograph Extension A, First>;Lo;0;L;;;;;N;;;;;
  23. // 4DBF;<CJK Ideograph Extension A, Last>;Lo;0;L;;;;;N;;;;;
  24. struct CodePointRange {
  25. u32 first;
  26. u32 last;
  27. };
  28. // SpecialCasing source: https://www.unicode.org/Public/13.0.0/ucd/SpecialCasing.txt
  29. // Field descriptions: https://www.unicode.org/reports/tr44/tr44-13.html#SpecialCasing.txt
  30. struct SpecialCasing {
  31. u32 index { 0 };
  32. u32 code_point { 0 };
  33. Vector<u32> lowercase_mapping;
  34. Vector<u32> uppercase_mapping;
  35. Vector<u32> titlecase_mapping;
  36. String locale;
  37. String condition;
  38. };
  39. // PropList source: https://www.unicode.org/Public/13.0.0/ucd/PropList.txt
  40. // Property descriptions: https://www.unicode.org/reports/tr44/tr44-13.html#PropList.txt
  41. using PropList = HashMap<String, Vector<CodePointRange>>;
  42. // PropertyAliases source: https://www.unicode.org/Public/13.0.0/ucd/PropertyAliases.txt
  43. struct Alias {
  44. String property;
  45. String alias;
  46. };
  47. // Normalization source: https://www.unicode.org/Public/13.0.0/ucd/DerivedNormalizationProps.txt
  48. // Normalization descriptions: https://www.unicode.org/reports/tr44/#DerivedNormalizationProps.txt
  49. enum class QuickCheck {
  50. Yes,
  51. No,
  52. Maybe,
  53. };
  54. struct Normalization {
  55. CodePointRange code_point_range;
  56. Vector<u32> value;
  57. QuickCheck quick_check { QuickCheck::Yes };
  58. };
  59. using NormalizationProps = HashMap<String, Vector<Normalization>>;
  60. // UnicodeData source: https://www.unicode.org/Public/13.0.0/ucd/UnicodeData.txt
  61. // Field descriptions: https://www.unicode.org/reports/tr44/tr44-13.html#UnicodeData.txt
  62. // https://www.unicode.org/reports/tr44/#General_Category_Values
  63. struct CodePointData {
  64. u32 code_point { 0 };
  65. String name;
  66. u8 canonical_combining_class { 0 };
  67. String bidi_class;
  68. String decomposition_type;
  69. Optional<i8> numeric_value_decimal;
  70. Optional<i8> numeric_value_digit;
  71. Optional<i8> numeric_value_numeric;
  72. bool bidi_mirrored { false };
  73. String unicode_1_name;
  74. String iso_comment;
  75. Optional<u32> simple_uppercase_mapping;
  76. Optional<u32> simple_lowercase_mapping;
  77. Optional<u32> simple_titlecase_mapping;
  78. Vector<u32> special_casing_indices;
  79. };
  80. struct UnicodeData {
  81. Vector<SpecialCasing> special_casing;
  82. u32 largest_casing_transform_size { 0 };
  83. u32 largest_special_casing_size { 0 };
  84. Vector<String> conditions;
  85. Vector<CodePointData> code_point_data;
  86. Vector<CodePointRange> code_point_ranges;
  87. PropList general_categories;
  88. Vector<Alias> general_category_aliases;
  89. // The Unicode standard defines additional properties (Any, Assigned, ASCII) which are not in
  90. // any UCD file. Assigned code point ranges are derived as this generator is executed.
  91. // https://unicode.org/reports/tr18/#General_Category_Property
  92. PropList prop_list {
  93. { "Any"sv, { { 0, 0x10ffff } } },
  94. { "Assigned"sv, {} },
  95. { "ASCII"sv, { { 0, 0x7f } } },
  96. };
  97. Vector<Alias> prop_aliases;
  98. PropList script_list {
  99. { "Unknown"sv, {} },
  100. };
  101. Vector<Alias> script_aliases;
  102. PropList script_extensions;
  103. // FIXME: We are not yet doing anything with this data. It will be needed for String.prototype.normalize.
  104. NormalizationProps normalization_props;
  105. };
  106. static constexpr auto s_desired_fields = Array {
  107. "simple_uppercase_mapping"sv,
  108. "simple_lowercase_mapping"sv,
  109. };
  110. static void write_to_file_if_different(Core::File& file, StringView contents)
  111. {
  112. auto const current_contents = file.read_all();
  113. if (StringView { current_contents.bytes() } == contents)
  114. return;
  115. VERIFY(file.seek(0));
  116. VERIFY(file.truncate(0));
  117. VERIFY(file.write(contents));
  118. }
  119. static Vector<u32> parse_code_point_list(StringView const& list)
  120. {
  121. Vector<u32> code_points;
  122. auto segments = list.split_view(' ');
  123. for (auto const& code_point : segments)
  124. code_points.append(AK::StringUtils::convert_to_uint_from_hex<u32>(code_point).value());
  125. return code_points;
  126. }
  127. static CodePointRange parse_code_point_range(StringView const& list)
  128. {
  129. CodePointRange code_point_range {};
  130. if (list.contains(".."sv)) {
  131. auto segments = list.split_view(".."sv);
  132. VERIFY(segments.size() == 2);
  133. auto begin = AK::StringUtils::convert_to_uint_from_hex<u32>(segments[0]).value();
  134. auto end = AK::StringUtils::convert_to_uint_from_hex<u32>(segments[1]).value();
  135. code_point_range = { begin, end };
  136. } else {
  137. auto code_point = AK::StringUtils::convert_to_uint_from_hex<u32>(list).value();
  138. code_point_range = { code_point, code_point };
  139. }
  140. return code_point_range;
  141. }
  142. static void parse_special_casing(Core::File& file, UnicodeData& unicode_data)
  143. {
  144. while (file.can_read_line()) {
  145. auto line = file.read_line();
  146. if (line.is_empty() || line.starts_with('#'))
  147. continue;
  148. if (auto index = line.find('#'); index.has_value())
  149. line = line.substring(0, *index);
  150. auto segments = line.split(';', true);
  151. VERIFY(segments.size() == 5 || segments.size() == 6);
  152. SpecialCasing casing {};
  153. casing.index = static_cast<u32>(unicode_data.special_casing.size());
  154. casing.code_point = AK::StringUtils::convert_to_uint_from_hex<u32>(segments[0]).value();
  155. casing.lowercase_mapping = parse_code_point_list(segments[1]);
  156. casing.titlecase_mapping = parse_code_point_list(segments[2]);
  157. casing.uppercase_mapping = parse_code_point_list(segments[3]);
  158. if (auto condition = segments[4].trim_whitespace(); !condition.is_empty()) {
  159. auto conditions = condition.split(' ', true);
  160. VERIFY(conditions.size() == 1 || conditions.size() == 2);
  161. if (conditions.size() == 2) {
  162. casing.locale = move(conditions[0]);
  163. casing.condition = move(conditions[1]);
  164. } else if (all_of(conditions[0], is_ascii_lower_alpha)) {
  165. casing.locale = move(conditions[0]);
  166. } else {
  167. casing.condition = move(conditions[0]);
  168. }
  169. if (!casing.locale.is_empty())
  170. casing.locale = String::formatted("{:c}{}", to_ascii_uppercase(casing.locale[0]), casing.locale.substring_view(1));
  171. casing.condition.replace("_", "", true);
  172. if (!casing.condition.is_empty() && !unicode_data.conditions.contains_slow(casing.condition))
  173. unicode_data.conditions.append(casing.condition);
  174. }
  175. unicode_data.largest_casing_transform_size = max(unicode_data.largest_casing_transform_size, casing.lowercase_mapping.size());
  176. unicode_data.largest_casing_transform_size = max(unicode_data.largest_casing_transform_size, casing.titlecase_mapping.size());
  177. unicode_data.largest_casing_transform_size = max(unicode_data.largest_casing_transform_size, casing.uppercase_mapping.size());
  178. unicode_data.special_casing.append(move(casing));
  179. }
  180. }
  181. static void parse_prop_list(Core::File& file, PropList& prop_list, bool multi_value_property = false)
  182. {
  183. while (file.can_read_line()) {
  184. auto line = file.read_line();
  185. if (line.is_empty() || line.starts_with('#'))
  186. continue;
  187. if (auto index = line.find('#'); index.has_value())
  188. line = line.substring(0, *index);
  189. auto segments = line.split_view(';', true);
  190. VERIFY(segments.size() == 2);
  191. auto code_point_range = parse_code_point_range(segments[0].trim_whitespace());
  192. Vector<StringView> properties;
  193. if (multi_value_property)
  194. properties = segments[1].trim_whitespace().split_view(' ');
  195. else
  196. properties = { segments[1].trim_whitespace() };
  197. for (auto const& property : properties) {
  198. auto& code_points = prop_list.ensure(property.trim_whitespace());
  199. code_points.append(code_point_range);
  200. }
  201. }
  202. }
  203. static void parse_alias_list(Core::File& file, PropList const& prop_list, Vector<Alias>& prop_aliases)
  204. {
  205. String current_property;
  206. auto append_alias = [&](auto alias, auto property) {
  207. // Note: The alias files contain lines such as "Hyphen = Hyphen", which we should just skip.
  208. if (alias == property)
  209. return;
  210. // FIXME: We will, eventually, need to find where missing properties are located and parse them.
  211. if (!prop_list.contains(property))
  212. return;
  213. prop_aliases.append({ property, alias });
  214. };
  215. while (file.can_read_line()) {
  216. auto line = file.read_line();
  217. if (line.is_empty() || line.starts_with('#')) {
  218. if (line.ends_with("Properties"sv))
  219. current_property = line.substring(2);
  220. continue;
  221. }
  222. // Note: For now, we only care about Binary Property aliases for Unicode property escapes.
  223. if (current_property != "Binary Properties"sv)
  224. continue;
  225. auto segments = line.split_view(';', true);
  226. VERIFY((segments.size() == 2) || (segments.size() == 3));
  227. auto alias = segments[0].trim_whitespace();
  228. auto property = segments[1].trim_whitespace();
  229. append_alias(alias, property);
  230. if (segments.size() == 3) {
  231. alias = segments[2].trim_whitespace();
  232. append_alias(alias, property);
  233. }
  234. }
  235. }
  236. static void parse_value_alias_list(Core::File& file, StringView desired_category, Vector<String> const& value_list, Vector<Alias>& prop_aliases, bool primary_value_is_first = true)
  237. {
  238. VERIFY(file.seek(0));
  239. auto append_alias = [&](auto alias, auto value) {
  240. // Note: The value alias file contains lines such as "Ahom = Ahom", which we should just skip.
  241. if (alias == value)
  242. return;
  243. // FIXME: We will, eventually, need to find where missing properties are located and parse them.
  244. if (!value_list.contains_slow(value))
  245. return;
  246. prop_aliases.append({ value, alias });
  247. };
  248. while (file.can_read_line()) {
  249. auto line = file.read_line();
  250. if (line.is_empty() || line.starts_with('#'))
  251. continue;
  252. if (auto index = line.find('#'); index.has_value())
  253. line = line.substring(0, *index);
  254. auto segments = line.split_view(';', true);
  255. auto category = segments[0].trim_whitespace();
  256. if (category != desired_category)
  257. continue;
  258. VERIFY((segments.size() == 3) || (segments.size() == 4));
  259. auto value = primary_value_is_first ? segments[1].trim_whitespace() : segments[2].trim_whitespace();
  260. auto alias = primary_value_is_first ? segments[2].trim_whitespace() : segments[1].trim_whitespace();
  261. append_alias(alias, value);
  262. if (segments.size() == 4) {
  263. alias = segments[3].trim_whitespace();
  264. append_alias(alias, value);
  265. }
  266. }
  267. }
  268. static void parse_normalization_props(Core::File& file, UnicodeData& unicode_data)
  269. {
  270. while (file.can_read_line()) {
  271. auto line = file.read_line();
  272. if (line.is_empty() || line.starts_with('#'))
  273. continue;
  274. if (auto index = line.find('#'); index.has_value())
  275. line = line.substring(0, *index);
  276. auto segments = line.split_view(';', true);
  277. VERIFY((segments.size() == 2) || (segments.size() == 3));
  278. auto code_point_range = parse_code_point_range(segments[0].trim_whitespace());
  279. auto property = segments[1].trim_whitespace().to_string();
  280. Vector<u32> value;
  281. QuickCheck quick_check = QuickCheck::Yes;
  282. if (segments.size() == 3) {
  283. auto value_or_quick_check = segments[2].trim_whitespace();
  284. if ((value_or_quick_check == "N"sv))
  285. quick_check = QuickCheck::No;
  286. else if ((value_or_quick_check == "M"sv))
  287. quick_check = QuickCheck::Maybe;
  288. else
  289. value = parse_code_point_list(value_or_quick_check);
  290. }
  291. auto& normalizations = unicode_data.normalization_props.ensure(property);
  292. normalizations.append({ code_point_range, move(value), quick_check });
  293. auto& prop_list = unicode_data.prop_list.ensure(property);
  294. prop_list.append(move(code_point_range));
  295. }
  296. }
  297. static void parse_unicode_data(Core::File& file, UnicodeData& unicode_data)
  298. {
  299. Optional<u32> code_point_range_start;
  300. auto& assigned_code_points = unicode_data.prop_list.find("Assigned"sv)->value;
  301. Optional<u32> assigned_code_point_range_start = 0;
  302. u32 previous_code_point = 0;
  303. while (file.can_read_line()) {
  304. auto line = file.read_line();
  305. if (line.is_empty())
  306. continue;
  307. auto segments = line.split(';', true);
  308. VERIFY(segments.size() == 15);
  309. CodePointData data {};
  310. data.code_point = AK::StringUtils::convert_to_uint_from_hex<u32>(segments[0]).value();
  311. data.name = move(segments[1]);
  312. data.canonical_combining_class = AK::StringUtils::convert_to_uint<u8>(segments[3]).value();
  313. data.bidi_class = move(segments[4]);
  314. data.decomposition_type = move(segments[5]);
  315. data.numeric_value_decimal = AK::StringUtils::convert_to_int<i8>(segments[6]);
  316. data.numeric_value_digit = AK::StringUtils::convert_to_int<i8>(segments[7]);
  317. data.numeric_value_numeric = AK::StringUtils::convert_to_int<i8>(segments[8]);
  318. data.bidi_mirrored = segments[9] == "Y"sv;
  319. data.unicode_1_name = move(segments[10]);
  320. data.iso_comment = move(segments[11]);
  321. data.simple_uppercase_mapping = AK::StringUtils::convert_to_uint_from_hex<u32>(segments[12]);
  322. data.simple_lowercase_mapping = AK::StringUtils::convert_to_uint_from_hex<u32>(segments[13]);
  323. data.simple_titlecase_mapping = AK::StringUtils::convert_to_uint_from_hex<u32>(segments[14]);
  324. if (!assigned_code_point_range_start.has_value())
  325. assigned_code_point_range_start = data.code_point;
  326. if (data.name.starts_with("<"sv) && data.name.ends_with(", First>")) {
  327. VERIFY(!code_point_range_start.has_value() && assigned_code_point_range_start.has_value());
  328. code_point_range_start = data.code_point;
  329. data.name = data.name.substring(1, data.name.length() - 9);
  330. assigned_code_points.append({ *assigned_code_point_range_start, previous_code_point });
  331. assigned_code_point_range_start.clear();
  332. } else if (data.name.starts_with("<"sv) && data.name.ends_with(", Last>")) {
  333. VERIFY(code_point_range_start.has_value());
  334. CodePointRange code_point_range { *code_point_range_start, data.code_point };
  335. unicode_data.code_point_ranges.append(code_point_range);
  336. assigned_code_points.append(code_point_range);
  337. data.name = data.name.substring(1, data.name.length() - 8);
  338. code_point_range_start.clear();
  339. } else if ((data.code_point > 0) && (data.code_point - previous_code_point) != 1) {
  340. VERIFY(assigned_code_point_range_start.has_value());
  341. assigned_code_points.append({ *assigned_code_point_range_start, previous_code_point });
  342. assigned_code_point_range_start = data.code_point;
  343. }
  344. for (auto const& casing : unicode_data.special_casing) {
  345. if (casing.code_point == data.code_point)
  346. data.special_casing_indices.append(casing.index);
  347. }
  348. unicode_data.largest_special_casing_size = max(unicode_data.largest_special_casing_size, data.special_casing_indices.size());
  349. previous_code_point = data.code_point;
  350. unicode_data.code_point_data.append(move(data));
  351. }
  352. }
  353. static void generate_unicode_data_header(Core::File& file, UnicodeData& unicode_data)
  354. {
  355. StringBuilder builder;
  356. SourceGenerator generator { builder };
  357. generator.set("casing_transform_size", String::number(unicode_data.largest_casing_transform_size));
  358. generator.set("special_casing_size", String::number(unicode_data.largest_special_casing_size));
  359. auto generate_enum = [&](StringView name, StringView default_, Vector<String> values, Vector<Alias> aliases = {}) {
  360. quick_sort(values);
  361. quick_sort(aliases, [](auto& alias1, auto& alias2) { return alias1.alias < alias2.alias; });
  362. generator.set("name", name);
  363. generator.set("underlying", String::formatted("{}UnderlyingType", name));
  364. generator.append(R"~~~(
  365. using @underlying@ = u8;
  366. enum class @name@ : @underlying@ {)~~~");
  367. if (!default_.is_empty()) {
  368. generator.set("default", default_);
  369. generator.append(R"~~~(
  370. @default@,)~~~");
  371. }
  372. for (auto const& value : values) {
  373. generator.set("value", value);
  374. generator.append(R"~~~(
  375. @value@,)~~~");
  376. }
  377. for (auto const& alias : aliases) {
  378. generator.set("alias", alias.alias);
  379. generator.set("value", alias.property);
  380. generator.append(R"~~~(
  381. @alias@ = @value@,)~~~");
  382. }
  383. generator.append(R"~~~(
  384. };
  385. )~~~");
  386. };
  387. generator.append(R"~~~(
  388. #pragma once
  389. #include <AK/Optional.h>
  390. #include <AK/Types.h>
  391. #include <LibUnicode/Forward.h>
  392. #include <LibUnicode/UnicodeLocale.h>
  393. namespace Unicode {
  394. )~~~");
  395. generate_enum("Condition"sv, "None"sv, move(unicode_data.conditions));
  396. generate_enum("GeneralCategory"sv, {}, unicode_data.general_categories.keys(), unicode_data.general_category_aliases);
  397. generate_enum("Property"sv, {}, unicode_data.prop_list.keys(), unicode_data.prop_aliases);
  398. generate_enum("Script"sv, {}, unicode_data.script_list.keys(), unicode_data.script_aliases);
  399. generator.append(R"~~~(
  400. struct SpecialCasing {
  401. u32 code_point { 0 };
  402. u32 lowercase_mapping[@casing_transform_size@];
  403. u32 lowercase_mapping_size { 0 };
  404. u32 uppercase_mapping[@casing_transform_size@];
  405. u32 uppercase_mapping_size { 0 };
  406. u32 titlecase_mapping[@casing_transform_size@];
  407. u32 titlecase_mapping_size { 0 };
  408. Locale locale { Locale::None };
  409. Condition condition { Condition::None };
  410. };
  411. struct UnicodeData {
  412. u32 code_point;)~~~");
  413. auto append_field = [&](StringView type, StringView name) {
  414. if (!s_desired_fields.span().contains_slow(name))
  415. return;
  416. generator.set("type", type);
  417. generator.set("name", name);
  418. generator.append(R"~~~(
  419. @type@ @name@;)~~~");
  420. };
  421. // Note: For compile-time performance, only primitive types are used.
  422. append_field("char const*"sv, "name"sv);
  423. append_field("u8"sv, "canonical_combining_class"sv);
  424. append_field("char const*"sv, "bidi_class"sv);
  425. append_field("char const*"sv, "decomposition_type"sv);
  426. append_field("i8"sv, "numeric_value_decimal"sv);
  427. append_field("i8"sv, "numeric_value_digit"sv);
  428. append_field("i8"sv, "numeric_value_numeric"sv);
  429. append_field("bool"sv, "bidi_mirrored"sv);
  430. append_field("char const*"sv, "unicode_1_name"sv);
  431. append_field("char const*"sv, "iso_comment"sv);
  432. append_field("u32"sv, "simple_uppercase_mapping"sv);
  433. append_field("u32"sv, "simple_lowercase_mapping"sv);
  434. append_field("u32"sv, "simple_titlecase_mapping"sv);
  435. generator.append(R"~~~(
  436. SpecialCasing const* special_casing[@special_casing_size@] {};
  437. u32 special_casing_size { 0 };
  438. };
  439. namespace Detail {
  440. Optional<UnicodeData> unicode_data_for_code_point(u32 code_point);
  441. bool code_point_has_general_category(u32 code_point, GeneralCategory general_category);
  442. Optional<GeneralCategory> general_category_from_string(StringView const& general_category);
  443. bool code_point_has_property(u32 code_point, Property property);
  444. Optional<Property> property_from_string(StringView const& property);
  445. bool code_point_has_script(u32 code_point, Script script);
  446. bool code_point_has_script_extension(u32 code_point, Script script);
  447. Optional<Script> script_from_string(StringView const& script);
  448. }
  449. }
  450. )~~~");
  451. write_to_file_if_different(file, generator.as_string_view());
  452. }
  453. static void generate_unicode_data_implementation(Core::File& file, UnicodeData const& unicode_data)
  454. {
  455. StringBuilder builder;
  456. SourceGenerator generator { builder };
  457. generator.set("special_casing_size", String::number(unicode_data.special_casing.size()));
  458. generator.set("code_point_data_size", String::number(unicode_data.code_point_data.size()));
  459. generator.append(R"~~~(
  460. #include <AK/Array.h>
  461. #include <AK/BinarySearch.h>
  462. #include <AK/CharacterTypes.h>
  463. #include <AK/HashMap.h>
  464. #include <AK/String.h>
  465. #include <AK/StringView.h>
  466. #include <LibUnicode/UnicodeData.h>
  467. namespace Unicode {
  468. )~~~");
  469. auto append_list_and_size = [&](auto const& list, StringView format) {
  470. if (list.is_empty()) {
  471. generator.append(", {}, 0");
  472. return;
  473. }
  474. bool first = true;
  475. generator.append(", {");
  476. for (auto const& item : list) {
  477. generator.append(first ? " " : ", ");
  478. generator.append(String::formatted(format, item));
  479. first = false;
  480. }
  481. generator.append(String::formatted(" }}, {}", list.size()));
  482. };
  483. generator.append(R"~~~(
  484. static constexpr Array<SpecialCasing, @special_casing_size@> s_special_casing { {)~~~");
  485. for (auto const& casing : unicode_data.special_casing) {
  486. generator.set("code_point", String::formatted("{:#x}", casing.code_point));
  487. generator.append(R"~~~(
  488. { @code_point@)~~~");
  489. constexpr auto format = "0x{:x}"sv;
  490. append_list_and_size(casing.lowercase_mapping, format);
  491. append_list_and_size(casing.uppercase_mapping, format);
  492. append_list_and_size(casing.titlecase_mapping, format);
  493. generator.set("locale", casing.locale.is_empty() ? "None" : casing.locale);
  494. generator.append(", Locale::@locale@");
  495. generator.set("condition", casing.condition.is_empty() ? "None" : casing.condition);
  496. generator.append(", Condition::@condition@");
  497. generator.append(" },");
  498. }
  499. generator.append(R"~~~(
  500. } };
  501. static constexpr Array<UnicodeData, @code_point_data_size@> s_unicode_data { {)~~~");
  502. auto append_field = [&](StringView name, String value) {
  503. if (!s_desired_fields.span().contains_slow(name))
  504. return;
  505. generator.set("value", move(value));
  506. generator.append(", @value@");
  507. };
  508. for (auto const& data : unicode_data.code_point_data) {
  509. generator.set("code_point", String::formatted("{:#x}", data.code_point));
  510. generator.append(R"~~~(
  511. { @code_point@)~~~");
  512. append_field("name", String::formatted("\"{}\"", data.name));
  513. append_field("canonical_combining_class", String::number(data.canonical_combining_class));
  514. append_field("bidi_class", String::formatted("\"{}\"", data.bidi_class));
  515. append_field("decomposition_type", String::formatted("\"{}\"", data.decomposition_type));
  516. append_field("numeric_value_decimal", String::number(data.numeric_value_decimal.value_or(-1)));
  517. append_field("numeric_value_digit", String::number(data.numeric_value_digit.value_or(-1)));
  518. append_field("numeric_value_numeric", String::number(data.numeric_value_numeric.value_or(-1)));
  519. append_field("bidi_mirrored", String::formatted("{}", data.bidi_mirrored));
  520. append_field("unicode_1_name", String::formatted("\"{}\"", data.unicode_1_name));
  521. append_field("iso_comment", String::formatted("\"{}\"", data.iso_comment));
  522. append_field("simple_uppercase_mapping", String::formatted("{:#x}", data.simple_uppercase_mapping.value_or(data.code_point)));
  523. append_field("simple_lowercase_mapping", String::formatted("{:#x}", data.simple_lowercase_mapping.value_or(data.code_point)));
  524. append_field("simple_titlecase_mapping", String::formatted("{:#x}", data.simple_titlecase_mapping.value_or(data.code_point)));
  525. append_list_and_size(data.special_casing_indices, "&s_special_casing[{}]"sv);
  526. generator.append(" },");
  527. }
  528. generator.append(R"~~~(
  529. } };
  530. struct CodePointRange {
  531. u32 first { 0 };
  532. u32 last { 0 };
  533. };
  534. struct CodePointRangeComparator {
  535. constexpr int operator()(u32 code_point, CodePointRange const& range)
  536. {
  537. return (code_point > range.last) - (code_point < range.first);
  538. }
  539. };
  540. )~~~");
  541. auto append_code_point_range_list = [&](String name, Vector<CodePointRange> const& ranges) {
  542. generator.set("name", name);
  543. generator.set("size", String::number(ranges.size()));
  544. generator.append(R"~~~(
  545. static constexpr Array<CodePointRange, @size@> @name@ { {
  546. )~~~");
  547. constexpr size_t max_ranges_per_row = 20;
  548. size_t ranges_in_current_row = 0;
  549. for (auto const& range : ranges) {
  550. if (ranges_in_current_row++ > 0)
  551. generator.append(" ");
  552. generator.set("first", String::formatted("{:#x}", range.first));
  553. generator.set("last", String::formatted("{:#x}", range.last));
  554. generator.append("{ @first@, @last@ },");
  555. if (ranges_in_current_row == max_ranges_per_row) {
  556. ranges_in_current_row = 0;
  557. generator.append("\n ");
  558. }
  559. }
  560. generator.append(R"~~~(
  561. } };
  562. )~~~");
  563. };
  564. auto append_prop_list = [&](StringView collection_name, StringView property_format, PropList const& property_list) {
  565. for (auto const& property : property_list) {
  566. auto name = String::formatted(property_format, property.key);
  567. append_code_point_range_list(move(name), property.value);
  568. }
  569. auto property_names = property_list.keys();
  570. quick_sort(property_names);
  571. generator.set("name", collection_name);
  572. generator.set("size", String::number(property_names.size()));
  573. generator.append(R"~~~(
  574. static constexpr Array<Span<CodePointRange const>, @size@> @name@ { {)~~~");
  575. for (auto const& property_name : property_names) {
  576. generator.set("name", String::formatted(property_format, property_name));
  577. generator.append(R"~~~(
  578. @name@.span(),)~~~");
  579. }
  580. generator.append(R"~~~(
  581. } };
  582. )~~~");
  583. };
  584. append_prop_list("s_general_categories"sv, "s_general_category_{}"sv, unicode_data.general_categories);
  585. append_prop_list("s_properties"sv, "s_property_{}"sv, unicode_data.prop_list);
  586. append_prop_list("s_scripts"sv, "s_script_{}"sv, unicode_data.script_list);
  587. append_prop_list("s_script_extensions"sv, "s_script_extension_{}"sv, unicode_data.script_extensions);
  588. generator.append(R"~~~(
  589. static HashMap<u32, UnicodeData const*> const& ensure_code_point_map()
  590. {
  591. static HashMap<u32, UnicodeData const*> code_point_to_data_map;
  592. code_point_to_data_map.ensure_capacity(s_unicode_data.size());
  593. for (auto const& unicode_data : s_unicode_data)
  594. code_point_to_data_map.set(unicode_data.code_point, &unicode_data);
  595. return code_point_to_data_map;
  596. }
  597. static Optional<u32> index_of_code_point_in_range(u32 code_point)
  598. {)~~~");
  599. for (auto const& range : unicode_data.code_point_ranges) {
  600. generator.set("first", String::formatted("{:#x}", range.first));
  601. generator.set("last", String::formatted("{:#x}", range.last));
  602. generator.append(R"~~~(
  603. if ((code_point > @first@) && (code_point < @last@))
  604. return @first@;)~~~");
  605. }
  606. generator.append(R"~~~(
  607. return {};
  608. }
  609. namespace Detail {
  610. Optional<UnicodeData> unicode_data_for_code_point(u32 code_point)
  611. {
  612. static auto const& code_point_to_data_map = ensure_code_point_map();
  613. VERIFY(is_unicode(code_point));
  614. if (auto data = code_point_to_data_map.get(code_point); data.has_value())
  615. return *(data.value());
  616. if (auto index = index_of_code_point_in_range(code_point); index.has_value()) {
  617. auto data_for_range = *(code_point_to_data_map.get(*index).value());
  618. data_for_range.simple_uppercase_mapping = code_point;
  619. data_for_range.simple_lowercase_mapping = code_point;
  620. return data_for_range;
  621. }
  622. return {};
  623. }
  624. )~~~");
  625. auto append_prop_search = [&](StringView enum_title, StringView enum_snake, StringView collection_name) {
  626. generator.set("enum_title", enum_title);
  627. generator.set("enum_snake", enum_snake);
  628. generator.set("collection_name", collection_name);
  629. generator.append(R"~~~(
  630. bool code_point_has_@enum_snake@(u32 code_point, @enum_title@ @enum_snake@)
  631. {
  632. auto index = static_cast<@enum_title@UnderlyingType>(@enum_snake@);
  633. auto const& ranges = @collection_name@.at(index);
  634. auto const* range = binary_search(ranges, code_point, nullptr, CodePointRangeComparator {});
  635. return range != nullptr;
  636. }
  637. )~~~");
  638. };
  639. auto append_from_string = [&](StringView enum_title, StringView enum_snake, PropList const& prop_list, Vector<Alias> const& aliases) {
  640. generator.set("enum_title", enum_title);
  641. generator.set("enum_snake", enum_snake);
  642. auto properties = prop_list.keys();
  643. for (auto const& alias : aliases)
  644. properties.append(alias.alias);
  645. quick_sort(properties);
  646. generator.append(R"~~~(
  647. Optional<@enum_title@> @enum_snake@_from_string(StringView const& @enum_snake@)
  648. {
  649. static HashMap<String, @enum_title@> @enum_snake@_values { {)~~~");
  650. for (auto const& property : properties) {
  651. generator.set("property", property);
  652. generator.append(R"~~~(
  653. { "@property@"sv, @enum_title@::@property@ },)~~~");
  654. }
  655. generator.append(R"~~~(
  656. } };
  657. if (auto value = @enum_snake@_values.get(@enum_snake@); value.has_value())
  658. return value.value();
  659. return {};
  660. }
  661. )~~~");
  662. };
  663. append_prop_search("GeneralCategory"sv, "general_category"sv, "s_general_categories"sv);
  664. append_from_string("GeneralCategory"sv, "general_category"sv, unicode_data.general_categories, unicode_data.general_category_aliases);
  665. append_prop_search("Property"sv, "property"sv, "s_properties"sv);
  666. append_from_string("Property"sv, "property"sv, unicode_data.prop_list, unicode_data.prop_aliases);
  667. append_prop_search("Script"sv, "script"sv, "s_scripts"sv);
  668. append_prop_search("Script"sv, "script_extension"sv, "s_script_extensions"sv);
  669. append_from_string("Script"sv, "script"sv, unicode_data.script_list, unicode_data.script_aliases);
  670. generator.append(R"~~~(
  671. }
  672. }
  673. )~~~");
  674. write_to_file_if_different(file, generator.as_string_view());
  675. }
  676. static Vector<u32> flatten_code_point_ranges(Vector<CodePointRange> const& code_points)
  677. {
  678. Vector<u32> flattened;
  679. for (auto const& range : code_points) {
  680. flattened.grow_capacity(range.last - range.first);
  681. for (u32 code_point = range.first; code_point <= range.last; ++code_point)
  682. flattened.append(code_point);
  683. }
  684. return flattened;
  685. }
  686. static Vector<CodePointRange> form_code_point_ranges(Vector<u32> code_points)
  687. {
  688. Vector<CodePointRange> ranges;
  689. u32 range_start = code_points[0];
  690. u32 range_end = range_start;
  691. for (size_t i = 1; i < code_points.size(); ++i) {
  692. u32 code_point = code_points[i];
  693. if ((code_point - range_end) == 1) {
  694. range_end = code_point;
  695. } else {
  696. ranges.append({ range_start, range_end });
  697. range_start = code_point;
  698. range_end = code_point;
  699. }
  700. }
  701. ranges.append({ range_start, range_end });
  702. return ranges;
  703. }
  704. static void sort_and_merge_code_point_ranges(Vector<CodePointRange>& code_points)
  705. {
  706. quick_sort(code_points, [](auto const& range1, auto const& range2) {
  707. return range1.first < range2.first;
  708. });
  709. for (size_t i = 0; i < code_points.size() - 1;) {
  710. if (code_points[i].last >= code_points[i + 1].first) {
  711. code_points[i].last = max(code_points[i].last, code_points[i + 1].last);
  712. code_points.remove(i + 1);
  713. } else {
  714. ++i;
  715. }
  716. }
  717. auto all_code_points = flatten_code_point_ranges(code_points);
  718. code_points = form_code_point_ranges(all_code_points);
  719. }
  720. static void populate_general_category_unions(PropList& general_categories)
  721. {
  722. // The Unicode standard defines General Category values which are not in any UCD file. These
  723. // values are simply unions of other values.
  724. // https://www.unicode.org/reports/tr44/#GC_Values_Table
  725. auto populate_union = [&](auto alias, auto categories) {
  726. auto& code_points = general_categories.ensure(alias);
  727. for (auto const& category : categories)
  728. code_points.extend(general_categories.find(category)->value);
  729. sort_and_merge_code_point_ranges(code_points);
  730. };
  731. populate_union("LC"sv, Array { "Ll"sv, "Lu"sv, "Lt"sv });
  732. populate_union("L"sv, Array { "Lu"sv, "Ll"sv, "Lt"sv, "Lm"sv, "Lo"sv });
  733. populate_union("M"sv, Array { "Mn"sv, "Mc"sv, "Me"sv });
  734. populate_union("N"sv, Array { "Nd"sv, "Nl"sv, "No"sv });
  735. populate_union("P"sv, Array { "Pc"sv, "Pd"sv, "Ps"sv, "Pe"sv, "Pi"sv, "Pf"sv, "Po"sv });
  736. populate_union("S"sv, Array { "Sm"sv, "Sc"sv, "Sk"sv, "So"sv });
  737. populate_union("Z"sv, Array { "Zs"sv, "Zl"sv, "Zp"sv });
  738. populate_union("C"sv, Array { "Cc"sv, "Cf"sv, "Cs"sv, "Co"sv, "Cn"sv });
  739. }
  740. static void normalize_script_extensions(PropList& script_extensions, PropList const& script_list, Vector<Alias> const& script_aliases)
  741. {
  742. // The ScriptExtensions UCD file lays out its code point ranges rather uniquely compared to
  743. // other files. The Script listed on each line may either be a full Script string or an aliased
  744. // abbreviation. Further, the extensions may or may not include the base Script list. Normalize
  745. // the extensions here to be keyed by the full Script name and always include the base list.
  746. auto extensions = move(script_extensions);
  747. script_extensions = script_list;
  748. for (auto const& extension : extensions) {
  749. auto it = find_if(script_aliases.begin(), script_aliases.end(), [&](auto const& alias) { return extension.key == alias.alias; });
  750. auto const& key = (it == script_aliases.end()) ? extension.key : it->property;
  751. auto& code_points = script_extensions.find(key)->value;
  752. code_points.extend(extension.value);
  753. sort_and_merge_code_point_ranges(code_points);
  754. }
  755. // Lastly, the Common and Inherited script extensions are special. They must not contain any
  756. // code points which appear in other script extensions. The ScriptExtensions UCD file does not
  757. // list these extensions, therefore this peculiarity must be handled programatically.
  758. // https://www.unicode.org/reports/tr24/#Assignment_ScriptX_Values
  759. auto code_point_has_other_extension = [&](StringView key, u32 code_point) {
  760. for (auto const& extension : extensions) {
  761. if (extension.key == key)
  762. continue;
  763. if (any_of(extension.value, [&](auto const& r) { return (r.first <= code_point) && (code_point <= r.last); }))
  764. return true;
  765. }
  766. return false;
  767. };
  768. auto get_code_points_without_other_extensions = [&](StringView key) {
  769. auto code_points = flatten_code_point_ranges(script_list.find(key)->value);
  770. code_points.remove_all_matching([&](u32 c) { return code_point_has_other_extension(key, c); });
  771. return code_points;
  772. };
  773. auto common_code_points = get_code_points_without_other_extensions("Common"sv);
  774. script_extensions.set("Common"sv, form_code_point_ranges(common_code_points));
  775. auto inherited_code_points = get_code_points_without_other_extensions("Inherited"sv);
  776. script_extensions.set("Inherited"sv, form_code_point_ranges(inherited_code_points));
  777. }
  778. int main(int argc, char** argv)
  779. {
  780. char const* generated_header_path = nullptr;
  781. char const* generated_implementation_path = nullptr;
  782. char const* unicode_data_path = nullptr;
  783. char const* special_casing_path = nullptr;
  784. char const* derived_general_category_path = nullptr;
  785. char const* prop_list_path = nullptr;
  786. char const* derived_core_prop_path = nullptr;
  787. char const* derived_binary_prop_path = nullptr;
  788. char const* prop_alias_path = nullptr;
  789. char const* prop_value_alias_path = nullptr;
  790. char const* scripts_path = nullptr;
  791. char const* script_extensions_path = nullptr;
  792. char const* emoji_data_path = nullptr;
  793. char const* normalization_path = nullptr;
  794. Core::ArgsParser args_parser;
  795. args_parser.add_option(generated_header_path, "Path to the Unicode Data header file to generate", "generated-header-path", 'h', "generated-header-path");
  796. args_parser.add_option(generated_implementation_path, "Path to the Unicode Data implementation file to generate", "generated-implementation-path", 'c', "generated-implementation-path");
  797. args_parser.add_option(unicode_data_path, "Path to UnicodeData.txt file", "unicode-data-path", 'u', "unicode-data-path");
  798. args_parser.add_option(special_casing_path, "Path to SpecialCasing.txt file", "special-casing-path", 's', "special-casing-path");
  799. args_parser.add_option(derived_general_category_path, "Path to DerivedGeneralCategory.txt file", "derived-general-category-path", 'g', "derived-general-category-path");
  800. args_parser.add_option(prop_list_path, "Path to PropList.txt file", "prop-list-path", 'p', "prop-list-path");
  801. args_parser.add_option(derived_core_prop_path, "Path to DerivedCoreProperties.txt file", "derived-core-prop-path", 'd', "derived-core-prop-path");
  802. args_parser.add_option(derived_binary_prop_path, "Path to DerivedBinaryProperties.txt file", "derived-binary-prop-path", 'b', "derived-binary-prop-path");
  803. args_parser.add_option(prop_alias_path, "Path to PropertyAliases.txt file", "prop-alias-path", 'a', "prop-alias-path");
  804. args_parser.add_option(prop_value_alias_path, "Path to PropertyValueAliases.txt file", "prop-value-alias-path", 'v', "prop-value-alias-path");
  805. args_parser.add_option(scripts_path, "Path to Scripts.txt file", "scripts-path", 'r', "scripts-path");
  806. args_parser.add_option(script_extensions_path, "Path to ScriptExtensions.txt file", "script-extensions-path", 'x', "script-extensions-path");
  807. args_parser.add_option(emoji_data_path, "Path to emoji-data.txt file", "emoji-data-path", 'e', "emoji-data-path");
  808. args_parser.add_option(normalization_path, "Path to DerivedNormalizationProps.txt file", "normalization-path", 'n', "normalization-path");
  809. args_parser.parse(argc, argv);
  810. auto open_file = [&](StringView path, StringView flags, Core::OpenMode mode = Core::OpenMode::ReadOnly) {
  811. if (path.is_empty()) {
  812. warnln("{} is required", flags);
  813. args_parser.print_usage(stderr, argv[0]);
  814. exit(1);
  815. }
  816. auto file_or_error = Core::File::open(path, mode);
  817. if (file_or_error.is_error()) {
  818. warnln("Failed to open {}: {}", path, file_or_error.release_error());
  819. exit(1);
  820. }
  821. return file_or_error.release_value();
  822. };
  823. auto generated_header_file = open_file(generated_header_path, "-h/--generated-header-path", Core::OpenMode::ReadWrite);
  824. auto generated_implementation_file = open_file(generated_implementation_path, "-c/--generated-implementation-path", Core::OpenMode::ReadWrite);
  825. auto unicode_data_file = open_file(unicode_data_path, "-u/--unicode-data-path");
  826. auto derived_general_category_file = open_file(derived_general_category_path, "-g/--derived-general-category-path");
  827. auto special_casing_file = open_file(special_casing_path, "-s/--special-casing-path");
  828. auto prop_list_file = open_file(prop_list_path, "-p/--prop-list-path");
  829. auto derived_core_prop_file = open_file(derived_core_prop_path, "-d/--derived-core-prop-path");
  830. auto derived_binary_prop_file = open_file(derived_binary_prop_path, "-b/--derived-binary-prop-path");
  831. auto prop_alias_file = open_file(prop_alias_path, "-a/--prop-alias-path");
  832. auto prop_value_alias_file = open_file(prop_value_alias_path, "-v/--prop-value-alias-path");
  833. auto scripts_file = open_file(scripts_path, "-r/--scripts-path");
  834. auto script_extensions_file = open_file(script_extensions_path, "-x/--script-extensions-path");
  835. auto emoji_data_file = open_file(emoji_data_path, "-e/--emoji-data-path");
  836. auto normalization_file = open_file(normalization_path, "-n/--normalization-path");
  837. UnicodeData unicode_data {};
  838. parse_special_casing(special_casing_file, unicode_data);
  839. parse_prop_list(derived_general_category_file, unicode_data.general_categories);
  840. parse_prop_list(prop_list_file, unicode_data.prop_list);
  841. parse_prop_list(derived_core_prop_file, unicode_data.prop_list);
  842. parse_prop_list(derived_binary_prop_file, unicode_data.prop_list);
  843. parse_prop_list(emoji_data_file, unicode_data.prop_list);
  844. parse_normalization_props(normalization_file, unicode_data);
  845. parse_alias_list(prop_alias_file, unicode_data.prop_list, unicode_data.prop_aliases);
  846. parse_prop_list(scripts_file, unicode_data.script_list);
  847. parse_prop_list(script_extensions_file, unicode_data.script_extensions, true);
  848. populate_general_category_unions(unicode_data.general_categories);
  849. parse_unicode_data(unicode_data_file, unicode_data);
  850. parse_value_alias_list(prop_value_alias_file, "gc"sv, unicode_data.general_categories.keys(), unicode_data.general_category_aliases);
  851. parse_value_alias_list(prop_value_alias_file, "sc"sv, unicode_data.script_list.keys(), unicode_data.script_aliases, false);
  852. normalize_script_extensions(unicode_data.script_extensions, unicode_data.script_list, unicode_data.script_aliases);
  853. generate_unicode_data_header(generated_header_file, unicode_data);
  854. generate_unicode_data_implementation(generated_implementation_file, unicode_data);
  855. return 0;
  856. }