Question

我正试图在Ubuntu 10.04下使用Python截取部分屏幕截图。

这是我的代码（假设IMAGE_GRAB为假）：

def screenshot_roi(regions):
    if IMAGE_GRAB:
        return map(ImageGrab.grab, regions)
    else:
        w = gtk.gdk.get_default_root_window()
        sz = w.get_size()
        pb = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB,False,8,sz[0],sz[1])
        src = pb.get_from_drawable(w,w.get_colormap(),0,0,0,0,sz[0],sz[1])
        results = []
        for roi in regions:
            if not roi:
                results.append(None)
                continue
            x,y,width,height = roi
            dst = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB,False,8,width,height)
            src.copy_area(x,y,width,height,dst,0,0)
            im = PIL.Image.fromstring("RGB", (width, height), dst.get_pixels())
            results.append(im)
        return results

这里没什么好看的。将整个drawable捕获到像素缓冲区中，然后在将每个像素缓冲区转换为所需的PIL对象之前继续裁剪每个像素缓冲区。

这是主线：

def main():
    regions = [(845, 219, 248, 82), (1101, 243, 109, 59), 
               (1213, 245, 66, 57), (1281, 245, 74, 58)]    
    images = screenshot_roi(regions)
    for i,roi in enumerate(images):
        if roi:
            roi.save('%d.png' % i)

if __name__ == '__main__':
    main()

结果图像（除第一个之外的所有图像）都存在大步问题：

1： enter image description here 2： 3： 4：

现在，如果我在PIL中进行裁剪，一切运作良好：

def screenshot_roi(regions):
    if IMAGE_GRAB:
        return map(ImageGrab.grab, regions)
    else:
        w = gtk.gdk.get_default_root_window()
        sz = w.get_size()
        pb = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB,False,8,sz[0],sz[1])
        src = pb.get_from_drawable(w,w.get_colormap(),0,0,0,0,sz[0],sz[1])
        entire_im = PIL.Image.fromstring("RGB", sz, src.get_pixels())
        results = []
        for roi in regions:
            if not roi:
                results.append(None)
                continue
            x,y,width,height = roi
            crop = entire_im.crop((x,y,x+width,y+height))
            crop.load()
            results.append(crop)
        return results

我想不要这样做，因为从GTK到PIL的转换非常昂贵。这是一种浪费，因为我正在转换整个图像只是为了从中得到小的子区域。

有人可以建议为什么GTK版本有大步错误吗？

修改的

工作来源：

x,y,w,h = region
win = gtk.gdk.get_default_root_window()
pb = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB,False,8,w,h)
src = pb.get_from_drawable(win,win.get_colormap(),x,y,0,0,w,h)
im = PIL.Image.frombuffer('RGB', (w,h), src.get_pixels(), 'raw', 
                'RGB', src.get_rowstride(), 1)
return im

Answer 1

我认为这是因为get_pixels会将数据存储在内存中。你在这里缺少的是，由于性能原因（由于内存对齐限制），一行可能在其末尾有一些填充。

参见GdkPixbuf structure：有趣的信息是rowstride，它是行开头和下一行开头之间的字节数。它等于一行中的数据字节数+填充字节数。

请注意最后一行没有填充。

使用GTK的屏幕子区域的屏幕截图

1 个答案: