added triangulator class

This commit is contained in:
zzzzrrr 2009-11-12 16:54:10 -05:00
parent a932c1b709
commit 76a60ae144

View File

@ -29,6 +29,7 @@
# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS # NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. # SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
# #
from random import shuffle
### ###
### Based on Raimund Seidel'e paper "A simple and fast incremental randomized ### Based on Raimund Seidel'e paper "A simple and fast incremental randomized
@ -42,14 +43,112 @@ cdef extern from 'math.h':
double floor(double) double floor(double)
double sqrt(double) double sqrt(double)
class Triangulator:
def __init__(self, points):
self.polygons = []
self.edge_list = self.init_edges(points)
self.trapezoids = []
self.trapezoidal_map = TrapezoidalMap()
boundingBox = self.trapezoidal_map.boundingBox(self.edge_list)
self.query_graph = QueryGraph(Sink(boundingBox))
self.xmono_poly = []
self.process()
def trapezoidMap(self):
return self.trapezoidal_map.map
# Build the trapezoidal map and query graph
def process(self):
for e in self.edge_list:
traps = self.query_graph.followSegment(e)
for t in traps:
self.trapezoidal_map.map.remove(t)
for t in traps:
tlist = []
cp = t.contains(e.p)
cq = t.contains(e.q)
if cp and cq:
tlist = self.trapezoidal_map.case1(t, e)
self.query_graph.case1(t.sink, e, tlist)
elif cp and not cq:
tlist = self.trapezoidal_map.case2(t, e)
self.query_graph.case2(t.sink, e, tlist)
elif not cp and not cq:
tlist = self.trapezoidal_map.case3(t, e)
self.query_graph.case3(t.sink, e, tlist)
else:
tlist = self.trapezoidal_map.case4(t, e)
self.query_graph.case4(t.sink, e, tlist)
# Add new trapezoids to map
for t in tlist:
self.trapezoidal_map.map.append(t)
self.trapezoidal_map.clear()
# Mark outside trapezoids
for t in self.trapezoidal_map.map:
self.mark_outside(t)
# Collect interior trapezoids
for t in self.trapezoidal_map.map:
if t.inside():
self.trapezoids.append(t)
t.add_points()
self.create_mountains()
def mono_polies(self):
polies = []
for x in self.xmono_poly:
polies.append(x.monoPoly)
return polies
def create_mountains(self):
for s in self.edge_list:
if len(s.mpoints) > 0:
mountain = MonotoneMountain()
k = merge_sort(s.mpoints)
points = [s.p] + k + [s.q]
for p in points:
mountain.append(p)
mountain.process()
for t in mountain.triangles:
self.polygons.append(t)
self.xmono_poly.append(mountain)
def mark_outside(self, t):
if t.top is self.boundingBox.top or t.bottom is self.boundingBox.bottom:
t.trimNeighbors()
def init_edges(self, points):
edges = []
for i in range(len(points)-1):
edges.append(Edge(points[i], points[i+1]))
edges.append(Edge(points[0], points[-1]))
return self.order_edges(edges)
def order_edges(self, edges):
segs = []
for s in edges:
p = self.shearTransform(s.p)
q = self.shearTransform(s.q)
if p.x > q.x: segs.append(Edge(q, p))
elif p.x < q.x: segs.append(Edge(p, q))
shuffle(segs)
return segs
def shearTransform(self, point):
return Point(point.x + 1e-4 * point.y, point.y)
cdef list merge_sort(list l): cdef list merge_sort(list l):
cdef list lleft, lright cdef list lleft, lright
cdef int p1, p2, p cdef int p1, p2, p
if len(l)>1 : if len(l)>1 :
lleft = merge_sort(l[:len(l)/2]) lleft = merge_sort(l[:len(l)/2])
lright = merge_sort(l[len(l)/2:]) lright = merge_sort(l[len(l)/2:])
#do merge here p1, p2, p = 0, 0, 0
p1,p2,p = 0,0,0
while p1<len(lleft) and p2<len(lright): while p1<len(lleft) and p2<len(lright):
if lleft[p1][0] < lright[p2][0]: if lleft[p1][0] < lright[p2][0]:
l[p]=lleft[p1] l[p]=lleft[p1]
@ -387,26 +486,20 @@ class Node:
def __init__(self, left, right): def __init__(self, left, right):
self.left = left self.left = left
self.right = right self.right = right
if left is not None: if left is not None: left.parent_list.append(self)
left.parent_list.append(self) if right is not None: right.parent_list.append(self)
if right is not None:
right.parent_list.append(self)
def replace(self, node): def replace(self, node):
for parent in node.parent_list: for parent in node.parent_list:
if parent.left is node: if parent.left is node: parent.left = self
parent.left = self else: parent.right = self
else:
parent.right = self
self.parent_list.append(parent) self.parent_list.append(parent)
class Sink(Node): class Sink(Node):
def __new__(cls, trapezoid): def __new__(cls, trapezoid):
if trapezoid.sink is not None: if trapezoid.sink is not None: return trapezoid.sink
return trapezoid.sink else: return Sink(trapezoid)
else:
return Sink(trapezoid)
def __init__(self, trapezoid): def __init__(self, trapezoid):
Node.__init__(self, None, None) Node.__init__(self, None, None)
@ -424,10 +517,8 @@ class XNode(Node):
self.rchild = rchild self.rchild = rchild
def locate(self, e): def locate(self, e):
if e.p.x >= self.point.x: if e.p.x >= self.point.x: return self.right.locate(e)
return self.right.locate(e) else: return self.left.locate(e)
else:
return self.left.locate(e)
class YNode(Node): class YNode(Node):
@ -438,15 +529,11 @@ class YNode(Node):
self.rchild = rchild self.rchild = rchild
def locate(self, e): def locate(self, e):
if self.edge.is_above(e.p): if self.edge.is_above(e.p): return self.right.locate(e)
return self.right.locate(e) elif self.edge.is_below(e.p): return self.left.locate(e)
elif self.edge.is_below(e.p):
return self.left.locate(e)
else: else:
if e.slope < self.edge.slope: if e.slope < self.edge.slope: return self.right.locate(e)
return self.right.locate(e) else: return self.left.locate(e)
else:
return self.left.locate(e)
class QueryGraph: class QueryGraph: