SubtitleEdit/libse/SubtitleFormats/CsvNuendo.cs
2020-02-03 20:51:25 +01:00

105 lines
3.8 KiB
C#

using System;
using System.Collections.Generic;
using System.Text;
using System.Text.RegularExpressions;
namespace Nikse.SubtitleEdit.Core.SubtitleFormats
{
public class CsvNuendo : SubtitleFormat
{
private static readonly Regex CsvLine = new Regex("^(\"(.*)\")*,\\d+:\\d+:\\d+:\\d+,\\d+:\\d+:\\d+:\\d+,(\"(.*)\")*", RegexOptions.Compiled);
private const string LineFormat = "{1}{0}{2}{0}{3}{0}{4}";
private static readonly string Header = string.Format(LineFormat, ",", "\"Character\"", "\"Timecode In\"", "\"Timecode Out\"", "\"Dialogue\"");
public override string Extension => ".csv";
public override string Name => "Csv Nuendo";
public override bool IsMine(List<string> lines, string fileName)
{
int fine = 0;
var sb = new StringBuilder();
foreach (string line in lines)
{
sb.Append(line);
if (line.IndexOf(':') > 0 && CsvLine.IsMatch(line))
{
fine++;
}
}
return fine > 0 && sb.ToString().Contains(Header);
}
public override string ToText(Subtitle subtitle, string title)
{
var sb = new StringBuilder();
sb.AppendLine(Header);
foreach (Paragraph p in subtitle.Paragraphs)
{
string text = string.IsNullOrEmpty(p.Text) ? string.Empty : "\"" + p.Text.Replace("\"", "\"\"").Replace(Environment.NewLine, "\n") + "\"";
string actor = string.IsNullOrEmpty(p.Actor) ? string.Empty : "\"" + p.Actor.Replace(",", " ").Replace("\"", string.Empty) + "\"";
sb.AppendLine(string.Format(LineFormat, ",", actor, p.StartTime.ToHHMMSSFF(), p.EndTime.ToHHMMSSFF(), text));
}
return sb.ToString().Trim();
}
public override void LoadSubtitle(Subtitle subtitle, List<string> lines, string fileName)
{
_errorCount = 0;
bool continuation = false;
Paragraph p = null;
foreach (string line in lines)
{
if (CsvLine.IsMatch(line))
{
string[] parts = line.Split(',');
if (parts.Length == 4)
{
try
{
var actor = Utilities.FixQuotes(parts[0]);
var start = DecodeTime(parts[1]);
var end = DecodeTime(parts[2]);
string text = Utilities.FixQuotes(parts[3]);
p = new Paragraph(start, end, text);
if (!string.IsNullOrEmpty(actor))
{
p.Actor = actor;
}
subtitle.Paragraphs.Add(p);
continuation = parts[3].StartsWith('"') && !parts[3].EndsWith('"');
}
catch
{
_errorCount++;
}
}
}
else
{
if (continuation)
{
if (p.Text.Length < 300)
{
p.Text = (p.Text + Environment.NewLine + line.TrimEnd('"')).Trim();
}
continuation = !line.TrimEnd().EndsWith('"');
}
else
{
_errorCount++;
}
}
}
subtitle.Renumber();
}
private static TimeCode DecodeTime(string s)
{
return DecodeTimeCodeFramesFourParts(s.Split(':'));
}
}
}