mirror of
https://github.com/ShareX/ShareX.git
synced 2024-10-04 12:14:45 +13:00
139 lines
No EOL
4.2 KiB
C#
139 lines
No EOL
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-2014 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 ScreenCaptureLib
|
|
{
|
|
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();
|
|
}
|
|
}
|
|
}
|
|
} |