OPENCM3_DIR 	?= libopencm3
CMSIS_DIR 	?= cmsis
MSPDEBUG_DIR 	?= mspdebug
LIBSODIUM_DIR	?= libsodium
TINYAES_DIR	?= tinyaes
MUSL_DIR	?= musl

C_SOURCES		:= src/main.c src/mspdebug_wrapper.c src/spi_flash.c src/freq_meas.c
C_SOURCES		+= $(MSPDEBUG_DIR)/drivers/jtaglib.c

CMSIS_SOURCES		+= $(CMSIS_DIR)/CMSIS/DSP/Source/TransformFunctions/arm_rfft_fast_init_f32.c
CMSIS_SOURCES		+= $(CMSIS_DIR)/CMSIS/DSP/Source/TransformFunctions/arm_rfft_fast_f32.c
CMSIS_SOURCES		+= $(CMSIS_DIR)/CMSIS/DSP/Source/TransformFunctions/arm_cfft_init_f32.c
CMSIS_SOURCES		+= $(CMSIS_DIR)/CMSIS/DSP/Source/TransformFunctions/arm_cfft_f32.c
CMSIS_SOURCES		+= $(CMSIS_DIR)/CMSIS/DSP/Source/TransformFunctions/arm_cfft_radix8_f32.c
CMSIS_SOURCES		+= $(CMSIS_DIR)/CMSIS/DSP/Source/CommonTables/arm_const_structs.c
CMSIS_SOURCES		+= $(CMSIS_DIR)/CMSIS/DSP/Source/CommonTables/arm_common_tables.c
CMSIS_SOURCES		+= $(CMSIS_DIR)/CMSIS/DSP/Source/TransformFunctions/arm_bitreversal.c
CMSIS_SOURCES		+= $(CMSIS_DIR)/CMSIS/DSP/Source/TransformFunctions/arm_bitreversal2.c

MUSL_SOURCES		+= $(MUSL_DIR)/src/math/tanhf.c $(MUSL_DIR)/src/math/atanhf.c
MUSL_SOURCES		+= $(MUSL_DIR)/src/math/expm1f.c $(MUSL_DIR)/src/math/log1pf.c
MUSL_SOURCES		+= $(MUSL_DIR)/src/math/expf.c $(MUSL_DIR)/src/math/exp2f_data.c
MUSL_SOURCES		+= $(MUSL_DIR)/src/math/powf.c
MUSL_SOURCES		+= $(MUSL_DIR)/src/math/__math_oflowf.c
MUSL_SOURCES		+= $(MUSL_DIR)/src/math/__math_uflowf.c
MUSL_SOURCES		+= $(MUSL_DIR)/src/math/__math_xflowf.c
MUSL_SOURCES		+= $(MUSL_DIR)/src/math/__math_invalidf.c
MUSL_SOURCES		+= $(MUSL_DIR)/src/math/powf_data.c

C_SOURCES += $(CMSIS_SOURCES) $(MUSL_SOURCES)

CXX_SOURCES		+= src/ldpc_wrapper.cpp

BUILDDIR 	?= build
BINARY 		:= safetyreset.elf
LDSCRIPT 	:= stm32f407.ld
OPENCM3_LIB	:= opencm3_stm32f4

PREFIX		?= arm-none-eabi-


GOLD_CODE_NBITS 		?= 5
FMEAS_ADC_SAMPLING_RATE ?= 1000
FMEAS_ADC_MAX			?= 4096
FMEAS_FFT_LEN			?= 256
FMEAS_FFT_WINDOW 		?= gaussian
FMEAS_FFT_WINDOW_SIGMA 	?= 8.0


CC		:= $(PREFIX)gcc
CXX		:= $(PREFIX)g++
LD		:= $(PREFIX)gcc
AR		:= $(PREFIX)ar
AS		:= $(PREFIX)as
OBJCOPY		:= $(PREFIX)objcopy
OBJDUMP		:= $(PREFIX)objdump
GDB		:= $(PREFIX)gdb
PYTHON3 ?= python3

