From 723b4fec35e8942df06933831024ec32559d55a5 Mon Sep 17 00:00:00 2001
From: Peter Hutterer <peter.hutterer@who-t.net>
Date: Fri, 23 Aug 2019 09:58:54 +1000
Subject: [PATCH] tools: add a format toggle for switching between json and svg

Signed-off-by: Peter Hutterer <peter.hutterer@who-t.net>
---
 tools/raw-log-converter.py | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/tools/raw-log-converter.py b/tools/raw-log-converter.py
index 47a59c2..1c7cd50 100755
--- a/tools/raw-log-converter.py
+++ b/tools/raw-log-converter.py
@@ -40,7 +40,7 @@ logging.basicConfig(format='%(asctime)s %(levelname)s: %(name)s: %(message)s',
 logger = logging.getLogger('tuhi')  # set the pseudo-root logger to take advantage of the other loggers
 
 
-def parse_file(filename, tablet_model, orientation):
+def parse_file(filename, file_format, tablet_model, orientation):
     width = tablet_model.width
     height = tablet_model.height
     pressure = tablet_model.pressure
@@ -81,8 +81,10 @@ def parse_file(filename, tablet_model, orientation):
                 stroke.new_abs((p.x * ps, p.y * ps), normalize(p.p))
             stroke.seal()
         d.seal()
-        with open(jsonname, 'w') as fd:
-            fd.write(d.to_json())
+        if file_format == 'json':
+            with open(jsonname, 'w') as fd:
+                fd.write(d.to_json())
+            return
 
         from io import StringIO
         js = json.load(StringIO(d.to_json()))
@@ -106,7 +108,7 @@ def main(args=sys.argv):
     Input data is a raw log file. These are usually stored in
     \t$XDG_DATA_HOME/tuhi/<bluetooth address>/raw/
 
-    Pass the log file to this tool and it will convert it to a JSON file and
+    Pass the log file to this tool and it will convert it to a JSON file or
     an SVG file. Alternatively, use --all to convert all
     all log files containing pen data in the above directory.
 
@@ -137,6 +139,10 @@ def main(args=sys.argv):
                         help='Use defaults from the given tablet model',
                         default='intuos-pro',
                         choices=['intuos-pro', 'slate', 'spark'])
+    parser.add_argument('--format',
+                        help='The format to generate. Default: svg',
+                        default='svg',
+                        choices=['svg', 'json'])
 
     ns = parser.parse_args(args[1:])
     if ns.verbose:
@@ -156,7 +162,7 @@ def main(args=sys.argv):
         'spark': WacomProtocolSpark,
     }
     for f in files:
-        parse_file(f, model_map[ns.tablet_model], ns.orientation)
+        parse_file(f, ns.format, model_map[ns.tablet_model], ns.orientation)
 
 
 if __name__ == '__main__':