Mirrors/SubtitleEdit
1
0
Fork 0
mirror of https://github.com/SubtitleEdit/subtitleedit.git synced 2024-11-25 04:33:04 +01:00
Code Issues Projects Releases Wiki Activity

Improve post processor a little

Browse Source
This commit is contained in:
Nikolaj Olsson 2024-04-05 13:14:28 +02:00
parent 440312fa54
commit 85a8076098
1 changed files with 43 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
src/libse/AudioToText/AudioToTextPostProcessor.cs
View File

@ -169,8 +169,8 @@ namespace Nikse.SubtitleEdit.Core.AudioToText
continue;
}
// check for period in last part of p
var lastPeriodIdx = p.Text.LastIndexOfAny(new char[] { '.', '?', '!' });
// check for period in last part of current
var lastPeriodIdx = p.Text.LastIndexOfAny(new[] { '.', '?', '!' });
if (lastPeriodIdx > 3 && lastPeriodIdx > p.Text.Length - maxMoveChunkSize)
{
var newCurrentText = p.Text.Substring(0, lastPeriodIdx + 1).Trim();
@ -189,6 +189,8 @@ namespace Nikse.SubtitleEdit.Core.AudioToText
if (currentOk && nextOk && allOk)
{
var oldP = new Paragraph(p);
p.Text = newCurrentText;
next.Text = newNextText;
@ -199,7 +201,9 @@ namespace Nikse.SubtitleEdit.Core.AudioToText
}
else
{
//TODO: calc time
var durationToMove = CalcDurationToMove(oldP, p, next);
p.EndTime.TotalMilliseconds += durationToMove;
next.StartTime.TotalMilliseconds += durationToMove;
}
continue;
@ -207,7 +211,7 @@ namespace Nikse.SubtitleEdit.Core.AudioToText
}
// check for period in beginning of next
var firstPeriodIdx = next.Text.IndexOfAny(new char[] { '.', '?', '!' });
var firstPeriodIdx = next.Text.IndexOfAny(new[] { '.', '?', '!' });
if (firstPeriodIdx >= 3 && firstPeriodIdx < maxMoveChunkSize)
{
var newCurrentText = next.Text.Substring(0, firstPeriodIdx + 1).Trim();
@ -226,6 +230,8 @@ namespace Nikse.SubtitleEdit.Core.AudioToText
if (currentOk && nextOk && allOk)
{
var oldP = new Paragraph(p);
p.Text = newCurrentText;
next.Text = newNextText;
@ -236,7 +242,9 @@ namespace Nikse.SubtitleEdit.Core.AudioToText
}
else
{
//TODO: calc time
var durationToMove = CalcDurationToMove(oldP, p, next);
p.EndTime.TotalMilliseconds += durationToMove;
next.StartTime.TotalMilliseconds += durationToMove;
}
}
}
@ -372,9 +380,15 @@ namespace Nikse.SubtitleEdit.Core.AudioToText
arr = text.SplitToLines();
if (arr.Count == 2)
{
var oldP = new Paragraph(p);
p.Text = Utilities.AutoBreakLine(arr[0], language);
next.Text = Utilities.AutoBreakLine(arr[1], language);
//TODO: calc time
var durationToMove = CalcDurationToMove(oldP, p, next);
p.EndTime.TotalMilliseconds += durationToMove;
next.StartTime.TotalMilliseconds += durationToMove;
splitDone = true;
}
@ -465,8 +479,14 @@ namespace Nikse.SubtitleEdit.Core.AudioToText
}
else
{
var oldP = new Paragraph(p);
p.Text = Utilities.AutoBreakLine(arr[0], language);
next.Text = Utilities.AutoBreakLine(arr[1], language);
var durationToMove = CalcDurationToMove(oldP, p, next);
p.EndTime.TotalMilliseconds += durationToMove;
next.StartTime.TotalMilliseconds += durationToMove;
}
}
}
@ -480,6 +500,22 @@ namespace Nikse.SubtitleEdit.Core.AudioToText
return s;
}
private static double CalcDurationToMove(Paragraph oldCurrent, Paragraph current, Paragraph next)
{
if (current.DurationTotalMilliseconds < 0 || next.DurationTotalMilliseconds < 0)
{
return 0;
}
var totalDuration = current.DurationTotalMilliseconds + next.DurationTotalMilliseconds;
var totalChars = current.Text.Length + next.Text.Length;
var durChar = totalDuration / totalChars;
var diffLength = current.Text.Length - oldCurrent.Text.Length;
var result = durChar * diffLength;
return result;
}
private bool IsNextCloseAndAlone(Paragraph p, Paragraph next, Paragraph nextNext, int maxMillisecondsBetweenLines, bool onlyContinuousLines)
{
if (nextNext == null || next.Text.Length > 12)