mirror of
https://github.com/jellyfin/jellyfin.git
synced 2026-03-14 22:26:34 +00:00
Refactor and add scheduled task
This commit is contained in:
@@ -4,73 +4,72 @@ using System.IO;
|
||||
using Jellyfin.MediaEncoding.Keyframes.Matroska.Extensions;
|
||||
using NEbml.Core;
|
||||
|
||||
namespace Jellyfin.MediaEncoding.Keyframes.Matroska
|
||||
namespace Jellyfin.MediaEncoding.Keyframes.Matroska;
|
||||
|
||||
/// <summary>
|
||||
/// The keyframe extractor for the matroska container.
|
||||
/// </summary>
|
||||
public static class MatroskaKeyframeExtractor
|
||||
{
|
||||
/// <summary>
|
||||
/// The keyframe extractor for the matroska container.
|
||||
/// Extracts the keyframes in ticks (scaled using the container timestamp scale) from the matroska container.
|
||||
/// </summary>
|
||||
public static class MatroskaKeyframeExtractor
|
||||
/// <param name="filePath">The file path.</param>
|
||||
/// <returns>An instance of <see cref="KeyframeData"/>.</returns>
|
||||
public static KeyframeData GetKeyframeData(string filePath)
|
||||
{
|
||||
/// <summary>
|
||||
/// Extracts the keyframes in ticks (scaled using the container timestamp scale) from the matroska container.
|
||||
/// </summary>
|
||||
/// <param name="filePath">The file path.</param>
|
||||
/// <returns>An instance of <see cref="KeyframeData"/>.</returns>
|
||||
public static KeyframeData GetKeyframeData(string filePath)
|
||||
using var stream = File.OpenRead(filePath);
|
||||
using var reader = new EbmlReader(stream);
|
||||
|
||||
var seekHead = reader.ReadSeekHead();
|
||||
var info = reader.ReadInfo(seekHead.InfoPosition);
|
||||
var videoTrackNumber = reader.FindFirstTrackNumberByType(seekHead.TracksPosition, MatroskaConstants.TrackTypeVideo);
|
||||
|
||||
var keyframes = new List<long>();
|
||||
reader.ReadAt(seekHead.CuesPosition);
|
||||
reader.EnterContainer();
|
||||
|
||||
while (reader.FindElement(MatroskaConstants.CuePoint))
|
||||
{
|
||||
using var stream = File.OpenRead(filePath);
|
||||
using var reader = new EbmlReader(stream);
|
||||
|
||||
var seekHead = reader.ReadSeekHead();
|
||||
var info = reader.ReadInfo(seekHead.InfoPosition);
|
||||
var videoTrackNumber = reader.FindFirstTrackNumberByType(seekHead.TracksPosition, MatroskaConstants.TrackTypeVideo);
|
||||
|
||||
var keyframes = new List<long>();
|
||||
reader.ReadAt(seekHead.CuesPosition);
|
||||
reader.EnterContainer();
|
||||
ulong? trackNumber = null;
|
||||
// Mandatory element
|
||||
reader.FindElement(MatroskaConstants.CueTime);
|
||||
var cueTime = reader.ReadUInt();
|
||||
|
||||
while (reader.FindElement(MatroskaConstants.CuePoint))
|
||||
// Mandatory element
|
||||
reader.FindElement(MatroskaConstants.CueTrackPositions);
|
||||
reader.EnterContainer();
|
||||
if (reader.FindElement(MatroskaConstants.CuePointTrackNumber))
|
||||
{
|
||||
reader.EnterContainer();
|
||||
ulong? trackNumber = null;
|
||||
// Mandatory element
|
||||
reader.FindElement(MatroskaConstants.CueTime);
|
||||
var cueTime = reader.ReadUInt();
|
||||
|
||||
// Mandatory element
|
||||
reader.FindElement(MatroskaConstants.CueTrackPositions);
|
||||
reader.EnterContainer();
|
||||
if (reader.FindElement(MatroskaConstants.CuePointTrackNumber))
|
||||
{
|
||||
trackNumber = reader.ReadUInt();
|
||||
}
|
||||
|
||||
reader.LeaveContainer();
|
||||
|
||||
if (trackNumber == videoTrackNumber)
|
||||
{
|
||||
keyframes.Add(ScaleToTicks(cueTime, info.TimestampScale));
|
||||
}
|
||||
|
||||
reader.LeaveContainer();
|
||||
trackNumber = reader.ReadUInt();
|
||||
}
|
||||
|
||||
reader.LeaveContainer();
|
||||
|
||||
var result = new KeyframeData(ScaleToTicks(info.Duration ?? 0, info.TimestampScale), keyframes);
|
||||
return result;
|
||||
if (trackNumber == videoTrackNumber)
|
||||
{
|
||||
keyframes.Add(ScaleToTicks(cueTime, info.TimestampScale));
|
||||
}
|
||||
|
||||
reader.LeaveContainer();
|
||||
}
|
||||
|
||||
private static long ScaleToTicks(ulong unscaledValue, long timestampScale)
|
||||
{
|
||||
// TimestampScale is in nanoseconds, scale it to get the value in ticks, 1 tick == 100 ns
|
||||
return (long)unscaledValue * timestampScale / 100;
|
||||
}
|
||||
reader.LeaveContainer();
|
||||
|
||||
private static long ScaleToTicks(double unscaledValue, long timestampScale)
|
||||
{
|
||||
// TimestampScale is in nanoseconds, scale it to get the value in ticks, 1 tick == 100 ns
|
||||
return Convert.ToInt64(unscaledValue * timestampScale / 100);
|
||||
}
|
||||
var result = new KeyframeData(ScaleToTicks(info.Duration ?? 0, info.TimestampScale), keyframes);
|
||||
return result;
|
||||
}
|
||||
|
||||
private static long ScaleToTicks(ulong unscaledValue, long timestampScale)
|
||||
{
|
||||
// TimestampScale is in nanoseconds, scale it to get the value in ticks, 1 tick == 100 ns
|
||||
return (long)unscaledValue * timestampScale / 100;
|
||||
}
|
||||
|
||||
private static long ScaleToTicks(double unscaledValue, long timestampScale)
|
||||
{
|
||||
// TimestampScale is in nanoseconds, scale it to get the value in ticks, 1 tick == 100 ns
|
||||
return Convert.ToInt64(unscaledValue * timestampScale / 100);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user