# Copyright (c) 2014-2024, Chris Pressey, Cat's Eye Technologies.
# This file is distributed under a 2-clause BSD license.  See LICENSES/ dir.
# SPDX-License-Identifier: LicenseRef-BSD-2-Clause-X-SixtyPical

"""Emittables for 6502 machine code."""

from sixtypical.emitter import Emittable, Byte, Label, Offset, LowAddressByte, HighAddressByte


class AddressingMode(Emittable):
    def size(self):
        """Size of the operand for the mode (not including the opcode)"""
        raise NotImplementedError

    def serialize(self, addr):
        raise NotImplementedError

    def __repr__(self):
        return "%s(%r)" % (self.__class__.__name__, self.value)


class Implied(AddressingMode):
    def size(self):
        return 0

    def serialize(self, addr):
        return []

    def __repr__(self):
        return "%s()" % (self.__class__.__name__)


class Immediate(AddressingMode):
    def __init__(self, value):
        assert isinstance(value, (Byte, LowAddressByte, HighAddressByte))
        self.value = value

    def size(self):
        return 1

    def serialize(self, addr):
        return self.value.serialize(addr)


class Absolute(AddressingMode):
    def __init__(self, value):
        assert isinstance(value, (Label, Offset))
        self.value = value

    def size(self):
        return 2

    def serialize(self, addr):
        return self.value.serialize(addr)


class AbsoluteX(Absolute):
    pass


class AbsoluteY(Absolute):
    pass


class ZeroPage(AddressingMode):
    def __init__(self, value):
        assert isinstance(value, (Label, Offset))
        self.value = value

    def size(self):
        return 1

    def serialize(self, addr):
        return self.value.serialize_as_zero_page(addr)


class Indirect(AddressingMode):
    def __init__(self, value):
        assert isinstance(value, Label)
        self.value = value

    def size(self):
        return 2

    def serialize(self, addr):
        return self.value.serialize(addr)


class IndirectY(ZeroPage):
    pass


class Relative(AddressingMode):
    def __init__(self, value):
        assert isinstance(value, Label)
        self.value = value

    def size(self):
        return 1

    def serialize(self, addr):
        return self.value.serialize_relative_to(addr)


# - - - -


class Instruction(Emittable):
    def __init__(self, operand=None):
        self.operand = operand or Implied()

    def size(self):
        return 1 + self.operand.size() if self.operand else 0

    def serialize(self, addr):
        return [self.opcodes[self.operand.__class__]] + self.operand.serialize(addr)

    def __repr__(self):
        return "%s(%r)" % (self.__class__.__name__, self.operand)


class ADC(Instruction):
    opcodes = {
        Immediate: 0x69,
        Absolute:  0x6d,
        AbsoluteX: 0x7d,
        AbsoluteY: 0x79,
    }


class AND(Instruction):
    opcodes = {
        Immediate: 0x29,
        Absolute:  0x2d,
        AbsoluteX: 0x3d,
        AbsoluteY: 0x39,
        ZeroPage:  0x25,
    }


class BCC(Instruction):
    opcodes = {
        Relative:  0x90,
    }


class BCS(Instruction):
    opcodes = {
        Relative:  0xb0,
    }


class BEQ(Instruction):
    opcodes = {
        Relative:  0xf0,
    }


class BNE(Instruction):
    opcodes = {
        Relative:  0xd0,
    }


class BPL(Instruction):
    opcodes = {
        Relative:  0x10,
    }


class BMI(Instruction):
    opcodes = {
        Relative:  0x30,
    }


class CLC(Instruction):
    opcodes = {
        Implied:   0x18
    }


class CLI(Instruction):
    opcodes = {
        Implied:   0x58,
    }


class CMP(Instruction):
    opcodes = {
        Immediate: 0xc9,
        Absolute:  0xcd,
        AbsoluteX: 0xdd,
        AbsoluteY: 0xd9,
    }


