Code

support for 5 point spline (Bug 685707)
[inkscape.git] / share / extensions / dxf_input.py
index d2e645ea1dec9a44babd97633cd30c7118749f7e..ee52b8c30b447a061776e5bce298c2a17c4e2c3a 100644 (file)
@@ -24,24 +24,35 @@ Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
 import inkex, simplestyle, math
 from StringIO import StringIO
+from urllib import quote
 
 def export_MTEXT():
-    # mandatory group codes : (1, 10, 20) (text, x, y)
-    if vals[groups['1']] and vals[groups['10']] and vals[groups['20']]:
+    # mandatory group codes : (1 or 3, 10, 20) (text, x, y)
+    if (vals[groups['1']] or vals[groups['3']]) and vals[groups['10']] and vals[groups['20']]:
         x = vals[groups['10']][0]
         y = vals[groups['20']][0]
-        # optional group codes : (40, 50) (text height mm, text angle)
+        # optional group codes : (21, 40, 50) (direction, text height mm, text angle)
         size = 12                       # default fontsize in px
         if vals[groups['40']]:
             size = scale*vals[groups['40']][0]
-        attribs = {'x': '%f' % x, 'y': '%f' % y, 'style': 'font-size: %dpx; fill: %s' % (size, color)}
+        attribs = {'x': '%f' % x, 'y': '%f' % y, 'style': 'font-size: %.1fpx; fill: %s; font-family: %s' % (size, color, options.font)}
         angle = 0                       # default angle in degrees
         if vals[groups['50']]:
             angle = vals[groups['50']][0]
             attribs.update({'transform': 'rotate (%f %f %f)' % (-angle, x, y)})
+        elif vals[groups['21']]:
+            if vals[groups['21']][0] == 1.0:
+                attribs.update({'transform': 'rotate (%f %f %f)' % (-90, x, y)})
+            elif vals[groups['21']][0] == -1.0:
+                attribs.update({'transform': 'rotate (%f %f %f)' % (90, x, y)})
         attribs.update({inkex.addNS('linespacing','sodipodi'): '125%'})
         node = inkex.etree.SubElement(layer, 'text', attribs)
-        text = vals[groups['1']][0]
+        text = ''
+        if vals[groups['3']]:
+            for i in range (0, len(vals[groups['3']])):
+                text += vals[groups['3']][i]
+        if vals[groups['1']]:
+            text += vals[groups['1']][0]
         found = text.find('\P')         # new line
         while found > -1:
             tspan = inkex.etree.SubElement(node , 'tspan', {inkex.addNS('role','sodipodi'): 'line'})
@@ -54,7 +65,10 @@ def export_MTEXT():
 def export_POINT():
     # mandatory group codes : (10, 20) (x, y)
     if vals[groups['10']] and vals[groups['20']]:
-        generate_ellipse(vals[groups['10']][0], vals[groups['20']][0], w/2, 0.0, 1.0, 0.0, 0.0)
+        if options.gcodetoolspoints:
+            generate_gcodetools_point(vals[groups['10']][0], vals[groups['20']][0])
+        else:
+            generate_ellipse(vals[groups['10']][0], vals[groups['20']][0], w/2, 0.0, 1.0, 0.0, 0.0)
 
 def export_LINE():
     # mandatory group codes : (10, 11, 20, 21) (x1, x2, y1, y2)
@@ -74,6 +88,10 @@ def export_SPLINE():
             path = 'M %f,%f Q %f,%f %f,%f' % (vals[groups['10']][0], vals[groups['20']][0], vals[groups['10']][1], vals[groups['20']][1], vals[groups['10']][2], vals[groups['20']][2])
             attribs = {'d': path, 'style': style}
             inkex.etree.SubElement(layer, 'path', attribs)
+        if not (vals[groups['70']][0] & 3) and len(vals[groups['10']]) == 5 and len(vals[groups['20']]) == 5:
+            path = 'M %f,%f Q %f,%f %f,%f Q %f,%f %f,%f' % (vals[groups['10']][0], vals[groups['20']][0], vals[groups['10']][1], vals[groups['20']][1], vals[groups['10']][2], vals[groups['20']][2], vals[groups['10']][3], vals[groups['20']][3], vals[groups['10']][4], vals[groups['20']][4])
+            attribs = {'d': path, 'style': style}
+            inkex.etree.SubElement(layer, 'path', attribs)
 
 def export_CIRCLE():
     # mandatory group codes : (10, 20, 40) (x, y, radius)
