示例#1
0
        protected override GeneratorResult ProcessInternal(PreviewGeneratorSettings settings, GeneratorEntry entry)
        {
            entry.State = JobStates.Processing;

            List <string> tempFiles = new List <string>();

            try
            {
                _wrapper = new FfmpegWrapper(FfmpegExePath);

                List <string> sectionFileNames = new List <string>();

                if (settings.TimeFrames.Any(tf => tf.IsFactor))
                {
                    VideoInfo info = _wrapper.GetVideoInfo(settings.VideoFile);

                    if (!info.IsGoodEnough())
                    {
                        entry.State    = JobStates.Done;
                        entry.DoneType = JobDoneTypes.Failure;
                        entry.Update("Failed", 1);
                        return(GeneratorResult.Failed());
                    }

                    TimeSpan duration = info.Duration;

                    settings.TimeFrames.ForEach(tf => tf.CalculateStart(duration));
                }

                ClipExtractorArguments clipArguments = new ClipExtractorArguments
                {
                    InputFile = settings.VideoFile,
                    Width     = settings.Width,
                    Height    = settings.Height,
                    Framerate = settings.Framerate
                };

                for (int i = 0; i < settings.TimeFrames.Count; i++)
                {
                    string sectionFileName = Path.Combine(Path.GetTempPath(),
                                                          Path.GetFileName(settings.VideoFile) + $"-clip_{i}.mkv");
                    sectionFileNames.Add(sectionFileName);
                    tempFiles.Add(sectionFileName);

                    var timeFrame = settings.TimeFrames[i];

                    clipArguments.Duration      = timeFrame.Duration;
                    clipArguments.StartTimeSpan = timeFrame.StartTimeSpan;
                    clipArguments.OutputFile    = sectionFileName;

                    entry.Update($"Generating GIF (1/4): Clipping Video Section {i + 1}/{settings.TimeFrames.Count}",
                                 ((i / (double)settings.TimeFrames.Count)) / 4.0);

                    if (!_wrapper.Execute(clipArguments))
                    {
                        return(GeneratorResult.Failed());
                    }

                    if (_canceled)
                    {
                        return(GeneratorResult.Failed());
                    }
                }

                entry.Update("Generating GIF (2/4): Merging Clips", 1 / 4.0);

                string clipFileName = "";

                if (sectionFileNames.Count == 1)
                {
                    clipFileName = sectionFileNames[0];
                }
                else
                {
                    ClipMergeArguments mergeArguments = new ClipMergeArguments
                    {
                        InputFile  = settings.VideoFile,
                        ClipFiles  = sectionFileNames.ToArray(),
                        OutputFile = Path.Combine(Path.GetTempPath(), Path.GetFileName(settings.VideoFile) + $"-clip.mkv")
                    };

                    clipFileName = mergeArguments.OutputFile;
                    tempFiles.Add(clipFileName);

                    if (!_wrapper.Execute(mergeArguments))
                    {
                        return(GeneratorResult.Failed());
                    }

                    if (_canceled)
                    {
                        return(GeneratorResult.Failed());
                    }
                }

                entry.Update("Generating GIF (3/4): Extracting Palette", 2 / 4.0);

                string paletteFile = clipFileName + "-palette.png";

                PaletteExtractorArguments paletteArguments = new PaletteExtractorArguments
                {
                    InputFile  = clipFileName,
                    OutputFile = paletteFile
                };

                tempFiles.Add(paletteFile);

                if (!_wrapper.Execute(paletteArguments))
                {
                    return(GeneratorResult.Failed());
                }

                if (_canceled)
                {
                    return(GeneratorResult.Failed());
                }

                entry.Update("Generating GIF (3/4): Creating GIF", 3 / 4.0);

                string gifFileName = settings.OutputFile;

                GifCreatorArguments gifArguments = new GifCreatorArguments();
                gifArguments.InputFile   = clipFileName;
                gifArguments.PaletteFile = paletteFile;
                gifArguments.OutputFile  = gifFileName;
                gifArguments.Framerate   = settings.Framerate;

                if (!_wrapper.Execute(gifArguments))
                {
                    return(GeneratorResult.Failed());
                }

                if (_canceled)
                {
                    return(GeneratorResult.Failed());
                }

                entry.Update("Done!", 4 / 4.0);
                bool success = File.Exists(gifFileName);

                if (success)
                {
                    entry.DoneType = JobDoneTypes.Success;
                }
                else
                {
                    entry.DoneType = JobDoneTypes.Failure;
                }

                OnDone(new Tuple <bool, string>(success, gifFileName));

                if (success)
                {
                    return(GeneratorResult.Succeeded(gifFileName));
                }

                return(GeneratorResult.Failed());
            }
            catch
            {
                entry.DoneType = JobDoneTypes.Failure;
                return(GeneratorResult.Failed());
            }
            finally
            {
                entry.State = JobStates.Done;

                if (_canceled)
                {
                    entry.DoneType = JobDoneTypes.Cancelled;
                }

                foreach (string tempFile in tempFiles)
                {
                    if (File.Exists(tempFile))
                    {
                        File.Delete(tempFile);
                    }
                }
            }
        }
