User:Jonas Lund/Svg-Python

Helper for creating SVG files with python.
See randShape() for example.
Outputs this

#!/usr/bin/env python
"""\
SVG.py - Construct/display SVG scenes.

The following code is a lightweight wrapper around SVG files. The metaphor
is to construct a scene, add objects to it, and then write it to a file
to display it.

This program uses ImageMagick to display the SVG files. ImageMagick also 
does a remarkable job of converting SVG files into other formats.
"""

import os
import random
display_prog = 'mirage' # Command to execute to display images.
      
class Scene:
    def __init__(self,name="svg",height=700,width=700):
        self.name = name
        self.items = []
        self.height = height
        self.width = width
        return

    def add(self,item): self.items.append(item)

    def strarray(self):
        var = ["<?xml version=\"1.0\"?>\n",
               "<svg height=\"%d\" width=\"%d\" >\n" % (self.height,self.width),
               " <g style=\"fill-opacity:1.0; stroke:black;\n",
               "  stroke-width:1;\">\n"]
        for item in self.items: var += item.strarray()            
        var += [" </g>\n</svg>\n"]
        return var

    def write_svg(self,filename=None):
        if filename:
            self.svgname = filename
        else:
            self.svgname = self.name + ".svg"
        file = open(self.svgname,'w')
        file.writelines(self.strarray())
        file.close()
        return

    def display(self,prog=display_prog):
        os.system("%s %s" % (prog,self.svgname))
        return        
        

class Line:
    def __init__(self,start,end):
        self.start = start #xy tuple
        self.end = end     #xy tuple
        return

    def strarray(self):
        return ["  <line x1=\"%d\" y1=\"%d\" x2=\"%d\" y2=\"%d\" />\n" %\
                (self.start[0],self.start[1],self.end[0],self.end[1])]


class Circle:
    def __init__(self,center,radius,color):
        self.center = center #xy tuple
        self.radius = radius #xy tuple
        self.color = color   #rgb tuple in range(0,256)
        return

    def strarray(self):
        return ["  <circle cx=\"%d\" cy=\"%d\" r=\"%d\"\n" %\
                (self.center[0],self.center[1],self.radius),
                "    style=\"fill:%s;\"  />\n" % colorstr(self.color)]

class Rectangle:
    def __init__(self,origin,height,width,color):
        self.origin = origin
        self.height = height
        self.width = width
        self.color = color
        return

    def strarray(self):
        return ["  <rect x=\"%d\" y=\"%d\" height=\"%d\"\n" %\
                (self.origin[0],self.origin[1],self.height),
                "    width=\"%d\" style=\"fill:%s;\" />\n" %\
                (self.width,colorstr(self.color))]

class Text:
    def __init__(self,origin,text,size=24):
        self.origin = origin
        self.text = text
        self.size = size
        return

    def strarray(self):
        return ["  <text x=\"%d\" y=\"%d\" font-size=\"%d\">\n" %\
                (self.origin[0],self.origin[1],self.size),
                "   %s\n" % self.text,
                "  </text>\n"]
        
    
def colorstr(rgb): return "#%x%x%x" % (rgb[0]/16,rgb[1]/16,rgb[2]/16)

lyrics = """Soldier, soldier, will you marry me
With your musket, fife and drum?
Oh how can I marry such a pretty little girl
When I have no coat to put on?

Off to the tailor she did go
As fast as she could run
She brought him back the finest that was there
Now soldier, put it on

Soldier, soldier, will you marry me
With your musket, fife and drum?
Oh how can I marry such a pretty little girl
When I have no shoes to put on?

Off to the cobbler she did go
As fast as she could run
She brought him back the finest that was there
Now soldier, put them on

Soldier, soldier, will you marry me
With your musket, fife and drum?
Oh how can I marry such a pretty little girl
When I have no hat to put on?

Off to the hatter she did go
As fast as she could run
She brought him back the finest that was there
Now soldier, put it on

Soldier, soldier, will you marry me
With your musket, fife and drum?
Oh how can I marry such a pretty little girl
When I've a wife and child at home?"""


def randShape():
    scene = Scene('randShape4')
    
    #Split Lyrics into an array of words
    words = lyrics.split()

    scene.add(Rectangle((0,0),700,700,(128,128,128)))

    i = 0
    while i < 280:
        px = random.randint(0,700)
        py = random.randint(0,700)        
        rgb = random.randint(0,255)
        rgb1 = random.randint(0,255)
        rgb2 = random.randint(0,255)
        size = random.randint(0,500)        
        rad = random.randint(10,55)        
        rad2 = random.randint(0,80)        

        scene.add(Rectangle((px,py),rad,rad2,(rgb,rgb1,rgb2)))
        #scene.add(Line((px,py),(size,size)))
        #scene.add(Circle((px,py),rad,(rgb1,rgb,rgb2)))

        if(i < len(words)): 
            scene.add(Text((px,py), words[i],rad))        

        i+=1

    scene.write_svg()
    return
        
randShape()