@@ -107,6 +125,10 @@ def export_LWPOLYLINE():
             # optional group codes : (42) (bulge)
             iseqs = 0
             ibulge = 0
+            if vals[groups['70']][0]:           # closed path
+                seqs.append('20')
+                vals[groups['10']].append(vals[groups['10']][0])
+                vals[groups['20']].append(vals[groups['20']][0])
             while seqs[iseqs] != '20':
                 iseqs += 1
             path = 'M %f,%f' % (vals[groups['10']][0], vals[groups['20']][0])
@@ -135,14 +157,14 @@ def export_LWPOLYLINE():
                     path += ' L %f,%f' % (vals[groups['10']][i], vals[groups['20']][i])
                 xold = vals[groups['10']][i]
                 yold = vals[groups['20']][i]
-            if vals[groups['70']][0] == 1:      # closed path
+            if vals[groups['70']][0]:           # closed path
                 path += ' z'
             attribs = {'d': path, 'style': style}
             inkex.etree.SubElement(layer, 'path', attribs)
 
 def export_HATCH():
-    # mandatory group codes : (10, 20, 72, 93) (x, y, Edge Type, Number of edges)
-    if vals[groups['10']] and vals[groups['20']] and vals[groups['72']] and vals[groups['93']]:
+    # mandatory group codes : (10, 20, 70, 72, 92, 93) (x, y, fill, Edge Type, Path Type, Number of edges)
+    if vals[groups['10']] and vals[groups['20']] and vals[groups['70']] and vals[groups['72']] and vals[groups['92']] and vals[groups['93']]:
         if len(vals[groups['10']]) > 1 and len(vals[groups['20']]) == len(vals[groups['10']]):
             # optional group codes : (11, 21, 40, 50, 51, 73) (x, y, r, angle1, angle2, CCW)
             i10 = 1    # count start points
@@ -161,7 +183,12 @@ def export_HATCH():
                     a1 = 0
                     path += 'M %f,%f ' % (xc, yc)
                 for j in range(0, vals[groups['93']][i]):
-                    if vals[groups['72']][i72] == 2:        # arc
+                    if vals[groups['92']][i] & 2:           # polyline
+                        if j > 0:
+                            path += 'L %f,%f ' % (vals[groups['10']][i10], vals[groups['20']][i10])
+                        if j == vals[groups['93']][i] - 1:
+                            i72 += 1
+                    elif vals[groups['72']][i72] == 2:      # arc
                         xc = vals[groups['10']][i10]
                         yc = vals[groups['20']][i10]
                         rm = scale*vals[groups['40']][i40]
@@ -180,14 +207,12 @@ def export_HATCH():
                         path += 'L %f,%f ' % (scale*(vals[groups['11']][i11] - xmin), -scale*(vals[groups['21']][i11] - ymax))
                         i11 += 1
                         i72 += 1
-                    elif vals[groups['72']][i72] == 0:      # polyline
-                        if j > 0:
-                            path += 'L %f,%f ' % (vals[groups['10']][i10], vals[groups['20']][i10])
-                        if j == vals[groups['93']][i] - 1:
-                            i72 += 1
                     i10 += 1
                 path += "z "
-            style = simplestyle.formatStyle({'fill': '%s' % color})
+            if vals[groups['70']][0]:
+                style = simplestyle.formatStyle({'fill': '%s' % color})
+            else:
+                style = simplestyle.formatStyle({'fill': 'url(#Hatch)', 'fill-opacity': '1.0'})
             attribs = {'d': path, 'style': style}
             inkex.etree.SubElement(layer, 'path', attribs)
 
@@ -206,23 +231,52 @@ def export_DIMENSION():
             path = 'M %f,%f %f,%f' % (vals[groups['10']][0], vals[groups['20']][0], vals[groups['10']][0], vals[groups['23']][0])
         else:
             return
