# enable the overrides for the context of the conf only
OVERRIDES .= ":qemuboot-xilinx"

# Default machine targets for Xilinx QEMU (FDT Generic)
# Allow QB_MACHINE to be overridden by a BSP config
QB_MACHINE ?= "${QB_MACHINE_XILINX}"
QB_RNG=""
QB_MACHINE_XILINX:aarch64 = "-machine arm-generic-fdt"
QB_MACHINE_XILINX:arm = "-M arm-generic-fdt-7series"
QB_MACHINE_XILINX:microblaze = "-M microblaze-fdt-plnx"

QB_SYSTEM_NAME ?= "${@qemu_target_binary(d)}"
QB_DEFAULT_FSTYPE ?= "${@qemu_rootfs_params(d,'fstype')}"
QB_ROOTFS ?= "${@qemu_rootfs_params(d,'rootfs')}"
QB_ROOTFS_OPT ?= "${@qemu_rootfs_params(d,'rootfs-opt')}"
QB_DTB ?= "${@qemu_default_dtb(d)}"

# defaults
QB_DEFAULT_KERNEL ?= "none"
QB_DEFAULT_KERNEL:zynq ?= "${@'zImage' if \
		d.getVar('INITRAMFS_IMAGE_BUNDLE') != '1' else 'zImage-initramfs-${MACHINE}.bin'}"
QB_DEFAULT_KERNEL:microblaze ?= "${@'simpleImage.mb' if \
		d.getVar('INITRAMFS_IMAGE_BUNDLE') != '1' else 'simpleImage.mb-initramfs-${MACHINE}.bin'}"

inherit qemuboot

def qemu_target_binary(data):
    package_arch = data.getVar("PACKAGE_ARCH")
    qemu_target_binary = (data.getVar("QEMU_TARGET_BINARY_%s" % package_arch) or "")
    if qemu_target_binary:
        return qemu_target_binary

    target_arch = data.getVar("TARGET_ARCH")
    if target_arch == "microblazeeb":
        target_arch = "microblaze"
    elif target_arch == "aarch64":
        target_arch += "-multiarch"
    elif target_arch == "arm":
        target_arch = "aarch64"
    return "qemu-system-%s" % target_arch

def qemu_add_extra_args(data):
    initramfs_image = data.getVar('INITRAMFS_IMAGE') or ""
    bundle_image = data.getVar('INITRAMFS_IMAGE_BUNDLE') or ""
    deploy_dir = data.getVar('DEPLOY_DIR_IMAGE') or ""
    machine_name = data.getVar('MACHINE') or ""
    soc_family = data.getVar('SOC_FAMILY') or ""
    qb_extra_args = ''
    # Add kernel image and boot.scr to qemu boot command when initramfs_image supplied
    kernel_name = ''
    bootscr_image = '%s/boot.scr' % deploy_dir
    if soc_family in ('zynqmp', 'versal'):
        kernel_name = 'Image'
        bootscr_loadaddr = '0x20000000'
    if initramfs_image:
        kernel_image = '%s/%s' % (deploy_dir, kernel_name)
        if bundle_image == "1":
            kernel_image = '%s/%s-initramfs-%s.bin' % (deploy_dir, kernel_name, machine_name)
        kernel_loadaddr = '0x200000'
        if kernel_name:
            qb_extra_args = ' -device loader,file=%s,addr=%s,force-raw=on' % (kernel_image, kernel_loadaddr)
            qb_extra_args += ' -device loader,file=%s,addr=%s,force-raw=on' % (bootscr_image, bootscr_loadaddr)
        if soc_family == 'versal':
            qb_extra_args += ' -boot mode=5'
    else:
        if soc_family in ('zynqmp', 'versal'):
            qb_extra_args = ' -boot mode=5'
    return qb_extra_args

def qemu_rootfs_params(data, param):
    initramfs_image = data.getVar('INITRAMFS_IMAGE') or ""
    bundle_image = data.getVar('INITRAMFS_IMAGE_BUNDLE') or ""
    soc_family = data.getVar('SOC_FAMILY') or ""
    tune_features = (data.getVar('TUNE_FEATURES') or []).split()
    if 'microblaze' in tune_features:
        soc_family = 'microblaze'
    soc_variant = data.getVar('SOC_VARIANT') or ""

    if param == 'rootfs':
        return 'none' if bundle_image == "1" else ''

    elif param == 'fstype':
        fstype_dict = {
            "microblaze": "cpio.gz",
            "zynq": "cpio.gz",
            "zynqmp": "cpio.gz.u-boot",
            "versal": "cpio.gz.u-boot.qemu-sd-fatimg"
        }
        if not initramfs_image:
            image_fs = data.getVar('IMAGE_FSTYPES')
            if 'wic.qemu-sd' in image_fs:
                return 'wic.qemu-sd'
        if soc_family not in fstype_dict:
            return ""
        return fstype_dict[soc_family]

    elif param == 'rootfs-opt':
        sd_index = "1"
        if soc_family == 'zynq':
            sd_index = "0"
        if soc_family == 'versal' and soc_variant == 'net':
            sd_index = "0"

        # Device is using a disk
        if not initramfs_image:
            return ' -drive if=sd,index=%s,file=@ROOTFS@,format=raw' % (sd_index)

        # Device is using a ramdisk
        if soc_family not in ('zynq', 'microblaze'):
            return ' -device loader,file=@ROOTFS@,addr=0x04000000,force-raw=on'

        # Ramdisk must be compiled into the kernel
        return ''

def qemu_default_dtb(data):
    if data.getVar("IMAGE_BOOT_FILES", True):
        dtbs = data.getVar("IMAGE_BOOT_FILES", True).split(" ")
        # IMAGE_BOOT_FILES has extra renaming info in the format '<source>;<target>'
        # Note: Wildcard sources work here only because runqemu expands them at run time
        dtbs = [f.split(";")[0] for f in dtbs]
        dtbs = [f for f in dtbs if f.endswith(".dtb")]
        if len(dtbs) != 0:
            return dtbs[0]
    return ""

def qemu_default_serial(data):
    if data.getVar("SERIAL_CONSOLES", True):
        first_console = data.getVar("SERIAL_CONSOLES", True).split(" ")[0]
        speed, console = first_console.split(";", 1)
        # zynqmp uses earlycon and stdout (in dtb)
        if "zynqmp" in data.getVar("MACHINEOVERRIDES", True).split(":"):
            return ""
        return "console=%s,%s earlyprintk" % (console, speed)
    return ""

def qemu_zynqmp_unhalt(data, multiarch):
    if multiarch:
        return "-global xlnx,zynqmp-boot.cpu-num=0 -global xlnx,zynqmp-boot.use-pmufw=true"
    return "-device loader,addr=0xfd1a0104,data=0x8000000e,data-len=4 -device loader,addr=0xfd1a0104,data=0x8000000e,data-len=4"