SsaParser.cs 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Globalization;
  4. using System.IO;
  5. using System.Linq;
  6. using System.Text.RegularExpressions;
  7. using System.Threading;
  8. namespace MediaBrowser.MediaEncoding.Subtitles
  9. {
  10. public class SsaParser : ISubtitleParser
  11. {
  12. private readonly CultureInfo _usCulture = new CultureInfo("en-US");
  13. public SubtitleTrackInfo Parse(Stream stream, CancellationToken cancellationToken)
  14. {
  15. var trackInfo = new SubtitleTrackInfo();
  16. var eventIndex = 1;
  17. using (var reader = new StreamReader(stream))
  18. {
  19. string line;
  20. while (reader.ReadLine() != "[Events]")
  21. {}
  22. var headers = ParseFieldHeaders(reader.ReadLine());
  23. while ((line = reader.ReadLine()) != null)
  24. {
  25. cancellationToken.ThrowIfCancellationRequested();
  26. if (string.IsNullOrWhiteSpace(line))
  27. {
  28. continue;
  29. }
  30. if(line.StartsWith("["))
  31. break;
  32. if(string.IsNullOrEmpty(line))
  33. continue;
  34. var subEvent = new SubtitleTrackEvent { Id = eventIndex.ToString(_usCulture) };
  35. eventIndex++;
  36. var sections = line.Substring(10).Split(',');
  37. subEvent.StartPositionTicks = GetTicks(sections[headers["Start"]]);
  38. subEvent.EndPositionTicks = GetTicks(sections[headers["End"]]);
  39. subEvent.Text = string.Join(",", sections.Skip(headers["Text"]));
  40. subEvent.Text = Regex.Replace(subEvent.Text, @"\{(\\[\w]+\(?([\w\d]+,?)+\)?)+\}", string.Empty, RegexOptions.IgnoreCase);
  41. trackInfo.TrackEvents.Add(subEvent);
  42. }
  43. }
  44. return trackInfo;
  45. }
  46. long GetTicks(string time)
  47. {
  48. TimeSpan span;
  49. return TimeSpan.TryParseExact(time, @"h\:mm\:ss\.ff", _usCulture, out span)
  50. ? span.Ticks: 0;
  51. }
  52. private Dictionary<string,int> ParseFieldHeaders(string line) {
  53. var fields = line.Substring(8).Split(',').Select(x=>x.Trim()).ToList();
  54. var result = new Dictionary<string, int> {
  55. {"Start", fields.IndexOf("Start")},
  56. {"End", fields.IndexOf("End")},
  57. {"Text", fields.IndexOf("Text")}
  58. };
  59. return result;
  60. }
  61. }
  62. }