Skip to content

Commit

Permalink
feat(core): ldml normalization 🙀
Browse files Browse the repository at this point in the history
- more progress in the normalization pipeline. Trying to keep it from leaking.
- normalize output to NFC.
- normalize json test data to NFC (both context and expected).
- we do NOT try to normalize the 'embedded' strings currently.
- yet more test data fixes (turns out u+00e0 ≠ u+00e8)

For: #9468
  • Loading branch information
srl295 committed Oct 12, 2023
1 parent 736af4b commit a577bd2
Show file tree
Hide file tree
Showing 8 changed files with 172 additions and 87 deletions.
175 changes: 98 additions & 77 deletions core/src/ldml/ldml_processor.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
*/

#include <fstream>
#include <algorithm>
#include "ldml/ldml_processor.hpp"
#include "state.hpp"
#include "kmx_file.h"
Expand Down Expand Up @@ -260,81 +261,100 @@ ldml_processor::process_event(

void
ldml_processor::process_key_string(km_core_state *state, const std::u16string &key_str) const {
// found a string - push it into the context and actions
// we convert it here instead of using the emit_text() overload
// so that we don't have to reconvert it inside the transform code.
std::u32string str32 = kmx::u16string_to_u32string(key_str);
UErrorCode status = U_ZERO_ERROR;
// We know that key_str is not empty per the caller.

if (!transforms) {
// No transforms: just emit the string.
emit_text(state, str32);
// we convert the keys str to UTF-32 here instead of using the emit_text() overload
// so that we don't have to reconvert it inside the transform code.
std::u32string key_str32 = kmx::u16string_to_u32string(key_str);
// normalize the keystroke to NFD
ldml::normalize_nfd(key_str32, status);

// extract context string, in NFC
std::u32string old_ctxtstr_nfc;
(void)context_to_string(state, old_ctxtstr_nfc, false);
ldml::normalize_nfc(old_ctxtstr_nfc, status);
assert(U_SUCCESS(status));

// context string in NFD
std::u32string ctxtstr;
(void)context_to_string(state, ctxtstr, true); // with markers
// add the newly added key output to ctxtstr
ctxtstr.append(key_str32);
ldml::normalize_nfd(ctxtstr, status);
assert(U_SUCCESS(status));

/** transform output string */
std::u32string outputString;
/** how many chars of the ctxtstr to replace */
size_t matchedContext = 0; // zero if no transforms

// begin modifications to the string

if(transforms) {
matchedContext = transforms->apply(ctxtstr, outputString);
} else {
// Process transforms here
/**
* a copy of the current/changed context, for transform use.
*
*/
std::u32string ctxtstr;
(void)context_to_string(state, ctxtstr);
// add the newly added key output to ctxtstr
ctxtstr.append(str32);
// and normalize

/** the output buffer for transforms */
std::u32string outputString;

// apply the transform, get how much matched (at the end)
const size_t matchedContext = transforms->apply(ctxtstr, outputString);


if (matchedContext == 0) {
// No match, just emit the original string
emit_text(state, str32);
} else {
// We have a match.

ctxtstr.resize(ctxtstr.length() - str32.length());
/** how many chars of the context we need to clear */
auto charsToDelete = matchedContext - str32.length(); /* we don't need to clear the output of the current key */

/** how many context items need to be removed */
size_t contextRemoved = 0;
for (auto c = state->context().rbegin(); charsToDelete > 0 && c != state->context().rend(); c++, contextRemoved++) {
/** last char of context */
km_core_usv lastCtx = ctxtstr.back();
uint8_t type = c->type;
assert(type == KM_CORE_BT_CHAR || type == KM_CORE_BT_MARKER);
if (type == KM_CORE_BT_CHAR) {
// single char, drop it
charsToDelete--;
assert(c->character == lastCtx);
ctxtstr.pop_back();
state->actions().push_backspace(KM_CORE_BT_CHAR, c->character); // Cause prior char to be removed
} else if (type == KM_CORE_BT_MARKER) {
// it's a marker, 'worth' 3 uchars
assert(charsToDelete >= 3);
assert(lastCtx == c->marker); // end of list
charsToDelete -= 3;
// pop off the three-part sentinel string
ctxtstr.pop_back();
ctxtstr.pop_back();
ctxtstr.pop_back();
// push a special backspace to delete the marker
state->actions().push_backspace(KM_CORE_BT_MARKER, c->marker);
}
}
// now, pop the right number of context items
for (size_t i = 0; i < contextRemoved; i++) {
// we don't pop during the above loop because the iterator gets confused
state->context().pop_back();
}
// Now, add in the updated text. This will convert UC_SENTINEL, etc back to marker actions.
emit_text(state, outputString);
// If we needed it further. we could update ctxtstr here:
// ctxtstr.append(outputString);
// ... but it is no longer needed at this point.
} // end of transform match
} // end of processing transforms
// no transforms, no output
}

// drop last 'matchedContext':
ctxtstr.resize(ctxtstr.length() - matchedContext);
ctxtstr.append(outputString); // TODO-LDML: should be able to do a normalization-safe append here.
ldml::normalize_nfd(ctxtstr, status);
assert(U_SUCCESS(status));

// Ok. We've done all the happy manipulations.

/** NFC and no markers */
std::u32string ctxtstr_cleanedup = ctxtstr;
// TODO-LDML: remove markers!
ldml::normalize_nfc(ctxtstr_cleanedup, status);

// find common prefix
auto ctxt_prefix = mismatch(old_ctxtstr_nfc.begin(), old_ctxtstr_nfc.end(), ctxtstr_cleanedup.begin(), ctxtstr_cleanedup.end());
/** the part of the old str that changed */
std::u32string old_ctxtstr_changed(ctxt_prefix.first,old_ctxtstr_nfc.end());
std::u32string new_ctxtstr_changed(ctxt_prefix.second,ctxtstr_cleanedup.end());

// drop the old suffix. Note: this mutates old_ctxtstr_changed.
remove_text(state, old_ctxtstr_changed, old_ctxtstr_changed.length());
assert(old_ctxtstr_changed.length() == 0);
emit_text(state, new_ctxtstr_changed);
}

void
ldml_processor::remove_text(km_core_state *state, std::u32string &str, size_t length) {
/** how many context items need to be removed */
size_t contextRemoved = 0;
for (auto c = state->context().rbegin(); length > 0 && c != state->context().rend(); c++, contextRemoved++) {
/** last char of context */
km_core_usv lastCtx = str.back();
uint8_t type = c->type;
assert(type == KM_CORE_BT_CHAR || type == KM_CORE_BT_MARKER);
if (type == KM_CORE_BT_CHAR) {
// single char, drop it
length--;
assert(c->character == lastCtx);
str.pop_back();
state->actions().push_backspace(KM_CORE_BT_CHAR, c->character); // Cause prior char to be removed
} else if (type == KM_CORE_BT_MARKER) {
// it's a marker, 'worth' 3 uchars
assert(length >= 3);
assert(lastCtx == c->marker); // end of list
length -= 3;
// pop off the three-part sentinel string
str.pop_back();
str.pop_back();
str.pop_back();
// push a special backspace to delete the marker
state->actions().push_backspace(KM_CORE_BT_MARKER, c->marker);
}
}
// now, pop the right number of context items
for (size_t i = 0; i < contextRemoved; i++) {
// we don't pop during the above loop because the iterator gets confused
state->context().pop_back();
}
}

km_core_attr const & ldml_processor::attributes() const {
Expand Down Expand Up @@ -402,25 +422,26 @@ ldml_processor::emit_marker(km_core_state *state, KMX_DWORD marker_no) {
}

size_t
ldml_processor::context_to_string(km_core_state *state, std::u32string &str) {
ldml_processor::context_to_string(km_core_state *state, std::u32string &str, bool include_markers) {
str.clear();
auto &cp = state->context();
size_t ctxlen = 0; // TODO-LDML: is this needed?
size_t ctxlen = 0; // TODO-LDML: not used by callers?
uint8_t last_type = KM_CORE_BT_UNKNOWN;
for (auto c = cp.rbegin(); c != cp.rend(); c++, ctxlen++) {
last_type = c->type;
if (last_type == KM_CORE_BT_CHAR) {
str.insert(0, 1, c->character);
} else if (last_type == KM_CORE_BT_MARKER) {
assert(km::kbp::kmx::is_valid_marker(c->marker));
prepend_marker(str, c->marker);
if (include_markers) {
prepend_marker(str, c->marker);
}
} else {
break;
}
}
return ctxlen; // consumed the entire context buffer.
}


} // namespace kbp
} // namespace km
8 changes: 7 additions & 1 deletion core/src/ldml/ldml_processor.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -93,6 +93,12 @@ namespace kbp {
static void emit_text(km_core_state *state, km_core_usv ch);
/** emit a marker */
static void emit_marker(km_core_state *state, KMX_DWORD marker);
/**
* Delete text from the state.
* @param str string with text to remove, from the end
* @param length number of chars from the end of str to drop
*/
static void remove_text(km_core_state *state, std::u32string &str, size_t length);

/** process a typed key */
void process_key_string(km_core_state *state, const std::u16string &key_str) const;
Expand All @@ -103,7 +109,7 @@ namespace kbp {
* Convert markers into the UC_SENTINEL format.
* @return the number of context items consumed
*/
static size_t context_to_string(km_core_state *state, std::u32string &str);
static size_t context_to_string(km_core_state *state, std::u32string &str, bool include_markers = true);

/** prepend the marker string in UC_SENTINEL format to the str */
inline static void prepend_marker(std::u32string &str, KMX_DWORD marker);
Expand Down
47 changes: 47 additions & 0 deletions core/src/ldml/ldml_transforms.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -846,6 +846,7 @@ transforms::load(

// string

// TODO-LDML: copypasta -> refactor
std::u32string &normalize_nfd(std::u32string &str, UErrorCode &status) {
const icu::Normalizer2 *nfd = icu::Normalizer2::getNFDInstance(status);
if (U_FAILURE(status)) {
Expand Down Expand Up @@ -873,6 +874,7 @@ std::u32string &normalize_nfd(std::u32string &str, UErrorCode &status) {
return str;
}

// TODO-LDML: copypasta -> refactor
std::u16string &normalize_nfd(std::u16string &str, UErrorCode &status) {
const icu::Normalizer2 *nfd = icu::Normalizer2::getNFDInstance(status);
if (U_FAILURE(status)) {
Expand All @@ -888,6 +890,51 @@ std::u16string &normalize_nfd(std::u16string &str, UErrorCode &status) {
return str;
}

// TODO-LDML: copypasta -> refactor
std::u32string &normalize_nfc(std::u32string &str, UErrorCode &status) {
const icu::Normalizer2 *nfc = icu::Normalizer2::getNFCInstance(status);
if (U_FAILURE(status)) {
return str;
}
icu::UnicodeString dest;
const std::u16string rstr = km::kbp::kmx::u32string_to_u16string(str);
icu::UnicodeString src = icu::UnicodeString(rstr.data(), (int32_t)rstr.length());
nfc->normalize(src, dest, status);
if (U_FAILURE(status)) {
return str;
}

UErrorCode preflightStatus = U_ZERO_ERROR;
// calculate how big the buffer is
auto out32len = dest.toUTF32(nullptr, 0, preflightStatus); // preflightStatus will be an err, because we know the buffer overruns zero bytes
// allocate
char32_t *s = new char32_t[out32len + 1];
assert(s != nullptr);
// convert
dest.toUTF32((UChar32 *)s, out32len + 1, status);
assert(U_SUCCESS(status));
str.assign(s, out32len);
delete [] s;
return str;
}

// TODO-LDML: copypasta -> refactor
std::u16string &normalize_nfc(std::u16string &str, UErrorCode &status) {
const icu::Normalizer2 *nfc = icu::Normalizer2::getNFCInstance(status);
if (U_FAILURE(status)) {
return str;
}
icu::UnicodeString dest;
icu::UnicodeString src = icu::UnicodeString(str.data(), (int32_t)str.length());
nfc->normalize(src, dest, status);
if (U_FAILURE(status)) {
return str;
}
str.assign(dest.getBuffer(), dest.length());
return str;
}




} // namespace ldml
Expand Down
4 changes: 4 additions & 0 deletions core/src/ldml/ldml_transforms.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -277,6 +277,10 @@ class transforms {
std::u32string &normalize_nfd(std::u32string &str, UErrorCode &status);
/** Normalize a u16string inplace. Returns a reference to the same string. */
std::u16string &normalize_nfd(std::u16string &str, UErrorCode &status);
/** Normalize a u32string inplace. Returns a reference to the same string. */
std::u32string &normalize_nfc(std::u32string &str, UErrorCode &status);
/** Normalize a u16string inplace. Returns a reference to the same string. */
std::u16string &normalize_nfc(std::u16string &str, UErrorCode &status);

} // namespace ldml
} // namespace kbp
Expand Down
2 changes: 1 addition & 1 deletion core/tests/unit/ldml/keyboards/k_003_transform.xml
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from https://github.com/unicode-org/cldr/blob/keyboard-preview/docs/ldml/tr35-keyboards.md#element-transform
@@keys: [K_Q][K_U][K_BKQUOTE][K_E]
@@expected: que\u0302
@@expected: qu\u00EA
-->
<!DOCTYPE keyboard3 SYSTEM "../../../../../resources/standards-data/ldml-keyboards/techpreview/dtd/ldmlKeyboard3.dtd">
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
</test>
<test name="regex-test-0-c">
<startContext to="" />
<keystroke key="u-00e0" />
<keystroke key="u-00e8" />
<keystroke key="u-0320" />
<check result="e\u{0320}\u{0300}" />
</test>
Expand Down
12 changes: 6 additions & 6 deletions core/tests/unit/ldml/keyboards/k_008_transform_norm.xml
Original file line number Diff line number Diff line change
Expand Up @@ -13,11 +13,11 @@ from https://github.com/unicode-org/cldr/blob/keyboard-preview/docs/ldml/tr35-ke
</names>

<keys>
<key id="u-0320" to="\u{0320}" />
<key id="u-0300" to="\u{0300}" />
<key id="u-00e0" to="\u{00e0}" />
<key id="u-0320" to="\u{0320}" /> <!-- ◌̠ -->
<key id="u-0300" to="\u{0300}" /> <!-- ◌̀ -->
<key id="u-00e8" to="\u{00e8}" /> <!-- è -->
<key id="nfd" to="e\u{0320}\u{0300}" />
<key id="nfc" to="\u{00E0}\u{0320}" />
<key id="nfc" to="\u{00E8}\u{0320}" />
<key id="not-nfd" to="e\u{0300}\u{0320}" />
</keys>

Expand All @@ -30,7 +30,7 @@ from https://github.com/unicode-org/cldr/blob/keyboard-preview/docs/ldml/tr35-ke
<row keys="space" />
</layer>
<layer modifier="shift" id="shift">
<row keys="u-0300 u-00e0 nfd nfc not-nfd" />
<row keys="u-0300 u-00e8 nfd nfc not-nfd" />
</layer>
</layers>

Expand All @@ -46,7 +46,7 @@ from https://github.com/unicode-org/cldr/blob/keyboard-preview/docs/ldml/tr35-ke
<!-- these SHOULD match -->
<transform from="1e\u{0300}\u{0320}" to="YES1" /> <!-- out of order but should get normalized -->
<transform from="2e\u{0320}\u{0300}" to="YES2" /> <!-- in order NFD -->
<transform from="3\u{00E0}\u{0320}" to="YES3" /> <!-- in order NFC -->
<transform from="3\u{00E8}\u{0320}" to="YES3" /> <!-- in order NFC -->
</transformGroup>
</transforms>
</keyboard3>
Loading

0 comments on commit a577bd2

Please sign in to comment.