mirror of
https://github.com/ShareX/ShareX.git
synced 2024-10-06 05:05:42 +13:00
273 lines
No EOL
8 KiB
C#
273 lines
No EOL
8 KiB
C#
#region License Information (GPL v3)
|
|
|
|
/*
|
|
ShareX - A program that allows you to take screenshots and share any file type
|
|
Copyright (c) 2007-2019 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 System;
|
|
using System.Diagnostics;
|
|
using System.Drawing;
|
|
using System.IO;
|
|
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 FFmpegHelper ffmpegCli;
|
|
private bool stopRequest;
|
|
|
|
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:
|
|
ffmpegCli = new FFmpegHelper(Options);
|
|
ffmpegCli.RecordingStarted += OnRecordingStarted;
|
|
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;
|
|
stopRequest = false;
|
|
|
|
if (OutputType == ScreenRecordOutput.FFmpeg)
|
|
{
|
|
ffmpegCli.Record();
|
|
}
|
|
else
|
|
{
|
|
OnRecordingStarted();
|
|
RecordUsingCache();
|
|
}
|
|
}
|
|
|
|
IsRecording = false;
|
|
}
|
|
|
|
private void RecordUsingCache()
|
|
{
|
|
try
|
|
{
|
|
for (int i = 0; !stopRequest && (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 (!stopRequest && (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()
|
|
{
|
|
stopRequest = true;
|
|
|
|
if (ffmpegCli != null)
|
|
{
|
|
ffmpegCli.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);
|
|
bool result = ffmpegCli.EncodeVideo(input, output);
|
|
//DebugHelper.WriteLine("Video encoding result:\nInput file size: {0}\nOutput file size: {1}", new FileInfo(input).Length.ToSizeString(), new FileInfo(output).Length.ToSizeString());
|
|
return result;
|
|
}
|
|
|
|
public bool FFmpegEncodeAsGIF(string sourceFilePath, string targetFilePath, string tempFolder)
|
|
{
|
|
Helpers.CreateDirectoryFromFilePath(targetFilePath);
|
|
Helpers.CreateDirectoryFromDirectoryPath(tempFolder);
|
|
return ffmpegCli.EncodeGIF(sourceFilePath, targetFilePath, tempFolder);
|
|
}
|
|
|
|
protected void OnRecordingStarted()
|
|
{
|
|
if (RecordingStarted != null)
|
|
{
|
|
RecordingStarted();
|
|
}
|
|
}
|
|
|
|
protected void OnEncodingProgressChanged(int progress)
|
|
{
|
|
if (EncodingProgressChanged != null && progress != previousProgress)
|
|
{
|
|
EncodingProgressChanged(progress);
|
|
previousProgress = progress;
|
|
}
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
if (imgCache != null)
|
|
{
|
|
imgCache.Dispose();
|
|
}
|
|
}
|
|
}
|
|
} |