010. LDA 标志位

在立即寻址 LDA 之后更新 Zero 和 Negative 标志位。

第 10 / 356 · tests/chapter_01_cpu/test_010_LDA_flags.py

需要更新的文件

emulator/cpu/cpu.py

位置

CPU.step, inside the existing $A9 immediate-LDA branch

此步骤存在的原因

6502 会记录 LDA 结果是否为零，以及第 7 位是否被置位。后续的分支指令会用到这些状态位，因此置位和清零都必须正确工作。

完整示例实现

class CPU:
    # Keep the existing methods from Tests 004, 008, and 009.

    def step(self) -> None:
        opcode = self.fetch_byte()

        if opcode == 0xA9:
            self.a = self.fetch_byte()

            if self.a == 0:
                self.p |= 1 << 1
            else:
                self.p &= ~(1 << 1)

            if self.a & (1 << 7):
                self.p |= 1 << 7
            else:
                self.p &= ~(1 << 7)

            return

        raise NotImplementedError(
            f"Opcode {opcode:02X} not implemented"
        )

结果表

A=$00 -> Z=1, N=0
A=$7F -> Z=0, N=0
A=$80 -> Z=0, N=1

重要不变量

只有 Zero 和 Negative 位会被改变；复位时的 Interrupt Disable 位以及其他所有状态位都保持先前的值。

常见误解

这里的“Negative”并不是在执行有符号运算。它只是表示 8 位结果的第 7 位被置位了。

范围之外

测试 012 中引入的具名标志常量和共享辅助函数
绝对寻址和零页寻址的 LDA
指令周期计数

运行本课

uv run pytest tests/chapter_01_cpu/test_010_LDA_flags.py -v

示例实现

完整示例实现

class CPU:
    # Keep the existing methods from Tests 004, 008, and 009.

    def step(self) -> None:
        opcode = self.fetch_byte()

        if opcode == 0xA9:
            self.a = self.fetch_byte()

            if self.a == 0:
                self.p |= 1 << 1
            else:
                self.p &= ~(1 << 1)

            if self.a & (1 << 7):
                self.p |= 1 << 7
            else:
                self.p &= ~(1 << 7)

            return

        raise NotImplementedError(
            f"Opcode {opcode:02X} not implemented"
        )

结果表

A=$00 -> Z=1, N=0
A=$7F -> Z=0, N=0
A=$80 -> Z=0, N=1

查看测试代码test_010_LDA_flags.py

from emulator.cpu.cpu import CPU
from emulator.bus.cpu_bus import CpuBus
from emulator.memory.fake_rom import FakeROM

def test_lda_clears_zero_flag():
    """Now it's time configure LDA flags
    Ensure that zero flag is unset (clear)

    If result == 0 -> Flag Zero is set 

    7  bit  0
    ---- ----
    NV1B DIZC
    |||| ||||
    |||| |||+- Carry
    |||| ||+-- Zero
    |||| |+--- Interrupt Disable
    |||| +---- Decimal
    |||+------ (No CPU effect; see: the B flag)
    ||+------- (No CPU effect; always pushed as 1)
    |+-------- Overflow
    +--------- Negative
    
    Pseudocode:
    if LDA:
      if self.a == 0:
        set flag Z
     else: 
        clear flag Z

    """
    rom = FakeROM()
    
    rom.write(0x7FFC, 0x00)
    rom.write(0x7FFD, 0x80)

    rom.write(0x0000, 0xA9)
    rom.write(0x0001, 0x42)

    bus = CpuBus(program_rom=rom)
    cpu = CPU(bus)

    cpu.reset()
    cpu.step() 

    assert cpu.a == 0x42
    assert (cpu.p & (1 << 1)) == 0 # Flag Zero is clear (0)


def test_lda_sets_zero_flag():
    """
    Ensure that zero flag is set

    if self.a == 0 -> set flag Z else Z is cleared
    Pseudocode:
    if LDA:
     if self.a == 0:
        set flag Z
     else: 
        clear flag Z

    """
    rom = FakeROM()
    
    rom.write(0x7FFC, 0x00)
    rom.write(0x7FFD, 0x80)

    rom.write(0x0000, 0xA9)
    rom.write(0x0001, 0x00)

    bus = CpuBus(program_rom=rom)
    cpu = CPU(bus)

    cpu.reset()
    cpu.step() 

    assert cpu.a == 0x00
    assert (cpu.p & (1 << 1)) != 0 # Flag Zero is set

def test_lda_sets_negative_flag():
    """
    Ensure that Negastive Flag is set on bit 7 active
    for cpu.a
    """
    rom = FakeROM()
    
    rom.write(0x7FFC, 0x00)
    rom.write(0x7FFD, 0x80)
                            #        bit 7
    rom.write(0x0000, 0xA9) #         v
    rom.write(0x0001, 0x80) # 0x80 = b1000_0000

    bus = CpuBus(program_rom=rom)
    cpu = CPU(bus)

    cpu.reset()
    cpu.step()

    assert cpu.a == 0x80
    assert (cpu.p & ( 1 << 7 )) != 0 # N 


def test_lda_clears_negative_flag():
    """
    Ensure that Negastive Flag is unset on bit 7 inactive
    for cpu.a
    """
    rom = FakeROM()
    
    rom.write(0x7FFC, 0x00)
    rom.write(0x7FFD, 0x80)
                            #        bit 7
    rom.write(0x0000, 0xA9) #         v
    rom.write(0x0001, 0x7F) # 0x00 = b0111_1111

    bus = CpuBus(program_rom=rom)
    cpu = CPU(bus)

    cpu.reset()
    cpu.step()

    assert cpu.a == 0x7F
    assert (cpu.p & ( 1 << 7 )) == 0 # N = 0

测试函数 (4)

test_lda_clears_zero_flag
test_lda_sets_zero_flag
test_lda_sets_negative_flag
test_lda_clears_negative_flag