···

       
3
       
 
       
import base64

     

       
4
       
 
       
import urllib.parse

     

       
5
       
 
       
import re

     

       
0
       
 
       

     

       
6
       
 
       


     

       
7
       
 
       
from . import Raster

     

       
8
       
 
       
from . import elements as elementsModule

     
···

       
32
       
 
       
        self.viewBox = (self.viewBox[0], -self.viewBox[1]-self.viewBox[3],

     

       
33
       
 
       
                        self.viewBox[2], self.viewBox[3])

     

       
34
       
 
       
        self.elements = []

     

       
0
       
 
       

     

       
35
       
 
       
        self.otherDefs = []

     

       
36
       
 
       
        self.pixelScale = 1

     

       
37
       
 
       
        self.renderWidth = None

     
···

       
67
       
 
       
            return self.renderWidth, self.height * s

     

       
68
       
 
       
        else:

     

       
69
       
 
       
            return self.renderWidth, self.renderHeight

     

       
70
       
-
       
    def draw(self, obj, **kwargs):

     

       
71
       
 
       
        if obj is None:

     

       
72
       
 
       
            return

     

       
73
       
 
       
        if not hasattr(obj, 'writeSvgElement'):

     
···

       
75
       
 
       
        else:

     

       
76
       
 
       
            assert len(kwargs) == 0

     

       
77
       
 
       
            elements = (obj,)

     

       
78
       
-
       
        self.extend(elements)

     

       
79
       
-
       
    def append(self, element):

     

       
80
       
-
       
        self.elements.append(element)

     

       
81
       
-
       
    def extend(self, iterable):

     

       
82
       
-
       
        self.elements.extend(iterable)

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
83
       
 
       
    def insert(self, i, element):

     

       
84
       
 
       
        self.elements.insert(i, element)

     

       
85
       
 
       
    def remove(self, element):

     
···

       
101
       
 
       
        self.otherDefs.extend(elements)

     

       
102
       
 
       
    def appendDef(self, element):

     

       
103
       
 
       
        self.otherDefs.append(element)

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
104
       
 
       
    def asSvg(self, outputFile=None):

     

       
105
       
 
       
        returnString = outputFile is None

     

       
106
       
 
       
        if returnString:

     
···

       
131
       
 
       
                outputFile.write('\n')

     

       
132
       
 
       
            except AttributeError:

     

       
133
       
 
       
                pass

     

       
134
       
-
       
        for element in self.elements:

     

       
0
       
 
       

     

       
135
       
 
       
            try:

     

       
136
       
 
       
                element.writeSvgDefs(idGen, isDuplicate, outputFile, False)

     

       
137
       
 
       
            except AttributeError:

     
···

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

     

       
140
       
 
       
        # Generate ids for normal elements

     

       
141
       
 
       
        prevDefSet = set(prevSet)

     

       
142
       
-
       
        for element in self.elements:

     

       
143
       
 
       
            try:

     

       
144
       
 
       
                element.writeSvgElement(idGen, isDuplicate, outputFile, True)

     

       
145
       
 
       
            except AttributeError:

     

       
146
       
 
       
                pass

     

       
147
       
 
       
        prevSet = prevDefSet

     

       
148
       
 
       
        # Write normal elements

     

       
149
       
-
       
        for element in self.elements:

     

       
150
       
 
       
            try:

     

       
151
       
 
       
                element.writeSvgElement(idGen, isDuplicate, outputFile, False)

     

       
152
       
 
       
                outputFile.write('\n')

     

···

       
3
       
 
       
import base64

     

       
4
       
 
       
import urllib.parse

     

       
5
       
 
       
import re

     

       
6
       
+
       
from collections import defaultdict

     

       
7
       
 
       


     

       
8
       
 
       
from . import Raster

     

       
9
       
 
       
from . import elements as elementsModule

     
···

       
33
       
 
       
        self.viewBox = (self.viewBox[0], -self.viewBox[1]-self.viewBox[3],

     

       
34
       
 
       
                        self.viewBox[2], self.viewBox[3])

     

       
