From 972cd9f9ea7236f6a05ab8efa644ec177bf3538d Mon Sep 17 00:00:00 2001 From: Steven Robertson Date: Mon, 2 May 2011 19:30:14 +0000 Subject: [PATCH] Add image writing, and revert the buffer flip --- cuburn/code/iter.py | 2 +- main.py | 7 +++++-- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/cuburn/code/iter.py b/cuburn/code/iter.py index 4d947dc..a705396 100644 --- a/cuburn/code/iter.py +++ b/cuburn/code/iter.py @@ -109,7 +109,7 @@ void iter(mwc_st *msts, const iter_info *infos, float *accbuf, float *denbuf) { } // TODO: dither? - int i = ((int)((1.0f - y) * 255.0f) * 512) + int i = ((int)((y + 1.0f) * 255.0f) * 512) + (int)((x + 1.0f) * 255.0f); // since info was declared const, C++ barfs unless it's loaded first diff --git a/main.py b/main.py index 38a9bb8..5557a29 100644 --- a/main.py +++ b/main.py @@ -16,6 +16,7 @@ from ctypes import * import numpy as np np.set_printoptions(precision=5, edgeitems=20) +import scipy from fr0stlib.pyflam3 import * from fr0stlib.pyflam3._flam3 import * @@ -37,11 +38,13 @@ def main(args): if '-g' not in args: return + noalpha = np.delete(accum, 3, axis=2) + scipy.misc.imsave('rendered.png', noalpha) + imgbuf = (np.minimum(accum * 255, 255)).astype(np.uint8) window = pyglet.window.Window(1600, 900) - image = pyglet.image.ImageData(512, 512, 'RGBA', imgbuf.tostring()) - #-anim.features.hist_stride*4) + image = pyglet.image.ImageData(512, 512, 'RGBA', imgbuf.tostring(), -2048) tex = image.texture #pal = (anim.ctx.ptx.instances[PaletteLookup].pal * 255.).astype(np.uint8)