SubtitleEdit/libse/SubtitleFormats/SubViewer20.cs

using System;
using System.Collections.Generic;
using System.Text;
using System.Text.RegularExpressions;

namespace Nikse.SubtitleEdit.Core.SubtitleFormats
{
    public class SubViewer20 : SubtitleFormat
    {
        private static readonly Regex RegexTimeCodes = new Regex(@"^\d\d:\d\d:\d\d.\d+,\d\d:\d\d:\d\d.\d+$", RegexOptions.Compiled);

        private enum ExpectingLine
        {
            TimeCodes,
            Text
        }

        public override string Extension
        {
            get { return ".sub"; }
        }

        public override string Name
        {
            get { return "SubViewer 2.0"; }
        }

        public override bool IsTimeBased
        {
            get { return true; }
        }

        public override bool IsMine(List<string> lines, string fileName)
        {
            var sbv = new YouTubeSbv();
            if (sbv.IsMine(lines, fileName) && !string.Join(string.Empty, lines.ToArray()).Contains("[br]"))
                return false;

            var subtitle = new Subtitle();
            LoadSubtitle(subtitle, lines, fileName);
            return subtitle.Paragraphs.Count > _errorCount;
        }

        public override string ToText(Subtitle subtitle, string title)
        {
            const string paragraphWriteFormat = "{0:00}:{1:00}:{2:00}.{3:00},{4:00}:{5:00}:{6:00}.{7:00}{8}{9}";
            const string header = @"[INFORMATION]
[TITLE]{0}
[AUTHOR]
[SOURCE]
[PRG]
[FILEPATH]
[DELAY]0
[CD TRACK]0
[COMMENT]
[END INFORMATION]
[SUBTITLE]
[COLF]&H000000,[STYLE]bd,[SIZE]25,[FONT]Arial
";
            //00:00:06.61,00:00:13.75
            //text1[br]text2
            var sb = new StringBuilder();
            sb.AppendFormat(header, title);
            foreach (Paragraph p in subtitle.Paragraphs)
            {

                string text = EncodeFormatting(p.Text);
                sb.AppendLine(string.Format(paragraphWriteFormat,
                                        p.StartTime.Hours,
                                        p.StartTime.Minutes,
                                        p.StartTime.Seconds,
                                        RoundTo2Cifres(p.StartTime.Milliseconds),
                                        p.EndTime.Hours,
                                        p.EndTime.Minutes,
                                        p.EndTime.Seconds,
                                        RoundTo2Cifres(p.EndTime.Milliseconds),
                                        Environment.NewLine,
                                        text));
                sb.AppendLine();
            }
            return sb.ToString().Trim();
        }

        private static int RoundTo2Cifres(int milliseconds)
        {
            return (int)Math.Round((double)milliseconds / 10);
        }

        public override void LoadSubtitle(Subtitle subtitle, List<string> lines, string fileName)
        {
            var paragraph = new Paragraph();
            ExpectingLine expecting = ExpectingLine.TimeCodes;
            _errorCount = 0;
            char[] splitChars = { ':', ',', '.' };
            subtitle.Paragraphs.Clear();
            foreach (string line in lines)
            {
                string lineTrimmed = line.Trim();
                if (lineTrimmed.Length >= 23 && RegexTimeCodes.IsMatch(lineTrimmed))
                {
                    string[] parts = lineTrimmed.Split(splitChars, StringSplitOptions.RemoveEmptyEntries);
                    try
                    {
                        int startHours = int.Parse(parts[0]);
                        int startMinutes = int.Parse(parts[1]);
                        int startSeconds = int.Parse(parts[2]);
                        int startMilliseconds = int.Parse(parts[3]);
                        int endHours = int.Parse(parts[4]);
                        int endMinutes = int.Parse(parts[5]);
                        int endSeconds = int.Parse(parts[6]);
                        int endMilliseconds = int.Parse(parts[7]);
                        paragraph.StartTime = new TimeCode(startHours, startMinutes, startSeconds, startMilliseconds);
                        paragraph.EndTime = new TimeCode(endHours, endMinutes, endSeconds, endMilliseconds);
                        expecting = ExpectingLine.Text;
                    }
                    catch
                    {
                        _errorCount++;
                    }
                }
                else if (expecting == ExpectingLine.Text && lineTrimmed.Length > 0)
                {
                    paragraph.Text = DecodeForamtting(lineTrimmed);
                    subtitle.Paragraphs.Add(paragraph);
                    paragraph = new Paragraph();
                    expecting = ExpectingLine.TimeCodes;
                }
                else
                {
                    _errorCount++;
                }
            }
            subtitle.Renumber();
        }

        private string DecodeForamtting(string text)
        {
            text = text.Replace("[br]", Environment.NewLine);
            text = text.Replace("{\\i1}", "<i>");
            text = text.Replace("{\\i0}", "</i>");
            text = text.Replace("{\\i}", "</i>");
            text = text.Replace("{\\b1}", "<b>'");
            text = text.Replace("{\\b0}", "</b>");
            text = text.Replace("{\\b}", "</b>");
            text = text.Replace("{\\u1}", "<u>");
            text = text.Replace("{\\u0}", "</u>");
            return text.Replace("{\\u}", "</u>");
        }

        private string EncodeFormatting(string text)
        {
            text = text.Replace(Environment.NewLine, "[br]");
            text = text.Replace("<i>", "{\\i1}");
            text = text.Replace("</i>", "{\\i0}");
            text = text.Replace("</i>", "{\\i}");
            text = text.Replace("<b>", "{\\b1}'");
            text = text.Replace("</b>", "{\\b0}");
            text = text.Replace("</b>", "{\\b}");
            text = text.Replace("<u>", "{\\u1}");
            text = text.Replace("</u>", "{\\u0}");
            return text.Replace("</u>", "{\\u}");
        }
    }
}