Question

我是一个业余爱好程序员，并尝试使用python在Ubuntu中制作屏幕录像机。使用此代码可以截取屏幕截图。

import wx
app = wx.App(False)
s = wx.ScreenDC()
w, h = s.Size.Get()
b = wx.EmptyBitmap(w, h)
m = wx.MemoryDCFromDC(s)
m.SelectObject(b)
m.Blit(0, 0, w, h, s, 0, 0)
m.SelectObject(wx.NullBitmap)
b.SaveFile("screenshot.png", wx.BITMAP_TYPE_PNG)

使用循环我拍摄更多照片并使用这些屏幕截图创建视频。我的代码如下：

import wx,os
app=wx.App(False)
s = wx.ScreenDC()
w, h = s.Size.Get()
b = wx.EmptyBitmap(w, h)
m = wx.MemoryDCFromDC(s)
i=0
while i<50:
   m.SelectObject(b)
   m.Blit(0, 0, w, h, s, 0, 0)
   m.SelectObject(wx.NullBitmap)
   b.SaveFile('{0:05d}.png'.format(i), wx.BITMAP_TYPE_PNG)
   i+=1
os.system('ffmpeg -f image2 -r 8 -i %05d.png -vcodec mpeg4 -y movie1.mp4')
i=0  
while i<50:
   os.remove('{0:05d}.png'.format(i))
   i += 1   `

在上面的代码中，我拍摄了50张照片并将其存储为00000.png到00049.png并使用ffmpeg制作视频。创建视频后，我删除了所有图片。

当前问题：

屏幕拍摄之间的延迟非常小。如果尝试使用此代码录制视频，则输出并不完美。
长时间录音效率不高。存储屏幕拍摄需要大量的硬盘内存。并使用更多的CPU。

我做了什么让代码更有效率？使用纯python如何从图片创建视频？有没有其他方法来记录屏幕？我喜欢改进我的代码。

Answer 1

您编写代码的方式，在获取下一个屏幕截图之前，必须等待每个文件保存。这就是你的“非常小的延迟”来自的地方。

你可以抓住内存中的所有快照，然后在最后写下它们：

snapshots = []
for i in range(20):
   b = wx.EmptyBitmap(w, h)
   m.SelectObject(b)
   m.Blit(0, 0, w, h, s, 0, 0)
   m.SelectObject(wx.NullBitmap)
   snapshots.append(b)
for snapshot in snapshots:
   snapshot.SaveFile('{0:05d}.png'.format(i), wx.BITMAP_TYPE_PNG)

但这会占用大量内存 - 可能足以让malloc次调用（或者更糟糕的是，交换thrash）让你慢下来。

另一种选择是将写入推送到后台线程。（由于工作可能是I / O绑定的，因此普通的Python线程应该没问题。）例如：

with concurrent.futures.ThreadPoolExecutor() as executor:
    for i in range(20):
       b = wx.EmptyBitmap(w, h)
       m.SelectObject(b)
       m.Blit(0, 0, w, h, s, 0, 0)
       m.SelectObject(wx.NullBitmap)
       executor.submit(b.SaveFile, '{0:05d}.png'.format(i), wx.BITMAP_TYPE_PNG)

但不管怎样，如果你解决这个问题......你不希望快照尽快出现。例如，如果您的屏幕刷新，比如说，60次/秒，并且您的20个循环都在前1/60秒内完成，那么您只需获得相同帧的20个副本。那不好。所以，你真的需要为快照编写一些简单的调度程序。（通常，屏幕截图实际上比屏幕刷新率 - 1/20秒或甚至1/6或1/2更慢。）

幸运的是，wx有一些很好的方法可以做到这一点。例如，使用wx.Timer每1000/60毫秒运行一次代码。

这样做有一个额外的好处，就是你在拍摄快照时不会阻挡整个事件循环，只是每隔1/60（或1/20或任何一秒）暂停一次。

当然，您可以并且可能必须将这两种解决方案结合起来：使用计时器或其他调度程序来设置最大帧速率，并卸载一些工作以尽可能接近尽可能最大。

至于处理更长时间的屏幕抓取：如果你试图将数万个帧保存为PNG文件，那将占用大量空间。（为什么？嗯，PNG使用无损压缩 - 它可以让你的4MB屏幕下降到400K。但H264视频使用有损压缩，更重要的是，只需要跟踪从一帧到下一帧的变化，所以它可能只每个关键帧需要100K，每个差异帧需要4K。）

你可以做的一件事就是每隔一段时间（例如，每120帧）启动一项工作，将它们压缩成一个视频并删除它们，最后将所有视频连接起来。为简单起见，您可能希望每个批处理的第一帧成为新的关键帧。 MP4 / H264并不是最有趣的连接格式，但它仍然可能比切换到流式压缩器更容易。

使用python在Ubuntu中的屏幕录像机

1 个答案: