From 862c331897c9ac2451fb1d2437c3768a8c9747c4 Mon Sep 17 00:00:00 2001
From: Dmitry Selyutin <ghostmansd@gmail.com>
Date: Sun, 6 Nov 2022 12:29:55 +0300
Subject: [PATCH] power_insn: add support for a trivial assembly

---
 src/openpower/decoder/power_insn.py | 35 +++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/src/openpower/decoder/power_insn.py b/src/openpower/decoder/power_insn.py
index 4aa2fff9..5a89a7eb 100644
--- a/src/openpower/decoder/power_insn.py
+++ b/src/openpower/decoder/power_insn.py
@@ -532,6 +532,14 @@ class Operand:
     def span(self, record):
         return record.fields[self.name]
 
+    def assemble(self, value, insn, record):
+        span = self.span(record=record)
+        if isinstance(insn, SVP64Instruction):
+            span = tuple(map(lambda bit: (bit + 32), span))
+        if isinstance(value, str):
+            value = int(value, 0)
+        insn[span] = value
+
     def disassemble(self, insn, record,
             verbosity=Verbosity.NORMAL, indent=""):
         raise NotImplementedError
@@ -575,6 +583,10 @@ class SignedOperand(DynamicOperand):
 class StaticOperand(Operand):
     value: int
 
+    def assemble(self, insn, record):
+        return super().assemble(value=self.value,
+            insn=insn, record=record)
+
     def disassemble(self, insn, record,
             verbosity=Verbosity.NORMAL, indent=""):
         span = self.span(record=record)
@@ -1176,6 +1188,10 @@ class Instruction(_Mapping):
         for operand in record.mdwn.operands.static:
             yield (operand.name, operand.value)
 
+    @classmethod
+    def assemble(cls, db, opcode, arguments):
+        raise NotImplementedError(f"{cls.__name__}.assemble")
+
     def disassemble(self, db,
             byteorder="little",
             verbosity=Verbosity.NORMAL):
@@ -1186,6 +1202,25 @@ class WordInstruction(Instruction):
     _: _Field = range(0, 32)
     po: _Field = range(0, 6)
 
+    @classmethod
+    def assemble(cls, db, opcode, arguments):
+        value = 0
+        mask = 0
+        record = db[opcode]
+        for opcode in record.opcodes:
+            value |= int(opcode.value)
+            mask |= int(opcode.mask)
+        insn = cls.integer(value=(value & mask))
+
+        operands = tuple(record.mdwn.operands.dynamic)
+        if len(operands) != len(arguments):
+            raise ValueError("operands count mismatch")
+        for (index, operand) in enumerate(operands):
+            value = arguments[index]
+            operand.assemble(value=value, insn=insn, record=record)
+
+        return insn
+
     @classmethod
     def integer(cls, value, byteorder="little"):
         return super().integer(bits=32, value=value, byteorder=byteorder)
-- 
2.30.2