Add word alignment to hybrid engine

sillsdev · Nov 5, 2024 · 08a2719 · 08a2719
1 parent 16fb76d
commit 08a2719
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 2 deletions.
diff --git a/src/SIL.Machine/Translation/HybridTranslationEngine.cs b/src/SIL.Machine/Translation/HybridTranslationEngine.cs
@@ -401,6 +401,49 @@ public void TrainSegment(
             InteractiveEngine.TrainSegment(sourceSegment, targetSegment, sentenceStart);
         }
 
+        public TranslationResult GetBestPhraseAlignment(
+            IReadOnlyList<string> sourceSegment,
+            IReadOnlyList<string> targetSegment
+        )
+        {
+            CheckDisposed();
+
+            return InteractiveEngine.GetBestPhraseAlignment(sourceSegment, targetSegment);
+        }
+
+        public TranslationResult GetBestPhraseAlignment(string sourceSegment, string targetSegment)
+        {
+            CheckDisposed();
+
+            return InteractiveEngine.GetBestPhraseAlignment(sourceSegment, targetSegment);
+        }
+
+        async Task<TranslationResult> IWordAlignerEngine.GetBestPhraseAlignmentAsync(
+            string sourceSegment,
+            string targetSegment,
+            CancellationToken cancellationToken
+        )
+        {
+            CheckDisposed();
+
+            return await InteractiveEngine
+                .GetBestPhraseAlignmentAsync(sourceSegment, targetSegment, cancellationToken)
+                .ConfigureAwait(false);
+        }
+
+        public async Task<TranslationResult> GetBestPhraseAlignmentAsync(
+            IReadOnlyList<string> sourceSegment,
+            IReadOnlyList<string> targetSegment,
+            CancellationToken cancellationToken = default
+        )
+        {
+            CheckDisposed();
+
+            return await InteractiveEngine
+                .GetBestPhraseAlignmentAsync(sourceSegment, targetSegment, cancellationToken)
+                .ConfigureAwait(false);
+        }
+
         private TranslationResult Merge(TranslationResult interactiveResult, TranslationResult ruleResult)
         {
             var mergedTargetSegment = new List<string>();

diff --git a/src/SIL.Machine/Translation/IInteractiveTranslationEngine.cs b/src/SIL.Machine/Translation/IInteractiveTranslationEngine.cs
@@ -4,7 +4,7 @@
 
 namespace SIL.Machine.Translation
 {
-    public interface IInteractiveTranslationEngine : ITranslationEngine
+    public interface IInteractiveTranslationEngine : ITranslationEngine, IWordAlignerEngine
     {
         Task<WordGraph> GetWordGraphAsync(string segment, CancellationToken cancellationToken = default);
 

diff --git a/src/SIL.Machine/Translation/IInteractiveTranslationModel.cs b/src/SIL.Machine/Translation/IInteractiveTranslationModel.cs
@@ -3,7 +3,7 @@
 
 namespace SIL.Machine.Translation
 {
-    public interface IInteractiveTranslationModel : IInteractiveTranslationEngine, IWordAlignerEngine, ITranslationModel
+    public interface IInteractiveTranslationModel : IInteractiveTranslationEngine, ITranslationModel
     {
         Task SaveAsync(CancellationToken cancellationToken = default);
         void Save();