CUR_DIR := $(shell dirname $(realpath $(lastword $(MAKEFILE_LIST))))
XGBOOST_TRACKER := $(realpath $(CUR_DIR)/xgboost_src/dmlc-core/tracker/)
BIN := xgboost
DATA := data

.PHONY: all test test-native test-local-cluster clean

all: occlum_workspace

$(BIN):
	cp xgboost_src/$(BIN) .

$(DATA):
	rm -rf $(DATA) && mkdir -p $(DATA)
	cp xgboost_src/demo/binary_classification/agaricus.txt.test $(DATA)
	cp xgboost_src/demo/binary_classification/agaricus.txt.train $(DATA)
	cp xgboost_src/demo/binary_classification/mushroom.conf $(DATA)
	sed -i 's/agaricus/.\/data\/agaricus/g' $(DATA)/mushroom.conf
	sed -i 's/num_round = 2/num_round = 10/g' $(DATA)/mushroom.conf
	sed -i '$$a\model_out = "./host/result.model"' $(DATA)/mushroom.conf
	sed -i '$$a\verbosity = 2' $(DATA)/mushroom.conf
	sed -i '$$a\nthread = 2' $(DATA)/mushroom.conf

occlum_workspace: $(BIN) $(DATA)
	rm -rf occlum_workspace && mkdir occlum_workspace
	cd occlum_workspace && \
	occlum init && \
	jq '.process.default_mmap_size = "96MB"' Occlum.json > temp_Occlum.json && \
	mv temp_Occlum.json Occlum.json && \
	cp ../$(BIN) image/bin && \
	cp ../xgboost_src/lib/libxgboost.so image/lib && \
	mkdir -p image/$(DATA) && \
	cp -r ../$(DATA)/* image/$(DATA) && \
	occlum build

test: occlum_workspace
	@./run_xgboost.sh

test-native: $(BIN) $(DATA)
	rm -rf host
	mkdir host
	./$(BIN) ./$(DATA)/mushroom.conf

test-local-cluster: occlum_workspace
	@$(XGBOOST_TRACKER)/dmlc-submit --cluster=local --num-workers=2 --log-level=DEBUG ./run_xgboost.sh

clean:
	rm -rf $(BIN) $(DATA) tmp_* host occlum_workspace