1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
#!/usr/bin/env python3
# Licensed BSD-3-Clause: https://spdx.org/licenses/BSD-3-Clause.html
"""
Pandoc filter to process code blocks with class "graphviz" into
SVG based graphviz-generated images.
Needs pygraphviz
"""
# Based on https://github.com/jgm/pandocfilters/blob/master/examples/graphviz.py
import sys, os, pygraphviz
from pandocfilters import toJSONFilter, Para, Image, RawBlock, get_filename4code, get_caption, get_extension, get_value
# https://pandoc.org/lua-filters.html
def graphviz (key, value, frmat, _):
if key == 'CodeBlock':
[[ident, classes, keyvals], code] = value
if "graphviz" in classes:
# sys.stderr.write ('Debug: ' + str (value) + '\n')
caption, typef, keyvals = get_caption (keyvals)
prog, keyvals = get_value (keyvals, u"prog", u"dot")
force, keyvals = get_value (keyvals, u"force", u"false")
filetype = get_extension (frmat, "svg", html ="svg", latex ="pdf")
dest = get_filename4code ("graphviz", str (value), filetype)
if force != u"false" or not os.path.isfile (dest):
g = pygraphviz.AGraph (string = code)
g.layout ()
g.draw (dest, prog = prog)
sys.stderr.write ('Create image: ' + dest + '\n')
if dest.endswith ('.svg'):
return RawBlock (u'html', '<object type="image/svg+xml" data="%s"></object>' % dest)
return Para ([Image ([ident, [], keyvals], caption, [dest, typef])])
if __name__ == "__main__":
toJSONFilter (graphviz)
|