-        attribs = {'d': path, 'style': style + '; marker-start: url(#DistanceX); marker-end: url(#DistanceX)'}
+        attribs = {'d': path, 'style': style + '; marker-start: url(#DistanceX); marker-end: url(#DistanceX); stroke-width: 0.25px'}
         inkex.etree.SubElement(layer, 'path', attribs)
         x = scale*(vals[groups['11']][0] - xmin)
         y = - scale*(vals[groups['21']][0] - ymax)
-        size = 3                    # default fontsize in px
-        attribs = {'x': '%f' % x, 'y': '%f' % y, 'style': 'font-size: %dpx; fill: %s' % (size, color)}
+        size = 12                   # default fontsize in px
+        if vals[groups['3']]:
+            if DIMTXT.has_key(vals[groups['3']][0]):
+                size = scale*DIMTXT[vals[groups['3']][0]]
+                if size < 2:
+                    size = 2
+        attribs = {'x': '%f' % x, 'y': '%f' % y, 'style': 'font-size: %.1fpx; fill: %s; font-family: %s; text-anchor: middle; text-align: center' % (size, color, options.font)}
         if dx == 0:
             attribs.update({'transform': 'rotate (%f %f %f)' % (-90, x, y)})
         node = inkex.etree.SubElement(layer, 'text', attribs)
         tspan = inkex.etree.SubElement(node , 'tspan', {inkex.addNS('role','sodipodi'): 'line'})
-        tspan.text = '%.2f' % d
+        tspan.text = str(float('%.2f' % d))
+
+def export_INSERT():
+    # mandatory group codes : (2, 10, 20) (block name, x, y)
+    if vals[groups['2']] and vals[groups['10']] and vals[groups['20']]:
+        x = vals[groups['10']][0]
+        y = vals[groups['20']][0] - scale*ymax
+        attribs = {'x': '%f' % x, 'y': '%f' % y, inkex.addNS('href','xlink'): '#' + quote(vals[groups['2']][0].encode("utf-8"))}
+        inkex.etree.SubElement(layer, 'use', attribs)
+
+def export_BLOCK():
+    # mandatory group codes : (2) (block name)
+    if vals[groups['2']]:
+        global block
+        block = inkex.etree.SubElement(defs, 'symbol', {'id': vals[groups['2']][0]})
+
+def export_ENDBLK():
+    global block
+    block = defs                                    # initiallize with dummy
+
+def export_ATTDEF():
+    # mandatory group codes : (1, 2) (default, tag)
+    if vals[groups['1']] and vals[groups['2']]:
+        vals[groups['1']][0] = vals[groups['2']][0]
+        export_MTEXT()
 
 def generate_ellipse(xc, yc, xm, ym, w, a1, a2):
     rm = math.sqrt(xm*xm + ym*ym)
     a = math.atan2(ym, xm)
     diff = (a2 - a1 + 2*math.pi) % (2*math.pi)
-    if diff:                        # open arc
+    if abs(diff) > 0.0000001 and abs(diff - 2*math.pi) > 0.0000001: # open arc
         large = 0                   # large-arc-flag
         if diff > math.pi:
             large = 1
@@ -240,6 +294,11 @@ def generate_ellipse(xc, yc, xm, ym, w, a1, a2):
     attribs = {'d': path, 'style': style}
     inkex.etree.SubElement(layer, 'path', attribs)
 
+def generate_gcodetools_point(xc, yc):
+    path= 'm %s,%s 2.9375,-6.34375 0.8125,1.90625 6.84375,-6.84375 0,0 0.6875,0.6875 -6.84375,6.84375 1.90625,0.8125 z' % (xc,yc)
+    attribs = {'d': path, inkex.addNS('dxfpoint','inkscape'):'1', 'style': 'stroke:#ff0000;fill:#ff0000'}
+    inkex.etree.SubElement(layer, 'path', attribs)
+
 def get_line():
     return (stream.readline().strip(), stream.readline().strip())
 
@@ -252,57 +311,118 @@ def get_group(group):
 
 #   define DXF Entities and specify which Group Codes to monitor
 
