def readObstacles(fileName, yFlip=False):
print "READ OBSTACLES: ", fileName
if (fileName[-3:] == 'xml'):
parser = make_parser()
obstHandler = ObstXMLParser()
parser.setContentHandler(obstHandler)
parser.parse(fileName)
if (yFlip):
for o in obstHandler.obstacles:
o.flipY()
obstHandler.bb.flipY()
elif (fileName[-3:] == 'txt'):
f = open(fileName, 'r')
try:
oCount = int(f.readline())
except ValueError:
raise Exception, "Tried to parse Stephen's file format, but the first line was not edge count"
bb = AABB()
obstacles = ObstacleSet()
for o in xrange(oCount):
try:
x0, y0, x1, y1 = map(lambda x: float(x), f.readline().split())
except:
raise Exception, "Error trying to parse the sjguy obstacle file -- didn't find the specified number of edges"
obst = GLPoly()
obst.closed = False
obst.vertices.append(Vector3(x0, y0, 0))
obst.vertices.append(Vector3(x1, y1, 0))
bb.expand(obst.vertices)
obstacles.append(obst)
f.close()
return obstacles, bb
else:
raise Exception, "Invalid obstacle extension: %s" % (fileName)
return obstHandler.obstacles, obstHandler.bb
class ObstXMLParser(handler.ContentHandler):
def __init__(self):
self.bb = AABB()
self.obstacles = ObstacleSet()
self.currObst = None
def startElement(self, name, attrs):
if (name == 'Obstacle'):
# assume all obstacles have a closed attribute
self.currObst = GLPoly()
if (int(attrs['closed']) != 0):
self.currObst.closed = True
elif (name == 'Vertex' and self.currObst != None):
x = float(attrs['p_x'])
y = float(attrs['p_y'])
self.currObst.vertices.append(Vector3(x, y, 0))
def endElement(self, name):
if (name == "Obstacle"):
self.currObst.close()
self.obstacles.append(self.currObst)
self.bb.expand(self.currObst.vertices)
self.currObst = None
def endDocument(self):
print "Found %d obstacles" % (len(self.obstacles))
print "Overal BB:", self.bb
print
def frameDrawables(self):
'''Modifies the view to frame all of the drawables'''
if (self.drawables):
bb = AABB()
for drawable in self.drawables:
bb.extend(drawable.getBB())
size = bb.getSize()
w = size.x
h = size.y
# set the background and the view to be the same
self.setBG((w, h), (bb.min.x, bb.min.y))
center = bb.getCenter()
ar = float(self.wWidth) / self.wHeight
if (ar > 1):
# wider than tall, height constrains
h *= 1.05
w = h * ar
else:
# higher than wide, width constrains
w *= 1.05
h = w / ar
corner = (center.x - w / 2, center.y - h / 2)
self.setView((w, h), corner)
self._setOrtho()
self.updateGL()
def __init__(self):
self.bb = AABB()
self.obstacles = ObstacleSet()
self.currObst = None
def getBB(self):
'''Returns a bounding box spanning all of the obstacles'''
bb = AABB()
for poly in self.polys:
bb.extend(poly.getBB())
return bb