Draft riscv-opcodes generator

Author: BrianAnakPintar

opcodes_maker/generate_instr_dict.py

@@ -0,0 +1,148 @@
+from typing import List, Dict
+import os
+import yaml
+import json
+
+
+def range_size(range_str: str) -> int:
+    try:
+        end, start = map(int, range_str.split('-'))
+        return abs(end - start) + 1
+    except ValueError:
+        return 0;
+
+reg_names = {'qs1', 'qs2', 'qd',
+             'fs1', 'fs2', 'fd'}
+
+def GetVariables(vars: List[Dict[str, str]]):
+    var_names = []
+    for var in vars:
+        var_name = var['name']
+        if var_name in reg_names:
+            # Since strings are immutable.
+            lst_var_name = list(var_name)
+            lst_var_name[0] = 'r'
+            var_name = "".join(lst_var_name)
+        elif var_name == "shamt":
+            size = range_size(var['location'])
+            if size == 5:
+                var_name = "shamtw"
+            elif size == 6:
+                var_name = "shamtd"
+        var_names.append(var_name)
+    var_names.reverse()
+
+    return var_names
+
+def BitStringToHex(bit_str: str) -> str:
+    new_bit_str = ""
+    for bit in bit_str:
+        if bit == '-':
+            new_bit_str += '0'
+        else:
+            new_bit_str += bit
+    return hex(int(new_bit_str, 2))
+
+def GetMask(bit_str: str) -> str:
+    mask_str = ""
+    for bit in bit_str:
+        if bit == '-':
+            mask_str += '0'
+        else:
+            mask_str += '1'
+    return hex(int(mask_str, 2))
+
+def GetExtension(ext, base):
+    prefix = f'rv{base}_'
+    final_extensions = []
+
+    if isinstance(ext, str):
+        final_extensions.append(prefix + ext.lower())
+    elif isinstance(ext, dict):
+        for _, extensions in ext.items():
+            for extension in extensions:
+                final_extensions.append(prefix + extension.lower())
+            final_extensions.reverse()
+
+    return final_extensions
+
+def find_first_match(data):
+    if isinstance(data, dict):
+        for key, value in data.items():
+            if key == "match":
+                return value
+            elif isinstance(value, (dict, list)):
+                result = find_first_match(value)
+                if result is not None:
+                    return result
+    elif isinstance(data, list):
+        for item in data:
+            result = find_first_match(item)
+            if result is not None:
+                return result
+    return ""
+
+def GetEncodings(enc: str):
+    n = len(enc)
+    if n < 32:
+        return '-' * (32-n) + enc
+    return enc
+
+def convert(file_dir: str, json_out):
+    with open(file_dir, 'r') as file:
+        data = yaml.safe_load(file)
+        instr_name = data['name'].replace('.', '_')
+
+        print(instr_name)
+        encodings = data['encoding']
+         
+        # USE RV_64
+        rv64_flag = False
+        if 'RV64' in encodings:
+            encodings = encodings['RV64']
+            rv64_flag = True
+        enc_match = GetEncodings(encodings['match'])
+
+        var_names = []
+        if 'variables' in encodings:
+            var_names = GetVariables(encodings['variables'])
+
+        extension = []
+        prefix = ""
+        if rv64_flag:
+            prefix = "64"
+        if "base" in data:
+            extension = GetExtension(data["definedBy"], data["base"])
+        else:
+            extension = GetExtension(data["definedBy"], prefix) 
+
+        match_hex = BitStringToHex(enc_match)
+        match_mask = GetMask(enc_match)
+
+        json_out[instr_name] = {
+            "encoding": enc_match,
+            "variable_fields": var_names,
+            "extension": extension,
+            "match": match_hex,
+            "mask": match_mask
+        }
+
+def read_yaml_insts(path: str):
+    yaml_files = []
+    for root, _, files in os.walk(path):
+        for file in files:
+            if file.endswith('.yaml') or file.endswith('.yml'):
+                yaml_files.append(os.path.join(root, file))
+    return yaml_files
+
+def main():
+    directory = "../arch/inst/"
+    insts = read_yaml_insts(directory)
+    
+    inst_dict = {}
+    with open('data.json', 'w') as outfile:
+        for inst_dir in insts:
+            convert(inst_dir, inst_dict)
+        json.dump(inst_dict, outfile, indent=4)
+main()
+