diff --git a/amrbatch/main.py b/amrbatch/main.py index b9dce1d42a0427cb6b9bc44f060d4cdbbf9736b1..075ae680cc19018448154bae8714bb49f09258aa 100644 --- a/amrbatch/main.py +++ b/amrbatch/main.py @@ -104,19 +104,28 @@ def __generate_dot_amr_graph(filepath_manager, data): """ AMR graph generation in dot and png format """ try: - # -- generating dot/png files using AMRLib and GraphViz + # -- generating dot/png/svg files using AMRLib and GraphViz dot_filename = data.get_dot_amr_graph_output_filepath() - format = 'png' logger.debug(f'----- AMR Graph file (dot): {os.path.basename(dot_filename)}') + format = 'png' plot = AMRPlot(dot_filename, format) plot.build_from_graph(data.graph) - plot.graph.render() - + plot.graph.render() render_fn = dot_filename + '.' + format # -- renaming PNG file good_png_fn = data.get_png_amr_graph_output_filepath() logger.debug(f'----- AMR Graph file (png): {{os.path.basename(good_png_fn)}}') os.rename(render_fn, good_png_fn) + format = 'svg' + plot = AMRPlot(dot_filename, format) + plot.build_from_graph(data.graph) + plot.graph.render() + render_fn = dot_filename + '.' + format # -- renaming PNG file + good_svg_fn = good_png_fn.replace('.png','.svg') + logger.debug(f'----- AMR Graph file (svg): {{os.path.basename(good_svg_fn)}}') + os.rename(render_fn, good_svg_fn) + + except: logger.warning('Exception when trying to plot') traceback.print_exc()