GenerateUnicodeData.cpp 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733
  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/HashMap.h>
  10. #include <AK/Optional.h>
  11. #include <AK/QuickSort.h>
  12. #include <AK/SourceGenerator.h>
  13. #include <AK/String.h>
  14. #include <AK/StringUtils.h>
  15. #include <AK/Types.h>
  16. #include <AK/Vector.h>
  17. #include <LibCore/ArgsParser.h>
  18. #include <LibCore/File.h>
  19. // Some code points are excluded from UnicodeData.txt, and instead are part of a "range" of code
  20. // points, as indicated by the "name" field. For example:
  21. // 3400;<CJK Ideograph Extension A, First>;Lo;0;L;;;;;N;;;;;
  22. // 4DBF;<CJK Ideograph Extension A, Last>;Lo;0;L;;;;;N;;;;;
  23. struct CodePointRange {
  24. u32 index;
  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. // https://www.unicode.org/reports/tr44/tr44-13.html#WordBreakProperty.txt
  42. using PropList = HashMap<String, Vector<CodePointRange>>;
  43. // PropertyAliases source: https://www.unicode.org/Public/13.0.0/ucd/PropertyAliases.txt
  44. struct Alias {
  45. String property;
  46. String alias;
  47. };
  48. // UnicodeData source: https://www.unicode.org/Public/13.0.0/ucd/UnicodeData.txt
  49. // Field descriptions: https://www.unicode.org/reports/tr44/tr44-13.html#UnicodeData.txt
  50. // https://www.unicode.org/reports/tr44/#General_Category_Values
  51. struct CodePointData {
  52. u32 index { 0 };
  53. u32 code_point { 0 };
  54. String name;
  55. String general_category;
  56. u8 canonical_combining_class { 0 };
  57. String bidi_class;
  58. String decomposition_type;
  59. Optional<i8> numeric_value_decimal;
  60. Optional<i8> numeric_value_digit;
  61. Optional<i8> numeric_value_numeric;
  62. bool bidi_mirrored { false };
  63. String unicode_1_name;
  64. String iso_comment;
  65. Optional<u32> simple_uppercase_mapping;
  66. Optional<u32> simple_lowercase_mapping;
  67. Optional<u32> simple_titlecase_mapping;
  68. Vector<u32> special_casing_indices;
  69. Vector<StringView> prop_list;
  70. StringView word_break_property;
  71. };
  72. struct UnicodeData {
  73. Vector<SpecialCasing> special_casing;
  74. u32 largest_casing_transform_size { 0 };
  75. u32 largest_special_casing_size { 0 };
  76. Vector<String> locales;
  77. Vector<String> conditions;
  78. Vector<CodePointData> code_point_data;
  79. Vector<CodePointRange> code_point_ranges;
  80. Vector<String> general_categories;
  81. u32 last_contiguous_code_point { 0 };
  82. PropList prop_list;
  83. Vector<Alias> prop_aliases;
  84. PropList word_break_prop_list;
  85. };
  86. static constexpr auto s_desired_fields = Array {
  87. "general_category"sv,
  88. "simple_uppercase_mapping"sv,
  89. "simple_lowercase_mapping"sv,
  90. };
  91. static void parse_special_casing(Core::File& file, UnicodeData& unicode_data)
  92. {
  93. auto parse_code_point_list = [&](auto const& line) {
  94. Vector<u32> code_points;
  95. auto segments = line.split(' ');
  96. for (auto const& code_point : segments)
  97. code_points.append(AK::StringUtils::convert_to_uint_from_hex<u32>(code_point).value());
  98. return code_points;
  99. };
  100. while (file.can_read_line()) {
  101. auto line = file.read_line();
  102. if (line.is_empty() || line.starts_with('#'))
  103. continue;
  104. if (auto index = line.find('#'); index.has_value())
  105. line = line.substring(0, *index);
  106. auto segments = line.split(';', true);
  107. VERIFY(segments.size() == 5 || segments.size() == 6);
  108. SpecialCasing casing {};
  109. casing.index = static_cast<u32>(unicode_data.special_casing.size());
  110. casing.code_point = AK::StringUtils::convert_to_uint_from_hex<u32>(segments[0]).value();
  111. casing.lowercase_mapping = parse_code_point_list(segments[1]);
  112. casing.titlecase_mapping = parse_code_point_list(segments[2]);
  113. casing.uppercase_mapping = parse_code_point_list(segments[3]);
  114. if (auto condition = segments[4].trim_whitespace(); !condition.is_empty()) {
  115. auto conditions = condition.split(' ', true);
  116. VERIFY(conditions.size() == 1 || conditions.size() == 2);
  117. if (conditions.size() == 2) {
  118. casing.locale = move(conditions[0]);
  119. casing.condition = move(conditions[1]);
  120. } else if (all_of(conditions[0], is_ascii_lower_alpha)) {
  121. casing.locale = move(conditions[0]);
  122. } else {
  123. casing.condition = move(conditions[0]);
  124. }
  125. casing.locale = casing.locale.to_uppercase();
  126. casing.condition.replace("_", "", true);
  127. if (!casing.locale.is_empty() && !unicode_data.locales.contains_slow(casing.locale))
  128. unicode_data.locales.append(casing.locale);
  129. if (!casing.condition.is_empty() && !unicode_data.conditions.contains_slow(casing.condition))
  130. unicode_data.conditions.append(casing.condition);
  131. }
  132. unicode_data.largest_casing_transform_size = max(unicode_data.largest_casing_transform_size, casing.lowercase_mapping.size());
  133. unicode_data.largest_casing_transform_size = max(unicode_data.largest_casing_transform_size, casing.titlecase_mapping.size());
  134. unicode_data.largest_casing_transform_size = max(unicode_data.largest_casing_transform_size, casing.uppercase_mapping.size());
  135. unicode_data.special_casing.append(move(casing));
  136. }
  137. }
  138. static void parse_prop_list(Core::File& file, PropList& prop_list)
  139. {
  140. while (file.can_read_line()) {
  141. auto line = file.read_line();
  142. if (line.is_empty() || line.starts_with('#'))
  143. continue;
  144. if (auto index = line.find('#'); index.has_value())
  145. line = line.substring(0, *index);
  146. auto segments = line.split_view(';', true);
  147. VERIFY(segments.size() == 2);
  148. auto code_point_range = segments[0].trim_whitespace();
  149. auto property = segments[1].trim_whitespace();
  150. auto& code_points = prop_list.ensure(property);
  151. if (code_point_range.contains(".."sv)) {
  152. segments = code_point_range.split_view(".."sv);
  153. VERIFY(segments.size() == 2);
  154. auto begin = AK::StringUtils::convert_to_uint_from_hex<u32>(segments[0]).value();
  155. auto end = AK::StringUtils::convert_to_uint_from_hex<u32>(segments[1]).value();
  156. code_points.append({ 0, begin, end });
  157. } else {
  158. auto code_point = AK::StringUtils::convert_to_uint_from_hex<u32>(code_point_range).value();
  159. code_points.append({ 0, code_point, code_point });
  160. }
  161. }
  162. }
  163. static void parse_alias_list(Core::File& file, PropList const& prop_list, Vector<Alias>& prop_aliases)
  164. {
  165. String current_property;
  166. auto append_alias = [&](auto alias, auto property) {
  167. // Note: The alias files contain lines such as "Hyphen = Hyphen", which we should just skip.
  168. if (alias == property)
  169. return;
  170. // FIXME: We will, eventually, need to find where missing properties are located and parse them.
  171. if (!prop_list.contains(property))
  172. return;
  173. prop_aliases.append({ property, alias });
  174. };
  175. while (file.can_read_line()) {
  176. auto line = file.read_line();
  177. if (line.is_empty() || line.starts_with('#')) {
  178. if (line.ends_with("Properties"sv))
  179. current_property = line.substring(2);
  180. continue;
  181. }
  182. // Note: For now, we only care about Binary Property aliases for Unicode property escapes.
  183. if (current_property != "Binary Properties"sv)
  184. continue;
  185. auto segments = line.split_view(';', true);
  186. VERIFY((segments.size() == 2) || (segments.size() == 3));
  187. auto alias = segments[0].trim_whitespace();
  188. auto property = segments[1].trim_whitespace();
  189. append_alias(alias, property);
  190. if (segments.size() == 3) {
  191. alias = segments[2].trim_whitespace();
  192. append_alias(alias, property);
  193. }
  194. }
  195. }
  196. static void parse_unicode_data(Core::File& file, UnicodeData& unicode_data)
  197. {
  198. Optional<u32> code_point_range_start;
  199. Optional<u32> code_point_range_index;
  200. Optional<u32> last_contiguous_code_point;
  201. u32 previous_code_point = 0;
  202. while (file.can_read_line()) {
  203. auto line = file.read_line();
  204. if (line.is_empty())
  205. continue;
  206. auto segments = line.split(';', true);
  207. VERIFY(segments.size() == 15);
  208. CodePointData data {};
  209. data.index = static_cast<u32>(unicode_data.code_point_data.size());
  210. data.code_point = AK::StringUtils::convert_to_uint_from_hex<u32>(segments[0]).value();
  211. data.name = move(segments[1]);
  212. data.general_category = move(segments[2]);
  213. data.canonical_combining_class = AK::StringUtils::convert_to_uint<u8>(segments[3]).value();
  214. data.bidi_class = move(segments[4]);
  215. data.decomposition_type = move(segments[5]);
  216. data.numeric_value_decimal = AK::StringUtils::convert_to_int<i8>(segments[6]);
  217. data.numeric_value_digit = AK::StringUtils::convert_to_int<i8>(segments[7]);
  218. data.numeric_value_numeric = AK::StringUtils::convert_to_int<i8>(segments[8]);
  219. data.bidi_mirrored = segments[9] == "Y"sv;
  220. data.unicode_1_name = move(segments[10]);
  221. data.iso_comment = move(segments[11]);
  222. data.simple_uppercase_mapping = AK::StringUtils::convert_to_uint_from_hex<u32>(segments[12]);
  223. data.simple_lowercase_mapping = AK::StringUtils::convert_to_uint_from_hex<u32>(segments[13]);
  224. data.simple_titlecase_mapping = AK::StringUtils::convert_to_uint_from_hex<u32>(segments[14]);
  225. if (data.name.starts_with("<"sv) && data.name.ends_with(", First>")) {
  226. VERIFY(!code_point_range_start.has_value() && !code_point_range_index.has_value());
  227. code_point_range_start = data.code_point;
  228. code_point_range_index = data.index;
  229. data.name = data.name.substring(1, data.name.length() - 9);
  230. } else if (data.name.starts_with("<"sv) && data.name.ends_with(", Last>")) {
  231. VERIFY(code_point_range_start.has_value() && code_point_range_index.has_value());
  232. unicode_data.code_point_ranges.append({ *code_point_range_index, *code_point_range_start, data.code_point });
  233. data.name = data.name.substring(1, data.name.length() - 8);
  234. code_point_range_start.clear();
  235. code_point_range_index.clear();
  236. } else if ((data.code_point > 0) && (data.code_point - previous_code_point) != 1) {
  237. if (!last_contiguous_code_point.has_value())
  238. last_contiguous_code_point = previous_code_point;
  239. }
  240. for (auto const& casing : unicode_data.special_casing) {
  241. if (casing.code_point == data.code_point)
  242. data.special_casing_indices.append(casing.index);
  243. }
  244. for (auto const& property : unicode_data.prop_list) {
  245. for (auto const& range : property.value) {
  246. if ((range.first <= data.code_point) && (data.code_point <= range.last)) {
  247. data.prop_list.append(property.key);
  248. break;
  249. }
  250. }
  251. }
  252. if (data.prop_list.is_empty())
  253. data.prop_list.append("Assigned"sv);
  254. for (auto const& property : unicode_data.word_break_prop_list) {
  255. for (auto const& range : property.value) {
  256. if ((range.first <= data.code_point) && (data.code_point <= range.last)) {
  257. data.word_break_property = property.key;
  258. break;
  259. }
  260. }
  261. if (!data.word_break_property.is_empty())
  262. break;
  263. }
  264. if (data.word_break_property.is_empty())
  265. data.word_break_property = "Other"sv;
  266. unicode_data.largest_special_casing_size = max(unicode_data.largest_special_casing_size, data.special_casing_indices.size());
  267. if (!unicode_data.general_categories.contains_slow(data.general_category))
  268. unicode_data.general_categories.append(data.general_category);
  269. previous_code_point = data.code_point;
  270. unicode_data.code_point_data.append(move(data));
  271. }
  272. unicode_data.last_contiguous_code_point = *last_contiguous_code_point;
  273. }
  274. static void generate_unicode_data_header(UnicodeData& unicode_data)
  275. {
  276. StringBuilder builder;
  277. SourceGenerator generator { builder };
  278. generator.set("casing_transform_size", String::number(unicode_data.largest_casing_transform_size));
  279. generator.set("special_casing_size", String::number(unicode_data.largest_special_casing_size));
  280. auto generate_enum = [&](StringView name, StringView default_, Vector<String> values, Vector<Alias> aliases = {}, bool as_bitmask = false) {
  281. VERIFY((values.size() + !default_.is_empty()) <= 64);
  282. quick_sort(values);
  283. quick_sort(aliases, [](auto& alias1, auto& alias2) { return alias1.alias < alias2.alias; });
  284. generator.set("name", name);
  285. generator.set("underlying", String::formatted("{}UnderlyingType", name));
  286. if (as_bitmask) {
  287. generator.append(R"~~~(
  288. using @underlying@ = u64;
  289. enum class @name@ : @underlying@ {)~~~");
  290. } else {
  291. generator.append(R"~~~(
  292. enum class @name@ {)~~~");
  293. }
  294. if (!default_.is_empty()) {
  295. generator.set("default", default_);
  296. generator.append(R"~~~(
  297. @default@,)~~~");
  298. }
  299. u8 index = 0;
  300. for (auto const& value : values) {
  301. generator.set("value", value);
  302. if (as_bitmask) {
  303. generator.set("index", String::number(index++));
  304. generator.append(R"~~~(
  305. @value@ = static_cast<@underlying@>(1) << @index@,)~~~");
  306. } else {
  307. generator.append(R"~~~(
  308. @value@,)~~~");
  309. }
  310. }
  311. for (auto const& alias : aliases) {
  312. generator.set("alias", alias.alias);
  313. generator.set("value", alias.property);
  314. generator.append(R"~~~(
  315. @alias@ = @value@,)~~~");
  316. }
  317. generator.append(R"~~~(
  318. };
  319. )~~~");
  320. if (as_bitmask) {
  321. generator.append(R"~~~(
  322. constexpr @name@ operator&(@name@ value1, @name@ value2)
  323. {
  324. return static_cast<@name@>(static_cast<@underlying@>(value1) & static_cast<@underlying@>(value2));
  325. }
  326. constexpr @name@ operator|(@name@ value1, @name@ value2)
  327. {
  328. return static_cast<@name@>(static_cast<@underlying@>(value1) | static_cast<@underlying@>(value2));
  329. }
  330. )~~~");
  331. }
  332. };
  333. generator.append(R"~~~(
  334. #pragma once
  335. #include <AK/Optional.h>
  336. #include <AK/Types.h>
  337. #include <LibUnicode/Forward.h>
  338. namespace Unicode {
  339. )~~~");
  340. generate_enum("Locale"sv, "None"sv, move(unicode_data.locales));
  341. generate_enum("Condition"sv, "None"sv, move(unicode_data.conditions));
  342. generate_enum("GeneralCategory"sv, {}, move(unicode_data.general_categories));
  343. generate_enum("Property"sv, "Assigned"sv, unicode_data.prop_list.keys(), unicode_data.prop_aliases, true);
  344. generate_enum("WordBreakProperty"sv, "Other"sv, unicode_data.word_break_prop_list.keys());
  345. generator.append(R"~~~(
  346. struct SpecialCasing {
  347. u32 code_point { 0 };
  348. u32 lowercase_mapping[@casing_transform_size@];
  349. u32 lowercase_mapping_size { 0 };
  350. u32 uppercase_mapping[@casing_transform_size@];
  351. u32 uppercase_mapping_size { 0 };
  352. u32 titlecase_mapping[@casing_transform_size@];
  353. u32 titlecase_mapping_size { 0 };
  354. Locale locale { Locale::None };
  355. Condition condition { Condition::None };
  356. };
  357. struct UnicodeData {
  358. u32 code_point;)~~~");
  359. auto append_field = [&](StringView type, StringView name) {
  360. if (!s_desired_fields.span().contains_slow(name))
  361. return;
  362. generator.set("type", type);
  363. generator.set("name", name);
  364. generator.append(R"~~~(
  365. @type@ @name@;)~~~");
  366. };
  367. // Note: For compile-time performance, only primitive types are used.
  368. append_field("char const*"sv, "name"sv);
  369. append_field("GeneralCategory"sv, "general_category"sv);
  370. append_field("u8"sv, "canonical_combining_class"sv);
  371. append_field("char const*"sv, "bidi_class"sv);
  372. append_field("char const*"sv, "decomposition_type"sv);
  373. append_field("i8"sv, "numeric_value_decimal"sv);
  374. append_field("i8"sv, "numeric_value_digit"sv);
  375. append_field("i8"sv, "numeric_value_numeric"sv);
  376. append_field("bool"sv, "bidi_mirrored"sv);
  377. append_field("char const*"sv, "unicode_1_name"sv);
  378. append_field("char const*"sv, "iso_comment"sv);
  379. append_field("u32"sv, "simple_uppercase_mapping"sv);
  380. append_field("u32"sv, "simple_lowercase_mapping"sv);
  381. append_field("u32"sv, "simple_titlecase_mapping"sv);
  382. generator.append(R"~~~(
  383. SpecialCasing const* special_casing[@special_casing_size@] {};
  384. u32 special_casing_size { 0 };
  385. Property properties { Property::Assigned };
  386. WordBreakProperty word_break_property { WordBreakProperty::Other };
  387. };
  388. namespace Detail {
  389. Optional<UnicodeData> unicode_data_for_code_point(u32 code_point);
  390. Optional<Property> property_from_string(StringView const& property);
  391. }
  392. })~~~");
  393. outln("{}", generator.as_string_view());
  394. }
  395. static void generate_unicode_data_implementation(UnicodeData unicode_data)
  396. {
  397. StringBuilder builder;
  398. SourceGenerator generator { builder };
  399. generator.set("special_casing_size", String::number(unicode_data.special_casing.size()));
  400. generator.set("code_point_data_size", String::number(unicode_data.code_point_data.size()));
  401. generator.set("last_contiguous_code_point", String::formatted("0x{:x}", unicode_data.last_contiguous_code_point));
  402. generator.append(R"~~~(
  403. #include <AK/Array.h>
  404. #include <AK/CharacterTypes.h>
  405. #include <AK/Find.h>
  406. #include <AK/StringView.h>
  407. #include <LibUnicode/UnicodeData.h>
  408. namespace Unicode {
  409. )~~~");
  410. auto append_list_and_size = [&](auto const& list, StringView format) {
  411. if (list.is_empty()) {
  412. generator.append(", {}, 0");
  413. return;
  414. }
  415. bool first = true;
  416. generator.append(", {");
  417. for (auto const& item : list) {
  418. generator.append(first ? " " : ", ");
  419. generator.append(String::formatted(format, item));
  420. first = false;
  421. }
  422. generator.append(String::formatted(" }}, {}", list.size()));
  423. };
  424. generator.append(R"~~~(
  425. static constexpr Array<SpecialCasing, @special_casing_size@> s_special_casing { {)~~~");
  426. for (auto const& casing : unicode_data.special_casing) {
  427. generator.set("code_point", String::formatted("{:#x}", casing.code_point));
  428. generator.append(R"~~~(
  429. { @code_point@)~~~");
  430. constexpr auto format = "0x{:x}"sv;
  431. append_list_and_size(casing.lowercase_mapping, format);
  432. append_list_and_size(casing.uppercase_mapping, format);
  433. append_list_and_size(casing.titlecase_mapping, format);
  434. generator.set("locale", casing.locale.is_empty() ? "None" : casing.locale);
  435. generator.append(", Locale::@locale@");
  436. generator.set("condition", casing.condition.is_empty() ? "None" : casing.condition);
  437. generator.append(", Condition::@condition@");
  438. generator.append(" },");
  439. }
  440. generator.append(R"~~~(
  441. } };
  442. static constexpr Array<UnicodeData, @code_point_data_size@> s_unicode_data { {)~~~");
  443. auto append_field = [&](StringView name, String value) {
  444. if (!s_desired_fields.span().contains_slow(name))
  445. return;
  446. generator.set("value", move(value));
  447. generator.append(", @value@");
  448. };
  449. for (auto const& data : unicode_data.code_point_data) {
  450. generator.set("code_point", String::formatted("{:#x}", data.code_point));
  451. generator.append(R"~~~(
  452. { @code_point@)~~~");
  453. append_field("name", String::formatted("\"{}\"", data.name));
  454. append_field("general_category", String::formatted("GeneralCategory::{}", data.general_category));
  455. append_field("canonical_combining_class", String::number(data.canonical_combining_class));
  456. append_field("bidi_class", String::formatted("\"{}\"", data.bidi_class));
  457. append_field("decomposition_type", String::formatted("\"{}\"", data.decomposition_type));
  458. append_field("numeric_value_decimal", String::number(data.numeric_value_decimal.value_or(-1)));
  459. append_field("numeric_value_digit", String::number(data.numeric_value_digit.value_or(-1)));
  460. append_field("numeric_value_numeric", String::number(data.numeric_value_numeric.value_or(-1)));
  461. append_field("bidi_mirrored", String::formatted("{}", data.bidi_mirrored));
  462. append_field("unicode_1_name", String::formatted("\"{}\"", data.unicode_1_name));
  463. append_field("iso_comment", String::formatted("\"{}\"", data.iso_comment));
  464. append_field("simple_uppercase_mapping", String::formatted("{:#x}", data.simple_uppercase_mapping.value_or(data.code_point)));
  465. append_field("simple_lowercase_mapping", String::formatted("{:#x}", data.simple_lowercase_mapping.value_or(data.code_point)));
  466. append_field("simple_titlecase_mapping", String::formatted("{:#x}", data.simple_titlecase_mapping.value_or(data.code_point)));
  467. append_list_and_size(data.special_casing_indices, "&s_special_casing[{}]"sv);
  468. bool first = true;
  469. for (auto const& property : data.prop_list) {
  470. generator.append(first ? ", " : " | ");
  471. generator.append(String::formatted("Property::{}", property));
  472. first = false;
  473. }
  474. generator.append(String::formatted(", WordBreakProperty::{}", data.word_break_property));
  475. generator.append(" },");
  476. }
  477. generator.append(R"~~~(
  478. } };
  479. static Optional<u32> index_of_code_point_in_range(u32 code_point)
  480. {)~~~");
  481. for (auto const& range : unicode_data.code_point_ranges) {
  482. generator.set("index", String::formatted("{}", range.index));
  483. generator.set("first", String::formatted("{:#x}", range.first));
  484. generator.set("last", String::formatted("{:#x}", range.last));
  485. generator.append(R"~~~(
  486. if ((code_point > @first@) && (code_point < @last@))
  487. return @index@;)~~~");
  488. }
  489. generator.append(R"~~~(
  490. return {};
  491. }
  492. namespace Detail {
  493. Optional<UnicodeData> unicode_data_for_code_point(u32 code_point)
  494. {
  495. VERIFY(is_unicode(code_point));
  496. if (code_point <= @last_contiguous_code_point@)
  497. return s_unicode_data[code_point];
  498. if (auto index = index_of_code_point_in_range(code_point); index.has_value()) {
  499. auto data_for_range = s_unicode_data[*index];
  500. data_for_range.simple_uppercase_mapping = code_point;
  501. data_for_range.simple_lowercase_mapping = code_point;
  502. return data_for_range;
  503. }
  504. auto it = AK::find_if(s_unicode_data.begin(), s_unicode_data.end(), [code_point](auto const& data) { return data.code_point == code_point; });
  505. if (it != s_unicode_data.end())
  506. return *it;
  507. return {};
  508. }
  509. Optional<Property> property_from_string(StringView const& property)
  510. {
  511. if (property == "Assigned"sv)
  512. return Property::Assigned;)~~~");
  513. for (auto const& property : unicode_data.prop_list) {
  514. generator.set("property", property.key);
  515. generator.append(R"~~~(
  516. if (property == "@property@"sv)
  517. return Property::@property@;)~~~");
  518. }
  519. for (auto const& alias : unicode_data.prop_aliases) {
  520. generator.set("property", alias.alias);
  521. generator.append(R"~~~(
  522. if (property == "@property@"sv)
  523. return Property::@property@;)~~~");
  524. }
  525. generator.append(R"~~~(
  526. return {};
  527. }
  528. }
  529. })~~~");
  530. outln("{}", generator.as_string_view());
  531. }
  532. int main(int argc, char** argv)
  533. {
  534. bool generate_header = false;
  535. bool generate_implementation = false;
  536. char const* unicode_data_path = nullptr;
  537. char const* special_casing_path = nullptr;
  538. char const* prop_list_path = nullptr;
  539. char const* derived_core_prop_path = nullptr;
  540. char const* prop_alias_path = nullptr;
  541. char const* word_break_path = nullptr;
  542. Core::ArgsParser args_parser;
  543. args_parser.add_option(generate_header, "Generate the Unicode Data header file", "generate-header", 'h');
  544. args_parser.add_option(generate_implementation, "Generate the Unicode Data implementation file", "generate-implementation", 'c');
  545. args_parser.add_option(unicode_data_path, "Path to UnicodeData.txt file", "unicode-data-path", 'u', "unicode-data-path");
  546. args_parser.add_option(special_casing_path, "Path to SpecialCasing.txt file", "special-casing-path", 's', "special-casing-path");
  547. args_parser.add_option(prop_list_path, "Path to PropList.txt file", "prop-list-path", 'p', "prop-list-path");
  548. args_parser.add_option(derived_core_prop_path, "Path to DerivedCoreProperties.txt file", "derived-core-prop-path", 'd', "derived-core-prop-path");
  549. args_parser.add_option(prop_alias_path, "Path to PropertyAliases.txt file", "prop-alias-path", 'a', "prop-alias-path");
  550. args_parser.add_option(word_break_path, "Path to WordBreakProperty.txt file", "word-break-path", 'w', "word-break-path");
  551. args_parser.parse(argc, argv);
  552. if (!generate_header && !generate_implementation) {
  553. warnln("At least one of -h/--generate-header or -c/--generate-implementation is required");
  554. args_parser.print_usage(stderr, argv[0]);
  555. return 1;
  556. }
  557. auto open_file = [&](StringView path, StringView flags) {
  558. if (path.is_empty()) {
  559. warnln("{} is required", flags);
  560. args_parser.print_usage(stderr, argv[0]);
  561. exit(1);
  562. }
  563. auto file_or_error = Core::File::open(path, Core::OpenMode::ReadOnly);
  564. if (file_or_error.is_error()) {
  565. warnln("Failed to open {}: {}", path, file_or_error.release_error());
  566. exit(1);
  567. }
  568. return file_or_error.release_value();
  569. };
  570. auto unicode_data_file = open_file(unicode_data_path, "-u/--unicode-data-path");
  571. auto special_casing_file = open_file(special_casing_path, "-s/--special-casing-path");
  572. auto prop_list_file = open_file(prop_list_path, "-p/--prop-list-path");
  573. auto derived_core_prop_file = open_file(derived_core_prop_path, "-d/--derived-core-prop-path");
  574. auto prop_alias_file = open_file(prop_alias_path, "-a/--prop-alias-path");
  575. auto word_break_file = open_file(word_break_path, "-w/--word-break-path");
  576. UnicodeData unicode_data {};
  577. parse_special_casing(special_casing_file, unicode_data);
  578. parse_prop_list(prop_list_file, unicode_data.prop_list);
  579. parse_prop_list(derived_core_prop_file, unicode_data.prop_list);
  580. parse_alias_list(prop_alias_file, unicode_data.prop_list, unicode_data.prop_aliases);
  581. parse_prop_list(word_break_file, unicode_data.word_break_prop_list);
  582. // The Unicode standard defines additional properties (Any, Assigned, ASCII) which are not in
  583. // any UCD file. Assigned is set as the default enum value 0 so "property & Assigned == Assigned"
  584. // is always true. Any is not assigned code points here because this file only parses assigned
  585. // code points, whereas Any will include unassigned code points.
  586. // https://unicode.org/reports/tr18/#General_Category_Property
  587. unicode_data.prop_list.set("Any"sv, {});
  588. unicode_data.prop_list.set("ASCII"sv, { { 0, 0, 0x7f } });
  589. parse_unicode_data(unicode_data_file, unicode_data);
  590. if (generate_header)
  591. generate_unicode_data_header(unicode_data);
  592. if (generate_implementation)
  593. generate_unicode_data_implementation(move(unicode_data));
  594. return 0;
  595. }