Question

我正在尝试使用ffmpeg录制VTK 3D动画的视频。我发现http://www.itk.org/Wiki/VTK/Examples/Cxx/Video/FFMPEG的例子对我来说有点奇怪，因为它不会创建渲染，因此在这种情况下如何附加actor并不清楚。那么，我的问题是如何将vtkRenderer连接到vtkFFMPEGWriter ？我写了一个简化的例子

import vtk

# draw a plane
plane_source = vtk.vtkPlaneSource()
plane_source.SetCenter(0, 0, 0)
plane_source.SetNormal(0, 0, 1)
plane_mapper = vtk.vtkPolyDataMapper()
plane_mapper.SetInput(plane_source.GetOutput())
plane_actor = vtk.vtkActor()
plane_actor.SetMapper(plane_mapper)

# create render
render = vtk.vtkRenderer()
renWin = vtk.vtkRenderWindow()
renWin.AddRenderer(render)
renWin.SetSize(800, 600)
renInter = vtk.vtkRenderWindowInteractor()
renInter.SetRenderWindow(renWin)
render.AddActor(plane_actor)
renWin.Render()
renInter.Initialize()

# set callback
def cb(interactor, event):
    global plane_actor
    plane_actor.RotateZ(0.1)
    interactor.GetRenderWindow().Render()

renInter.AddObserver('TimerEvent', cb)
timerId = renInter.CreateRepeatingTimer(100);

# put something here?
writer = vtk.vtkFFMPEGWriter()
# ??? writer.SetInputConnection(...)
writer.SetFileName("test.avi")
writer.Start()

# run
renInter.Start()

Answer 1

使用vtkWindowToImageFilter（description here）可以轻松解决问题：

windowToImageFilter = vtk.vtkWindowToImageFilter()
windowToImageFilter.SetInput(renWin)
windowToImageFilter.SetInputBufferTypeToRGBA()
windowToImageFilter.ReadFrontBufferOff()
windowToImageFilter.Update()

writer = vtk.vtkAVIWriter()
writer.SetInputConnection(windowToImageFilter.GetOutputPort())
writer.SetFileName("test.avi")
...

Answer 2

使用vtkWindowToImageFilter。当新帧准备就绪时，您必须告诉两者电影作家和图像过滤器。此外，一些电影作者需要显式调用end才能正确完成文件。下面的例子显示了整个生命周期。

#Setup filter
renWin = vtk.vtkRenderWindow()
imageFilter = vtk.vtkWindowToImageFilter()
imageFilter.SetInput(renWin)
imageFilter.SetInputBufferTypeToRGB()
imageFilter.ReadFrontBufferOff()
imageFilter.Update()

#Setup movie writer
moviewriter = vtk.vtkOggTheoraWriter() 
moviewriter.SetInputConnection(windowToImageFilter.GetOutputPort())
moviewriter.SetFileName("movie.ogv")
moviewriter.Start()
renWin.start()
...

#Export a single frame
imageFilter.Modified()
moviewriter.Write()
...

#Finish movie
moviewriter.End()

导出帧的对联可以根据需要多次调用以完成电影。

使用SetInputBufferTypeToRGB已解决的问题仅出现在我的某个安装中。使用SetInputBufferTypeToRGBA，我看到了条带和多个部分副本。使用SetInputBufferTypeToRGB它可以在我尝试的每台机器上运行。

该示例使用OggTheora编码器，因为默认情况下它在许多平台上都可用。

Python VTK，如何录制视频？

2 个答案: