From 72ca02032a11d752c89463dd9ea92954080aea4f Mon Sep 17 00:00:00 2001 From: Steven Robertson Date: Tue, 11 Oct 2011 20:58:09 -0400 Subject: [PATCH] Fix flicker (and a few other things). --- main.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/main.py b/main.py index 856d7fa..51b1ba9 100644 --- a/main.py +++ b/main.py @@ -35,6 +35,7 @@ os.environ['PATH'] = ('/usr/x86_64-pc-linux-gnu/gcc-bin/4.4.5:' def main(args): if '-t' in args: MWCTest.test_mwc() + return with open(args[1]) as fp: genome_ptr, ngenomes = pyflam3.Genome.from_string(fp.read()) @@ -44,14 +45,17 @@ def main(args): anim.compile(keep=True, cmp_options=('-use_fast_math', '-maxrregcount', '32', '-G')) else: - anim.compile() + anim.compile(keep='-k' in args) anim.load() for n, out in enumerate(anim.render_frames()): noalpha = np.delete(out, 3, axis=2) - name = 'rendered_%03d' % n - scipy.misc.imsave(name+'.png', noalpha) - # Convert using imagemagick, to set custom quality - Popen(['convert', name+'.png', '-quality', '90', name+'.jpg']) + name = 'rendered_%05d' % n + scipy.misc.toimage(noalpha, cmin=0, cmax=1).save(name+'.png') + + if '-j' in args: + # Convert using imagemagick, to set custom quality + Popen(['convert', name+'.png', '-quality', '90', name+'.jpg']) + print 'saved', name, np.min(noalpha), np.max(noalpha) return #if '-g' not in args: