mirror-github-bspeice/cuburn
1
0
Fork 0
mirror of https://github.com/stevenrobertson/cuburn.git synced 2025-02-05 11:40:04 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix xfsearch some more

Browse Source
This commit is contained in:
Steven Robertson 2012-07-04 23:04:46 -07:00
parent 27e0a2adb8
commit 5c079b2718
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
helpers/xfsearch.py
View File

@ -9,14 +9,16 @@ import os, sys, json, scipy, pycuda.autoinit
import numpy as np import numpy as np
from copy import deepcopy from copy import deepcopy
sys.path.insert(0, '.') from os.path import abspath, join, dirname
sys.path.insert(0, abspath(join(dirname(__file__), '..')))
from cuburn import genome, profile, render from cuburn import genome, profile, render
from main import save from main import save
def main(gnm_path, time): def main(gnm_path, time):
basename = os.path.basename(gnm_path).rsplit('.', 1)[0] gdb = genome.db.connect('.')
gnm, basename = gdb.get_anim(gnm_path)
rmgr = render.RenderManager() rmgr = render.RenderManager()
def go(gj, name): def go(gj, name):
gprof = profile.wrap(profile.BUILTIN['720p'], gj) gprof = profile.wrap(profile.BUILTIN['720p'], gj)
@ -24,13 +26,16 @@ def main(gnm_path, time):
for out in rmgr.render(gnm, gprof, rt): for out in rmgr.render(gnm, gprof, rt):
save(out) save(out)
gnm = json.load(open(gnm_path))
for i in gnm['xforms']: for i in gnm['xforms']:
xf = gnm['xforms'].pop(i) xf = gnm['xforms'].pop(i)
go(gnm, 'noxf_' + i) go(gnm, 'noxf_' + i)
gnm['xforms'][i] = xf gnm['xforms'][i] = xf
if 'final_xform' in gnm:
xf = gnm.pop('final_xform')
go(gnm, 'noxf_final')
gnm['final_xform'] = xf
vars = set([v for g in gnm['xforms'].values() for v in g['variations']]) vars = set([v for g in gnm['xforms'].values() for v in g['variations']])
for v in vars: for v in vars:
g2 = deepcopy(gnm) g2 = deepcopy(gnm)