depot/third_party/nixpkgs/pkgs/development/python-modules/imageio/default.nix
Default email fa5436e0a7 Project import generated by Copybara.
GitOrigin-RevId: e8057b67ebf307f01bdcc8fba94d94f75039d1f6
2024-06-05 17:53:02 +02:00

123 lines
2.5 KiB
Nix

{
lib,
stdenv,
buildPythonPackage,
pythonOlder,
fetchFromGitHub,
isPyPy,
substituteAll,
# build-system
setuptools,
# native dependencies
libGL,
# dependencies
numpy,
pillow,
# optional-dependencies
astropy,
av,
imageio-ffmpeg,
pillow-heif,
psutil,
tifffile,
# tests
pytestCheckHook,
fsspec,
}:
buildPythonPackage rec {
pname = "imageio";
version = "2.34.1";
pyproject = true;
disabled = pythonOlder "3.8";
src = fetchFromGitHub {
owner = "imageio";
repo = "imageio";
rev = "refs/tags/v${version}";
hash = "sha256-/VZUifiz8iImq+JLvckFDr7YMIqu0Xro2t3GFj0obg0=";
};
patches = lib.optionals (!stdenv.isDarwin) [
(substituteAll {
src = ./libgl-path.patch;
libgl = "${libGL.out}/lib/libGL${stdenv.hostPlatform.extensions.sharedLibrary}";
})
];
build-system = [ setuptools ];
dependencies = [
numpy
pillow
];
passthru.optional-dependencies = {
bsdf = [ ];
dicom = [ ];
feisem = [ ];
ffmpeg = [
imageio-ffmpeg
psutil
];
fits = lib.optionals (!isPyPy) [ astropy ];
freeimage = [ ];
lytro = [ ];
numpy = [ ];
pillow = [ ];
simpleitk = [ ];
spe = [ ];
swf = [ ];
tifffile = [ tifffile ];
pyav = [ av ];
heif = [ pillow-heif ];
};
nativeCheckInputs =
[
fsspec
psutil
pytestCheckHook
]
++ fsspec.optional-dependencies.github
++ lib.flatten (builtins.attrValues passthru.optional-dependencies);
pytestFlagsArray = [ "-m 'not needs_internet'" ];
preCheck = ''
export IMAGEIO_USERDIR="$TMP"
export HOME=$TMPDIR
'';
disabledTestPaths = [
# tries to fetch fixtures over the network
"tests/test_freeimage.py"
"tests/test_pillow.py"
"tests/test_spe.py"
"tests/test_swf.py"
];
disabledTests = lib.optionals stdenv.isDarwin [
# Segmentation fault
"test_bayer_write"
# RuntimeError: No valid H.264 encoder was found with the ffmpeg installation
"test_writer_file_properly_closed"
"test_writer_pixelformat_size_verbose"
"test_writer_ffmpeg_params"
"test_reverse_read"
];
meta = with lib; {
description = "Library for reading and writing a wide range of image, video, scientific, and volumetric data formats";
homepage = "https://imageio.readthedocs.io";
changelog = "https://github.com/imageio/imageio/blob/v${version}/CHANGELOG.md";
license = licenses.bsd2;
maintainers = with maintainers; [ Luflosi ];
};
}