From 4ca21d74ee2f3962c2f3e50c32f50ca1fc864a23 Mon Sep 17 00:00:00 2001
From: Dmitry Selyutin <ghostmansd@gmail.com>
Date: Sun, 4 Jun 2023 11:49:54 +0300
Subject: [PATCH] insndb/db: introduce instruction argument type

---
 src/openpower/insndb/db.py | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

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()
-- 
2.30.2