SubtitleEdit/libse/SubtitleFormats/UnknownSubtitle70.cs
2016-02-08 21:11:03 +01:00

566 lines
29 KiB
C#

using System;
using System.Collections.Generic;
using System.Text;
using System.Text.RegularExpressions;
namespace Nikse.SubtitleEdit.Core.SubtitleFormats
{
public class UnknownSubtitle70 : SubtitleFormat
{
private static readonly Regex RegexMicroDvdLine = new Regex(@"^\[-?\d+\]\[-?\d+\].*$", RegexOptions.Compiled);
public string Errors { get; private set; }
private StringBuilder _errors;
private int _lineNumber;
public override string Extension
{
get { return ".txt"; }
}
public override string Name
{
get { return "Unknown 70"; }
}
public override bool IsTimeBased
{
get { return false; }
}
public override bool IsMine(List<string> lines, string fileName)
{
var trimmedLines = new List<string>();
int errors = 0;
foreach (string line in lines)
{
if (string.IsNullOrWhiteSpace(line))
{
continue;
}
if (line.Contains('['))
{
string s = RemoveIllegalSpacesAndFixEmptyCodes(line);
if (RegexMicroDvdLine.IsMatch(s))
trimmedLines.Add(s);
else
errors++;
}
else
{
errors++;
}
}
Errors = null;
return trimmedLines.Count > errors;
}
private static string RemoveIllegalSpacesAndFixEmptyCodes(string line)
{
int index = line.IndexOf(']');
if (index >= 0 && index < line.Length)
{
index = line.IndexOf(']', index + 1);
if (index >= 0 && index + 1 < line.Length)
{
var indexOfBrackets = line.IndexOf("[]", StringComparison.Ordinal);
if (indexOfBrackets >= 0 && indexOfBrackets < index)
{
line = line.Insert(indexOfBrackets + 1, "0"); // set empty time codes to zero
index++;
}
while (line.Contains(' ') && line.IndexOf(' ') < index)
{
line = line.Remove(line.IndexOf(' '), 1);
index--;
}
}
}
return line;
}
public override string ToText(Subtitle subtitle, string title)
{
var sb = new StringBuilder();
foreach (Paragraph p in subtitle.Paragraphs)
{
sb.Append('[');
sb.Append(p.StartFrame);
sb.Append("][");
sb.Append(p.EndFrame);
sb.Append(']');
//{y:b} is italics for single line
//{Y:b} is italics for both lines
var parts = p.Text.SplitToLines();
int count = 0;
bool italicOn = false;
bool boldOn = false;
bool underlineOn = false;
var lineSb = new StringBuilder();
foreach (string line in parts)
{
if (count > 0)
lineSb.Append('|');
bool alreadyItalic = italicOn;
bool alreadyBold = boldOn;
bool alreadyUnderline = underlineOn;
var pre = new StringBuilder();
string s = line;
for (int i = 0; i < 5; i++)
{
if (alreadyItalic && i == 0)
{
italicOn = true;
boldOn = false;
underlineOn = false;
pre.Append("{y:i}"); // italic single line
alreadyItalic = false;
}
else if (s.StartsWith("<i>"))
{
italicOn = true;
boldOn = false;
underlineOn = false;
if (pre.ToString() != "{y:i}")
pre.Append("{y:i}"); // italic single line
s = s.Remove(0, 3);
}
else if (alreadyBold && i == 0)
{
italicOn = false;
boldOn = true;
underlineOn = false;
pre.Append("{y:b}"); // bold single line
}
else if (s.StartsWith("<b>"))
{
italicOn = false;
boldOn = true;
underlineOn = false;
if (pre.ToString() != "{y:b}")
pre.Append("{y:b}"); // bold single line
s = s.Remove(0, 3);
}
else if (alreadyUnderline && i == 0)
{
italicOn = false;
boldOn = false;
underlineOn = true;
pre.Append("{y:u}"); // underline single line
}
else if (s.StartsWith("<u>"))
{
italicOn = false;
boldOn = false;
underlineOn = true;
if (pre.ToString() != "{y:u}")
pre.Append("{y:u}"); // underline single line
s = s.Remove(0, 3);
}
if (s.StartsWith("<font "))
{
int end = s.IndexOf('>');
if (end > 0)
{
string tag = s.Substring(0, end);
if (tag.Contains(" color="))
{
int colorStart = tag.IndexOf(" color=", StringComparison.Ordinal);
int colorEnd = tag.IndexOf('"', colorStart + " color=".Length + 1);
if (colorEnd > 0)
{
string color = tag.Substring(colorStart, colorEnd - colorStart);
color = color.Remove(0, " color=".Length);
color = color.Trim('"');
color = color.Trim('\'');
color = color.TrimStart('#');
if (color.Length == 6)
{
if (s.Contains(Environment.NewLine) && s.Contains("</font>" + Environment.NewLine))
pre.Append("{c:$" + color.Substring(4, 2) + color.Substring(2, 2) + color.Substring(0, 2) + "}");
else
pre.Append("{C:$" + color.Substring(4, 2) + color.Substring(2, 2) + color.Substring(0, 2) + "}");
}
}
}
if (tag.Contains(" face="))
{
var faceStart = tag.IndexOf(" face=", StringComparison.Ordinal);
var faceEnd = tag.IndexOf('"', faceStart + " face=".Length + 1);
if (faceEnd > 0)
{
string fontName = tag.Substring(faceStart, faceEnd - faceStart);
fontName = fontName.Remove(0, " face=".Length).Trim();
fontName = fontName.Trim('"');
fontName = fontName.Trim('\'');
if (fontName.Length > 0)
{
if (s.Contains(Environment.NewLine) && s.Contains("</font>" + Environment.NewLine))
pre.Append("{f:" + fontName + "}");
else
pre.Append("{F:" + fontName + "}");
}
}
}
if (tag.Contains(" size="))
{
var sizeStart = tag.IndexOf(" size=", StringComparison.Ordinal);
var sizeEnd = tag.IndexOf('"', sizeStart + " size=".Length + 1);
if (sizeEnd > 0)
{
string fontSize = tag.Substring(sizeStart, sizeEnd - sizeStart);
fontSize = fontSize.Remove(0, " size=".Length).Trim();
fontSize = fontSize.Trim('"');
fontSize = fontSize.Trim('\'');
if (fontSize.Length > 0)
{
if (s.Contains(Environment.NewLine) && s.Contains("</font>" + Environment.NewLine))
pre.Append("{s:" + fontSize + "}");
else
pre.Append("{S:" + fontSize + "}");
}
}
}
s = s.Remove(0, end + 1);
}
}
}
if (s.Contains("</i>"))
italicOn = false;
if (s.Contains("</b>"))
boldOn = false;
if (s.Contains("</u>"))
underlineOn = false;
lineSb.Append(HtmlUtil.RemoveHtmlTags(pre + line));
count++;
}
string text = lineSb.ToString();
int noOfLines = Utilities.CountTagInText(text, '|') + 1;
if (Utilities.CountTagInText(text, "{y:i}") == noOfLines && text.StartsWith("{y:i}"))
text = "{Y:i}" + text.Replace("{y:i}", string.Empty);
else if (Utilities.CountTagInText(text, "{y:b}") == noOfLines && text.StartsWith("{y:b}"))
text = "{Y:b}" + text.Replace("{y:b}", string.Empty);
else if (Utilities.CountTagInText(text, "{y:u}") == noOfLines && text.StartsWith("{y:u}"))
text = "{Y:u}" + text.Replace("{y:u}", string.Empty);
else if (Utilities.CountTagInText(text, "{y:u}{y:i}") == noOfLines && text.StartsWith("{y:u}{y:i}"))
text = "{Y:u}{Y:i}" + text.Replace("{y:u}", string.Empty).Replace("{y:i}", string.Empty);
else if (Utilities.CountTagInText(text, "{y:i}{y:u}") == noOfLines && text.StartsWith("{y:i}{y:u}"))
text = "{Y:i}{Y:u}" + text.Replace("{y:i}", string.Empty).Replace("{y:u}", string.Empty);
else if (Utilities.CountTagInText(text, "{y:i}{y:b}") == noOfLines && text.StartsWith("{y:i}{y:b}"))
text = "{Y:i}{Y:b}" + text.Replace("{y:i}", string.Empty).Replace("{y:b}", string.Empty);
else if (Utilities.CountTagInText(text, "{y:b}{y:i}") == noOfLines && text.StartsWith("{y:b}{y:i}"))
text = "{Y:b}{Y:i}" + text.Replace("{y:i}", string.Empty).Replace("{y:b}", string.Empty);
else if (Utilities.CountTagInText(text, "{y:b}{y:u}") == noOfLines && text.StartsWith("{y:b}{y:u}"))
text = "{Y:b}{Y:u}" + text.Replace("{y:b}", string.Empty).Replace("{y:u}", string.Empty);
else if (Utilities.CountTagInText(text, "{y:u}{y:b}") == noOfLines && text.StartsWith("{y:u}{y:b}"))
text = "{Y:u}{Y:b}" + text.Replace("{y:u}", string.Empty).Replace("{y:b}", string.Empty);
if (Utilities.CountTagInText(text, "{y:i}") == 1 && noOfLines == 1)
text = text.Replace("{y:i}", "{Y:i}");
if (Utilities.CountTagInText(text, "{y:b}") == 1 && noOfLines == 1)
text = text.Replace("{y:b}", "{Y:b}");
if (Utilities.CountTagInText(text, "{y:u}") == 1 && noOfLines == 1)
text = text.Replace("{y:u}", "{Y:u}");
sb.AppendLine(HtmlUtil.RemoveHtmlTags(text));
}
return sb.ToString().Trim();
}
public override void LoadSubtitle(Subtitle subtitle, List<string> lines, string fileName)
{
_errorCount = 0;
_errors = new StringBuilder();
Errors = null;
_lineNumber = 0;
foreach (string line in lines)
{
_lineNumber++;
string s = RemoveIllegalSpacesAndFixEmptyCodes(line);
if (RegexMicroDvdLine.IsMatch(s))
{
try
{
int textIndex = GetTextStartIndex(s);
if (textIndex < s.Length)
{
string text = s.Substring(textIndex);
string temp = s.Substring(0, textIndex - 1);
string[] frames = temp.Replace("][", ":").Replace("[", string.Empty).Replace("]", string.Empty).Split(':');
int startFrame = int.Parse(frames[0]);
int endFrame = int.Parse(frames[1]);
string post = string.Empty;
string[] parts = text.Split(new[] { '|' }, StringSplitOptions.RemoveEmptyEntries);
int count = 0;
var lineSb = new StringBuilder();
foreach (string s2 in parts)
{
if (count > 0)
lineSb.AppendLine();
s = s2.Trim();
var pre = new StringBuilder();
string singlePost = string.Empty;
for (int i = 0; i < 5; i++)
{
if (s.StartsWith("{Y:i}"))
{
s = s.Remove(0, 5);
pre.Append("<i>");
post = "</i>" + post;
}
else if (s.StartsWith("{Y:b}"))
{
s = s.Remove(0, 5);
pre.Append("<b>");
post = "</b>" + post;
}
else if (s.StartsWith("{Y:u}"))
{
s = s.Remove(0, 5);
pre.Append("<u>");
post = "</u>" + post;
}
else if (s.StartsWith("{y:i}"))
{
s = s.Remove(0, 5);
singlePost = "</i>" + singlePost;
pre.Append("<i>");
}
else if (s.StartsWith("{y:b}"))
{
s = s.Remove(0, 5);
singlePost = "</b>" + singlePost;
pre.Append("<b>");
}
else if (s.StartsWith("{y:u}"))
{
s = s.Remove(0, 5);
singlePost = "</u>" + singlePost;
pre.Append("<u>");
}
else if (s.StartsWith("{y:b,u}"))
{
s = s.Remove(0, 7);
singlePost = "</u></b>" + singlePost;
pre.Append("<b><u>");
}
else if (s.StartsWith("{y:u,b}"))
{
s = s.Remove(0, 7);
singlePost = "</b></u>" + singlePost;
pre.Append("<u><b>");
}
else if (s.StartsWith("{y:b,i}"))
{
s = s.Remove(0, 7);
singlePost = "</i></b>" + singlePost;
pre.Append("<b><i>");
}
else if (s.StartsWith("{y:i,b}"))
{
s = s.Remove(0, 7);
singlePost = "</b></i>" + singlePost;
pre.Append("<i><b>");
}
else if (s.StartsWith("{y:i,u}"))
{
s = s.Remove(0, 7);
singlePost = "</u></i>" + singlePost;
pre.Append("<i><u>");
}
else if (s.StartsWith("{y:u,i}"))
{
s = s.Remove(0, 7);
singlePost = "</i></u>" + singlePost;
pre.Append("<><i>");
}
else if (s.StartsWith("{Y:b,u}") || s.StartsWith("{Y:u,b}"))
{
s = s.Remove(0, 7);
pre.Append("<b><u>");
post = "</u></b>" + post;
}
else if (s.StartsWith("{Y:b,i}") || s.StartsWith("{Y:i,b}"))
{
s = s.Remove(0, 7);
pre.Append("<b><i>");
post = "</i></b>" + post;
}
else if (s.StartsWith("{Y:i,u}") || s.StartsWith("{Y:u,i}"))
{
s = s.Remove(0, 7);
pre.Append("<i><u>");
post = "</u></i>" + post;
}
else if (s.Contains("{c:$"))
{
int start = s.IndexOf("{c:$", StringComparison.Ordinal);
int end = s.IndexOf('}', start);
if (end > start)
{
string tag = s.Substring(start, end - start);
tag = tag.Remove(0, 4);
if (tag.Length == 6)
{
pre.Append("<font color=\"#" + tag.Substring(4, 2) + tag.Substring(2, 2) + tag.Substring(0, 2) + "\">");
s = s.Remove(start, end - start + 1);
singlePost = "</font>" + singlePost;
}
}
}
else if (s.Contains("{C:$")) // uppercase=all lines
{
int start = s.IndexOf("{C:$", StringComparison.Ordinal);
int end = s.IndexOf('}', start);
if (end > start)
{
string tag = s.Substring(start, end - start);
tag = tag.Remove(0, 4);
if (tag.Length == 6)
{
pre.Append("<font color=\"#" + tag.Substring(4, 2) + tag.Substring(2, 2) + tag.Substring(0, 2) + "\">");
s = s.Remove(start, end - start + 1);
post = "</font>" + post;
}
}
}
else if (s.Contains("{f:"))
{
int start = s.IndexOf("{f:", StringComparison.Ordinal);
int end = s.IndexOf('}', start);
if (end > start)
{
string tag = s.Substring(start, end - start);
tag = tag.Remove(0, 3).Trim();
if (tag.Length > 0)
{
pre.Append("<font face=\"" + tag + "\">");
s = s.Remove(start, end - start + 1);
singlePost = "</font>" + singlePost;
}
}
}
else if (s.Contains("{F:")) // uppercase=all lines
{
int start = s.IndexOf("{F:", StringComparison.Ordinal);
int end = s.IndexOf('}', start);
if (end > start)
{
string tag = s.Substring(start, end - start);
tag = tag.Remove(0, 3).Trim();
if (tag.Length > 0)
{
pre.Append("<font face=\"" + tag + "\">");
s = s.Remove(start, end - start + 1);
post = "</font>" + post;
}
}
}
else if (s.Contains("{s:"))
{
int start = s.IndexOf("{s:", StringComparison.Ordinal);
int end = s.IndexOf('}', start);
if (end > start)
{
string tag = s.Substring(start, end - start);
tag = tag.Remove(0, 3).Trim();
if (tag.Length > 0)
{
pre.Append("<font size=\"" + tag + "\">");
s = s.Remove(start, end - start + 1);
singlePost = "</font>" + singlePost;
}
}
}
else if (s.Contains("{S:")) // uppercase=all lines
{
int start = s.IndexOf("{S:", StringComparison.Ordinal);
int end = s.IndexOf('}', start);
if (end > start)
{
string tag = s.Substring(start, end - start);
tag = tag.Remove(0, 3).Trim();
if (tag.Length > 0)
{
pre.Append("<font size=\"" + tag + "\">");
s = s.Remove(start, end - start + 1);
post = "</font>" + post;
}
}
}
}
s = s.Replace("{Y:i}", string.Empty).Replace("{y:i}", string.Empty);
s = s.Replace("{Y:b}", string.Empty).Replace("{y:b}", string.Empty);
s = s.Replace("{Y:u}", string.Empty).Replace("{y:u}", string.Empty);
lineSb.Append(pre + s + singlePost);
count++;
}
text = lineSb + post;
subtitle.Paragraphs.Add(new Paragraph(startFrame, endFrame, text));
}
}
catch
{
_errorCount++;
if (_errors.Length < 2000)
_errors.AppendLine(string.Format(Configuration.Settings.Language.Main.LineNumberXErrorReadingFromSourceLineY, _lineNumber, line));
}
}
else
{
_errorCount++;
if (_errors.Length < 2000)
_errors.AppendLine(string.Format(Configuration.Settings.Language.Main.LineNumberXErrorReadingFromSourceLineY, _lineNumber, line));
}
}
int j = 0;
foreach (Paragraph p in subtitle.Paragraphs)
{
Paragraph previous = subtitle.GetParagraphOrDefault(j - 1);
if (p.StartFrame == 0 && previous != null)
{
p.StartFrame = previous.EndFrame + 1;
}
if (p.EndFrame == 0)
{
p.EndFrame = p.StartFrame;
}
j++;
}
subtitle.Renumber();
Errors = _errors.ToString();
}
private static int GetTextStartIndex(string line)
{
int i = 0;
int tagCount = 0;
while (i < line.Length && tagCount < 4)
{
if (line[i] == '[' || line[i] == ']')
tagCount++;
i++;
}
return i;
}
}
}