share/extensions/eqtexsvg.py

   1 #!/usr/bin/env python
   2 # -*- coding: cp1252 -*-
   3 """
   4 eqtexsvg.py
   5 functions for converting LaTeX equation string into SVG path
   6 This extension need, to work properly:
   7     - a TeX/LaTeX distribution (MiKTeX ...)
   8     - pstoedit software: <http://www.pstoedit.net/pstoedit>
   9
  10 Copyright (C) 2006 Julien Vitard <julienvitard@gmail.com>
  11
  12 2010-04-04: Added support for custom packages
  13             Christoph Schmidt-Hieber <christsc@gmx.de>
  14
  15 This program is free software; you can redistribute it and/or modify
  16 it under the terms of the GNU General Public License as published by
  17 the Free Software Foundation; either version 2 of the License, or
  18 (at your option) any later version.
  19
  20 This program is distributed in the hope that it will be useful,
  21 but WITHOUT ANY WARRANTY; without even the implied warranty of
  22 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  23 GNU General Public License for more details.
  24
  25 You should have received a copy of the GNU General Public License
  26 along with this program; if not, write to the Free Software
  27 Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
  28
  29 """
  30
  31 import inkex, os, tempfile, sys, xml.dom.minidom
  32
  33 def parse_pkgs(pkgstring):
  34     pkglist = pkgstring.replace(" ","").split(",")
  35     header = ""
  36     for pkg in pkglist:
  37         header += "\\usepackage{%s}\n" % pkg
  38
  39     return header
  40
  41 def create_equation_tex(filename, equation, add_header=""):
  42     tex = open(filename, 'w')
  43     tex.write("""%% processed with eqtexsvg.py
  44 \\documentclass{article}
  45 \\usepackage{amsmath}
  46 \\usepackage{amssymb}
  47 \\usepackage{amsfonts}
  48 """)
  49     tex.write(add_header)
  50     tex.write("""\\thispagestyle{empty}
  51 \\begin{document}
  52 """)
  53     tex.write(equation)
  54     tex.write("\n\\end{document}\n")
  55     tex.close()
  56
  57 def svg_open(self,filename):
  58     doc_width = inkex.unittouu(self.document.getroot().get('width'))
  59     doc_height = inkex.unittouu(self.document.getroot().get('height'))
  60     doc_sizeH = min(doc_width,doc_height)
  61     doc_sizeW = max(doc_width,doc_height)
  62
  63     def clone_and_rewrite(self, node_in):
  64         in_tag = node_in.tag.rsplit('}',1)[-1]
  65         if in_tag != 'svg':
  66             node_out = inkex.etree.Element(inkex.addNS(in_tag,'svg'))
  67             for name in node_in.attrib:
  68                 node_out.set(name, node_in.attrib[name])
  69         else:
  70             node_out = inkex.etree.Element(inkex.addNS('g','svg'))
  71         for c in node_in.iterchildren():
  72             c_tag = c.tag.rsplit('}',1)[-1]
  73             if c_tag in ('g', 'path', 'polyline', 'polygon'):
  74                 child = clone_and_rewrite(self, c)
  75                 if c_tag == 'g':
  76                     child.set('transform','matrix('+str(doc_sizeH/700.)+',0,0,'+str(-doc_sizeH/700.)+','+str(-doc_sizeH*0.25)+','+str(doc_sizeW*0.75)+')')
  77                 node_out.append(child)
  78
  79         return node_out
  80
  81     doc = inkex.etree.parse(filename)
  82     svg = doc.getroot()
  83     group = clone_and_rewrite(self, svg)
  84     self.current_layer.append(group)
  85
  86 class EQTEXSVG(inkex.Effect):
  87     def __init__(self):
  88         inkex.Effect.__init__(self)
  89         self.OptionParser.add_option("-f", "--formule",
  90                         action="store", type="string",
  91                         dest="formula", default="",
  92                         help="LaTeX formula")
  93         self.OptionParser.add_option("-p", "--packages",
  94                         action="store", type="string",
  95                         dest="packages", default="",
  96                         help="Additional packages")
  97     def effect(self):
  98
  99         base_dir = tempfile.mkdtemp("", "inkscape-");
 100         latex_file = os.path.join(base_dir, "eq.tex")
 101         aux_file = os.path.join(base_dir, "eq.aux")
 102         log_file = os.path.join(base_dir, "eq.log")
 103         ps_file = os.path.join(base_dir, "eq.ps")
 104         dvi_file = os.path.join(base_dir, "eq.dvi")
 105         svg_file = os.path.join(base_dir, "eq.svg")
 106         out_file = os.path.join(base_dir, "eq.out")
 107         err_file = os.path.join(base_dir, "eq.err")
 108
 109         def clean():
 110             os.remove(latex_file)
 111             os.remove(aux_file)
 112             os.remove(log_file)
 113             os.remove(ps_file)
 114             os.remove(dvi_file)
 115             os.remove(svg_file)
 116             os.remove(out_file)
 117             if os.path.exists(err_file):
 118                 os.remove(err_file)
 119             os.rmdir(base_dir)
 120
 121         add_header = parse_pkgs(self.options.packages)
 122         create_equation_tex(latex_file, self.options.formula, add_header)
 123         os.system('latex "-output-directory=%s" -halt-on-error "%s" > "%s"' \
 124                   % (base_dir, latex_file, out_file))
 125         try:
 126             os.stat(dvi_file)
 127         except OSError:
 128             print >>sys.stderr, "invalid LaTeX input:"
 129             print >>sys.stderr, self.options.formula
 130             print >>sys.stderr, "temporary files were left in:", base_dir
 131             sys.exit(1)
 132
 133         os.system('dvips -q -f -E -D 600 -y 5000 -o "%s" "%s"' % (ps_file, dvi_file))
 134         # cd to base_dir is necessary, because pstoedit writes
 135         # temporary files to cwd and needs write permissions
 136         separator = ';'
 137         if os.name == 'nt':
 138             separator = '&&'
 139         os.system('cd "%s" %s pstoedit -f plot-svg -dt -ssp "%s" "%s" > "%s" 2> "%s"' \
 140                   % (base_dir, separator, ps_file, svg_file, out_file, err_file))
 141
 142         # forward errors to stderr but skip pstoedit header
 143         if os.path.exists(err_file):
 144             err_stream = open(err_file, 'r')
 145             for line in err_stream:
 146                 if not line.startswith('pstoedit: version'):
 147                     sys.stderr.write(line + '\n')
 148             err_stream.close()
 149
 150         svg_open(self, svg_file)
 151
 152         clean()
 153
 154 if __name__ == '__main__':
 155     e = EQTEXSVG()
 156     e.affect()
 157
 158
 159 # vim: expandtab shiftwidth=4 tabstop=8 softtabstop=4 fileencoding=utf-8 textwidth=99