mirror of
https://github.com/ShareX/ShareX.git
synced 2024-07-04 14:11:38 +12:00
139 lines
4.2 KiB
C#
139 lines
4.2 KiB
C#
|
#region License Information (GPL v3)
|
|||
|
|
|||
|
/*
|
|||
|
ShareX - A program that allows you to take screenshots and share any file type
|
|||
|
Copyright (C) 2008-2013 ShareX Developers
|
|||
|
|
|||
|
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 HelpersLib;
|
|||
|
using System;
|
|||
|
using System.Collections.Concurrent;
|
|||
|
using System.Drawing;
|
|||
|
using System.Threading.Tasks;
|
|||
|
|
|||
|
namespace ScreenCapture
|
|||
|
{
|
|||
|
public class AVICache : IDisposable
|
|||
|
{
|
|||
|
public bool IsWorking { get; private set; }
|
|||
|
public string OutputPath { get; private set; }
|
|||
|
public int FPS { get; private set; }
|
|||
|
public Size Size { get; private set; }
|
|||
|
public bool ShowOptions { get; private set; }
|
|||
|
|
|||
|
private Task task;
|
|||
|
private BlockingCollection<Image> imageQueue;
|
|||
|
private int position;
|
|||
|
|
|||
|
public AVICache(string outputPath, int fps, Size size, bool showOptions = false)
|
|||
|
{
|
|||
|
OutputPath = outputPath;
|
|||
|
FPS = fps;
|
|||
|
Size = size;
|
|||
|
ShowOptions = showOptions;
|
|||
|
imageQueue = new BlockingCollection<Image>();
|
|||
|
StartConsumerThread();
|
|||
|
}
|
|||
|
|
|||
|
private void StartConsumerThread()
|
|||
|
{
|
|||
|
if (!IsWorking)
|
|||
|
{
|
|||
|
IsWorking = true;
|
|||
|
|
|||
|
Helpers.CreateDirectoryIfNotExist(OutputPath);
|
|||
|
|
|||
|
AVIWriter aviWriter = new AVIWriter(OutputPath, FPS, Size.Width, Size.Height, ShowOptions);
|
|||
|
|
|||
|
task = TaskEx.Run(() =>
|
|||
|
{
|
|||
|
try
|
|||
|
{
|
|||
|
position = 0;
|
|||
|
|
|||
|
while (!imageQueue.IsCompleted)
|
|||
|
{
|
|||
|
Image img = null;
|
|||
|
|
|||
|
try
|
|||
|
{
|
|||
|
img = imageQueue.Take();
|
|||
|
|
|||
|
if (img != null)
|
|||
|
{
|
|||
|
//img.Save("Test\\" + position + ".bmp", ImageFormat.Bmp);
|
|||
|
aviWriter.AddFrame((Bitmap)img);
|
|||
|
position++;
|
|||
|
}
|
|||
|
}
|
|||
|
catch (InvalidOperationException)
|
|||
|
{
|
|||
|
}
|
|||
|
finally
|
|||
|
{
|
|||
|
if (img != null) img.Dispose();
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
finally
|
|||
|
{
|
|||
|
IsWorking = false;
|
|||
|
|
|||
|
if (aviWriter != null)
|
|||
|
{
|
|||
|
aviWriter.Dispose();
|
|||
|
}
|
|||
|
}
|
|||
|
});
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
public void AddImageAsync(Image img)
|
|||
|
{
|
|||
|
if (IsWorking)
|
|||
|
{
|
|||
|
/*if (imageQueue.Count > 0)
|
|||
|
{
|
|||
|
Debug.WriteLine("ImageQueue count: " + imageQueue.Count);
|
|||
|
}*/
|
|||
|
|
|||
|
imageQueue.Add(img);
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
public void Finish()
|
|||
|
{
|
|||
|
if (IsWorking)
|
|||
|
{
|
|||
|
imageQueue.CompleteAdding();
|
|||
|
task.Wait();
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
public void Dispose()
|
|||
|
{
|
|||
|
if (imageQueue != null)
|
|||
|
{
|
|||
|
imageQueue.Dispose();
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
}
|