TSPlaylistFile.cs 43 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292
  1. //============================================================================
  2. // BDInfo - Blu-ray Video and Audio Analysis Tool
  3. // Copyright © 2010 Cinema Squid
  4. //
  5. // This library is free software; you can redistribute it and/or
  6. // modify it under the terms of the GNU Lesser General Public
  7. // License as published by the Free Software Foundation; either
  8. // version 2.1 of the License, or (at your option) any later version.
  9. //
  10. // This library is distributed in the hope that it will be useful,
  11. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  13. // Lesser General Public License for more details.
  14. //
  15. // You should have received a copy of the GNU Lesser General Public
  16. // License along with this library; if not, write to the Free Software
  17. // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  18. //=============================================================================
  19. #undef DEBUG
  20. using System;
  21. using System.Collections.Generic;
  22. using System.IO;
  23. using System.Text;
  24. using MediaBrowser.Model.IO;
  25. using MediaBrowser.Model.Text;
  26. namespace BDInfo
  27. {
  28. public class TSPlaylistFile
  29. {
  30. private readonly IFileSystem _fileSystem;
  31. private readonly ITextEncoding _textEncoding;
  32. private FileSystemMetadata FileInfo = null;
  33. public string FileType = null;
  34. public bool IsInitialized = false;
  35. public string Name = null;
  36. public BDROM BDROM = null;
  37. public bool HasHiddenTracks = false;
  38. public bool HasLoops = false;
  39. public bool IsCustom = false;
  40. public List<double> Chapters = new List<double>();
  41. public Dictionary<ushort, TSStream> Streams =
  42. new Dictionary<ushort, TSStream>();
  43. public Dictionary<ushort, TSStream> PlaylistStreams =
  44. new Dictionary<ushort, TSStream>();
  45. public List<TSStreamClip> StreamClips =
  46. new List<TSStreamClip>();
  47. public List<Dictionary<ushort, TSStream>> AngleStreams =
  48. new List<Dictionary<ushort, TSStream>>();
  49. public List<Dictionary<double, TSStreamClip>> AngleClips =
  50. new List<Dictionary<double, TSStreamClip>>();
  51. public int AngleCount = 0;
  52. public List<TSStream> SortedStreams =
  53. new List<TSStream>();
  54. public List<TSVideoStream> VideoStreams =
  55. new List<TSVideoStream>();
  56. public List<TSAudioStream> AudioStreams =
  57. new List<TSAudioStream>();
  58. public List<TSTextStream> TextStreams =
  59. new List<TSTextStream>();
  60. public List<TSGraphicsStream> GraphicsStreams =
  61. new List<TSGraphicsStream>();
  62. public TSPlaylistFile(
  63. BDROM bdrom,
  64. FileSystemMetadata fileInfo, IFileSystem fileSystem, ITextEncoding textEncoding)
  65. {
  66. BDROM = bdrom;
  67. FileInfo = fileInfo;
  68. _fileSystem = fileSystem;
  69. _textEncoding = textEncoding;
  70. Name = fileInfo.Name.ToUpper();
  71. }
  72. public TSPlaylistFile(
  73. BDROM bdrom,
  74. string name,
  75. List<TSStreamClip> clips, IFileSystem fileSystem, ITextEncoding textEncoding)
  76. {
  77. BDROM = bdrom;
  78. Name = name;
  79. _fileSystem = fileSystem;
  80. _textEncoding = textEncoding;
  81. IsCustom = true;
  82. foreach (TSStreamClip clip in clips)
  83. {
  84. TSStreamClip newClip = new TSStreamClip(
  85. clip.StreamFile, clip.StreamClipFile);
  86. newClip.Name = clip.Name;
  87. newClip.TimeIn = clip.TimeIn;
  88. newClip.TimeOut = clip.TimeOut;
  89. newClip.Length = newClip.TimeOut - newClip.TimeIn;
  90. newClip.RelativeTimeIn = TotalLength;
  91. newClip.RelativeTimeOut = newClip.RelativeTimeIn + newClip.Length;
  92. newClip.AngleIndex = clip.AngleIndex;
  93. newClip.Chapters.Add(clip.TimeIn);
  94. StreamClips.Add(newClip);
  95. if (newClip.AngleIndex > AngleCount)
  96. {
  97. AngleCount = newClip.AngleIndex;
  98. }
  99. if (newClip.AngleIndex == 0)
  100. {
  101. Chapters.Add(newClip.RelativeTimeIn);
  102. }
  103. }
  104. LoadStreamClips();
  105. IsInitialized = true;
  106. }
  107. public override string ToString()
  108. {
  109. return Name;
  110. }
  111. public ulong InterleavedFileSize
  112. {
  113. get
  114. {
  115. ulong size = 0;
  116. foreach (TSStreamClip clip in StreamClips)
  117. {
  118. size += clip.InterleavedFileSize;
  119. }
  120. return size;
  121. }
  122. }
  123. public ulong FileSize
  124. {
  125. get
  126. {
  127. ulong size = 0;
  128. foreach (TSStreamClip clip in StreamClips)
  129. {
  130. size += clip.FileSize;
  131. }
  132. return size;
  133. }
  134. }
  135. public double TotalLength
  136. {
  137. get
  138. {
  139. double length = 0;
  140. foreach (TSStreamClip clip in StreamClips)
  141. {
  142. if (clip.AngleIndex == 0)
  143. {
  144. length += clip.Length;
  145. }
  146. }
  147. return length;
  148. }
  149. }
  150. public double TotalAngleLength
  151. {
  152. get
  153. {
  154. double length = 0;
  155. foreach (TSStreamClip clip in StreamClips)
  156. {
  157. length += clip.Length;
  158. }
  159. return length;
  160. }
  161. }
  162. public ulong TotalSize
  163. {
  164. get
  165. {
  166. ulong size = 0;
  167. foreach (TSStreamClip clip in StreamClips)
  168. {
  169. if (clip.AngleIndex == 0)
  170. {
  171. size += clip.PacketSize;
  172. }
  173. }
  174. return size;
  175. }
  176. }
  177. public ulong TotalAngleSize
  178. {
  179. get
  180. {
  181. ulong size = 0;
  182. foreach (TSStreamClip clip in StreamClips)
  183. {
  184. size += clip.PacketSize;
  185. }
  186. return size;
  187. }
  188. }
  189. public ulong TotalBitRate
  190. {
  191. get
  192. {
  193. if (TotalLength > 0)
  194. {
  195. return (ulong)Math.Round(((TotalSize * 8.0) / TotalLength));
  196. }
  197. return 0;
  198. }
  199. }
  200. public ulong TotalAngleBitRate
  201. {
  202. get
  203. {
  204. if (TotalAngleLength > 0)
  205. {
  206. return (ulong)Math.Round(((TotalAngleSize * 8.0) / TotalAngleLength));
  207. }
  208. return 0;
  209. }
  210. }
  211. public void Scan(
  212. Dictionary<string, TSStreamFile> streamFiles,
  213. Dictionary<string, TSStreamClipFile> streamClipFiles)
  214. {
  215. Stream fileStream = null;
  216. BinaryReader fileReader = null;
  217. try
  218. {
  219. Streams.Clear();
  220. StreamClips.Clear();
  221. fileStream = _fileSystem.OpenRead(FileInfo.FullName);
  222. fileReader = new BinaryReader(fileStream);
  223. byte[] data = new byte[fileStream.Length];
  224. int dataLength = fileReader.Read(data, 0, data.Length);
  225. int pos = 0;
  226. FileType = ReadString(data, 8, ref pos);
  227. if (FileType != "MPLS0100" && FileType != "MPLS0200")
  228. {
  229. throw new Exception(string.Format(
  230. "Playlist {0} has an unknown file type {1}.",
  231. FileInfo.Name, FileType));
  232. }
  233. int playlistOffset = ReadInt32(data, ref pos);
  234. int chaptersOffset = ReadInt32(data, ref pos);
  235. int extensionsOffset = ReadInt32(data, ref pos);
  236. pos = playlistOffset;
  237. int playlistLength = ReadInt32(data, ref pos);
  238. int playlistReserved = ReadInt16(data, ref pos);
  239. int itemCount = ReadInt16(data, ref pos);
  240. int subitemCount = ReadInt16(data, ref pos);
  241. List<TSStreamClip> chapterClips = new List<TSStreamClip>();
  242. for (int itemIndex = 0; itemIndex < itemCount; itemIndex++)
  243. {
  244. int itemStart = pos;
  245. int itemLength = ReadInt16(data, ref pos);
  246. string itemName = ReadString(data, 5, ref pos);
  247. string itemType = ReadString(data, 4, ref pos);
  248. TSStreamFile streamFile = null;
  249. string streamFileName = string.Format(
  250. "{0}.M2TS", itemName);
  251. if (streamFiles.ContainsKey(streamFileName))
  252. {
  253. streamFile = streamFiles[streamFileName];
  254. }
  255. if (streamFile == null)
  256. {
  257. // Error condition
  258. }
  259. TSStreamClipFile streamClipFile = null;
  260. string streamClipFileName = string.Format(
  261. "{0}.CLPI", itemName);
  262. if (streamClipFiles.ContainsKey(streamClipFileName))
  263. {
  264. streamClipFile = streamClipFiles[streamClipFileName];
  265. }
  266. if (streamClipFile == null)
  267. {
  268. throw new Exception(string.Format(
  269. "Playlist {0} referenced missing file {1}.",
  270. FileInfo.Name, streamFileName));
  271. }
  272. pos += 1;
  273. int multiangle = (data[pos] >> 4) & 0x01;
  274. int condition = data[pos] & 0x0F;
  275. pos += 2;
  276. int inTime = ReadInt32(data, ref pos);
  277. if (inTime < 0) inTime &= 0x7FFFFFFF;
  278. double timeIn = (double)inTime / 45000;
  279. int outTime = ReadInt32(data, ref pos);
  280. if (outTime < 0) outTime &= 0x7FFFFFFF;
  281. double timeOut = (double)outTime / 45000;
  282. TSStreamClip streamClip = new TSStreamClip(
  283. streamFile, streamClipFile);
  284. streamClip.Name = streamFileName; //TODO
  285. streamClip.TimeIn = timeIn;
  286. streamClip.TimeOut = timeOut;
  287. streamClip.Length = streamClip.TimeOut - streamClip.TimeIn;
  288. streamClip.RelativeTimeIn = TotalLength;
  289. streamClip.RelativeTimeOut = streamClip.RelativeTimeIn + streamClip.Length;
  290. StreamClips.Add(streamClip);
  291. chapterClips.Add(streamClip);
  292. pos += 12;
  293. if (multiangle > 0)
  294. {
  295. int angles = data[pos];
  296. pos += 2;
  297. for (int angle = 0; angle < angles - 1; angle++)
  298. {
  299. string angleName = ReadString(data, 5, ref pos);
  300. string angleType = ReadString(data, 4, ref pos);
  301. pos += 1;
  302. TSStreamFile angleFile = null;
  303. string angleFileName = string.Format(
  304. "{0}.M2TS", angleName);
  305. if (streamFiles.ContainsKey(angleFileName))
  306. {
  307. angleFile = streamFiles[angleFileName];
  308. }
  309. if (angleFile == null)
  310. {
  311. throw new Exception(string.Format(
  312. "Playlist {0} referenced missing angle file {1}.",
  313. FileInfo.Name, angleFileName));
  314. }
  315. TSStreamClipFile angleClipFile = null;
  316. string angleClipFileName = string.Format(
  317. "{0}.CLPI", angleName);
  318. if (streamClipFiles.ContainsKey(angleClipFileName))
  319. {
  320. angleClipFile = streamClipFiles[angleClipFileName];
  321. }
  322. if (angleClipFile == null)
  323. {
  324. throw new Exception(string.Format(
  325. "Playlist {0} referenced missing angle file {1}.",
  326. FileInfo.Name, angleClipFileName));
  327. }
  328. TSStreamClip angleClip =
  329. new TSStreamClip(angleFile, angleClipFile);
  330. angleClip.AngleIndex = angle + 1;
  331. angleClip.TimeIn = streamClip.TimeIn;
  332. angleClip.TimeOut = streamClip.TimeOut;
  333. angleClip.RelativeTimeIn = streamClip.RelativeTimeIn;
  334. angleClip.RelativeTimeOut = streamClip.RelativeTimeOut;
  335. angleClip.Length = streamClip.Length;
  336. StreamClips.Add(angleClip);
  337. }
  338. if (angles - 1 > AngleCount) AngleCount = angles - 1;
  339. }
  340. int streamInfoLength = ReadInt16(data, ref pos);
  341. pos += 2;
  342. int streamCountVideo = data[pos++];
  343. int streamCountAudio = data[pos++];
  344. int streamCountPG = data[pos++];
  345. int streamCountIG = data[pos++];
  346. int streamCountSecondaryAudio = data[pos++];
  347. int streamCountSecondaryVideo = data[pos++];
  348. int streamCountPIP = data[pos++];
  349. pos += 5;
  350. #if DEBUG
  351. Debug.WriteLine(string.Format(
  352. "{0} : {1} -> V:{2} A:{3} PG:{4} IG:{5} 2A:{6} 2V:{7} PIP:{8}",
  353. Name, streamFileName, streamCountVideo, streamCountAudio, streamCountPG, streamCountIG,
  354. streamCountSecondaryAudio, streamCountSecondaryVideo, streamCountPIP));
  355. #endif
  356. for (int i = 0; i < streamCountVideo; i++)
  357. {
  358. TSStream stream = CreatePlaylistStream(data, ref pos);
  359. if (stream != null) PlaylistStreams[stream.PID] = stream;
  360. }
  361. for (int i = 0; i < streamCountAudio; i++)
  362. {
  363. TSStream stream = CreatePlaylistStream(data, ref pos);
  364. if (stream != null) PlaylistStreams[stream.PID] = stream;
  365. }
  366. for (int i = 0; i < streamCountPG; i++)
  367. {
  368. TSStream stream = CreatePlaylistStream(data, ref pos);
  369. if (stream != null) PlaylistStreams[stream.PID] = stream;
  370. }
  371. for (int i = 0; i < streamCountIG; i++)
  372. {
  373. TSStream stream = CreatePlaylistStream(data, ref pos);
  374. if (stream != null) PlaylistStreams[stream.PID] = stream;
  375. }
  376. for (int i = 0; i < streamCountSecondaryAudio; i++)
  377. {
  378. TSStream stream = CreatePlaylistStream(data, ref pos);
  379. if (stream != null) PlaylistStreams[stream.PID] = stream;
  380. pos += 2;
  381. }
  382. for (int i = 0; i < streamCountSecondaryVideo; i++)
  383. {
  384. TSStream stream = CreatePlaylistStream(data, ref pos);
  385. if (stream != null) PlaylistStreams[stream.PID] = stream;
  386. pos += 6;
  387. }
  388. /*
  389. * TODO
  390. *
  391. for (int i = 0; i < streamCountPIP; i++)
  392. {
  393. TSStream stream = CreatePlaylistStream(data, ref pos);
  394. if (stream != null) PlaylistStreams[stream.PID] = stream;
  395. }
  396. */
  397. pos += itemLength - (pos - itemStart) + 2;
  398. }
  399. pos = chaptersOffset + 4;
  400. int chapterCount = ReadInt16(data, ref pos);
  401. for (int chapterIndex = 0;
  402. chapterIndex < chapterCount;
  403. chapterIndex++)
  404. {
  405. int chapterType = data[pos+1];
  406. if (chapterType == 1)
  407. {
  408. int streamFileIndex =
  409. ((int)data[pos + 2] << 8) + data[pos + 3];
  410. long chapterTime =
  411. ((long)data[pos + 4] << 24) +
  412. ((long)data[pos + 5] << 16) +
  413. ((long)data[pos + 6] << 8) +
  414. ((long)data[pos + 7]);
  415. TSStreamClip streamClip = chapterClips[streamFileIndex];
  416. double chapterSeconds = (double)chapterTime / 45000;
  417. double relativeSeconds =
  418. chapterSeconds -
  419. streamClip.TimeIn +
  420. streamClip.RelativeTimeIn;
  421. // TODO: Ignore short last chapter?
  422. if (TotalLength - relativeSeconds > 1.0)
  423. {
  424. streamClip.Chapters.Add(chapterSeconds);
  425. this.Chapters.Add(relativeSeconds);
  426. }
  427. }
  428. else
  429. {
  430. // TODO: Handle other chapter types?
  431. }
  432. pos += 14;
  433. }
  434. }
  435. finally
  436. {
  437. if (fileReader != null)
  438. {
  439. fileReader.Dispose();
  440. }
  441. if (fileStream != null)
  442. {
  443. fileStream.Dispose();
  444. }
  445. }
  446. }
  447. public void Initialize()
  448. {
  449. LoadStreamClips();
  450. Dictionary<string, List<double>> clipTimes = new Dictionary<string, List<double>>();
  451. foreach (TSStreamClip clip in StreamClips)
  452. {
  453. if (clip.AngleIndex == 0)
  454. {
  455. if (clipTimes.ContainsKey(clip.Name))
  456. {
  457. if (clipTimes[clip.Name].Contains(clip.TimeIn))
  458. {
  459. HasLoops = true;
  460. break;
  461. }
  462. else
  463. {
  464. clipTimes[clip.Name].Add(clip.TimeIn);
  465. }
  466. }
  467. else
  468. {
  469. clipTimes[clip.Name] = new List<double> { clip.TimeIn };
  470. }
  471. }
  472. }
  473. ClearBitrates();
  474. IsInitialized = true;
  475. }
  476. protected TSStream CreatePlaylistStream(byte[] data, ref int pos)
  477. {
  478. TSStream stream = null;
  479. int start = pos;
  480. int headerLength = data[pos++];
  481. int headerPos = pos;
  482. int headerType = data[pos++];
  483. int pid = 0;
  484. int subpathid = 0;
  485. int subclipid = 0;
  486. switch (headerType)
  487. {
  488. case 1:
  489. pid = ReadInt16(data, ref pos);
  490. break;
  491. case 2:
  492. subpathid = data[pos++];
  493. subclipid = data[pos++];
  494. pid = ReadInt16(data, ref pos);
  495. break;
  496. case 3:
  497. subpathid = data[pos++];
  498. pid = ReadInt16(data, ref pos);
  499. break;
  500. case 4:
  501. subpathid = data[pos++];
  502. subclipid = data[pos++];
  503. pid = ReadInt16(data, ref pos);
  504. break;
  505. default:
  506. break;
  507. }
  508. pos = headerPos + headerLength;
  509. int streamLength = data[pos++];
  510. int streamPos = pos;
  511. TSStreamType streamType = (TSStreamType)data[pos++];
  512. switch (streamType)
  513. {
  514. case TSStreamType.MVC_VIDEO:
  515. // TODO
  516. break;
  517. case TSStreamType.AVC_VIDEO:
  518. case TSStreamType.MPEG1_VIDEO:
  519. case TSStreamType.MPEG2_VIDEO:
  520. case TSStreamType.VC1_VIDEO:
  521. TSVideoFormat videoFormat = (TSVideoFormat)
  522. (data[pos] >> 4);
  523. TSFrameRate frameRate = (TSFrameRate)
  524. (data[pos] & 0xF);
  525. TSAspectRatio aspectRatio = (TSAspectRatio)
  526. (data[pos + 1] >> 4);
  527. stream = new TSVideoStream();
  528. ((TSVideoStream)stream).VideoFormat = videoFormat;
  529. ((TSVideoStream)stream).AspectRatio = aspectRatio;
  530. ((TSVideoStream)stream).FrameRate = frameRate;
  531. #if DEBUG
  532. Debug.WriteLine(string.Format(
  533. "\t{0} {1} {2} {3} {4}",
  534. pid,
  535. streamType,
  536. videoFormat,
  537. frameRate,
  538. aspectRatio));
  539. #endif
  540. break;
  541. case TSStreamType.AC3_AUDIO:
  542. case TSStreamType.AC3_PLUS_AUDIO:
  543. case TSStreamType.AC3_PLUS_SECONDARY_AUDIO:
  544. case TSStreamType.AC3_TRUE_HD_AUDIO:
  545. case TSStreamType.DTS_AUDIO:
  546. case TSStreamType.DTS_HD_AUDIO:
  547. case TSStreamType.DTS_HD_MASTER_AUDIO:
  548. case TSStreamType.DTS_HD_SECONDARY_AUDIO:
  549. case TSStreamType.LPCM_AUDIO:
  550. case TSStreamType.MPEG1_AUDIO:
  551. case TSStreamType.MPEG2_AUDIO:
  552. int audioFormat = ReadByte(data, ref pos);
  553. TSChannelLayout channelLayout = (TSChannelLayout)
  554. (audioFormat >> 4);
  555. TSSampleRate sampleRate = (TSSampleRate)
  556. (audioFormat & 0xF);
  557. string audioLanguage = ReadString(data, 3, ref pos);
  558. stream = new TSAudioStream();
  559. ((TSAudioStream)stream).ChannelLayout = channelLayout;
  560. ((TSAudioStream)stream).SampleRate = TSAudioStream.ConvertSampleRate(sampleRate);
  561. ((TSAudioStream)stream).LanguageCode = audioLanguage;
  562. #if DEBUG
  563. Debug.WriteLine(string.Format(
  564. "\t{0} {1} {2} {3} {4}",
  565. pid,
  566. streamType,
  567. audioLanguage,
  568. channelLayout,
  569. sampleRate));
  570. #endif
  571. break;
  572. case TSStreamType.INTERACTIVE_GRAPHICS:
  573. case TSStreamType.PRESENTATION_GRAPHICS:
  574. string graphicsLanguage = ReadString(data, 3, ref pos);
  575. stream = new TSGraphicsStream();
  576. ((TSGraphicsStream)stream).LanguageCode = graphicsLanguage;
  577. if (data[pos] != 0)
  578. {
  579. }
  580. #if DEBUG
  581. Debug.WriteLine(string.Format(
  582. "\t{0} {1} {2}",
  583. pid,
  584. streamType,
  585. graphicsLanguage));
  586. #endif
  587. break;
  588. case TSStreamType.SUBTITLE:
  589. int code = ReadByte(data, ref pos); // TODO
  590. string textLanguage = ReadString(data, 3, ref pos);
  591. stream = new TSTextStream();
  592. ((TSTextStream)stream).LanguageCode = textLanguage;
  593. #if DEBUG
  594. Debug.WriteLine(string.Format(
  595. "\t{0} {1} {2}",
  596. pid,
  597. streamType,
  598. textLanguage));
  599. #endif
  600. break;
  601. default:
  602. break;
  603. }
  604. pos = streamPos + streamLength;
  605. if (stream != null)
  606. {
  607. stream.PID = (ushort)pid;
  608. stream.StreamType = streamType;
  609. }
  610. return stream;
  611. }
  612. private void LoadStreamClips()
  613. {
  614. AngleClips.Clear();
  615. if (AngleCount > 0)
  616. {
  617. for (int angleIndex = 0; angleIndex < AngleCount; angleIndex++)
  618. {
  619. AngleClips.Add(new Dictionary<double, TSStreamClip>());
  620. }
  621. }
  622. TSStreamClip referenceClip = null;
  623. if (StreamClips.Count > 0)
  624. {
  625. referenceClip = StreamClips[0];
  626. }
  627. foreach (TSStreamClip clip in StreamClips)
  628. {
  629. if (clip.StreamClipFile.Streams.Count > referenceClip.StreamClipFile.Streams.Count)
  630. {
  631. referenceClip = clip;
  632. }
  633. else if (clip.Length > referenceClip.Length)
  634. {
  635. referenceClip = clip;
  636. }
  637. if (AngleCount > 0)
  638. {
  639. if (clip.AngleIndex == 0)
  640. {
  641. for (int angleIndex = 0; angleIndex < AngleCount; angleIndex++)
  642. {
  643. AngleClips[angleIndex][clip.RelativeTimeIn] = clip;
  644. }
  645. }
  646. else
  647. {
  648. AngleClips[clip.AngleIndex - 1][clip.RelativeTimeIn] = clip;
  649. }
  650. }
  651. }
  652. foreach (TSStream clipStream
  653. in referenceClip.StreamClipFile.Streams.Values)
  654. {
  655. if (!Streams.ContainsKey(clipStream.PID))
  656. {
  657. TSStream stream = clipStream.Clone();
  658. Streams[clipStream.PID] = stream;
  659. if (!IsCustom && !PlaylistStreams.ContainsKey(stream.PID))
  660. {
  661. stream.IsHidden = true;
  662. HasHiddenTracks = true;
  663. }
  664. if (stream.IsVideoStream)
  665. {
  666. VideoStreams.Add((TSVideoStream)stream);
  667. }
  668. else if (stream.IsAudioStream)
  669. {
  670. AudioStreams.Add((TSAudioStream)stream);
  671. }
  672. else if (stream.IsGraphicsStream)
  673. {
  674. GraphicsStreams.Add((TSGraphicsStream)stream);
  675. }
  676. else if (stream.IsTextStream)
  677. {
  678. TextStreams.Add((TSTextStream)stream);
  679. }
  680. }
  681. }
  682. if (referenceClip.StreamFile != null)
  683. {
  684. // TODO: Better way to add this in?
  685. if (BDInfoSettings.EnableSSIF &&
  686. referenceClip.StreamFile.InterleavedFile != null &&
  687. referenceClip.StreamFile.Streams.ContainsKey(4114) &&
  688. !Streams.ContainsKey(4114))
  689. {
  690. TSStream stream = referenceClip.StreamFile.Streams[4114].Clone();
  691. Streams[4114] = stream;
  692. if (stream.IsVideoStream)
  693. {
  694. VideoStreams.Add((TSVideoStream)stream);
  695. }
  696. }
  697. foreach (TSStream clipStream
  698. in referenceClip.StreamFile.Streams.Values)
  699. {
  700. if (Streams.ContainsKey(clipStream.PID))
  701. {
  702. TSStream stream = Streams[clipStream.PID];
  703. if (stream.StreamType != clipStream.StreamType) continue;
  704. if (clipStream.BitRate > stream.BitRate)
  705. {
  706. stream.BitRate = clipStream.BitRate;
  707. }
  708. stream.IsVBR = clipStream.IsVBR;
  709. if (stream.IsVideoStream &&
  710. clipStream.IsVideoStream)
  711. {
  712. ((TSVideoStream)stream).EncodingProfile =
  713. ((TSVideoStream)clipStream).EncodingProfile;
  714. }
  715. else if (stream.IsAudioStream &&
  716. clipStream.IsAudioStream)
  717. {
  718. TSAudioStream audioStream = (TSAudioStream)stream;
  719. TSAudioStream clipAudioStream = (TSAudioStream)clipStream;
  720. if (clipAudioStream.ChannelCount > audioStream.ChannelCount)
  721. {
  722. audioStream.ChannelCount = clipAudioStream.ChannelCount;
  723. }
  724. if (clipAudioStream.LFE > audioStream.LFE)
  725. {
  726. audioStream.LFE = clipAudioStream.LFE;
  727. }
  728. if (clipAudioStream.SampleRate > audioStream.SampleRate)
  729. {
  730. audioStream.SampleRate = clipAudioStream.SampleRate;
  731. }
  732. if (clipAudioStream.BitDepth > audioStream.BitDepth)
  733. {
  734. audioStream.BitDepth = clipAudioStream.BitDepth;
  735. }
  736. if (clipAudioStream.DialNorm < audioStream.DialNorm)
  737. {
  738. audioStream.DialNorm = clipAudioStream.DialNorm;
  739. }
  740. if (clipAudioStream.AudioMode != TSAudioMode.Unknown)
  741. {
  742. audioStream.AudioMode = clipAudioStream.AudioMode;
  743. }
  744. if (clipAudioStream.CoreStream != null &&
  745. audioStream.CoreStream == null)
  746. {
  747. audioStream.CoreStream = (TSAudioStream)
  748. clipAudioStream.CoreStream.Clone();
  749. }
  750. }
  751. }
  752. }
  753. }
  754. for (int i = 0; i < AngleCount; i++)
  755. {
  756. AngleStreams.Add(new Dictionary<ushort, TSStream>());
  757. }
  758. if (!BDInfoSettings.KeepStreamOrder)
  759. {
  760. VideoStreams.Sort(CompareVideoStreams);
  761. }
  762. foreach (TSStream stream in VideoStreams)
  763. {
  764. SortedStreams.Add(stream);
  765. for (int i = 0; i < AngleCount; i++)
  766. {
  767. TSStream angleStream = stream.Clone();
  768. angleStream.AngleIndex = i + 1;
  769. AngleStreams[i][angleStream.PID] = angleStream;
  770. SortedStreams.Add(angleStream);
  771. }
  772. }
  773. if (!BDInfoSettings.KeepStreamOrder)
  774. {
  775. AudioStreams.Sort(CompareAudioStreams);
  776. }
  777. foreach (TSStream stream in AudioStreams)
  778. {
  779. SortedStreams.Add(stream);
  780. }
  781. if (!BDInfoSettings.KeepStreamOrder)
  782. {
  783. GraphicsStreams.Sort(CompareGraphicsStreams);
  784. }
  785. foreach (TSStream stream in GraphicsStreams)
  786. {
  787. SortedStreams.Add(stream);
  788. }
  789. if (!BDInfoSettings.KeepStreamOrder)
  790. {
  791. TextStreams.Sort(CompareTextStreams);
  792. }
  793. foreach (TSStream stream in TextStreams)
  794. {
  795. SortedStreams.Add(stream);
  796. }
  797. }
  798. public void ClearBitrates()
  799. {
  800. foreach (TSStreamClip clip in StreamClips)
  801. {
  802. clip.PayloadBytes = 0;
  803. clip.PacketCount = 0;
  804. clip.PacketSeconds = 0;
  805. if (clip.StreamFile != null)
  806. {
  807. foreach (TSStream stream in clip.StreamFile.Streams.Values)
  808. {
  809. stream.PayloadBytes = 0;
  810. stream.PacketCount = 0;
  811. stream.PacketSeconds = 0;
  812. }
  813. if (clip.StreamFile != null &&
  814. clip.StreamFile.StreamDiagnostics != null)
  815. {
  816. clip.StreamFile.StreamDiagnostics.Clear();
  817. }
  818. }
  819. }
  820. foreach (TSStream stream in SortedStreams)
  821. {
  822. stream.PayloadBytes = 0;
  823. stream.PacketCount = 0;
  824. stream.PacketSeconds = 0;
  825. }
  826. }
  827. public bool IsValid
  828. {
  829. get
  830. {
  831. if (!IsInitialized) return false;
  832. if (BDInfoSettings.FilterShortPlaylists &&
  833. TotalLength < BDInfoSettings.FilterShortPlaylistsValue)
  834. {
  835. return false;
  836. }
  837. if (HasLoops &&
  838. BDInfoSettings.FilterLoopingPlaylists)
  839. {
  840. return false;
  841. }
  842. return true;
  843. }
  844. }
  845. public int CompareVideoStreams(
  846. TSVideoStream x,
  847. TSVideoStream y)
  848. {
  849. if (x == null && y == null)
  850. {
  851. return 0;
  852. }
  853. else if (x == null && y != null)
  854. {
  855. return 1;
  856. }
  857. else if (x != null && y == null)
  858. {
  859. return -1;
  860. }
  861. else
  862. {
  863. if (x.Height > y.Height)
  864. {
  865. return -1;
  866. }
  867. else if (y.Height > x.Height)
  868. {
  869. return 1;
  870. }
  871. else if (x.PID > y.PID)
  872. {
  873. return 1;
  874. }
  875. else if (y.PID > x.PID)
  876. {
  877. return -1;
  878. }
  879. else
  880. {
  881. return 0;
  882. }
  883. }
  884. }
  885. public int CompareAudioStreams(
  886. TSAudioStream x,
  887. TSAudioStream y)
  888. {
  889. if (x == y)
  890. {
  891. return 0;
  892. }
  893. else if (x == null && y == null)
  894. {
  895. return 0;
  896. }
  897. else if (x == null && y != null)
  898. {
  899. return -1;
  900. }
  901. else if (x != null && y == null)
  902. {
  903. return 1;
  904. }
  905. else
  906. {
  907. if (x.ChannelCount > y.ChannelCount)
  908. {
  909. return -1;
  910. }
  911. else if (y.ChannelCount > x.ChannelCount)
  912. {
  913. return 1;
  914. }
  915. else
  916. {
  917. int sortX = GetStreamTypeSortIndex(x.StreamType);
  918. int sortY = GetStreamTypeSortIndex(y.StreamType);
  919. if (sortX > sortY)
  920. {
  921. return -1;
  922. }
  923. else if (sortY > sortX)
  924. {
  925. return 1;
  926. }
  927. else
  928. {
  929. if (x.LanguageCode == "eng")
  930. {
  931. return -1;
  932. }
  933. else if (y.LanguageCode == "eng")
  934. {
  935. return 1;
  936. }
  937. else if (x.LanguageCode != y.LanguageCode)
  938. {
  939. return string.Compare(
  940. x.LanguageName, y.LanguageName);
  941. }
  942. else if (x.PID < y.PID)
  943. {
  944. return -1;
  945. }
  946. else if (y.PID < x.PID)
  947. {
  948. return 1;
  949. }
  950. return 0;
  951. }
  952. }
  953. }
  954. }
  955. public int CompareTextStreams(
  956. TSTextStream x,
  957. TSTextStream y)
  958. {
  959. if (x == y)
  960. {
  961. return 0;
  962. }
  963. else if (x == null && y == null)
  964. {
  965. return 0;
  966. }
  967. else if (x == null && y != null)
  968. {
  969. return -1;
  970. }
  971. else if (x != null && y == null)
  972. {
  973. return 1;
  974. }
  975. else
  976. {
  977. if (x.LanguageCode == "eng")
  978. {
  979. return -1;
  980. }
  981. else if (y.LanguageCode == "eng")
  982. {
  983. return 1;
  984. }
  985. else
  986. {
  987. if (x.LanguageCode == y.LanguageCode)
  988. {
  989. if (x.PID > y.PID)
  990. {
  991. return 1;
  992. }
  993. else if (y.PID > x.PID)
  994. {
  995. return -1;
  996. }
  997. else
  998. {
  999. return 0;
  1000. }
  1001. }
  1002. else
  1003. {
  1004. return string.Compare(
  1005. x.LanguageName, y.LanguageName);
  1006. }
  1007. }
  1008. }
  1009. }
  1010. private int CompareGraphicsStreams(
  1011. TSGraphicsStream x,
  1012. TSGraphicsStream y)
  1013. {
  1014. if (x == y)
  1015. {
  1016. return 0;
  1017. }
  1018. else if (x == null && y == null)
  1019. {
  1020. return 0;
  1021. }
  1022. else if (x == null && y != null)
  1023. {
  1024. return -1;
  1025. }
  1026. else if (x != null && y == null)
  1027. {
  1028. return 1;
  1029. }
  1030. else
  1031. {
  1032. int sortX = GetStreamTypeSortIndex(x.StreamType);
  1033. int sortY = GetStreamTypeSortIndex(y.StreamType);
  1034. if (sortX > sortY)
  1035. {
  1036. return -1;
  1037. }
  1038. else if (sortY > sortX)
  1039. {
  1040. return 1;
  1041. }
  1042. else if (x.LanguageCode == "eng")
  1043. {
  1044. return -1;
  1045. }
  1046. else if (y.LanguageCode == "eng")
  1047. {
  1048. return 1;
  1049. }
  1050. else
  1051. {
  1052. if (x.LanguageCode == y.LanguageCode)
  1053. {
  1054. if (x.PID > y.PID)
  1055. {
  1056. return 1;
  1057. }
  1058. else if (y.PID > x.PID)
  1059. {
  1060. return -1;
  1061. }
  1062. else
  1063. {
  1064. return 0;
  1065. }
  1066. }
  1067. else
  1068. {
  1069. return string.Compare(x.LanguageName, y.LanguageName);
  1070. }
  1071. }
  1072. }
  1073. }
  1074. private int GetStreamTypeSortIndex(TSStreamType streamType)
  1075. {
  1076. switch (streamType)
  1077. {
  1078. case TSStreamType.Unknown:
  1079. return 0;
  1080. case TSStreamType.MPEG1_VIDEO:
  1081. return 1;
  1082. case TSStreamType.MPEG2_VIDEO:
  1083. return 2;
  1084. case TSStreamType.AVC_VIDEO:
  1085. return 3;
  1086. case TSStreamType.VC1_VIDEO:
  1087. return 4;
  1088. case TSStreamType.MVC_VIDEO:
  1089. return 5;
  1090. case TSStreamType.MPEG1_AUDIO:
  1091. return 1;
  1092. case TSStreamType.MPEG2_AUDIO:
  1093. return 2;
  1094. case TSStreamType.AC3_PLUS_SECONDARY_AUDIO:
  1095. return 3;
  1096. case TSStreamType.DTS_HD_SECONDARY_AUDIO:
  1097. return 4;
  1098. case TSStreamType.AC3_AUDIO:
  1099. return 5;
  1100. case TSStreamType.DTS_AUDIO:
  1101. return 6;
  1102. case TSStreamType.AC3_PLUS_AUDIO:
  1103. return 7;
  1104. case TSStreamType.DTS_HD_AUDIO:
  1105. return 8;
  1106. case TSStreamType.AC3_TRUE_HD_AUDIO:
  1107. return 9;
  1108. case TSStreamType.DTS_HD_MASTER_AUDIO:
  1109. return 10;
  1110. case TSStreamType.LPCM_AUDIO:
  1111. return 11;
  1112. case TSStreamType.SUBTITLE:
  1113. return 1;
  1114. case TSStreamType.INTERACTIVE_GRAPHICS:
  1115. return 2;
  1116. case TSStreamType.PRESENTATION_GRAPHICS:
  1117. return 3;
  1118. default:
  1119. return 0;
  1120. }
  1121. }
  1122. protected string ReadString(
  1123. byte[] data,
  1124. int count,
  1125. ref int pos)
  1126. {
  1127. string val =
  1128. _textEncoding.GetASCIIEncoding().GetString(data, pos, count);
  1129. pos += count;
  1130. return val;
  1131. }
  1132. protected int ReadInt32(
  1133. byte[] data,
  1134. ref int pos)
  1135. {
  1136. int val =
  1137. ((int)data[pos] << 24) +
  1138. ((int)data[pos + 1] << 16) +
  1139. ((int)data[pos + 2] << 8) +
  1140. ((int)data[pos + 3]);
  1141. pos += 4;
  1142. return val;
  1143. }
  1144. protected int ReadInt16(
  1145. byte[] data,
  1146. ref int pos)
  1147. {
  1148. int val =
  1149. ((int)data[pos] << 8) +
  1150. ((int)data[pos + 1]);
  1151. pos += 2;
  1152. return val;
  1153. }
  1154. protected byte ReadByte(
  1155. byte[] data,
  1156. ref int pos)
  1157. {
  1158. return data[pos++];
  1159. }
  1160. }
  1161. }