400 lines
14 KiB
C#
400 lines
14 KiB
C#
using System;
|
|
using System.Diagnostics;
|
|
using System.IO;
|
|
using System.Text;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
using MediaBrowser.Common.Configuration;
|
|
using MediaBrowser.Controller.Configuration;
|
|
using MediaBrowser.Controller.Library;
|
|
using MediaBrowser.Controller.MediaEncoding;
|
|
using MediaBrowser.Controller.Session;
|
|
using MediaBrowser.Model.Configuration;
|
|
using MediaBrowser.Model.Diagnostics;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Model.IO;
|
|
using MediaBrowser.Model.MediaInfo;
|
|
using Microsoft.Extensions.Logging;
|
|
|
|
namespace MediaBrowser.MediaEncoding.Encoder
|
|
{
|
|
public abstract class BaseEncoder
|
|
{
|
|
protected readonly MediaEncoder MediaEncoder;
|
|
protected readonly ILogger Logger;
|
|
protected readonly IServerConfigurationManager ConfigurationManager;
|
|
protected readonly IFileSystem FileSystem;
|
|
protected readonly IIsoManager IsoManager;
|
|
protected readonly ILibraryManager LibraryManager;
|
|
protected readonly ISessionManager SessionManager;
|
|
protected readonly ISubtitleEncoder SubtitleEncoder;
|
|
protected readonly IMediaSourceManager MediaSourceManager;
|
|
protected IProcessFactory ProcessFactory;
|
|
|
|
protected EncodingHelper EncodingHelper;
|
|
|
|
protected BaseEncoder(
|
|
MediaEncoder mediaEncoder,
|
|
ILogger logger,
|
|
IServerConfigurationManager configurationManager,
|
|
IFileSystem fileSystem,
|
|
IIsoManager isoManager,
|
|
ILibraryManager libraryManager,
|
|
ISessionManager sessionManager,
|
|
ISubtitleEncoder subtitleEncoder,
|
|
IMediaSourceManager mediaSourceManager,
|
|
IProcessFactory processFactory)
|
|
{
|
|
MediaEncoder = mediaEncoder;
|
|
Logger = logger;
|
|
ConfigurationManager = configurationManager;
|
|
FileSystem = fileSystem;
|
|
IsoManager = isoManager;
|
|
LibraryManager = libraryManager;
|
|
SessionManager = sessionManager;
|
|
SubtitleEncoder = subtitleEncoder;
|
|
MediaSourceManager = mediaSourceManager;
|
|
ProcessFactory = processFactory;
|
|
|
|
EncodingHelper = new EncodingHelper(MediaEncoder, FileSystem, SubtitleEncoder);
|
|
}
|
|
|
|
public async Task<EncodingJob> Start(
|
|
EncodingJobOptions options,
|
|
IProgress<double> progress,
|
|
CancellationToken cancellationToken)
|
|
{
|
|
var encodingJob = await new EncodingJobFactory(Logger, LibraryManager, MediaSourceManager, MediaEncoder)
|
|
.CreateJob(options, EncodingHelper, IsVideoEncoder, progress, cancellationToken).ConfigureAwait(false);
|
|
|
|
encodingJob.OutputFilePath = GetOutputFilePath(encodingJob);
|
|
Directory.CreateDirectory(Path.GetDirectoryName(encodingJob.OutputFilePath));
|
|
|
|
encodingJob.ReadInputAtNativeFramerate = options.ReadInputAtNativeFramerate;
|
|
|
|
await AcquireResources(encodingJob, cancellationToken).ConfigureAwait(false);
|
|
|
|
var commandLineArgs = GetCommandLineArguments(encodingJob);
|
|
|
|
Process process = new Process
|
|
{
|
|
StartInfo = new ProcessStartInfo
|
|
{
|
|
WindowStyle = ProcessWindowStyle.Hidden,
|
|
CreateNoWindow = true,
|
|
UseShellExecute = false,
|
|
|
|
// Must consume both stdout and stderr or deadlocks may occur
|
|
//RedirectStandardOutput = true,
|
|
RedirectStandardError = true,
|
|
RedirectStandardInput = true,
|
|
|
|
FileName = MediaEncoder.EncoderPath,
|
|
Arguments = commandLineArgs,
|
|
|
|
ErrorDialog = false
|
|
},
|
|
EnableRaisingEvents = true
|
|
};
|
|
|
|
var workingDirectory = GetWorkingDirectory(options);
|
|
if (!string.IsNullOrWhiteSpace(workingDirectory))
|
|
{
|
|
process.StartInfo.WorkingDirectory = workingDirectory;
|
|
}
|
|
|
|
OnTranscodeBeginning(encodingJob);
|
|
|
|
var commandLineLogMessage = process.StartInfo.FileName + " " + process.StartInfo.Arguments;
|
|
Logger.LogInformation(commandLineLogMessage);
|
|
|
|
var logFilePath = Path.Combine(ConfigurationManager.CommonApplicationPaths.LogDirectoryPath, "transcode-" + Guid.NewGuid() + ".txt");
|
|
Directory.CreateDirectory(Path.GetDirectoryName(logFilePath));
|
|
|
|
// FFMpeg writes debug/error info to stderr. This is useful when debugging so let's put it in the log directory.
|
|
encodingJob.LogFileStream = FileSystem.GetFileStream(logFilePath, FileOpenMode.Create, FileAccessMode.Write, FileShareMode.Read, true);
|
|
|
|
var commandLineLogMessageBytes = Encoding.UTF8.GetBytes(commandLineLogMessage + Environment.NewLine + Environment.NewLine);
|
|
await encodingJob.LogFileStream.WriteAsync(commandLineLogMessageBytes, 0, commandLineLogMessageBytes.Length, cancellationToken).ConfigureAwait(false);
|
|
|
|
process.Exited += (sender, args) => OnFfMpegProcessExited(process, encodingJob);
|
|
|
|
try
|
|
{
|
|
process.Start();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Logger.LogError(ex, "Error starting ffmpeg");
|
|
|
|
OnTranscodeFailedToStart(encodingJob.OutputFilePath, encodingJob);
|
|
|
|
throw;
|
|
}
|
|
|
|
cancellationToken.Register(async () => await Cancel(process, encodingJob));
|
|
|
|
// Important - don't await the log task or we won't be able to kill ffmpeg when the user stops playback
|
|
new JobLogger(Logger).StartStreamingLog(encodingJob, process.StandardError.BaseStream, encodingJob.LogFileStream);
|
|
|
|
Logger.LogInformation("test0");
|
|
|
|
if (File.Exists(encodingJob.OutputFilePath))
|
|
{
|
|
return encodingJob;
|
|
}
|
|
|
|
Logger.LogInformation("test1");
|
|
|
|
using (var watcher = new FileSystemWatcher(Path.GetDirectoryName(encodingJob.OutputFilePath)))
|
|
{
|
|
var tcs = new TaskCompletionSource<bool>();
|
|
string fileName = Path.GetFileName(encodingJob.OutputFilePath);
|
|
|
|
watcher.Created += (s, e) =>
|
|
{
|
|
if (e.Name == fileName)
|
|
{
|
|
tcs.TrySetResult(true);
|
|
}
|
|
};
|
|
|
|
watcher.EnableRaisingEvents = true;
|
|
|
|
Logger.LogInformation("test2");
|
|
|
|
// Wait for the file to or for the process to stop
|
|
await Task.WhenAny(encodingJob.TaskCompletionSource.Task, tcs.Task).ConfigureAwait(false);
|
|
|
|
Logger.LogInformation("test3");
|
|
|
|
return encodingJob;
|
|
}
|
|
}
|
|
|
|
private async Task Cancel(Process process, EncodingJob job)
|
|
{
|
|
Logger.LogInformation("Killing ffmpeg process for {0}", job.OutputFilePath);
|
|
|
|
//process.Kill();
|
|
await process.StandardInput.WriteLineAsync("q");
|
|
|
|
job.IsCancelled = true;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Processes the exited.
|
|
/// </summary>
|
|
/// <param name="process">The process.</param>
|
|
/// <param name="job">The job.</param>
|
|
private void OnFfMpegProcessExited(Process process, EncodingJob job)
|
|
{
|
|
job.HasExited = true;
|
|
|
|
Logger.LogDebug("Disposing stream resources");
|
|
job.Dispose();
|
|
|
|
var isSuccessful = false;
|
|
|
|
try
|
|
{
|
|
var exitCode = process.ExitCode;
|
|
Logger.LogInformation("FFMpeg exited with code {0}", exitCode);
|
|
|
|
isSuccessful = exitCode == 0;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Logger.LogError(ex, "FFMpeg exited with an error.");
|
|
}
|
|
|
|
if (isSuccessful && !job.IsCancelled)
|
|
{
|
|
job.TaskCompletionSource.TrySetResult(true);
|
|
}
|
|
else if (job.IsCancelled)
|
|
{
|
|
try
|
|
{
|
|
DeleteFiles(job);
|
|
}
|
|
catch
|
|
{
|
|
}
|
|
try
|
|
{
|
|
job.TaskCompletionSource.TrySetException(new OperationCanceledException());
|
|
}
|
|
catch
|
|
{
|
|
}
|
|
}
|
|
else
|
|
{
|
|
try
|
|
{
|
|
DeleteFiles(job);
|
|
}
|
|
catch
|
|
{
|
|
}
|
|
try
|
|
{
|
|
job.TaskCompletionSource.TrySetException(new Exception("Encoding failed"));
|
|
}
|
|
catch
|
|
{
|
|
}
|
|
}
|
|
|
|
// This causes on exited to be called twice:
|
|
//try
|
|
//{
|
|
// // Dispose the process
|
|
// process.Dispose();
|
|
//}
|
|
//catch (Exception ex)
|
|
//{
|
|
// Logger.LogError("Error disposing ffmpeg.", ex);
|
|
//}
|
|
}
|
|
|
|
protected virtual void DeleteFiles(EncodingJob job)
|
|
{
|
|
FileSystem.DeleteFile(job.OutputFilePath);
|
|
}
|
|
|
|
private void OnTranscodeBeginning(EncodingJob job)
|
|
{
|
|
job.ReportTranscodingProgress(null, null, null, null, null);
|
|
}
|
|
|
|
private void OnTranscodeFailedToStart(string path, EncodingJob job)
|
|
{
|
|
if (!string.IsNullOrWhiteSpace(job.Options.DeviceId))
|
|
{
|
|
SessionManager.ClearTranscodingInfo(job.Options.DeviceId);
|
|
}
|
|
}
|
|
|
|
protected abstract bool IsVideoEncoder { get; }
|
|
|
|
protected virtual string GetWorkingDirectory(EncodingJobOptions options)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
protected EncodingOptions GetEncodingOptions()
|
|
{
|
|
return ConfigurationManager.GetConfiguration<EncodingOptions>("encoding");
|
|
}
|
|
|
|
protected abstract string GetCommandLineArguments(EncodingJob job);
|
|
|
|
private string GetOutputFilePath(EncodingJob state)
|
|
{
|
|
var folder = string.IsNullOrWhiteSpace(state.Options.TempDirectory) ?
|
|
ConfigurationManager.ApplicationPaths.TranscodingTempPath :
|
|
state.Options.TempDirectory;
|
|
|
|
var outputFileExtension = GetOutputFileExtension(state);
|
|
|
|
var filename = state.Id + (outputFileExtension ?? string.Empty).ToLowerInvariant();
|
|
return Path.Combine(folder, filename);
|
|
}
|
|
|
|
protected virtual string GetOutputFileExtension(EncodingJob state)
|
|
{
|
|
if (!string.IsNullOrWhiteSpace(state.Options.Container))
|
|
{
|
|
return "." + state.Options.Container;
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the name of the output video codec
|
|
/// </summary>
|
|
/// <param name="state">The state.</param>
|
|
/// <returns>System.String.</returns>
|
|
protected string GetVideoDecoder(EncodingJob state)
|
|
{
|
|
if (string.Equals(state.OutputVideoCodec, "copy", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
return null;
|
|
}
|
|
|
|
// Only use alternative encoders for video files.
|
|
// When using concat with folder rips, if the mfx session fails to initialize, ffmpeg will be stuck retrying and will not exit gracefully
|
|
// Since transcoding of folder rips is expiremental anyway, it's not worth adding additional variables such as this.
|
|
if (state.VideoType != VideoType.VideoFile)
|
|
{
|
|
return null;
|
|
}
|
|
|
|
if (state.VideoStream != null && !string.IsNullOrWhiteSpace(state.VideoStream.Codec))
|
|
{
|
|
if (string.Equals(GetEncodingOptions().HardwareAccelerationType, "qsv", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
switch (state.MediaSource.VideoStream.Codec.ToLowerInvariant())
|
|
{
|
|
case "avc":
|
|
case "h264":
|
|
if (MediaEncoder.SupportsDecoder("h264_qsv"))
|
|
{
|
|
// Seeing stalls and failures with decoding. Not worth it compared to encoding.
|
|
return "-c:v h264_qsv ";
|
|
}
|
|
break;
|
|
case "mpeg2video":
|
|
if (MediaEncoder.SupportsDecoder("mpeg2_qsv"))
|
|
{
|
|
return "-c:v mpeg2_qsv ";
|
|
}
|
|
break;
|
|
case "vc1":
|
|
if (MediaEncoder.SupportsDecoder("vc1_qsv"))
|
|
{
|
|
return "-c:v vc1_qsv ";
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
// leave blank so ffmpeg will decide
|
|
return null;
|
|
}
|
|
|
|
private async Task AcquireResources(EncodingJob state, CancellationToken cancellationToken)
|
|
{
|
|
if (state.VideoType == VideoType.Iso && state.IsoType.HasValue && IsoManager.CanMount(state.MediaPath))
|
|
{
|
|
state.IsoMount = await IsoManager.Mount(state.MediaPath, cancellationToken).ConfigureAwait(false);
|
|
}
|
|
|
|
if (state.MediaSource.RequiresOpening && string.IsNullOrWhiteSpace(state.Options.LiveStreamId))
|
|
{
|
|
var liveStreamResponse = await MediaSourceManager.OpenLiveStream(new LiveStreamRequest
|
|
{
|
|
OpenToken = state.MediaSource.OpenToken
|
|
|
|
}, cancellationToken).ConfigureAwait(false);
|
|
|
|
EncodingHelper.AttachMediaSourceInfo(state, liveStreamResponse.MediaSource, null);
|
|
|
|
if (state.IsVideoRequest)
|
|
{
|
|
EncodingHelper.TryStreamCopy(state);
|
|
}
|
|
}
|
|
|
|
if (state.MediaSource.BufferMs.HasValue)
|
|
{
|
|
await Task.Delay(state.MediaSource.BufferMs.Value, cancellationToken).ConfigureAwait(false);
|
|
}
|
|
}
|
|
}
|
|
}
|