#1559

- Adds `hs.image:colorAt(point)` - Closes #1559
Hammerspoon · Jun 14, 2018 · 596a280 · 596a280 · chdiza · Jun 15, 2018
1 parent bf43dd7
commit 596a280
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/extensions/image/internal.m b/extensions/image/internal.m
@@ -1272,6 +1272,31 @@ static int getImageSize(lua_State* L) {
     return 1 ;
 }
 
+/// hs.image:colorAt(point) -> table
+/// Method
+/// Reads the color of the pixel at the specified location.
+///
+/// Parameters:
+///  * `point` - a `hs.geometry.point`
+///
+/// Returns:
+///  * A `hs.drawing.color` object
+static int colorAt(lua_State* L) {
+    LuaSkin *skin = [LuaSkin shared] ;
+    [skin checkArgs:LS_TUSERDATA, USERDATA_TAG, LS_TTABLE, LS_TBREAK] ;
+    NSImage *theImage = [skin luaObjectAtIndex:1 toClass:"NSImage"] ;
+
+    NSPoint point  = [skin tableToPointAtIndex:2] ;
+
+	// Source: https://stackoverflow.com/a/48400410
+    [theImage lockFocus];
+	NSColor *pixelColor = NSReadPixel(point);
+	[theImage unlockFocus];
+
+    [skin pushNSObject:pixelColor];
+    return 1;
+}
+
 /// hs.image:croppedCopy(rectangle) -> object
 /// Method
 /// Returns a copy of the portion of the image specified by the rectangle specified.
@@ -1624,6 +1649,7 @@ static int userdata_gc(lua_State* L) {
     {"croppedCopy",       croppedCopy},
     {"saveToFile",        saveToFile},
     {"encodeAsURLString", encodeAsString},
+    {"colorAt",			  colorAt},
 
     {"__tostring",        userdata_tostring},
     {"__eq",              userdata_eq},