extra/cap/cap.factor

   1 ! Copyright (C) 2008 Doug Coleman, Joe Groff.
   2 ! See http://factorcode.org/license.txt for BSD license.
   3 USING: accessors arrays byte-arrays kernel math namespaces
   4 opengl.gl sequences math.vectors ui graphics.bitmap graphics.viewer
   5 models ui.gadgets.worlds ui.gadgets fry alien.syntax ;
   6 IN: cap
   7
   8 : screenshot-array ( world -- byte-array )
   9     dim>> [ first 3 * 4 align ] [ second ] bi * <byte-array> ;
  10
  11 : gl-screenshot ( gadget -- byte-array )
  12     [
  13         GL_BACK glReadBuffer
  14         GL_PACK_ALIGNMENT 4 glPixelStorei
  15         0 0
  16     ] dip
  17     [ dim>> first2 GL_BGR GL_UNSIGNED_BYTE ]
  18     [ screenshot-array ] bi
  19     [ glReadPixels ] keep ;
  20
  21 : screenshot ( window -- bitmap )
  22     [ gl-screenshot ]
  23     [ dim>> first2 ] bi
  24     bgr>bitmap ;
  25
  26 : save-screenshot ( window path -- )
  27     [ screenshot ] dip save-bitmap ;
  28
  29 : screenshot. ( window -- )
  30     [ screenshot <graphics-gadget> ] [ title>> ] bi open-window ;