ProbeResultNormalizer.cs 50 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321
  1. using MediaBrowser.Model.Dto;
  2. using MediaBrowser.Model.Entities;
  3. using MediaBrowser.Model.Extensions;
  4. using System;
  5. using System.Collections.Generic;
  6. using System.Globalization;
  7. using System.IO;
  8. using System.Linq;
  9. using System.Text;
  10. using System.Xml;
  11. using MediaBrowser.Model.IO;
  12. using MediaBrowser.Common.IO;
  13. using MediaBrowser.Controller.IO;
  14. using MediaBrowser.Controller.Library;
  15. using MediaBrowser.Model.IO;
  16. using MediaBrowser.Model.Logging;
  17. using MediaBrowser.Model.MediaInfo;
  18. namespace MediaBrowser.MediaEncoding.Probing
  19. {
  20. public class ProbeResultNormalizer
  21. {
  22. private readonly CultureInfo _usCulture = new CultureInfo("en-US");
  23. private readonly ILogger _logger;
  24. private readonly IFileSystem _fileSystem;
  25. private readonly IMemoryStreamFactory _memoryStreamProvider;
  26. public ProbeResultNormalizer(ILogger logger, IFileSystem fileSystem, IMemoryStreamFactory memoryStreamProvider)
  27. {
  28. _logger = logger;
  29. _fileSystem = fileSystem;
  30. _memoryStreamProvider = memoryStreamProvider;
  31. }
  32. public MediaInfo GetMediaInfo(InternalMediaInfoResult data, VideoType videoType, bool isAudio, string path, MediaProtocol protocol)
  33. {
  34. var info = new MediaInfo
  35. {
  36. Path = path,
  37. Protocol = protocol
  38. };
  39. FFProbeHelpers.NormalizeFFProbeResult(data);
  40. SetSize(data, info);
  41. var internalStreams = data.streams ?? new MediaStreamInfo[] { };
  42. info.MediaStreams = internalStreams.Select(s => GetMediaStream(isAudio, s, data.format))
  43. .Where(i => i != null)
  44. // Drop subtitle streams if we don't know the codec because it will just cause failures if we don't know how to handle them
  45. .Where(i => i.Type != MediaStreamType.Subtitle || !string.IsNullOrWhiteSpace(i.Codec))
  46. .ToList();
  47. if (data.format != null)
  48. {
  49. info.Container = data.format.format_name;
  50. if (!string.IsNullOrEmpty(data.format.bit_rate))
  51. {
  52. int value;
  53. if (int.TryParse(data.format.bit_rate, NumberStyles.Any, _usCulture, out value))
  54. {
  55. info.Bitrate = value;
  56. }
  57. }
  58. }
  59. var tags = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
  60. var tagStreamType = isAudio ? "audio" : "video";
  61. if (data.streams != null)
  62. {
  63. var tagStream = data.streams.FirstOrDefault(i => string.Equals(i.codec_type, tagStreamType, StringComparison.OrdinalIgnoreCase));
  64. if (tagStream != null && tagStream.tags != null)
  65. {
  66. foreach (var pair in tagStream.tags)
  67. {
  68. tags[pair.Key] = pair.Value;
  69. }
  70. }
  71. }
  72. if (data.format != null && data.format.tags != null)
  73. {
  74. foreach (var pair in data.format.tags)
  75. {
  76. tags[pair.Key] = pair.Value;
  77. }
  78. }
  79. FetchGenres(info, tags);
  80. var shortOverview = FFProbeHelpers.GetDictionaryValue(tags, "description");
  81. var overview = FFProbeHelpers.GetDictionaryValue(tags, "synopsis");
  82. if (string.IsNullOrWhiteSpace(overview))
  83. {
  84. overview = shortOverview;
  85. shortOverview = null;
  86. }
  87. if (string.IsNullOrWhiteSpace(overview))
  88. {
  89. overview = FFProbeHelpers.GetDictionaryValue(tags, "desc");
  90. }
  91. if (!string.IsNullOrWhiteSpace(overview))
  92. {
  93. info.Overview = overview;
  94. }
  95. if (!string.IsNullOrWhiteSpace(shortOverview))
  96. {
  97. info.ShortOverview = shortOverview;
  98. }
  99. var title = FFProbeHelpers.GetDictionaryValue(tags, "title");
  100. if (!string.IsNullOrWhiteSpace(title))
  101. {
  102. info.Name = title;
  103. }
  104. info.ProductionYear = FFProbeHelpers.GetDictionaryNumericValue(tags, "date");
  105. // Several different forms of retaildate
  106. info.PremiereDate = FFProbeHelpers.GetDictionaryDateTime(tags, "retaildate") ??
  107. FFProbeHelpers.GetDictionaryDateTime(tags, "retail date") ??
  108. FFProbeHelpers.GetDictionaryDateTime(tags, "retail_date") ??
  109. FFProbeHelpers.GetDictionaryDateTime(tags, "date");
  110. if (isAudio)
  111. {
  112. SetAudioRuntimeTicks(data, info);
  113. // tags are normally located under data.format, but we've seen some cases with ogg where they're part of the info stream
  114. // so let's create a combined list of both
  115. SetAudioInfoFromTags(info, tags);
  116. }
  117. else
  118. {
  119. FetchStudios(info, tags, "copyright");
  120. var iTunEXTC = FFProbeHelpers.GetDictionaryValue(tags, "iTunEXTC");
  121. if (!string.IsNullOrWhiteSpace(iTunEXTC))
  122. {
  123. var parts = iTunEXTC.Split(new[] { '|' }, StringSplitOptions.RemoveEmptyEntries);
  124. // Example
  125. // mpaa|G|100|For crude humor
  126. if (parts.Length > 1)
  127. {
  128. info.OfficialRating = parts[1];
  129. if (parts.Length > 3)
  130. {
  131. info.OfficialRatingDescription = parts[3];
  132. }
  133. }
  134. }
  135. var itunesXml = FFProbeHelpers.GetDictionaryValue(tags, "iTunMOVI");
  136. if (!string.IsNullOrWhiteSpace(itunesXml))
  137. {
  138. FetchFromItunesInfo(itunesXml, info);
  139. }
  140. if (data.format != null && !string.IsNullOrEmpty(data.format.duration))
  141. {
  142. info.RunTimeTicks = TimeSpan.FromSeconds(double.Parse(data.format.duration, _usCulture)).Ticks;
  143. }
  144. FetchWtvInfo(info, data);
  145. if (data.Chapters != null)
  146. {
  147. info.Chapters = data.Chapters.Select(GetChapterInfo).ToList();
  148. }
  149. ExtractTimestamp(info);
  150. var stereoMode = GetDictionaryValue(tags, "stereo_mode");
  151. if (string.Equals(stereoMode, "left_right", StringComparison.OrdinalIgnoreCase))
  152. {
  153. info.Video3DFormat = Video3DFormat.FullSideBySide;
  154. }
  155. var videoStreamsBitrate = info.MediaStreams.Where(i => i.Type == MediaStreamType.Video).Select(i => i.BitRate ?? 0).Sum();
  156. // If ffprobe reported the container bitrate as being the same as the video stream bitrate, then it's wrong
  157. if (videoStreamsBitrate == (info.Bitrate ?? 0))
  158. {
  159. var streamBitrates = info.MediaStreams.Where(i => !i.IsExternal).Select(i => i.BitRate ?? 0).Sum();
  160. if (streamBitrates > (info.Bitrate ?? 0))
  161. {
  162. info.Bitrate = streamBitrates;
  163. }
  164. }
  165. }
  166. return info;
  167. }
  168. private void FetchFromItunesInfo(string xml, MediaInfo info)
  169. {
  170. // Make things simpler and strip out the dtd
  171. var plistIndex = xml.IndexOf("<plist", StringComparison.OrdinalIgnoreCase);
  172. if (plistIndex != -1)
  173. {
  174. xml = xml.Substring(plistIndex);
  175. }
  176. xml = "<?xml version=\"1.0\"?>" + xml;
  177. // <?xml version=\"1.0\" encoding=\"UTF-8\"?>\n<!DOCTYPE plist PUBLIC \"-//Apple//DTD PLIST 1.0//EN\" \"http://www.apple.com/DTDs/PropertyList-1.0.dtd\">\n<plist version=\"1.0\">\n<dict>\n\t<key>cast</key>\n\t<array>\n\t\t<dict>\n\t\t\t<key>name</key>\n\t\t\t<string>Blender Foundation</string>\n\t\t</dict>\n\t\t<dict>\n\t\t\t<key>name</key>\n\t\t\t<string>Janus Bager Kristensen</string>\n\t\t</dict>\n\t</array>\n\t<key>directors</key>\n\t<array>\n\t\t<dict>\n\t\t\t<key>name</key>\n\t\t\t<string>Sacha Goedegebure</string>\n\t\t</dict>\n\t</array>\n\t<key>studio</key>\n\t<string>Blender Foundation</string>\n</dict>\n</plist>\n
  178. using (var stream = _memoryStreamProvider.CreateNew(Encoding.UTF8.GetBytes(xml)))
  179. {
  180. using (var streamReader = new StreamReader(stream))
  181. {
  182. try
  183. {
  184. // Use XmlReader for best performance
  185. using (var reader = XmlReader.Create(streamReader))
  186. {
  187. reader.MoveToContent();
  188. reader.Read();
  189. // Loop through each element
  190. while (!reader.EOF && reader.ReadState == ReadState.Interactive)
  191. {
  192. if (reader.NodeType == XmlNodeType.Element)
  193. {
  194. switch (reader.Name)
  195. {
  196. case "dict":
  197. if (reader.IsEmptyElement)
  198. {
  199. reader.Read();
  200. continue;
  201. }
  202. using (var subtree = reader.ReadSubtree())
  203. {
  204. ReadFromDictNode(subtree, info);
  205. }
  206. break;
  207. default:
  208. reader.Skip();
  209. break;
  210. }
  211. }
  212. else
  213. {
  214. reader.Read();
  215. }
  216. }
  217. }
  218. }
  219. catch (XmlException)
  220. {
  221. // I've seen probe examples where the iTunMOVI value is just "<"
  222. // So we should not allow this to fail the entire probing operation
  223. }
  224. }
  225. }
  226. }
  227. private void ReadFromDictNode(XmlReader reader, MediaInfo info)
  228. {
  229. string currentKey = null;
  230. List<NameValuePair> pairs = new List<NameValuePair>();
  231. reader.MoveToContent();
  232. reader.Read();
  233. // Loop through each element
  234. while (!reader.EOF && reader.ReadState == ReadState.Interactive)
  235. {
  236. if (reader.NodeType == XmlNodeType.Element)
  237. {
  238. switch (reader.Name)
  239. {
  240. case "key":
  241. if (!string.IsNullOrWhiteSpace(currentKey))
  242. {
  243. ProcessPairs(currentKey, pairs, info);
  244. }
  245. currentKey = reader.ReadElementContentAsString();
  246. pairs = new List<NameValuePair>();
  247. break;
  248. case "string":
  249. var value = reader.ReadElementContentAsString();
  250. if (!string.IsNullOrWhiteSpace(value))
  251. {
  252. pairs.Add(new NameValuePair
  253. {
  254. Name = value,
  255. Value = value
  256. });
  257. }
  258. break;
  259. case "array":
  260. if (reader.IsEmptyElement)
  261. {
  262. reader.Read();
  263. continue;
  264. }
  265. using (var subtree = reader.ReadSubtree())
  266. {
  267. if (!string.IsNullOrWhiteSpace(currentKey))
  268. {
  269. pairs.AddRange(ReadValueArray(subtree));
  270. }
  271. }
  272. break;
  273. default:
  274. reader.Skip();
  275. break;
  276. }
  277. }
  278. else
  279. {
  280. reader.Read();
  281. }
  282. }
  283. }
  284. private List<NameValuePair> ReadValueArray(XmlReader reader)
  285. {
  286. List<NameValuePair> pairs = new List<NameValuePair>();
  287. reader.MoveToContent();
  288. reader.Read();
  289. // Loop through each element
  290. while (!reader.EOF && reader.ReadState == ReadState.Interactive)
  291. {
  292. if (reader.NodeType == XmlNodeType.Element)
  293. {
  294. switch (reader.Name)
  295. {
  296. case "dict":
  297. if (reader.IsEmptyElement)
  298. {
  299. reader.Read();
  300. continue;
  301. }
  302. using (var subtree = reader.ReadSubtree())
  303. {
  304. var dict = GetNameValuePair(subtree);
  305. if (dict != null)
  306. {
  307. pairs.Add(dict);
  308. }
  309. }
  310. break;
  311. default:
  312. reader.Skip();
  313. break;
  314. }
  315. }
  316. else
  317. {
  318. reader.Read();
  319. }
  320. }
  321. return pairs;
  322. }
  323. private void ProcessPairs(string key, List<NameValuePair> pairs, MediaInfo info)
  324. {
  325. if (string.Equals(key, "studio", StringComparison.OrdinalIgnoreCase))
  326. {
  327. foreach (var pair in pairs)
  328. {
  329. info.Studios.Add(pair.Value);
  330. }
  331. info.Studios = info.Studios
  332. .Where(i => !string.IsNullOrWhiteSpace(i))
  333. .Distinct(StringComparer.OrdinalIgnoreCase)
  334. .ToList();
  335. }
  336. else if (string.Equals(key, "screenwriters", StringComparison.OrdinalIgnoreCase))
  337. {
  338. foreach (var pair in pairs)
  339. {
  340. info.People.Add(new BaseItemPerson
  341. {
  342. Name = pair.Value,
  343. Type = PersonType.Writer
  344. });
  345. }
  346. }
  347. else if (string.Equals(key, "producers", StringComparison.OrdinalIgnoreCase))
  348. {
  349. foreach (var pair in pairs)
  350. {
  351. info.People.Add(new BaseItemPerson
  352. {
  353. Name = pair.Value,
  354. Type = PersonType.Producer
  355. });
  356. }
  357. }
  358. else if (string.Equals(key, "directors", StringComparison.OrdinalIgnoreCase))
  359. {
  360. foreach (var pair in pairs)
  361. {
  362. info.People.Add(new BaseItemPerson
  363. {
  364. Name = pair.Value,
  365. Type = PersonType.Director
  366. });
  367. }
  368. }
  369. }
  370. private NameValuePair GetNameValuePair(XmlReader reader)
  371. {
  372. string name = null;
  373. string value = null;
  374. reader.MoveToContent();
  375. reader.Read();
  376. // Loop through each element
  377. while (!reader.EOF && reader.ReadState == ReadState.Interactive)
  378. {
  379. if (reader.NodeType == XmlNodeType.Element)
  380. {
  381. switch (reader.Name)
  382. {
  383. case "key":
  384. name = reader.ReadElementContentAsString();
  385. break;
  386. case "string":
  387. value = reader.ReadElementContentAsString();
  388. break;
  389. default:
  390. reader.Skip();
  391. break;
  392. }
  393. }
  394. else
  395. {
  396. reader.Read();
  397. }
  398. }
  399. if (string.IsNullOrWhiteSpace(name) ||
  400. string.IsNullOrWhiteSpace(value))
  401. {
  402. return null;
  403. }
  404. return new NameValuePair
  405. {
  406. Name = name,
  407. Value = value
  408. };
  409. }
  410. private string NormalizeSubtitleCodec(string codec)
  411. {
  412. if (string.Equals(codec, "dvb_subtitle", StringComparison.OrdinalIgnoreCase))
  413. {
  414. codec = "dvbsub";
  415. }
  416. else if ((codec ?? string.Empty).IndexOf("PGS", StringComparison.OrdinalIgnoreCase) != -1)
  417. {
  418. codec = "PGSSUB";
  419. }
  420. else if ((codec ?? string.Empty).IndexOf("DVD", StringComparison.OrdinalIgnoreCase) != -1)
  421. {
  422. codec = "DVDSUB";
  423. }
  424. return codec;
  425. }
  426. /// <summary>
  427. /// Converts ffprobe stream info to our MediaStream class
  428. /// </summary>
  429. /// <param name="isAudio">if set to <c>true</c> [is info].</param>
  430. /// <param name="streamInfo">The stream info.</param>
  431. /// <param name="formatInfo">The format info.</param>
  432. /// <returns>MediaStream.</returns>
  433. private MediaStream GetMediaStream(bool isAudio, MediaStreamInfo streamInfo, MediaFormatInfo formatInfo)
  434. {
  435. // These are mp4 chapters
  436. if (string.Equals(streamInfo.codec_name, "mov_text", StringComparison.OrdinalIgnoreCase))
  437. {
  438. // Edit: but these are also sometimes subtitles?
  439. //return null;
  440. }
  441. var stream = new MediaStream
  442. {
  443. Codec = streamInfo.codec_name,
  444. Profile = streamInfo.profile,
  445. Level = streamInfo.level,
  446. Index = streamInfo.index,
  447. PixelFormat = streamInfo.pix_fmt,
  448. NalLengthSize = streamInfo.nal_length_size,
  449. TimeBase = streamInfo.time_base,
  450. CodecTimeBase = streamInfo.codec_time_base
  451. };
  452. if (string.Equals(streamInfo.is_avc, "true", StringComparison.OrdinalIgnoreCase) ||
  453. string.Equals(streamInfo.is_avc, "1", StringComparison.OrdinalIgnoreCase))
  454. {
  455. stream.IsAVC = true;
  456. }
  457. else if (string.Equals(streamInfo.is_avc, "false", StringComparison.OrdinalIgnoreCase) ||
  458. string.Equals(streamInfo.is_avc, "0", StringComparison.OrdinalIgnoreCase))
  459. {
  460. stream.IsAVC = false;
  461. }
  462. // Filter out junk
  463. if (!string.IsNullOrWhiteSpace(streamInfo.codec_tag_string) && streamInfo.codec_tag_string.IndexOf("[0]", StringComparison.OrdinalIgnoreCase) == -1)
  464. {
  465. stream.CodecTag = streamInfo.codec_tag_string;
  466. }
  467. if (streamInfo.tags != null)
  468. {
  469. stream.Language = GetDictionaryValue(streamInfo.tags, "language");
  470. stream.Comment = GetDictionaryValue(streamInfo.tags, "comment");
  471. stream.Title = GetDictionaryValue(streamInfo.tags, "title");
  472. }
  473. if (string.Equals(streamInfo.codec_type, "audio", StringComparison.OrdinalIgnoreCase))
  474. {
  475. stream.Type = MediaStreamType.Audio;
  476. stream.Channels = streamInfo.channels;
  477. if (!string.IsNullOrEmpty(streamInfo.sample_rate))
  478. {
  479. int value;
  480. if (int.TryParse(streamInfo.sample_rate, NumberStyles.Any, _usCulture, out value))
  481. {
  482. stream.SampleRate = value;
  483. }
  484. }
  485. stream.ChannelLayout = ParseChannelLayout(streamInfo.channel_layout);
  486. if (streamInfo.bits_per_sample > 0)
  487. {
  488. stream.BitDepth = streamInfo.bits_per_sample;
  489. }
  490. else if (streamInfo.bits_per_raw_sample > 0)
  491. {
  492. stream.BitDepth = streamInfo.bits_per_raw_sample;
  493. }
  494. }
  495. else if (string.Equals(streamInfo.codec_type, "subtitle", StringComparison.OrdinalIgnoreCase))
  496. {
  497. stream.Type = MediaStreamType.Subtitle;
  498. stream.Codec = NormalizeSubtitleCodec(stream.Codec);
  499. }
  500. else if (string.Equals(streamInfo.codec_type, "video", StringComparison.OrdinalIgnoreCase))
  501. {
  502. stream.Type = isAudio || string.Equals(stream.Codec, "mjpeg", StringComparison.OrdinalIgnoreCase) || string.Equals(stream.Codec, "gif", StringComparison.OrdinalIgnoreCase) || string.Equals(stream.Codec, "png", StringComparison.OrdinalIgnoreCase)
  503. ? MediaStreamType.EmbeddedImage
  504. : MediaStreamType.Video;
  505. stream.Width = streamInfo.width;
  506. stream.Height = streamInfo.height;
  507. stream.AspectRatio = GetAspectRatio(streamInfo);
  508. stream.AverageFrameRate = GetFrameRate(streamInfo.avg_frame_rate);
  509. stream.RealFrameRate = GetFrameRate(streamInfo.r_frame_rate);
  510. if (streamInfo.bits_per_sample > 0)
  511. {
  512. stream.BitDepth = streamInfo.bits_per_sample;
  513. }
  514. else if (streamInfo.bits_per_raw_sample > 0)
  515. {
  516. stream.BitDepth = streamInfo.bits_per_raw_sample;
  517. }
  518. //stream.IsAnamorphic = string.Equals(streamInfo.sample_aspect_ratio, "0:1", StringComparison.OrdinalIgnoreCase) ||
  519. // string.Equals(stream.AspectRatio, "2.35:1", StringComparison.OrdinalIgnoreCase) ||
  520. // string.Equals(stream.AspectRatio, "2.40:1", StringComparison.OrdinalIgnoreCase);
  521. // http://stackoverflow.com/questions/17353387/how-to-detect-anamorphic-video-with-ffprobe
  522. stream.IsAnamorphic = string.Equals(streamInfo.sample_aspect_ratio, "0:1", StringComparison.OrdinalIgnoreCase);
  523. if (streamInfo.refs > 0)
  524. {
  525. stream.RefFrames = streamInfo.refs;
  526. }
  527. }
  528. else
  529. {
  530. return null;
  531. }
  532. // Get stream bitrate
  533. var bitrate = 0;
  534. if (!string.IsNullOrEmpty(streamInfo.bit_rate))
  535. {
  536. int value;
  537. if (int.TryParse(streamInfo.bit_rate, NumberStyles.Any, _usCulture, out value))
  538. {
  539. bitrate = value;
  540. }
  541. }
  542. if (bitrate == 0 && formatInfo != null && !string.IsNullOrEmpty(formatInfo.bit_rate) && stream.Type == MediaStreamType.Video)
  543. {
  544. // If the stream info doesn't have a bitrate get the value from the media format info
  545. int value;
  546. if (int.TryParse(formatInfo.bit_rate, NumberStyles.Any, _usCulture, out value))
  547. {
  548. bitrate = value;
  549. }
  550. }
  551. if (bitrate > 0)
  552. {
  553. stream.BitRate = bitrate;
  554. }
  555. if (streamInfo.disposition != null)
  556. {
  557. var isDefault = GetDictionaryValue(streamInfo.disposition, "default");
  558. var isForced = GetDictionaryValue(streamInfo.disposition, "forced");
  559. stream.IsDefault = string.Equals(isDefault, "1", StringComparison.OrdinalIgnoreCase);
  560. stream.IsForced = string.Equals(isForced, "1", StringComparison.OrdinalIgnoreCase);
  561. }
  562. NormalizeStreamTitle(stream);
  563. return stream;
  564. }
  565. private void NormalizeStreamTitle(MediaStream stream)
  566. {
  567. if (string.Equals(stream.Title, "cc", StringComparison.OrdinalIgnoreCase))
  568. {
  569. stream.Title = null;
  570. }
  571. if (stream.Type == MediaStreamType.EmbeddedImage)
  572. {
  573. stream.Title = null;
  574. }
  575. }
  576. /// <summary>
  577. /// Gets a string from an FFProbeResult tags dictionary
  578. /// </summary>
  579. /// <param name="tags">The tags.</param>
  580. /// <param name="key">The key.</param>
  581. /// <returns>System.String.</returns>
  582. private string GetDictionaryValue(Dictionary<string, string> tags, string key)
  583. {
  584. if (tags == null)
  585. {
  586. return null;
  587. }
  588. string val;
  589. tags.TryGetValue(key, out val);
  590. return val;
  591. }
  592. private string ParseChannelLayout(string input)
  593. {
  594. if (string.IsNullOrEmpty(input))
  595. {
  596. return input;
  597. }
  598. return input.Split('(').FirstOrDefault();
  599. }
  600. private string GetAspectRatio(MediaStreamInfo info)
  601. {
  602. var original = info.display_aspect_ratio;
  603. int height;
  604. int width;
  605. var parts = (original ?? string.Empty).Split(':');
  606. if (!(parts.Length == 2 &&
  607. int.TryParse(parts[0], NumberStyles.Any, _usCulture, out width) &&
  608. int.TryParse(parts[1], NumberStyles.Any, _usCulture, out height) &&
  609. width > 0 &&
  610. height > 0))
  611. {
  612. width = info.width;
  613. height = info.height;
  614. }
  615. if (width > 0 && height > 0)
  616. {
  617. double ratio = width;
  618. ratio /= height;
  619. if (IsClose(ratio, 1.777777778, .03))
  620. {
  621. return "16:9";
  622. }
  623. if (IsClose(ratio, 1.3333333333, .05))
  624. {
  625. return "4:3";
  626. }
  627. if (IsClose(ratio, 1.41))
  628. {
  629. return "1.41:1";
  630. }
  631. if (IsClose(ratio, 1.5))
  632. {
  633. return "1.5:1";
  634. }
  635. if (IsClose(ratio, 1.6))
  636. {
  637. return "1.6:1";
  638. }
  639. if (IsClose(ratio, 1.66666666667))
  640. {
  641. return "5:3";
  642. }
  643. if (IsClose(ratio, 1.85, .02))
  644. {
  645. return "1.85:1";
  646. }
  647. if (IsClose(ratio, 2.35, .025))
  648. {
  649. return "2.35:1";
  650. }
  651. if (IsClose(ratio, 2.4, .025))
  652. {
  653. return "2.40:1";
  654. }
  655. }
  656. return original;
  657. }
  658. private bool IsClose(double d1, double d2, double variance = .005)
  659. {
  660. return Math.Abs(d1 - d2) <= variance;
  661. }
  662. /// <summary>
  663. /// Gets a frame rate from a string value in ffprobe output
  664. /// This could be a number or in the format of 2997/125.
  665. /// </summary>
  666. /// <param name="value">The value.</param>
  667. /// <returns>System.Nullable{System.Single}.</returns>
  668. private float? GetFrameRate(string value)
  669. {
  670. if (!string.IsNullOrEmpty(value))
  671. {
  672. var parts = value.Split('/');
  673. float result;
  674. if (parts.Length == 2)
  675. {
  676. result = float.Parse(parts[0], _usCulture) / float.Parse(parts[1], _usCulture);
  677. }
  678. else
  679. {
  680. result = float.Parse(parts[0], _usCulture);
  681. }
  682. return float.IsNaN(result) ? (float?)null : result;
  683. }
  684. return null;
  685. }
  686. private void SetAudioRuntimeTicks(InternalMediaInfoResult result, MediaInfo data)
  687. {
  688. if (result.streams != null)
  689. {
  690. // Get the first info stream
  691. var stream = result.streams.FirstOrDefault(s => string.Equals(s.codec_type, "audio", StringComparison.OrdinalIgnoreCase));
  692. if (stream != null)
  693. {
  694. // Get duration from stream properties
  695. var duration = stream.duration;
  696. // If it's not there go into format properties
  697. if (string.IsNullOrEmpty(duration))
  698. {
  699. duration = result.format.duration;
  700. }
  701. // If we got something, parse it
  702. if (!string.IsNullOrEmpty(duration))
  703. {
  704. data.RunTimeTicks = TimeSpan.FromSeconds(double.Parse(duration, _usCulture)).Ticks;
  705. }
  706. }
  707. }
  708. }
  709. private void SetSize(InternalMediaInfoResult data, Model.MediaInfo.MediaInfo info)
  710. {
  711. if (data.format != null)
  712. {
  713. if (!string.IsNullOrEmpty(data.format.size))
  714. {
  715. info.Size = long.Parse(data.format.size, _usCulture);
  716. }
  717. else
  718. {
  719. info.Size = null;
  720. }
  721. }
  722. }
  723. private void SetAudioInfoFromTags(MediaInfo audio, Dictionary<string, string> tags)
  724. {
  725. var composer = FFProbeHelpers.GetDictionaryValue(tags, "composer");
  726. if (!string.IsNullOrWhiteSpace(composer))
  727. {
  728. foreach (var person in Split(composer, false))
  729. {
  730. audio.People.Add(new BaseItemPerson { Name = person, Type = PersonType.Composer });
  731. }
  732. }
  733. //var conductor = FFProbeHelpers.GetDictionaryValue(tags, "conductor");
  734. //if (!string.IsNullOrWhiteSpace(conductor))
  735. //{
  736. // foreach (var person in Split(conductor, false))
  737. // {
  738. // audio.People.Add(new BaseItemPerson { Name = person, Type = PersonType.Conductor });
  739. // }
  740. //}
  741. //var lyricist = FFProbeHelpers.GetDictionaryValue(tags, "lyricist");
  742. //if (!string.IsNullOrWhiteSpace(lyricist))
  743. //{
  744. // foreach (var person in Split(lyricist, false))
  745. // {
  746. // audio.People.Add(new BaseItemPerson { Name = person, Type = PersonType.Lyricist });
  747. // }
  748. //}
  749. // Check for writer some music is tagged that way as alternative to composer/lyricist
  750. var writer = FFProbeHelpers.GetDictionaryValue(tags, "writer");
  751. if (!string.IsNullOrWhiteSpace(writer))
  752. {
  753. foreach (var person in Split(writer, false))
  754. {
  755. audio.People.Add(new BaseItemPerson { Name = person, Type = PersonType.Writer });
  756. }
  757. }
  758. audio.Album = FFProbeHelpers.GetDictionaryValue(tags, "album");
  759. var artists = FFProbeHelpers.GetDictionaryValue(tags, "artists");
  760. if (!string.IsNullOrWhiteSpace(artists))
  761. {
  762. audio.Artists = SplitArtists(artists, new[] { '/', ';' }, false)
  763. .DistinctNames()
  764. .ToList();
  765. }
  766. else
  767. {
  768. var artist = FFProbeHelpers.GetDictionaryValue(tags, "artist");
  769. if (string.IsNullOrWhiteSpace(artist))
  770. {
  771. audio.Artists.Clear();
  772. }
  773. else
  774. {
  775. audio.Artists = SplitArtists(artist, _nameDelimiters, true)
  776. .DistinctNames()
  777. .ToList();
  778. }
  779. }
  780. var albumArtist = FFProbeHelpers.GetDictionaryValue(tags, "albumartist");
  781. if (string.IsNullOrWhiteSpace(albumArtist))
  782. {
  783. albumArtist = FFProbeHelpers.GetDictionaryValue(tags, "album artist");
  784. }
  785. if (string.IsNullOrWhiteSpace(albumArtist))
  786. {
  787. albumArtist = FFProbeHelpers.GetDictionaryValue(tags, "album_artist");
  788. }
  789. if (string.IsNullOrWhiteSpace(albumArtist))
  790. {
  791. audio.AlbumArtists = new List<string>();
  792. }
  793. else
  794. {
  795. audio.AlbumArtists = SplitArtists(albumArtist, _nameDelimiters, true)
  796. .DistinctNames()
  797. .ToList();
  798. }
  799. if (audio.AlbumArtists.Count == 0)
  800. {
  801. audio.AlbumArtists = audio.Artists.Take(1).ToList();
  802. }
  803. // Track number
  804. audio.IndexNumber = GetDictionaryDiscValue(tags, "track");
  805. // Disc number
  806. audio.ParentIndexNumber = GetDictionaryDiscValue(tags, "disc");
  807. // If we don't have a ProductionYear try and get it from PremiereDate
  808. if (audio.PremiereDate.HasValue && !audio.ProductionYear.HasValue)
  809. {
  810. audio.ProductionYear = audio.PremiereDate.Value.ToLocalTime().Year;
  811. }
  812. // There's several values in tags may or may not be present
  813. FetchStudios(audio, tags, "organization");
  814. FetchStudios(audio, tags, "ensemble");
  815. FetchStudios(audio, tags, "publisher");
  816. FetchStudios(audio, tags, "label");
  817. // These support mulitple values, but for now we only store the first.
  818. var mb = GetMultipleMusicBrainzId(FFProbeHelpers.GetDictionaryValue(tags, "MusicBrainz Album Artist Id"));
  819. if (mb == null) mb = GetMultipleMusicBrainzId(FFProbeHelpers.GetDictionaryValue(tags, "MUSICBRAINZ_ALBUMARTISTID"));
  820. audio.SetProviderId(MetadataProviders.MusicBrainzAlbumArtist, mb);
  821. mb = GetMultipleMusicBrainzId(FFProbeHelpers.GetDictionaryValue(tags, "MusicBrainz Artist Id"));
  822. if (mb == null) mb = GetMultipleMusicBrainzId(FFProbeHelpers.GetDictionaryValue(tags, "MUSICBRAINZ_ARTISTID"));
  823. audio.SetProviderId(MetadataProviders.MusicBrainzArtist, mb);
  824. mb = GetMultipleMusicBrainzId(FFProbeHelpers.GetDictionaryValue(tags, "MusicBrainz Album Id"));
  825. if (mb == null) mb = GetMultipleMusicBrainzId(FFProbeHelpers.GetDictionaryValue(tags, "MUSICBRAINZ_ALBUMID"));
  826. audio.SetProviderId(MetadataProviders.MusicBrainzAlbum, mb);
  827. mb = GetMultipleMusicBrainzId(FFProbeHelpers.GetDictionaryValue(tags, "MusicBrainz Release Group Id"));
  828. if (mb == null) mb = GetMultipleMusicBrainzId(FFProbeHelpers.GetDictionaryValue(tags, "MUSICBRAINZ_RELEASEGROUPID"));
  829. audio.SetProviderId(MetadataProviders.MusicBrainzReleaseGroup, mb);
  830. mb = GetMultipleMusicBrainzId(FFProbeHelpers.GetDictionaryValue(tags, "MusicBrainz Release Track Id"));
  831. if (mb == null) mb = GetMultipleMusicBrainzId(FFProbeHelpers.GetDictionaryValue(tags, "MUSICBRAINZ_RELEASETRACKID"));
  832. audio.SetProviderId(MetadataProviders.MusicBrainzTrack, mb);
  833. }
  834. private string GetMultipleMusicBrainzId(string value)
  835. {
  836. if (string.IsNullOrWhiteSpace(value))
  837. {
  838. return null;
  839. }
  840. return value.Split(new[] { '/' }, StringSplitOptions.RemoveEmptyEntries)
  841. .Select(i => i.Trim())
  842. .FirstOrDefault(i => !string.IsNullOrWhiteSpace(i));
  843. }
  844. private readonly char[] _nameDelimiters = { '/', '|', ';', '\\' };
  845. /// <summary>
  846. /// Splits the specified val.
  847. /// </summary>
  848. /// <param name="val">The val.</param>
  849. /// <param name="allowCommaDelimiter">if set to <c>true</c> [allow comma delimiter].</param>
  850. /// <returns>System.String[][].</returns>
  851. private IEnumerable<string> Split(string val, bool allowCommaDelimiter)
  852. {
  853. // Only use the comma as a delimeter if there are no slashes or pipes.
  854. // We want to be careful not to split names that have commas in them
  855. var delimeter = !allowCommaDelimiter || _nameDelimiters.Any(i => val.IndexOf(i) != -1) ?
  856. _nameDelimiters :
  857. new[] { ',' };
  858. return val.Split(delimeter, StringSplitOptions.RemoveEmptyEntries)
  859. .Where(i => !string.IsNullOrWhiteSpace(i))
  860. .Select(i => i.Trim());
  861. }
  862. private const string ArtistReplaceValue = " | ";
  863. private IEnumerable<string> SplitArtists(string val, char[] delimiters, bool splitFeaturing)
  864. {
  865. if (splitFeaturing)
  866. {
  867. val = val.Replace(" featuring ", ArtistReplaceValue, StringComparison.OrdinalIgnoreCase)
  868. .Replace(" feat. ", ArtistReplaceValue, StringComparison.OrdinalIgnoreCase);
  869. }
  870. var artistsFound = new List<string>();
  871. foreach (var whitelistArtist in GetSplitWhitelist())
  872. {
  873. var originalVal = val;
  874. val = val.Replace(whitelistArtist, "|", StringComparison.OrdinalIgnoreCase);
  875. if (!string.Equals(originalVal, val, StringComparison.OrdinalIgnoreCase))
  876. {
  877. artistsFound.Add(whitelistArtist);
  878. }
  879. }
  880. var artists = val.Split(delimiters, StringSplitOptions.RemoveEmptyEntries)
  881. .Where(i => !string.IsNullOrWhiteSpace(i))
  882. .Select(i => i.Trim());
  883. artistsFound.AddRange(artists);
  884. return artistsFound;
  885. }
  886. private List<string> _splitWhiteList = null;
  887. private IEnumerable<string> GetSplitWhitelist()
  888. {
  889. if (_splitWhiteList == null)
  890. {
  891. _splitWhiteList = new List<string>
  892. {
  893. "AC/DC"
  894. };
  895. }
  896. return _splitWhiteList;
  897. }
  898. /// <summary>
  899. /// Gets the studios from the tags collection
  900. /// </summary>
  901. /// <param name="info">The info.</param>
  902. /// <param name="tags">The tags.</param>
  903. /// <param name="tagName">Name of the tag.</param>
  904. private void FetchStudios(MediaInfo info, Dictionary<string, string> tags, string tagName)
  905. {
  906. var val = FFProbeHelpers.GetDictionaryValue(tags, tagName);
  907. if (!string.IsNullOrEmpty(val))
  908. {
  909. var studios = Split(val, true);
  910. foreach (var studio in studios)
  911. {
  912. // Sometimes the artist name is listed here, account for that
  913. if (info.Artists.Contains(studio, StringComparer.OrdinalIgnoreCase))
  914. {
  915. continue;
  916. }
  917. if (info.AlbumArtists.Contains(studio, StringComparer.OrdinalIgnoreCase))
  918. {
  919. continue;
  920. }
  921. info.Studios.Add(studio);
  922. }
  923. info.Studios = info.Studios
  924. .Where(i => !string.IsNullOrWhiteSpace(i))
  925. .Distinct(StringComparer.OrdinalIgnoreCase)
  926. .ToList();
  927. }
  928. }
  929. /// <summary>
  930. /// Gets the genres from the tags collection
  931. /// </summary>
  932. /// <param name="info">The information.</param>
  933. /// <param name="tags">The tags.</param>
  934. private void FetchGenres(MediaInfo info, Dictionary<string, string> tags)
  935. {
  936. var val = FFProbeHelpers.GetDictionaryValue(tags, "genre");
  937. if (!string.IsNullOrEmpty(val))
  938. {
  939. foreach (var genre in Split(val, true))
  940. {
  941. info.Genres.Add(genre);
  942. }
  943. info.Genres = info.Genres
  944. .Where(i => !string.IsNullOrWhiteSpace(i))
  945. .Distinct(StringComparer.OrdinalIgnoreCase)
  946. .ToList();
  947. }
  948. }
  949. /// <summary>
  950. /// Gets the disc number, which is sometimes can be in the form of '1', or '1/3'
  951. /// </summary>
  952. /// <param name="tags">The tags.</param>
  953. /// <param name="tagName">Name of the tag.</param>
  954. /// <returns>System.Nullable{System.Int32}.</returns>
  955. private int? GetDictionaryDiscValue(Dictionary<string, string> tags, string tagName)
  956. {
  957. var disc = FFProbeHelpers.GetDictionaryValue(tags, tagName);
  958. if (!string.IsNullOrEmpty(disc))
  959. {
  960. disc = disc.Split('/')[0];
  961. int num;
  962. if (int.TryParse(disc, out num))
  963. {
  964. return num;
  965. }
  966. }
  967. return null;
  968. }
  969. private ChapterInfo GetChapterInfo(MediaChapter chapter)
  970. {
  971. var info = new ChapterInfo();
  972. if (chapter.tags != null)
  973. {
  974. string name;
  975. if (chapter.tags.TryGetValue("title", out name))
  976. {
  977. info.Name = name;
  978. }
  979. }
  980. // Limit accuracy to milliseconds to match xml saving
  981. var secondsString = chapter.start_time;
  982. double seconds;
  983. if (double.TryParse(secondsString, NumberStyles.Any, CultureInfo.InvariantCulture, out seconds))
  984. {
  985. var ms = Math.Round(TimeSpan.FromSeconds(seconds).TotalMilliseconds);
  986. info.StartPositionTicks = TimeSpan.FromMilliseconds(ms).Ticks;
  987. }
  988. return info;
  989. }
  990. private const int MaxSubtitleDescriptionExtractionLength = 100; // When extracting subtitles, the maximum length to consider (to avoid invalid filenames)
  991. private void FetchWtvInfo(MediaInfo video, InternalMediaInfoResult data)
  992. {
  993. if (data.format == null || data.format.tags == null)
  994. {
  995. return;
  996. }
  997. var genres = FFProbeHelpers.GetDictionaryValue(data.format.tags, "WM/Genre");
  998. if (!string.IsNullOrWhiteSpace(genres))
  999. {
  1000. var genreList = genres.Split(new[] { ';', '/', ',' }, StringSplitOptions.RemoveEmptyEntries)
  1001. .Where(i => !string.IsNullOrWhiteSpace(i))
  1002. .Select(i => i.Trim())
  1003. .ToList();
  1004. // If this is empty then don't overwrite genres that might have been fetched earlier
  1005. if (genreList.Count > 0)
  1006. {
  1007. video.Genres = genreList;
  1008. }
  1009. }
  1010. var officialRating = FFProbeHelpers.GetDictionaryValue(data.format.tags, "WM/ParentalRating");
  1011. if (!string.IsNullOrWhiteSpace(officialRating))
  1012. {
  1013. video.OfficialRating = officialRating;
  1014. }
  1015. var people = FFProbeHelpers.GetDictionaryValue(data.format.tags, "WM/MediaCredits");
  1016. if (!string.IsNullOrEmpty(people))
  1017. {
  1018. video.People = people.Split(new[] { ';', '/' }, StringSplitOptions.RemoveEmptyEntries)
  1019. .Where(i => !string.IsNullOrWhiteSpace(i))
  1020. .Select(i => new BaseItemPerson { Name = i.Trim(), Type = PersonType.Actor })
  1021. .ToList();
  1022. }
  1023. var year = FFProbeHelpers.GetDictionaryValue(data.format.tags, "WM/OriginalReleaseTime");
  1024. if (!string.IsNullOrWhiteSpace(year))
  1025. {
  1026. int val;
  1027. if (int.TryParse(year, NumberStyles.Integer, _usCulture, out val))
  1028. {
  1029. video.ProductionYear = val;
  1030. }
  1031. }
  1032. var premiereDateString = FFProbeHelpers.GetDictionaryValue(data.format.tags, "WM/MediaOriginalBroadcastDateTime");
  1033. if (!string.IsNullOrWhiteSpace(premiereDateString))
  1034. {
  1035. DateTime val;
  1036. // Credit to MCEBuddy: https://mcebuddy2x.codeplex.com/
  1037. // DateTime is reported along with timezone info (typically Z i.e. UTC hence assume None)
  1038. if (DateTime.TryParse(year, null, DateTimeStyles.None, out val))
  1039. {
  1040. video.PremiereDate = val.ToUniversalTime();
  1041. }
  1042. }
  1043. var description = FFProbeHelpers.GetDictionaryValue(data.format.tags, "WM/SubTitleDescription");
  1044. var subTitle = FFProbeHelpers.GetDictionaryValue(data.format.tags, "WM/SubTitle");
  1045. // For below code, credit to MCEBuddy: https://mcebuddy2x.codeplex.com/
  1046. // Sometimes for TV Shows the Subtitle field is empty and the subtitle description contains the subtitle, extract if possible. See ticket https://mcebuddy2x.codeplex.com/workitem/1910
  1047. // The format is -> EPISODE/TOTAL_EPISODES_IN_SEASON. SUBTITLE: DESCRIPTION
  1048. // OR -> COMMENT. SUBTITLE: DESCRIPTION
  1049. // e.g. -> 4/13. The Doctor's Wife: Science fiction drama. When he follows a Time Lord distress signal, the Doctor puts Amy, Rory and his beloved TARDIS in grave danger. Also in HD. [AD,S]
  1050. // e.g. -> CBeebies Bedtime Hour. The Mystery: Animated adventures of two friends who live on an island in the middle of the big city. Some of Abney and Teal's favourite objects are missing. [S]
  1051. if (String.IsNullOrWhiteSpace(subTitle) && !String.IsNullOrWhiteSpace(description) && description.Substring(0, Math.Min(description.Length, MaxSubtitleDescriptionExtractionLength)).Contains(":")) // Check within the Subtitle size limit, otherwise from description it can get too long creating an invalid filename
  1052. {
  1053. string[] parts = description.Split(':');
  1054. if (parts.Length > 0)
  1055. {
  1056. string subtitle = parts[0];
  1057. try
  1058. {
  1059. if (subtitle.Contains("/")) // It contains a episode number and season number
  1060. {
  1061. string[] numbers = subtitle.Split(' ');
  1062. video.IndexNumber = int.Parse(numbers[0].Replace(".", "").Split('/')[0]);
  1063. int totalEpisodesInSeason = int.Parse(numbers[0].Replace(".", "").Split('/')[1]);
  1064. description = String.Join(" ", numbers, 1, numbers.Length - 1).Trim(); // Skip the first, concatenate the rest, clean up spaces and save it
  1065. }
  1066. else
  1067. throw new Exception(); // Switch to default parsing
  1068. }
  1069. catch // Default parsing
  1070. {
  1071. if (subtitle.Contains(".")) // skip the comment, keep the subtitle
  1072. description = String.Join(".", subtitle.Split('.'), 1, subtitle.Split('.').Length - 1).Trim(); // skip the first
  1073. else
  1074. description = subtitle.Trim(); // Clean up whitespaces and save it
  1075. }
  1076. }
  1077. }
  1078. if (!string.IsNullOrWhiteSpace(description))
  1079. {
  1080. video.Overview = description;
  1081. }
  1082. }
  1083. private void ExtractTimestamp(MediaInfo video)
  1084. {
  1085. if (video.VideoType == VideoType.VideoFile)
  1086. {
  1087. if (string.Equals(video.Container, "mpeg2ts", StringComparison.OrdinalIgnoreCase) ||
  1088. string.Equals(video.Container, "m2ts", StringComparison.OrdinalIgnoreCase) ||
  1089. string.Equals(video.Container, "ts", StringComparison.OrdinalIgnoreCase))
  1090. {
  1091. try
  1092. {
  1093. video.Timestamp = GetMpegTimestamp(video.Path);
  1094. _logger.Debug("Video has {0} timestamp", video.Timestamp);
  1095. }
  1096. catch (Exception ex)
  1097. {
  1098. _logger.ErrorException("Error extracting timestamp info from {0}", ex, video.Path);
  1099. video.Timestamp = null;
  1100. }
  1101. }
  1102. }
  1103. }
  1104. private TransportStreamTimestamp GetMpegTimestamp(string path)
  1105. {
  1106. var packetBuffer = new byte['Å'];
  1107. using (var fs = _fileSystem.GetFileStream(path, FileOpenMode.Open, FileAccessMode.Read, FileShareMode.Read))
  1108. {
  1109. fs.Read(packetBuffer, 0, packetBuffer.Length);
  1110. }
  1111. if (packetBuffer[0] == 71)
  1112. {
  1113. return TransportStreamTimestamp.None;
  1114. }
  1115. if ((packetBuffer[4] == 71) && (packetBuffer['Ä'] == 71))
  1116. {
  1117. if ((packetBuffer[0] == 0) && (packetBuffer[1] == 0) && (packetBuffer[2] == 0) && (packetBuffer[3] == 0))
  1118. {
  1119. return TransportStreamTimestamp.Zero;
  1120. }
  1121. return TransportStreamTimestamp.Valid;
  1122. }
  1123. return TransportStreamTimestamp.None;
  1124. }
  1125. }
  1126. }