#
# This file is part of fastms.
#
# 2014 Evgeny Strekalovskiy <evgeny dot strekalovskiy at in dot tum dot de> (Technical University of Munich)
#
# fastms is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# fastms is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with fastms. If not, see <http://www.gnu.org/licenses/>.
#

all: targets

USE_CUDA:=0
USE_OPENMP:=0
USE_OPENCV:=0
USE_MEX:=1

TMP_DIR:=tmp



LIBS:=
DEFINES:=
INCLUDES:=
TARGETS:=

SOLVER_SOURCE_DIR:=src/libfastms
INCLUDES += -I$(SOLVER_SOURCE_DIR)


# check if mac or linux
MAC:=1


# c++

GXX:=x86_64-w64-mingw32-gcc
ARGS_GXX:=
ARGS_GXX += -Wall
ARGS_GXX += -O3
ARGS_GXX += -m64
#ARGS_GXX += -fPIC #causes warnings for some reason

ifeq ($(USE_OPENMP), 1)
	ARGS_GXX += -fopenmp
endif

COMMAND_COMPILE_GXX=$(GXX) -c -o $@ $< $(ARGS_GXX) $(INCLUDES) $(DEFINES)
COMMAND_GET_DEPENDENCIES_GXX=@$(GXX) -M $< $(ARGS_GXX) $(INCLUDES) $(DEFINES) > $@.dep


# openmp
ifeq ($(USE_OPENMP), 1)
    LIBS += -lgomp
else
    DEFINES += -DDISABLE_OPENMP
endif

DEFINES += -DDISABLE_OPENCV
DEFINES += -DDISABLE_CUDA



# target: solver
SOLVER_SOURCES:=
SOLVER_SOURCES += $(shell find $(SOLVER_SOURCE_DIR) -name '*.cpp')
ifeq ($(USE_CUDA), 1)
    SOLVER_SOURCES += $(shell find $(SOLVER_SOURCE_DIR) -name '*.cu')
endif
SOLVER_OBJECTS:=$(foreach file, $(SOLVER_SOURCES), $(TMP_DIR)/$(file).o)
SOLVER_DEPENDENCIES:=$(foreach file, $(SOLVER_OBJECTS), $(file).dep)
-include $(SOLVER_DEPENDENCIES)
SOLVER_TARGET:=$(TMP_DIR)/$(SOLVER_SOURCE_DIR)/libfastms.o
TARGETS += $(SOLVER_TARGET)
COMMAND_LINK_SOLVER=ld -r -o $@ $^



# target: main
MAIN_SOURCES:=
MAIN_SOURCES += $(shell find src/examples -name '*.cpp')
MAIN_OBJECTS:=$(foreach file, $(MAIN_SOURCES), $(TMP_DIR)/$(file).o)
MAIN_OBJECTS += $(SOLVER_TARGET)
MAIN_DEPENDENCIES:=$(foreach file, $(MAIN_OBJECTS), $(file).dep)
-include $(MAIN_DEPENDENCIES)
MAIN_TARGET:=main
TARGETS += $(MAIN_TARGET)
COMMAND_LINK_MAIN=$(GXX) -o $@ $^ $(LIBS)



# mex
ifeq ($(USE_MEX), 1)

	MATLAB_DIR:=/c/Program\ Files/MATLAB/r2017a

    ifeq ($(wildcard $(MATLAB_DIR)/bin/win64),)
        $(info WARNING: Did not find MATLAB in the specified directory $(MATLAB_DIR), disabling mex target compilation.)
        USE_MEX:=
    endif