-entities = {'MTEXT': export_MTEXT, 'TEXT': export_MTEXT, 'POINT': export_POINT, 'LINE': export_LINE, 'SPLINE': export_SPLINE, 'CIRCLE': export_CIRCLE, 'ARC': export_ARC, 'ELLIPSE': export_ELLIPSE, 'LEADER': export_LEADER, 'LWPOLYLINE': export_LWPOLYLINE, 'HATCH': export_HATCH, 'DIMENSION': export_DIMENSION, 'ENDSEC': ''}
-groups = {'1': 0, '8': 1, '10': 2, '11': 3, '13': 4, '14': 5, '20': 6, '21': 7, '23': 8, '24': 9, '40': 10, '41': 11, '42': 12, '50': 13, '51': 14, '62': 15, '70': 16, '72': 17, '73': 18, '93': 19, '370': 20}
+entities = {'MTEXT': export_MTEXT, 'TEXT': export_MTEXT, 'POINT': export_POINT, 'LINE': export_LINE, 'SPLINE': export_SPLINE, 'CIRCLE': export_CIRCLE, 'ARC': export_ARC, 'ELLIPSE': export_ELLIPSE, 'LEADER': export_LEADER, 'LWPOLYLINE': export_LWPOLYLINE, 'HATCH': export_HATCH, 'DIMENSION': export_DIMENSION, 'INSERT': export_INSERT, 'BLOCK': export_BLOCK, 'ENDBLK': export_ENDBLK, 'ATTDEF': export_ATTDEF, 'VIEWPORT': False, 'DICTIONARY': False}
+groups = {'1': 0, '2': 1, '3': 2, '6': 3, '8': 4, '10': 5, '11': 6, '13': 7, '14': 8, '20': 9, '21': 10, '23': 11, '24': 12, '40': 13, '41': 14, '42': 15, '50': 16, '51': 17, '62': 18, '70': 19, '72': 20, '73': 21, '92': 22, '93': 23, '370': 24}
 colors = {  1: '#FF0000',   2: '#FFFF00',   3: '#00FF00',   4: '#00FFFF',   5: '#0000FF',
-            6: '#FF00FF',   8: '#414141',   9: '#808080',  30: '#FF7F00',
+            6: '#FF00FF',   8: '#414141',   9: '#808080',  12: '#BD0000',  30: '#FF7F00',
           250: '#333333', 251: '#505050', 252: '#696969', 253: '#828282', 254: '#BEBEBE', 255: '#FFFFFF'}
 
-doc = inkex.etree.parse(StringIO('<svg xmlns:sodipodi="http://sodipodi.sourceforge.net/DTD/sodipodi-0.dtd"></svg>'))
-defs = inkex.etree.SubElement(doc.getroot(), 'defs', {} )
+parser = inkex.optparse.OptionParser(usage="usage: %prog [options] SVGfile", option_class=inkex.InkOption)
+parser.add_option("--auto", action="store", type="inkbool", dest="auto", default=True)
+parser.add_option("--scale", action="store", type="string", dest="scale", default="1.0")
+parser.add_option("--gcodetoolspoints", action="store", type="inkbool", dest="gcodetoolspoints", default=True)
+parser.add_option("--encoding", action="store", type="string", dest="input_encode", default="latin_1")
+parser.add_option("--font", action="store", type="string", dest="font", default="Arial")
+parser.add_option("--tab", action="store", type="string", dest="tab", default="Options")
+parser.add_option("--inputhelp", action="store", type="string", dest="inputhelp", default="")
+(options, args) = parser.parse_args(inkex.sys.argv[1:])
+doc = inkex.etree.parse(StringIO('<svg xmlns:sodipodi="http://sodipodi.sourceforge.net/DTD/sodipodi-0.dtd" width="%s" height="%s"></svg>' % (210*90/25.4, 297*90/25.4)))
+desc = inkex.etree.SubElement(doc.getroot(), 'desc', {})
+defs = inkex.etree.SubElement(doc.getroot(), 'defs', {})
 marker = inkex.etree.SubElement(defs, 'marker', {'id': 'DistanceX', 'orient': 'auto', 'refX': '0.0', 'refY': '0.0', 'style': 'overflow:visible'})
 inkex.etree.SubElement(marker, 'path', {'d': 'M 3,-3 L -3,3 M 0,-5 L  0,5', 'style': 'stroke:#000000; stroke-width:0.5'})
