import platform


def get_dataset_mrnet_args(parser, args=[]):
    """
    Get all relevant parameters to handle the dataset
    -> here: MRNET
    """

    # determine path
    if platform.system() == "Linux":
        path = "/home/biomech/Documents/OsteoData/MRNet-v1.0/"
    else:
        path = "C:/Users/Niko/Documents/data/MRNet-v1.0/MRNet-v1.0"
        # path = "C:/Users/ga46yeg/data/MRNet-v1.0"

    # Dataset MRNet:
    # ------------------------------------------------------------------------
    parser.add_argument(
        "--root_dir_mrnet", type=str, default=path, help="Directory of the dataset"
    )
    parser.add_argument(
        "--perspectives",
        type=list,
        default=["axial", "coronal", "sagittal"],
        help="Perspectives of the Mr Scans",
    )
    parser.add_argument(
        "--classes",
        type=list,
        default=["abn", "acl", "men"],
        help="Classify for these classes",
    )
    # ------------------------------------------------------------------------
    return parser