SubtitleEdit/libse/SubtitleFormats/ItunesTimedText.cs
2019-08-16 22:55:32 +02:00

377 lines
18 KiB
C#

using System;
using System.Collections.Generic;
using System.Xml;
namespace Nikse.SubtitleEdit.Core.SubtitleFormats
{
/// <summary>
/// Crappy format... should always be saved as UTF-8 without BOM (hacked Main.cs) and <br /> tags should be oldstyle <br/>
/// </summary>
public class ItunesTimedText : TimedText10
{
public override string Extension => ".itt";
public new const string NameOfFormat = "iTunes Timed Text";
public override string Name => NameOfFormat;
public override bool IsMine(List<string> lines, string fileName)
{
if (fileName != null && !fileName.EndsWith(Extension, StringComparison.OrdinalIgnoreCase))
{
return false;
}
if (new NetflixTimedText().IsMine(lines, fileName))
{
return false;
}
return base.IsMine(lines, fileName);
}
public override string ToText(Subtitle subtitle, string title)
{
XmlNode styleHead = null;
bool convertedFromSubStationAlpha = false;
if (subtitle.Header != null)
{
try
{
var x = new XmlDocument();
x.LoadXml(subtitle.Header);
var xnsmgr = new XmlNamespaceManager(x.NameTable);
xnsmgr.AddNamespace("ttml", "http://www.w3.org/ns/ttml");
styleHead = x.DocumentElement.SelectSingleNode("ttml:head", xnsmgr);
}
catch
{
styleHead = null;
}
if (styleHead == null && (subtitle.Header.Contains("[V4+ Styles]") || subtitle.Header.Contains("[V4 Styles]")))
{
var x = new XmlDocument();
x.LoadXml(new ItunesTimedText().ToText(new Subtitle(), "tt")); // load default xml
var xnsmgr = new XmlNamespaceManager(x.NameTable);
xnsmgr.AddNamespace("ttml", "http://www.w3.org/ns/ttml");
styleHead = x.DocumentElement.SelectSingleNode("ttml:head", xnsmgr);
styleHead.SelectSingleNode("ttml:styling", xnsmgr).RemoveAll();
foreach (string styleName in AdvancedSubStationAlpha.GetStylesFromHeader(subtitle.Header))
{
try
{
var ssaStyle = AdvancedSubStationAlpha.GetSsaStyle(styleName, subtitle.Header);
string fontStyle = "normal";
if (ssaStyle.Italic)
{
fontStyle = "italic";
}
string fontWeight = "normal";
if (ssaStyle.Bold)
{
fontWeight = "bold";
}
AddStyleToXml(x, styleHead, xnsmgr, ssaStyle.Name, ssaStyle.FontName, fontWeight, fontStyle, Utilities.ColorToHex(ssaStyle.Primary), ssaStyle.FontSize.ToString());
convertedFromSubStationAlpha = true;
}
catch
{
// ignored
}
}
subtitle.Header = x.OuterXml; // save new xml with styles in header
}
}
var xml = new XmlDocument { XmlResolver = null };
var nsmgr = new XmlNamespaceManager(xml.NameTable);
nsmgr.AddNamespace("ttml", "http://www.w3.org/ns/ttml");
nsmgr.AddNamespace("ttp", "http://www.w3.org/ns/10/ttml#parameter");
nsmgr.AddNamespace("tts", "http://www.w3.org/ns/10/ttml#style");
nsmgr.AddNamespace("ttm", "http://www.w3.org/ns/10/ttml#metadata");
string frameRate = ((int)Math.Round(Configuration.Settings.General.CurrentFrameRate)).ToString();
string frameRateMultiplier = "999 1000";
if (Configuration.Settings.General.CurrentFrameRate % 1.0 < 0.01)
{
frameRateMultiplier = "1 1";
}
string dropMode = "nonDrop";
if (Math.Abs(Configuration.Settings.General.CurrentFrameRate - 29.97) < 0.01)
{
dropMode = "dropNTSC";
}
const string language = "en-US";
string xmlStructure = "<?xml version=\"1.0\" encoding=\"utf-8\" ?>" + Environment.NewLine +
"<tt xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xmlns=\"http://www.w3.org/ns/ttml\" xmlns:tt=\"http://www.w3.org/ns/ttml\" xmlns:tts=\"http://www.w3.org/ns/ttml#styling\" xmlns:ttp=\"http://www.w3.org/ns/ttml#parameter\" xml:lang=\"" + language + "\" ttp:timeBase=\"smpte\" ttp:frameRate=\"" + frameRate + "\" ttp:frameRateMultiplier=\"" + frameRateMultiplier + "\" ttp:dropMode=\"" + dropMode + "\">" + Environment.NewLine +
" <head>" + Environment.NewLine +
" <styling>" + Environment.NewLine +
" <style tts:fontSize=\"100%\" tts:color=\"white\" tts:fontStyle=\"normal\" tts:fontWeight=\"normal\" tts:fontFamily=\"sansSerif\" xml:id=\"normal\"/>" + Environment.NewLine +
" <style tts:fontSize=\"100%\" tts:color=\"white\" tts:fontStyle=\"normal\" tts:fontWeight=\"bold\" tts:fontFamily=\"sansSerif\" xml:id=\"bold\"/>" + Environment.NewLine +
" <style tts:fontSize=\"100%\" tts:color=\"white\" tts:fontStyle=\"italic\" tts:fontWeight=\"normal\" tts:fontFamily=\"sansSerif\" xml:id=\"italic\"/>" + Environment.NewLine +
" </styling>" + Environment.NewLine +
" <layout>" + Environment.NewLine +
" <region xml:id=\"top\" tts:origin=\"0% 0%\" tts:extent=\"100% 15%\" tts:textAlign=\"center\" tts:displayAlign=\"before\"/>" + Environment.NewLine +
" <region xml:id=\"bottom\" tts:origin=\"0% 85%\" tts:extent=\"100% 15%\" tts:textAlign=\"center\" tts:displayAlign=\"after\"/>" + Environment.NewLine +
" </layout>" + Environment.NewLine +
" </head>" + Environment.NewLine +
" <body>" + Environment.NewLine +
" <div />" + Environment.NewLine +
" </body>" + Environment.NewLine +
"</tt>";
if (styleHead == null)
{
xml.LoadXml(xmlStructure);
}
else
{
try
{
xml.LoadXml(subtitle.Header);
XmlNode divNode = xml.DocumentElement.SelectSingleNode("//ttml:body", nsmgr).SelectSingleNode("ttml:div", nsmgr);
if (divNode == null)
{
divNode = xml.DocumentElement.SelectSingleNode("//ttml:body", nsmgr).FirstChild;
}
if (divNode != null)
{
var lst = new List<XmlNode>();
foreach (XmlNode child in divNode.ChildNodes)
{
lst.Add(child);
}
foreach (XmlNode child in lst)
{
divNode.RemoveChild(child);
}
}
else
{
xml.LoadXml(xmlStructure);
}
}
catch
{
xml.LoadXml(xmlStructure);
}
}
XmlNode body = xml.DocumentElement.SelectSingleNode("ttml:body", nsmgr);
string defaultStyle = Guid.NewGuid().ToString();
if (body.Attributes["style"] != null)
{
defaultStyle = body.Attributes["style"].InnerText;
}
XmlNode div = xml.DocumentElement.SelectSingleNode("//ttml:body", nsmgr).SelectSingleNode("ttml:div", nsmgr);
if (div == null)
{
div = xml.DocumentElement.SelectSingleNode("//ttml:body", nsmgr).FirstChild;
}
bool hasBottomRegion = false;
bool hasTopRegion = false;
foreach (XmlNode node in xml.DocumentElement.SelectNodes("//ttml:head/ttml:layout/ttml:region", nsmgr))
{
string id = null;
if (node.Attributes["xml:id"] != null)
{
id = node.Attributes["xml:id"].Value;
}
else if (node.Attributes["id"] != null)
{
id = node.Attributes["id"].Value;
}
if (id != null && id == "bottom")
{
hasBottomRegion = true;
}
if (id != null && id == "top")
{
hasTopRegion = true;
}
}
var headerStyles = GetStylesFromHeader(subtitle.Header);
foreach (Paragraph p in subtitle.Paragraphs)
{
XmlNode paragraph = xml.CreateElement("p", "http://www.w3.org/ns/ttml");
string text = p.Text;
XmlAttribute regionP = xml.CreateAttribute("region");
if (text.StartsWith("{\\an7}", StringComparison.Ordinal) || text.StartsWith("{\\an8}", StringComparison.Ordinal) || text.StartsWith("{\\an9}", StringComparison.Ordinal))
{
if (hasTopRegion)
{
regionP.InnerText = "top";
paragraph.Attributes.Append(regionP);
}
}
else if (hasBottomRegion)
{
regionP.InnerText = "bottom";
paragraph.Attributes.Append(regionP);
}
if (text.StartsWith("{\\an", StringComparison.Ordinal) && text.Length > 6 && text[5] == '}')
{
text = text.Remove(0, 6);
}
if (convertedFromSubStationAlpha)
{
if (string.IsNullOrEmpty(p.Style))
{
p.Style = p.Extra;
}
}
if (subtitle.Header != null && p.Style != null && headerStyles.Contains(p.Style))
{
if (p.Style != defaultStyle)
{
XmlAttribute styleAttr = xml.CreateAttribute("style");
styleAttr.InnerText = p.Style;
paragraph.Attributes.Append(styleAttr);
}
}
else if (!string.IsNullOrEmpty(p.Style))
{
XmlAttribute styleP = xml.CreateAttribute("style");
styleP.InnerText = p.Style;
paragraph.Attributes.Append(styleP);
}
bool first = true;
bool italicOn = false;
foreach (string line in text.SplitToLines())
{
if (!first)
{
XmlNode br = xml.CreateElement("br", "http://www.w3.org/ns/ttml");
paragraph.AppendChild(br);
}
Stack<XmlNode> styles = new Stack<XmlNode>();
XmlNode currentStyle = xml.CreateTextNode(string.Empty);
paragraph.AppendChild(currentStyle);
int skipCount = 0;
for (int i = 0; i < line.Length; i++)
{
if (skipCount > 0)
{
skipCount--;
}
else if (line.Substring(i).StartsWith("<i>", StringComparison.Ordinal))
{
styles.Push(currentStyle);
currentStyle = xml.CreateNode(XmlNodeType.Element, "span", null);
paragraph.AppendChild(currentStyle);
XmlAttribute attr = xml.CreateAttribute("tts:fontStyle", "http://www.w3.org/ns/10/ttml#style");
attr.InnerText = "italic";
currentStyle.Attributes.Append(attr);
skipCount = 2;
italicOn = true;
}
else if (line.Substring(i).StartsWith("<b>", StringComparison.Ordinal))
{
currentStyle = xml.CreateNode(XmlNodeType.Element, "span", null);
paragraph.AppendChild(currentStyle);
XmlAttribute attr = xml.CreateAttribute("tts:fontWeight", "http://www.w3.org/ns/10/ttml#style");
attr.InnerText = "bold";
currentStyle.Attributes.Append(attr);
skipCount = 2;
}
else if (line.Substring(i).StartsWith("<font ", StringComparison.Ordinal))
{
int endIndex = line.Substring(i + 1).IndexOf('>');
if (endIndex > 0)
{
skipCount = endIndex + 1;
string fontContent = line.Substring(i, skipCount);
if (fontContent.Contains(" color="))
{
var arr = fontContent.Substring(fontContent.IndexOf(" color=", StringComparison.Ordinal) + 7).Trim().Split(new[] { ' ' }, StringSplitOptions.RemoveEmptyEntries);
if (arr.Length > 0)
{
string fontColor = arr[0].Trim('\'').Trim('"').Trim('\'');
currentStyle = xml.CreateNode(XmlNodeType.Element, "span", null);
paragraph.AppendChild(currentStyle);
XmlAttribute attr = xml.CreateAttribute("tts:color", "http://www.w3.org/ns/10/ttml#style");
attr.InnerText = fontColor;
currentStyle.Attributes.Append(attr);
}
}
}
else
{
skipCount = line.Length;
}
}
else if (line.Substring(i).StartsWith("</i>", StringComparison.Ordinal) || line.Substring(i).StartsWith("</b>", StringComparison.Ordinal) || line.Substring(i).StartsWith("</font>", StringComparison.Ordinal))
{
currentStyle = xml.CreateTextNode(string.Empty);
if (styles.Count > 0)
{
currentStyle = styles.Pop().CloneNode(true);
currentStyle.InnerText = string.Empty;
}
paragraph.AppendChild(currentStyle);
if (line.Substring(i).StartsWith("</font>", StringComparison.Ordinal))
{
skipCount = 6;
}
else
{
skipCount = 3;
}
italicOn = false;
}
else
{
if (i == 0 && italicOn && !(line.Substring(i).StartsWith("<i>", StringComparison.Ordinal)))
{
styles.Push(currentStyle);
currentStyle = xml.CreateNode(XmlNodeType.Element, "span", null);
paragraph.AppendChild(currentStyle);
XmlAttribute attr = xml.CreateAttribute("tts:fontStyle", "http://www.w3.org/ns/10/ttml#style");
attr.InnerText = "italic";
currentStyle.Attributes.Append(attr);
}
currentStyle.InnerText = currentStyle.InnerText + line[i];
}
}
first = false;
}
XmlAttribute start = xml.CreateAttribute("begin");
start.InnerText = ConvertToTimeString(p.StartTime);
paragraph.Attributes.Append(start);
XmlAttribute end = xml.CreateAttribute("end");
end.InnerText = ConvertToTimeString(p.EndTime);
paragraph.Attributes.Append(end);
div.AppendChild(paragraph);
}
string xmlString = ToUtf8XmlString(xml).Replace(" xmlns=\"\"", string.Empty).Replace(" xmlns:tts=\"http://www.w3.org/ns/10/ttml#style\">", ">").Replace("<br />", "<br/>");
if (subtitle.Header == null)
{
subtitle.Header = xmlString;
}
return xmlString;
}
}
}