BUILD_DIR=./build
Python3_EXECUTABLE=
BINDINGS_PYTHON_ENABLED=ON
CCACHE=OFF

ifeq ($(CCACHE),ON)
        CMAKE_CCACHE_OPTIONS=-DCMAKE_CXX_COMPILER_LAUNCHER=ccache
else
        CMAKE_CCACHE_OPTIONS=
endif

$(BUILD_DIR)/configured.stamp:
	cmake -B $(BUILD_DIR) -GNinja ../llvm-project/llvm/ \
	$(CMAKE_CCACHE_OPTIONS) \
	-DLLVM_ENABLE_PROJECTS=mlir \
	-DLLVM_BUILD_EXAMPLES=OFF \
	-DLLVM_TARGETS_TO_BUILD="host" \
	-DCMAKE_BUILD_TYPE=Release \
	-DLLVM_ENABLE_ASSERTIONS=ON \
	-DMLIR_ENABLE_BINDINGS_PYTHON=$(BINDINGS_PYTHON_ENABLED) \
	-DZAMALANG_BINDINGS_PYTHON_ENABLED=$(BINDINGS_PYTHON_ENABLED) \
	-DCONCRETE_FFI_RELEASE=${CONCRETE_PROJECT}/target/release \
	-DLLVM_EXTERNAL_PROJECTS=zamalang \
	-DLLVM_EXTERNAL_ZAMALANG_SOURCE_DIR=. \
	-DPython3_EXECUTABLE=${Python3_EXECUTABLE}
	touch $@

build-initialized: $(BUILD_DIR)/configured.stamp

zamacompiler: build-initialized
	cmake --build $(BUILD_DIR) --target zamacompiler

python-bindings: build-initialized
	cmake --build $(BUILD_DIR) --target ZamalangMLIRPythonModules
	cmake --build $(BUILD_DIR) --target ZamalangPythonModules

test-check: zamacompiler file-check not
	$(BUILD_DIR)/bin/llvm-lit -v tests/

test-python: python-bindings zamacompiler
	PYTHONPATH=${PYTHONPATH}:$(BUILD_DIR)/tools/zamalang/python_packages/zamalang_core LD_PRELOAD=$(BUILD_DIR)/lib/libZamalangRuntime.so pytest -vs tests/python

test: test-check test-end-to-end-jit test-python

# Unittests

build-end-to-end-jit-test: build-initialized
	cmake --build $(BUILD_DIR) --target end_to_end_jit_test

build-end-to-end-jit-clear-tensor: build-initialized
	cmake --build $(BUILD_DIR) --target end_to_end_jit_clear_tensor

build-end-to-end-jit-encrypted-tensor: build-initialized
	cmake --build $(BUILD_DIR) --target end_to_end_jit_encrypted_tensor

build-end-to-end-jit-hlfhelinalg: build-initialized
	cmake --build $(BUILD_DIR) --target end_to_end_jit_hlfhelinalg

build-end-to-end-jit: build-end-to-end-jit-test build-end-to-end-jit-clear-tensor build-end-to-end-jit-encrypted-tensor build-end-to-end-jit-hlfhelinalg

test-end-to-end-jit-test: build-end-to-end-jit-test
	$(BUILD_DIR)/bin/end_to_end_jit_test

test-end-to-end-jit-clear-tensor: build-end-to-end-jit-clear-tensor
	$(BUILD_DIR)/bin/end_to_end_jit_clear_tensor

test-end-to-end-jit-encrypted-tensor: build-end-to-end-jit-encrypted-tensor
	$(BUILD_DIR)/bin/end_to_end_jit_encrypted_tensor

test-end-to-end-jit-hlfhelinalg: build-end-to-end-jit-hlfhelinalg
	$(BUILD_DIR)/bin/end_to_end_jit_hlfhelinalg

test-end-to-end-jit: test-end-to-end-jit-test test-end-to-end-jit-clear-tensor test-end-to-end-jit-encrypted-tensor test-end-to-end-jit-hlfhelinalg

# LLVM/MLIR dependencies

all-deps: file-check not

file-check:
	cmake --build $(BUILD_DIR) --target FileCheck
not:
	cmake --build $(BUILD_DIR) --target not

# Python packages

define build_image_and_copy_wheels
	docker image build -t concretefhe-compiler-manylinux:$(1) --build-arg python_tag=$(1) -f ../builders/Dockerfile.release_manylinux_2_24_x86_64 ..
	docker container run --rm -v ${PWD}/../wheels:/wheels_volume concretefhe-compiler-manylinux:$(1) cp -r /wheels/. /wheels_volume/.
endef

package_py38:
	$(call build_image_and_copy_wheels,cp38-cp38)

package_py39:
	$(call build_image_and_copy_wheels,cp39-cp39)

package_py310:
	$(call build_image_and_copy_wheels,cp310-cp310)

release_tarballs:
	docker image build -t concretefhe-compiler-manylinux:linux_x86_64_tarball -f ../builders/Dockerfile.release_tarball_linux_x86_64 ..
	docker container run --rm -v ${PWD}/../tarballs:/tarballs_volume concretefhe-compiler-manylinux:linux_x86_64_tarball cp -r /tarballs/. /tarballs_volume/.

.PHONY: build-initialized \
	build-end-to-end-jit \
	zamacompiler \
	python-bindings \
	test-check \
	test-end-to-end-jit \
	test-end-to-end-jit-test \
	test-end-to-end-jit-clear-tensor \
	test-end-to-end-jit-encrypted-tensor \
	test-end-to-end-jit-hlfhelinalg \
	test-python \
	test \
	add-deps \
	file-check \
	not \
	package_py38 \
	package_py39 \
	package_py310 \
	release_tarballs