Stubbed out subd.butterfly and added testing script

This commit is contained in:
Stephen M. McQuay 2012-05-29 17:33:15 -06:00
parent e85a99e95a
commit 2355deec9f
6 changed files with 40 additions and 11 deletions

24
bin/iterate.py Normal file
View File

@ -0,0 +1,24 @@
import argparse
import json
from surf.geometry import PolygonMesh
from surf.subd import cc
from surf.subd import butterfly
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Prune orphan PCDs from couchdb')
parser.add_argument('type', choices=('cc', 'butterfly'))
parser.add_argument('mesh')
opt = parser.parse_args()
cube = json.load(open(opt.mesh, 'r'))
p = PolygonMesh(**cube)
if opt.type == 'cc':
refine = cc.refine
else:
refine = butterfly.refine
q = refine(p)
print(q)

View File

@ -1,6 +1,8 @@
from .geometry import Vertex, PolygonMesh from .geometry import Vertex, PolygonMesh
from . import subd
__all__ = [ __all__ = [
Vertex.__name__, Vertex.__name__,
PolygonMesh.__name__, PolygonMesh.__name__,
'subd',
] ]

View File

@ -18,6 +18,11 @@ We have chosen to use a winged-edge style mesh for our purpopses.
''' '''
__all__ = [
'Vertex',
'PolygonMesh',
]
def cross(a, b): def cross(a, b):
i = a.y * b.z - a.z * b.y i = a.y * b.z - a.z * b.y

View File

@ -0,0 +1,7 @@
from . import cc, butterfly
__all__ = [
cc.__name__,
butterfly.__name__,
]

2
surf/subd/butterfly.py Normal file
View File

@ -0,0 +1,2 @@
def refine(mesh):
pass

View File

@ -109,14 +109,3 @@ def refine(mesh):
new_faces.append([fvid, common_ids[0], connected_vid, common_ids[1]]) new_faces.append([fvid, common_ids[0], connected_vid, common_ids[1]])
return PolygonMesh(vertices=new_vertices, edges=new_edges, faces=new_faces) return PolygonMesh(vertices=new_vertices, edges=new_edges, faces=new_faces)
if __name__ == '__main__':
import sys
import json
from surf.subd.cc import refine
input_file_name = sys.argv[1]
cube = json.load(open(input_file_name, 'r'))
p = PolygonMesh(**cube)
q = refine(p)
print q