mirror of
https://github.com/jellyfin/jellyfin.git
synced 2025-11-03 19:17:24 -05:00
483 lines
16 KiB
C#
483 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.Runtime.InteropServices;
|
|
using System.Text;
|
|
|
|
namespace BDInfo
|
|
{
|
|
public class BDROM
|
|
{
|
|
public DirectoryInfo DirectoryRoot = null;
|
|
public DirectoryInfo DirectoryBDMV = null;
|
|
public DirectoryInfo DirectoryBDJO = null;
|
|
public DirectoryInfo DirectoryCLIPINF = null;
|
|
public DirectoryInfo DirectoryPLAYLIST = null;
|
|
public DirectoryInfo DirectorySNP = null;
|
|
public DirectoryInfo DirectorySSIF = null;
|
|
public DirectoryInfo 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;
|
|
|
|
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)
|
|
{
|
|
//
|
|
// Locate BDMV directories.
|
|
//
|
|
|
|
DirectoryBDMV =
|
|
GetDirectoryBDMV(path);
|
|
|
|
if (DirectoryBDMV == null)
|
|
{
|
|
throw new Exception("Unable to locate BD structure.");
|
|
}
|
|
|
|
DirectoryRoot =
|
|
DirectoryBDMV.Parent;
|
|
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 &&
|
|
DirectoryBDJO.GetFiles().Length > 0)
|
|
{
|
|
IsBDJava = true;
|
|
}
|
|
|
|
if (DirectorySNP != null &&
|
|
(DirectorySNP.GetFiles("*.mnv").Length > 0 || DirectorySNP.GetFiles("*.MNV").Length > 0))
|
|
{
|
|
IsPSP = true;
|
|
}
|
|
|
|
if (DirectorySSIF != null &&
|
|
DirectorySSIF.GetFiles().Length > 0)
|
|
{
|
|
Is3D = true;
|
|
}
|
|
|
|
if (File.Exists(Path.Combine(DirectoryRoot.FullName, "FilmIndex.xml")))
|
|
{
|
|
IsDBOX = true;
|
|
}
|
|
|
|
//
|
|
// Initialize file lists.
|
|
//
|
|
|
|
if (DirectoryPLAYLIST != null)
|
|
{
|
|
FileInfo[] files = DirectoryPLAYLIST.GetFiles("*.mpls");
|
|
if (files.Length == 0)
|
|
{
|
|
files = DirectoryPLAYLIST.GetFiles("*.MPLS");
|
|
}
|
|
foreach (FileInfo file in files)
|
|
{
|
|
PlaylistFiles.Add(
|
|
file.Name.ToUpper(), new TSPlaylistFile(this, file));
|
|
}
|
|
}
|
|
|
|
if (DirectorySTREAM != null)
|
|
{
|
|
FileInfo[] files = DirectorySTREAM.GetFiles("*.m2ts");
|
|
if (files.Length == 0)
|
|
{
|
|
files = DirectoryPLAYLIST.GetFiles("*.M2TS");
|
|
}
|
|
foreach (FileInfo file in files)
|
|
{
|
|
StreamFiles.Add(
|
|
file.Name.ToUpper(), new TSStreamFile(file));
|
|
}
|
|
}
|
|
|
|
if (DirectoryCLIPINF != null)
|
|
{
|
|
FileInfo[] files = DirectoryCLIPINF.GetFiles("*.clpi");
|
|
if (files.Length == 0)
|
|
{
|
|
files = DirectoryPLAYLIST.GetFiles("*.CLPI");
|
|
}
|
|
foreach (FileInfo file in files)
|
|
{
|
|
StreamClipFiles.Add(
|
|
file.Name.ToUpper(), new TSStreamClipFile(file));
|
|
}
|
|
}
|
|
|
|
if (DirectorySSIF != null)
|
|
{
|
|
FileInfo[] files = DirectorySSIF.GetFiles("*.ssif");
|
|
if (files.Length == 0)
|
|
{
|
|
files = DirectorySSIF.GetFiles("*.SSIF");
|
|
}
|
|
foreach (FileInfo file in files)
|
|
{
|
|
InterleavedFiles.Add(
|
|
file.Name.ToUpper(), new TSInterleavedFile(file));
|
|
}
|
|
}
|
|
}
|
|
|
|
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 DirectoryInfo GetDirectoryBDMV(
|
|
string path)
|
|
{
|
|
DirectoryInfo dir = new DirectoryInfo(path);
|
|
|
|
while (dir != null)
|
|
{
|
|
if (dir.Name == "BDMV")
|
|
{
|
|
return dir;
|
|
}
|
|
dir = dir.Parent;
|
|
}
|
|
|
|
return GetDirectory("BDMV", new DirectoryInfo(path), 0);
|
|
}
|
|
|
|
private DirectoryInfo GetDirectory(
|
|
string name,
|
|
DirectoryInfo dir,
|
|
int searchDepth)
|
|
{
|
|
if (dir != null)
|
|
{
|
|
DirectoryInfo[] children = dir.GetDirectories();
|
|
foreach (DirectoryInfo child in children)
|
|
{
|
|
if (child.Name == name)
|
|
{
|
|
return child;
|
|
}
|
|
}
|
|
if (searchDepth > 0)
|
|
{
|
|
foreach (DirectoryInfo child in children)
|
|
{
|
|
GetDirectory(
|
|
name, child, searchDepth - 1);
|
|
}
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
private long GetDirectorySize(DirectoryInfo directoryInfo)
|
|
{
|
|
long size = 0;
|
|
|
|
//if (!ExcludeDirs.Contains(directoryInfo.Name.ToUpper())) // TODO: Keep?
|
|
{
|
|
FileInfo[] pathFiles = directoryInfo.GetFiles();
|
|
foreach (FileInfo pathFile in pathFiles)
|
|
{
|
|
if (pathFile.Extension.ToUpper() == ".SSIF")
|
|
{
|
|
continue;
|
|
}
|
|
size += pathFile.Length;
|
|
}
|
|
|
|
DirectoryInfo[] pathChildren = directoryInfo.GetDirectories();
|
|
foreach (DirectoryInfo pathChild in pathChildren)
|
|
{
|
|
size += GetDirectorySize(pathChild);
|
|
}
|
|
}
|
|
|
|
return size;
|
|
}
|
|
|
|
private string GetVolumeLabel(DirectoryInfo dir)
|
|
{
|
|
uint serialNumber = 0;
|
|
uint maxLength = 0;
|
|
uint volumeFlags = new uint();
|
|
StringBuilder volumeLabel = new StringBuilder(256);
|
|
StringBuilder fileSystemName = new StringBuilder(256);
|
|
string label = "";
|
|
|
|
try
|
|
{
|
|
long result = GetVolumeInformation(
|
|
dir.Name,
|
|
volumeLabel,
|
|
(uint)volumeLabel.Capacity,
|
|
ref serialNumber,
|
|
ref maxLength,
|
|
ref volumeFlags,
|
|
fileSystemName,
|
|
(uint)fileSystemName.Capacity);
|
|
|
|
label = volumeLabel.ToString();
|
|
}
|
|
catch { }
|
|
|
|
if (label.Length == 0)
|
|
{
|
|
label = dir.Name;
|
|
}
|
|
|
|
return label;
|
|
}
|
|
|
|
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;
|
|
}
|
|
}
|
|
}
|
|
|
|
[DllImport("kernel32.dll")]
|
|
private static extern long GetVolumeInformation(
|
|
string PathName,
|
|
StringBuilder VolumeNameBuffer,
|
|
uint VolumeNameSize,
|
|
ref uint VolumeSerialNumber,
|
|
ref uint MaximumComponentLength,
|
|
ref uint FileSystemFlags,
|
|
StringBuilder FileSystemNameBuffer,
|
|
uint FileSystemNameSize);
|
|
}
|
|
}
|