endif
ifeq ($(USE_MEX), 1)

    MEX_SUFFIX:=mexw64

    MEX_SOURCES:=$(shell find src/mex -name '*.cpp')
    MEX_OBJECTS:=$(foreach file, $(MEX_SOURCES), $(TMP_DIR)/$(file).o)
    MEX_OBJECTS += $(SOLVER_TARGET)
    MEX_DEPENDENCIES:=$(foreach file, $(MEX_OBJECTS), $(file).dep)
    -include $(MEX_DEPENDENCIES)

    MEX_DEFINES := $(DEFINES)

    MEX_INCLUDES :=$(INCLUDES) -I$(MATLAB_DIR)/extern/include
    MATLAB_LIB_DIR1:=$(MATLAB_DIR)/extern/lib/win64/mingw64
    MATLAB_LIB_DIR2:=$(MATLAB_DIR)/bin/win64

    MEX_LIBS:=$(LIBS) -L$(MATLAB_LIB_DIR1) -L$(MATLAB_LIB_DIR2) -lmex -lmx


    MEX_TARGET:=examples/matlab/fastms_mex.$(MEX_SUFFIX)
	TARGETS += $(MEX_TARGET)

    COMMAND_COMPILE_GXX_MEX=$(GXX) -c -o $@ $< $(ARGS_GXX) $(MEX_INCLUDES) $(MEX_DEFINES)
    COMMAND_GET_DEPENDENCIES_GXX_MEX=@$(GXX) -M $< $(ARGS_GXX) $(MEX_INCLUDES) $(MEX_DEFINES) > $@.dep
    COMMAND_LINK_MEX=$(GXX) -o $@ $^ -shared $(MEX_LIBS)

else
	DEFINES += -DDISABLE_MEX
endif



# common commands
COMMAND_POSTPROCESS_DEPENDENCIES=@echo $@:`sed 's/.*://' $@.dep | tr "\n" " " | sed 's/\\\\/ /g'` > $@.dep; sed -e 's/^.*://' -e 's/  */::/g' $@.dep | tr ":" "\n" | sed -e 's/$$/:/' -e 's/^:$$//' >> $@.dep; echo >> $@.dep
COMMAND_MAKE_TARGET_DIR=@mkdir -p $(shell dirname $@)
COMMAND_CLEAN=@rm -rf $(TMP_DIR) $(TARGETS)



targets: $(TARGETS)

# solver

$(TMP_DIR)/$(SOLVER_SOURCE_DIR)/%.cpp.o: $(SOLVER_SOURCE_DIR)/%.cpp Makefile

	$(COMMAND_MAKE_TARGET_DIR)
	$(COMMAND_COMPILE_GXX)
	$(COMMAND_GET_DEPENDENCIES_GXX)
	$(COMMAND_POSTPROCESS_DEPENDENCIES)

$(TMP_DIR)/$(SOLVER_SOURCE_DIR)/%.cu.o: $(SOLVER_SOURCE_DIR)/%.cu Makefile
	$(COMMAND_MAKE_TARGET_DIR)
	$(COMMAND_NVCC_COMPILE)
	$(COMMAND_GET_DEPENDENCIES_NVCC)
	$(COMMAND_POSTPROCESS_DEPENDENCIES)

$(SOLVER_TARGET): $(SOLVER_OBJECTS)
	$(COMMAND_MAKE_TARGET_DIR)
	$(COMMAND_LINK_SOLVER)

# main

$(TMP_DIR)/src/examples/%.cpp.o: src/examples/%.cpp Makefile
	$(COMMAND_MAKE_TARGET_DIR)
	$(COMMAND_COMPILE_GXX)
	$(COMMAND_GET_DEPENDENCIES_GXX)
	$(COMMAND_POSTPROCESS_DEPENDENCIES)

$(MAIN_TARGET): $(MAIN_OBJECTS)
	$(COMMAND_MAKE_TARGET_DIR)
	$(COMMAND_LINK_MAIN)



# mex
$(TMP_DIR)/src/mex/%.cpp.o: src/mex/%.cpp Makefile
	$(COMMAND_MAKE_TARGET_DIR)
	$(COMMAND_COMPILE_GXX_MEX)
	$(COMMAND_GET_DEPENDENCIES_GXX_MEX)
	$(COMMAND_POSTPROCESS_DEPENDENCIES)

$(MEX_TARGET): $(MEX_OBJECTS)
	$(COMMAND_MAKE_TARGET_DIR)
	$(COMMAND_LINK_MEX)



clean:
	$(COMMAND_CLEAN)