diff options
Diffstat (limited to 'make.py')
| -rw-r--r-- | make.py | 11 |
1 files changed, 7 insertions, 4 deletions
@@ -1,7 +1,7 @@ import numpy as np -from geometry import Mesh -from transform import rotate -from itertoolset import * +from chroma.geometry import Mesh +from chroma.transform import rotate +from chroma.itertoolset import * def mesh_grid(grid): return np.vstack(zip(grid[:-1].flatten(),grid[1:].flatten(),np.roll(grid[1:],-1,1).flatten()) + zip(grid[:-1].flatten(),np.roll(grid[1:],-1,1).flatten(),np.roll(grid[:-1],-1,1).flatten())) @@ -65,8 +65,11 @@ def rotate_extrude(x, y, nsteps=64): return Mesh(vertices, triangles, remove_duplicate_vertices=True) -def cube(size=1): +def cube(size=1, height=None): "Return a cube mesh whose sides have length `size`." + if height is None: + height = size + return linear_extrude([-size/2.0,size/2.0,size/2.0,-size/2.0],[-size/2.0,-size/2.0,size/2.0,size/2.0], height=size) def cylinder(radius=1, height=2, radius2=None, nsteps=64): |