public IEnumerable <TranslationResult> Translate(int n, IReadOnlyList <string> segment)
{
IEnumerable <IEnumerable <WordAnalysis> > sourceAnalyses = segment
.Select(word => _sourceAnalyzer.AnalyzeWord(word));
foreach (TransferResult transferResult in _transferer.Transfer(sourceAnalyses).Take(n))
{
IReadOnlyList <WordAnalysis> targetAnalyses = transferResult.TargetAnalyses;
WordAlignmentMatrix waMatrix = transferResult.WordAlignmentMatrix;
var translation = new List <string>();
var confidences = new List <double>();
var sources = new List <TranslationSources>();
var alignment = new WordAlignmentMatrix(segment.Count, targetAnalyses.Count);
double confidence = double.MaxValue;
for (int j = 0; j < targetAnalyses.Count; j++)
{
int[] sourceIndices = Enumerable.Range(0, waMatrix.RowCount)
.Where(i => waMatrix[i, j]).ToArray();
string targetWord = targetAnalyses[j].IsEmpty
? null
: _targetGenerator.GenerateWords(targetAnalyses[j]).FirstOrDefault();
double wordConfidence = 1.0;
TranslationSources source = TranslationSources.Transfer;
if (targetWord == null)
{
if (sourceIndices.Length > 0)
{
int i = sourceIndices[0];
targetWord = segment[i];
wordConfidence = 0;
source = TranslationSources.None;
alignment[i, j] = true;
}
}
else
{
foreach (int i in sourceIndices)
{
alignment[i, j] = true;
}
}
if (targetWord != null)
{
translation.Add(targetWord);
confidences.Add(wordConfidence);
sources.Add(source);
confidence = Math.Min(confidence, wordConfidence);
}
}
yield return(new TranslationResult(segment, translation, confidences, sources, alignment,
new[] { new Phrase(Range <int> .Create(0, segment.Count), translation.Count, confidence) }));
}
}
public static void AddGeneratedWords(this IMorphologicalGenerator targetGenerator, WordAnalysis analysis,
params string[] words)
{
targetGenerator.GenerateWords(Arg.Is(analysis)).Returns(words);
}