def record(window, filename, fps):
'''
Record an animated window to a movie file.
:param window: window that is being recorded.
:param filename: name of the file to write the movie to.
:param fps: framerate at which to record.
'''
writer = FFMPEG_VideoWriter(filename, (window.width, window.height), fps=fps)
fbuffer = np.zeros((window.height, window.width, 3), dtype=np.uint8)
# Function to write one frame
def write_frame(writer):
gl.glReadPixels(0, 0, window.width, window.height,
gl.GL_RGB, gl.GL_UNSIGNED_BYTE, fbuffer)
writer.write_frame(np.flipud(fbuffer))
# Modify the on_draw event handler to also write a
# movie frame every time it is called
old_on_draw = window.get_handler('on_draw')
@window.event
def on_draw(dt):
old_on_draw(dt)
write_frame(writer)
yield
writer.close()
def run(clock=None,
framerate=None,
interactive=None,
duration=sys.maxint,
framecount=sys.maxint):
""" Run the main loop
Parameters
----------
clock : Clock
clock to use to run the app (gives the elementary tick)
framerate : int
frames per second
duration : float
Duration after which the app will be stopped
framecount : int
Number of frame to display before stopping.
"""
global __running__
clock = __init__(clock=clock, framerate=framerate, backend=__backend__)
options = parser.get_options()
if interactive is None:
interactive = options.interactive
writer = None
if options.record:
from glumpy.ext.ffmpeg_writer import FFMPEG_VideoWriter
framerate = 60
window = __backend__.windows()[0]
width, height = window.width, window.height
filename = "movie.mp4"
writer = FFMPEG_VideoWriter(filename, (width, height), fps=framerate)
fbuffer = np.zeros((height, width, 3), dtype=np.uint8)
data = fbuffer.copy()
log.info("Recording movie in '%s'" % filename)
if interactive:
# Set interactive python session
os.environ['PYTHONINSPECT'] = '1'
import readline
readline.parse_and_bind("tab: complete")
def run():
while not stdin_ready():
__backend__.process(clock.tick())
return 0
inputhook_manager.set_inputhook(run)
else:
__running__ = True
count = len(__backend__.windows())
while count and duration > 0 and framecount > 0 and __running__:
dt = clock.tick()
duration -= dt
framecount -= 1
count = __backend__.process(dt)
# Record one frame (if there is writer available)
if writer is not None:
gl.glReadPixels(0, 0, window.width, window.height, gl.GL_RGB,
gl.GL_UNSIGNED_BYTE, fbuffer)
data[...] = fbuffer[::-1, :, :]
writer.write_frame(data)
if writer is not None:
writer.close()
def run(clock=None, framerate=None, interactive=None,
duration = sys.maxint, framecount = sys.maxint):
""" Run the main loop
Parameters
----------
clock : Clock
clock to use to run the app (gives the elementary tick)
framerate : int
frames per second
duration : float
Duration after which the app will be stopped
framecount : int
Number of frame to display before stopping.
"""
global __running__
clock = __init__(clock=clock, framerate=framerate, backend=__backend__)
options = parser.get_options()
if interactive is None:
interactive = options.interactive
writer = None
if options.record:
from glumpy.ext.ffmpeg_writer import FFMPEG_VideoWriter
framerate = 60
window = __backend__.windows()[0]
width, height = window.width, window.height
filename = "movie.mp4"
writer = FFMPEG_VideoWriter(filename, (width, height), fps=framerate)
fbuffer = np.zeros((height, width, 3), dtype=np.uint8)
data = fbuffer.copy()
log.info("Recording movie in '%s'" % filename)
if interactive:
# Set interactive python session
os.environ['PYTHONINSPECT'] = '1'
import readline
readline.parse_and_bind("tab: complete")
def run():
while not stdin_ready():
__backend__.process(clock.tick())
return 0
inputhook_manager.set_inputhook(run)
else:
__running__ = True
count = len(__backend__.windows())
while count and duration > 0 and framecount > 0 and __running__:
dt = clock.tick()
duration -= dt
framecount -= 1
count = __backend__.process(dt)
# Record one frame (if there is writer available)
if writer is not None:
gl.glReadPixels(0, 0, window.width, window.height,
gl.GL_RGB, gl.GL_UNSIGNED_BYTE, fbuffer)
data[...] = fbuffer[::-1,:,:]
writer.write_frame(data)
if writer is not None:
writer.close()