-stream = open(inkex.sys.argv[1], 'r')
+pattern = inkex.etree.SubElement(defs, 'pattern', {'id': 'Hatch', 'patternUnits': 'userSpaceOnUse', 'width': '8', 'height': '8', 'x': '0', 'y': '0'})
+inkex.etree.SubElement(pattern, 'path', {'d': 'M8 4 l-4,4', 'stroke': '#000000', 'stroke-width': '0.25', 'linecap': 'square'})
+inkex.etree.SubElement(pattern, 'path', {'d': 'M6 2 l-4,4', 'stroke': '#000000', 'stroke-width': '0.25', 'linecap': 'square'})
+inkex.etree.SubElement(pattern, 'path', {'d': 'M4 0 l-4,4', 'stroke': '#000000', 'stroke-width': '0.25', 'linecap': 'square'})
+stream = open(args[0], 'r')
 xmax = xmin = 0.0
 ymax = 297.0                                        # default A4 height in mm
 line = get_line()
-flag = 0
+flag = 0                                            # (0, 1, 2, 3) = (none, LAYER, LTYPE, DIMTXT)
 layer_colors = {}                                   # store colors by layer
 layer_nodes = {}                                    # store nodes by layer
-while line[0] and line[1] != 'ENTITIES':
+linetypes = {}                                      # store linetypes by name
+DIMTXT = {}                                         # store DIMENSION text sizes
+
+while line[0] and line[1] != 'BLOCKS':
     line = get_line()
-    if line[1] == '$EXTMIN':
-        xmin = get_group('10')
-    if line[1] == '$EXTMAX':
-        xmax = get_group('10')
-        ymax = get_group('20')
-    if flag and line[0] == '2':
-        name = unicode(line[1], "iso-8859-1")
-        attribs = {inkex.addNS('groupmode','inkscape'): 'layer', inkex.addNS('label','inkscape'): '%s' % name}
-        layer_nodes[name] = inkex.etree.SubElement(doc.getroot(), 'g', attribs)
+    if options.auto:
+        if line[1] == '$EXTMIN':
+            xmin = get_group('10')
+        if line[1] == '$EXTMAX':
+            xmax = get_group('10')
+            ymax = get_group('20')
+    if flag == 1 and line[0] == '2':
+        layername = unicode(line[1], options.input_encode)
+        attribs = {inkex.addNS('groupmode','inkscape'): 'layer', inkex.addNS('label','inkscape'): '%s' % layername}
+        layer_nodes[layername] = inkex.etree.SubElement(doc.getroot(), 'g', attribs)
+    if flag == 2 and line[0] == '2':
+        linename = unicode(line[1], options.input_encode)
+        linetypes[linename] = []
+    if flag == 3 and line[0] == '2':
+        stylename = unicode(line[1], options.input_encode)
     if line[0] == '2' and line[1] == 'LAYER':
         flag = 1
-    if flag and line[0] == '62':
-        layer_colors[name] = int(line[1])
+    if line[0] == '2' and line[1] == 'LTYPE':
+        flag = 2
+    if line[0] == '2' and line[1] == 'DIMSTYLE':
+        flag = 3
+    if flag == 1 and line[0] == '62':
+        layer_colors[layername] = int(line[1])
+    if flag == 2 and line[0] == '49':
+        linetypes[linename].append(float(line[1]))
+    if flag == 3 and line[0] == '140':
+        DIMTXT[stylename] = float(line[1])
     if line[0] == '0' and line[1] == 'ENDTAB':
         flag = 0
 