OPENCM3_DIR_ABS 	:= $(abspath $(OPENCM3_DIR))
CMSIS_DIR_ABS 		:= $(abspath $(CMSIS_DIR))
MSPDEBUG_DIR_ABS 	:= $(abspath $(MSPDEBUG_DIR))
LIBSODIUM_DIR_ABS	:= $(abspath $(LIBSODIUM_DIR))
TINYAES_DIR_ABS		:= $(abspath $(TINYAES_DIR))
MUSL_DIR_ABS		:= $(abspath $(MUSL_DIR))

COMMON_CFLAGS		+= -I$(OPENCM3_DIR_ABS)/include -Imspdebug/util -Imspdebug/drivers
COMMON_CFLAGS		+= -I$(CMSIS_DIR_ABS)/CMSIS/DSP/Include -I$(CMSIS_DIR_ABS)/CMSIS/Core/Include 
COMMON_CFLAGS		+= -I$(abspath musl_include_shims) -Ilevmarq

COMMON_CFLAGS	+= -Os -std=gnu11 -g -DSTM32F4
CFLAGS			+= -mthumb -mcpu=cortex-m4 -mfloat-abi=hard -mfpu=fpv4-sp-d16
SIM_CFLAGS		+= -mthumb -mcpu=cortex-m4 -mfloat-abi=soft
# Note: libopencm3 requires some standard libc definitions from stdint.h and stdbool.h, so we don't pass -nostdinc here.
COMMON_CFLAGS	+= -nostdlib -ffreestanding
CFLAGS			+= -fno-common -ffunction-sections -fdata-sections
COMMON_CFLAGS	+= -DGOLD_CODE_NBITS=$(GOLD_CODE_NBITS) -DFMEAS_FFT_LEN=$(FMEAS_FFT_LEN) -DFMEAS_ADC_MAX=$(FMEAS_ADC_MAX)
COMMON_CFLAGS	+= -DFMEAS_ADC_SAMPLING_RATE=$(FMEAS_ADC_SAMPLING_RATE)
COMMON_CFLAGS	+= -DFMEAS_FFT_WINDOW_SIGMA=$(FMEAS_FFT_WINDOW_SIGMA)
# for musl
CFLAGS			+= -Dhidden=

SIM_CFLAGS	+= -Isrc

INT_CFLAGS	+= -Wall -Wextra -Wpedantic -Wshadow -Wimplicit-function-declaration -Wundef
INT_CFLAGS	+= -Wredundant-decls -Wmissing-prototypes -Wstrict-prototypes

CXXFLAGS	+= -Os -g
CXXFLAGS	+= -nostdlib -ffreestanding
CXXFLAGS	+= -mthumb -mcpu=cortex-m4 -mfloat-abi=hard -mfpu=fpv4-sp-d16
CXXFLAGS	+= -fno-common -ffunction-sections -fdata-sections
CXXFLAGS	+= -Wall -Wextra -Wshadow -Wundef -Wredundant-decls
CXXFLAGS	+= -I.

COMMON_LDFLAGS	+= -mthumb -mcpu=cortex-m4 -mfloat-abi=hard -mfpu=fpv4-sp-d16 -g
COMMON_LDFLAGS	+= --static -nostartfiles
LDFLAGS			+= -Wl,--cre
#LDFLAGS		+= -Wl,--gc-sections
LDFLAGS			+= -Wl,--start-group -lc -lgcc -lnosys -Wl,--end-group
LDFLAGS			+= -L$(OPENCM3_DIR_ABS)/lib -l$(OPENCM3_LIB) $(shell $(CC) -print-libgcc-file-name)

all: $(BUILDDIR)/$(BINARY)

tests: tools/freq_meas_test.elf

OBJS := $(addprefix $(BUILDDIR)/,$(C_SOURCES:.c=.o) $(CXX_SOURCES:.cpp=.o))

ALL_OBJS := $(OBJS)
ALL_OBJS += $(OPENCM3_DIR)/lib/lib$(OPENCM3_LIB).a
ALL_OBJS += $(BUILDDIR)/libsodium/src/libsodium/.libs/libsodium.a
ALL_OBJS += $(BUILDDIR)/tinyaes/aes.o
ALL_OBJS += $(BUILDDIR)/levmarq/levmarq.o
ALL_OBJS += $(BUILDDIR)/generated/gold_code_$(GOLD_CODE_NBITS).o
ALL_OBJS += $(BUILDDIR)/generated/fmeas_fft_window.o

