SUMMARY = "Device Trees for BSPs"
DESCRIPTION = "Device Tree generation and packaging for BSP Device Trees."
SECTION = "bsp"

LICENSE = "MIT"
LIC_FILES_CHKSUM = "file://${COMMON_LICENSE_DIR}/MIT;md5=0835ade698e0bcf8506ecda2f7b4f302"

inherit deploy

INHIBIT_DEFAULT_DEPS = "1"
PACKAGE_ARCH = "${MACHINE_ARCH}"

DEPENDS += "dtc-native"

FILES_${PN} = "/boot/devicetree*"
DEVICETREE_FLAGS ?= "-R 8 -p 0x3000 \
		-i ${WORKDIR}/devicetree \
		${@' '.join(['-i %s' % i for i in d.getVar('KERNEL_DTS_INCLUDE', True).split()])} \
		"
DEVICETREE_PP_FLAGS ?= "-nostdinc -Ulinux \
		-I${WORKDIR}/devicetree \
		${@' '.join(['-I%s' % i for i in d.getVar('KERNEL_DTS_INCLUDE', True).split()])} \
		-x assembler-with-cpp \
		"

FILESEXTRAPATHS_prepend := "${THISDIR}/files:"
MACHINE_DEVICETREE_prepend_zynq = " \
		common/zynq-7000-qspi-dummy.dtsi \
		common/zynq7-base.dtsi \
		"

S = "${WORKDIR}"

KERNEL_DTS_INCLUDE ??= ""
KERNEL_DTS_INCLUDE_zynq = " \
		${STAGING_KERNEL_DIR}/arch/arm/boot/dts \
		${STAGING_KERNEL_DIR}/arch/arm/boot/dts/include \
		"
KERNEL_DTS_INCLUDE_zynqmp = " \
		${STAGING_KERNEL_DIR}/arch/arm64/boot/dts \
		${STAGING_KERNEL_DIR}/arch/arm64/boot/dts/include \
		${STAGING_KERNEL_DIR}/arch/arm64/boot/dts/xilinx \
		"

python () {
    # auto add dependency on kernel tree
    if d.getVar("KERNEL_DTS_INCLUDE", True) != "":
        d.setVarFlag("do_compile", "depends",
            " ".join([d.getVarFlag("do_compile", "depends", True) or "", "virtual/kernel:do_shared_workdir"]))
}

python do_deprecation_check() {
    # check for use of 'zynq7-base.dtsi'
    import re
    for i in d.getVar('MACHINE_DEVICETREE', True).split():
        filepath = os.path.join(d.getVar("WORKDIR", True), i)
        if os.path.exists(filepath):
            with open(filepath, 'rb') as f:
                if re.search("/include/ \"zynq7-base.dtsi\"", f.read()):
                    bb.warn("%s includes 'zynq7-base.dtsi', this include is deprecated." % i)
}
addtask do_deprecation_check after do_unpack

do_compile() {
	if test -n "${MACHINE_DEVICETREE}"; then
		mkdir -p ${WORKDIR}/devicetree
		for i in ${MACHINE_DEVICETREE}; do
			if test -e ${WORKDIR}/$i; then
				cp ${WORKDIR}/$i ${WORKDIR}/devicetree
			fi
		done
	fi

	for DTS_FILE in ${DEVICETREE}; do
		DTS_NAME=`basename -s .dts ${DTS_FILE}`
		${BUILD_CPP} ${DEVICETREE_PP_FLAGS} -o ${DTS_FILE}.pp ${DTS_FILE}
		dtc -I dts -O dtb ${DEVICETREE_FLAGS} -o ${DTS_NAME}.dtb ${DTS_FILE}.pp
	done
}

do_install() {
	for DTS_FILE in ${DEVICETREE}; do
		if [ ! -f ${DTS_FILE} ]; then
			echo "Warning: ${DTS_FILE} is not available!"
			continue
		fi
		DTS_NAME=`basename -s .dts ${DTS_FILE}`
		install -d ${D}/boot/devicetree
		install -m 0644 ${B}/${DTS_NAME}.dtb ${D}/boot/devicetree/${DTS_NAME}.dtb
	done
}

do_deploy() {
	for DTS_FILE in ${DEVICETREE}; do
		if [ ! -f ${DTS_FILE} ]; then
			echo "Warning: ${DTS_FILE} is not available!"
			continue
		fi
		DTS_NAME=`basename -s .dts ${DTS_FILE}`
		install -d ${DEPLOYDIR}
		install -m 0644 ${B}/${DTS_NAME}.dtb ${DEPLOYDIR}/${DTS_NAME}.dtb
	done
}

addtask deploy before do_build after do_install

inherit xilinx-utils

DEVICETREE ?= "${@expand_dir_basepaths_by_extension("MACHINE_DEVICETREE", os.path.join(d.getVar("WORKDIR", True), 'devicetree'), '.dts', d)}"
FILESEXTRAPATHS_append := "${@get_additional_bbpath_filespath('conf/machine/boards', d)}"

# Using the MACHINE_DEVICETREE and MACHINE_KCONFIG vars, append them to SRC_URI
SRC_URI += "${@paths_affix(d.getVar("MACHINE_DEVICETREE", True) or '', prefix = 'file://')}"