class CPX(Instruction):
    opcodes = {
        Immediate: 0xe0,
        Absolute:  0xec,
    }


class CPY(Instruction):
    opcodes = {
        Immediate: 0xc0,
        Absolute:  0xcc,
    }


class DEC(Instruction):
    opcodes = {
        Absolute:  0xce,
        AbsoluteX: 0xde,
    }


class DEX(Instruction):
    opcodes = {
        Implied:   0xca,
    }


class DEY(Instruction):
    opcodes = {
        Implied:   0x88,
    }


class EOR(Instruction):
    opcodes = {
        Immediate: 0x49,
        Absolute:  0x4d,
        AbsoluteX: 0x5d,
        AbsoluteY: 0x59,
        ZeroPage:  0x45,
    }


class INC(Instruction):
    opcodes = {
        Absolute:  0xee,
        AbsoluteX: 0xfe,
    }


class INX(Instruction):
    opcodes = {
        Implied:   0xe8,
    }


class INY(Instruction):
    opcodes = {
        Implied:   0xc8,
    }


class JMP(Instruction):
    opcodes = {
        Absolute:  0x4c,
        Indirect:  0x6c,
    }


class JSR(Instruction):
    opcodes = {
        Absolute:  0x20,
    }


class LDA(Instruction):
    opcodes = {
        Immediate: 0xa9,
        Absolute:  0xad,
        AbsoluteX: 0xbd,
        AbsoluteY: 0xb9,
        IndirectY: 0xb1,
        ZeroPage:  0xa5,
    }


class LDX(Instruction):
    opcodes = {
        Immediate: 0xa2,
        Absolute:  0xae,
        AbsoluteY: 0xbe,
    }


class LDY(Instruction):
    opcodes = {
        Immediate: 0xa0,
        Absolute:  0xac,
        AbsoluteX: 0xbc,
    }


class ORA(Instruction):
    opcodes = {
        Immediate: 0x09,
        Absolute:  0x0d,
        AbsoluteX: 0x1d,
        AbsoluteY: 0x19,
        ZeroPage:  0x05,
    }


class PHA(Instruction):
    opcodes = {
        Implied:   0x48,
    }


class PLA(Instruction):
    opcodes = {
        Implied:   0x68,
    }


class ROL(Instruction):
    opcodes = {
        Implied:   0x2a,    # Accumulator
        Absolute:  0x2e,
        AbsoluteX: 0x3e,
    }


class ROR(Instruction):
    opcodes = {
        Implied:   0x6a,    # Accumulator
        Absolute:  0x6e,
        AbsoluteX: 0x7e,
    }


class RTS(Instruction):
    opcodes = {
        Implied:   0x60,
    }


class NOP(Instruction):
    opcodes = {
        Implied:   0xEA,
    }


class SBC(Instruction):
    opcodes = {
        Immediate: 0xe9,
        Absolute:  0xed,
        AbsoluteX: 0xfd,
        AbsoluteY: 0xf9,
    }


class SEC(Instruction):
    opcodes = {
        Implied:   0x38,
    }


class SEI(Instruction):
    opcodes = {
        Implied:   0x78,
    }


class STA(Instruction):
    opcodes = {
        Absolute:  0x8d,
        AbsoluteX: 0x9d,
        AbsoluteY: 0x99,
        IndirectY: 0x91,
        ZeroPage:  0x85,
    }


class STX(Instruction):
    opcodes = {
        Absolute:  0x8e,
    }


class STY(Instruction):
    opcodes = {
        Absolute:  0x8c,
    }


class TAX(Instruction):
    opcodes = {
        Implied:   0xaa,
    }


class TAY(Instruction):
    opcodes = {
        Implied:   0xa8,
    }


class TXA(Instruction):
    opcodes = {
        Implied:   0x8a,
    }


class TYA(Instruction):
    opcodes = {
        Implied:   0x98,
    }