flowframes/CodeLegacy/Ui/MainUiFunctions.cs

using Flowframes.Media;
using Flowframes.IO;
using Flowframes.Magick;
using Flowframes.Main;
using Flowframes.Os;
using System;
using System.Collections.Generic;
using System.Drawing;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;
using Flowframes.Data;
using Flowframes.MiscUtils;

namespace Flowframes.Ui
{
    class MainUiFunctions
    {
        public static async Task InitInput (TextBox outputTbox, TextBox inputTbox, TextBox fpsInTbox, bool start = false)
        {
            Program.mainForm.Enabled = false;

            try
            {
                Program.mainForm.SetTab(Program.mainForm.interpOptsTab.Name);
                Program.mainForm.ResetInputInfo();
                string path = inputTbox.Text.Trim();

                GetFrameCountCached.Clear();
                GetMediaResolutionCached.Clear();

                if (Config.GetBool(Config.Key.clearLogOnInput))
                    Logger.ClearLogBox();

                SetOutPath(outputTbox, inputTbox.Text.Trim().GetParentDir());

                Program.lastInputPath = path;
                Program.lastInputPathIsSsd = OsUtils.DriveIsSSD(path);

                if (!Program.lastInputPathIsSsd)
                    Logger.Log("Your file seems to be on an HDD or USB device. It is recommended to interpolate videos on an SSD drive for best performance.");

                Logger.Log("Importing file...");
                Interpolate.currentMediaFile = new MediaFile(path);
                await Interpolate.currentMediaFile.Initialize();
                Program.mainForm.currInDuration = Interpolate.currentMediaFile.DurationMs;
                Program.mainForm.currInDurationCut = Program.mainForm.currInDuration;
                Fraction fps = Interpolate.currentMediaFile.VideoStreams.Count > 0 ? Interpolate.currentMediaFile.VideoStreams[0].Rate : new Fraction();
                string fpsStr = fps.Float > 0 ? FormatUtils.Fraction(fps) : "Not Found";
                Program.mainForm.currInFpsDetected = fps;
                fpsInTbox.Text = fps.GetString();
                Logger.Log($"Video FPS: {fpsStr} - Total Number Of Frames: {Interpolate.currentMediaFile.FrameCount}", false, true);
                Program.mainForm.GetInputFpsTextbox().ReadOnly = (fps.Float > 0 && !Config.GetBool("allowCustomInputRate", false));
                Program.mainForm.currInFps = fps;
                Program.mainForm.currInFrames = Interpolate.currentMediaFile.FrameCount;
                Program.mainForm.UpdateInputInfo();
                CheckExistingFolder(path, outputTbox.Text.Trim());
                // await Task.Delay(10);
                await PrintResolution(path);
                // await Task.Delay(10);
                // InterpolationProgress.SetPreviewImg(await GetThumbnail(path));

                if (AutoEncodeResume.resumeNextRun)
                    Logger.Log($"Incomplete interpolation detected. Flowframes will resume the interpolation.");

                if (start)
                    Program.mainForm.runBtn_Click(null, null);
            }
            catch { }

            if (!Cli.AutoRun)
            {
                Program.mainForm.Enabled = true;
            }
        }

        public static bool SetOutPath (TextBox outputTbox, string outPath)
        {
            bool customOutDir = Config.GetInt("outFolderLoc") == 1;
            outputTbox.Text = customOutDir ? Config.Get("custOutDir").Trim() : outPath;

            if (customOutDir)
            {
                try
                {
                    Directory.CreateDirectory(outputTbox.Text);
                }
                catch (Exception e)
                {
                    Logger.Log($"Failed to create output folder: {e.Message}");
                    outputTbox.Text = outPath;
                    return false;
                }
            }

            return true;
        }

