[1114] | 1 | using System;
|
---|
| 2 | using System.Collections.Generic;
|
---|
| 3 | using System.IO;
|
---|
| 4 | using System.Text;
|
---|
| 5 |
|
---|
| 6 | namespace Oni
|
---|
| 7 | {
|
---|
| 8 | internal sealed class SubtitleImporter : Importer
|
---|
| 9 | {
|
---|
| 10 | #region Private data
|
---|
| 11 | private List<byte> subtitles;
|
---|
| 12 | private List<int> offsets;
|
---|
| 13 | #endregion
|
---|
| 14 |
|
---|
| 15 | public override void Import(string filePath, string outputDirPath)
|
---|
| 16 | {
|
---|
| 17 | ReadTextFile(filePath);
|
---|
| 18 |
|
---|
| 19 | string name = Path.GetFileNameWithoutExtension(filePath);
|
---|
| 20 |
|
---|
| 21 | BeginImport();
|
---|
| 22 | WriteSUBT(name, subtitles.ToArray());
|
---|
| 23 | Write(outputDirPath);
|
---|
| 24 | }
|
---|
| 25 |
|
---|
| 26 | private void ReadTextFile(string filePath)
|
---|
| 27 | {
|
---|
| 28 | subtitles = new List<byte>();
|
---|
| 29 | offsets = new List<int>();
|
---|
| 30 |
|
---|
| 31 | var data = File.ReadAllBytes(filePath);
|
---|
| 32 | int i = SkipPreamble(data);
|
---|
| 33 |
|
---|
| 34 | while (i < data.Length)
|
---|
| 35 | {
|
---|
| 36 | int nameStart = i;
|
---|
| 37 |
|
---|
| 38 | while (!IsNewLine(data, i) && data[i] != '=')
|
---|
| 39 | i++;
|
---|
| 40 |
|
---|
| 41 | int nameEnd = i;
|
---|
| 42 |
|
---|
| 43 | if (IsNewLine(data, i))
|
---|
| 44 | {
|
---|
| 45 | //
|
---|
| 46 | // This was an empty line or a line that does not contain a =. Skip it.
|
---|
| 47 | //
|
---|
| 48 |
|
---|
| 49 | continue;
|
---|
| 50 | }
|
---|
| 51 |
|
---|
| 52 | i++;
|
---|
| 53 |
|
---|
| 54 | int textStart = i;
|
---|
| 55 |
|
---|
| 56 | while (!IsNewLine(data, i))
|
---|
| 57 | i++;
|
---|
| 58 |
|
---|
| 59 | int textEnd = i;
|
---|
| 60 |
|
---|
| 61 | if (nameEnd > nameStart)
|
---|
| 62 | {
|
---|
| 63 | offsets.Add(subtitles.Count);
|
---|
| 64 |
|
---|
| 65 | for (int j = nameStart; j < nameEnd; j++)
|
---|
| 66 | subtitles.Add(data[j]);
|
---|
| 67 |
|
---|
| 68 | subtitles.Add(0);
|
---|
| 69 |
|
---|
| 70 | for (int j = textStart; j < textEnd; j++)
|
---|
| 71 | subtitles.Add(data[j]);
|
---|
| 72 |
|
---|
| 73 | subtitles.Add(0);
|
---|
| 74 | }
|
---|
| 75 |
|
---|
| 76 | i = SkipNewLine(data, i);
|
---|
| 77 | }
|
---|
| 78 | }
|
---|
| 79 |
|
---|
| 80 | private static int SkipPreamble(byte[] data)
|
---|
| 81 | {
|
---|
| 82 | int offset = CheckPreamble(data, Encoding.UTF8.GetPreamble());
|
---|
| 83 |
|
---|
| 84 | if (offset > 0)
|
---|
| 85 | return offset;
|
---|
| 86 |
|
---|
| 87 | if (CheckPreamble(data, Encoding.Unicode.GetPreamble()) != 0
|
---|
| 88 | || CheckPreamble(data, Encoding.BigEndianUnicode.GetPreamble()) != 0
|
---|
| 89 | || CheckPreamble(data, Encoding.UTF32.GetPreamble()) != 0)
|
---|
| 90 | {
|
---|
| 91 | throw new InvalidDataException("UTF16/32 input text files are not supported.");
|
---|
| 92 | }
|
---|
| 93 |
|
---|
| 94 | if (data.Length >= 4)
|
---|
| 95 | {
|
---|
| 96 | if ((data[1] == 0 && data[3] == 0)
|
---|
| 97 | || (data[0] == 0 && data[2] == 0))
|
---|
| 98 | {
|
---|
| 99 | throw new InvalidDataException("UTF16/32 input text files are not supported.");
|
---|
| 100 | }
|
---|
| 101 | }
|
---|
| 102 |
|
---|
| 103 | return 0;
|
---|
| 104 | }
|
---|
| 105 |
|
---|
| 106 | private static int CheckPreamble(byte[] data, byte[] preamble)
|
---|
| 107 | {
|
---|
| 108 | if (data.Length < preamble.Length)
|
---|
| 109 | return 0;
|
---|
| 110 |
|
---|
| 111 | for (int i = 0; i < preamble.Length; i++)
|
---|
| 112 | {
|
---|
| 113 | if (data[i] != preamble[i])
|
---|
| 114 | return 0;
|
---|
| 115 | }
|
---|
| 116 |
|
---|
| 117 | return preamble.Length;
|
---|
| 118 | }
|
---|
| 119 |
|
---|
| 120 | private static bool IsNewLine(byte[] data, int offset)
|
---|
| 121 | {
|
---|
| 122 | if (offset >= data.Length)
|
---|
| 123 | return true;
|
---|
| 124 |
|
---|
| 125 | return (SkipNewLine(data, offset) > offset);
|
---|
| 126 | }
|
---|
| 127 |
|
---|
| 128 | private static int SkipNewLine(byte[] data, int offset)
|
---|
| 129 | {
|
---|
| 130 | if (offset < data.Length)
|
---|
| 131 | {
|
---|
| 132 | if (data[offset] == '\n')
|
---|
| 133 | {
|
---|
| 134 | offset++;
|
---|
| 135 | }
|
---|
| 136 | else if (data[offset] == '\r')
|
---|
| 137 | {
|
---|
| 138 | offset++;
|
---|
| 139 |
|
---|
| 140 | if (offset < data.Length && data[offset] == '\n')
|
---|
| 141 | offset++;
|
---|
| 142 | }
|
---|
| 143 | }
|
---|
| 144 |
|
---|
| 145 | return offset;
|
---|
| 146 | }
|
---|
| 147 |
|
---|
| 148 | private void WriteSUBT(string name, byte[] subtitles)
|
---|
| 149 | {
|
---|
| 150 | var subt = CreateInstance(TemplateTag.SUBT, name);
|
---|
| 151 |
|
---|
| 152 | using (var writer = subt.OpenWrite(16))
|
---|
| 153 | {
|
---|
| 154 | writer.Write(WriteRawPart(subtitles));
|
---|
| 155 | writer.Write(offsets.Count);
|
---|
| 156 | writer.Write(offsets.ToArray());
|
---|
| 157 | }
|
---|
| 158 | }
|
---|
| 159 | }
|
---|
| 160 | }
|
---|