#mem_alloc_method=0 : MEM=
#mem_alloc_method=1 : BigPhysArea
#mem_alloc_method=2 : Dynamic

# Test passing $(1) to $(CC). If $(CC) succeeds, then echo $(1).
# Copy from https://github.com/NVIDIA/open-gpu-kernel-modules/blob/main/utils.mk
TEST_CC_ARG = \
 $(shell $(CC) -c -x c /dev/null -Werror $(1) -o /dev/null > /dev/null 2>&1 && \
   echo $(1))

ifeq ($(ARCH),ARM64)
	ARCH:=arm64
endif

SHELL := /bin/bash

DRIVERNAME := flex-x400
UDEVRULENAME := delta-x400

BINARY-OBJS := X4XXDrvBoard.o X4XXDrvBoardHelper.o X400DrvBoard.o X410DrvBoard.o Scp.o ScpDv.o DvDrvBoard.o SdiDrvBoard.o XDrvBoard.o XDvDrv_Properties.o HpdMngr.o LtcMngr.o LowLatencyMngr.o UltimateLowLatencyMngr.o MailboxMngr.o ClockReconstructionMngr.o DelayedStreamStarterMngr.o ConfigurationSpaceMngr.o BufMngr.o DMABufMngr.o DMAMngr.o PropertiesMngr.o ShadowRegisters.o SkelDrv.o CommonBufMngr.o KernelObjectCommon.o SkelProperties.o Stream.o StreamProperties.o PCIeBridge_ADVLP.o FPGA_DMAController_ADVLP.o USBufMngr.o DrvDbg.o EventMngr.o XDv_Helper.o XSdi_Helper.o BramStatusMngr.o HDMIInfoFrames.o

SRC-OBJS := CommonBufMngr_Linux.o DriverAccess.o DriverEntry.o DrvParam.o EventMngr_Linux.o KernelObjects.o KernelSysCall.o LinuxAllocator.o LinuxDrvHlp.o USBufMngr_Linux.o KThreadMngr.o InterlockedOp.o DMAMngr_Linux.o KernelTime.o NvidiaP2PWrapper.o

# default to currently running kernel if KVER not specified
ifeq ($(KVER),)
	KVER := $(shell uname -r)
endif

KERNEL_MAJOR := $(shell echo $(KVER) | cut -d. -f1)

KDIR ?= /lib/modules/$(KVER)/build
INSTALLDIR ?= /lib/modules/$(KVER)
PWD := $(shell pwd)

ifeq ($(shell [ $(KERNEL_MAJOR) -le 5 ] && echo true),true)
	USE_SHIPPED := 0
else
	USE_SHIPPED := 1
endif

# Conditionally add -mno-outline-atomics to CFLAGS if it is supported
CONDITIONAL_CFLAGS := $(call TEST_CC_ARG, -mno-outline-atomics)

ifdef mem_alloc_method
EXTRA_CFLAGS += -DMEM_ALLOC_METHOD=$(mem_alloc_method)
endif
EXTRA_CFLAGS += -w -DDBG=0 -DDRIVER -DNO_C_LIB
EXTRA_CFLAGS += -fno-stack-protector
EXTRA_CFLAGS += $(CONDITIONAL_CFLAGS)
EXTRA_CFLAGS += -Wno-error=incompatible-pointer-types

ifneq ($(KERNELRELEASE),)
	obj-m := $(DRIVERNAME).o

ifeq ($(USE_SHIPPED),1)
	$(DRIVERNAME)-objs := $(SRC-OBJS) $(BINARY-OBJS:.o=.o_shipped)
else
	$(DRIVERNAME)-objs := $(SRC-OBJS) $(BINARY-OBJS)
endif
else
ifndef VERBOSE
.SILENT: all
endif

all: gencmd default

default:
	
	for file in *.o_shipped; do if ! file "$$file" | grep -q 'ELF'; then xxd -p -r "$$file" > "$${file/.o_shipped/}.c"; fi; done

	@trap '$(MAKE) shipped_cleanup' EXIT ; $(MAKE) -C $(KDIR) M=$(PWD) modules

gencmd:
ifeq ($(USE_SHIPPED),1)
		for file in $(BINARY-OBJS:.o=.o_shipped); do touch .$${file}.cmd; done
else
		for file in $(BINARY-OBJS); do touch .$${file}.cmd; done
endif

genbin:
	for file in $(BINARY-OBJS); do echo "X" > $${file/.o/}.o_shipped; done

install:
	# Driver installation
	cp $(DRIVERNAME).ko $(INSTALLDIR)/$(DRIVERNAME).ko	
	cp $(UDEVRULENAME).rules /etc/udev/$(UDEVRULENAME).rules
	if [ ! -f /etc/udev/rules.d/010-$(UDEVRULENAME).rules ]; then ln -sf /etc/udev/$(UDEVRULENAME).rules /etc/udev/rules.d/010-$(UDEVRULENAME).rules; fi
	depmod -a
	modprobe -rq $(DRIVERNAME)
	modprobe $(DRIVERNAME)
	@for files in /dev/$(DRIVERNAME)*; do if [ ! -r files ]; then ( dmesg | grep -i $(DRIVERNAME) ) | grep -i "System needs a restart" -o --colour=always; echo "">NULL; fi; done

uninstall:
	modprobe -r $(DRIVERNAME)
	rm $(INSTALLDIR)/$(DRIVERNAME).ko
	rm /etc/udev/$(UDEVRULENAME).rules
	rm /etc/udev/rules.d/010-$(UDEVRULENAME).rules


shipped_cleanup:
	for file in *.o_shipped; do if ! file "$$file" | grep -q 'ELF'; then rm -f "$${file/.o_shipped/}.c"; fi; done

clean:
	rm -f *.o
	rm -f Module.symvers
	rm -f modules.order
	rm -f *.ko
	rm -f .*.cmd

endif