示例#2
0
        protected override GeneratorResult ProcessInternal(PreviewGeneratorSettings settings, GeneratorEntry entry)
        {
            entry.State = JobStates.Processing;

            List <string> tempFiles = new List <string>();

            int totalSteps = settings.OverlayScriptPositions ? 5 : 4;
            int stepsDone  = 0;

            try
            {
                _wrapper = new FfmpegWrapper(FfmpegExePath);

                List <string> sectionFileNames = new List <string>();

                if (settings.TimeFrames.Any(tf => tf.IsFactor))
                {
                    VideoInfo info = _wrapper.GetVideoInfo(settings.VideoFile);

                    if (!info.IsGoodEnough())
                    {
                        entry.State    = JobStates.Done;
                        entry.DoneType = JobDoneTypes.Failure;
                        entry.Update("Failed", 1);
                        return(GeneratorResult.Failed());
                    }

                    TimeSpan duration = info.Duration;

                    settings.TimeFrames.ForEach(tf => tf.CalculateStart(duration));
                }

                ClipExtractorArguments clipArguments = new ClipExtractorArguments
                {
                    InputFile = settings.VideoFile,
                    Width     = settings.Width,
                    Height    = settings.Height,
                    Framerate = settings.Framerate,
                    ClipLeft  = settings.ClipLeft,
                    DeLense   = settings.ClipLeft
                };

                VideoInfo clipInfo = null;

                for (int i = 0; i < settings.TimeFrames.Count; i++)
                {
                    string sectionFileName = Path.Combine(Path.GetTempPath(),
                                                          Path.GetFileName(settings.VideoFile) + $"-clip_{i}.mkv");
                    sectionFileNames.Add(sectionFileName);
                    tempFiles.Add(sectionFileName);

                    var timeFrame = settings.TimeFrames[i];

                    clipArguments.Duration      = timeFrame.Duration;
                    clipArguments.StartTimeSpan = timeFrame.StartTimeSpan;
                    clipArguments.OutputFile    = sectionFileName;

                    entry.Update($"Generating GIF ({stepsDone + 1}/{totalSteps}): Clipping Video Section {i + 1}/{settings.TimeFrames.Count}",
                                 (stepsDone / (double)totalSteps) + ((i / (double)settings.TimeFrames.Count)) / totalSteps);

                    if (!_wrapper.Execute(clipArguments))
                    {
                        return(GeneratorResult.Failed());
                    }

                    if (clipInfo == null)
                    {
                        clipInfo = _wrapper.GetVideoInfo(sectionFileName);
                    }

                    if (_canceled)
                    {
                        return(GeneratorResult.Failed());
                    }
                }

                stepsDone++;

                if (settings.OverlayScriptPositions)
                {
                    string script  = ViewModel.GetScriptFile(settings.VideoFile);
                    var    actions = ViewModel.LoadScriptActions(script, null)?.OfType <FunScriptAction>().ToList();

                    Size barSize = new Size(clipInfo.Resolution.Horizontal, 20);

                    if (actions != null && actions.Count > 0)
                    {
                        PositionBar bar = new PositionBar
                        {
                            Width  = barSize.Width,
                            Height = barSize.Height,
                            TotalDisplayedDuration = TimeSpan.FromSeconds(5),
                            Background             = Brushes.Black,
                            Positions = new PositionCollection(actions.Select(a => new TimedPosition()
                            {
                                Position  = a.Position,
                                TimeStamp = a.TimeStamp
                            })),
                            DrawCircles = false,
                            DrawLines   = false,
                        };

                        for (int i = 0; i < settings.TimeFrames.Count; i++)
                        {
                            TimeSpan start = settings.TimeFrames[i].StartTimeSpan;
                            TimeSpan max   = start + settings.TimeFrames[i].Duration;

                            TimeSpan progress = start;

                            string tempDir = Path.Combine(Path.GetTempPath(), Guid.NewGuid().ToString("N"));
                            Directory.CreateDirectory(tempDir);

                            string overlayFileName = Path.Combine(Path.GetTempPath(),
                                                                  Path.GetFileName(settings.VideoFile) + $"-overlay_{i}.mkv");

                            tempFiles.Add(overlayFileName);

                            int frame = 0;

                            entry.Update($"Generating GIF ({stepsDone + 1}/{totalSteps}): Overlaying Positions {i + 1}/{settings.TimeFrames.Count}",
                                         (stepsDone / (double)totalSteps) + ((i / (double)settings.TimeFrames.Count)) / totalSteps);

                            double expectedFrames = 1 + (settings.TimeFrames[i].Duration.TotalSeconds * clipInfo.FrameRate);

                            while (progress <= max)
                            {
                                progress = start + TimeSpan.FromSeconds(frame / clipInfo.FrameRate);

                                bar.Progress = progress;

                                var bitmap = RenderToBitmap(bar, barSize);

                                PngBitmapEncoder encoder = new PngBitmapEncoder();

                                encoder.Frames.Add(BitmapFrame.Create(bitmap));

                                using (FileStream f = new FileStream(Path.Combine(tempDir, $"frame{frame:0000}.png"),
                                                                     FileMode.CreateNew))
                                    encoder.Save(f);

                                frame++;
                            }

                            FrameMergeArguments frameMergeArguments = new FrameMergeArguments
                            {
                                Framerate  = clipInfo.FrameRate,
                                InputFile  = Path.Combine(tempDir, "frame%04d.png"),
                                OutputFile = overlayFileName
                            };

                            if (!_wrapper.Execute(frameMergeArguments))
                            {
                                return(GeneratorResult.Failed());
                            }

                            var clipInfo2 = _wrapper.GetVideoInfo(overlayFileName);

                            Directory.Delete(tempDir, true);

                            string mergeFileName = Path.Combine(Path.GetTempPath(),
                                                                Path.GetFileName(settings.VideoFile) + $"-merge_{i}.mkv");

                            tempFiles.Add(mergeFileName);

                            VideoOverlayArguments overlayArguments = new VideoOverlayArguments
                            {
                                InputFile  = sectionFileNames[i],
                                Overlay    = overlayFileName,
                                OutputFile = mergeFileName,
                                PosX       = 0,
                                PosY       = clipInfo.Resolution.Vertical - 20
                            };

                            if (!_wrapper.Execute(overlayArguments))
                            {
                                return(GeneratorResult.Failed());
                            }

                            sectionFileNames[i] = mergeFileName;
                        }
                    }

                    stepsDone++;
                }

                entry.Update($"Generating GIF ({stepsDone + 1}/{totalSteps}): Merging Clips", stepsDone / (double)totalSteps);

                string clipFileName;

                if (sectionFileNames.Count == 1)
                {
                    clipFileName = sectionFileNames[0];
                }
                else
                {
                    ClipMergeArguments mergeArguments = new ClipMergeArguments
                    {
                        InputFile  = settings.VideoFile,
                        ClipFiles  = sectionFileNames.ToArray(),
                        OutputFile = Path.Combine(Path.GetTempPath(), Path.GetFileName(settings.VideoFile) + $"-clip.mkv")
                    };

                    clipFileName = mergeArguments.OutputFile;
                    tempFiles.Add(clipFileName);

                    if (!_wrapper.Execute(mergeArguments))
                    {
                        return(GeneratorResult.Failed());
                    }

                    if (_canceled)
                    {
                        return(GeneratorResult.Failed());
                    }
                }

                stepsDone++;

                entry.Update($"Generating GIF ({stepsDone + 1}/{totalSteps}): Extracting Palette", stepsDone / (double)totalSteps);

                string paletteFile = clipFileName + "-palette.png";

                PaletteExtractorArguments paletteArguments = new PaletteExtractorArguments
                {
                    InputFile  = clipFileName,
                    OutputFile = paletteFile
                };

                tempFiles.Add(paletteFile);

                if (!_wrapper.Execute(paletteArguments))
                {
                    return(GeneratorResult.Failed());
                }

                if (_canceled)
                {
                    return(GeneratorResult.Failed());
                }

                stepsDone++;

                entry.Update($"Generating GIF ({stepsDone + 1}/{totalSteps}): Creating GIF", stepsDone / (double)totalSteps);

                string gifFileName = settings.OutputFile;

                GifCreatorArguments gifArguments = new GifCreatorArguments();
                gifArguments.InputFile   = clipFileName;
                gifArguments.PaletteFile = paletteFile;
                gifArguments.OutputFile  = gifFileName;
                gifArguments.Framerate   = settings.Framerate;

                if (!_wrapper.Execute(gifArguments))
                {
                    return(GeneratorResult.Failed());
                }

                if (_canceled)
                {
                    return(GeneratorResult.Failed());
                }

                stepsDone++;

                entry.Update("Done", 1.0);
                bool success = File.Exists(gifFileName);

                if (success)
                {
                    entry.DoneType = JobDoneTypes.Success;
                }
                else
                {
                    entry.DoneType = JobDoneTypes.Failure;
                }

                OnDone(new Tuple <bool, string>(success, gifFileName));

                if (success)
                {
                    return(GeneratorResult.Succeeded(gifFileName));
                }

                return(GeneratorResult.Failed());
            }
            catch (Exception e)
            {
                entry.DoneType = JobDoneTypes.Failure;
                return(GeneratorResult.Failed());
            }
            finally
            {
                entry.State = JobStates.Done;

                if (_canceled)
                {
                    entry.DoneType = JobDoneTypes.Cancelled;
                }

                foreach (string tempFile in tempFiles)
                {
                    if (File.Exists(tempFile))
                    {
                        File.Delete(tempFile);
                    }
                }
            }
        }