def _delaunayMapFromData(nodePositions, edgeData, imageSize, sigmaOrbits = None):
if sigmaOrbits:
sys.stderr.write(
"TODO: re-use Delaunay sigma orbits instead of re-sorting!\n")
edges = [startEnd and
(startEnd[0], startEnd[1],
[nodePositions[startEnd[0]], nodePositions[startEnd[1]]])
for startEnd in edgeData]
result = GeoMap(nodePositions, edges, imageSize)
result.initializeMap(initLabelImage = False)
return result
def _delaunayMapFromData(nodePositions, edgeData, imageSize, sigmaOrbits = None):
if sigmaOrbits:
sys.stderr.write(
"TODO: re-use Delaunay sigma orbits instead of re-sorting!\n")
edges = [startEnd and
(startEnd[0], startEnd[1],
[nodePositions[startEnd[0]], nodePositions[startEnd[1]]])
for startEnd in edgeData]
result = GeoMap(nodePositions, edges, imageSize)
result.initializeMap(initLabelImage = False)
return result
def catMap(delaunayMap,
rectified = True,
includeTerminalPositions = False):
"""catMap(delaunayMap,
rectified = True,
includeTerminalPositions = False)
Extract a CAT (chordal axis transform) from a GeoMap object
containing a Delaunay Triangulation. Implements the rectified
version (from L.Prasad's DGCI'05 article), which also works for
GeoMaps with non-triangular Faces.
Expects outer faces of the delaunayMap to be marked with the
OUTER_FACE flag (in order to skip these faces and work only on the
inner parts of the shapes), and edges which belong to the
constrained segments of the CDT to be marked with the
CONTOUR_SEGMENT flag (i.e. an edge is a chord iff
'not edge.flag(CONTOUR_SEGMENT)')
Usually, you will do sth. along:
>>> del1 = delaunay.faceCDTMap(map1.face(173), map1.imageSize())
>>> cat1 = delaunay.catMap(del1, rectified = False)
The optional parameter `rectified` has been set to False here to
use the original definition of junction node positions (using the
circumcenter if possible) which works only for triangulations (and
is therefore disabled by default).
For triangulations, it is also possible to include the opposite
vertex of the terminal triangles into the skeleton by setting
`includeTerminalPositions` to True.
If you want to use the rectified CAT (with weak chords being
suppressed), you would use removeWeakChords before calling catMap:
>>> del2 = copy.copy(del1) # if you want to keep the original
>>> removeWeakChords(del2)
>>> rcat2 = delaunay.catMap(del2)
The resulting map will have additional attributes that relate the
skeleton with the original boundary:
subtendedLengths:
This is a list mapping edge labels of the skeleton to the
accumulated lengths of all contour segments within the contours
of the sleeve- and terminal-faces that comprise the sleeve
represented by that skeleton edge.
shapeWidths:
This is a list mapping edge labels of the skeleton to lists of
shape widths at each support point of the polygon (may be None
at the ends). In theory, this allows to reconstruct the
original boundary and thus makes the CAT fully reversible (see
Prasad's papers).
nodeChordLabels:
This is a list mapping node labels of the skeleton to lists of
(sleeveLabel, chordLabel) pairs, where sleeveLabel is the label
of a skeleton edge and chordLabel is the dart label of the first
chord from the `delaunayMap` within that sleeve. (This is
needed for later corrections of the junction node positions
after pruning.)"""
if rectified:
junctionPos = rectifiedJunctionNodePosition
assert not includeTerminalPositions, \
"includeTerminalPositions is not supported for the rectified CAT!"
else:
junctionPos = junctionNodePosition
result = GeoMap(delaunayMap.imageSize())
result.subtendedLengths = [0] # unused Edge 0
result.shapeWidths = [None]
result.nodeChordLabels = []
faceChords = [None] * delaunayMap.maxFaceLabel()
boundaryLength = [None] * delaunayMap.maxFaceLabel()
faceType = [None] * delaunayMap.maxFaceLabel()
nodeLabel = [None] * delaunayMap.maxFaceLabel()
for face in delaunayMap.faceIter(skipInfinite = True):
if face.flag(OUTER_FACE):
continue
assert face.holeCount() == 0
chords = []
bl = 0.0
for dart in face.contour().phiOrbit():
if not dart.edge().flag(CONTOUR_SEGMENT):
chords.append(dart)
else:
bl += dart.edge().length()
boundaryLength[face.label()] = bl
faceChords[face.label()] = chords
# classify into terminal, sleeve, and junction triangles:
if len(chords) < 2:
faceType[face.label()] = "T"
if chords:
nodePos = middlePoint(chords[0]) # (may be changed later)
elif len(chords) == 2:
faceType[face.label()] = "S"
else:
faceType[face.label()] = "J"
nodePos = junctionPos(chords)
# add nodes for non-sleeve triangles:
if faceType[face.label()] != "S" and chords:
nodeLabel[face.label()] = result.addNode(nodePos).label()
result.nodeChordLabels.append([])
for face in delaunayMap.faceIter(skipInfinite = True):
if face.flag(OUTER_FACE):
continue
if faceType[face.label()] != "S":
#print faceType[face.label()] + "-triangle", face.label()
for dart in faceChords[face.label()]:
#print "following limb starting with", dart
startNode = result.node(nodeLabel[face.label()])
startDart = dart.clone()
edgePoints = []
subtendedBoundaryLength = 0.0
shapeWidth = []
if faceType[face.label()] == "T":
subtendedBoundaryLength += boundaryLength[face.label()]
if includeTerminalPositions:
# include opposite position
startNode.setPosition((dart.clone().nextPhi())[-1])
while True:
edgePoints.append(middlePoint(dart))
shapeWidth.append(dart.edge().length())
dart.nextAlpha()
nextFace = dart.leftFace()
if faceType[nextFace.label()] == "S":
subtendedBoundaryLength += boundaryLength[nextFace.label()]
# continue with opposite dart:
if dart == faceChords[nextFace.label()][0]:
dart = faceChords[nextFace.label()][1]
else:
dart = faceChords[nextFace.label()][0]
else:
faceChords[nextFace.label()].remove(dart)
break
endNode = result.node(nodeLabel[nextFace.label()])
if faceType[nextFace.label()] == "T":
subtendedBoundaryLength += boundaryLength[nextFace.label()]
if includeTerminalPositions:
endNode.setPosition((dart.clone().nextPhi())[-1])
flags = 0
if not edgePoints or edgePoints[0] != startNode.position():
edgePoints.insert(0, startNode.position())
shapeWidth.insert(0, None)
flags |= START_NODE_ADDED
if edgePoints[-1] != endNode.position():
edgePoints.append(endNode.position())
shapeWidth.append(None)
flags |= END_NODE_ADDED
if len(edgePoints) < 2:
# don't add edges with only 1 point (two adjacent
# T-triangles) - may fail for J-faces?!
# but J-faces should not have nodes on their borders
assert endNode.position() == startNode.position() and \
endNode.isIsolated() and startNode.isIsolated()
nodeLabel[nextFace.label()] = nodeLabel[face.label()]
result.removeIsolatedNode(endNode)
continue
sleeve = result.addEdge(
startNode, endNode, edgePoints)
sleeve.setFlag(flags)
result.subtendedLengths.append(subtendedBoundaryLength)
result.shapeWidths.append(shapeWidth)
if faceType[face.label()] == "J":
result.nodeChordLabels[startNode.label()].append(
(sleeve.label(), startDart.label()))
if faceType[nextFace.label()] == "J":
result.nodeChordLabels[endNode.label()].append(
(sleeve.label(), dart.label()))
result.initializeMap(initLabelImage = False)
return result
def catMap(delaunayMap,
rectified = True,
includeTerminalPositions = False):
"""catMap(delaunayMap,
rectified = True,
includeTerminalPositions = False)
Extract a CAT (chordal axis transform) from a GeoMap object
containing a Delaunay Triangulation. Implements the rectified
version (from L.Prasad's DGCI'05 article), which also works for
GeoMaps with non-triangular Faces.
Expects outer faces of the delaunayMap to be marked with the
OUTER_FACE flag (in order to skip these faces and work only on the
inner parts of the shapes), and edges which belong to the
constrained segments of the CDT to be marked with the
CONTOUR_SEGMENT flag (i.e. an edge is a chord iff
'not edge.flag(CONTOUR_SEGMENT)')
Usually, you will do sth. along:
>>> del1 = delaunay.faceCDTMap(map1.face(173), map1.imageSize())
>>> cat1 = delaunay.catMap(del1, rectified = False)
The optional parameter `rectified` has been set to False here to
use the original definition of junction node positions (using the
circumcenter if possible) which works only for triangulations (and
is therefore disabled by default).
For triangulations, it is also possible to include the opposite
vertex of the terminal triangles into the skeleton by setting
`includeTerminalPositions` to True.
If you want to use the rectified CAT (with weak chords being
suppressed), you would use removeWeakChords before calling catMap:
>>> del2 = copy.copy(del1) # if you want to keep the original
>>> removeWeakChords(del2)
>>> rcat2 = delaunay.catMap(del2)
The resulting map will have additional attributes that relate the
skeleton with the original boundary:
subtendedLengths:
This is a list mapping edge labels of the skeleton to the
accumulated lengths of all contour segments within the contours
of the sleeve- and terminal-faces that comprise the sleeve
represented by that skeleton edge.
shapeWidths:
This is a list mapping edge labels of the skeleton to lists of
shape widths at each support point of the polygon (may be None
at the ends). In theory, this allows to reconstruct the
original boundary and thus makes the CAT fully reversible (see
Prasad's papers).
nodeChordLabels:
This is a list mapping node labels of the skeleton to lists of
(sleeveLabel, chordLabel) pairs, where sleeveLabel is the label
of a skeleton edge and chordLabel is the dart label of the first
chord from the `delaunayMap` within that sleeve. (This is
needed for later corrections of the junction node positions
after pruning.)"""
if rectified:
junctionPos = rectifiedJunctionNodePosition
assert not includeTerminalPositions, \
"includeTerminalPositions is not supported for the rectified CAT!"
else:
junctionPos = junctionNodePosition
result = GeoMap(delaunayMap.imageSize())
result.subtendedLengths = [0] # unused Edge 0
result.shapeWidths = [None]
result.nodeChordLabels = []
faceChords = [None] * delaunayMap.maxFaceLabel()
boundaryLength = [None] * delaunayMap.maxFaceLabel()
faceType = [None] * delaunayMap.maxFaceLabel()
nodeLabel = [None] * delaunayMap.maxFaceLabel()
for face in delaunayMap.faceIter(skipInfinite = True):
if face.flag(OUTER_FACE):
continue
assert face.holeCount() == 0
chords = []
bl = 0.0
for dart in face.contour().phiOrbit():
if not dart.edge().flag(CONTOUR_SEGMENT):
chords.append(dart)
else:
bl += dart.edge().length()
boundaryLength[face.label()] = bl
faceChords[face.label()] = chords
# classify into terminal, sleeve, and junction triangles:
if len(chords) < 2:
faceType[face.label()] = "T"
if chords:
nodePos = middlePoint(chords[0]) # (may be changed later)
elif len(chords) == 2:
faceType[face.label()] = "S"
else:
faceType[face.label()] = "J"
nodePos = junctionPos(chords)
# add nodes for non-sleeve triangles:
if faceType[face.label()] != "S" and chords:
nodeLabel[face.label()] = result.addNode(nodePos).label()
result.nodeChordLabels.append([])
for face in delaunayMap.faceIter(skipInfinite = True):
if face.flag(OUTER_FACE):
continue
if faceType[face.label()] != "S":
#print faceType[face.label()] + "-triangle", face.label()
for dart in faceChords[face.label()]:
#print "following limb starting with", dart
startNode = result.node(nodeLabel[face.label()])
startDart = dart.clone()
edgePoints = []
subtendedBoundaryLength = 0.0
shapeWidth = []
if faceType[face.label()] == "T":
subtendedBoundaryLength += boundaryLength[face.label()]
if includeTerminalPositions:
# include opposite position
startNode.setPosition((dart.clone().nextPhi())[-1])
while True:
edgePoints.append(middlePoint(dart))
shapeWidth.append(dart.edge().length())
dart.nextAlpha()
nextFace = dart.leftFace()
if faceType[nextFace.label()] == "S":
subtendedBoundaryLength += boundaryLength[nextFace.label()]
# continue with opposite dart:
if dart == faceChords[nextFace.label()][0]:
dart = faceChords[nextFace.label()][1]
else:
dart = faceChords[nextFace.label()][0]
else:
faceChords[nextFace.label()].remove(dart)
break
endNode = result.node(nodeLabel[nextFace.label()])
if faceType[nextFace.label()] == "T":
subtendedBoundaryLength += boundaryLength[nextFace.label()]
if includeTerminalPositions:
endNode.setPosition((dart.clone().nextPhi())[-1])
flags = 0
if not edgePoints or edgePoints[0] != startNode.position():
edgePoints.insert(0, startNode.position())
shapeWidth.insert(0, None)
flags |= START_NODE_ADDED
if edgePoints[-1] != endNode.position():
edgePoints.append(endNode.position())
shapeWidth.append(None)
flags |= END_NODE_ADDED
if len(edgePoints) < 2:
# don't add edges with only 1 point (two adjacent
# T-triangles) - may fail for J-faces?!
# but J-faces should not have nodes on their borders
assert endNode.position() == startNode.position() and \
endNode.isIsolated() and startNode.isIsolated()
nodeLabel[nextFace.label()] = nodeLabel[face.label()]
result.removeIsolatedNode(endNode)
continue
sleeve = result.addEdge(
startNode, endNode, edgePoints)
sleeve.setFlag(flags)
result.subtendedLengths.append(subtendedBoundaryLength)
result.shapeWidths.append(shapeWidth)
if faceType[face.label()] == "J":
result.nodeChordLabels[startNode.label()].append(
(sleeve.label(), startDart.label()))
if faceType[nextFace.label()] == "J":
result.nodeChordLabels[endNode.label()].append(
(sleeve.label(), dart.label()))
result.initializeMap(initLabelImage = False)
return result
execfile("testSPWS")
# --------------------------------------------------------------------
# flowline map creation / face.contains
# --------------------------------------------------------------------
# The following data contains edges that run out of the image range,
# which ATM leads to overlapping edges after border closing. That
# violates some assumptions and would lead to errors if we actually
# worked with that Map.
map = GeoMap(maxima2, [], Size2D(39, 39))
maputils.addFlowLinesToMap(flowlines2, map)
assert map.checkConsistency(), "graph inconsistent"
map.sortEdgesEventually(stepDist = 0.2, minDist = 0.05)
map.splitParallelEdges()
map.initializeMap()
assert map.checkConsistency(), "map inconsistent"
assert maputils.checkLabelConsistency(map), "map.labelImage() inconsistent"
# merge faces so that survivor has a hole:
hole = map.faceAt((16,17))
assert hole.contains((16,17))
assert contourPoly(hole.contour()).contains((16,17))
dart = hole.contour()
while True:
while dart.startNode().hasMinDegree(3):
face = maputils.removeEdge(dart.clone().prevSigma())
if dart.nextPhi() == hole.contour():
break
execfile("testSPWS")
# --------------------------------------------------------------------
# flowline map creation / face.contains
# --------------------------------------------------------------------
# The following data contains edges that run out of the image range,
# which ATM leads to overlapping edges after border closing. That
# violates some assumptions and would lead to errors if we actually
# worked with that Map.
map = GeoMap(maxima2, [], Size2D(39, 39))
maputils.addFlowLinesToMap(flowlines2, map)
assert map.checkConsistency(), "graph inconsistent"
map.sortEdgesEventually(stepDist=0.2, minDist=0.05)
map.splitParallelEdges()
map.initializeMap()
assert map.checkConsistency(), "map inconsistent"
assert maputils.checkLabelConsistency(map), "map.labelImage() inconsistent"
# merge faces so that survivor has a hole:
hole = map.faceAt((16, 17))
assert hole.contains((16, 17))
assert contourPoly(hole.contour()).contains((16, 17))
dart = hole.contour()
while True:
while dart.startNode().hasMinDegree(3):
face = maputils.removeEdge(dart.clone().prevSigma())
if dart.nextPhi() == hole.contour():
break
