win32screenshot 0.0.1 → 0.0.2

data/History.txt

@@ -0,0 +1,7 @@
+= 0.0.2 (trunk)
+* Added desktop method (patch from Ryan Schuft)
+* Added HTTP server example (patch from Ryan Schuft)
+* Added window(regexp) method
+
+= 0.0.1 2006-11-29
+* First release

data/lib/win32screenshot.rb

@@ -9,6 +9,11 @@ module Win32
 
     dlload "kernel32.dll","user32.dll","gdi32.dll"
 
+    USER32 = DL.dlopen("user32")
+    EnumWindows = USER32['EnumWindows', 'IPL']
+    GetWindowTextLength = USER32['GetWindowTextLengthA' ,'LI' ]
+    GetWindowText = USER32['GetWindowTextA', 'iLsL' ] 
+
     SRCCOPY = 0xCC0020
     GMEM_FIXED = 0
     DIB_RGB_COLORS = 0
@@ -17,6 +22,7 @@ module Win32
     typealias "LPRECT","unsigned int*"
 
     extern "HWND GetForegroundWindow()"
+    extern "HWND GetDesktopWindow()"
     extern "BOOL GetWindowRect(HWND, LPRECT)"
     extern "BOOL GetClientRect(HWND, LPRECT)"
     extern "HDC GetDC(HWND)"
@@ -34,15 +40,10 @@ module Win32
     extern "long DeleteObject(unsigned long)"
     extern "long DeleteDC(HDC)"
     extern "long ReleaseDC(long, HDC)"
+    extern "BOOL SetForegroundWindow(HWND)"
 
     module_function
-    def capture(hwnd)
-      hScreenDC = getDC(hwnd)
-
-      # Find the dimensions of the window
-      rect = DL.malloc(DL.sizeof('LLLL'))
-      getClientRect(hwnd, rect)
-      x1, y1, x2, y2 = rect.to_a('LLLL')
+    def capture(hScreenDC, x1, y1, x2, y2)
       w = x2-x1
       h = y2-y1
 
@@ -78,10 +79,52 @@ module Win32
  
       return [w, h, data]
     end
+
+    def capture_hwnd(hwnd)
+      hScreenDC = getDC(hwnd)
+
+      # Find the dimensions of the window
+      rect = DL.malloc(DL.sizeof('LLLL'))
+      getClientRect(hwnd, rect)
+      x1, y1, x2, y2 = rect.to_a('LLLL')
+
+      capture(hScreenDC, x1, y1, x2, y2)
+    end
   
     module_function
     def foreground
-      capture(getForegroundWindow())
+      hwnd = getForegroundWindow
+      capture_hwnd(hwnd)
+    end
+
+    module_function
+    def desktop
+      hwnd = getDesktopWindow
+      capture_hwnd(hwnd)
+    end
+
+    module_function
+    def window(title_query, delay=0.1)
+      hwnd = nil
+
+      proc = DL.callback('ILL') do |curr_hwnd, lparam|
+        textLength, a = GetWindowTextLength.call(curr_hwnd)
+        captionBuffer = " " * (textLength+1)
+        t, textCaption = GetWindowText.call(curr_hwnd, captionBuffer, textLength+1)    
+        text = textCaption[1].to_s
+        if text =~ title_query
+          hwnd = curr_hwnd
+          0
+        else
+          1
+        end
+      end
+      EnumWindows.call(proc, 0)
+
+      raise "Couldn't find window with title matching #{title_query}" if hwnd.nil?
+      setForegroundWindow(hwnd)
+      sleep(delay)
+      capture_hwnd(hwnd)
     end
   end
 end

data/lib/win32screenshot/version.rb

@@ -2,7 +2,7 @@ module Win32screenshot #:nodoc:
   module VERSION #:nodoc:
     MAJOR = 0
     MINOR = 0
-    TINY  = 1
+    TINY  = 2
 
     STRING = [MAJOR, MINOR, TINY].join('.')
   end

data/test/win32screenshot_test.rb

@@ -1,23 +1,43 @@
 require File.dirname(__FILE__) + '/test_helper.rb'
+require 'rubygems'
+require 'RMagick'
 
+# Prereqs for this test:
+# * Must be run from a command window with dimensions 1600x1200 pixels (160x60 characters, default font)
+# * Screen res must be 1600x1200
+# * Must have a maximised Firefox open
+#
 class Win32screenshotTest < Test::Unit::TestCase
 
-  def test_should_create_bmp
+  def test_should_capture_foreground
     width, height, bmp = Win32::Screenshot.foreground
     assert_equal 1280, width
     assert_equal 720,  height
-    assert_equal 'BM',  bmp[0..1]
+    assert_image(bmp)
   end
   
-  def test_should_be_convertible_to_png
-    require 'rubygems'
-    require 'RMagick'
+  def test_should_capture_desktop
+    width, height, bmp = Win32::Screenshot.desktop
+    assert_equal 1600, width
+    assert_equal 1200,  height
+    assert_image(bmp)
+  end
 
-    width, height, bmp = Win32::Screenshot.foreground
+  def test_should_capture_window_by_title
+    width, height, bmp = Win32::Screenshot.window(/Firefox/)
+    assert_equal 1600, width
+    assert_equal 1147,  height
+    assert_image(bmp, "ff.png")
+  end
+  
+
+  def assert_image(bmp, file=nil)
+    assert_equal 'BM',  bmp[0..1]
     img = Magick::Image.from_blob(bmp)
     png = img[0].to_blob do
       self.format = 'PNG'
     end
     assert_equal "\211PNG",  png[0..3]
+    File.open(file, "wb") {|io| io.write(png)} unless file.nil?
   end
 end

metadata

@@ -3,8 +3,8 @@ rubygems_version: 0.9.0
 specification_version: 1
 name: win32screenshot
 version: !ruby/object:Gem::Version 
-  version: 0.0.1
-date: 2006-11-29 00:00:00 +01:00
+  version: 0.0.2
+date: 2006-12-02 00:00:00 +01:00
 summary: Capture Screenshots on Windows with Ruby
 require_paths: 
 - lib