def __init__(self, tree):
if isinstance(tree, basestring):
tree = self.decode(tree)
if len(tree) != 2:
raise TracError("Tree must only have one trunk!")
(name,branch) = tree
self.xml = tag.map( self.node(name,branch), version="0.8.0")
def __init__(self, tree):
if isinstance(tree, basestring):
tree = self.decode(tree)
if len(tree) != 2:
raise TracError("Tree must only have one trunk!")
(name, branch) = tree
self.xml = tag.map(self.node(name, branch), version="0.8.0")
def expand_macro(self, formatter_or_context, name, content):
"""Return the HTML output of the macro.
:param formatter_or_context: a Formatter when called as a macro,
a Context when called by `GraphvizPlugin.render`
:param name: Wiki macro command that resulted in this method being
called. In this case, it should be 'graphviz', followed
(or not) by the processor name, then by an output
format, as following: graphviz.<processor>/<format>
Valid processor names are: dot, neato, twopi, circo,
and fdp. The default is dot.
Valid output formats are: jpg, png, gif, svg and svgz.
The default is the value specified in the out_format
configuration parameter. If out_format is not specified
in the configuration, then the default is png.
examples: graphviz.dot/png -> dot png
graphviz.neato/jpg -> neato jpg
graphviz.circo -> circo png
graphviz/svg -> dot svg
:param content: The text the user entered for the macro to process.
"""
# check and load the configuration
errmsg = self._load_config()
if errmsg:
return self._error_div(errmsg)
## Extract processor and format from name
processor = out_format = None
# first try with the RegExp engine
try:
m = re.match('graphviz\.?([a-z]*)\/?([a-z]*)', name)
(processor, out_format) = m.group(1, 2)
# or use the string.split method
except:
(d_sp, s_sp) = (name.split('.'), name.split('/'))
if len(d_sp) > 1:
s_sp = d_sp[1].split('/')
if len(s_sp) > 1:
out_format = s_sp[1]
processor = s_sp[0]
elif len(s_sp) > 1:
out_format = s_sp[1]
# assign default values, if instance ones are empty
if not out_format:
out_format = self.out_format
if not processor:
processor = self.processor
if processor in Graphviz.Processors:
proc_cmd = self.cmds[processor]
else:
self.log.error('render_macro: requested processor (%s) not found.' %
processor)
return self._error_div('requested processor (%s) not found.' %
processor)
if out_format not in Graphviz.Formats:
self.log.error('render_macro: requested format (%s) not found.' %
out_format)
return self._error_div(
tag.p(_("Graphviz macro processor error: "
"requested format (%(fmt)s) not valid.",
fmt=out_format)))
encoded_cmd = (processor + unicode(self.processor_options)) \
.encode(self.encoding)
encoded_content = content.encode(self.encoding)
sha_key = sha1(encoded_cmd + encoded_content).hexdigest()
img_name = '%s.%s.%s' % (sha_key, processor, out_format)
# cache: hash.<dot>.<png>
img_path = os.path.join(self.cache_dir, img_name)
map_name = '%s.%s.map' % (sha_key, processor)
# cache: hash.<dot>.map
map_path = os.path.join(self.cache_dir, map_name)
# Check for URL="" presence in graph code
URL_in_graph = 'URL=' in content
# Create image if not in cache
if not os.path.exists(img_path):
self._clean_cache()
if URL_in_graph: # translate wiki TracLinks in URL
if isinstance(formatter_or_context, Context):
context = formatter_or_context
else:
context = formatter_or_context.context
content = self._expand_wiki_links(context, out_format,
content)
encoded_content = content.encode(self.encoding)
# Antialias PNGs with rsvg, if requested
if out_format == 'png' and self.png_anti_alias == True:
# 1. SVG output
failure, errmsg = self._launch(
encoded_content, proc_cmd, '-Tsvg',
'-o%s.svg' % img_path, *self.processor_options)
if failure:
return self._error_div(errmsg)
# 2. SVG to PNG rasterization
failure, errmsg = self._launch(
None, self.rsvg_path, '--dpi-x=%d' % self.dpi,
'--dpi-y=%d' % self.dpi, '%s.svg' % img_path, img_path)
if failure:
return self._error_div(errmsg)
else: # Render other image formats
failure, errmsg = self._launch(
encoded_content, proc_cmd, '-T%s' % out_format,
'-o%s' % img_path, *self.processor_options)
if failure:
return self._error_div(errmsg)
# Generate a map file for binary formats
if URL_in_graph and out_format in Graphviz.Bitmap_Formats:
# Create the map if not in cache
if not os.path.exists(map_path):
failure, errmsg = self._launch(
encoded_content, proc_cmd, '-Tcmap',
'-o%s' % map_path, *self.processor_options)
if failure:
return self._error_div(errmsg)
if errmsg:
# there was a warning. Ideally we should be able to use
# `add_warning` here, but that's not possible as the warnings
# are already emitted at this point in the template processing
return self._error_div(errmsg)
# Generate HTML output
img_url = formatter_or_context.href.graphviz(img_name)
# for SVG(z)
if out_format in Graphviz.Vector_Formats:
try: # try to get SVG dimensions
f = open(img_path, 'r')
svg = f.readlines(1024) # don't read all
f.close()
svg = "".join(svg).replace('\n', '')
w = re.search('width="([0-9]+)(.*?)" ', svg)
h = re.search('height="([0-9]+)(.*?)"', svg)
(w_val, w_unit) = w.group(1,2)
(h_val, h_unit) = h.group(1,2)
# Graphviz seems to underestimate height/width for SVG images,
# so we have to adjust them.
# The correction factor seems to be constant.
w_val, h_val = [1.35 * float(x) for x in (w_val, h_val)]
width = unicode(w_val) + w_unit
height = unicode(h_val) + h_unit
except ValueError:
width = height = '100%'
# insert SVG, IE compatibility
return tag.object(
tag.embed(src=img_url, type="image/svg+xml",
width=width, height=height),
data=img_url, type="image/svg+xml",
width=width, height=height)
# for binary formats, add map
elif URL_in_graph and os.path.exists(map_path):
f = open(map_path, 'r')
map = f.readlines()
f.close()
map = "".join(map).replace('\n', '')
return tag(tag.map(Markup(map), id='G'+sha_key, name='G'+sha_key),
tag.img(src=img_url, usemap="#G"+sha_key,
alt=_("GraphViz image")))
else:
return tag.img(src=img_url, alt=_("GraphViz image"))
def expand_macro(self, formatter, name, content):
"""Return the HTML output of the macro."""
req = formatter.req
# check and load the configuration
errmsg = self._load_config()
if errmsg:
return self._error_div(errmsg)
## Extract processor and format from name
processor = out_format = None
# first try with the RegExp engine
try:
m = re.match('graphviz\.?([a-z]*)\/?([a-z]*)', name)
(processor, out_format) = m.group(1, 2)
# or use the string.split method
except:
(d_sp, s_sp) = (name.split('.'), name.split('/'))
if len(d_sp) > 1:
s_sp = d_sp[1].split('/')
if len(s_sp) > 1:
out_format = s_sp[1]
processor = s_sp[0]
elif len(s_sp) > 1:
out_format = s_sp[1]
# assign default values, if instance ones are empty
if not out_format:
out_format = self.out_format
if not processor:
processor = self.processor
if processor in Graphviz.Processors:
proc_cmd = self.cmds[processor]
else:
self.log.error('render_macro: requested processor (%s) not found.' %
processor)
return self._error_div('requested processor (%s) not found.' %
processor)
if out_format not in Graphviz.Formats:
self.log.error('render_macro: requested format (%s) not found.' %
out_format)
return self._error_div(
tag.p(_("Graphviz macro processor error: "
"requested format (%(fmt)s) not valid.",
fmt=out_format)))
encoded_cmd = (processor + unicode(self.processor_options)) \
.encode(self.encoding)
encoded_content = content.encode(self.encoding)
sha_key = sha.new(encoded_cmd + encoded_content).hexdigest()
img_name = '%s.%s.%s' % (sha_key, processor, out_format)
# cache: hash.<dot>.<png>
img_path = os.path.join(self.cache_dir, img_name)
map_name = '%s.%s.map' % (sha_key, processor)
# cache: hash.<dot>.map
map_path = os.path.join(self.cache_dir, map_name)
# Check for URL="" presence in graph code
URL_in_graph = 'URL=' in content
# Create image if not in cache
if not os.path.exists(img_path):
self._clean_cache()
if URL_in_graph: # translate wiki TracLinks in URL
content = self._expand_wiki_links(formatter, out_format,
content)
encoded_content = content.encode(self.encoding)
# Antialias PNGs with rsvg, if requested
if out_format == 'png' and self.png_anti_alias == True:
# 1. SVG output
errmsg = self._launch(encoded_content, proc_cmd, '-Tsvg',
'-o%s.svg' % img_path,
*self.processor_options)
if errmsg:
return self._error_div(errmsg)
# 2. SVG to PNG rasterization
errmsg = self._launch(None, self.rsvg_path,
'--dpi-x=%d' % self.dpi,
'--dpi-y=%d' % self.dpi,
'%s.svg' % img_path, img_path)
if errmsg:
return self._error_div(errmsg)
else: # Render other image formats
errmsg = self._launch(encoded_content, proc_cmd,
'-T%s' % out_format,
'-o%s' % img_path,
*self.processor_options)
if errmsg:
return self._error_div(errmsg)
# Generate a map file for binary formats
if URL_in_graph and out_format in Graphviz.Bitmap_Formats:
# Create the map if not in cache
if not os.path.exists(map_path):
errmsg = self._launch(encoded_content, proc_cmd, '-Tcmap',
'-o%s' % map_path,
*self.processor_options)
if errmsg:
return self._error_div(errmsg)
# Generate HTML output
img_url = formatter.href.graphviz(img_name)
# for SVG(z)
if out_format in Graphviz.Vector_Formats:
try: # try to get SVG dimensions
f = open(img_path, 'r')
svg = f.readlines(1024) # don't read all
f.close()
svg = "".join(svg).replace('\n', '')
w = re.search('width="([0-9]+)(.*?)" ', svg)
h = re.search('height="([0-9]+)(.*?)"', svg)
(w_val, w_unit) = w.group(1,2)
(h_val, h_unit) = h.group(1,2)
# Graphviz seems to underestimate height/width for SVG images,
# so we have to adjust them.
# The correction factor seems to be constant.
w_val, h_val = [1.35 * float(x) for x in (w_val, h_val)]
width = unicode(w_val) + w_unit
height = unicode(h_val) + h_unit
except ValueError:
width = height = '100%'
# insert SVG, IE compatibility
return tag.object(
tag.embed(src=img_url, type="image/svg+xml",
width=width, height=height),
data=img_url, type="image/svg+xml",
width=width, height=height)
# for binary formats, add map
elif URL_in_graph and os.path.exists(map_path):
f = open(map_path, 'r')
map = f.readlines()
f.close()
map = "".join(map).replace('\n', '')
return tag(tag.map(Markup(map), id='G'+sha_key, name='G'+sha_key),
tag.img(src=img_url, usemap="#G"+sha_key,
alt=_("GraphViz image")))
else:
return tag.img(src=img_url, alt=_("GraphViz image"))
def expand_macro(self, formatter_or_context, name, content):
"""Return the HTML output of the macro.
:param formatter_or_context: a Formatter when called as a macro,
a Context when called by `GraphvizPlugin.render`
:param name: Wiki macro command that resulted in this method being
called. In this case, it should be 'graphviz', followed
(or not) by the processor name, then by an output
format, as following: graphviz.<processor>/<format>
Valid processor names are: dot, neato, twopi, circo,
and fdp. The default is dot.
Valid output formats are: jpg, png, gif, svg and svgz.
The default is the value specified in the out_format
configuration parameter. If out_format is not specified
in the configuration, then the default is png.
examples: graphviz.dot/png -> dot png
graphviz.neato/jpg -> neato jpg
graphviz.circo -> circo png
graphviz/svg -> dot svg
:param content: The text the user entered for the macro to process.
"""
# check and load the configuration
errmsg = self._load_config()
if errmsg:
return self._error_div(errmsg)
## Extract processor and format from name
processor = out_format = None
# first try with the RegExp engine
try:
m = re.match('graphviz\.?([a-z]*)\/?([a-z]*)', name)
(processor, out_format) = m.group(1, 2)
# or use the string.split method
except:
(d_sp, s_sp) = (name.split('.'), name.split('/'))
if len(d_sp) > 1:
s_sp = d_sp[1].split('/')
if len(s_sp) > 1:
out_format = s_sp[1]
processor = s_sp[0]
elif len(s_sp) > 1:
out_format = s_sp[1]
# assign default values, if instance ones are empty
if not out_format:
out_format = self.out_format
if not processor:
processor = self.processor
if processor in Graphviz.Processors:
proc_cmd = self.cmds[processor]
else:
self.log.error(
'render_macro: requested processor (%s) not found.' %
processor)
return self._error_div('requested processor (%s) not found.' %
processor)
if out_format not in Graphviz.Formats:
self.log.error('render_macro: requested format (%s) not found.' %
out_format)
return self._error_div(
tag.p(
_(
"Graphviz macro processor error: "
"requested format (%(fmt)s) not valid.",
fmt=out_format)))
encoded_cmd = (processor + unicode(self.processor_options)) \
.encode(self.encoding)
encoded_content = content.encode(self.encoding)
sha_key = sha1(encoded_cmd + encoded_content).hexdigest()
img_name = '%s.%s.%s' % (sha_key, processor, out_format)
# cache: hash.<dot>.<png>
img_path = os.path.join(self.cache_dir, img_name)
map_name = '%s.%s.map' % (sha_key, processor)
# cache: hash.<dot>.map
map_path = os.path.join(self.cache_dir, map_name)
# Check for URL="" presence in graph code
URL_in_graph = 'URL=' in content
# Create image if not in cache
if not os.path.exists(img_path):
self._clean_cache()
if URL_in_graph: # translate wiki TracLinks in URL
if isinstance(formatter_or_context, Context):
context = formatter_or_context
else:
context = formatter_or_context.context
content = self._expand_wiki_links(context, out_format, content)
encoded_content = content.encode(self.encoding)
# Antialias PNGs with rsvg, if requested
if out_format == 'png' and self.png_anti_alias == True:
# 1. SVG output
failure, errmsg = self._launch(encoded_content, proc_cmd,
'-Tsvg', '-o%s.svg' % img_path,
*self.processor_options)
if failure:
return self._error_div(errmsg)
# 2. SVG to PNG rasterization
failure, errmsg = self._launch(None, self.rsvg_path,
'--dpi-x=%d' % self.dpi,
'--dpi-y=%d' % self.dpi,
'%s.svg' % img_path, img_path)
if failure:
return self._error_div(errmsg)
else: # Render other image formats
failure, errmsg = self._launch(encoded_content, proc_cmd,
'-T%s' % out_format,
'-o%s' % img_path,
*self.processor_options)
if failure:
return self._error_div(errmsg)
# Generate a map file for binary formats
if URL_in_graph and out_format in Graphviz.Bitmap_Formats:
# Create the map if not in cache
if not os.path.exists(map_path):
failure, errmsg = self._launch(encoded_content, proc_cmd,
'-Tcmap', '-o%s' % map_path,
*self.processor_options)
if failure:
return self._error_div(errmsg)
if errmsg:
# there was a warning. Ideally we should be able to use
# `add_warning` here, but that's not possible as the warnings
# are already emitted at this point in the template processing
return self._error_div(errmsg)
# Generate HTML output
img_url = formatter_or_context.href.graphviz(img_name)
# for SVG(z)
if out_format in Graphviz.Vector_Formats:
try: # try to get SVG dimensions
f = open(img_path, 'r')
svg = f.readlines(1024) # don't read all
f.close()
svg = "".join(svg).replace('\n', '')
w = re.search('width="([0-9]+)(.*?)" ', svg)
h = re.search('height="([0-9]+)(.*?)"', svg)
(w_val, w_unit) = w.group(1, 2)
(h_val, h_unit) = h.group(1, 2)
# Graphviz seems to underestimate height/width for SVG images,
# so we have to adjust them.
# The correction factor seems to be constant.
w_val, h_val = [1.35 * float(x) for x in (w_val, h_val)]
width = unicode(w_val) + w_unit
height = unicode(h_val) + h_unit
except ValueError:
width = height = '100%'
# insert SVG, IE compatibility
return tag.object(tag.embed(src=img_url,
type="image/svg+xml",
width=width,
height=height),
data=img_url,
type="image/svg+xml",
width=width,
height=height)
# for binary formats, add map
elif URL_in_graph and os.path.exists(map_path):
f = open(map_path, 'r')
map = f.readlines()
f.close()
map = "".join(map).replace('\n', '')
return tag(
tag.map(Markup(map), id='G' + sha_key, name='G' + sha_key),
tag.img(src=img_url,
usemap="#G" + sha_key,
alt=_("GraphViz image")))
else:
return tag.img(src=img_url, alt=_("GraphViz image"))