        static void CheckExistingFolder (string inpath, string outpath)
        {
            if (Interpolate.currentSettings == null || !Interpolate.currentSettings.stepByStep) return;
            string tmpFolder = InterpolateUtils.GetTempFolderLoc(inpath, outpath);

            if (Directory.Exists(tmpFolder))
            {
                int scnFrmAmount = IoUtils.GetAmountOfFiles(Path.Combine(tmpFolder, Paths.scenesDir), false, "*" + Interpolate.currentSettings.interpExt);  // TODO: Make this work if the frames extension was changed
                string scnFrames = scnFrmAmount > 0 ? $"{scnFrmAmount} scene frames" : "no scene frames";
                int srcFrmAmount = IoUtils.GetAmountOfFiles(Path.Combine(tmpFolder, Paths.framesDir), false, "*" + Interpolate.currentSettings.interpExt);
                string srcFrames = srcFrmAmount > 1 ? $"{srcFrmAmount} source frames" : "no source frames";
                int interpFrmAmount = IoUtils.GetAmountOfFiles(Path.Combine(tmpFolder, Paths.interpDir), false);
                string interpFrames = interpFrmAmount > 2 ? $"{interpFrmAmount} interpolated frames" : "no interpolated frames";
                string msg = $"A temporary folder for this video already exists. It contains {scnFrames}, {srcFrames}, {interpFrames}.";

                DialogResult dialogResult = UiUtils.ShowMessageBox($"{msg}\n\nClick \"Yes\" to use the existing files or \"No\" to delete them.", "Use files from existing temp folder?", MessageBoxButtons.YesNo);
                
                if (dialogResult == DialogResult.No)
                {
                    IoUtils.TryDeleteIfExists(tmpFolder);
                    Logger.Log("Deleted old temp folder.");
                }
            }
        }

        static async Task PrintResolution (string path)
        {
            Size res = new Size();

            if(path == Interpolate.currentSettings?.inPath)
                res = Interpolate.currentSettings.InputResolution;
            else
                res = await GetMediaResolutionCached.GetSizeAsync(path);

            if (res.Width > 1 && res.Height > 1)
                Logger.Log($"Input Resolution: {res.Width}x{res.Height}");

            Program.mainForm.currInRes = res;
            Program.mainForm.UpdateInputInfo();
        }

        public static async Task<Image> GetThumbnail (string path)
        {
            string imgOnDisk = Path.Combine(Paths.GetSessionDataPath(), "thumb-temp.jpg");

            try
            {
                if (!IoUtils.IsPathDirectory(path))     // If path is video - Extract first frame
                {
                    await FfmpegExtract.ExtractSingleFrame(path, imgOnDisk, 1);
                    return IoUtils.GetImage(imgOnDisk);
                }
                else     // Path is frame folder - Get first frame
                {
                    return IoUtils.GetImage(IoUtils.GetFilesSorted(path)[0]);
                }
            }
            catch (Exception e)
            {
                Logger.Log("GetThumbnail Error: " + e.Message, true);
                return null;
            }
        }

        public static float ValidateInterpFactor (float factor)
        {
            AiInfo ai = Program.mainForm.GetAi();

            if (ai.NameInternal == Implementations.rifeNcnn.NameInternal && !Program.mainForm.GetModel(ai).Dir.Contains("v4"))
            {
                if (factor != 2)
                    Logger.Log($"{ai.FriendlyName} models before 4.0 only support 2x interpolation!");

                return 2;
            }

            if (ai.FactorSupport == AiInfo.InterpFactorSupport.Fixed)
            {
                int closest = ai.SupportedFactors.Min(i => (Math.Abs(factor.RoundToInt() - i), i)).i;
                return (float)closest;
            }

            if(ai.FactorSupport == AiInfo.InterpFactorSupport.AnyPowerOfTwo)
            {
                return ToNearestPow2(factor.RoundToInt()).Clamp(2, 128);
            }

            if(ai.FactorSupport == AiInfo.InterpFactorSupport.AnyInteger)
            {
                return factor.RoundToInt().Clamp(2, 128);
            }

            if(ai.FactorSupport == AiInfo.InterpFactorSupport.AnyFloat)
            {
                return factor.Clamp(2, 128);
            }

            return factor;
        }

        static bool IsPowerOfTwo(int x)
        {
            return (x != 0) && ((x & (x - 1)) == 0);
        }

        static int ToNearestPow2(int x)
        {
            int next = ToNextNearestPow2(x);
            int prev = next >> 1;
            return next - x < x - prev ? next : prev;
        }

        static int ToNextNearestPow2(int x)
        {
            if (x < 0) { return 0; }
            --x;
            x |= x >> 1;
            x |= x >> 2;
            x |= x >> 4;
            x |= x >> 8;
            x |= x >> 16;
            return x + 1;
        }
    }
}