special_input_device 0.0.1 → 0.1.0

data/ext/special_input_device/window.c

@@ -1,6 +1,7 @@
 #include "special_input_device.h"
 #include "_system_time.h"
 #include "color.h"
+#include "image.h"
 #include "rectangle.h"
 
 #define MAXIMUM_WINDOW_NAME_LENGTH 256
@@ -136,13 +137,13 @@ static VALUE ruby__find(VARIABLE_ARGUMENTS_C) {
 	SCAN_ARGUMENTS("02", &windowName, &className);
 	if (windowName != Qnil && !RTEST(RUBY_OBJECT_IS_A_qs(windowName, rb_cRegexp)) &&
 	    !RTEST(RUBY_OBJECT_IS_A_qs(windowName, rb_cString)))
-		RAISE_TYPE_ERROR(windowName, "nil, Regexp or String");
+		RAISE_SIMPLE_TYPE_ERROR(windowName, "nil, Regexp or String");
 	if (className != Qnil && !RTEST(RUBY_OBJECT_IS_A_qs(className, rb_cRegexp)) &&
 	    !RTEST(RUBY_OBJECT_IS_A_qs(className, rb_cString)))
-		RAISE_TYPE_ERROR(className, "nil, Regexp or String");
-	cCallbackArguments.className   = className;
-	cCallbackArguments.windowName  = windowName;
-	cCallbackArguments.hWND = 0;
+		RAISE_SIMPLE_TYPE_ERROR(className, "nil, Regexp or String");
+	cCallbackArguments.className  = className;
+	cCallbackArguments.windowName = windowName;
+	cCallbackArguments.hWND       = 0;
 	EnumWindows(callback_find, (LPARAM) &cCallbackArguments);
 	cHWND = cCallbackArguments.hWND;
 	if (GetLastError())
@@ -163,7 +164,7 @@ static VALUE ruby__alloc(VALUE self) {
 }
 
 static VALUE ruby_window_name_EQ(VALUE self, VALUE value) {
-	HWND    cHWND = ruby___window_get_handle_code(self);
+	HWND    cHWND       = ruby___window_get_handle_code(self);
 	LPCTSTR cWindowName = RUBY_VALUE_TO_LPCTSTR(value);
 	SetWindowText(cHWND, cWindowName);
 	return rb_iv_set(self, "@window_name", value);
@@ -177,7 +178,7 @@ static VALUE ruby_window_name_EQ(VALUE self, VALUE value) {
  *   @param [Integer] handle_code the handle code of the window. Handle code is an Integer for identifying windows.
  */
 static VALUE ruby_initialize(VALUE self, VALUE handleCode) {
-	HWND   cHWND       = (HWND) NUM2ULL(handleCode);
+	HWND   cHWND             = (HWND) NUM2ULL(handleCode);
 	TCHAR  cClassName[MAXIMUM_WINDOW_NAME_LENGTH * sizeof(TCHAR)];
 	INT    cWindowNameLength = GetWindowTextLength(cHWND) + 1;
 	LPTSTR cWindowName       = (LPTSTR) VirtualAlloc((LPVOID) NULL, (SIZE_T) (cWindowNameLength) * sizeof(TCHAR),
@@ -204,7 +205,7 @@ static VALUE ruby_initialize(VALUE self, VALUE handleCode) {
 	return Qnil;
 }
 
-//<editor-fold desc="relationship">
+//region relationship
 
 /*
  * @overload find_child(window_name = nil, class_name = nil, after = nil)
@@ -228,7 +229,7 @@ static VALUE ruby_find_child(VARIABLE_ARGUMENTS_C) {
 	cWindowName = windowName == Qnil ? NULL : RUBY_VALUE_TO_LPCTSTR(windowName);
 	cClassName  = className == Qnil ? NULL : RUBY_VALUE_TO_LPCTSTR(className);
 	cAfter      = after == Qnil ? NULL : ruby___window_get_handle_code(after);
-	cHWND = FindWindowEx(ruby___window_get_handle_code(self), cAfter, cClassName, cWindowName);
+	cHWND       = FindWindowEx(ruby___window_get_handle_code(self), cAfter, cClassName, cWindowName);
 	if (!cHWND)
 		RAISE_WIN32_ERROR();
 	return ruby___window_new(cHWND);
@@ -262,9 +263,9 @@ static VALUE ruby_root(VARIABLE_ARGUMENTS_C) {
 	return ruby___window_new(GetAncestor(cHWND, includeOwner && includeOwner != Qnil ? GA_ROOT : GA_ROOTOWNER));
 }
 
-//</editor-fold>
+//endregion relationship
 
-//<editor-fold desc="size">
+//region size
 
 /*
  * Returns the rectangle of the window's client area. The coordinates of rectangle are relative to the client area
@@ -298,9 +299,9 @@ static VALUE ruby_window_rectangle(VALUE self) {
 	return ruby___rectangle_new_from_rect(&cWindowRectangle);
 }
 
-//</editor-fold>
+//endregion size
 
-//<editor-fold desc="control">
+//region control
 
 /*
  * Brings the window to the foreground.
@@ -402,9 +403,9 @@ static VALUE ruby_restore(VALUE self) {
 	return self;
 }
 
-//</editor-fold>
+//endregion control
 
-//<editor-fold desc="display">
+//region display
 
 /*
  * @overload color_at(x, y)
@@ -415,18 +416,162 @@ static VALUE ruby_restore(VALUE self) {
  * @see https://msdn.microsoft.com/en-us/library/windows/desktop/dd144909.aspx Windows Dev Center - GetPixel function
  */
 static VALUE ruby_color_at(VALUE self, VALUE x, VALUE y) {
-	INT      cX        = NUM2INT(x);
-	INT      cY        = NUM2INT(y);
-	HDC      cWindowDC = GetDC(ruby___window_get_handle_code(self));
-	COLORREF cColor    = GetPixel(cWindowDC, cX, cY);
+	HWND  cWindow = ruby___window_get_handle_code(self);
+	COLORREF cColor;
+	HDC   cWindowDC;
+	POINT cPoint;
+	RECT  cWindowRectangle;
+	cPoint.x = NUM2LONG(x);
+	cPoint.y = NUM2LONG(y);
+	if (!ClientToScreen(cWindow, &cPoint))
+		RAISE_WIN32_ERROR();
+	if (!GetWindowRect(cWindow, &cWindowRectangle))
+		RAISE_WIN32_ERROR();
+	cWindowDC = GetWindowDC(cWindow);
+	cColor = GetPixel(cWindowDC, cPoint.x - cWindowRectangle.left, cPoint.y - cWindowRectangle.top);
+	ReleaseDC(cWindow, cWindowDC);
 	return ruby___color_new_from_color_ref(cColor);
 }
 
-//</editor-fold>
+typedef struct {
+	INT    x;
+	INT    y;
+	INT    width;
+	INT    height;
+	HWND   window;
+	HDC    memoryDC;
+	HDC    windowDC;
+	LPSTR  bitmapFile;
+	HANDLE hBitmapFile;
+}            SID_WINDOW_CAPTURE_PARAMETER;
+
+static VALUE ruby_capture_helper(VALUE arguments) {
+	SID_WINDOW_CAPTURE_PARAMETER *cParameter = GET_DATA_OBJECT(arguments);
+	HBITMAP                      cHBitmap;
+
+	cParameter->windowDC = GetWindowDC(cParameter->window);
+	cParameter->memoryDC = CreateCompatibleDC(cParameter->windowDC);
+	cHBitmap = CreateCompatibleBitmap(cParameter->windowDC, cParameter->width, cParameter->height);
+
+	SelectObject(cParameter->memoryDC, cHBitmap);
+	BitBlt(cParameter->memoryDC, 0, 0, cParameter->width, cParameter->height,
+	       cParameter->windowDC, cParameter->x, cParameter->y, SRCCOPY);
+
+	return ruby___image_new_from_bitmap(cHBitmap);
+}
 
-//<editor-fold desc="input">
+/*
+ * @overload capture()
+ *   Capture the whole screen.
+ * @overload capture(x, y, width, height)
+ *   Capture a rectangle part of screen.
+ *   @param [Integer] x the x-coordinate of the left edge
+ *   @param [Integer] y the y-coordinate of the top edge
+ *   @param [Integer] width the width of rectangle
+ *   @param [Integer] height the height of rectangle
+ * @return [SpecialInputDevice::Image]
+ */
+static VALUE ruby_capture(VARIABLE_ARGUMENTS_C) {
+	VALUE                        x;
+	VALUE                        y;
+	VALUE                        width;
+	VALUE                        height;
+	VALUE                        parameter;
+	SID_WINDOW_CAPTURE_PARAMETER *cParameter;
+	INT                          cError;
+	VALUE                        result;
+	parameter          = Data_Make_Struct(rb_cData, SID_WINDOW_CAPTURE_PARAMETER, NULL, RUBY_DEFAULT_FREE, cParameter);
+	switch (cCount) {
+		case 0: {
+			HWND cWindow = ruby___window_get_handle_code(self);
+			RECT cClientRectangle;
+			RECT cWindowRectangle;
+			if (!GetClientRect(cWindow, &cClientRectangle))
+				RAISE_WIN32_ERROR();
+			if (!ClientToScreen(cWindow, (LPPOINT) &cClientRectangle))
+				RAISE_WIN32_ERROR();
+			if (!GetWindowRect(cWindow, &cWindowRectangle))
+				RAISE_WIN32_ERROR();
+			cParameter->x      = cClientRectangle.left - cWindowRectangle.left;
+			cParameter->y      = cClientRectangle.top - cWindowRectangle.top;
+			cParameter->width  = cClientRectangle.right;
+			cParameter->height = cClientRectangle.bottom;
+			break;
+		}
+		case 4:
+			SCAN_ARGUMENTS("4", &x, &y, &width, &height);
+			cParameter->x      = NUM2INT(x);
+			cParameter->y      = NUM2INT(y);
+			cParameter->width  = NUM2INT(width);
+			cParameter->height = NUM2INT(height);
+			break;
+		default:
+			RAISE_ARGUMENT_ERROR("0, 4");
+	}
+	cParameter->window = ruby___window_get_handle_code(self);
+	result = rb_protect(ruby_capture_helper, parameter, &cError);
+	GlobalUnlock(cParameter->hBitmapFile);
+	GlobalFree(cParameter->hBitmapFile);
+	ReleaseDC(NULL, cParameter->memoryDC);
+	ReleaseDC(NULL, cParameter->windowDC);
+	if (cError)
+		rb_jump_tag(cError);
+	return result;
+}
 
-//<editor-fold desc="keyboard">
+/*
+ * @overload find_image(image, area: SpecialInputDevice::Screen.primary_screen, similarity: 1.0, _return: RETURN_METHOD::FIRST)
+ *   @param [SpecialInputDevice::Image, String] image a picture to be search
+ *   @param [SpecialInputDevice::Rectangle] area the searching area
+ *   @param [Float] similarity minimum limit of similarity
+ *   @param [Symbol] _return indicate the behavior of this method, see SpecialInputDevice::Image::RETURN_METHOD
+ * @return [NilClass, SpecialInputDevice::Rectangle] a rectangle contain the position and the size of the result
+ * @see SpecialInputDevice::Image#find
+ * @see SpecialInputDevice::Image::RETURN_METHOD
+ */
+static VALUE ruby_find_image(VARIABLE_ARGUMENTS_C) {
+	VALUE image;
+	VALUE options;
+	ID    keywords[1] = {
+			rb_intern_const("area")
+	};
+	VALUE option_values[sizeof(keywords)];
+	VALUE *area       = option_values + 0;
+	VALUE x           = Qnil;
+	VALUE y           = Qnil;
+	VALUE capture;
+	VALUE captureArguments[4];
+	VALUE result;
+	SCAN_ARGUMENTS("1:", &image, &options);
+	rb_get_kwargs(options, keywords, 0, -2, option_values);
+	if (RTEST(RUBY_OBJECT_IS_A_qs(image, rb_cString)) || RTEST(RUBY_OBJECT_IS_A_qs(image, rb_cFile)))
+		image = rb_funcall(specialInputDevice_image, rb_intern_const("load"), 1, image);
+	else if (!RTEST(RUBY_OBJECT_IS_A_qs(image, specialInputDevice_image)))
+		RAISE_SIMPLE_TYPE_ERROR(image, "String, File or SpecialInputDevice::Image");
+	if (*area == Qundef) {
+		capture = ruby_capture(0, captureArguments, self);
+	} else {
+		CHECK_TYPE_ERROR(*area, ":area", specialInputDevice_rectangle);
+		x = rb_funcall(*area, rb_intern_const("x"), 0);
+		y = rb_funcall(*area, rb_intern_const("y"), 0);
+		captureArguments[0] = x;
+		captureArguments[1] = y;
+		captureArguments[2] = rb_funcall(*area, rb_intern_const("width"), 0);
+		captureArguments[3] = rb_funcall(*area, rb_intern_const("height"), 0);
+		capture = ruby_capture(4, captureArguments, self);
+	}
+	rb_funcall(capture, rb_intern_const("save"), 1, rb_str_new_cstr("D:/2.bmp"));
+	result     = rb_funcall(capture, rb_intern_const("find"), options == Qnil ? 1 : 2, image, options);
+	if (result != Qnil && *area != Qundef)
+		result = rb_funcall(result, rb_intern_const("move"), 2, x, y);
+	return result;
+}
+
+//endregion display
+
+//region input
+
+//region keyboard
 
 /*
  * @overload key_hold(key_code)
@@ -438,7 +583,7 @@ static VALUE ruby_color_at(VALUE self, VALUE x, VALUE y) {
  * @see https://msdn.microsoft.com/en-us/library/windows/desktop/ms646280.aspx Windows Dev Center - WM_KEYDOWN message
  */
 static VALUE ruby_key_hold(VALUE self, VALUE keyCode) {
-	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
+	SID_WINDOW_DATA *cData  = GET_SELF_DATA_OBJECT();
 	HWND            cHWnd   = ruby___window_get_handle_code(self);
 	WPARAM          cWParam = NUM2USHORT(keyCode);
 	LPARAM          cLParam = 1 | (WORD) MapVirtualKeyA(cWParam, MAPVK_VK_TO_VSC) << 16;
@@ -487,7 +632,7 @@ static VALUE ruby_key_hold(VALUE self, VALUE keyCode) {
  * @see https://msdn.microsoft.com/en-us/library/windows/desktop/ms646281.aspx Windows Dev Center - WM_KEYUP message
  */
 static VALUE ruby_key_release(VALUE self, VALUE keyCode) {
-	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
+	SID_WINDOW_DATA *cData  = GET_SELF_DATA_OBJECT();
 	HWND            cHWnd   = ruby___window_get_handle_code(self);
 	WPARAM          cWParam = NUM2USHORT(keyCode);
 	LPARAM          cLParam = 1;
@@ -539,7 +684,7 @@ static VALUE ruby_key_release(VALUE self, VALUE keyCode) {
  * @see https://msdn.microsoft.com/en-us/library/windows/desktop/ms646281.aspx Windows Dev Center - WM_KEYUP message
  */
 static VALUE ruby_key_press(VALUE self, VALUE keyCode) {
-	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
+	SID_WINDOW_DATA *cData  = GET_SELF_DATA_OBJECT();
 	HWND            cHWnd   = ruby___window_get_handle_code(self);
 	WPARAM          cWParam = NUM2USHORT(keyCode);
 	LPARAM          cLParam = 1 | (WORD) MapVirtualKeyA(cWParam, MAPVK_VK_TO_VSC) << 16;
@@ -580,11 +725,11 @@ static VALUE ruby_key_press(VALUE self, VALUE keyCode) {
 	return self;
 }
 
-//</editor-fold>
+//endregion keyboard
 
-//<editor-fold desc="mouse">
+//region mouse
 
-//<editor-fold desc="move">
+//region move
 
 /*
  * @overload mouse_move_relatively(x, y)
@@ -597,7 +742,7 @@ static VALUE ruby_key_press(VALUE self, VALUE keyCode) {
  */
 static VALUE ruby_mouse_move_relatively(VALUE self, VALUE x, VALUE y) {
 	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
-	HWND            cHWnd = ruby___window_get_handle_code(self);
+	HWND            cHWnd  = ruby___window_get_handle_code(self);
 	cData->x += NUM2SHORT(x);
 	cData->y += NUM2SHORT(y);
 	PostMessage(cHWnd, WM_MOUSEMOVE, createMouseMessageWParam(cData), createMouseMessageLParam(cData));
@@ -616,16 +761,16 @@ static VALUE ruby_mouse_move_relatively(VALUE self, VALUE x, VALUE y) {
  */
 static VALUE ruby_mouse_move_to(VALUE self, VALUE x, VALUE y) {
 	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
-	HWND            cHWnd = ruby___window_get_handle_code(self);
+	HWND            cHWnd  = ruby___window_get_handle_code(self);
 	cData->x = NUM2SHORT(x);
 	cData->y = NUM2SHORT(y);
 	PostMessage(cHWnd, WM_MOUSEMOVE, createMouseMessageWParam(cData), createMouseMessageLParam(cData));
 	return self;
 }
 
-//</editor-fold>
+//endregion move
 
-//<editor-fold desc="left_button">
+//region left_button
 
 /*
  * Simulate clicking the left button on the mouse in the window.
@@ -637,7 +782,7 @@ static VALUE ruby_mouse_move_to(VALUE self, VALUE x, VALUE y) {
  * @see https://msdn.microsoft.com/en-us/library/windows/desktop/ms645608.aspx Windows Dev Center - WM_LBUTTONUP message
  */
 static VALUE ruby_mouse_left_click(VALUE self) {
-	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
+	SID_WINDOW_DATA *cData       = GET_SELF_DATA_OBJECT();
 	ULONGLONG       cCurrentTime = getSystemTimeIdentifier();
 	HWND            cHWnd        = ruby___window_get_handle_code(self);
 	cData->flats |= FLAT_MOUSE_LEFT;
@@ -659,7 +804,7 @@ static VALUE ruby_mouse_left_click(VALUE self) {
  * @see https://msdn.microsoft.com/en-us/library/windows/desktop/ms645607.aspx Windows Dev Center - WM_LBUTTONDOWN message
  */
 static VALUE ruby_mouse_left_down(VALUE self) {
-	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
+	SID_WINDOW_DATA *cData       = GET_SELF_DATA_OBJECT();
 	ULONGLONG       cCurrentTime = getSystemTimeIdentifier();
 	HWND            cHWnd        = ruby___window_get_handle_code(self);
 	cData->flats |= FLAT_MOUSE_LEFT;
@@ -679,15 +824,15 @@ static VALUE ruby_mouse_left_down(VALUE self) {
  */
 static VALUE ruby_mouse_left_up(VALUE self) {
 	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
-	HWND            cHWnd = ruby___window_get_handle_code(self);
+	HWND            cHWnd  = ruby___window_get_handle_code(self);
 	cData->flats &= ~FLAT_MOUSE_LEFT;
 	PostMessage(cHWnd, WM_LBUTTONUP, createMouseMessageWParam(cData), createMouseMessageLParam(cData));
 	return self;
 }
 
-//</editor-fold>
+//endregion left_button
 
-//<editor-fold desc="right_button">
+//region right_button
 
 /*
  * Simulate clicking the right button on the mouse in the window.
@@ -699,7 +844,7 @@ static VALUE ruby_mouse_left_up(VALUE self) {
  * @see https://msdn.microsoft.com/en-us/library/windows/desktop/ms646243.aspx Windows Dev Center - WM_RBUTTONUP function
  */
 static VALUE ruby_mouse_right_click(VALUE self) {
-	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
+	SID_WINDOW_DATA *cData       = GET_SELF_DATA_OBJECT();
 	ULONGLONG       cCurrentTime = getSystemTimeIdentifier();
 	HWND            cHWnd        = ruby___window_get_handle_code(self);
 	cData->flats |= FLAT_MOUSE_RIGHT;
@@ -721,7 +866,7 @@ static VALUE ruby_mouse_right_click(VALUE self) {
  * @see https://msdn.microsoft.com/en-us/library/windows/desktop/ms646242.aspx Windows Dev Center - WM_RBUTTONDOWN function
  */
 static VALUE ruby_mouse_right_down(VALUE self) {
-	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
+	SID_WINDOW_DATA *cData       = GET_SELF_DATA_OBJECT();
 	ULONGLONG       cCurrentTime = getSystemTimeIdentifier();
 	HWND            cHWnd        = ruby___window_get_handle_code(self);
 	cData->flats |= FLAT_MOUSE_RIGHT;
@@ -741,15 +886,15 @@ static VALUE ruby_mouse_right_down(VALUE self) {
  */
 static VALUE ruby_mouse_right_up(VALUE self) {
 	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
-	HWND            cHWnd = ruby___window_get_handle_code(self);
+	HWND            cHWnd  = ruby___window_get_handle_code(self);
 	cData->flats &= ~FLAT_MOUSE_RIGHT;
 	PostMessage(cHWnd, WM_RBUTTONUP, createMouseMessageWParam(cData), createMouseMessageLParam(cData));
 	return self;
 }
 
-//</editor-fold>
+//endregion right_button
 
-//<editor-fold desc="middle_button">
+//region middle_button
 
 /*
  * Simulate clicking the middle button on the mouse in the window.
@@ -761,7 +906,7 @@ static VALUE ruby_mouse_right_up(VALUE self) {
  * @see https://msdn.microsoft.com/en-us/library/windows/desktop/ms645611.aspx Windows Dev Center - WM_MBUTTONUP function
  */
 static VALUE ruby_mouse_middle_click(VALUE self) {
-	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
+	SID_WINDOW_DATA *cData       = GET_SELF_DATA_OBJECT();
 	ULONGLONG       cCurrentTime = getSystemTimeIdentifier();
 	HWND            cHWnd        = ruby___window_get_handle_code(self);
 	cData->flats |= FLAT_MOUSE_MIDDLE;
@@ -783,7 +928,7 @@ static VALUE ruby_mouse_middle_click(VALUE self) {
  * @see https://msdn.microsoft.com/en-us/library/windows/desktop/ms645610.aspx Windows Dev Center - WM_MBUTTONDOWN function
  */
 static VALUE ruby_mouse_middle_down(VALUE self) {
-	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
+	SID_WINDOW_DATA *cData       = GET_SELF_DATA_OBJECT();
 	ULONGLONG       cCurrentTime = getSystemTimeIdentifier();
 	HWND            cHWnd        = ruby___window_get_handle_code(self);
 	cData->flats |= FLAT_MOUSE_MIDDLE;
@@ -803,15 +948,15 @@ static VALUE ruby_mouse_middle_down(VALUE self) {
  */
 static VALUE ruby_mouse_middle_up(VALUE self) {
 	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
-	HWND            cHWnd = ruby___window_get_handle_code(self);
+	HWND            cHWnd  = ruby___window_get_handle_code(self);
 	cData->flats &= ~FLAT_MOUSE_MIDDLE;
 	PostMessage(cHWnd, WM_MBUTTONUP, createMouseMessageWParam(cData), createMouseMessageLParam(cData));
 	return self;
 }
 
-//</editor-fold>
+//endregion middle_button
 
-//<editor-fold desc="x_button">
+//region x_button
 
 /*
  * Simulate clicking the x1 button on the mouse in the window.
@@ -823,7 +968,7 @@ static VALUE ruby_mouse_middle_up(VALUE self) {
  * @see https://msdn.microsoft.com/en-us/library/windows/desktop/ms646246.aspx Windows Dev Center - WM_XBUTTONUP function
  */
 static VALUE ruby_mouse_x1_click(VALUE self) {
-	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
+	SID_WINDOW_DATA *cData       = GET_SELF_DATA_OBJECT();
 	ULONGLONG       cCurrentTime = getSystemTimeIdentifier();
 	HWND            cHWnd        = ruby___window_get_handle_code(self);
 	cData->flats |= FLAT_MOUSE_X1;
@@ -847,7 +992,7 @@ static VALUE ruby_mouse_x1_click(VALUE self) {
  * @see https://msdn.microsoft.com/en-us/library/windows/desktop/ms646245.aspx Windows Dev Center - WM_XBUTTONDOWN function
  */
 static VALUE ruby_mouse_x1_down(VALUE self) {
-	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
+	SID_WINDOW_DATA *cData       = GET_SELF_DATA_OBJECT();
 	ULONGLONG       cCurrentTime = getSystemTimeIdentifier();
 	HWND            cHWnd        = ruby___window_get_handle_code(self);
 	cData->flats |= FLAT_MOUSE_X1;
@@ -869,7 +1014,7 @@ static VALUE ruby_mouse_x1_down(VALUE self) {
  */
 static VALUE ruby_mouse_x1_up(VALUE self) {
 	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
-	HWND            cHWnd = ruby___window_get_handle_code(self);
+	HWND            cHWnd  = ruby___window_get_handle_code(self);
 	cData->flats &= ~FLAT_MOUSE_X1;
 	PostMessage(cHWnd, WM_XBUTTONUP, createMouseMessageWParam(cData) | XBUTTON1 << 16, createMouseMessageLParam(cData));
 	return self;
@@ -885,7 +1030,7 @@ static VALUE ruby_mouse_x1_up(VALUE self) {
  * @see https://msdn.microsoft.com/en-us/library/windows/desktop/ms646246.aspx Windows Dev Center - WM_XBUTTONUP function
  */
 static VALUE ruby_mouse_x2_click(VALUE self) {
-	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
+	SID_WINDOW_DATA *cData       = GET_SELF_DATA_OBJECT();
 	ULONGLONG       cCurrentTime = getSystemTimeIdentifier();
 	HWND            cHWnd        = ruby___window_get_handle_code(self);
 	cData->flats |= FLAT_MOUSE_X2;
@@ -909,7 +1054,7 @@ static VALUE ruby_mouse_x2_click(VALUE self) {
  * @see https://msdn.microsoft.com/en-us/library/windows/desktop/ms646245.aspx Windows Dev Center - WM_XBUTTONDOWN function
  */
 static VALUE ruby_mouse_x2_down(VALUE self) {
-	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
+	SID_WINDOW_DATA *cData       = GET_SELF_DATA_OBJECT();
 	ULONGLONG       cCurrentTime = getSystemTimeIdentifier();
 	HWND            cHWnd        = ruby___window_get_handle_code(self);
 	cData->flats |= FLAT_MOUSE_X2;
@@ -931,15 +1076,15 @@ static VALUE ruby_mouse_x2_down(VALUE self) {
  */
 static VALUE ruby_mouse_x2_up(VALUE self) {
 	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
-	HWND            cHWnd = ruby___window_get_handle_code(self);
+	HWND            cHWnd  = ruby___window_get_handle_code(self);
 	cData->flats &= ~FLAT_MOUSE_X2;
 	PostMessage(cHWnd, WM_XBUTTONUP, createMouseMessageWParam(cData) | XBUTTON2 << 16, createMouseMessageLParam(cData));
 	return self;
 }
 
-//</editor-fold>
+//endregion x_button
 
-//<editor-fold desc="wheel">
+//region wheel
 
 /*
  * Simulate wheeling the scroll backward in the window.
@@ -950,7 +1095,7 @@ static VALUE ruby_mouse_x2_up(VALUE self) {
  */
 static VALUE ruby_mouse_scroll_wheel_backward(VALUE self) {
 	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
-	HWND            cHWnd = ruby___window_get_handle_code(self);
+	HWND            cHWnd  = ruby___window_get_handle_code(self);
 	PostMessage(cHWnd, WM_MOUSEWHEEL, createMouseMessageWParam(cData) | -WHEEL_DELTA << 16,
 	            createMouseMessageLParam(cData));
 	return self;
@@ -965,7 +1110,7 @@ static VALUE ruby_mouse_scroll_wheel_backward(VALUE self) {
  */
 static VALUE ruby_mouse_scroll_wheel_forward(VALUE self) {
 	SID_WINDOW_DATA *cData = GET_SELF_DATA_OBJECT();
-	HWND            cHWnd = ruby___window_get_handle_code(self);
+	HWND            cHWnd  = ruby___window_get_handle_code(self);
 	PostMessage(cHWnd, WM_MOUSEWHEEL, createMouseMessageWParam(cData) | WHEEL_DELTA << 16,
 	            createMouseMessageLParam(cData));
 	return self;
@@ -1007,11 +1152,11 @@ static VALUE ruby_mouse_scroll_wheel_to_right(VALUE self) {
 #endif
 }
 
-//</editor-fold>
+//endregion wheel
 
-//</editor-fold>
+//endregion mouse
 
-//</editor-fold>
+//endregion input
 
 VOID Init_window() {
 #ifdef YARD
@@ -1058,6 +1203,8 @@ VOID Init_window() {
 	rb_define_method(specialInputDevice_window, "resize_to", ruby_resize_to, 2);
 	rb_define_method(specialInputDevice_window, "restore", ruby_restore, 0);
 	rb_define_method(specialInputDevice_window, "color_at", ruby_color_at, 2);
+	rb_define_method(specialInputDevice_window, "capture", ruby_capture, -1);
+	rb_define_method(specialInputDevice_window, "find_image", ruby_find_image, -1);
 	rb_define_method(specialInputDevice_window, "key_hold", ruby_key_hold, 1);
 	rb_define_method(specialInputDevice_window, "key_release", ruby_key_release, 1);
 	rb_define_method(specialInputDevice_window, "key_press", ruby_key_press, 1);

data/lib/special_input_device/color.rb

@@ -6,28 +6,28 @@ class SpecialInputDevice::Color
   include(SpecialInputDevice::AttributesEqualChecker)
 
   # @return [Fixnum] the level of alpha
-  attr_reader :alpha
+  attr_reader(:alpha)
   # @param [Fixnum] value the level of alpha
   def alpha=(value)
     @alpha = [value.to_i, 0, 255].sort[1];
   end
 
   # @return [Fixnum] the level of blue
-  attr_reader :blue
+  attr_reader(:blue)
   # @param [Fixnum] value the level of blue
   def blue=(value)
     @blue = [value.to_i, 0, 255].sort[1];
   end
 
   # @return [Fixnum] the level of green
-  attr_reader :green
+  attr_reader(:green)
   # @param [Fixnum] value the level of green
   def green=(value)
     @green = [value.to_i, 0, 255].sort[1];
   end
 
   # @return [Fixnum] the level of red
-  attr_reader :red
+  attr_reader(:red)
   # @param [Fixnum] value the level of red
   def red=(value)
     @red = [value.to_i, 0, 255].sort[1];

data/lib/special_input_device/image.rb

@@ -35,7 +35,7 @@ class SpecialInputDevice::Image < Data
   end
 
   # Find the position of another smaller image in this image.
-  # @param [SpecialInputDevice::Image] image another image, should be smaller
+  # @param [SpecialInputDevice::Image, String] image another image, should be smaller
   # @param [Float] similarity minimum limit of similarity
   # @param [Symbol] _return indicate the behavior of this method, see SpecialInputDevice::Image::RETURN_METHOD
   # @return [SpecialInputDevice::Rectangle, NilClass] the position
@@ -43,6 +43,7 @@ class SpecialInputDevice::Image < Data
   def find(image, similarity: 1.0, _return: RETURN_METHOD::FIRST)
     methods = RETURN_METHOD.constants(false).map { |x| RETURN_METHOD.const_get(x) }
     raise("Undefined return method '#{_return.to_s}' passed, Defined methods are [#{methods.join(', ')}].") unless methods.include?(_return)
+    image = Image.load(image) if image.is_a?(String)
     closest_similarity   = 0.0
     closest_similarity_x = nil
     closest_similarity_y = nil

data/lib/special_input_device/point.rb

@@ -6,14 +6,14 @@ class SpecialInputDevice::Point
   include(SpecialInputDevice::AttributesEqualChecker)
 
   # @return [Integer] the x-coordinate of the point
-  attr_reader :x
+  attr_reader(:x)
   # @param [Integer] value the x-coordinate of the point
   def x=(value)
     @x = value.to_i
   end
 
   # @return [Integer] the y-coordinate of the point
-  attr_reader :y
+  attr_reader(:y)
   # @param [Integer] value the y-coordinate of the point
   def y=(value)
     @y = value.to_i