summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--src/exporter.py57
1 files changed, 29 insertions, 28 deletions
diff --git a/src/exporter.py b/src/exporter.py
index 547b6fd..f1afd4b 100644
--- a/src/exporter.py
+++ b/src/exporter.py
@@ -299,34 +299,35 @@ class KPMapExporter:
current = len(data)
data += struct.pack('>hhiiiiii', x+12, y+12, 0, 0, 0, 0, 0, 0)
- # this varies
- if node.isStop():
- # figure out the exits by direction
- leftExit, rightExit, upExit, downExit = None, None, None, None
-
- for exit in node.exits:
- start, end = exit._startNodeRef(), exit._endNodeRef()
- opposite = end if (start == node) else start
-
- oX, oY = opposite.position
- deltaX, deltaY = oX-x, oY-y
- angle = math.degrees(math.atan2(deltaX, deltaY)) % 360
-
- # Left = 270, Right = 90, Up = 180, Down = 0
- if angle >= 225 and angle <= 315:
- leftExit = exit
- elif angle >= 45 and angle <= 135:
- rightExit = exit
- elif angle > 135 and angle < 225:
- upExit = exit
- elif angle > 315 or angle < 45:
- downExit = exit
-
- exits = [leftExit, rightExit, upExit, downExit]
-
- else:
- # not a stop, so just dump them in
- exits = node.exits + [None,None,None,None]
+ # figure out the exits by direction
+ exits = [None, None, None, None]
+ left, right, up, down = 0, 1, 2, 3
+
+ for exit in node.exits:
+ start, end = exit._startNodeRef(), exit._endNodeRef()
+ opposite = end if (start == node) else start
+
+ oX, oY = opposite.position
+ deltaX, deltaY = oX-x, oY-y
+ angle = math.degrees(math.atan2(deltaX, deltaY)) % 360
+ print "Here: %d,%d Opposite %d,%d Delta: %d,%d Angle: %d" % (x,y,oX,oY,deltaX,deltaY,angle)
+
+ # Left = 270, Right = 90, Up = 180, Down = 0
+ if angle >= 225 and angle <= 315:
+ direction = left
+ elif angle >= 45 and angle <= 135:
+ direction = right
+ elif angle > 135 and angle < 225:
+ direction = up
+ elif angle > 315 or angle < 45:
+ direction = down
+
+ if exits[direction]:
+ print "Conflicting directions!"
+ while exits[direction]:
+ direction = (direction + 1) % 4
+
+ exits[direction] = exit
requiredFixUps.append((current+4, exits[0]))
requiredFixUps.append((current+8, exits[1]))