From d58918a31dd9a5e53a85db46aefa14847ff7430a Mon Sep 17 00:00:00 2001
From: daxid <david.rouquet@tetras-libre.fr>
Date: Wed, 15 Mar 2023 09:11:11 +0000
Subject: [PATCH] add svg file
---
amrbatch/main.py | 17 +++++++++++++----
1 file changed, 13 insertions(+), 4 deletions(-)
diff --git a/amrbatch/main.py b/amrbatch/main.py
index b9dce1d4..075ae680 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()
--
GitLab