Refactorize MATH labels

Author: boriel

src/arch/zx48k/backend/runtime/labels.py

@@ -7,13 +7,15 @@
 from . import core
 from . import datarestore
 from . import io
+from . import math
 from . import random
 
 
 class Labels(
     core.CoreLabels,
     datarestore.DataRestoreLabels,
     io.IOLabels,
+    math.MathLabels,
     random.RandomLabels
 ):
     """ All labels
@@ -40,6 +42,7 @@ def dict_join(*args: Dict[str, str]) -> Dict[str, str]:
 LABEL_REQUIRED_MODULES = dict_join(
     core.REQUIRED_MODULES,
     datarestore.REQUIRED_MODULES,
+    math.REQUIRED_MODULES,
     io.REQUIRED_MODULES,
     random.REQUIRED_MODULES
 )

src/arch/zx48k/backend/runtime/math.py

@@ -0,0 +1,48 @@
+# Runtime labels
+
+from .namespace import NAMESPACE
+
+
+class MathLabels:
+    ACS = f"{NAMESPACE}ACOS"
+    ASN = f"{NAMESPACE}ASIN"
+    ATN = f"{NAMESPACE}ATAN"
+    COS = f"{NAMESPACE}COS"
+    SIN = f"{NAMESPACE}SIN"
+    TAN = f"{NAMESPACE}TAN"
+
+    EXP = f"{NAMESPACE}EXP"
+    LN = f"{NAMESPACE}LN"
+    SQR = f"{NAMESPACE}SQRT"
+
+    SGNI8 = f"{NAMESPACE}__SGNI8"
+    SGNU8 = f"{NAMESPACE}__SGNU8"
+    SGNI16 = f"{NAMESPACE}__SGNI16"
+    SGNU16 = f"{NAMESPACE}__SGNU16"
+    SGNI32 = f"{NAMESPACE}__SGNI32"
+    SGNU32 = f"{NAMESPACE}__SGNU32"
+    SGNF16 = f"{NAMESPACE}__SGNF16"
+    SGNF = f"{NAMESPACE}__SGNF"
+
+
+REQUIRED_MODULES = {
+    MathLabels.ACS: 'acos.asm',
+    MathLabels.ASN: 'asin.asm',
+    MathLabels.ATN: 'atan.asm',
+    MathLabels.COS: 'cos.asm',
+    MathLabels.SIN: 'sin.asm',
+    MathLabels.TAN: 'tan.asm',
+
+    MathLabels.EXP: 'exp.asm',
+    MathLabels.LN: 'logn.asm',
+    MathLabels.SQR: 'sqrt.asm',
+
+    MathLabels.SGNI8: 'sgni8.asm',
+    MathLabels.SGNU8: 'sgnu8.asm',
+    MathLabels.SGNI16: 'sgni16.asm',
+    MathLabels.SGNU16: 'sgnu16.asm',
+    MathLabels.SGNI32: 'sgni32.asm',
+    MathLabels.SGNU32: 'sgnu32.asm',
+    MathLabels.SGNF16: 'sgnf16.asm',
+    MathLabels.SGNF: 'sgnf.asm'
+}

src/arch/zx48k/translator.py

@@ -1276,54 +1276,55 @@ def visit_PEEK(self, node):
     # region MATH Functions
     def visit_SIN(self, node):
         self.ic_fparam(node.operand.type_, node.operand.t)
-        self.ic_call('SIN', node.size)
-        self.REQUIRES.add('sin.asm')
+        self.runtime_call(RuntimeLabel.SIN, node.size)
 
     def visit_COS(self, node):
         self.ic_fparam(node.operand.type_, node.operand.t)
-        self.ic_call('COS', node.size)
-        self.REQUIRES.add('cos.asm')
+        self.runtime_call(RuntimeLabel.COS, node.size)
 
     def visit_TAN(self, node):
         self.ic_fparam(node.operand.type_, node.operand.t)
-        self.ic_call('TAN', node.size)
-        self.REQUIRES.add('tan.asm')
+        self.runtime_call(RuntimeLabel.TAN, node.size)
 
     def visit_ASN(self, node):
         self.ic_fparam(node.operand.type_, node.operand.t)
-        self.ic_call('ASIN', node.size)
-        self.REQUIRES.add('asin.asm')
+        self.runtime_call(RuntimeLabel.ASN, node.size)
 
     def visit_ACS(self, node):
         self.ic_fparam(node.operand.type_, node.operand.t)
-        self.ic_call('ACOS', node.size)
-        self.REQUIRES.add('acos.asm')
+        self.runtime_call(RuntimeLabel.ACS, node.size)
 
     def visit_ATN(self, node):
         self.ic_fparam(node.operand.type_, node.operand.t)
-        self.ic_call('ATAN', node.size)
-        self.REQUIRES.add('atan.asm')
+        self.runtime_call(RuntimeLabel.ATN, node.size)
 
     def visit_EXP(self, node):
         self.ic_fparam(node.operand.type_, node.operand.t)
-        self.ic_call('EXP', node.size)
-        self.REQUIRES.add('exp.asm')
+        self.runtime_call(RuntimeLabel.EXP, node.size)
 
     def visit_LN(self, node):
         self.ic_fparam(node.operand.type_, node.operand.t)
-        self.ic_call('LN', node.size)
-        self.REQUIRES.add('logn.asm')
+        self.runtime_call(RuntimeLabel.LN, node.size)
 
     def visit_SGN(self, node):
         s = self.TSUFFIX(node.operand.type_)
         self.ic_fparam(node.operand.type_, node.operand.t)
-        self.ic_call('__SGN%s' % s.upper(), node.size)
-        self.REQUIRES.add('sgn%s.asm' % s)
+
+        label = {
+            'i8': RuntimeLabel.SGNI8,
+            'u8': RuntimeLabel.SGNU8,
+            'i16': RuntimeLabel.SGNI16,
+            'u16': RuntimeLabel.SGNU16,
+            'i32': RuntimeLabel.SGNI32,
+            'u32': RuntimeLabel.SGNU32,
+            'f16': RuntimeLabel.SGNF16,
+            'f': RuntimeLabel.SGNF
+        }[s]
+        self.runtime_call(label, node.size)
 
     def visit_SQR(self, node):
         self.ic_fparam(node.operand.type_, node.operand.t)
-        self.ic_call('SQRT', node.size)
-        self.REQUIRES.add('sqrt.asm')
+        self.runtime_call(RuntimeLabel.SQR, node.size)
 
     # endregion