$(BUILDDIR)/$(BINARY): $(ALL_OBJS)
	$(LD) -T$(LDSCRIPT) $(COMMON_LDFLAGS) $(LDFLAGS) -o $@ -Wl,-Map=$(BUILDDIR)/src/$*.map $^

tools/freq_meas_test.elf: tools/freq_meas_test.c src/freq_meas.c $(CMSIS_SOURCES) $(BUILDDIR)/musl/lib/libc.a
	$(CC) $(COMMON_CFLAGS) $(SIM_CFLAGS) -lm -lgcc -lnosys -o $@ $^

$(BUILDDIR)/src/%.o: src/%.c
	mkdir -p $(@D)
	$(CC) $(COMMON_CFLAGS) $(CFLAGS) $(INT_CFLAGS) -o $@ -c $<

$(BUILDDIR)/src/%.o: src/%.cpp
	mkdir -p $(@D)
	$(CXX) $(CXXFLAGS) -o $@ -c $<

$(BUILDDIR)/generated/%.o: $(BUILDDIR)/generated/%.c
	mkdir -p $(@D)
	$(CC) $(COMMON_CFLAGS) $(CFLAGS) $(INT_CFLAGS) -o $@ -c $<

$(BUILDDIR)/%.o: %.c
	mkdir -p $(@D)
	$(CC) $(COMMON_CFLAGS) $(CFLAGS) $(EXT_CFLAGS) -o $@ -c $<

$(OPENCM3_DIR)/lib/lib$(OPENCM3_LIB).a:
	$(MAKE) -C $(OPENCM3_DIR) -j $(shell nproc)

$(BUILDDIR)/libsodium/src/libsodium/.libs/libsodium.a:
	mkdir -p $(BUILDDIR)/libsodium
	cd $(BUILDDIR)/libsodium && CFLAGS="$(COMMON_CFLAGS) $(CFLAGS) -DDEV_MODE=1" $(LIBSODIUM_DIR_ABS)/configure --host=arm-none-eabi && $(MAKE) -j $(shell nproc)

$(BUILDDIR)/tinyaes/aes.o:
	mkdir -p $(@D)
	make -C $(@D) -f $(TINYAES_DIR_ABS)/Makefile VPATH=$(TINYAES_DIR_ABS) CFLAGS="$(COMMON_CFLAGS) $(CFLAGS) -c" CC=$(CC) LD=$(LD) AR=$(AR) aes.o

$(BUILDDIR)/generated/gold_code_%.c:
	mkdir -p $(@D)
	$(PYTHON3) tools/gold_code_header_gen.py $* > $@

$(BUILDDIR)/generated/fmeas_fft_window.c:
	mkdir -p $(@D)
	$(PYTHON3) tools/fft_window_header_gen.py -v fmeas_fft_window_table $(FMEAS_FFT_WINDOW) $(FMEAS_FFT_LEN) $(FMEAS_FFT_WINDOW_SIGMA) > $@

$(BUILDDIR)/musl/lib/libc.a:
	mkdir -p $(BUILDDIR)/musl
	cd $(BUILDDIR)/musl && CFLAGS="$(SIM_CFLAGS) $(COMMON_CFLAGS)" CC=$(CC) LD=$(LD) AR=$(AR) $(MUSL_DIR_ABS)/configure && $(MAKE) TARGET=arm-linux-musleabihf GCC_CONFIG="-mcpu=cortex-m4 -mfloat-abi=soft" -j $(shell nproc)

build/ldpc_decoder_test.so: src/ldpc_decoder.c
	gcc -fPIC -shared -Wall -Wextra -Wpedantic -std=gnu11 -O0 -g -o $@ $^

clean:
	-rm -r $(BUILDDIR)/src
	-rm -r $(BUILDDIR)/generated
	-rm $(BUILDDIR)/$(BINARY)

mrproper: clean
	-rm -r build
	-$(MAKE) -C $(OPENCM3_DIR) clean

.PHONY: clean mrproper

-include $(OBJS:.o=.d)