From: Dmitry Selyutin <ghostmansd@gmail.com>
Date: Sun, 4 Jun 2023 08:49:54 +0000 (+0300)
Subject: insndb/db: introduce instruction argument type
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=4ca21d74ee2f3962c2f3e50c32f50ca1fc864a23;p=openpower-isa.git

insndb/db: introduce instruction argument type
---

diff --git a/src/openpower/insndb/db.py b/src/openpower/insndb/db.py
index 4f922c65..b8311f31 100644
--- a/src/openpower/insndb/db.py
+++ b/src/openpower/insndb/db.py
@@ -11,6 +11,21 @@ from openpower.insndb.core import (
 )
 
 
+class Instruction(str):
+    def __new__(cls, string):
+        svp64 = False
+        if string.startswith("sv."):
+            string = string[len("sv."):]
+            svp64 = True
+        self = super().__new__(cls, string)
+        self.__svp64 = svp64
+        return self
+
+    @property
+    def svp64(self):
+        return self.__svp64
+
+
 class BaseVisitor(Visitor):
     def __init__(self, **_):
         pass
@@ -89,7 +104,8 @@ def main():
     for (command, (visitor, help)) in commands.items():
         parser = main_subparser.add_parser(command, help=help)
         if issubclass(visitor, ConcreteInstructionVisitor):
-            parser.add_argument("insn", metavar="INSN", help="instruction")
+            parser.add_argument("insn", type=Instruction,
+                metavar="INSN", help="instruction")
 
     args = vars(main_parser.parse_args())
     command = args.pop("command")
@@ -100,3 +116,7 @@ def main():
 
     db = Database(find_wiki_dir())
     db.visit(visitor=visitor)
+
+
+if __name__ == "__main__":
+    main()