#!/usr/bin/python

# --------------------------------------------------------------
# Copyleft (k) 2008, by Jose M. Rodriguez-Rosa
# (a.k.a. Boriel, http://www.boriel.com)
#
# This module contains cls.f16 (fixed) bit boolean, arithmetic and
# comparison intermediate-code translations
# --------------------------------------------------------------
from src.arch.interface.quad import Quad

from ._32bit import Bits32
from ._float import Float
from .common import _f_ops, is_float, runtime_call
from .runtime import Labels as RuntimeLabel


# -----------------------------------------------------
# Fixed Point (16.16) bits operands
# -----------------------------------------------------
class Fixed16:
    @classmethod
    def f16(cls, op: str | int | float) -> tuple[int, int]:
        """Returns a floating point operand converted to 32 bits unsigned int.
        Negative numbers are returned in 2 complement.
        The result is returned in a tuple (DE, HL) => High16 (Int part), Low16 (Decimal part)
        """
        op = float(op)

        negative = op < 0
        if negative:
            op = -op

        DE = int(op)
        HL = int((op - DE) * 2**16) & 0xFFFF
        DE &= 0xFFFF

        if negative:  # Do C2
            DE ^= 0xFFFF
            HL ^= 0xFFFF

            DEHL = ((DE << 16) | HL) + 1
            HL = DEHL & 0xFFFF
            DE = (DEHL >> 16) & 0xFFFF

        return DE, HL

    @classmethod
    def get_oper(cls, op1: str, op2: str | None = None, *, use_bc: bool = False, reversed: bool = False) -> list[str]:
        """Returns pop sequence for 32 bits operands
        1st operand in HLDE, 2nd operand remains in the stack

        Now it does support operands inversion calling __SWAP32.

        However, if 1st operand is integer (immediate) or indirect, the stack
        will be rearranged, so it contains a 32 bit pushed parameter value for the
        subroutine to be called.

        If use_bc is True, then BC will be used instead of HL for lower part
        for the 1st operand.
        """
        output: list[str] = []

        op1 = str(op1)

        if op2 is not None:
            op2 = str(op2)

        op = op2 if op2 is not None else op1

        float1 = False  # whether op1 (2nd operand) is float

        indirect = op[0] == "*"
        if indirect:
            op = op[1:]

        immediate = op[0] == "#"
        if immediate:
            op = op[1:]

        hl = "hl" if not use_bc else "bc"

        if is_float(op):
            float1 = True
            op = float(op)

            if indirect:
                op = int(op) & 0xFFFF
                if immediate:
                    output.append(f"ld hl, {op}")
                else:
                    output.append(f"ld hl, ({op})")

                output.append(runtime_call(RuntimeLabel.ILOAD32))
                if use_bc:
                    output.append("ld b, h")
                    output.append("ld c, l")
            else:
                DE, HL = cls.f16(op)
                output.append("ld de, %i" % DE)
                output.append("ld %s, %i" % (hl, HL))
        else:
            if op[0] == "_":
                if immediate:
                    output.append("ld %s, %s" % (hl, op))
                else:
                    output.append("ld %s, (%s)" % (hl, op))
            else:
                output.append("pop %s" % hl)

            if indirect:
                output.append(runtime_call(RuntimeLabel.ILOAD32))
                if use_bc:
                    output.append("ld b, h")
                    output.append("ld c, l")
            else:
                if op[0] == "_":
                    output.append("ld de, (%s + 2)" % op)
                else:
                    output.append("pop de")

        if op2 is not None:
            op = op1

            indirect = op[0] == "*"
            if indirect:
                op = op[1:]

            immediate = op[0] == "#"
            if immediate:
                op = op[1:]

            if is_float(op):
                op = float(op)

                if indirect:
                    op = int(op)
                    output.append("exx")
                    if immediate:
                        output.append("ld hl, %i" % (op & 0xFFFF))
                    else:
                        output.append("ld hl, (%i)" % (op & 0xFFFF))

                    output.append(runtime_call(RuntimeLabel.ILOAD32))  # TODO: Check if this is ever used
                    output.append("push de")
                    output.append("push hl")
                    output.append("exx")
                else:
                    DE, HL = cls.f16(op)
                    output.append("ld bc, %i" % DE)
                    output.append("push bc")
                    output.append("ld bc, %i" % HL)
                    output.append("push bc")
            else:
                if indirect:
                    output.append("exx")  # uses alternate set to put it on the stack
                    if op[0] == "_":
                        if immediate:
                            output.append("ld hl, %s" % op)
                        else:
                            output.append("ld hl, (%s)" % op)
                    else:
                        output.append("pop hl")  # Pointers are only 16 bits ***

                    output.append(runtime_call(RuntimeLabel.ILOAD32))  # TODO: Check if this is ever used
                    output.append("push de")
                    output.append("push hl")
                    output.append("exx")
                elif op[0] == "_":  # an address
                    if float1 or op1[0] == "_":  # If previous op was constant, we can use hl in advance
                        tmp = output
                        output = []
                        output.append("ld hl, (%s + 2)" % op)
                        output.append("push hl")
                        output.append("ld hl, (%s)" % op)
                        output.append("push hl")
                        output.extend(tmp)
                    else:
                        output.append("ld bc, (%s + 2)" % op)
                        output.append("push bc")
                        output.append("ld bc, (%s)" % op)
                        output.append("push bc")
                else:
                    pass  # 2nd operand remains in the stack

        if op2 is not None and reversed:
            output.append(runtime_call(RuntimeLabel.SWAP32))

        return output

    @classmethod
    def f16_to_32bit(cls, ins: Quad) -> Quad:
        """If any of the operands within the ins(truction) are numeric,
        convert them to its 32bit representation, otherwise leave them
        as they are.
        """
        quads = list(ins)
        for i, instruction in enumerate(quads[1:], 1):
            if is_float(instruction):
                de, hl = cls.f16(instruction)
                quads[i] = str((de << 16) | hl)

        ins = Quad(*quads)
        return ins

    @classmethod
    def f16_binary(cls, ins: Quad, label: str, *, reversible: bool = False) -> list[str]:
        op1, op2 = tuple(ins[2:])
        rev = reversible and not is_float(op1) and op1[0] != "t" and op2[0] == "t"
        output = cls.get_oper(op1, op2, reversed=rev)
        output.append(runtime_call(label))
        output.append("push de")
        output.append("push hl")
        return output

    @classmethod
    def addf16(cls, ins: Quad) -> list[str]:
        """Pops last 2 bytes from the stack and adds them.
        Then push the result onto the stack.

        Optimizations:
          * If any of the operands is ZERO,
            then do NOTHING: A + 0 = 0 + A = A
        """
        return Bits32.add32(cls.f16_to_32bit(ins))

    @classmethod
    def subf16(cls, ins: Quad) -> list[str]:
        """Pops last 2 dwords from the stack and subtract them.
        Then push the result onto the stack.
        NOTE: The operation is TOP[0] = TOP[-1] - TOP[0]

        If TOP[0] is 0, nothing is done
        """
        return Bits32.sub32(cls.f16_to_32bit(ins))

    @classmethod
    def mulf16(cls, ins: Quad) -> list[str]:
        """Multiplies 2 32bit (16.16) fixed point numbers. The result is pushed onto the stack."""
        op1, op2 = tuple(ins[2:])

        if _f_ops(op1, op2) is not None:  # TODO: move this to the optimizer
            op1, op2 = _f_ops(op1, op2)

            if op2 == 1:  # A * 1 => A
                output = cls.get_oper(op1)
                output.append("push de")
                output.append("push hl")
                return output

            if op2 == -1:
                return Bits32.neg32(ins)

            output = cls.get_oper(op1)
            if op2 == 0:
                output.append("ld hl, 0")
                output.append("ld e, h")
                output.append("ld d, l")
                output.append("push de")
                output.append("push hl")
                return output

        return cls.f16_binary(ins, RuntimeLabel.MULF16)

    @classmethod
    def divf16(cls, ins: Quad) -> list[str]:
        """Divides 2 32bit (16.16) fixed point numbers. The result is pushed onto the stack.

        Optimizations:

         * If 2nd operand is 1, do nothing
         * If 2nd operand is -1, do NEG32
        """
        op1, op2 = tuple(ins[2:])

        if is_float(op2):
            if float(op2) == 1:  # TODO move this to the optimizer
                output = cls.get_oper(op1)
                output.append("push de")
                output.append("push hl")
                return output

            if float(op2) == -1:
                return Float.negf(ins)

        return cls.f16_binary(ins, RuntimeLabel.DIVF16, reversible=True)

    @classmethod
    def modf16(cls, ins: Quad) -> list[str]:
        """Reminder of div. 2 32bit (16.16) fixed point numbers. The result is pushed onto the stack.
        Optimizations:
         * If 2nd op is 1. Returns 0
        """
        op1, op2 = tuple(ins[2:])

        if is_float(op2) and float(op2) == 1:  # TODO move this to the optimizer
            output = cls.get_oper(op1)
            output.append("ld hl, 0")
            output.append("push hl")
            output.append("push hl")
            return output

        return cls.f16_binary(ins, RuntimeLabel.MODF16, reversible=True)

    @classmethod
    def negf16(cls, ins: Quad) -> list[str]:
        """Negates (arithmetic) top of the stack (Fixed point in DE.HL)

        Fixed point signed version
        """
        return Bits32.neg32(cls.f16_to_32bit(ins))

    @classmethod
    def ltf16(cls, ins: Quad) -> list[str]:
        """Compares & pops top 2 operands out of the stack, and checks
        if the 1st operand < 2nd operand (top of the stack).
        Pushes 0 if False, 1 if True.

        Fixed point signed version
        """
        return Bits32.lti32(cls.f16_to_32bit(ins))

    @classmethod
    def gtf16(cls, ins: Quad) -> list[str]:
        """Compares & pops top 2 operands out of the stack, and checks
        if the 1st operand > 2nd operand (top of the stack).
        Pushes 0 if False, 1 if True.

        Fixed point signed version
        """
        return Bits32.gti32(cls.f16_to_32bit(ins))

    @classmethod
    def lef16(cls, ins: Quad) -> list[str]:
        """Compares & pops top 2 operands out of the stack, and checks
        if the 1st operand <= 2nd operand (top of the stack).
        Pushes 0 if False, 1 if True.

        Fixed point signed version
        """
        return Bits32.lei32(cls.f16_to_32bit(ins))

    @classmethod
    def gef16(cls, ins: Quad) -> list[str]:
        """Compares & pops top 2 operands out of the stack, and checks
        if the 1st operand >= 2nd operand (top of the stack).
        Pushes 0 if False, 1 if True.

        Fixed point signed version
        """
        return Bits32.gei32(cls.f16_to_32bit(ins))

    @classmethod
    def eqf16(cls, ins: Quad) -> list[str]:
        """Compares & pops top 2 operands out of the stack, and checks
        if the 1st operand == 2nd operand (top of the stack).
        Pushes 0 if False, 1 if True.

        Fixed point signed version
        """
        return Bits32.eq32(cls.f16_to_32bit(ins))

    @classmethod
    def nef16(cls, ins: Quad) -> list[str]:
        """Compares & pops top 2 operands out of the stack, and checks
        if the 1st operand != 2nd operand (top of the stack).
        Pushes 0 if False, 1 if True.

        Fixed point signed version
        """
        return Bits32.ne32(cls.f16_to_32bit(ins))

    @classmethod
    def orf16(cls, ins: Quad) -> list[str]:
        """Compares & pops top 2 operands out of the stack, and checks
        if the 1st operand OR (Logical) 2nd operand (top of the stack).
        Pushes 0 if False, 1 if True.

        Fixed point signed version
        """
        return Bits32.or32(cls.f16_to_32bit(ins))

    @classmethod
    def xorf16(cls, ins: Quad) -> list[str]:
        """Compares & pops top 2 operands out of the stack, and checks
        if the 1st operand XOR (Logical) 2nd operand (top of the stack).
        Pushes 0 if False, 1 if True.

        Fixed point signed version
        """
        return Bits32.xor32(cls.f16_to_32bit(ins))

    @classmethod
    def andf16(cls, ins: Quad) -> list[str]:
        """Compares & pops top 2 operands out of the stack, and checks
        if the 1st operand AND (Logical) 2nd operand (top of the stack).
        Pushes 0 if False, 1 if True.

        Fixed point signed version
        """
        return Bits32.and32(cls.f16_to_32bit(ins))

    @classmethod
    def notf16(cls, ins: Quad) -> list[str]:
        """Negates top of the stack (Fixed point in DE.HL)

        Fixed point signed version
        """
        return Bits32.not32(cls.f16_to_32bit(ins))

    @classmethod
    def absf16(cls, ins: Quad) -> list[str]:
        """Absolute value of top of the stack (Fixed point in DE.HL)

        Fixed point signed version
        """
        return Bits32.abs32(cls.f16_to_32bit(ins))

    @classmethod
    def loadf16(cls, ins: Quad) -> list[str]:
        """Load a 32 bit (16.16) fixed point value from a memory address
        If 2nd arg. start with '*', it is always treated as
        an indirect value.
        """
        output = cls.get_oper(ins[2])
        output.append("push de")
        output.append("push hl")
        return output

    @classmethod
    def storef16(cls, ins: Quad) -> list[str]:
        """Stores 2º operand content into address of 1st operand.
        store16 a, x =>  *(&a) = x
        """
        value = ins[2]
        if is_float(value):
            val = float(ins[2])  # Immediate?
            (de, hl) = cls.f16(val)
            q = list(ins)
            q[2] = (de << 16) | hl
            ins = Quad(*q)

        return Bits32.store32(ins)

    @classmethod
    def jzerof16(cls, ins: Quad) -> list[str]:
        """Jumps if top of the stack (32bit) is 0 to arg(1)
        (For Fixed point 16.16 bit values)
        """
        value = ins[1]
        if is_float(value):
            if float(value) == 0:
                return ["jp %s" % str(ins[2])]  # Always true
            else:
                return []

        output = cls.get_oper(value)
        output.append("ld a, h")
        output.append("or l")
        output.append("or e")
        output.append("or d")
        output.append("jp z, %s" % str(ins[2]))
        return output

    @classmethod
    def jnzerof16(cls, ins: Quad) -> list[str]:
        """Jumps if top of the stack (32bit) is !=0 to arg(1)
        Fixed Point (16.16 bit) values.
        """
        value = ins[1]
        if is_float(value):
            if float(value) != 0:
                return ["jp %s" % str(ins[2])]  # Always true
            else:
                return []

        output = cls.get_oper(value)
        output.append("ld a, h")
        output.append("or l")
        output.append("or e")
        output.append("or d")
        output.append("jp nz, %s" % str(ins[2]))
        return output

    @classmethod
    def jgezerof16(cls, ins: Quad) -> list[str]:
        """Jumps if top of the stack (32bit, fixed point) is >= 0 to arg(1)"""
        value = ins[1]
        if is_float(value):
            if float(value) >= 0:
                return ["jp %s" % str(ins[2])]  # Always true

        output = cls.get_oper(value)
        output.append("ld a, d")
        output.append("add a, a")  # Puts sign into carry
        output.append("jp nc, %s" % str(ins[2]))
        return output

    @classmethod
    def retf16(cls, ins: Quad) -> list[str]:
        """Returns from a procedure / function a Fixed Point (32bits) value"""
        output = cls.get_oper(ins[1])
        output.append("#pragma opt require hl,de")
        output.append("jp %s" % str(ins[2]))
        return output

    @classmethod
    def paramf16(cls, ins: Quad) -> list[str]:
        """Pushes 32bit fixed point param into the stack"""
        output = cls.get_oper(ins[1])
        output.append("push de")
        output.append("push hl")
        return output

    @classmethod
    def fparamf16(cls, ins: Quad) -> list[str]:
        """Passes a 16.16 fixed point as a __FASTCALL__ parameter.
        This is done by popping out of the stack for a
        value, or by loading it from memory (indirect)
        or directly (immediate)
        """
        return cls.get_oper(ins[1])
