add Screenshot

This commit is contained in:
NaBian committed 2020-03-06 14:36:12 +08:00
1 parent 08ae754ea3
commit 46b1297aa7
28 files changed
+923 -100

No files matched your search

@@ -13,11 +13,11 @@ namespace HandyControl.Controls
public static Visibility GetVisibility(DependencyObject element) => (Visibility) element.GetValue(VisibilityProperty);
public static readonly DependencyProperty PlacementProperty = DependencyProperty.RegisterAttached(
"Placement", typeof(TipPlacement), typeof(TipElement), new PropertyMetadata(default(TipPlacement)));
"Placement", typeof(PlacementType), typeof(TipElement), new PropertyMetadata(default(PlacementType)));
public static void SetPlacement(DependencyObject element, TipPlacement value) => element.SetValue(PlacementProperty, value);
public static void SetPlacement(DependencyObject element, PlacementType value) => element.SetValue(PlacementProperty, value);
public static TipPlacement GetPlacement(DependencyObject element) => (TipPlacement) element.GetValue(PlacementProperty);
public static PlacementType GetPlacement(DependencyObject element) => (PlacementType) element.GetValue(PlacementProperty);
public static readonly DependencyProperty StringFormatProperty = DependencyProperty.RegisterAttached(
"StringFormat", typeof(string), typeof(TipElement), new PropertyMetadata("#0.0"));
@@ -60,7 +60,7 @@ namespace HandyControl.Controls
if (!_colorPicker.IsMouseOver && window.IsMouseOver)
{
UpdateCursor(true);
if (e.Message == MouseHookMessageType.LeftButtonDown)
if (e.MessageType == MouseHookMessageType.LeftButtonDown)
{
var brush = new SolidColorBrush(GetColorAt(e.Point.X, e.Point.Y));
_colorPicker.SelectedBrush = brush;
@@ -106,19 +106,19 @@ namespace HandyControl.Controls
set => SetValue(OffsetProperty, value);
}
public static readonly DependencyProperty PlacementProperty = DependencyProperty.RegisterAttached(
"Placement", typeof(TipPlacement), typeof(Poptip), new PropertyMetadata(TipPlacement.Top));
public static readonly DependencyProperty PlacementTypeProperty = DependencyProperty.RegisterAttached(
"PlacementType", typeof(PlacementType), typeof(Poptip), new PropertyMetadata(PlacementType.Top));
public static void SetPlacement(DependencyObject element, TipPlacement value)
=> element.SetValue(PlacementProperty, value);
public static void SetPlacement(DependencyObject element, PlacementType value)
=> element.SetValue(PlacementTypeProperty, value);
public static TipPlacement GetPlacement(DependencyObject element)
=> (TipPlacement) element.GetValue(PlacementProperty);
public static PlacementType GetPlacement(DependencyObject element)
=> (PlacementType) element.GetValue(PlacementTypeProperty);
public TipPlacement Placement
public PlacementType PlacementType
{
get => (TipPlacement)GetValue(PlacementProperty);
set => SetValue(PlacementProperty, value);
get => (PlacementType)GetValue(PlacementTypeProperty);
set => SetValue(PlacementTypeProperty, value);
}
public static readonly DependencyProperty IsOpenProperty = DependencyProperty.RegisterAttached(
@@ -192,65 +192,65 @@ namespace HandyControl.Controls
var offsetY = .0;
var poptip = (Poptip)GetInstance(Target);
var popupPlacement = poptip.Placement;
var popupPlacement = poptip.PlacementType;
var popupOffset = poptip.Offset;
PlacementMode placementMode;
switch (popupPlacement)
{
case TipPlacement.LeftTop:
case PlacementType.LeftTop:
offsetX = -popupOffset;
placementMode = PlacementMode.Left;
break;
case TipPlacement.Left:
case PlacementType.Left:
offsetX = -popupOffset;
offsetY = (targetHeight - height) / 2;
placementMode = PlacementMode.Left;
break;
case TipPlacement.LeftBottom:
case PlacementType.LeftBottom:
offsetX = -popupOffset;
offsetY = targetHeight - height;
placementMode = PlacementMode.Left;
break;
case TipPlacement.TopLeft:
case PlacementType.TopLeft:
offsetY = -popupOffset;
placementMode = PlacementMode.Top;
break;
case TipPlacement.Top:
case PlacementType.Top:
offsetX = (targetWidth - width) / 2;
offsetY = -popupOffset;
placementMode = PlacementMode.Top;
break;
case TipPlacement.TopRight:
case PlacementType.TopRight:
offsetX = targetWidth - width;
offsetY = -popupOffset;
placementMode = PlacementMode.Top;
break;
case TipPlacement.RightTop:
case PlacementType.RightTop:
offsetX = popupOffset;
placementMode = PlacementMode.Right;
break;
case TipPlacement.Right:
case PlacementType.Right:
offsetX = popupOffset;
offsetY = (targetHeight - height) / 2;
placementMode = PlacementMode.Right;
break;
case TipPlacement.RightBottom:
case PlacementType.RightBottom:
offsetX = popupOffset;
offsetY = targetHeight - height;
placementMode = PlacementMode.Right;
break;
case TipPlacement.BottomLeft:
case PlacementType.BottomLeft:
offsetY = popupOffset;
placementMode = PlacementMode.Bottom;
break;
case TipPlacement.Bottom:
case PlacementType.Bottom:
offsetX = (targetWidth - width) / 2;
offsetY = popupOffset;
placementMode = PlacementMode.Bottom;
break;
case TipPlacement.BottomRight:
case PlacementType.BottomRight:
offsetX = targetWidth - width;
offsetY = popupOffset;
placementMode = PlacementMode.Bottom;
@@ -271,7 +271,7 @@ namespace HandyControl.Controls
if (!GetIsInstance(Target))
{
SetCurrentValue(ContentProperty, GetContent(Target));
SetCurrentValue(PlacementProperty, GetPlacement(Target));
SetCurrentValue(PlacementTypeProperty, GetPlacement(Target));
SetCurrentValue(HitModeProperty, GetHitMode(Target));
SetCurrentValue(OffsetProperty, GetOffset(Target));
SetCurrentValue(IsOpenProperty, GetIsOpen(Target));
@@ -0,0 +1,16 @@
using System;
using System.Windows.Media;
namespace HandyControl.Controls
{
public class Screenshot
{
public event EventHandler Snapped;
public ImageSource Source { get; set; }
public void Start() => new ScreenshotWindow(this).Show();
internal void OnSnapped() => Snapped?.Invoke(this, EventArgs.Empty);
}
}
@@ -0,0 +1,572 @@
using System;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Input;
using System.Windows.Interop;
using System.Windows.Media;
using System.Windows.Media.Imaging;
using HandyControl.Data;
using HandyControl.Tools;
using HandyControl.Tools.Interop;
namespace HandyControl.Controls
{
[TemplatePart(Name = ElementCanvas, Type = typeof(InkCanvas))]
[TemplatePart(Name = ElementMaskAreaLeft, Type = typeof(FrameworkElement))]
[TemplatePart(Name = ElementMaskAreaTop, Type = typeof(FrameworkElement))]
[TemplatePart(Name = ElementMaskAreaRight, Type = typeof(FrameworkElement))]
[TemplatePart(Name = ElementMaskAreaBottom, Type = typeof(FrameworkElement))]
[TemplatePart(Name = ElementTargetArea, Type = typeof(InkCanvas))]
[TemplatePart(Name = ElementTextSize, Type = typeof(TextBlock))]
public class ScreenshotWindow : System.Windows.Window
{
#region fields
private readonly Screenshot _screenshot;
#region const
private const int RectMinLength = 14;
private const int IntervalLength = 1;
private const int IntervalBigLength = 10;
private const int SnapLength = 4;
#endregion
#region IntPtr
private IntPtr _desktopWindowHandle;
private IntPtr _mouseOverWindowHandle;
private readonly IntPtr _screenshotWindowHandle;
#endregion
#region status
private InteropValues.RECT _desktopWindowRect;
private InteropValues.RECT _targetWindowRect;
private readonly int[] _flagArr = new int[4];
private bool _canDrag;
private bool _receiveMoveMsg = true;
private Point _mousePointOld;
#endregion
#endregion
#region Elements
internal InkCanvas Canvas { get; set; }
internal FrameworkElement MaskAreaLeft { get; set; }
internal FrameworkElement MaskAreaTop { get; set; }
internal FrameworkElement MaskAreaRight { get; set; }
internal FrameworkElement MaskAreaBottom { get; set; }
internal FrameworkElement TargetArea { get; set; }
internal TextBlock TextSize { get; set; }
#endregion
#region const
private const string ElementCanvas = "PART_Canvas";
private const string ElementMaskAreaLeft = "PART_MaskAreaLeft";
private const string ElementMaskAreaTop = "PART_MaskAreaTop";
private const string ElementMaskAreaRight = "PART_MaskAreaRight";
private const string ElementMaskAreaBottom = "PART_MaskAreaBottom";
private const string ElementTargetArea = "PART_TargetArea";
private const string ElementTextSize = "PART_TextSize";
#endregion
#region prop
public static readonly DependencyProperty IsDrawingProperty = DependencyProperty.Register(
"IsDrawing", typeof(bool), typeof(ScreenshotWindow), new PropertyMetadata(ValueBoxes.FalseBox));
public bool IsDrawing
{
get => (bool)GetValue(IsDrawingProperty);
internal set => SetValue(IsDrawingProperty, value);
}
public static readonly DependencyProperty IsSelectingProperty = DependencyProperty.Register(
"IsSelecting", typeof(bool), typeof(ScreenshotWindow), new PropertyMetadata(ValueBoxes.FalseBox));
public bool IsSelecting
{
get => (bool)GetValue(IsSelectingProperty);
internal set => SetValue(IsSelectingProperty, value);
}
#endregion
public ScreenshotWindow(Screenshot screenshot)
{
Style = ResourceHelper.GetResource<Style>(ResourceToken.Window4ScreenshotStyle);
_screenshot = screenshot;
DataContext = this;
_screenshotWindowHandle = this.GetHandle();
InteropMethods.EnableWindow(_screenshotWindowHandle, false);
Loaded += ScreenshotWindow_Loaded;
Closed += ScreenshotWindow_Closed;
}
public override void OnApplyTemplate()
{
base.OnApplyTemplate();
Canvas = GetTemplateChild(ElementCanvas) as InkCanvas;
MaskAreaLeft = GetTemplateChild(ElementMaskAreaLeft) as FrameworkElement;
MaskAreaTop = GetTemplateChild(ElementMaskAreaTop) as FrameworkElement;
MaskAreaRight = GetTemplateChild(ElementMaskAreaRight) as FrameworkElement;
MaskAreaBottom = GetTemplateChild(ElementMaskAreaBottom) as FrameworkElement;
TargetArea = GetTemplateChild(ElementTargetArea) as FrameworkElement;
TextSize = GetTemplateChild(ElementTextSize) as TextBlock;
}
protected override void OnPreviewKeyDown(KeyEventArgs e)
{
if (e.Key == Key.Escape) Close();
if (IsDrawing)
{
switch (e.Key)
{
case Key.Left:
{
MoveTargetArea(MoveRect(_targetWindowRect, -1, rightFlag: -1));
}
break;
case Key.Up:
{
MoveTargetArea(MoveRect(_targetWindowRect, bottomFlag: -1, topFlag: -1));
}
break;
case Key.Right:
{
MoveTargetArea(MoveRect(_targetWindowRect, 1, rightFlag: 1));
}
break;
case Key.Down:
{
MoveTargetArea(MoveRect(_targetWindowRect, bottomFlag: 1, topFlag: 1));
}
break;
case Key.Enter:
SaveScreenshot();
break;
}
}
}
protected override void OnPreviewMouseLeftButtonDown(MouseButtonEventArgs e) => _mousePointOld = e.GetPosition(this);
protected override void OnPreviewMouseDoubleClick(MouseButtonEventArgs e) => SaveScreenshot();
protected override void OnPreviewMouseMove(MouseEventArgs e)
{
if (e.LeftButton != MouseButtonState.Pressed)
{
UpdateStatus(e.GetPosition(TargetArea));
return;
}
var newPoint = Mouse.GetPosition(this);
var offsetX = (int)(newPoint.X - _mousePointOld.X);
var offsetY = (int)(newPoint.Y - _mousePointOld.Y);
var rect = _targetWindowRect;
if (IsDrawing)
{
if (_canDrag)
{
rect.Left += offsetX;
rect.Top += offsetY;
rect.Right += offsetX;
rect.Bottom += offsetY;
}
else
{
var leftAddend = offsetX * _flagArr[0];
var topAddend = offsetY * _flagArr[1];
var rightAddend = offsetX * _flagArr[2];
var bottomAddend = offsetY * _flagArr[3];
rect.Left += leftAddend;
rect.Top += topAddend;
rect.Right += rightAddend;
rect.Bottom += bottomAddend;
if (rect.Width < 0)
{
if (_flagArr[0] > 0)
{
_flagArr[0] = 0;
_flagArr[2] = 1;
rect.Left -= leftAddend;
rightAddend = offsetX * _flagArr[2];
rect.Right += rightAddend;
}
else if (_flagArr[2] > 0)
{
_flagArr[0] = 1;
_flagArr[2] = 0;
rect.Right -= rightAddend;
leftAddend = offsetX * _flagArr[0];
rect.Left += leftAddend;
}
}
if (rect.Height < 0)
{
if (_flagArr[1] > 0)
{
_flagArr[1] = 0;
_flagArr[3] = 1;
rect.Top -= topAddend;
bottomAddend = offsetY * _flagArr[3];
rect.Bottom += bottomAddend;
}
else if (_flagArr[3] > 0)
{
_flagArr[1] = 1;
_flagArr[3] = 0;
rect.Bottom -= bottomAddend;
topAddend = offsetY * _flagArr[1];
rect.Top += topAddend;
}
}
}
MoveTargetArea(rect);
_mousePointOld = newPoint;
}
else
{
if (!IsSelecting && offsetX > 0 && offsetY > 0)
{
rect = new InteropValues.RECT((int)_mousePointOld.X, (int)_mousePointOld.Y, 0, 0);
IsSelecting = true;
}
if (IsSelecting)
{
MoveTargetArea(new InteropValues.RECT(rect.Left, rect.Top, rect.Left + offsetX, rect.Top + offsetY));
}
}
}
private void ScreenshotWindow_Closed(object sender, EventArgs e)
{
StopHooks();
IsDrawing = false;
Loaded -= ScreenshotWindow_Loaded;
Closed -= ScreenshotWindow_Closed;
}
private void ScreenshotWindow_Loaded(object sender, RoutedEventArgs e)
{
Canvas.Background = new ImageBrush(GetDesktopSnapshot());
StartHooks();
InteropMethods.GetCursorPos(out var point);
MoveElement2Point(point);
}
private void UpdateStatus(Point point)
{
Cursor cursor;
var leftAbs = Math.Abs(point.X);
var topAbs = Math.Abs(point.Y);
var rightAbs = Math.Abs(point.X - TargetArea.Width);
var downAbs = Math.Abs(point.Y - TargetArea.Height);
_canDrag = false;
_flagArr[0] = 0;
_flagArr[1] = 0;
_flagArr[2] = 0;
_flagArr[3] = 0;
if (leftAbs <= SnapLength)
{
if (topAbs > SnapLength)
{
if (downAbs > SnapLength)
{
// left
cursor = Cursors.SizeWE;
_flagArr[0] = 1;
}
else
{
//left bottom
cursor = Cursors.SizeNESW;
_flagArr[0] = 1;
_flagArr[3] = 1;
}
}
else
{
// left top
cursor = Cursors.SizeNWSE;
_flagArr[0] = 1;
_flagArr[1] = 1;
}
}
else if (rightAbs > SnapLength)
{
if (topAbs > SnapLength)
{
if (downAbs > SnapLength)
{
if (TargetArea.IsMouseOver)
{
//drag
cursor = Cursors.SizeAll;
_canDrag = true;
}
else
{
//out
cursor = Cursors.Arrow;
}
}
else
{
//bottom
cursor = Cursors.SizeNS;
_flagArr[3] = 1;
}
}
else
{
//top
cursor = Cursors.SizeNS;
_flagArr[1] = 1;
}
}
else if (rightAbs <= SnapLength)
{
if (topAbs > SnapLength)
{
if (downAbs > SnapLength)
{
//right
cursor = Cursors.SizeWE;
_flagArr[2] = 1;
}
else
{
//right bottom
cursor = Cursors.SizeNWSE;
_flagArr[2] = 1;
_flagArr[3] = 1;
}
}
else
{
// right top
cursor = Cursors.SizeNESW;
_flagArr[1] = 1;
_flagArr[2] = 1;
}
}
else
{
//out
cursor = Cursors.Arrow;
}
TargetArea.Cursor = cursor;
}
private void StopHooks()
{
MouseHook.Stop();
MouseHook.StatusChanged -= MouseHook_StatusChanged;
}
private void StartHooks()
{
MouseHook.Start();
MouseHook.StatusChanged += MouseHook_StatusChanged;
}
private void MouseHook_StatusChanged(object sender, MouseHookEventArgs e)
{
switch (e.MessageType)
{
case MouseHookMessageType.MouseMove:
MoveElement2Point(e.Point);
break;
case MouseHookMessageType.LeftButtonDown:
_receiveMoveMsg = false;
_mousePointOld = new Point(e.Point.X, e.Point.Y);
InteropMethods.EnableWindow(_screenshotWindowHandle, true);
break;
case MouseHookMessageType.LeftButtonUp:
StopHooks();
IsSelecting = false;
IsDrawing = true;
break;
}
}
private void SaveScreenshot()
{
if (Canvas.Background is ImageBrush imageBrush && imageBrush.ImageSource is BitmapSource bitmapSource)
{
var cb = new CroppedBitmap(bitmapSource, new Int32Rect(_targetWindowRect.Left, _targetWindowRect.Top, _targetWindowRect.Width, _targetWindowRect.Height));
_screenshot.Source = cb;
}
_screenshot.OnSnapped();
Close();
}
private ImageSource GetDesktopSnapshot()
{
_desktopWindowHandle = InteropMethods.GetDesktopWindow();
var hdcSrc = InteropMethods.GetWindowDC(_desktopWindowHandle);
var hdcDest = InteropMethods.CreateCompatibleDC(hdcSrc);
InteropMethods.GetWindowRect(_desktopWindowHandle, out _desktopWindowRect);
var desktopWindowWidth = _desktopWindowRect.Right - _desktopWindowRect.Left;
var desktopWindowHeight = _desktopWindowRect.Bottom - _desktopWindowRect.Top;
var bitmap = InteropMethods.CreateCompatibleBitmap(hdcSrc, desktopWindowWidth, desktopWindowHeight);
var hOld = InteropMethods.SelectObject(hdcDest, bitmap);
InteropMethods.BitBlt(hdcDest, 0, 0, desktopWindowWidth, desktopWindowHeight, hdcSrc, 0, 0, InteropValues.SRCCOPY);
InteropMethods.SelectObject(hdcDest, hOld);
InteropMethods.DeleteDC(hdcDest);
InteropMethods.ReleaseDC(_desktopWindowHandle, hdcSrc);
return Imaging.CreateBitmapSourceFromHBitmap(bitmap, IntPtr.Zero, Int32Rect.Empty, BitmapSizeOptions.FromEmptyOptions());
}
private void MoveElement2Point(InteropValues.POINT point)
{
if (!_receiveMoveMsg) return;
var mouseOverWindowHandle = InteropMethods.ChildWindowFromPointEx(_desktopWindowHandle,
new InteropValues.POINT
{
X = point.X,
Y = point.Y
}, 1 | 2);
if (mouseOverWindowHandle != _mouseOverWindowHandle && mouseOverWindowHandle != IntPtr.Zero)
{
_mouseOverWindowHandle = mouseOverWindowHandle;
InteropMethods.GetWindowRect(_mouseOverWindowHandle, out var windowRect);
MoveTargetArea(windowRect);
}
}
private void RegularRect(ref InteropValues.RECT rect)
{
if (rect.Left < 0)
{
rect.Right -= rect.Left;
rect.Left = 0;
}
if (rect.Top < 0)
{
rect.Bottom -= rect.Top;
rect.Top = 0;
}
if (rect.Right > _desktopWindowRect.Width)
{
rect.Left -= rect.Right - _desktopWindowRect.Width;
rect.Right = _desktopWindowRect.Width;
}
if (rect.Bottom > _desktopWindowRect.Height)
{
rect.Top -= rect.Bottom - _desktopWindowRect.Height;
rect.Bottom = _desktopWindowRect.Height;
}
rect.Left = Math.Max(0, rect.Left);
rect.Top = Math.Max(0, rect.Top);
}
private static InteropValues.RECT MoveRect(InteropValues.RECT rect, int leftFlag = 0, int topFlag = 0, int rightFlag = 0, int bottomFlag = 0)
{
var moveLength = Keyboard.IsKeyDown(Key.LeftCtrl) || Keyboard.IsKeyDown(Key.RightCtrl)
? IntervalBigLength
: IntervalLength;
rect.Left += leftFlag * moveLength;
rect.Top += topFlag * moveLength;
rect.Right += rightFlag * moveLength;
rect.Bottom += bottomFlag * moveLength;
return rect;
}
private void MoveTargetArea(InteropValues.RECT rect)
{
RegularRect(ref rect);
var width = rect.Width;
var height = rect.Height;
var left = rect.Left;
var top = rect.Top;
TargetArea.Width = width;
TargetArea.Height = height;
TargetArea.Margin = new Thickness(left, top, 0, 0);
_targetWindowRect = new InteropValues.RECT(left, top, left + width, top + height);
TextSize.Text = $"{_targetWindowRect.Width} x {_targetWindowRect.Height}";
MoveMaskArea();
}
private void MoveMaskArea()
{
MaskAreaLeft.Width = TargetArea.Margin.Left;
MaskAreaLeft.Height = _desktopWindowRect.Height;
MaskAreaTop.Margin = new Thickness(TargetArea.Margin.Left, 0, 0, 0);
MaskAreaTop.Width = TargetArea.Width;
MaskAreaTop.Height = TargetArea.Margin.Top;
MaskAreaRight.Margin = new Thickness(TargetArea.Width + TargetArea.Margin.Left, 0, 0, 0);
MaskAreaRight.Width = _desktopWindowRect.Width - TargetArea.Margin.Left - TargetArea.Width;
MaskAreaRight.Height = _desktopWindowRect.Height;
MaskAreaBottom.Margin = new Thickness(TargetArea.Margin.Left, TargetArea.Height + TargetArea.Margin.Top, 0, 0);
MaskAreaBottom.Width = TargetArea.Width;
MaskAreaBottom.Height = _desktopWindowRect.Height - TargetArea.Height - TargetArea.Margin.Top;
}
}
}