Added main.py and opcodes.example.yaml files

Author: tekord

main.py

@@ -0,0 +1,132 @@
+import argparse
+import os
+import typing
+import yaml
+from enum import Enum
+from string import Template
+from time import gmtime, strftime
+
+parser = argparse.ArgumentParser()
+parser.add_argument("input", type=str, help="OpCode definition file path")
+parser.add_argument("output", type=str, help="Output file name")
+
+args = parser.parse_args()
+
+class Indent:
+ stepSize = 4
+ value = 0
+
+ def __init__(self):
+ pass
+
+ def increase(self):
+ self.value += 1
+
+ def decrease(self):
+ self.value -= 1
+
+ def reset(self):
+ self.value = 0
+
+ def generateSpaces(self):
+ return " " * (self.stepSize * self.value)
+
+class OpCodeCppListGenerator:
+ prefix = ''
+ indent = Indent()
+
+ def formatName(self, name):
+ return str.upper(name).replace('.', '_')
+
+ def generateKey(self, name):
+ return self.prefix + name
+
+ def generateValue(self, code):
+ return code
+
+ def prepareItems(self, items):
+ result = []
+
+ for i in items:
+ if type(i['name']) == type(list()):
+ primaryName = i['name'][0]
+
+ for n in i['name']:
+ stringToAppend = self.indent.generateSpaces() + self.generateKey(self.formatName(n)) \
+ + " = " + self.generateValue(i['code'])
+
+ if (n != primaryName):
+ stringToAppend += " /* alias for " + self.formatName(primaryName) + " */"
+
+ result.append(stringToAppend)
+ else:
+ result.append(self.indent.generateSpaces() + self.generateKey(self.formatName(i['name'])) + " = " + self.generateValue(i['code']))
+
+ return result
+
+ def generateEnumerationLines(self, items):
+ lines = self.prepareItems(items)
+
+ # Do not use 'os.linesep' on Windows. It generates extra line break
+ return str.join(',\n', lines)
+
+currentTime = strftime("%Y-%m-%d %H:%M:%S %z", gmtime())
+
+cppHeaderFileTemplate = """\
+#pragma once
+
+/**
+ * ATTENTION! This file was generated automatically. Do NOT change it directly!
+ * All direct changes may be overwritten.
+ *
+ * Generated at: $generatedAt
+ */
+
+$content
+"""
+
+cppEnumerationTemplate = """\
+enum $name {
+$items
+};\
+"""
+
+outputFileName = args.output
+inputFileName = args.input
+
+if outputFileName == None:
+ outputFileName = '/._opcodes.h'
+
+inputFile = open(inputFileName, 'r')
+
+opCodeList = yaml.load(inputFile)
+opCodeList = opCodeList['opcodes']
+
+opCodesFileGeneratorSettings = {
+ "outputFileName": outputFileName,
+ "fileTemplate": cppHeaderFileTemplate,
+ "name": "_GeneratedOpCodes",
+ "prefix": "OPCODE_",
+}
+
+with open(opCodesFileGeneratorSettings["outputFileName"], 'w') as f:
+ fileTemplate = Template(opCodesFileGeneratorSettings["fileTemplate"])
+
+ cppEnumerationGenerator = OpCodeCppListGenerator()
+ cppEnumerationGenerator.prefix = opCodesFileGeneratorSettings["prefix"]
+
+ cppEnumerationGenerator.indent.increase()
+ renderedItems = cppEnumerationGenerator.generateEnumerationLines(
+ opCodeList)
+ cppEnumerationGenerator.indent.reset()
+
+ enumerationTemplate = Template(cppEnumerationTemplate)
+
+ f.write(
+ fileTemplate.substitute(
+ generatedAt=currentTime,
+ content=enumerationTemplate.substitute(
+ name=opCodesFileGeneratorSettings["name"],
+ items=renderedItems)))
+
+inputFile.close()

opcodes.example.yaml

@@ -0,0 +1,12 @@
+opcodes:
+ - name: nop
+ code: "0x00"
+
+ - name: b
+ code: "0xA0"
+
+ - name: [bz, b_false, b_zero, b_null] # the first opcode is lead, all others are aliases
+ code: "0xA1"
+
+ - name: [bnz, b_true, b_not_zero, b_not_null] # the first opcode is lead, all others are aliases
+ code: "0xA2"