| | 1 | | using System; |
| | 2 | | using System.IO; |
| | 3 | | using System.Text; |
| | 4 | | using System.Text.RegularExpressions; |
| | 5 | | using System.Threading; |
| | 6 | | using MediaBrowser.Model.MediaInfo; |
| | 7 | |
|
| | 8 | | namespace MediaBrowser.MediaEncoding.Subtitles |
| | 9 | | { |
| | 10 | | /// <summary> |
| | 11 | | /// Subtitle writer for the WebVTT format. |
| | 12 | | /// </summary> |
| | 13 | | public partial class VttWriter : ISubtitleWriter |
| | 14 | | { |
| | 15 | | [GeneratedRegex(@"\\n", RegexOptions.IgnoreCase)] |
| | 16 | | private static partial Regex NewlineEscapeRegex(); |
| | 17 | |
|
| | 18 | | /// <inheritdoc /> |
| | 19 | | public void Write(SubtitleTrackInfo info, Stream stream, CancellationToken cancellationToken) |
| | 20 | | { |
| 0 | 21 | | using (var writer = new StreamWriter(stream, Encoding.UTF8, 1024, true)) |
| | 22 | | { |
| 0 | 23 | | writer.WriteLine("WEBVTT"); |
| 0 | 24 | | writer.WriteLine(); |
| 0 | 25 | | writer.WriteLine("Region: id:subtitle width:80% lines:3 regionanchor:50%,100% viewportanchor:50%,90%"); |
| 0 | 26 | | writer.WriteLine(); |
| 0 | 27 | | foreach (var trackEvent in info.TrackEvents) |
| | 28 | | { |
| 0 | 29 | | cancellationToken.ThrowIfCancellationRequested(); |
| | 30 | |
|
| 0 | 31 | | var startTime = TimeSpan.FromTicks(trackEvent.StartPositionTicks); |
| 0 | 32 | | var endTime = TimeSpan.FromTicks(trackEvent.EndPositionTicks); |
| | 33 | |
|
| | 34 | | // make sure the start and end times are different and sequential |
| 0 | 35 | | if (endTime.TotalMilliseconds <= startTime.TotalMilliseconds) |
| | 36 | | { |
| 0 | 37 | | endTime = startTime.Add(TimeSpan.FromMilliseconds(1)); |
| | 38 | | } |
| | 39 | |
|
| 0 | 40 | | writer.WriteLine(@"{0:hh\:mm\:ss\.fff} --> {1:hh\:mm\:ss\.fff} region:subtitle line:90%", startTime, |
| | 41 | |
|
| 0 | 42 | | var text = trackEvent.Text; |
| | 43 | |
|
| | 44 | | // TODO: Not sure how to handle these |
| 0 | 45 | | text = NewlineEscapeRegex().Replace(text, " "); |
| | 46 | |
|
| 0 | 47 | | writer.WriteLine(text); |
| 0 | 48 | | writer.WriteLine(); |
| | 49 | | } |
| | 50 | | } |
| 0 | 51 | | } |
| | 52 | | } |
| | 53 | | } |