factor/extra/cap/cap.factor

USING: accessors arrays byte-arrays kernel math namespaces
opengl.gl sequences math.vectors ui graphics.bitmap graphics.viewer
models opengl.framebuffers ui.gadgets.worlds ui.gadgets fry ;
IN: cap

: screenshot-array ( world -- byte-array )
    dim>> product 3 * <byte-array> ;

: gl-screenshot ( gadget -- byte-array )
    [
        GL_BACK glReadBuffer
        GL_PACK_ALIGNMENT 1 glPixelStorei
        0 0
    ] dip
    [ dim>> first2 GL_BGR GL_UNSIGNED_BYTE ]
    [ screenshot-array ] bi
    [ glReadPixels ] keep ;

: screenshot ( window -- bitmap )
    [ gl-screenshot ]
    [ dim>> first2 ] bi
    bgr>bitmap ;

: save-screenshot ( window path -- )
    [ screenshot ] dip save-bitmap ;

: screenshot. ( window -- )
    [ screenshot <graphics-gadget> ] [ title>> ] bi open-window ;