public List <Tuple <TimeSpan, BitmapSource> > ExtractFrames(FrameConverterArguments arguments)
{
bool isTempDir = false;
if (string.IsNullOrWhiteSpace(arguments.OutputDirectory))
{
isTempDir = true;
arguments.OutputDirectory = CreateRandomTempDirectory();
}
FrameConverterWrapper wrapper = new FrameConverterWrapper(arguments, FfmpegExePath);
ExecuteWrapper(wrapper);
List <Tuple <TimeSpan, BitmapSource> > result = new List <Tuple <TimeSpan, BitmapSource> >();
List <string> usedFiles = new List <string>();
foreach (string file in Directory.EnumerateFiles(arguments.OutputDirectory))
{
string number = Path.GetFileNameWithoutExtension(file);
int index = int.Parse(number);
TimeSpan position = TimeSpan.FromSeconds((index - 0.5) * arguments.Intervall);
var frame = new BitmapImage();
frame.BeginInit();
frame.CacheOption = BitmapCacheOption.OnLoad;
frame.UriSource = new Uri(file, UriKind.Absolute);
frame.EndInit();
result.Add(new Tuple <TimeSpan, BitmapSource>(position, frame));
usedFiles.Add(file);
}
foreach (string tempFile in usedFiles)
{
File.Delete(tempFile);
}
if (isTempDir)
{
Directory.Delete(arguments.OutputDirectory);
}
return(result);
}
protected override GeneratorResult ProcessInternal(ThumbnailBannerGeneratorSettings settings, GeneratorEntry entry)
{
try
{
entry.State = JobStates.Processing;
_wrapper = new FfmpegWrapper(FfmpegExePath);
var videoInfo = _wrapper.GetVideoInfo(settings.VideoFile);
if (!videoInfo.IsGoodEnough())
{
entry.State = JobStates.Done;
entry.DoneType = JobDoneTypes.Failure;
entry.Update("Failed", 1);
return(GeneratorResult.Failed());
}
TimeSpan duration = videoInfo.Duration;
TimeSpan intervall = duration.Divide(settings.Columns * settings.Rows + 1);
var frameArguments = new FrameConverterArguments
{
Width = 800,
Intervall = intervall.TotalSeconds,
StatusUpdateHandler = (progress) => { entry.Update(null, progress); },
InputFile = settings.VideoFile,
OutputDirectory = FfmpegWrapper.CreateRandomTempDirectory()
};
entry.Update("Extracting Frames", 0);
var frames = _wrapper.ExtractFrames(frameArguments);
if (_canceled)
{
return(GeneratorResult.Failed());
}
entry.Update("Saving Thumbnails", 1);
List <ThumbnailBannerGeneratorImage> images = new List <ThumbnailBannerGeneratorImage>();
foreach (var frame in frames)
{
images.Add(new ThumbnailBannerGeneratorImage
{
Image = frame.Item2,
Position = frame.Item1
});
}
ThumbnailBannerGeneratorData data = new ThumbnailBannerGeneratorData();
data.Settings = settings;
data.Images = images.ToArray();
data.VideoName = settings.VideoFile;
data.VideoInfo = videoInfo;
data.FileSize = new FileInfo(settings.VideoFile).Length;
var result = CreateBanner(data);
JpegBitmapEncoder encoder = new JpegBitmapEncoder();
encoder.Frames.Add(BitmapFrame.Create((BitmapSource)result));
using (FileStream stream = new FileStream(settings.OutputFile, FileMode.Create))
encoder.Save(stream);
Directory.Delete(frameArguments.OutputDirectory);
entry.DoneType = JobDoneTypes.Success;
entry.Update("Done", 1);
return(GeneratorResult.Succeeded(settings.OutputFile));
}
catch (Exception)
{
entry.Update("Failed", 1);
entry.DoneType = JobDoneTypes.Failure;
return(GeneratorResult.Failed());
}
finally
{
entry.State = JobStates.Done;
if (_canceled)
{
entry.DoneType = JobDoneTypes.Cancelled;
entry.Update("Cancelled", 1);
}
}
}
protected override GeneratorResult ProcessInternal(ThumbnailGeneratorSettings settings, GeneratorEntry entry)
{
try
{
entry.State = JobStates.Processing;
_wrapper = new FfmpegWrapper(FfmpegExePath);
double intervall = settings.Intervall;
if (settings.Intervall < 1)
{
var info = _wrapper.GetVideoInfo(settings.VideoFile);
if (!info.IsGoodEnough())
{
entry.Update("Failed", 1);
entry.DoneType = JobDoneTypes.Failure;
return(GeneratorResult.Failed());
}
const double targetFrameCount = 500.0;
intervall = info.Duration.TotalSeconds / targetFrameCount;
intervall = Math.Min(Math.Max(1, intervall), 10);
}
FrameConverterArguments arguments = new FrameConverterArguments
{
StatusUpdateHandler = (progress) => { entry.Update(null, progress); },
InputFile = settings.VideoFile,
OutputDirectory = FfmpegWrapper.CreateRandomTempDirectory(),
Intervall = intervall,
Width = settings.Width,
Height = settings.Height,
ClipLeft = settings.ClipLeft,
DeLense = settings.ClipLeft
};
string thumbfile = Path.ChangeExtension(settings.VideoFile, "thumbs");
entry.Update("Extracting Frames", 0);
var frames = _wrapper.ExtractFrames(arguments);
if (_canceled)
{
return(GeneratorResult.Failed());
}
entry.Update("Saving Thumbnails", 1);
VideoThumbnailCollection thumbnails = new VideoThumbnailCollection();
foreach (var frame in frames)
{
thumbnails.Add(frame.Item1, frame.Item2);
}
using (FileStream stream = new FileStream(thumbfile, FileMode.Create, FileAccess.Write))
{
thumbnails.Save(stream);
}
thumbnails.Dispose();
entry.DoneType = JobDoneTypes.Success;
entry.Update("Done", 1);
return(GeneratorResult.Succeeded(thumbfile));
}
catch (Exception)
{
entry.DoneType = JobDoneTypes.Failure;
return(GeneratorResult.Failed());
}
finally
{
entry.State = JobStates.Done;
if (_canceled)
{
entry.DoneType = JobDoneTypes.Cancelled;
}
}
}
