commit 37c0b2a7fb65a7dc8b91066b676e1a76ded757e5 · cduck.me/drawsvg

+10

__init__.py

···

       41
        
       ```

     

       42
        
       '''

     

       43
        
       

     

       0
        
       
     

       44
        
       from .raster import Raster

     

       45
        
       from .drawing import Drawing

     

       46
        
       from .elements import *

     

       47
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0

···

       41
        
       ```

     

       42
        
       '''

     

       43
        
       

     

       44
       +
       from .defs import *

     

       45
        
       from .raster import Raster

     

       46
        
       from .drawing import Drawing

     

       47
        
       from .elements import *

     

       48
        
       

     

       49
       +
       # Make all elements available in the elements module

     

       50
       +
       from . import defs

     

       51
       +
       from . import elements

     

       52
       +
       elementsDir = dir(elements)

     

       53
       +
       for k in dir(defs):

     

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

     

       55
       +
           if k in elementsDir: continue

     

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

     

       57
       +

+46

defs.py

···

       1
       +
       

     

       2
       +
       from .elements import DrawingElement, DrawingParentElement

     

       3
       +
       

     

       4
       +
       

     

       5
       +
       class DrawingDef(DrawingParentElement):

     

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

     

       7
       +
           @property

     

       8
       +
           def id(self):

     

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

     

       10
       +
           @id.setter

     

       11
       +
           def id(self, newId):

     

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

     

       13
       +
           def getSvgDefs(self):

     

       14
       +
               return (self,)

     

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

     

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

     

       17
       +
       

     

       18
       +
       class DrawingDefSub(DrawingParentElement):

     

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

     

       20
       +
           pass

     

       21
       +
       

     

       22
       +
       class RadialGradient(DrawingDef):

     

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

     

       24
       +
       

     

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

     

       26
       +
           TAG_NAME = 'radialGradient'

     

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

     

       28
       +
               yShift = 0

     

       29
       +
               if gradientUnits != 'userSpaceOnUse':

     

       30
       +
                   yShift = 1

     

       31
       +
               try: cy = yShift - cy

     

       32
       +
               except TypeError: pass

     

       33
       +
               try: fy = yShift - fy

     

       34
       +
               except TypeError: pass

     

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

     

       36
       +
                                fy=fy, **kwargs)

     

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

     

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

     

       39
       +
                                   stop_opacity=opacity, **kwargs)

     

       40
       +
               self.append(stop)

     

       41
       +
       

     

       42
       +
       class GradientStop(DrawingDefSub):

     

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

     

       44
       +
           TAG_NAME = 'stop'

     

       45
       +
           hasContent = False

     

       46
       +

+21 -1

drawing.py

···

       82
        
                   imgWidth, imgHeight, *self.viewBox)

     

       83
        
               endStr = '</svg>'

     

       84
        
               outputFile.write(startStr)

     

       85
       -
               outputFile.write('\n')

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       86
        
               for element in self.elements:

     

       87
        
                   try:

     

       88
        
                       element.writeSvgElement(outputFile)

···

       82
        
                   imgWidth, imgHeight, *self.viewBox)

     

       83
        
               endStr = '</svg>'

     

       84
        
               outputFile.write(startStr)

     

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

     

       86
       +
               # Write definition elements

     

       87
       +
               idIndex = 0

     

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

     

       89
       +
                   nonlocal idIndex

     

       90
       +
                   idStr = base + str(idIndex)

     

       91
       +
                   idIndex += 1

     

       92
       +
                   return idStr

     

       93
       +
               prevSet = set()

     

       94
       +
               def isDuplicate(obj):

     

       95
       +
                   nonlocal prevSet

     

       96
       +
                   dup = id(obj) in prevSet

     

       97
       +
                   prevSet.add(id(obj))

     

       98
       +
                   return dup

     

       99
       +
               for element in self.elements:

     

       100
       +
                   try:

     

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

     

       102
       +
                   except AttributeError:

     

       103
       +
                       pass

     

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

     

       105
       +
               # Write normal elements

     

       106
        
               for element in self.elements:

     

       107
        
                   try:

     

       108
        
                       element.writeSvgElement(outputFile)

+55 -3

elements.py

···

       1
        
       

     

       0
        
       
     

       2
        
       import math

     

       3
        
       import os.path

     

       4
        
       import base64

     

       5
        
       import warnings

     

       6
        
       import xml.sax.saxutils as xml

     

       7
        
       

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       8
        
       # TODO: Support drawing ellipses without manually using Path

     

       9
        
       

     

       10
        
       

     
···

       14
        
               Subclasses must implement writeSvgElement '''

     

       15
        
           def writeSvgElement(self, outputFile):

     

       16
        
               raise NotImplementedError('Abstract base class')

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       17
        
           def __eq__(self, other):

     

       18
        
               return self is other

     

       19
        
       

     
