@ECHO off

if [%1] == [test] shift & goto :test
if [%1] == [build] shift & goto :build
if [%1] == [clean] shift & goto :clean

:parseArgs
if [%1] == [WORKERS] set NATTEN_N_WORKERS=%2 & shift & shift & goto :parseargs
if [%1] == [CUDA_ARCH] set NATTEN_CUDA_ARCH=%2 & shift & shift & goto :parseargs
if [%1] == [FORCE_CUDA] set NATTEN_WITH_CUDA=%2 & shift & shift & goto :parseargs
if [%1] == [VERBOSE] set NATTEN_VERBOSE=%2 & shift & shift & goto :parseargs
goto :buildContinue
:end

:test
echo "Testing NATTEN"
pip install -r requirements-dev.txt
pytest -v -x ./tests
goto :eof
:end

:build
goto :buildStart
:end

:buildStart
goto :parseargs
:end

:buildFinalize
set NATTEN_N_WORKERS=
set NATTEN_CUDA_ARCH=
set NATTEN_VERBOSE=
set NATTEN_IS_BUILDING_DIST=
set NATTEN_WITH_CUDA=
goto :eof
:end

:buildContinue
echo NATTEN_N_WORKERS: %NATTEN_N_WORKERS%
echo NATTEN_CUDA_ARCH: %NATTEN_CUDA_ARCH%
echo NATTEN_WITH_CUDA: %NATTEN_WITH_CUDA%
echo NATTEN_VERBOSE: %NATTEN_VERBOSE%
set NATTEN_IS_BUILDING_DIST=1
rem set CUDA_ARCH=8.6
pip install -r requirements.txt
python setup.py bdist_wheel
goto :buildFinalize
:end

:clean
echo Cleaning up
echo "Removing %CD%\build"
del %CD%\build
goto :eof
:end