#region License Information (GPL v3)
/*
ShareX - A program that allows you to take screenshots and share any file type
Copyright (c) 2007-2020 ShareX Team
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
Optionally you can also view the license at .
*/
#endregion License Information (GPL v3)
using ShareX.HelpersLib;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Drawing;
using System.IO;
namespace ShareX.MediaLib
{
public class VideoThumbnailer
{
public delegate void ProgressChangedEventHandler(int current, int length);
public event ProgressChangedEventHandler ProgressChanged;
public string FFmpegPath { get; private set; }
public VideoThumbnailOptions Options { get; private set; }
public string MediaPath { get; private set; }
public VideoInfo VideoInfo { get; private set; }
public VideoThumbnailer(string ffmpegPath, VideoThumbnailOptions options)
{
FFmpegPath = ffmpegPath;
Options = options;
}
private void UpdateVideoInfo()
{
using (FFmpegCLIManager ffmpeg = new FFmpegCLIManager(FFmpegPath))
{
VideoInfo = ffmpeg.GetVideoInfo(MediaPath);
}
}
public List TakeThumbnails(string mediaPath)
{
MediaPath = mediaPath;
UpdateVideoInfo();
if (VideoInfo == null || VideoInfo.Duration == TimeSpan.Zero)
{
return null;
}
List tempThumbnails = new List();
for (int i = 0; i < Options.ThumbnailCount; i++)
{
string mediaFileName = Path.GetFileNameWithoutExtension(MediaPath);
int timeSliceElapsed;
if (Options.RandomFrame)
{
timeSliceElapsed = GetRandomTimeSlice(i);
}
else
{
timeSliceElapsed = GetTimeSlice(Options.ThumbnailCount) * (i + 1);
}
string filename = string.Format("{0}-{1}.{2}", mediaFileName, timeSliceElapsed, Options.ImageFormat.GetDescription());
string tempThumbnailPath = Path.Combine(GetOutputDirectory(), filename);
using (Process process = new Process())
{
ProcessStartInfo psi = new ProcessStartInfo()
{
FileName = FFmpegPath,
Arguments = $"-ss {timeSliceElapsed} -i \"{MediaPath}\" -f image2 -vframes 1 -y \"{tempThumbnailPath}\"",
UseShellExecute = false,
CreateNoWindow = true
};
process.StartInfo = psi;
process.Start();
process.WaitForExit(1000 * 30);
}
if (File.Exists(tempThumbnailPath))
{
VideoThumbnailInfo screenshotInfo = new VideoThumbnailInfo(tempThumbnailPath)
{
Timestamp = TimeSpan.FromSeconds(timeSliceElapsed)
};
tempThumbnails.Add(screenshotInfo);
}
OnProgressChanged(i + 1, Options.ThumbnailCount);
}
return Finish(tempThumbnails);
}
private List Finish(List tempThumbnails)
{
List thumbnails = new List();
if (tempThumbnails != null && tempThumbnails.Count > 0)
{
if (Options.CombineScreenshots)
{
using (Image img = CombineScreenshots(tempThumbnails))
{
string tempFilepath = Path.Combine(GetOutputDirectory(), Path.GetFileNameWithoutExtension(MediaPath) + Options.FilenameSuffix + "." + Options.ImageFormat.GetDescription());
ImageHelpers.SaveImage(img, tempFilepath);
thumbnails.Add(new VideoThumbnailInfo(tempFilepath));
}
if (!Options.KeepScreenshots)
{
tempThumbnails.ForEach(x => File.Delete(x.Filepath));
}
}
else
{
thumbnails.AddRange(tempThumbnails);
}
if (Options.OpenDirectory && thumbnails.Count > 0)
{
Helpers.OpenFolderWithFile(thumbnails[0].Filepath);
}
}
return thumbnails;
}
protected void OnProgressChanged(int current, int length)
{
ProgressChanged?.Invoke(current, length);
}
private string GetOutputDirectory()
{
string directory;
switch (Options.OutputLocation)
{
default:
case ThumbnailLocationType.DefaultFolder:
directory = Options.DefaultOutputDirectory;
break;
case ThumbnailLocationType.ParentFolder:
directory = Path.GetDirectoryName(MediaPath);
break;
case ThumbnailLocationType.CustomFolder:
directory = Helpers.ExpandFolderVariables(Options.CustomOutputDirectory);
break;
}
Helpers.CreateDirectory(directory);
return directory;
}
private int GetTimeSlice(int count)
{
return (int)(VideoInfo.Duration.TotalSeconds / count);
}
private int GetRandomTimeSlice(int start)
{
List mediaSeekTimes = new List();
for (int i = 1; i < Options.ThumbnailCount + 2; i++)
{
mediaSeekTimes.Add(GetTimeSlice(Options.ThumbnailCount + 2) * i);
}
return (int)((RandomFast.NextDouble() * (mediaSeekTimes[start + 1] - mediaSeekTimes[start])) + mediaSeekTimes[start]);
}
private Image CombineScreenshots(List thumbnails)
{
List images = new List();
Image finalImage = null;
try
{
string infoString = "";
int infoStringHeight = 0;
if (Options.AddVideoInfo)
{
infoString = VideoInfo.ToString();
using (Font font = new Font("Arial", 12))
{
infoStringHeight = Helpers.MeasureText(infoString, font).Height;
}
}
foreach (VideoThumbnailInfo thumbnail in thumbnails)
{
Bitmap bmp = ImageHelpers.LoadImage(thumbnail.Filepath);
if (Options.MaxThumbnailWidth > 0 && bmp.Width > Options.MaxThumbnailWidth)
{
int maxThumbnailHeight = (int)((float)Options.MaxThumbnailWidth / bmp.Width * bmp.Height);
bmp = ImageHelpers.ResizeImage(bmp, Options.MaxThumbnailWidth, maxThumbnailHeight);
}
images.Add(bmp);
}
int columnCount = Options.ColumnCount;
int thumbWidth = images[0].Width;
int width = (Options.Padding * 2) +
(thumbWidth * columnCount) +
((columnCount - 1) * Options.Spacing);
int rowCount = (int)Math.Ceiling(images.Count / (float)columnCount);
int thumbHeight = images[0].Height;
int height = (Options.Padding * 3) +
infoStringHeight +
(thumbHeight * rowCount) +
((rowCount - 1) * Options.Spacing);
finalImage = new Bitmap(width, height);
using (Graphics g = Graphics.FromImage(finalImage))
{
g.Clear(Color.WhiteSmoke);
if (!string.IsNullOrEmpty(infoString))
{
using (Font font = new Font("Arial", 12))
{
g.DrawString(infoString, font, Brushes.Black, Options.Padding, Options.Padding);
}
}
int i = 0;
int offsetY = (Options.Padding * 2) + infoStringHeight;
for (int y = 0; y < rowCount; y++)
{
int offsetX = Options.Padding;
for (int x = 0; x < columnCount; x++)
{
if (Options.DrawShadow)
{
int shadowOffset = 3;
using (Brush shadowBrush = new SolidBrush(Color.FromArgb(75, Color.Black)))
{
g.FillRectangle(shadowBrush, offsetX + shadowOffset, offsetY + shadowOffset, thumbWidth, thumbHeight);
}
}
g.DrawImage(images[i], offsetX, offsetY, thumbWidth, thumbHeight);
if (Options.DrawBorder)
{
g.DrawRectangleProper(Pens.Black, offsetX, offsetY, thumbWidth, thumbHeight);
}
if (Options.AddTimestamp)
{
int timestampOffset = 10;
using (Font font = new Font("Arial", 10, FontStyle.Bold))
{
g.DrawTextWithShadow(thumbnails[i].Timestamp.ToString(), new Point(offsetX + timestampOffset, offsetY + timestampOffset), font, Brushes.White, Brushes.Black);
}
}
i++;
if (i >= images.Count)
{
return finalImage;
}
offsetX += thumbWidth + Options.Spacing;
}
offsetY += thumbHeight + Options.Spacing;
}
}
return finalImage;
}
catch
{
if (finalImage != null)
{
finalImage.Dispose();
}
throw;
}
finally
{
foreach (Bitmap image in images)
{
if (image != null)
{
image.Dispose();
}
}
}
}
}
}