35
       
 
       
        self.elements = []

     

       
36
       
+
       
        self.orderedElements = defaultdict(list)

     

       
37
       
 
       
        self.otherDefs = []

     

       
38
       
 
       
        self.pixelScale = 1

     

       
39
       
 
       
        self.renderWidth = None

     
···

       
69
       
 
       
            return self.renderWidth, self.height * s

     

       
70
       
 
       
        else:

     

       
71
       
 
       
            return self.renderWidth, self.renderHeight

     

       
72
       
+
       
    def draw(self, obj, *, z=None, **kwargs):

     

       
73
       
 
       
        if obj is None:

     

       
74
       
 
       
            return

     

       
75
       
 
       
        if not hasattr(obj, 'writeSvgElement'):

     
···

       
77
       
 
       
        else:

     

       
78
       
 
       
            assert len(kwargs) == 0

     

       
79
       
 
       
            elements = (obj,)

     

       
80
       
+
       
        self.extend(elements, z=z)

     

       
81
       
+
       
    def append(self, element, *, z=None):

     

       
82
       
+
       
        if z is not None:

     

       
83
       
+
       
            self.orderedElements[z].append(element)

     

       
84
       
+
       
        else:

     

       
85
       
+
       
            self.elements.append(element)

     

       
86
       
+
       
    def extend(self, iterable, *, z=None):

     

       
87
       
+
       
        if z is not None:

     

       
88
       
+
       
            self.orderedElements[z].extend(iterable)

     

       
89
       
+
       
        else:

     

       
90
       
+
       
            self.elements.extend(iterable)

     

       
91
       
 
       
    def insert(self, i, element):

     

       
92
       
 
       
        self.elements.insert(i, element)

     

       
93
       
 
       
    def remove(self, element):

     
···

       
109
       
 
       
        self.otherDefs.extend(elements)

     

       
110
       
 
       
    def appendDef(self, element):

     

       
111
       
 
       
        self.otherDefs.append(element)

     

       
112
       
+
       
    def allElements(self):

     

       
113
       
+
       
        ''' Returns self.elements and self.orderedElements as a single list. '''

     

       
114
       
+
       
        output = list(self.elements)

     

       
115
       
+
       
        for z in sorted(self.orderedElements):

     

       
116
       
+
       
            output.extend(self.orderedElements[z])

     

       
117
       
+
       
        return output

     

       
118
       
 
       
    def asSvg(self, outputFile=None):

     

       
119
       
 
       
        returnString = outputFile is None

     

       
120
       
 
       
        if returnString:

     
···

       
145
       
 
       
                outputFile.write('\n')

     

       
146
       
 
       
            except AttributeError:

     

       
147
       
 
       
                pass

     

       
148
       
+
       
        allElements = self.allElements()

     

       
149
       
+
       
        for element in allElements:

     

       
150
       
 
       
            try:

     

       
151
       
 
       
                element.writeSvgDefs(idGen, isDuplicate, outputFile, False)

     

       
152
       
 
       
            except AttributeError:

     
···

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

     

       
155
       
 
       
        # Generate ids for normal elements

     

       
156
       
 
       
        prevDefSet = set(prevSet)

     

       
157
       
+
       
        for element in allElements:

     

       
158
       
 
       
            try:

     

       
159
       
 
       
                element.writeSvgElement(idGen, isDuplicate, outputFile, True)

     

       
160
       
 
       
            except AttributeError:

     

       
161
       
 
       
                pass

     

       
162
       
 
       
        prevSet = prevDefSet

     

       
163
       
 
       
        # Write normal elements

     

       
164
       
+
       
        for element in allElements:

     

       
165
       
 
       
            try:

     

       
166
       
 
       
                element.writeSvgElement(idGen, isDuplicate, outputFile, False)

     

       
167
       
 
       
                outputFile.write('\n')