461 lines
16 KiB
C#
461 lines
16 KiB
C#
//============================================================================
|
|
// BDInfo - Blu-ray Video and Audio Analysis Tool
|
|
// Copyright © 2010 Cinema Squid
|
|
//
|
|
// This library is free software; you can redistribute it and/or
|
|
// modify it under the terms of the GNU Lesser General Public
|
|
// License as published by the Free Software Foundation; either
|
|
// version 2.1 of the License, or (at your option) any later version.
|
|
//
|
|
// This library is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
// Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public
|
|
// License along with this library; if not, write to the Free Software
|
|
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
//=============================================================================
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using MediaBrowser.Model.IO;
|
|
using MediaBrowser.Model.Text;
|
|
|
|
namespace BDInfo
|
|
{
|
|
public class BDROM
|
|
{
|
|
public FileSystemMetadata DirectoryRoot = null;
|
|
public FileSystemMetadata DirectoryBDMV = null;
|
|
public FileSystemMetadata DirectoryBDJO = null;
|
|
public FileSystemMetadata DirectoryCLIPINF = null;
|
|
public FileSystemMetadata DirectoryPLAYLIST = null;
|
|
public FileSystemMetadata DirectorySNP = null;
|
|
public FileSystemMetadata DirectorySSIF = null;
|
|
public FileSystemMetadata DirectorySTREAM = null;
|
|
|
|
public string VolumeLabel = null;
|
|
public ulong Size = 0;
|
|
public bool IsBDPlus = false;
|
|
public bool IsBDJava = false;
|
|
public bool IsDBOX = false;
|
|
public bool IsPSP = false;
|
|
public bool Is3D = false;
|
|
public bool Is50Hz = false;
|
|
|
|
private readonly IFileSystem _fileSystem;
|
|
|
|
public Dictionary<string, TSPlaylistFile> PlaylistFiles =
|
|
new Dictionary<string, TSPlaylistFile>();
|
|
public Dictionary<string, TSStreamClipFile> StreamClipFiles =
|
|
new Dictionary<string, TSStreamClipFile>();
|
|
public Dictionary<string, TSStreamFile> StreamFiles =
|
|
new Dictionary<string, TSStreamFile>();
|
|
public Dictionary<string, TSInterleavedFile> InterleavedFiles =
|
|
new Dictionary<string, TSInterleavedFile>();
|
|
|
|
private static List<string> ExcludeDirs = new List<string> { "ANY!", "AACS", "BDSVM", "ANYVM", "SLYVM" };
|
|
|
|
public delegate bool OnStreamClipFileScanError(
|
|
TSStreamClipFile streamClipFile, Exception ex);
|
|
|
|
public event OnStreamClipFileScanError StreamClipFileScanError;
|
|
|
|
public delegate bool OnStreamFileScanError(
|
|
TSStreamFile streamClipFile, Exception ex);
|
|
|
|
public event OnStreamFileScanError StreamFileScanError;
|
|
|
|
public delegate bool OnPlaylistFileScanError(
|
|
TSPlaylistFile playlistFile, Exception ex);
|
|
|
|
public event OnPlaylistFileScanError PlaylistFileScanError;
|
|
|
|
public BDROM(
|
|
string path, IFileSystem fileSystem, ITextEncoding textEncoding)
|
|
{
|
|
if (string.IsNullOrWhiteSpace(path))
|
|
{
|
|
throw new ArgumentNullException("path");
|
|
}
|
|
|
|
_fileSystem = fileSystem;
|
|
//
|
|
// Locate BDMV directories.
|
|
//
|
|
|
|
DirectoryBDMV =
|
|
GetDirectoryBDMV(path);
|
|
|
|
if (DirectoryBDMV == null)
|
|
{
|
|
throw new Exception("Unable to locate BD structure.");
|
|
}
|
|
|
|
DirectoryRoot =
|
|
_fileSystem.GetDirectoryInfo(Path.GetDirectoryName(DirectoryBDMV.FullName));
|
|
DirectoryBDJO =
|
|
GetDirectory("BDJO", DirectoryBDMV, 0);
|
|
DirectoryCLIPINF =
|
|
GetDirectory("CLIPINF", DirectoryBDMV, 0);
|
|
DirectoryPLAYLIST =
|
|
GetDirectory("PLAYLIST", DirectoryBDMV, 0);
|
|
DirectorySNP =
|
|
GetDirectory("SNP", DirectoryRoot, 0);
|
|
DirectorySTREAM =
|
|
GetDirectory("STREAM", DirectoryBDMV, 0);
|
|
DirectorySSIF =
|
|
GetDirectory("SSIF", DirectorySTREAM, 0);
|
|
|
|
if (DirectoryCLIPINF == null
|
|
|| DirectoryPLAYLIST == null)
|
|
{
|
|
throw new Exception("Unable to locate BD structure.");
|
|
}
|
|
|
|
//
|
|
// Initialize basic disc properties.
|
|
//
|
|
|
|
VolumeLabel = GetVolumeLabel(DirectoryRoot);
|
|
Size = (ulong)GetDirectorySize(DirectoryRoot);
|
|
|
|
if (null != GetDirectory("BDSVM", DirectoryRoot, 0))
|
|
{
|
|
IsBDPlus = true;
|
|
}
|
|
if (null != GetDirectory("SLYVM", DirectoryRoot, 0))
|
|
{
|
|
IsBDPlus = true;
|
|
}
|
|
if (null != GetDirectory("ANYVM", DirectoryRoot, 0))
|
|
{
|
|
IsBDPlus = true;
|
|
}
|
|
|
|
if (DirectoryBDJO != null &&
|
|
_fileSystem.GetFilePaths(DirectoryBDJO.FullName).Any())
|
|
{
|
|
IsBDJava = true;
|
|
}
|
|
|
|
if (DirectorySNP != null &&
|
|
GetFilePaths(DirectorySNP.FullName, ".mnv").Any())
|
|
{
|
|
IsPSP = true;
|
|
}
|
|
|
|
if (DirectorySSIF != null &&
|
|
_fileSystem.GetFilePaths(DirectorySSIF.FullName).Any())
|
|
{
|
|
Is3D = true;
|
|
}
|
|
|
|
if (_fileSystem.FileExists(Path.Combine(DirectoryRoot.FullName, "FilmIndex.xml")))
|
|
{
|
|
IsDBOX = true;
|
|
}
|
|
|
|
//
|
|
// Initialize file lists.
|
|
//
|
|
|
|
if (DirectoryPLAYLIST != null)
|
|
{
|
|
FileSystemMetadata[] files = GetFiles(DirectoryPLAYLIST.FullName, ".mpls").ToArray();
|
|
foreach (FileSystemMetadata file in files)
|
|
{
|
|
PlaylistFiles.Add(
|
|
file.Name.ToUpper(), new TSPlaylistFile(this, file, _fileSystem, textEncoding));
|
|
}
|
|
}
|
|
|
|
if (DirectorySTREAM != null)
|
|
{
|
|
FileSystemMetadata[] files = GetFiles(DirectorySTREAM.FullName, ".m2ts").ToArray();
|
|
foreach (FileSystemMetadata file in files)
|
|
{
|
|
StreamFiles.Add(
|
|
file.Name.ToUpper(), new TSStreamFile(file, _fileSystem));
|
|
}
|
|
}
|
|
|
|
if (DirectoryCLIPINF != null)
|
|
{
|
|
FileSystemMetadata[] files = GetFiles(DirectoryCLIPINF.FullName, ".clpi").ToArray();
|
|
foreach (FileSystemMetadata file in files)
|
|
{
|
|
StreamClipFiles.Add(
|
|
file.Name.ToUpper(), new TSStreamClipFile(file, _fileSystem, textEncoding));
|
|
}
|
|
}
|
|
|
|
if (DirectorySSIF != null)
|
|
{
|
|
FileSystemMetadata[] files = GetFiles(DirectorySSIF.FullName, ".ssif").ToArray();
|
|
foreach (FileSystemMetadata file in files)
|
|
{
|
|
InterleavedFiles.Add(
|
|
file.Name.ToUpper(), new TSInterleavedFile(file));
|
|
}
|
|
}
|
|
}
|
|
|
|
private IEnumerable<FileSystemMetadata> GetFiles(string path, string extension)
|
|
{
|
|
return _fileSystem.GetFiles(path, new[] { extension }, false, false);
|
|
}
|
|
|
|
private IEnumerable<string> GetFilePaths(string path, string extension)
|
|
{
|
|
return _fileSystem.GetFilePaths(path, new[] { extension }, false, false);
|
|
}
|
|
|
|
public void Scan()
|
|
{
|
|
List<TSStreamClipFile> errorStreamClipFiles = new List<TSStreamClipFile>();
|
|
foreach (TSStreamClipFile streamClipFile in StreamClipFiles.Values)
|
|
{
|
|
try
|
|
{
|
|
streamClipFile.Scan();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
errorStreamClipFiles.Add(streamClipFile);
|
|
if (StreamClipFileScanError != null)
|
|
{
|
|
if (StreamClipFileScanError(streamClipFile, ex))
|
|
{
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
else throw ex;
|
|
}
|
|
}
|
|
|
|
foreach (TSStreamFile streamFile in StreamFiles.Values)
|
|
{
|
|
string ssifName = Path.GetFileNameWithoutExtension(streamFile.Name) + ".SSIF";
|
|
if (InterleavedFiles.ContainsKey(ssifName))
|
|
{
|
|
streamFile.InterleavedFile = InterleavedFiles[ssifName];
|
|
}
|
|
}
|
|
|
|
TSStreamFile[] streamFiles = new TSStreamFile[StreamFiles.Count];
|
|
StreamFiles.Values.CopyTo(streamFiles, 0);
|
|
Array.Sort(streamFiles, CompareStreamFiles);
|
|
|
|
List<TSPlaylistFile> errorPlaylistFiles = new List<TSPlaylistFile>();
|
|
foreach (TSPlaylistFile playlistFile in PlaylistFiles.Values)
|
|
{
|
|
try
|
|
{
|
|
playlistFile.Scan(StreamFiles, StreamClipFiles);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
errorPlaylistFiles.Add(playlistFile);
|
|
if (PlaylistFileScanError != null)
|
|
{
|
|
if (PlaylistFileScanError(playlistFile, ex))
|
|
{
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
else throw ex;
|
|
}
|
|
}
|
|
|
|
List<TSStreamFile> errorStreamFiles = new List<TSStreamFile>();
|
|
foreach (TSStreamFile streamFile in streamFiles)
|
|
{
|
|
try
|
|
{
|
|
List<TSPlaylistFile> playlists = new List<TSPlaylistFile>();
|
|
foreach (TSPlaylistFile playlist in PlaylistFiles.Values)
|
|
{
|
|
foreach (TSStreamClip streamClip in playlist.StreamClips)
|
|
{
|
|
if (streamClip.Name == streamFile.Name)
|
|
{
|
|
playlists.Add(playlist);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
streamFile.Scan(playlists, false);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
errorStreamFiles.Add(streamFile);
|
|
if (StreamFileScanError != null)
|
|
{
|
|
if (StreamFileScanError(streamFile, ex))
|
|
{
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
else throw ex;
|
|
}
|
|
}
|
|
|
|
foreach (TSPlaylistFile playlistFile in PlaylistFiles.Values)
|
|
{
|
|
playlistFile.Initialize();
|
|
if (!Is50Hz)
|
|
{
|
|
foreach (TSVideoStream videoStream in playlistFile.VideoStreams)
|
|
{
|
|
if (videoStream.FrameRate == TSFrameRate.FRAMERATE_25 ||
|
|
videoStream.FrameRate == TSFrameRate.FRAMERATE_50)
|
|
{
|
|
Is50Hz = true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private FileSystemMetadata GetDirectoryBDMV(
|
|
string path)
|
|
{
|
|
if (string.IsNullOrWhiteSpace(path))
|
|
{
|
|
throw new ArgumentNullException("path");
|
|
}
|
|
|
|
FileSystemMetadata dir = _fileSystem.GetDirectoryInfo(path);
|
|
|
|
while (dir != null)
|
|
{
|
|
if (string.Equals(dir.Name, "BDMV", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return dir;
|
|
}
|
|
var parentFolder = Path.GetDirectoryName(dir.FullName);
|
|
if (string.IsNullOrEmpty(parentFolder))
|
|
{
|
|
dir = null;
|
|
}
|
|
else
|
|
{
|
|
dir = _fileSystem.GetDirectoryInfo(parentFolder);
|
|
}
|
|
}
|
|
|
|
return GetDirectory("BDMV", _fileSystem.GetDirectoryInfo(path), 0);
|
|
}
|
|
|
|
private FileSystemMetadata GetDirectory(
|
|
string name,
|
|
FileSystemMetadata dir,
|
|
int searchDepth)
|
|
{
|
|
if (dir != null)
|
|
{
|
|
FileSystemMetadata[] children = _fileSystem.GetDirectories(dir.FullName).ToArray();
|
|
foreach (FileSystemMetadata child in children)
|
|
{
|
|
if (string.Equals(child.Name, name, StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return child;
|
|
}
|
|
}
|
|
if (searchDepth > 0)
|
|
{
|
|
foreach (FileSystemMetadata child in children)
|
|
{
|
|
GetDirectory(
|
|
name, child, searchDepth - 1);
|
|
}
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
private long GetDirectorySize(FileSystemMetadata directoryInfo)
|
|
{
|
|
long size = 0;
|
|
|
|
//if (!ExcludeDirs.Contains(directoryInfo.Name.ToUpper())) // TODO: Keep?
|
|
{
|
|
FileSystemMetadata[] pathFiles = _fileSystem.GetFiles(directoryInfo.FullName).ToArray();
|
|
foreach (FileSystemMetadata pathFile in pathFiles)
|
|
{
|
|
if (pathFile.Extension.ToUpper() == ".SSIF")
|
|
{
|
|
continue;
|
|
}
|
|
size += pathFile.Length;
|
|
}
|
|
|
|
FileSystemMetadata[] pathChildren = _fileSystem.GetDirectories(directoryInfo.FullName).ToArray();
|
|
foreach (FileSystemMetadata pathChild in pathChildren)
|
|
{
|
|
size += GetDirectorySize(pathChild);
|
|
}
|
|
}
|
|
|
|
return size;
|
|
}
|
|
|
|
private string GetVolumeLabel(FileSystemMetadata dir)
|
|
{
|
|
return dir.Name;
|
|
}
|
|
|
|
public static int CompareStreamFiles(
|
|
TSStreamFile x,
|
|
TSStreamFile y)
|
|
{
|
|
// TODO: Use interleaved file sizes
|
|
|
|
if ((x == null || x.FileInfo == null) && (y == null || y.FileInfo == null))
|
|
{
|
|
return 0;
|
|
}
|
|
else if ((x == null || x.FileInfo == null) && (y != null && y.FileInfo != null))
|
|
{
|
|
return 1;
|
|
}
|
|
else if ((x != null || x.FileInfo != null) && (y == null || y.FileInfo == null))
|
|
{
|
|
return -1;
|
|
}
|
|
else
|
|
{
|
|
if (x.FileInfo.Length > y.FileInfo.Length)
|
|
{
|
|
return 1;
|
|
}
|
|
else if (y.FileInfo.Length > x.FileInfo.Length)
|
|
{
|
|
return -1;
|
|
}
|
|
else
|
|
{
|
|
return 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|