Initial import0.1

1 files changed, 112 insertions, 0 deletions

diff --git a/z80count.py b/z80count.py
new file mode 100755
index 0000000..f361134
--- /dev/null
+++ b/z80count.py
@@ -0,0 +1,112 @@
+#!/usr/bin/env python3
+#
+# Copyright (C) 2019 by Juan J. Martinez <jjm@usebox.net>
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+#
+
+__version__ = "0.1"
+
+import json
+import sys
+import re
+import argparse
+
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        description='Z80 Cycle Count', epilog="Copyright (C) 2019 Juan J Martinez <jjm@usebox.net>")
+
+    parser.add_argument(
+        "--version", action="version", version="%(prog)s " + __version__)
+    parser.add_argument('-d', dest='debug', action='store_true',
+                        help="Enable debug (show the matched case)")
+    parser.add_argument('-s', dest='subt', action='store_true',
+                        help="Include subtotal")
+    parser.add_argument('-u', dest='update', action='store_true',
+                        help="Update existing count if available")
+    parser.add_argument('-t', dest='tabstop', type=int,
+                        help="Number of tabs for new comments", default=2)
+    parser.add_argument(
+        "infile", nargs="?", type=argparse.FileType('r'), default=sys.stdin,
+            help="Input file")
+    parser.add_argument(
+        "outfile", nargs="?", type=argparse.FileType('w'), default=sys.stdout,
+            help="Output file")
+    args = parser.parse_args()
+
+    in_f = args.infile
+    out_f = args.outfile
+
+    with open("z80table.json", "rt") as fd:
+        table = json.load(fd)
+
+    for i in range(len(table)):
+        table[i]["cregex"] = re.compile(table[i]["regex"] + r"\s?(;.*)?")
+
+    our_comment = re.compile(r"(\[[0-9.\s/]+\])")
+
+    total = [0, 0]
+    while True:
+        line = in_f.readline()
+        if not line:
+            break
+
+        found = False
+        for entry in sorted(table, key=lambda o: o["w"]):
+            if entry["cregex"].search(line):
+                cycles = entry["cycles"]
+                if "/" in cycles:
+                    c = cycles.split("/")
+                    total[0] += int(c[0])
+                    total[1] += int(c[1])
+                else:
+                    total[0] += int(cycles)
+                    total[1] += int(cycles)
+
+                line = line.rstrip().rsplit(";", 1)
+                comment = "; [%s" % cycles
+                if args.subt:
+                    comment += " .. %d/%d]" % (total[0], total[1])
+                else:
+                    comment += "]"
+                if args.debug:
+                    comment += " case{%s}" % entry["case"]
+
+                if len(line) == 1:
+                    comment = "\t" * args.tabstop + comment
+                out_f.write(line[0] + comment)
+                if len(line) > 1:
+                    if args.update:
+                        m = our_comment.search(line[1])
+                        if m:
+                            line[1] = line[1].replace(m.group(0), "")
+                    out_f.write(" ")
+                    out_f.write(line[1].lstrip())
+                out_f.write("\n")
+                found = True
+                break
+
+        if not found:
+            out_f.write(line.rstrip() + "\n")
+
+
+if __name__ == "__main__":
+    main()