-scale = 90.0/25.4                                   # default convert from mm to pixels
-if xmax > xmin:
-    scale *= 210.0/(xmax - xmin)                    # scale to A4 width
+if options.auto:
+    scale = 1.0
+    if xmax > xmin:
+        scale = 210.0/(xmax - xmin)                 # scale to A4 width
+else:
+    scale = float(options.scale)                    # manual scale factor
+desc.text = '%s - scale = %f' % (unicode(args[0], options.input_encode), scale)
+scale *= 90.0/25.4                                  # convert from mm to pixels
+
+if not layer_nodes:
+    attribs = {inkex.addNS('groupmode','inkscape'): 'layer', inkex.addNS('label','inkscape'): '0'}
+    layer_nodes['0'] = inkex.etree.SubElement(doc.getroot(), 'g', attribs)
+    layer_colors['0'] = 7
+
+for linename in linetypes.keys():                   # scale the dashed lines
+    linetype = ''
+    for length in linetypes[linename]:
+        if length == 0:                             # test for dot
+            linetype += ' 0.5,'
+        else:
+            linetype += '%.4f,' % math.fabs(length*scale)
+    if linetype == '':
+        linetypes[linename] = 'stroke-linecap: round'
+    else:
+        linetypes[linename] = 'stroke-dasharray:' + linetype
+
 entity = ''
-while line[0] and line[1] != 'ENDSEC':
+block = defs                                        # initiallize with dummy
+while line[0] and line[1] != 'DICTIONARY':
     line = get_line()
     if entity and groups.has_key(line[0]):
         seqs.append(line[0])                        # list of group codes
-        if line[0] == '1' or line[0] == '8':        # text value
+        if line[0] == '1' or line[0] == '2' or line[0] == '3' or line[0] == '6' or line[0] == '8':  # text value
             val = line[1].replace('\~', ' ')
             val = inkex.re.sub( '\\\\A.*;', '', val)
             val = inkex.re.sub( '\\\\H.*;', '', val)
+            val = inkex.re.sub( '\\^I', '', val)
+            val = inkex.re.sub( '{\\\\L', '', val)
+            val = inkex.re.sub( '}', '', val)
             val = inkex.re.sub( '\\\\S.*;', '', val)
             val = inkex.re.sub( '\\\\W.*;', '', val)
-            val = unicode(val, "iso-8859-1")
-        elif line[0] == '62' or line[0] == '70' or line[0] == '93':
+            val = unicode(val, options.input_encode)
+            val = val.encode('unicode_escape')
+            val = inkex.re.sub( '\\\\\\\\U\+([0-9A-Fa-f]{4})', '\\u\\1', val)
+            val = val.decode('unicode_escape')
+        elif line[0] == '62' or line[0] == '70' or line[0] == '92' or line[0] == '93':
             val = int(line[1])
         elif line[0] == '10' or line[0] == '13' or line[0] == '14': # scaled float x value
             val = scale*(float(line[1]) - xmin)
@@ -313,9 +433,12 @@ while line[0] and line[1] != 'ENDSEC':
         vals[groups[line[0]]].append(val)
     elif entities.has_key(line[1]):
         if entities.has_key(entity):
-            color = '#000000'                       # default color
-            if vals[groups['8']]:                   # Common Layer Name
+            if block != defs:                       # in a BLOCK
+                layer = block
+            elif vals[groups['8']]:                 # use Common Layer Name
                 layer = layer_nodes[vals[groups['8']][0]]
+            color = '#000000'                       # default color
+            if vals[groups['8']]:
                 if layer_colors.has_key(vals[groups['8']][0]):
                     if colors.has_key(layer_colors[vals[groups['8']][0]]):
                         color = colors[layer_colors[vals[groups['8']][0]]]
@@ -330,11 +453,15 @@ while line[0] and line[1] != 'ENDSEC':
                     if w < 0.5:
                         w = 0.5
                     style = simplestyle.formatStyle({'stroke': '%s' % color, 'fill': 'none', 'stroke-width': '%.1f' % w})
-            entities[entity]()
+            if vals[groups['6']]:                   # Common Linetype
+                if linetypes.has_key(vals[groups['6']][0]):
+                    style += ';' + linetypes[vals[groups['6']][0]]
+            if entities[entity]:
+                entities[entity]()
         entity = line[1]
-        vals = [[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[]]
+        vals = [[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[],[]]
         seqs = []
 
 doc.write(inkex.sys.stdout)
 
-# vim: expandtab shiftwidth=4 tabstop=8 softtabstop=4 encoding=utf-8 textwidth=99
+# vim: expandtab shiftwidth=4 tabstop=8 softtabstop=4 fileencoding=utf-8 textwidth=99