···

       27
        
           def writeSvgElement(self, outputFile):

     

       28
        
               outputFile.write('<')

     

       29
        
               outputFile.write(self.TAG_NAME)

     

       30
       -
               outputFile.write(' ')

     

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

     

       0
        
       
     

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

     

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

     

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

     

       0
        
       
     

       0
        
       
     

       35
        
               if not self.hasContent:

     

       36
       -
                   outputFile.write('/>')

     

       37
        
               else:

     

       38
        
                   outputFile.write('>')

     

       39
        
                   self.writeContent(outputFile)

     
···

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

     

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

     

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

     

       0
        
       
     

       0
        
       
     

       47
        
           def __eq__(self, other):

     

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

     

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

     

       50
        
                           self.args == other.args)

     

       51
        
               return False

     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       0
        
       
     

       52
        
       

     

       53
        
       class NoElement(DrawingElement):

     

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

···

       1
        
       

     

       2
       +
       import sys

     

       3
        
       import math

     

       4
        
       import os.path

     

       5
        
       import base64

     

       6
        
       import warnings

     

       7
        
       import xml.sax.saxutils as xml

     

       8
        
       

     

       9
       +
       from . import defs

     

       10
       +
       

     

       11
       +
       elementsModule = sys.modules[__name__]

     

       12
       +
       

     

       13
        
       # TODO: Support drawing ellipses without manually using Path

     

       14
        
       

     

       15
        
       

     
···

       19
        
               Subclasses must implement writeSvgElement '''

     

       20
        
           def writeSvgElement(self, outputFile):

     

       21
        
               raise NotImplementedError('Abstract base class')

     

       22
       +
           def getSvgDefs(self):

     

       23
       +
               return ()

     

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

     

       25
       +
               for defn in self.getSvgDefs():

     

       26
       +
                   if isDuplicate(defn): continue

     

       27
       +
                   defn.id = idGen()

     

       28
       +
                   defn.writeSvgElement(outputFile)

     

       29
       +
                   outputFile.write('\n')

     

       30
        
           def __eq__(self, other):

     

       31
        
               return self is other

     

       32
        
       

     
···

       40
        
           def writeSvgElement(self, outputFile):

     

       41
        
               outputFile.write('<')

     

       42
        
               outputFile.write(self.TAG_NAME)

     

       0
        
       
     

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

     

       44
       +
                   if v is None: continue

     

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

     

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

     

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

     

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

     

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

     

       50
        
               if not self.hasContent:

     

       51
       +
                   outputFile.write(' />')

     

       52
        
               else:

     

       53
        
                   outputFile.write('>')

     

       54
        
                   self.writeContent(outputFile)

     
···

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

     

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

     

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

     

       62
       +
           def getSvgDefs(self):

     

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

     

       64
        
           def __eq__(self, other):

     

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

     

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

     

       67
        
                           self.args == other.args)

     

       68
        
               return False

     

       69
       +
       

     

       70
       +
       class DrawingParentElement(DrawingBasicElement):

     

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

     

       72
       +
           hasContent = True

     

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

     

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

     

       75
       +
               self.children = list(children)

     

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

     

       77
       +
                   self.checkChildrenAllowed()

     

       78
       +
           def checkChildrenAllowed(self):

     

       79
       +
               if not self.hasContent:

     

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

     

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

     

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

     

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

     

       84
       +
                   self.extend(elements)

     

       85
       +
               else:

     

       86
       +
                   assert len(kwargs) == 0

     

       87
       +
                   self.append(obj)

     

       88
       +
           def append(self, element):

     

       89
       +
               self.checkChildrenAllowed()

     

       90
       +
               self.children.append(element)

     

       91
       +
           def extend(self, iterable):

     

       92
       +
               self.checkChildrenAllowed()

     

       93
       +
               self.children.extend(iterable)

     

       94
       +
           def writeContent(self, outputFile):

     

       95
       +
               outputFile.write('\n')

     

       96
       +
               for child in self.children:

     

       97
       +
                   child.writeSvgElement(outputFile)

     

       98
       +
                   outputFile.write('\n')

     

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

     

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

     

       101
       +
               for child in self.children:

     

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

     

       103
       +
                   outputFile.write('\n')

     

       104
        
       

     

       105
        
       class NoElement(DrawingElement):

     

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