···

       
3
       
 
       
import os.path

     

       
4
       
 
       
import base64

     

       
5
       
 
       
import warnings

     

       
0
       
 
       

     

       
6
       
 
       


     

       
7
       
 
       
# TODO: Support drawing ellipses without manually using Path

     

       
8
       
 
       


     
···

       
20
       
 
       
    ''' Base class for SVG drawing elements that are a single node with no

     

       
21
       
 
       
        child nodes '''

     

       
22
       
 
       
    TAG_NAME = '_'

     

       
0
       
 
       

     

       
23
       
 
       
    def __init__(self, **args):

     

       
24
       
 
       
        self.args = args

     

       
25
       
 
       
    def writeSvgElement(self, outputFile):

     
···

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

     

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

     

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

     

       
33
       
-
       
        outputFile.write('/>')

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
34
       
 
       
    def __eq__(self, other):

     

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

     

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

     
···

       
90
       
 
       
            encData = base64.b64encode(data).decode()

     

       
91
       
 
       
            uri = 'data:{};base64,{}'.format(mimeType, encData)

     

       
92
       
 
       
        super().__init__(x=x, y=-y-height, width=width, height=height,

     

       
93
       
-
       
                         href=uri, **kwargs)

     

       
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
       
 
       

     

       
94
       
 
       


     

       
95
       
 
       
class Rectangle(DrawingBasicElement):

     

       
96
       
 
       
    ''' A rectangle

     

···

       
3
       
 
       
import os.path

     

       
4
       
 
       
import base64

     

       
5
       
 
       
import warnings

     

       
6
       
+
       
import xml.sax.saxutils as xml

     

       
7
       
 
       


     

       
8
       
 
       
# TODO: Support drawing ellipses without manually using Path

     

       
9
       
 
       


     
···

       
21
       
 
       
    ''' Base class for SVG drawing elements that are a single node with no

     

       
22
       
 
       
        child nodes '''

     

       
23
       
 
       
    TAG_NAME = '_'

     

       
24
       
+
       
    hasContent = False

     

       
25
       
 
       
    def __init__(self, **args):

     

       
26
       
 
       
        self.args = args

     

       
27
       
 
       
    def writeSvgElement(self, outputFile):

     
···

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

     

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

     

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

     

       
35
       
+
       
        if not self.hasContent:

     

       
36
       
+
       
            outputFile.write('/>')

     

       
37
       
+
       
        else:

     

       
38
       
+
       
            outputFile.write('>')

     

       
39
       
+
       
            self.writeContent(outputFile)

     

       
40
       
+
       
            outputFile.write('</')

     

       
41
       
+
       
            outputFile.write(self.TAG_NAME)

     

       
42
       
+
       
            outputFile.write('>')

     

       
43
       
+
       
    def writeContent(self, 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')

     

       
47
       
 
       
    def __eq__(self, other):

     

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

     

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

     
···

       
103
       
 
       
            encData = base64.b64encode(data).decode()

     

       
104
       
 
       
            uri = 'data:{};base64,{}'.format(mimeType, encData)

     

       
105
       
 
       
        super().__init__(x=x, y=-y-height, width=width, height=height,

     

       
106
       
+
       
                         xlink__href=uri, **kwargs)

     

       
107
       
+
       


     

       
108
       
+
       
class Text(DrawingBasicElement):

     

       
109
       
+
       
    ''' Text

     

       
110
       
+
       


     

       
111
       
+
       
        Additional keyword arguments are output as additional arguments to the

     

       
112
       
+
       
        SVG node e.g. fill="red", font_size=20, text_anchor="middle". '''

     

       
113
       
+
       
    TAG_NAME = 'text'

     

       
114
       
+
       
    hasContent = True

     

       
115
       
+
       
    def __init__(self, text, fontSize, x, y, center=False, **kwargs):

     

       
116
       
+
       
        if center:

     

       
117
       
+
       
            if 'text_anchor' not in kwargs:

     

       
118
       
+
       
                kwargs['text_anchor'] = 'middle'

     

       
119
       
+
       
            try:

     

       
120
       
+
       
                fontSize = float(fontSize)

     

       
121
       
+
       
                translate = 'translate(0,{})'.format(fontSize*0.5*center)

     

       
122
       
+
       
                if 'transform' in kwargs:

     

       
123
       
+
       
                    kwargs['transform'] = translate + ' ' + kwargs['transform']

     

       
124
       
+
       
                else:

     

       
125
       
+
       
                    kwargs['transform'] = translate

     

       
126
       
+
       
            except TypeError:

     

       
127
       
+
       
                pass

     

       
128
       
+
       
        super().__init__(x=x, y=-y, font_size=fontSize, **kwargs)

     

       
129
       
+
       
        self.escapedText = xml.escape(text)

     

       
130
       
+
       
    def writeContent(self, outputFile):

     

       
131
       
+
       
        outputFile.write(self.escapedText)

     

       
132
       
 
       


     

       
133
       
 
       
class Rectangle(DrawingBasicElement):

     

       
134
       
 
       
    ''' A rectangle