public void PrepareJobFiles(MediaEncoderSettings settings)
{
settings.JobIndex = ++JobIndex;
// Files must be prepared before adding to queue so that user can replace preview files.
PathManager.DeleteJobFiles(settings.JobIndex);
File.Delete(PathManager.PreviewScriptFile);
File.Delete(PathManager.PreviewSettingsFile);
if (settings.ConvertToAvi)
{
PathManager.SafeMove(PathManager.PreviewSourceFile, settings.InputFile);
}
if (settings.Deshaker)
{
PathManager.SafeMove(PathManager.PreviewDeshakerLog, settings.DeshakerLog);
}
settings.Save(settings.SettingsFile);
AviSynthScriptBuilder Script = new AviSynthScriptBuilder(settings.CustomScript);
if (Script.IsEmpty)
{
Script = MediaEncoderScript.GenerateVideoScript(settings, settings.InputFile, false, true);
}
else
{
Script.Replace(Script.GetAsciiPath(PathManager.PreviewSourceFile), Script.GetAsciiPath(settings.InputFile));
Script.Replace(Script.GetAsciiPath(PathManager.PreviewDeshakerLog), Script.GetAsciiPath(settings.DeshakerLog));
}
Script.WriteToFile(settings.ScriptFile);
// if (settings.DeshakerSettings.PrescanAction == PrescanType.Full)
}
/// <summary>
/// Saves the audio output of specified script into a WAV file.
/// </summary>
/// <param name="settings">An object containing the encoding settings.</param>
/// <param name="destination">The WAV file to write.</param>
/// <param name="options">The options that control the behaviors of the process.</param>
public static void SaveAudioToWav(MediaEncoderSettings settings, string destination, ProcessStartOptions options)
{
string TempFile = settings.TempFile + ".avs";
AviSynthScriptBuilder Script = new AviSynthScriptBuilder();
if (settings.VideoAction != VideoAction.Copy)
{
// Read source script.
Script.Script = File.ReadAllText(settings.ScriptFile);
// Remote MT code.
Script.RemoveMT();
Script.AppendLine("Trim(0,0)");
}
else
{
// Read full video file.
Script.AddPluginPath();
if (settings.ConvertToAvi || settings.InputFile.ToLower().EndsWith(".avi"))
{
Script.OpenAvi(settings.InputFile, !string.IsNullOrEmpty(settings.SourceAudioFormat));
}
else
{
Script.OpenDirect(settings.InputFile, !string.IsNullOrEmpty(settings.SourceAudioFormat));
}
Script.AppendLine("KillVideo()");
}
Script.AppendLine();
// Add audio gain.
if (settings.AudioGain.HasValue && settings.AudioGain != 0)
{
Script.AppendLine("AmplifydB({0})", settings.AudioGain.Value);
}
if (settings.ChangeAudioPitch)
{
// Change pitch to 432hz.
Script.LoadPluginDll("TimeStretch.dll");
Script.AppendLine("ResampleAudio(48000)");
Script.AppendLine("TimeStretchPlugin(pitch = 100.0 * 0.98181819915771484)");
}
// Add TWriteWAV.
Script.AppendLine();
Script.LoadPluginDll("TWriteAVI.dll");
Script.AppendLine(@"TWriteWAV(""{0}"", true)", Script.GetAsciiPath(destination));
Script.AppendLine("ForceProcessWAV()");
// Write temp script.
Script.WriteToFile(TempFile);
// Execute. It aways returns an error but the file is generated.
FFmpegProcess Worker = new FFmpegProcess(options);
Worker.RunAvisynth(TempFile);
File.Delete(TempFile);
}
public void GenerateScript(MediaEncoderSettings settings, bool preview, bool multiThreaded) {
AviSynthScriptBuilder Script = new AviSynthScriptBuilder(settings.CustomScript);
if (Script.IsEmpty)
Script = MediaEncoderScript.GenerateVideoScript(settings, GetPreviewSourceFile(settings), preview, multiThreaded);
else if (preview) {
Script.RemoveMT();
Script.AppendLine(@"ConvertToRGB32(matrix=""Rec709"")");
} else if (!multiThreaded)
Script.RemoveMT();
Script.WriteToFile(PathManager.PreviewScriptFile);
settings.Save(PathManager.PreviewSettingsFile);
}
/// <summary>
/// Gets an AviSynth clip information by running a script that outputs the frame count to a file.
/// </summary>
/// <param name="source">The AviSynth script to get information for.</param>
/// <param name="silent">If true, the x264 window will be hidden.</param>
/// <returns>The clip information.</returns>
public static ClipInfo GetClipInfo(MediaEncoderSettings settings, string scriptFile, bool silent)
{
string TempScript = settings.TempFile + ".avs";
string TempResult = settings.TempFile + ".txt";
string TempOut = settings.TempFile + ".y4m";
// Read source script and remove MT. Also remove Deshaker if present.
string FileContent = File.ReadAllText(scriptFile);
FileContent.Replace(Environment.NewLine + "Deshaker", Environment.NewLine + "#Deshaker");
AviSynthScriptBuilder Script = new AviSynthScriptBuilder(FileContent);
Script.RemoveMT();
//Script.DitherOut(false);
// Get frame count.
Script.AppendLine();
Script.AppendLine(@"WriteFileStart(""{0}"", ""FrameRate""{1}""Framecount"")", TempResult, @", """""" "","" """""", ");
Script.AppendLine("Trim(0,-1)");
Script.WriteToFile(TempScript);
// Run script.
Run(@"Encoder\avs2yuv.exe", String.Format(@"""{0}"" -o {1}", TempScript, TempOut), true);
// Read frame count
ClipInfo Result = null;
if (File.Exists(TempResult))
{
string FileString = File.ReadAllText(TempResult);
string[] FileValues = FileString.Split(',');
Result = new ClipInfo();
try {
Result.FrameRate = float.Parse(FileValues[0], CultureInfo.InvariantCulture);
Result.FrameCount = int.Parse(FileValues[1]);
}
catch {
Result = null;
}
}
// Delete temp files.
File.Delete(TempScript);
File.Delete(TempResult);
File.Delete(TempOut); // Dummy file that received avs2yuv output.
return(Result);
}
/// <summary>
/// Returns whether the GPU supports the latest version of KNLMeans with OpenCL 1.2
/// </summary>
/// <param name="supports11">If true, it will instead test whether the GPU supports OpenCL 1.1</param>
/// <returns>True if OpenCL is supported.</returns>
private static bool GpuSupportsOpenCL(bool version12)
{
string TempScript = Settings.TempFilesPath + "Temp.avs";
string TempResult = Settings.TempFilesPath + "Temp.txt";
string TempOut = Settings.TempFilesPath + "Temp.y4m";
AviSynthScriptBuilder Script = new AviSynthScriptBuilder();
Script.AddPluginPath();
Script.LoadPluginDll(string.Format("KNLMeansCL{0}.dll", version12 ? "" : "-6.11"));
Script.AppendLine(@"colorbars(pixel_type = ""yv12"").killaudio().trim(1, 1)");
Script.AppendLine("Result = true");
Script.AppendLine("try {");
Script.AppendLine(@"KNLMeansCL(device_type=""GPU"")");
Script.AppendLine("} catch(error_msg) {");
Script.AppendLine("Result = false");
Script.AppendLine("}");
Script.AppendLine(@"WriteFileStart(""{0}"", string(Result))", TempResult);
Script.WriteToFile(TempScript);
// Run script.
FfmpegBusiness.Run(@"Encoder\avs2yuv.exe", String.Format(@"""{0}"" -o ""{1}""", TempScript, TempOut), true);
// Read frame count
bool Result = false;
if (File.Exists(TempResult))
{
string FileString = File.ReadAllText(TempResult);
try {
Result = bool.Parse(FileString.TrimEnd());
}
catch {
Result = false;
}
}
// Delete temp files.
File.Delete(TempScript);
File.Delete(TempResult);
File.Delete(TempOut); // Dummy file that received avs2yuv output.
return(Result);
}
/// <summary>
/// Calculates auto-crop coordinates for specified video script.
/// </summary>
/// <param name="filePath">The script to analyze.</param>
/// <param name="sourceWidth">The width of the source video.</param>
/// <param name="sourceHeight">The height of the source video.</param>
/// <param name="options">The options that control the behaviors of the process.</param>
/// <returns>The auto-crop coordinates.</returns>
public static Rect GetAutoCropRect(string filePath, int sourceWidth, int sourceHeight, ProcessStartOptions options)
{
string TempScript = PathManager.GetTempFile(".avs");
string TempResult = Path.ChangeExtension(TempScript, ".txt");
// Create script to get auto-crop coordinates
AviSynthScriptBuilder Script = new AviSynthScriptBuilder();
Script.AddPluginPath();
Script.OpenDirect(filePath, false);
Script.LoadPluginDll("RoboCrop26.dll");
Script.AppendLine(@"RoboCrop(LogFn=""{0}"")", Script.GetAsciiPath(TempResult));
Script.AppendLine("Trim(0,-1)");
Script.WriteToFile(TempScript);
// Run script.
FFmpegProcess Worker = new FFmpegProcess(options);
Worker.RunAvisynth(TempScript);
// Read auto crop coordinates
Rect Result = new Rect();
if (File.Exists(TempResult))
{
string[] Values = File.ReadAllText(TempResult).Split(' ');
if (Values.Length >= 13)
{
Result.Left = Math.Max(int.Parse(Values[10]), 0);
Result.Top = Math.Max(int.Parse(Values[11]), 0);
Result.Right = Math.Max(sourceWidth - int.Parse(Values[12]), 0);
Result.Bottom = Math.Max(sourceHeight - int.Parse(Values[13]), 0);
// Make result conservative, we have to visually see a line of black border to know the right dimensions.
if (Result.Left > 0)
{
Result.Left--;
}
if (Result.Top > 0)
{
Result.Top--;
}
if (Result.Right > 0)
{
Result.Right--;
}
if (Result.Bottom > 0)
{
Result.Bottom--;
}
}
}
// Delete temp files.
Exception LastError = null;
for (int i = 0; i < 10; i++)
{
try {
File.Delete(TempScript);
File.Delete(TempResult);
break;
} catch (Exception e) {
LastError = e;
System.Threading.Thread.Sleep(500);
}
}
if (LastError != null)
{
throw LastError;
}
return(Result);
}
public static Rect GetAutoCropRect(MediaEncoderSettings settings, bool silent)
{
string TempScript = settings.TempFile + ".avs";
string TempResult = settings.TempFile + ".txt";
string TempOut = settings.TempFile + ".y4m";
// Create script to get auto-crop coordinates
AviSynthScriptBuilder Script = new AviSynthScriptBuilder();
Script.AddPluginPath();
Script.OpenDirect(settings.FilePath, false);
Script.LoadPluginDll("RoboCrop26.dll");
Script.AppendLine(@"RoboCrop(LogFn=""{0}"")", Script.GetAsciiPath(TempResult));
Script.AppendLine("Trim(0,-1)");
Script.WriteToFile(TempScript);
// Run script.
Run(@"Encoder\avs2yuv.exe", String.Format(@"""{0}"" -o ""{1}""", TempScript, TempOut), true);
// Read auto crop coordinates
Rect Result = new Rect();
if (File.Exists(TempResult))
{
string[] Values = File.ReadAllText(TempResult).Split(' ');
if (Values.Length >= 13)
{
Result.Left = int.Parse(Values[10]);
Result.Top = int.Parse(Values[11]);
Result.Right = settings.SourceWidth.Value - int.Parse(Values[12]);
Result.Bottom = settings.SourceHeight.Value - int.Parse(Values[13]);
// Make result conservative, we have to visually see a line of black border to know the right dimensions.
if (Result.Left > 0)
{
Result.Left--;
}
if (Result.Top > 0)
{
Result.Top--;
}
if (Result.Right > 0)
{
Result.Right--;
}
if (Result.Bottom > 0)
{
Result.Bottom--;
}
}
}
// Delete temp files.
Exception LastError = null;
for (int i = 0; i < 10; i++)
{
try {
File.Delete(TempScript);
File.Delete(TempResult);
File.Delete(TempOut); // Dummy file that received avs2yuv output.
break;
}
catch (Exception e) {
LastError = e;
System.Threading.Thread.Sleep(500);
}
}
if (LastError != null)
{
throw LastError;
}
return(Result);
}