Fix #107 – postgenerator fails when ~ in middle of wblanked word

There was already support for [[w:text:foo]]~zzy[[/]] but not for [[w:text:foo]]xy~zzy[[/]] This change should keep the old feature of moving wblank-start after the post-generated part of [[w:text:foo]]~zzy[[/]] (so if there was a rule turning ~z into Z we get Z[[w:text:foo]]zy[[/]]), but if the wake-up-mark ~ is seen later in the wblank, it stays surrounded, so [[w:text:foo]]xy~zzy[[/]] turns into [[w:text:foo]]xyZzy[[/]]. + tests for #107
apertium · Jun 11, 2021 · b99925c · b99925c
1 parent de5983a
commit b99925c
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 3 deletions.
diff --git a/lttoolbox/fst_processor.cc b/lttoolbox/fst_processor.cc
@@ -280,11 +280,28 @@ FSTProcessor::wblankPostGen(FILE *input, FILE *output)
   wstring result = L"";
   result += L"[[";
   wchar_t c = 0;
+  bool in_content = false;
 
   while(!feof(input))
   {
     c = static_cast<wchar_t>(fgetwc_unlocked(input));
-    result += c;
+
+    if(in_content && c == L'~')
+    {
+      if(result[result.size()-1] == L']') {
+        // We just saw the end of a wblank, may want to merge
+        wblankqueue.push(result);
+      }
+      else {
+        // wake-up-mark happened some characters into the wblanked word
+        fputws(result.c_str(), output);
+      }
+      return true;
+    }
+    else
+    {
+      result += c;
+    }
 
     if(c == L'\\')
     {
@@ -315,6 +332,7 @@ FSTProcessor::wblankPostGen(FILE *input, FILE *output)
           {
             result += c;
           }
+          in_content = true; // Assumption: No nested wblanks, always balanced
         }
       }
     }

diff --git a/tests/data/postgen.dix b/tests/data/postgen.dix
@@ -92,6 +92,8 @@
         </p>
       </e>
 
+    <e><p><l><a/>sss</l><r>ss</r></p></e>
+
   </section>
 
 </dictionary>

diff --git a/tests/lt_proc/__init__.py b/tests/lt_proc/__init__.py
@@ -173,7 +173,9 @@ class PostgenerationWordboundBlankTest(ProcTest):
                         "[[t:b:Z9eiLA]]abc[[/]] [[t:i:123456]]~les[[/]] [[t:i:4_tPUA]]~de[[/]] [[t:b:Z9eiLA]]el[[/]] [[t:i:wSM6RQ]]testword[[/]]",
                         "[[t:b:Z9eiLA]]abc[[/]] [[t:i:123456]]~les pes[[/]] [[t:i:4_tPUA]]~de[[/]] [[t:b:Z9eiLA]]el[[/]] [[t:i:wSM6RQ]]testword[[/]]",
                         "[[t:b:Z9eiLA]]abc[[/]] [[t:i:123456]]~les[[/]] [[t:b:12bsa23]]pes[[/]] [[t:i:4_tPUA]]~de[[/]] [[t:b:Z9eiLA]]el[[/]] [[t:i:wSM6RQ]]testword[[/]]",
-                        "[[t:b:Z9eiLA]]abc[[/]] ~les [[t:b:12bsa23]]pes[[/]] [[t:i:4_tPUA]]~de[[/]] [[t:b:Z9eiLA]]el[[/]] [[t:i:wSM6RQ]]testword[[/]]"]
+                        "[[t:b:Z9eiLA]]abc[[/]] ~les [[t:b:12bsa23]]pes[[/]] [[t:i:4_tPUA]]~de[[/]] [[t:b:Z9eiLA]]el[[/]] [[t:i:wSM6RQ]]testword[[/]]",
+                        "[[t:text:NaNaNa]]pla~ssar[[/]]",
+                        "[[t:text:NaNaNa]]pla~sssar[[/]]"]
 
     expectedOutputs = [ "xyz ejemplo [[t:i:123456; t:b:abc123; t:i:123456]]u ho[[/]] [[t:b:iopmnb]]nombre[[/]].",
                         "xyz ejemplo [[t:b:poim230]]u ho[[/]] [[t:i:mnbj203]]nombre[[/]].",
@@ -193,7 +195,9 @@ class PostgenerationWordboundBlankTest(ProcTest):
                         "[[t:b:Z9eiLA]]abc[[/]] [[t:i:123456]]le pe test[[/]] [[t:i:4_tPUA; t:b:Z9eiLA]]del[[/]] [[t:i:wSM6RQ]]testword[[/]]",
                         "[[t:b:Z9eiLA]]abc[[/]] [[t:i:123456]]les pes test[[/]] [[t:i:4_tPUA; t:b:Z9eiLA]]del[[/]] [[t:i:wSM6RQ]]testword[[/]]",
                         "[[t:b:Z9eiLA]]abc[[/]] [[t:i:123456; t:b:12bsa23]]les pes test[[/]] [[t:i:4_tPUA; t:b:Z9eiLA]]del[[/]] [[t:i:wSM6RQ]]testword[[/]]",
-                        "[[t:b:Z9eiLA]]abc[[/]] [[t:b:12bsa23]]les pes test[[/]] [[t:i:4_tPUA; t:b:Z9eiLA]]del[[/]] [[t:i:wSM6RQ]]testword[[/]]"]
+                        "[[t:b:Z9eiLA]]abc[[/]] [[t:b:12bsa23]]les pes test[[/]] [[t:i:4_tPUA; t:b:Z9eiLA]]del[[/]] [[t:i:wSM6RQ]]testword[[/]]",
+                        "[[t:text:NaNaNa]]plassar[[/]]",
+                        "[[t:text:NaNaNa]]plassar[[/]]"]
 
 
 class PostgenerationWordboundBlankEscapingTest(ProcTest):