mirror of
https://github.com/jellyfin/jellyfin.git
synced 2024-09-07 03:48:06 +02:00
196 lines
6.1 KiB
C#
196 lines
6.1 KiB
C#
using MediaBrowser.Controller.Library;
|
|
using MediaBrowser.Controller.MediaEncoding;
|
|
using MediaBrowser.Model.Dlna;
|
|
using MediaBrowser.Model.Drawing;
|
|
using MediaBrowser.Model.Dto;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Model.IO;
|
|
using MediaBrowser.Model.Logging;
|
|
using MediaBrowser.Model.MediaInfo;
|
|
using MediaBrowser.Model.Net;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace MediaBrowser.MediaEncoding.Encoder
|
|
{
|
|
public class EncodingJob : EncodingJobInfo, IDisposable
|
|
{
|
|
public bool HasExited { get; internal set; }
|
|
public bool IsCancelled { get; internal set; }
|
|
|
|
public Stream LogFileStream { get; set; }
|
|
public IProgress<double> Progress { get; set; }
|
|
public TaskCompletionSource<bool> TaskCompletionSource;
|
|
|
|
public EncodingJobOptions Options
|
|
{
|
|
get { return (EncodingJobOptions) BaseRequest; }
|
|
set { BaseRequest = value; }
|
|
}
|
|
|
|
public string Id { get; set; }
|
|
|
|
public string MimeType { get; set; }
|
|
public bool EstimateContentLength { get; set; }
|
|
public TranscodeSeekInfo TranscodeSeekInfo { get; set; }
|
|
public long? EncodingDurationTicks { get; set; }
|
|
|
|
public string ItemType { get; set; }
|
|
|
|
public string GetMimeType(string outputPath)
|
|
{
|
|
if (!string.IsNullOrEmpty(MimeType))
|
|
{
|
|
return MimeType;
|
|
}
|
|
|
|
return MimeTypes.GetMimeType(outputPath);
|
|
}
|
|
|
|
private readonly ILogger _logger;
|
|
private readonly IMediaSourceManager _mediaSourceManager;
|
|
|
|
public EncodingJob(ILogger logger, IMediaSourceManager mediaSourceManager) :
|
|
base(logger, mediaSourceManager, TranscodingJobType.Progressive)
|
|
{
|
|
_logger = logger;
|
|
_mediaSourceManager = mediaSourceManager;
|
|
Id = Guid.NewGuid().ToString("N");
|
|
|
|
_logger = logger;
|
|
TaskCompletionSource = new TaskCompletionSource<bool>();
|
|
}
|
|
|
|
public override void Dispose()
|
|
{
|
|
DisposeLiveStream();
|
|
DisposeLogStream();
|
|
}
|
|
|
|
private void DisposeLogStream()
|
|
{
|
|
if (LogFileStream != null)
|
|
{
|
|
try
|
|
{
|
|
LogFileStream.Dispose();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.ErrorException("Error disposing log stream", ex);
|
|
}
|
|
|
|
LogFileStream = null;
|
|
}
|
|
}
|
|
|
|
private async void DisposeLiveStream()
|
|
{
|
|
if (MediaSource.RequiresClosing && string.IsNullOrWhiteSpace(Options.LiveStreamId) && !string.IsNullOrWhiteSpace(MediaSource.LiveStreamId))
|
|
{
|
|
try
|
|
{
|
|
await _mediaSourceManager.CloseLiveStream(MediaSource.LiveStreamId).ConfigureAwait(false);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.ErrorException("Error closing media source", ex);
|
|
}
|
|
}
|
|
}
|
|
|
|
public string OutputFilePath { get; set; }
|
|
|
|
public string ActualOutputVideoCodec
|
|
{
|
|
get
|
|
{
|
|
var codec = OutputVideoCodec;
|
|
|
|
if (string.Equals(codec, "copy", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
var stream = VideoStream;
|
|
|
|
if (stream != null)
|
|
{
|
|
return stream.Codec;
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
return codec;
|
|
}
|
|
}
|
|
|
|
public string ActualOutputAudioCodec
|
|
{
|
|
get
|
|
{
|
|
var codec = OutputAudioCodec;
|
|
|
|
if (string.Equals(codec, "copy", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
var stream = AudioStream;
|
|
|
|
if (stream != null)
|
|
{
|
|
return stream.Codec;
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
return codec;
|
|
}
|
|
}
|
|
|
|
public void ReportTranscodingProgress(TimeSpan? transcodingPosition, float? framerate, double? percentComplete, long? bytesTranscoded, int? bitRate)
|
|
{
|
|
var ticks = transcodingPosition.HasValue ? transcodingPosition.Value.Ticks : (long?)null;
|
|
|
|
// job.Framerate = framerate;
|
|
|
|
if (!percentComplete.HasValue && ticks.HasValue && RunTimeTicks.HasValue)
|
|
{
|
|
var pct = ticks.Value / RunTimeTicks.Value;
|
|
percentComplete = pct * 100;
|
|
}
|
|
|
|
if (percentComplete.HasValue)
|
|
{
|
|
Progress.Report(percentComplete.Value);
|
|
}
|
|
|
|
// job.TranscodingPositionTicks = ticks;
|
|
// job.BytesTranscoded = bytesTranscoded;
|
|
|
|
var deviceId = Options.DeviceId;
|
|
|
|
if (!string.IsNullOrWhiteSpace(deviceId))
|
|
{
|
|
var audioCodec = ActualOutputVideoCodec;
|
|
var videoCodec = ActualOutputVideoCodec;
|
|
|
|
// SessionManager.ReportTranscodingInfo(deviceId, new TranscodingInfo
|
|
// {
|
|
// Bitrate = job.TotalOutputBitrate,
|
|
// AudioCodec = audioCodec,
|
|
// VideoCodec = videoCodec,
|
|
// Container = job.Options.OutputContainer,
|
|
// Framerate = framerate,
|
|
// CompletionPercentage = percentComplete,
|
|
// Width = job.OutputWidth,
|
|
// Height = job.OutputHeight,
|
|
// AudioChannels = job.OutputAudioChannels,
|
|
// IsAudioDirect = string.Equals(job.OutputAudioCodec, "copy", StringComparison.OrdinalIgnoreCase),
|
|
// IsVideoDirect = string.Equals(job.OutputVideoCodec, "copy", StringComparison.OrdinalIgnoreCase)
|
|
// });
|
|
}
|
|
}
|
|
}
|
|
}
|