Source code for medical_image.data.in_memory_image

from typing import Optional, Union

import numpy as np
import torch

from medical_image.data.image import Image




class InMemoryImage(Image):
    """Concrete image that lives only in memory (no file I/O).

    Both :meth:`load` and :meth:`save` are no-ops.  Useful for
    intermediate processing results, temporary images, and test fixtures.
    """



    def __init__(
        self,
        file_path: Optional[str] = None,
        array: Optional[Union[np.ndarray, torch.Tensor]] = None,
        width: Optional[int] = None,
        height: Optional[int] = None,
        source_image: Optional[Image] = None,
    ):
        """Initialise an in-memory image.

        Accepts the same parameters as :class:`Image`.
        """
        super().__init__(
            file_path=file_path,
            array=array,
            width=width,
            height=height,
            source_image=source_image,
        )




    def load(self) -> None:
        """No-op (in-memory images have no file to load)."""
        pass




    def save(self) -> None:
        """No-op (in-memory images have no file to save to)."""
        pass