SHELL:=/bin/bash

setup_env:
	poetry install
	poetry run python -m pip install -U pip wheel setuptools
	poetry run python -m pip install -r torch_requirements.txt \
		-f https://download.pytorch.org/whl/torch_stable.html
.PHONY: setup_env

sync_env:
	poetry install --remove-untracked
	make setup_env
.PHONY: sync_env

python_format:
	poetry run env bash ./script/source_format/format_python.sh --dir hdk --dir tests --dir benchmarks
.PHONY: python_format

check_python_format:
	poetry run env bash ./script/source_format/format_python.sh --dir hdk --dir tests --dir benchmarks --check
.PHONY: check_python_format

pylint:
	poetry run pylint --rcfile=pylintrc hdk tests benchmarks
.PHONY: pylint

flake8:
	poetry run flake8 --max-line-length 100 --per-file-ignores="__init__.py:F401" hdk/ tests/ benchmarks/
.PHONY: flake8

python_linting: pylint flake8
.PHONY: python_linting

conformance: python_format
.PHONY: conformance

pcc:
	@$(MAKE) --keep-going --jobs $$(nproc) --output-sync --no-print-directory pcc_internal
.PHONY: pcc

pcc_internal: check_python_format python_linting mypy_ci pydocstyle
.PHONY: pcc_internal

pytest:
	poetry run pytest --cov=hdk -vv --cov-report=xml tests/
.PHONY: pytest

# Not a huge fan of ignoring missing imports, but some packages do not have typing stubs
mypy:
	poetry run mypy -p hdk --ignore-missing-imports
.PHONY: mypy

# Friendly target to run mypy without ignoring missing stubs and still have errors messages
# Allows to see which stubs we are missing
mypy_ns:
	poetry run mypy -p hdk
.PHONY: mypy_ns

mypy_test:
	find ./tests/ -name "*.py" | xargs poetry run mypy --ignore-missing-imports
.PHONY: mypy_test

mypy_benchmark:
	find ./benchmarks/ -name "*.py" | xargs poetry run mypy --ignore-missing-imports
.PHONY: mypy_benchmark

mypy_ci:
	@$(MAKE) --no-print-directory mypy
	@$(MAKE) --no-print-directory mypy_test
	@$(MAKE) --no-print-directory mypy_benchmark
.PHONY: mypy_ci

pytest_and_coverage: pytest coverage
.PHONY: pytest_and_coverage

coverage:
	@if [[ "$$BB" == "" ]]; then BB=origin/main; fi && poetry run diff-cover coverage.xml --fail-under 100 --html-report coverage.html --compare-branch $$BB
.PHONY: coverage

docker_build:
	docker build -t hdk:mlir -f docker/Dockerfile .
.PHONY: docker_build

docker_rebuild:
	docker build --no-cache -t hdk:mlir -f docker/Dockerfile .
.PHONY: docker_rebuild

docker_start:
	@# the slash before pwd is for Windows
	docker run --rm -it --volume /"$$(pwd)":/hdk hdk:mlir
.PHONY: docker_start

docker_build_and_start: docker_build docker_start
.PHONY: docker_build_and_start

docker_bas: docker_build_and_start
.PHONY: docker_bas

docs:
	@# Generate the auto summary of documentations
	poetry run sphinx-apidoc -o docs/_apidoc hdk

	@# Docs
	cd docs && poetry run make html
.PHONY: docs

clean_docs:
	rm -rf docs/_apidoc docs/_build
.PHONY: clean_docs

open_docs:
	@# This is macOS only. On other systems, one would use `start` or `xdg-open`
	open docs/_build/html/index.html
.PHONY: open_docs

build_and_open_docs: clean_docs docs open_docs
.PHONY: build_and_open_docs

pydocstyle:
	@# From http://www.pydocstyle.org/en/stable/error_codes.html
	poetry run pydocstyle hdk --convention google --add-ignore=D1,D202
.PHONY: pydocstyle

strip_nb:
	poetry run python ./script/nbmake_utils/notebook_sanitize.py examples
.PHONY: strip_nb

notebook_timeout:
	poetry run python ./script/nbmake_utils/notebook_test_timeout.py examples
.PHONY: notebook_timeout

benchmark:
	poetry run pytest benchmarks/ --benchmark-save=findings
.PHONY: benchmark