mirror of
https://github.com/ShareX/ShareX.git
synced 2024-08-19 20:21:33 +12:00
307 lines
No EOL
9 KiB
C#
307 lines
No EOL
9 KiB
C#
#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 <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#endregion License Information (GPL v3)
|
|
|
|
using ShareX.HelpersLib;
|
|
using ShareX.MediaLib;
|
|
using System;
|
|
using System.Diagnostics;
|
|
using System.Drawing;
|
|
using System.Threading;
|
|
|
|
namespace ShareX.ScreenCaptureLib
|
|
{
|
|
public class ScreenRecorder : IDisposable
|
|
{
|
|
public bool IsRecording { get; private set; }
|
|
|
|
public int FPS
|
|
{
|
|
get
|
|
{
|
|
return fps;
|
|
}
|
|
set
|
|
{
|
|
if (!IsRecording)
|
|
{
|
|
fps = value;
|
|
UpdateInfo();
|
|
}
|
|
}
|
|
}
|
|
|
|
public float DurationSeconds
|
|
{
|
|
get
|
|
{
|
|
return durationSeconds;
|
|
}
|
|
set
|
|
{
|
|
if (!IsRecording)
|
|
{
|
|
durationSeconds = value;
|
|
UpdateInfo();
|
|
}
|
|
}
|
|
}
|
|
|
|
public Rectangle CaptureRectangle
|
|
{
|
|
get
|
|
{
|
|
return captureRectangle;
|
|
}
|
|
private set
|
|
{
|
|
if (!IsRecording)
|
|
{
|
|
captureRectangle = value;
|
|
}
|
|
}
|
|
}
|
|
|
|
public string CachePath { get; private set; }
|
|
|
|
public ScreenRecordOutput OutputType { get; private set; }
|
|
|
|
public ScreencastOptions Options { get; set; }
|
|
|
|
public event Action RecordingStarted;
|
|
|
|
public delegate void ProgressEventHandler(int progress);
|
|
public event ProgressEventHandler EncodingProgressChanged;
|
|
|
|
private int fps, delay, frameCount, previousProgress;
|
|
private float durationSeconds;
|
|
private Screenshot screenshot;
|
|
private Rectangle captureRectangle;
|
|
private ImageCache imgCache;
|
|
private FFmpegCLIManager ffmpeg;
|
|
private bool stopRequested;
|
|
|
|
public ScreenRecorder(ScreenRecordOutput outputType, ScreencastOptions options, Screenshot screenshot, Rectangle captureRectangle)
|
|
{
|
|
if (string.IsNullOrEmpty(options.OutputPath))
|
|
{
|
|
throw new Exception("Screen recorder cache path is empty.");
|
|
}
|
|
|
|
FPS = options.FPS;
|
|
DurationSeconds = options.Duration;
|
|
CaptureRectangle = captureRectangle;
|
|
CachePath = options.OutputPath;
|
|
OutputType = outputType;
|
|
|
|
Options = options;
|
|
|
|
switch (OutputType)
|
|
{
|
|
default:
|
|
case ScreenRecordOutput.FFmpeg:
|
|
Helpers.CreateDirectoryFromFilePath(Options.OutputPath);
|
|
ffmpeg = new FFmpegCLIManager(Options.FFmpeg.FFmpegPath);
|
|
ffmpeg.ShowError = true;
|
|
ffmpeg.EncodeStarted += OnRecordingStarted;
|
|
ffmpeg.EncodeProgressChanged += OnEncodingProgressChanged;
|
|
break;
|
|
case ScreenRecordOutput.GIF:
|
|
imgCache = new HardDiskCache(Options);
|
|
break;
|
|
}
|
|
|
|
this.screenshot = screenshot;
|
|
}
|
|
|
|
private void UpdateInfo()
|
|
{
|
|
delay = 1000 / fps;
|
|
frameCount = (int)(fps * durationSeconds);
|
|
}
|
|
|
|
public void StartRecording()
|
|
{
|
|
if (!IsRecording)
|
|
{
|
|
IsRecording = true;
|
|
stopRequested = false;
|
|
|
|
if (OutputType == ScreenRecordOutput.FFmpeg)
|
|
{
|
|
ffmpeg.Run(Options.GetFFmpegCommands());
|
|
}
|
|
else
|
|
{
|
|
OnRecordingStarted();
|
|
RecordUsingCache();
|
|
}
|
|
}
|
|
|
|
IsRecording = false;
|
|
}
|
|
|
|
private void RecordUsingCache()
|
|
{
|
|
try
|
|
{
|
|
for (int i = 0; !stopRequested && (frameCount == 0 || i < frameCount); i++)
|
|
{
|
|
Stopwatch timer = Stopwatch.StartNew();
|
|
|
|
Image img = screenshot.CaptureRectangle(CaptureRectangle);
|
|
//DebugHelper.WriteLine("Screen capture: " + (int)timer.ElapsedMilliseconds);
|
|
|
|
imgCache.AddImageAsync(img);
|
|
|
|
if (!stopRequested && (frameCount == 0 || i + 1 < frameCount))
|
|
{
|
|
int sleepTime = delay - (int)timer.ElapsedMilliseconds;
|
|
|
|
if (sleepTime > 0)
|
|
{
|
|
Thread.Sleep(sleepTime);
|
|
}
|
|
else if (sleepTime < 0)
|
|
{
|
|
// Need to handle FPS drops
|
|
}
|
|
}
|
|
}
|
|
}
|
|
finally
|
|
{
|
|
imgCache.Finish();
|
|
}
|
|
}
|
|
|
|
public void StopRecording()
|
|
{
|
|
stopRequested = true;
|
|
|
|
if (ffmpeg != null)
|
|
{
|
|
ffmpeg.Close();
|
|
}
|
|
}
|
|
|
|
public void SaveAsGIF(string path, GIFQuality quality)
|
|
{
|
|
if (imgCache != null && imgCache is HardDiskCache && !IsRecording)
|
|
{
|
|
Helpers.CreateDirectoryFromFilePath(path);
|
|
|
|
HardDiskCache hdCache = imgCache as HardDiskCache;
|
|
|
|
using (AnimatedGifCreator gifEncoder = new AnimatedGifCreator(path, delay))
|
|
{
|
|
int i = 0;
|
|
int count = hdCache.Count;
|
|
|
|
foreach (Image img in hdCache.GetImageEnumerator())
|
|
{
|
|
i++;
|
|
OnEncodingProgressChanged((int)((float)i / count * 100));
|
|
|
|
using (img)
|
|
{
|
|
gifEncoder.AddFrame(img, quality);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public bool FFmpegEncodeVideo(string input, string output)
|
|
{
|
|
Helpers.CreateDirectoryFromFilePath(output);
|
|
|
|
Options.IsRecording = false;
|
|
Options.IsLossless = false;
|
|
Options.InputPath = input;
|
|
Options.OutputPath = output;
|
|
|
|
try
|
|
{
|
|
ffmpeg.TrackEncodeProgress = true;
|
|
|
|
return ffmpeg.Run(Options.GetFFmpegCommands());
|
|
}
|
|
finally
|
|
{
|
|
ffmpeg.TrackEncodeProgress = false;
|
|
}
|
|
}
|
|
|
|
public bool FFmpegEncodeAsGIF(string input, string output)
|
|
{
|
|
Helpers.CreateDirectoryFromFilePath(output);
|
|
|
|
try
|
|
{
|
|
ffmpeg.TrackEncodeProgress = true;
|
|
|
|
// https://ffmpeg.org/ffmpeg-filters.html#palettegen-1
|
|
// https://ffmpeg.org/ffmpeg-filters.html#paletteuse
|
|
return ffmpeg.Run($"-i \"{input}\" -lavfi \"palettegen=stats_mode={Options.FFmpeg.GIFStatsMode}[palette]," +
|
|
$"[0:v][palette]paletteuse=dither={Options.FFmpeg.GIFDither}" +
|
|
$"{(Options.FFmpeg.GIFDither == FFmpegPaletteUseDither.bayer ? ":bayer_scale=" + Options.FFmpeg.GIFBayerScale : "")}\"" +
|
|
$" -y \"{output}\"");
|
|
}
|
|
finally
|
|
{
|
|
ffmpeg.TrackEncodeProgress = false;
|
|
}
|
|
}
|
|
|
|
protected void OnRecordingStarted()
|
|
{
|
|
RecordingStarted?.Invoke();
|
|
}
|
|
|
|
protected void OnEncodingProgressChanged(float progress)
|
|
{
|
|
int currentProgress = (int)progress;
|
|
|
|
if (EncodingProgressChanged != null && currentProgress != previousProgress)
|
|
{
|
|
EncodingProgressChanged(currentProgress);
|
|
previousProgress = currentProgress;
|
|
}
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
if (ffmpeg != null)
|
|
{
|
|
ffmpeg.Dispose();
|
|
}
|
|
|
|
if (imgCache != null)
|
|
{
|
|
imgCache.Dispose();
|
|
}
|
|
}
|
|
}
|
|
} |