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

98 lines
3.4 KiB
C#

using System;
using System.Collections.Generic;
using System.Text;
using System.Text.RegularExpressions;
namespace Nikse.SubtitleEdit.Core.SubtitleFormats
{
public class Csv : SubtitleFormat
{
private const string Separator = ";";
private static readonly Regex CsvLine = new Regex(@"^""?\d+""?" + Separator + @"""?\d+""?" + Separator + @"""?\d+""?" + Separator + @"""?[^""]*""?$", RegexOptions.Compiled);
public override string Extension
{
get { return ".csv"; }
}
public override string Name
{
get { return "Csv"; }
}
public override bool IsTimeBased
{
get { return true; }
}
public override bool IsMine(List<string> lines, string fileName)
{
int fine = 0;
int failed = 0;
foreach (string line in lines)
{
if (CsvLine.IsMatch(line))
fine++;
else
failed++;
}
return fine > failed;
}
public override string ToText(Subtitle subtitle, string title)
{
const string format = "{1}{0}{2}{0}{3}{0}\"{4}\"";
var sb = new StringBuilder();
sb.AppendLine(string.Format(format, Separator, "Number", "Start time in milliseconds", "End time in milliseconds", "Text"));
foreach (Paragraph p in subtitle.Paragraphs)
{
sb.AppendLine(string.Format(format, Separator, p.Number, p.StartTime.TotalMilliseconds, p.EndTime.TotalMilliseconds, p.Text.Replace(Environment.NewLine, "\n")));
}
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(Separator.ToCharArray(), StringSplitOptions.RemoveEmptyEntries);
if (parts.Length == 4)
try
{
int start = Convert.ToInt32(Utilities.FixQuotes(parts[1]));
int end = Convert.ToInt32(Utilities.FixQuotes(parts[2]));
string text = Utilities.FixQuotes(parts[3]);
p = new Paragraph(text, start, end);
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();
}
}
}