#!/usr/bin/env python
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved

import os
from os import path

from setuptools import find_packages, setup


def get_version():
    init_py_path = path.join(
        path.abspath(path.dirname(__file__)), "comp2comp", "__init__.py"
    )
    init_py = open(init_py_path, "r").readlines()
    version_line = [line.strip() for line in init_py if line.startswith("__version__")][
        0
    ]
    version = version_line.split("=")[-1].strip().strip("'\"")

    # The following is used to build release packages.
    # Users should never use it.
    suffix = os.getenv("ABCTSEG_VERSION_SUFFIX", "")
    version = version + suffix
    if os.getenv("BUILD_NIGHTLY", "0") == "1":
        from datetime import datetime

        date_str = datetime.today().strftime("%y%m%d")
        version = version + ".dev" + date_str

        new_init_py = [line for line in init_py if not line.startswith("__version__")]
        new_init_py.append('__version__ = "{}"\n'.format(version))
        with open(init_py_path, "w") as f:
            f.write("".join(new_init_py))
    return version


setup(
    name="comp2comp",
    version=get_version(),
    author="StanfordMIMI",
    url="https://github.com/StanfordMIMI/Comp2Comp",
    description="Computed tomography to body composition.",
    packages=find_packages(exclude=("configs", "tests")),
    python_requires=">=3.9",
    install_requires=[
        "pydicom",
        "moviepy",
        "numpy==1.23.5",
        "h5py",
        "tabulate",
        "tqdm",
        "silx",
        "yacs",
        "pandas",
        "dosma",
        "opencv-python",
        "huggingface_hub",
        "pycocotools",
        "wget",
        "tensorflow==2.12.0",
        "totalsegmentator @ git+https://github.com/StanfordMIMI/TotalSegmentator.git",
        "totalsegmentatorv2 @ git+https://github.com/StanfordMIMI/TotalSegmentatorV2.git",
    ],
    extras_require={
        "all": ["shapely", "psutil"],
        "dev": [
            # Formatting
            "flake8",
            "isort",
            "black==22.8.0",
            "flake8-bugbear",
            "flake8-comprehensions",
            # Docs
            "mock",
            "sphinx",
            "sphinx-rtd-theme",
            "recommonmark",
            "myst-parser",
        ],
        "contrast_phase": ["xgboost"],
    },
)