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

Support symmetry element when importing XML

Browse Source
This commit is contained in:
Steven Robertson 2011-12-18 21:38:08 -05:00
parent cbb8e74383
commit e9c9cd7db4
1 changed files with 23 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
cuburn/genome.py
View File

@ -273,6 +273,8 @@ class XMLGenomeParser(object):
idx = int(attrs['index']) idx = int(attrs['index'])
self._flame['palette'][idx][:3] = [float(v) / 255.0 self._flame['palette'][idx][:3] = [float(v) / 255.0
for v in attrs['rgb'].split()] for v in attrs['rgb'].split()]
elif name == 'symmetry':
self._flame['symmetry'] = int(attrs['kind'])
def end_element(self, name): def end_element(self, name):
if name == 'flame': if name == 'flame':
self.flames.append(self._flame) self.flames.append(self._flame)
@ -326,6 +328,8 @@ def convert_flame(flame):
num_xf = len(flame['xforms']) num_xf = len(flame['xforms'])
xfs = dict([(str(k), convert_xform(v, num_xf)) xfs = dict([(str(k), convert_xform(v, num_xf))
for k, v in enumerate(flame['xforms'])]) for k, v in enumerate(flame['xforms'])])
if 'symmetry' in flame:
xfs.update(make_symm_xforms(flame['symmetry'], len(xfs)))
if 'finalxform' in flame: if 'finalxform' in flame:
xfs['final'] = convert_xform(flame['finalxform'], num_xf, True) xfs['final'] = convert_xform(flame['finalxform'], num_xf, True)
@ -392,6 +396,25 @@ def convert_affine(aff, animate=False):
return dict(spread=spread, magnitude={'x': xm, 'y': ym}, return dict(spread=spread, magnitude={'x': xm, 'y': ym},
angle=angle, offset={'x': xo, 'y': yo}) angle=angle, offset={'x': xo, 'y': yo})
def make_symm_xforms(kind, offset):
assert kind != 0, 'Pick your own damn symmetry.'
out = []
boring_xf = dict(color=1, color_speed=0, density=1, opacity=1,
variations={'linear': {'weight': 1}})
if kind < 0:
out.append(boring_xf.copy())
out[-1]['affine'] = dict(angle=135, magnitude={'x': 1, 'y': 1},
spread=-45, offset={'x': 0, 'y': 0})
kind = -kind
for i in range(1, kind):
out.append(boring_xf.copy())
if kind >= 3:
out[-1]['color'] = (i - 1) / (kind - 2.0)
ang = (45 + 360 * i / float(kind)) % 360
out[-1]['affine'] = dict(angle=ang, magnitude={'x': 1, 'y': 1},
spread=-45, offset={'x': 0, 'y': 0})
return dict((str(i+offset), v) for i, v in enumerate(out))
def convert_file(path): def convert_file(path):
"""Quick one-shot conversion for an XML genome.""" """Quick one-shot conversion for an XML genome."""
flames = XMLGenomeParser.parse(open(path).read()) flames = XMLGenomeParser.parse(open(path).read())