mirror of
https://github.com/stevenrobertson/cuburn.git
synced 2025-03-13 06:51:28 -04:00
Drop GL mode in main.py; sleep to reduce load
This commit is contained in:
Steven Robertson
parent
7dc58a0e1c
commit
636efcd059
154
main.py
154
main.py
@ -25,78 +25,6 @@ sys.path.insert(0, os.path.dirname(__file__))
|
|||||||
from cuburn import render, filters, output, profile
|
from cuburn import render, filters, output, profile
|
||||||
from cuburn.genome import convert, use, db
|
from cuburn.genome import convert, use, db
|
||||||
|
|
||||||
def save(output_module, name, rendered_frame):
|
|
||||||
out, log = output_module.encode(rendered_frame)
|
|
||||||
for suffix, file_like in out.items():
|
|
||||||
with open(name + suffix, 'w') as fp:
|
|
||||||
fp.write(file_like.read())
|
|
||||||
if getattr(file_like, 'close', None):
|
|
||||||
file_like.close()
|
|
||||||
for key, val in log:
|
|
||||||
print '\n=== %s ===' % key
|
|
||||||
print val
|
|
||||||
|
|
||||||
def pyglet_preview(args, gprof, itr):
|
|
||||||
import pyglet
|
|
||||||
import pyglet.gl as gl
|
|
||||||
w, h = gprof.width, gprof.height
|
|
||||||
window = pyglet.window.Window(w, h, vsync=False)
|
|
||||||
image = pyglet.image.CheckerImagePattern().create_image(w, h)
|
|
||||||
tex = image.texture
|
|
||||||
label = pyglet.text.Label('Rendering first frame', x=5, y=h-5,
|
|
||||||
width=w, anchor_y='top', font_size=16,
|
|
||||||
bold=True, multiline=True)
|
|
||||||
|
|
||||||
@window.event
|
|
||||||
def on_draw():
|
|
||||||
window.clear()
|
|
||||||
tex.blit(0, 0, 0)
|
|
||||||
label.draw()
|
|
||||||
|
|
||||||
@window.event
|
|
||||||
def on_key_press(sym, mod):
|
|
||||||
if sym == pyglet.window.key.Q:
|
|
||||||
pyglet.app.exit()
|
|
||||||
|
|
||||||
@window.event
|
|
||||||
def on_mouse_motion(x, y, dx, dy):
|
|
||||||
pass
|
|
||||||
|
|
||||||
last_time = [time.time()]
|
|
||||||
|
|
||||||
def poll(dt):
|
|
||||||
out = next(itr, False)
|
|
||||||
if out is False:
|
|
||||||
if args.pause:
|
|
||||||
label.text = "Done. ('q' to quit)"
|
|
||||||
else:
|
|
||||||
pyglet.app.exit()
|
|
||||||
elif out is not None:
|
|
||||||
name, buf = out
|
|
||||||
real_dt = time.time() - last_time[0]
|
|
||||||
last_time[0] = time.time()
|
|
||||||
if buf.dtype == np.uint8:
|
|
||||||
fmt = gl.GL_UNSIGNED_BYTE
|
|
||||||
elif buf.dtype == np.uint16:
|
|
||||||
fmt = gl.GL_UNSIGNED_SHORT
|
|
||||||
else:
|
|
||||||
label.text = 'Unsupported format: ' + buf.dtype
|
|
||||||
return
|
|
||||||
|
|
||||||
h, w, ch = buf.shape
|
|
||||||
gl.glEnable(tex.target)
|
|
||||||
gl.glBindTexture(tex.target, tex.id)
|
|
||||||
gl.glTexImage2D(tex.target, 0, gl.GL_RGB8, w, h, 0, gl.GL_RGBA,
|
|
||||||
fmt, buf.tostring())
|
|
||||||
gl.glDisable(tex.target)
|
|
||||||
label.text = '%s (%g fps)' % (name, 1./real_dt)
|
|
||||||
else:
|
|
||||||
label.text += '.'
|
|
||||||
|
|
||||||
pyglet.clock.set_fps_limit(20)
|
|
||||||
pyglet.clock.schedule_interval(poll, 1/20.)
|
|
||||||
pyglet.app.run()
|
|
||||||
|
|
||||||
def main(args, prof):
|
def main(args, prof):
|
||||||
gdb = db.connect(args.genomedb)
|
gdb = db.connect(args.genomedb)
|
||||||
gnm, basename = gdb.get_anim(args.flame, args.half)
|
gnm, basename = gdb.get_anim(args.flame, args.half)
|
||||||
@ -121,7 +49,7 @@ def main(args, prof):
|
|||||||
import pycuda.driver as cuda
|
import pycuda.driver as cuda
|
||||||
cuda.init()
|
cuda.init()
|
||||||
dev = cuda.Device(args.device or 0)
|
dev = cuda.Device(args.device or 0)
|
||||||
cuctx = dev.make_context(flags=cuda.ctx_flags.SCHED_YIELD)
|
cuctx = dev.make_context(flags=cuda.ctx_flags.SCHED_BLOCKING_SYNC)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
rmgr = render.RenderManager()
|
rmgr = render.RenderManager()
|
||||||
@ -129,40 +57,56 @@ def main(args, prof):
|
|||||||
dev.get_attribute(cuda.device_attribute.COMPUTE_CAPABILITY_MAJOR),
|
dev.get_attribute(cuda.device_attribute.COMPUTE_CAPABILITY_MAJOR),
|
||||||
dev.get_attribute(cuda.device_attribute.COMPUTE_CAPABILITY_MINOR))
|
dev.get_attribute(cuda.device_attribute.COMPUTE_CAPABILITY_MINOR))
|
||||||
rdr = render.Renderer(gnm, gprof, keep=args.keep, arch=arch)
|
rdr = render.Renderer(gnm, gprof, keep=args.keep, arch=arch)
|
||||||
|
last_render_time_ms = 0
|
||||||
|
|
||||||
def render_iter():
|
m = os.path.getmtime(args.flame)
|
||||||
m = os.path.getmtime(args.flame)
|
for name, times in frames:
|
||||||
first = True
|
if args.resume:
|
||||||
for name, times in frames:
|
fp = name + rdr.out.suffix
|
||||||
if args.resume:
|
if os.path.isfile(fp) and m < os.path.getmtime(fp):
|
||||||
fp = name + rdr.out.suffix
|
continue
|
||||||
if os.path.isfile(fp) and m < os.path.getmtime(fp):
|
|
||||||
continue
|
|
||||||
|
|
||||||
for idx, t in enumerate(times):
|
def save(buf):
|
||||||
evt, buf = rmgr.queue_frame(rdr, gnm, gprof, t, first)
|
out, log = rdr.out.encode(buf)
|
||||||
first = False
|
for suffix, file_like in out.items():
|
||||||
while not evt.query():
|
with open(name + suffix, 'w') as fp:
|
||||||
time.sleep(0.01)
|
fp.write(file_like.read())
|
||||||
yield None
|
if getattr(file_like, 'close', None):
|
||||||
save(rdr.out, name, buf)
|
file_like.close()
|
||||||
if args.rawfn:
|
for key, val in log:
|
||||||
try:
|
print '\n=== %s ===' % key
|
||||||
buf.tofile(args.rawfn + '.tmp')
|
print val
|
||||||
os.rename(args.rawfn + '.tmp', args.rawfn)
|
|
||||||
except:
|
evt = buf = next_evt = next_buf = None
|
||||||
import traceback
|
for idx, t in enumerate(list(times) + [None]):
|
||||||
print 'Failed to write %s: %s' % (args.rawfn,
|
evt, buf = next_evt, next_buf
|
||||||
traceback.format_exc())
|
if t is not None:
|
||||||
print '%s (%3d/%3d), %dms' % (name, idx, len(times), evt.time())
|
next_evt, next_buf = rmgr.queue_frame(rdr, gnm, gprof, t)
|
||||||
sys.stdout.flush()
|
if not evt: continue
|
||||||
yield name, buf
|
if last_render_time_ms > 2000:
|
||||||
save(rdr.out, name, None)
|
while not evt.query():
|
||||||
|
time.sleep(0.2)
|
||||||
|
else:
|
||||||
|
evt.synchronize()
|
||||||
|
last_render_time_ms = evt.time()
|
||||||
|
|
||||||
|
save(buf)
|
||||||
|
|
||||||
|
if args.rawfn:
|
||||||
|
try:
|
||||||
|
buf.tofile(args.rawfn + '.tmp')
|
||||||
|
os.rename(args.rawfn + '.tmp', args.rawfn)
|
||||||
|
except:
|
||||||
|
import traceback
|
||||||
|
print 'Failed to write %s: %s' % (
|
||||||
|
args.rawfn, traceback.format_exc())
|
||||||
|
print '%s%s (%3d/%3d), %dms' % (
|
||||||
|
('%d: ' % args.device) if args.device >= 0 else '',
|
||||||
|
name, idx, len(times), last_render_time_ms)
|
||||||
|
sys.stdout.flush()
|
||||||
|
|
||||||
|
save(None)
|
||||||
|
|
||||||
if args.gfx:
|
|
||||||
pyglet_preview(args, gprof, render_iter())
|
|
||||||
else:
|
|
||||||
for i in render_iter(): pass
|
|
||||||
finally:
|
finally:
|
||||||
cuda.Context.pop()
|
cuda.Context.pop()
|
||||||
|
|
||||||
@ -172,8 +116,6 @@ if __name__ == "__main__":
|
|||||||
|
|
||||||
parser.add_argument('flame', metavar='ID', type=str,
|
parser.add_argument('flame', metavar='ID', type=str,
|
||||||
help="Filename or flame ID of genome to render")
|
help="Filename or flame ID of genome to render")
|
||||||
parser.add_argument('-g', action='store_true', dest='gfx',
|
|
||||||
help="Show output in OpenGL window")
|
|
||||||
parser.add_argument('-n', metavar='NAME', type=str, dest='name',
|
parser.add_argument('-n', metavar='NAME', type=str, dest='name',
|
||||||
help="Prefix to use when saving files (default is basename of input)")
|
help="Prefix to use when saving files (default is basename of input)")
|
||||||
parser.add_argument('--suffix', metavar='NAME', type=str, dest='suffix',
|
parser.add_argument('--suffix', metavar='NAME', type=str, dest='suffix',
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user