torchio.datasets.visible_human

Complexity

Total Complexity

7

Size/Duplication

Total Lines	75
Duplicated Lines	0 %

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	VisibleHuman.__init__()	0	13	2
A	VisibleHuman.cache_part_dir()	0	3	1
A	VisibleHuman._parse_part()	0	6	2
A	VisibleHuman.cache_class_dir()	0	3	1
A	VisibleHuman.url()	0	3	1

import abc
import tempfile

from .. import Subject, ScalarImage
from ..utils import get_torchio_cache_dir
from ..download import download_and_extract_archive


class VisibleHuman(abc.ABC, Subject):

    URL = 'https://mri.radiology.uiowa.edu/website_documents/visible_human_tar_files/{}{}.tar.gz'  # noqa: E501, FS003

    def __init__(self, part: str):
        self.part = self._parse_part(part)
        if not self.cache_part_dir.is_dir():
            tempdir = tempfile.gettempdir()
            filename = f'{self.__class__.__name__}-{self.part}.tar.gz'
            download_and_extract_archive(
                self.url,
                tempdir,
                filename=filename,
                extract_root=self.cache_class_dir,
                remove_finished=True,
            )
        super().__init__({self.part.lower(): ScalarImage(self.cache_part_dir)})

    @property
    def cache_class_dir(self):
        return get_torchio_cache_dir() / self.__class__.__name__

    @property
    def cache_part_dir(self):
        return self.cache_class_dir / self.part

    @property
    def url(self):
        return self.URL.format(self.PREFIX, self.part)

    def _parse_part(self, part: str) -> None:
        part_capital = part.capitalize()
        if part_capital not in self.PARTS:
            message = f'Part "{part}" not in available parts: {self.PARTS}'
            raise ValueError(message)
        return part_capital


class VisibleMale(VisibleHuman):
    """Visible Male CT Datasets.

    Args:
        part: Can be ``'Head'``, ``'Hip'``, ``'Pelvis'`` or ``'Shoulder'``.
    """

    PREFIX = 'VHMCT1mm_'
    PARTS = (
        'Head',
        'Hip',
        'Pelvis',
        'Shoulder',
    )


class VisibleFemale(VisibleHuman):
    """Visible Female CT Datasets.

    Args:
        part: Can be ``'Ankle'``, ``'Head'``, ``'Hip'``, ``'Knee'``,
            ``'Pelvis'`` or ``'Shoulder'``.
    """

    PREFIX = 'VHF-'
    PARTS = VisibleMale.PARTS + (
        'Ankle',
        'Knee',
    )