#region License Information (GPL v3)
/*
ShareX - A program that allows you to take screenshots and share any file type
Copyright (c) 2007-2024 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 .
*/
#endregion License Information (GPL v3)
using ShareX.HelpersLib;
using System;
using System.Diagnostics;
using System.Drawing;
using System.Drawing.Drawing2D;
using System.Drawing.Imaging;
using System.Runtime.InteropServices;
using System.Threading.Tasks;
namespace ShareX.ScreenCaptureLib
{
internal class ScrollingCaptureManager : IDisposable
{
public ScrollingCaptureOptions Options { get; private set; }
public Bitmap Result { get; private set; }
public bool IsCapturing { get; private set; }
private Bitmap lastScreenshot;
private Bitmap previousScreenshot;
private bool stopRequested;
private ScrollingCaptureStatus status;
private int bestMatchCount, bestMatchIndex;
private WindowInfo selectedWindow;
private Rectangle selectedRectangle;
public ScrollingCaptureManager(ScrollingCaptureOptions options)
{
Options = options;
}
public void Dispose()
{
Reset();
}
private void Reset(bool keepResult = false)
{
if (lastScreenshot != null)
{
lastScreenshot.Dispose();
lastScreenshot = null;
}
if (previousScreenshot != null)
{
previousScreenshot.Dispose();
previousScreenshot = null;
}
if (!keepResult && Result != null)
{
Result.Dispose();
Result = null;
}
}
public async Task StartCapture()
{
if (!IsCapturing && selectedWindow != null && !selectedRectangle.IsEmpty)
{
IsCapturing = true;
stopRequested = false;
status = ScrollingCaptureStatus.Failed;
bestMatchCount = 0;
bestMatchIndex = 0;
int ignoreSideOffset = 50;
int ignoreBottomOffset = 50;
Reset();
ScrollingCaptureRegionForm regionForm = null;
if (Options.ShowRegion)
{
regionForm = new ScrollingCaptureRegionForm(selectedRectangle);
regionForm.Show();
}
try
{
selectedWindow.Activate();
await Task.Delay(Options.StartDelay);
if (Options.AutoScrollTop)
{
InputHelpers.SendKeyPress(VirtualKeyCode.HOME);
NativeMethods.SendMessage(selectedWindow.Handle, (int)WindowsMessages.VSCROLL, (int)ScrollBarCommands.SB_TOP, 0);
await Task.Delay(Options.ScrollDelay);
}
Screenshot screenshot = new Screenshot()
{
CaptureCursor = false
};
while (!stopRequested)
{
lastScreenshot = screenshot.CaptureRectangle(selectedRectangle);
if (CompareLastTwoImages())
{
break;
}
InputHelpers.SendMouseWheel(-120 * Options.ScrollAmount);
Stopwatch timer = Stopwatch.StartNew();
if (lastScreenshot != null)
{
if (Options.AutoIgnoreBottomEdge && previousScreenshot != null)
{
int autoIgnoreBottomOffset = GuessBottomEdge(previousScreenshot, lastScreenshot, ignoreSideOffset);
ignoreBottomOffset = Math.Max(ignoreBottomOffset, autoIgnoreBottomOffset + 50);
}
Bitmap newResult = await CombineImagesAsync(Result, lastScreenshot, ignoreSideOffset, ignoreBottomOffset);
if (newResult != null)
{
Result?.Dispose();
Result = newResult;
}
else
{
break;
}
}
if (stopRequested)
{
break;
}
if (lastScreenshot != null)
{
if (previousScreenshot != null)
{
previousScreenshot.Dispose();
}
previousScreenshot = lastScreenshot;
lastScreenshot = null;
}
int delay = Options.ScrollDelay - (int)timer.ElapsedMilliseconds;
if (delay > 0)
{
await Task.Delay(delay);
}
}
}
finally
{
regionForm?.Close();
Reset(true);
IsCapturing = false;
}
}
return status;
}
public void StopCapture()
{
if (IsCapturing)
{
stopRequested = true;
}
}
public bool SelectWindow()
{
return RegionCaptureTasks.GetRectangleRegion(out selectedRectangle, out selectedWindow, new RegionCaptureOptions());
}
private bool IsScrollReachedBottom(IntPtr handle)
{
SCROLLINFO scrollInfo = new SCROLLINFO();
scrollInfo.cbSize = (uint)Marshal.SizeOf(scrollInfo);
scrollInfo.fMask = (uint)(ScrollInfoMask.SIF_RANGE | ScrollInfoMask.SIF_PAGE | ScrollInfoMask.SIF_TRACKPOS);
if (NativeMethods.GetScrollInfo(handle, (int)SBOrientation.SB_VERT, ref scrollInfo))
{
return scrollInfo.nMax == scrollInfo.nTrackPos + scrollInfo.nPage - 1;
}
return CompareLastTwoImages();
}
private bool CompareLastTwoImages()
{
if (lastScreenshot != null && previousScreenshot != null)
{
return ImageHelpers.CompareImages(lastScreenshot, previousScreenshot);
}
return false;
}
private async Task CombineImagesAsync(Bitmap result, Bitmap currentImage, int ignoreSideOffset = 50, int ignoreBottomOffset = 50)
{
return await Task.Run(() => CombineImages(result, currentImage, ignoreSideOffset, ignoreBottomOffset));
}
private Bitmap CombineImages(Bitmap result, Bitmap currentImage, int ignoreSideOffset = 50, int ignoreBottomOffset = 50)
{
if (result == null)
{
status = ScrollingCaptureStatus.Successful;
return (Bitmap)currentImage.Clone();
}
int matchCount = 0;
int matchIndex = 0;
int matchLimit = currentImage.Height / 2;
ignoreSideOffset = Math.Max(ignoreSideOffset, currentImage.Width / 20);
ignoreSideOffset = Math.Min(ignoreSideOffset, currentImage.Width / 3);
ignoreBottomOffset = Math.Max(ignoreBottomOffset, currentImage.Height / 20);
ignoreBottomOffset = Math.Min(ignoreBottomOffset, currentImage.Height / 3);
Rectangle rect = new Rectangle(ignoreSideOffset, result.Height - currentImage.Height,
currentImage.Width - ignoreSideOffset * 2, currentImage.Height - ignoreBottomOffset);
BitmapData bdResult = result.LockBits(new Rectangle(0, 0, result.Width, result.Height), ImageLockMode.ReadOnly, PixelFormat.Format32bppArgb);
BitmapData bdCurrentImage = currentImage.LockBits(new Rectangle(0, 0, currentImage.Width, currentImage.Height), ImageLockMode.ReadOnly, PixelFormat.Format32bppArgb);
int stride = bdResult.Stride;
int pixelSize = stride / result.Width;
IntPtr resultScan0 = bdResult.Scan0 + pixelSize * ignoreSideOffset;
IntPtr currentImageScan0 = bdCurrentImage.Scan0 + pixelSize * ignoreSideOffset;
int rectBottom = rect.Bottom - 1;
int compareLength = pixelSize * rect.Width;
for (int currentImageY = currentImage.Height - 1; currentImageY >= 0 && matchCount < matchLimit; currentImageY--)
{
int currentMatchCount = 0;
for (int y = 0; currentImageY - y >= 0 && currentMatchCount < matchLimit; y++)
{
if (NativeMethods.memcmp(resultScan0 + ((rectBottom - y) * stride), currentImageScan0 + ((currentImageY - y) * stride), compareLength) == 0)
{
currentMatchCount++;
}
else
{
break;
}
}
if (currentMatchCount > matchCount)
{
matchCount = currentMatchCount;
matchIndex = currentImageY;
}
}
result.UnlockBits(bdResult);
currentImage.UnlockBits(bdCurrentImage);
bool bestGuess = false;
if (matchCount == 0 && bestMatchCount > 0)
{
matchCount = bestMatchCount;
matchIndex = bestMatchIndex;
bestGuess = true;
}
if (matchCount > 0)
{
int matchHeight = currentImage.Height - matchIndex - 1;
if (matchHeight > 0)
{
if (matchCount > bestMatchCount)
{
bestMatchCount = matchCount;
bestMatchIndex = matchIndex;
}
Bitmap newResult = new Bitmap(result.Width, result.Height - ignoreBottomOffset + matchHeight);
using (Graphics g = Graphics.FromImage(newResult))
{
g.CompositingMode = CompositingMode.SourceCopy;
g.InterpolationMode = InterpolationMode.NearestNeighbor;
g.DrawImage(result, new Rectangle(0, 0, result.Width, result.Height - ignoreBottomOffset),
new Rectangle(0, 0, result.Width, result.Height - ignoreBottomOffset), GraphicsUnit.Pixel);
g.DrawImage(currentImage, new Rectangle(0, result.Height - ignoreBottomOffset, currentImage.Width, matchHeight),
new Rectangle(0, matchIndex + 1, currentImage.Width, matchHeight), GraphicsUnit.Pixel);
}
if (bestGuess)
{
status = ScrollingCaptureStatus.PartiallySuccessful;
}
else if (status != ScrollingCaptureStatus.PartiallySuccessful)
{
status = ScrollingCaptureStatus.Successful;
}
return newResult;
}
}
status = ScrollingCaptureStatus.Failed;
return null;
}
private int GuessBottomEdge(Bitmap img1, Bitmap img2, int ignoreSideOffset)
{
Rectangle rect = new Rectangle(ignoreSideOffset, 0, img1.Width - ignoreSideOffset * 2, img1.Height);
using (UnsafeBitmap bmp1 = new UnsafeBitmap(img1, true, ImageLockMode.ReadOnly))
using (UnsafeBitmap bmp2 = new UnsafeBitmap(img2, true, ImageLockMode.ReadOnly))
{
for (int y = rect.Height - 1; y >= rect.X; y--)
{
for (int x = rect.X; x < rect.Width; x++)
{
if (bmp1.GetPixel(x, y) != bmp2.GetPixel(x, y))
{
return rect.Height - y - 1;
}
}
}
}
return 0;
}
}
}