commit 37c0b2a7fb65a7dc8b91066b676e1a76ded757e5 · cduck.me/drawsvg

+10

__init__.py

···

       41
       41
        
       ```

     

       42
       42
        
       '''

     

       43
       43
        
       

     

       44
       44
       +
       from .defs import *

     

       44
       45
        
       from .raster import Raster

     

       45
       46
        
       from .drawing import Drawing

     

       46
       47
        
       from .elements import *

     

       47
       48
        
       

     

       49
       49
       +
       # Make all elements available in the elements module

     

       50
       50
       +
       from . import defs

     

       51
       51
       +
       from . import elements

     

       52
       52
       +
       elementsDir = dir(elements)

     

       53
       53
       +
       for k in dir(defs):

     

       54
       54
       +
           if k.startswith('_'): continue

     

       55
       55
       +
           if k in elementsDir: continue

     

       56
       56
       +
           setattr(elements, k, getattr(defs, k))

     

       57
       57
       +

+46

defs.py

···

       1
       1
       +
       

     

       2
       2
       +
       from .elements import DrawingElement, DrawingParentElement

     

       3
       3
       +
       

     

       4
       4
       +
       

     

       5
       5
       +
       class DrawingDef(DrawingParentElement):

     

       6
       6
       +
           ''' Parent class of SVG nodes that must be direct children of <defs> '''

     

       7
       7
       +
           @property

     

       8
       8
       +
           def id(self):

     

       9
       9
       +
               return self.args.get('id', None)

     

       10
       10
       +
           @id.setter

     

       11
       11
       +
           def id(self, newId):

     

       12
       12
       +
               self.args['id'] = newId

     

       13
       13
       +
           def getSvgDefs(self):

     

       14
       14
       +
               return (self,)

     

       15
       15
       +
           def writeSvgDefs(self, idGen, isDuplicate, outputFile):

     

       16
       16
       +
               DrawingElement.writeSvgDefs(idGen, isDuplicate, outputFile)

     

       17
       17
       +
       

     

       18
       18
       +
       class DrawingDefSub(DrawingParentElement):

     

       19
       19
       +
           ''' Parent class of SVG nodes that are meant to be descendants of a Def '''

     

       20
       20
       +
           pass

     

       21
       21
       +
       

     

       22
       22
       +
       class RadialGradient(DrawingDef):

     

       23
       23
       +
           ''' A radial gradient to use as a fill or other color

     

       24
       24
       +
       

     

       25
       25
       +
               Has <stop> nodes as children. '''

     

       26
       26
       +
           TAG_NAME = 'radialGradient'

     

       27
       27
       +
           def __init__(self, cx, cy, r, gradientUnits='userSpaceOnUse', fy=None, **kwargs):

     

       28
       28
       +
               yShift = 0

     

       29
       29
       +
               if gradientUnits != 'userSpaceOnUse':

     

       30
       30
       +
                   yShift = 1

     

       31
       31
       +
               try: cy = yShift - cy

     

       32
       32
       +
               except TypeError: pass

     

       33
       33
       +
               try: fy = yShift - fy

     

       34
       34
       +
               except TypeError: pass

     

       35
       35
       +
               super().__init__(cx=cx, cy=cy, r=r, gradientUnits=gradientUnits,

     

       36
       36
       +
                                fy=fy, **kwargs)

     

       37
       37
       +
           def addStop(self, offset, color, opacity=None, **kwargs):

     

       38
       38
       +
               stop = GradientStop(offset=offset, stop_color=color,

     

       39
       39
       +
                                   stop_opacity=opacity, **kwargs)

     

       40
       40
       +
               self.append(stop)

     

       41
       41
       +
       

     

       42
       42
       +
       class GradientStop(DrawingDefSub):

     

       43
       43
       +
           ''' A control point for a radial or linear gradient '''

     

       44
       44
       +
           TAG_NAME = 'stop'

     

       45
       45
       +
           hasContent = False

     

       46
       46
       +

+21 -1

drawing.py

···

       82
       82
        
                   imgWidth, imgHeight, *self.viewBox)

     

       83
       83
        
               endStr = '</svg>'

     

       84
       84
        
               outputFile.write(startStr)

     

       85
       85
       -
               outputFile.write('\n')

     

       85
       85
       +
               outputFile.write('\n<defs>\n')

     

       86
       86
       +
               # Write definition elements

     

       87
       87
       +
               idIndex = 0

     

       88
       88
       +
               def idGen(base='d'):

     

       89
       89
       +
                   nonlocal idIndex

     

       90
       90
       +
                   idStr = base + str(idIndex)

     

       91
       91
       +
                   idIndex += 1

     

       92
       92
       +
                   return idStr

     

       93
       93
       +
               prevSet = set()

     

       94
       94
       +
               def isDuplicate(obj):

     

       95
       95
       +
                   nonlocal prevSet

     

       96
       96
       +
                   dup = id(obj) in prevSet

     

       97
       97
       +
                   prevSet.add(id(obj))

     

       98
       98
       +
                   return dup

     

       99
       99
       +
               for element in self.elements:

     

       100
       100
       +
                   try:

     

       101
       101
       +
                       element.writeSvgDefs(idGen, isDuplicate, outputFile)

     

       102
       102
       +
                   except AttributeError:

     

       103
       103
       +
                       pass

     

       104
       104
       +
               outputFile.write('</defs>\n')

     

       105
       105
       +
               # Write normal elements

     

       86
       106
        
               for element in self.elements:

     

       87
       107
        
                   try:

     

       88
       108
        
                       element.writeSvgElement(outputFile)

+55 -3

elements.py

···

       1
       1
        
       

     

       2
       2
       +
       import sys

     

       2
       3
        
       import math

     

       3
       4
        
       import os.path

     

       4
       5
        
       import base64

     

       5
       6
        
       import warnings

     

       6
       7
        
       import xml.sax.saxutils as xml

     

       7
       8
        
       

     

       9
       9
       +
       from . import defs

     

       10
       10
       +
       

     

       11
       11
       +
       elementsModule = sys.modules[__name__]

     

       12
       12
       +
       

     

       8
       13
        
       # TODO: Support drawing ellipses without manually using Path

     

       9
       14
        
       

     

       10
       15
        
       

     
···

       14
       19
        
               Subclasses must implement writeSvgElement '''

     

       15
       20
        
           def writeSvgElement(self, outputFile):

     

       16
       21
        
               raise NotImplementedError('Abstract base class')

     

       22
       22
       +
           def getSvgDefs(self):

     

       23
       23
       +
               return ()

     

       24
       24
       +
           def writeSvgDefs(self, idGen, isDuplicate, outputFile):

     

       25
       25
       +
               for defn in self.getSvgDefs():

     

       26
       26
       +
                   if isDuplicate(defn): continue

     

       27
       27
       +
                   defn.id = idGen()

     

       28
       28
       +
                   defn.writeSvgElement(outputFile)

     

       29
       29
       +
                   outputFile.write('\n')

     

       17
       30
        
           def __eq__(self, other):

     

       18
       31
        
               return self is other

     

       19
       32
        
       

     
···

       27
       40
        
           def writeSvgElement(self, outputFile):

     

       28
       41
        
               outputFile.write('<')

     

       29
       42
        
               outputFile.write(self.TAG_NAME)

     

       30
       30
       -
               outputFile.write(' ')

     

       31
       43
        
               for k, v in self.args.items():

     

       44
       44
       +
                   if v is None: continue

     

       32
       45
        
                   k = k.replace('__', ':')

     

       33
       46
        
                   k = k.replace('_', '-')

     

       34
       34
       -
                   outputFile.write('{}="{}" '.format(k,v))

     

       47
       47
       +
                   if isinstance(v, defs.DrawingDef):

     

       48
       48
       +
                       v = 'url(#{})'.format(v.id)

     

       49
       49
       +
                   outputFile.write(' {}="{}"'.format(k,v))

     

       35
       50
        
               if not self.hasContent:

     

       36
       36
       -
                   outputFile.write('/>')

     

       51
       51
       +
                   outputFile.write(' />')

     

       37
       52
        
               else:

     

       38
       53
        
                   outputFile.write('>')

     

       39
       54
        
                   self.writeContent(outputFile)

     
···

       44
       59
        
               ''' Override in a subclass to add data between the start and end

     

       45
       60
        
                   tags.  This will not be called if hasContent is False. '''

     

       46
       61
        
               raise RuntimeError('This element has no content')

     

       62
       62
       +
           def getSvgDefs(self):

     

       63
       63
       +
               return [v for v in self.args.values() if isinstance(v, defs.DrawingDef)]

     

       47
       64
        
           def __eq__(self, other):

     

       48
       65
        
               if isinstance(other, type(self)):

     

       49
       66
        
                   return (self.tagName == other.tagName and

     

       50
       67
        
                           self.args == other.args)

     

       51
       68
        
               return False

     

       69
       69
       +
       

     

       70
       70
       +
       class DrawingParentElement(DrawingBasicElement):

     

       71
       71
       +
           ''' Base class for SVG elements that can have child nodes '''

     

       72
       72
       +
           hasContent = True

     

       73
       73
       +
           def __init__(self, children=(), **args):

     

       74
       74
       +
               super().__init__(**args)

     

       75
       75
       +
               self.children = list(children)

     

       76
       76
       +
               if len(self.children) > 0:

     

       77
       77
       +
                   self.checkChildrenAllowed()

     

       78
       78
       +
           def checkChildrenAllowed(self):

     

       79
       79
       +
               if not self.hasContent:

     

       80
       80
       +
                   raise RuntimeError('{} does not support children'.format(type(self)))

     

       81
       81
       +
           def draw(self, obj, **kwargs):

     

       82
       82
       +
               if not hasattr(obj, 'writeSvgElement'):

     

       83
       83
       +
                   elements = obj.toDrawables(elements=elementsModule, **kwargs)

     

       84
       84
       +
                   self.extend(elements)

     

       85
       85
       +
               else:

     

       86
       86
       +
                   assert len(kwargs) == 0

     

       87
       87
       +
                   self.append(obj)

     

       88
       88
       +
           def append(self, element):

     

       89
       89
       +
               self.checkChildrenAllowed()

     

       90
       90
       +
               self.children.append(element)

     

       91
       91
       +
           def extend(self, iterable):

     

       92
       92
       +
               self.checkChildrenAllowed()

     

       93
       93
       +
               self.children.extend(iterable)

     

       94
       94
       +
           def writeContent(self, outputFile):

     

       95
       95
       +
               outputFile.write('\n')

     

       96
       96
       +
               for child in self.children:

     

       97
       97
       +
                   child.writeSvgElement(outputFile)

     

       98
       98
       +
                   outputFile.write('\n')

     

       99
       99
       +
           def writeSvgDefs(self, idGen, isDuplicate, outputFile):

     

       100
       100
       +
               super().writeSvgDefs(idGen, isDuplicate, outputFile)

     

       101
       101
       +
               for child in self.children:

     

       102
       102
       +
                   child.writeSvgDefs(idGen, isDuplicate, outputFile)

     

       103
       103
       +
                   outputFile.write('\n')

     

       52
       104
        
       

     

       53
       105
        
       class NoElement(DrawingElement):

     

       54
       106
        
           ''' A drawing element that has no effect '''