diff options
| -rwxr-xr-x | scripts/tikz.py | 67 |
1 files changed, 67 insertions, 0 deletions
diff --git a/scripts/tikz.py b/scripts/tikz.py new file mode 100755 index 000000000..7e1ed7927 --- /dev/null +++ b/scripts/tikz.py @@ -0,0 +1,67 @@ +#!/usr/bin/env python + +""" +Pandoc filter to process raw latex tikz environments into images. +Assumes that pdflatex is in the path, and that the standalone +package is available. Also assumes that ImageMagick's convert +is in the path. Images are put in the tikz-images directory. +""" + +import hashlib +import re +import os +import sys +import shutil +from pandoc import toJSONFilter +from subprocess import Popen, PIPE, call +from tempfile import mkdtemp + +imagedir = "tikz-images" + +def sha1(x): + return hashlib.sha1(x).hexdigest() + +def tikz2image(tikz, filetype, outfile): + tmpdir = mkdtemp() + olddir = os.getcwd() + os.chdir(tmpdir) + f = open('tikz.tex', 'w') + f.write("""\\documentclass{standalone} + \\usepackage{tikz} + \\begin{document} + """) + f.write(tikz) + f.write("\n\\end{document}\n") + f.close() + p = call(["pdflatex", 'tikz.tex'], stdout=sys.stderr) + os.chdir(olddir) + if filetype == 'pdf': + shutil.copyfile(tmpdir + '/tikz.pdf', outfile + '.pdf') + else: + call(["convert", tmpdir + '/tikz.pdf', outfile + '.' + filetype]) + shutil.rmtree(tmpdir) + +def tikz(key, value, format): + if key == 'RawBlock': + [fmt, code] = value + if fmt['unFormat'] == "latex" and re.match("\\\\begin{tikzpicture}", code): + outfile = imagedir + '/' + sha1(code) + if format == "html": + filetype = "png" + elif format == "latex": + filetype = "pdf" + else: + filetype = "png" + src = outfile + '.' + filetype + if not os.path.isfile(src): + try: + os.mkdir(imagedir) + sys.stderr.write('Created directory ' + imagedir + '\n') + except OSError: + pass + tikz2image(code, filetype, outfile) + sys.stderr.write('Created image ' + src + '\n') + return {'Para': [{'Image': [[], [src,""]]}]} + +if __name__ == "__main__": + toJSONFilter(tikz) |