import sys from random import seed, randrange # ========================================================================== # Recursive quadtree data structure # ========================================================================== def quadtree(points,topleft,botright): if len(points) > 1: # print >>sys.stderr,"splitting",len(points),topleft,botright mid = [(topleft[i]+botright[i])*0.5 for i in range(2)] svgLine((mid[0],topleft[1]),(mid[0],botright[1])) svgLine((topleft[0],mid[1]),(botright[0],mid[1])) quadtree([p for p in points if p[0]<mid[0] and p[1]<mid[1]], topleft,mid) quadtree([p for p in points if p[0]<mid[0] and p[1]>=mid[1]], (topleft[0],mid[1]),(mid[0],botright[1])) quadtree([p for p in points if p[0]>=mid[0] and p[1]<mid[1]], (mid[0],topleft[1]),(botright[0],mid[1])) quadtree([p for p in points if p[0]>=mid[0] and p[1]>=mid[1]], mid,botright) # ========================================================================== # SVG output utility routines # ========================================================================== nestingLevel = None def svgTag(s, deltaIndentation = 0): """Send a single XML tag to the SVG file. First argument is the tag with all its attributes appended. Second arg is +1, -1, or 0 if tag is open, close, or both respectively. """ global nestingLevel if deltaIndentation < 0: nestingLevel -= 1 if nestingLevel: sys.stdout.write('\t' * nestingLevel) sys.stdout.write('<') if deltaIndentation < 0: sys.stdout.write('/') sys.stdout.write(s) if not deltaIndentation: sys.stdout.write(' /') sys.stdout.write('>\n') if deltaIndentation > 0: nestingLevel += 1 def svgHeader(maxX, maxY): """Start producing an SVG object. The output bounding box runs from (0,0) to (maxX,maxY). Must be followed by svg content and a call to svgTrailer(). """ global nestingLevel if nestingLevel is None: sys.stdout.write('''<?xml version="1.0" encoding="iso-8859-1"?> <!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd"> ''') nestingLevel = 0 svgTag('svg xmlns="http://www.w3.org/2000/svg" version="1.1" ' 'width="%dpt" height="%dpt" viewBox="0 0 %d %d"' % (maxX, maxY, maxX, maxY), 1) def svgTrailer(): """End of SVG object.""" svgTag('svg', -1) def svgStyle(style): """Start a group of svg items with the given style. Argument is a string in the form of a list of svg item attributes. Must be followed by svg content and a call to svgEndStyle(). """ svgTag('g ' + style, 1) def svgEdgeStyle(index): """Look up edge style and call svgStyle with it.""" svgStyle(globals()['edgeStyle' + str(index)]) def svgEndStyle(): """Finish group of styled svg items.""" svgTag('g', -1) def svgLine(start, end): """Output line segment from start to end coordinates. Must be called within an svgStyle()/svgEndStyle() call pair. """ svgTag('line x1="%d" y1="%d" x2="%d" y2="%d"' % (start+end) ) def svgCircle(center, radius): """Output circle with given center and radius coordinates. Must be called within an svgStyle()/svgEndStyle() call pair. """ svgTag('circle cx="%d" cy="%d" r="%s"' % (center+(radius,)) ) # ========================================================================== # Test code driver # ========================================================================== seed(27) svgRadius = 2 svgBound = 400 margin = 4 def clusteredCoordinate(clusteringExponent,fractalExponent): x = long(randrange(1000)**clusteringExponent) y = 0 p = 1 while x: if x & 1: y += p p *= fractalExponent x >>= 1 return y def clusteredPoint(): return clusteredCoordinate(2,2.5),clusteredCoordinate(1.5,3.2) points = [clusteredPoint() for i in range(250)] maxX = max(x for x,y in points) maxY = max(y for x,y in points) scaleX = 1.0*(svgBound-2*margin)/maxX scaleY = 1.0*(svgBound-2*margin)/maxY points = [(x*scaleX+margin,y*scaleY+margin) for x,y in points] svgHeader(svgBound,svgBound) svgStyle('fill="none" stroke="blue"') svgLine((1,1),(1,svgBound-1)) svgLine((1,svgBound-1),(svgBound-1,svgBound-1)) svgLine((svgBound-1,svgBound-1),(svgBound-1,1)) svgLine((svgBound-1,1),(1,1)) quadtree(points,(1,1),(svgBound-1,svgBound-1)) svgEndStyle() svgStyle('fill="white" stroke="black"') for p in points: svgCircle(p,svgRadius) svgEndStyle() svgTrailer()