fix: bug on O4 optimization

Author: boriel

src/arch/zx48k/optimizer/basicblock.py

@@ -381,18 +381,22 @@ def is_used(self, regs, i, top=None):
             top -= 1
 
         if regs and regs[0][0] == '(' and regs[0][-1] == ')':  # A memory address
-            r16 = helpers.single_registers(regs[0][1:-1])\
+            r16 = helpers.single_registers(regs[0][1:-1]) \
                 if helpers.is_16bit_oper_register(regs[0][1:-1]) else []
             ix = helpers.single_registers(helpers.idx_args(regs[0][1:-1])[0]) \
                 if helpers.idx_args(regs[0][1:-1]) else []
 
             rr = set(r16 + ix)
             mem_vars = set([] if rr else RE_ID_OR_NUMBER.findall(regs[0]))
 
-            for mem in self[i:top]:  # For memory accesses only mark as NOT uses if it's overwritten
+            for mem in self[i:top]:  # For memory accesses only mark as NOT used if it's overwritten
                 if mem.inst == 'ld' and mem.opers[0] == regs[0]:
                     return False
 
+                # And, Or, Xor uses both operands
+                if mem.inst in {'and', 'or', 'xor'} and mem.opers[0] == regs[0]:
+                    return True
+
                 if mem.opers and mem.opers[-1] == regs[0]:
                     return True
 

src/arch/zx48k/peephole/opts/109_o4_ld_mem_op_ld_mem.opt

@@ -1,6 +1,5 @@
-;; Removes useless LD's
-;; Tries to guess the value of the 1st and 2nd operands.
-;; If they're the same, this LD is useless
+;; Removes useless operation
+;; Tries to guess if the result of the operation LD, OR, AND, XOR, ADC, ADD, SBC is later used
 
 OLEVEL: 4
 OFLAG: 109

tests/arch/zx48k/optimizer/test_basicblock.py

@@ -169,3 +169,27 @@ def test_mempos_requires(self):
         """
         self.blk.code = [x for x in code.split('\n') if x.strip()]
         self.assertTrue(self.blk.is_used(['(_k)'], 0))
+
+    def test_is_used_or_ix(self):
+        code = """
+        or (ix-1)
+        ld (ix-1), a
+        """
+        self.blk.code = [x for x in code.split('\n') if x.strip()]
+        assert self.blk.is_used(['(ix-1)'], 0)
+
+    def test_is_used_and_ix(self):
+        code = """
+        and (ix-1)
+        ld (ix-1), a
+        """
+        self.blk.code = [x for x in code.split('\n') if x.strip()]
+        assert self.blk.is_used(['(ix-1)'], 0)
+
+    def test_is_used_xor_ix(self):
+        code = """
+        xor (ix-1)
+        ld (ix-1), a
+        """
+        self.blk.code = [x for x in code.split('\n') if x.strip()]
+        assert self.blk.is_used(['(ix-1)'], 0)

tests/functional/opt4_due0.asm

@@ -0,0 +1,54 @@
+	org 32768
+.core.__START_PROGRAM:
+	di
+	push ix
+	push iy
+	exx
+	push hl
+	exx
+	ld hl, 0
+	add hl, sp
+	ld (.core.__CALL_BACK__), hl
+	ei
+	jp .core.__MAIN_PROGRAM__
+.core.__CALL_BACK__:
+	DEFW 0
+.core.ZXBASIC_USER_DATA:
+	; Defines USER DATA Length in bytes
+.core.ZXBASIC_USER_DATA_LEN EQU .core.ZXBASIC_USER_DATA_END - .core.ZXBASIC_USER_DATA
+	.core.__LABEL__.ZXBASIC_USER_DATA_LEN EQU .core.ZXBASIC_USER_DATA_LEN
+	.core.__LABEL__.ZXBASIC_USER_DATA EQU .core.ZXBASIC_USER_DATA
+.core.ZXBASIC_USER_DATA_END:
+.core.__MAIN_PROGRAM__:
+	call _Demo_PutPlayer
+	ld b, h
+	ld c, l
+.core.__END_PROGRAM:
+	di
+	ld hl, (.core.__CALL_BACK__)
+	ld sp, hl
+	exx
+	pop hl
+	pop iy
+	pop ix
+	exx
+	ei
+	ret
+_Demo_PutPlayer:
+	push ix
+	ld ix, 0
+	add ix, sp
+	ld hl, 0
+	push hl
+	ld a, (ix-1)
+	and 199
+	ld (ix-1), a
+	ld a, (ix-2)
+	or (ix-1)
+	ld (ix-1), a
+_Demo_PutPlayer__leave:
+	ld sp, ix
+	pop ix
+	ret
+	;; --- end of user code ---
+	END

tests/functional/opt4_due0.bas

@@ -0,0 +1,9 @@
+sub Demo_PutPlayer()
+    dim  atrSprite, atrGame as ubyte
+
+    atrSprite=atrSprite bAND %11000111
+    atrSprite=atrGame bOR atrSprite
+end sub
+
+Demo_PutPlayer()
+