From a24c54a570f109311a39747322f1a7a7ff0a9dd6 Mon Sep 17 00:00:00 2001
From: Peter Hutterer <peter.hutterer@who-t.net>
Date: Fri, 16 Aug 2019 11:25:57 +1000
Subject: [PATCH] tools: add an svg generation tool

This takes the raw log YAML file tuhi saves for each interaction, extracts the
pen data and then creates an SVG from that. Good for testing whether the
stroke parsing is sane.

Where --all is given, it will search through the directory that tuhi uses to
store the raw logs.

Two fixmes: the pointsize is hardcoded to 5 (intuos pro) because there's no
easy way here to get the point size. Same for the pressure range.

Signed-off-by: Peter Hutterer <peter.hutterer@who-t.net>
---
 tools/test-svg.py | 134 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 134 insertions(+)
 create mode 100755 tools/test-svg.py

diff --git a/tools/test-svg.py b/tools/test-svg.py
new file mode 100755
index 0000000..c429924
--- /dev/null
+++ b/tools/test-svg.py
@@ -0,0 +1,134 @@
+#!/bin/env python3
+# -*- coding: utf-8 -*-
+#
+# Copyright (c) 2019 Red Hat, Inc.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+
+import argparse
+import os
+import sys
+from pathlib import Path
+import yaml
+import json
+import logging
+
+# This tool isn't installed, so we can assume that the tuhi module is always
+# in the parent directory
+sys.path.insert(0, os.path.dirname(os.path.realpath(__file__)) + '/..')  # noqa
+from tuhi.util import flatten
+from tuhi.drawing import Drawing
+from tuhi.protocol import StrokeFile
+from tuhi.svg import JsonSvg
+from tuhi.wacom import WacomProtocolSpark, WacomProtocolIntuosPro, WacomProtocolSlate
+
+logging.basicConfig(format='%(asctime)s %(levelname)s: %(name)s: %(message)s',
+                    level=logging.INFO,
+                    datefmt='%H:%M:%S')
+logger = logging.getLogger('tuhi')  # set the pseudo-root logger to take advantage of the other loggers
+
+
+def parse_file(filename, tablet_model, orientation):
+    width = tablet_model.width
+    height = tablet_model.height
+    pressure = tablet_model.pressure
+    orientation = orientation or tablet_model.orientation
+
+    stem = Path(filename).stem
+    with open(filename) as fd:
+        yml = yaml.load(fd, Loader=yaml.Loader)
+        # all recv lists that have source PEN
+        pendata = [d['recv'] for d in yml['data'] if 'recv' in d and 'source' in d and d['source'] == 'PEN']
+        data = list(flatten(pendata))
+        if not data:
+            print(f'{filename}: no pen data.')
+            return
+
+        f = StrokeFile(data)
+        # gotta convert to Drawings, then to json string, then to json, then
+        # to svg. ffs.
+        svgname = f'{stem}.svg'
+        jsonname = f'{stem}.json'
+        ps = 5  # FIXME: fetch the point size from the settings.ini
+        d = Drawing(svgname, (width * ps, height * ps), f.timestamp)
+
+        def normalize(p):
+            NORMALIZED_RANGE = 0x10000
+            return NORMALIZED_RANGE * p / pressure
+
+        for s in f.strokes:
+            stroke = d.new_stroke()
+            for p in s.points:
+                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())
+
+        from io import StringIO
+        js = json.load(StringIO(d.to_json()))
+        JsonSvg(js, orientation, d.name)
+
+
+def fetch_files():
+    import xdg.BaseDirectory
+    basedir = Path(xdg.BaseDirectory.xdg_data_home, 'tuhi')
+
+    return [f for f in basedir.rglob('raw/*.yaml')]
+
+
+def main(args=sys.argv):
+    parser = argparse.ArgumentParser(description='YAML log to SVG converter')
+    parser.add_argument('filename', help='The YAML file to load', nargs='?')
+    parser.add_argument('--verbose',
+                        help='Show some debugging informations',
+                        action='store_true',
+                        default=False)
+    parser.add_argument('--all',
+                        help='Convert all files in $XDG_DATA_DIR/tuhi/',
+                        action='store_true',
+                        default=False)
+    parser.add_argument('--orientation',
+                        help='The orientation of the tablet. Default: the tablet model\'s default',
+                        default=None,
+                        choices=['landscape', 'portrait', 'reverse-landscape', 'reverse-portrait'])
+    parser.add_argument('--tablet-model',
+                        help='Use defaults from the given tablet model',
+                        default='intuos-pro',
+                        choices=['intuos-pro', 'slate', 'spark'])
+
+    ns = parser.parse_args(args[1:])
+    if ns.verbose:
+        logger.setLevel(logging.DEBUG)
+
+    if not ns.all:
+        if ns.filename is None:
+            print('filename is required, or use --all', file=sys.stderr)
+            sys.exit(1)
+        files = [ns.filename]
+    else:
+        files = fetch_files()
+
+    model_map = {
+            'intuos-pro': WacomProtocolIntuosPro,
+            'slate': WacomProtocolSlate,
+            'spark': WacomProtocolSpark,
+    }
+    for f in files:
+        parse_file(f, model_map[ns.tablet_model], ns.orientation)
+
+
+if __name__ == '__main__':
+    main()