depot/third_party/nixpkgs/pkgs/development/python-modules/pillow-heif/default.nix

{
  lib,
  stdenv,
  buildPythonPackage,
  fetchFromGitHub,
  fetchpatch2,

  # build-system
  cmake,
  nasm,
  pkg-config,
  setuptools,

  # native dependencies
  libheif,
  libaom,
  libde265,
  x265,

  # dependencies
  pillow,

  # tests
  opencv4,
  numpy,
  pympler,
  pytestCheckHook,
}:

buildPythonPackage rec {
  pname = "pillow-heif";
  version = "0.17.0";
  pyproject = true;

  src = fetchFromGitHub {
    owner = "bigcat88";
    repo = "pillow_heif";
    rev = "refs/tags/v${version}";
    hash = "sha256-fKh4UbTVj74YxH2vvL24DNmMxg10GSYAmduwuRneE+0=";
  };

  patches = [
    (fetchpatch2 {
      # fix libheif 1.18 support in tests
      url = "https://github.com/bigcat88/pillow_heif/commit/a59434e9ca1138e47e322ddef2adc79e684384f1.patch";
      hash = "sha256-yVT/pnO5KWMnsO95EPCZgyhx6FIJOhsna7t0zpTjWpE=";
    })
  ];

  postPatch = ''
    sed -i '/addopts/d' pyproject.toml
  '';

  nativeBuildInputs = [
    cmake
    nasm
    pkg-config
    setuptools
  ];

  dontUseCmakeConfigure = true;

  buildInputs = [
    libaom
    libde265
    libheif
    x265
  ];

  env = {
    # clang-16: error: argument unused during compilation: '-fno-strict-overflow'
    NIX_CFLAGS_COMPILE = lib.optionalString stdenv.cc.isClang "-Wno-unused-command-line-argument";

    RELEASE_FULL_FLAG = 1;
  };

  propagatedBuildInputs = [ pillow ];

  pythonImportsCheck = [ "pillow_heif" ];

  nativeCheckInputs = [
    opencv4
    numpy
    pympler
    pytestCheckHook
  ];

  disabledTests =
    [
      # Time based
      "test_decode_threads"
    ]
    ++ lib.optionals stdenv.hostPlatform.isDarwin [
      # https://github.com/bigcat88/pillow_heif/issues/89
      # not reproducible in nixpkgs
      "test_opencv_crash"
    ]
    ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [
      # RuntimeError: Encoder plugin generated an error: Unsupported bit depth: Bit depth not supported by x265
      "test_open_heif_compare_non_standard_modes_data"
      "test_open_save_disable_16bit"
      "test_save_bgr_16bit_to_10_12_bit"
      "test_save_bgra_16bit_to_10_12_bit"
      "test_premultiplied_alpha"
      "test_hdr_save"
      "test_I_color_modes_to_10_12_bit"
    ];

  meta = {
    changelog = "https://github.com/bigcat88/pillow_heif/releases/tag/v${version}";
    description = "Python library for working with HEIF images and plugin for Pillow";
    homepage = "https://github.com/bigcat88/pillow_heif";
    license = with lib.licenses; [
      bsd3
      lgpl3
    ];
    maintainers = with lib.maintainers; [ dandellion ];
  };
}