open3d python package from wheel

pkgs/development/python-modules/open3d/default.nix

@@ -0,0 +1,71 @@
+{ stdenv, lib, buildPythonPackage, fetchPypi, scipy, scikitlearn, numpy
+, matplotlib, ipywidgets, plyfile, pandas, pyyaml, tqdm, tree, unzip, zip
+, autoPatchelfHook, pytorchWithCuda, libtensorflow-bin, libusb, cudaPackages
+, libGL }:
+
+let
+  addict = buildPythonPackage {
+    pname = "addict";
+    version = "2.4.0";
+
+    src = fetchPypi {
+      pname = "addict";
+      version = "2.4.0";
+      sha256 = "1574sicy5ydx9pvva3lbx8qp56z9jbdwbj26aqgjhyh61q723cmk";
+    };
+  };
+
+in buildPythonPackage {
+  pname = "open3d";
+  version = "0.12.0";
+  format = "wheel";
+
+  src = fetchPypi {
+    pname = "open3d";
+    version = "0.12.0";
+    format = "wheel";
+    python = "cp37";
+    abi = "cp37m";
+    platform = "manylinux2014_x86_64";
+    sha256 = "11ysnk2wni4ffg5gipi5yx0i7qa2p11rfaqgwzlp6f1z4fgfsiac";
+  };
+
+  # sklearn dependency does not exist ofc... Why can't people
+  # package their shit normally. This tilts me so much.
+  patchPhase = ''
+    ${unzip}/bin/unzip ./dist/open3d-0.12.0-cp37-cp37m-manylinux2014_x86_64.whl -d tmp
+    rm ./dist/open3d-0.12.0-cp37-cp37m-manylinux2014_x86_64.whl
+    sed -i 's/sklearn/scikit-learn/g' tmp/open3d-0.12.0.dist-info/METADATA
+    cd tmp
+    ${zip}/bin/zip -0 -r ../dist/open3d-0.12.0-cp37-cp37m-manylinux2014_x86_64.whl ./*
+    cd ../
+  '';
+
+  nativeBuildInputs = [
+    autoPatchelfHook
+  ];
+
+  buildInputs = [
+    # so deps
+    stdenv.cc.cc.lib
+    libusb.out
+    pytorchWithCuda
+    libtensorflow-bin
+    cudaPackages.cudatoolkit_10_1.lib
+    libGL
+  ];
+
+  propagatedBuildInputs = [
+    # py deps
+    ipywidgets
+    tqdm
+    pyyaml
+    pandas
+    plyfile
+    scipy
+    scikitlearn
+    numpy
+    addict
+    matplotlib
+  ];
+}