FFMpegCore/FFMpegCore.Test/ArgumentBuilderTest.cs

using Microsoft.VisualStudio.TestTools.UnitTesting;
using System;
using FFMpegCore.Arguments;
using FFMpegCore.Enums;

namespace FFMpegCore.Test
{
    [TestClass]
    public class ArgumentBuilderTest
    {
        private readonly string[] _concatFiles = { "1.mp4", "2.mp4", "3.mp4", "4.mp4" };


        [TestMethod]
        public void Builder_BuildString_IO_1()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4").OutputToFile("output.mp4").Arguments;
            Assert.AreEqual("-i \"input.mp4\" \"output.mp4\" -y", str);
        }

        [TestMethod]
        public void Builder_BuildString_Scale()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", true, opt => opt
                    .WithVideoFilters(filterOptions => filterOptions
                        .Scale(VideoSize.Hd)))
                .Arguments;
            Assert.AreEqual("-i \"input.mp4\" -vf \"scale=-1:720\" \"output.mp4\" -y", str);
        }

        [TestMethod]
        public void Builder_BuildString_AudioCodec()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", true, opt => opt.WithAudioCodec(AudioCodec.Aac)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -c:a aac \"output.mp4\" -y", str);
        }

        [TestMethod]
        public void Builder_BuildString_AudioBitrate()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", true, opt => opt.WithAudioBitrate(AudioQuality.Normal)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -b:a 128k \"output.mp4\" -y", str);
        }

        [TestMethod]
        public void Builder_BuildString_Quiet()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4").WithGlobalOptions(opt => opt.WithVerbosityLevel())
                .OutputToFile("output.mp4", false).Arguments;
            Assert.AreEqual("-hide_banner -loglevel error -i \"input.mp4\" \"output.mp4\"", str);
        }


        [TestMethod]
        public void Builder_BuildString_AudioCodec_Fluent()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4").OutputToFile("output.mp4", false,
                opt => opt.WithAudioCodec(AudioCodec.Aac).WithAudioBitrate(128)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -c:a aac -b:a 128k \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_BitStream()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4").OutputToFile("output.mp4", false,
                opt => opt.WithBitStreamFilter(Channel.Audio, Filter.H264_Mp4ToAnnexB)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -bsf:a h264_mp4toannexb \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_HardwareAcceleration_Auto()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.WithHardwareAcceleration()).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -hwaccel \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_HardwareAcceleration_Specific()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4").OutputToFile("output.mp4", false,
                opt => opt.WithHardwareAcceleration(HardwareAccelerationDevice.CUVID)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -hwaccel cuvid \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Concat()
        {
            var str = FFMpegArguments.FromConcatInput(_concatFiles).OutputToFile("output.mp4", false).Arguments;
            Assert.AreEqual("-i \"concat:1.mp4|2.mp4|3.mp4|4.mp4\" \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Copy_Audio()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.CopyChannel(Channel.Audio)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -c:a copy \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Copy_Video()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.CopyChannel(Channel.Video)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -c:v copy \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Copy_Both()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.CopyChannel()).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -c copy \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_DisableChannel_Audio()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.DisableChannel(Channel.Audio)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -an \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_DisableChannel_Video()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.DisableChannel(Channel.Video)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -vn \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_AudioSamplingRate_Default()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.WithAudioSamplingRate()).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -ar 48000 \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_AudioSamplingRate()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.WithAudioSamplingRate(44000)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -ar 44000 \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_VariableBitrate()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.WithVariableBitrate(5)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -vbr 5 \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Faststart()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.WithFastStart()).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -movflags faststart \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Overwrite()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.OverwriteExisting()).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -y \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_RemoveMetadata()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.WithoutMetadata()).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -map_metadata -1 \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Transpose()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt
                    .WithVideoFilters(filterOptions => filterOptions
                        .Transpose(Transposition.CounterClockwise90)))
                .Arguments;
            Assert.AreEqual("-i \"input.mp4\" -vf \"transpose=2\" \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_TransposeScale()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt
                    .WithVideoFilters(filterOptions => filterOptions
                        .Transpose(Transposition.CounterClockwise90)
                        .Scale(200, 300)))
                .Arguments;
            Assert.AreEqual("-i \"input.mp4\" -vf \"transpose=2, scale=200:300\" \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_ForceFormat()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4", false, opt => opt.ForceFormat(VideoType.Mp4))
                .OutputToFile("output.mp4", false, opt => opt.ForceFormat(VideoType.Mp4)).Arguments;
            Assert.AreEqual("-f mp4 -i \"input.mp4\" -f mp4 \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_FrameOutputCount()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.WithFrameOutputCount(50)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -vframes 50 \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_FrameRate()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.WithFramerate(50)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -r 50 \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Loop()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4").OutputToFile("output.mp4", false, opt => opt.Loop(50))
                .Arguments;
            Assert.AreEqual("-i \"input.mp4\" -loop 50 \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Seek()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4", false, opt => opt.Seek(TimeSpan.FromSeconds(10)))
                .OutputToFile("output.mp4", false, opt => opt.Seek(TimeSpan.FromSeconds(10))).Arguments;
            Assert.AreEqual("-ss 00:00:10 -i \"input.mp4\" -ss 00:00:10 \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Shortest()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.UsingShortest()).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -shortest \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Size()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.Resize(1920, 1080)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -s 1920x1080 \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Speed()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.WithSpeedPreset(Speed.Fast)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -preset fast \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_DrawtextFilter()
        {
            var str = FFMpegArguments
                .FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt
                    .WithVideoFilters(filterOptions => filterOptions
                        .DrawText(DrawTextOptions
                            .Create("Stack Overflow", "/path/to/font.ttf")
                            .WithParameter("fontcolor", "white")
                            .WithParameter("fontsize", "24")
                            .WithParameter("box", "1")
                            .WithParameter("boxcolor", "black@0.5")
                            .WithParameter("boxborderw", "5")
                            .WithParameter("x", "(w-text_w)/2")
                            .WithParameter("y", "(h-text_h)/2"))))
                .Arguments;

            Assert.AreEqual(
                "-i \"input.mp4\" -vf \"drawtext=text='Stack Overflow':fontfile=/path/to/font.ttf:fontcolor=white:fontsize=24:box=1:boxcolor=black@0.5:boxborderw=5:x=(w-text_w)/2:y=(h-text_h)/2\" \"output.mp4\"",
                str);
        }

        [TestMethod]
        public void Builder_BuildString_DrawtextFilter_Alt()
        {
            var str = FFMpegArguments
                .FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt
                    .WithVideoFilters(filterOptions => filterOptions
                        .DrawText(DrawTextOptions
                            .Create("Stack Overflow", "/path/to/font.ttf", ("fontcolor", "white"), ("fontsize", "24")))))
                .Arguments;

            Assert.AreEqual(
                "-i \"input.mp4\" -vf \"drawtext=text='Stack Overflow':fontfile=/path/to/font.ttf:fontcolor=white:fontsize=24\" \"output.mp4\"",
                str);
        }

        [TestMethod]
        public void Builder_BuildString_StartNumber()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.WithStartNumber(50)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -start_number 50 \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Threads_1()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.UsingThreads(50)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -threads 50 \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Threads_2()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.UsingMultithreading(true)).Arguments;
            Assert.AreEqual($"-i \"input.mp4\" -threads {Environment.ProcessorCount} \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Codec()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.WithVideoCodec(VideoCodec.LibX264)).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -c:v libx264 \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Codec_Override()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4").OutputToFile("output.mp4", true,
                opt => opt.WithVideoCodec(VideoCodec.LibX264).ForcePixelFormat("yuv420p")).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -c:v libx264 -pix_fmt yuv420p \"output.mp4\" -y", str);
        }


        [TestMethod]
        public void Builder_BuildString_Duration()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.WithDuration(TimeSpan.FromSeconds(20))).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -t 00:00:20 \"output.mp4\"", str);
        }

        [TestMethod]
        public void Builder_BuildString_Raw()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4", false, opt => opt.WithCustomArgument(null!))
                .OutputToFile("output.mp4", false, opt => opt.WithCustomArgument(null!)).Arguments;
            Assert.AreEqual(" -i \"input.mp4\"  \"output.mp4\"", str);

            str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.WithCustomArgument("-acodec copy")).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -acodec copy \"output.mp4\"", str);
        }


        [TestMethod]
        public void Builder_BuildString_ForcePixelFormat()
        {
            var str = FFMpegArguments.FromFileInput("input.mp4")
                .OutputToFile("output.mp4", false, opt => opt.ForcePixelFormat("yuv444p")).Arguments;
            Assert.AreEqual("-i \"input.mp4\" -pix_fmt yuv444p \"output.mp4\"", str);
        }
    }
}