caffe: add Python support

Move to CMake in the process.

pkgs/applications/science/math/caffe/default.nix

@@ -1,24 +1,26 @@
-{ stdenv
+{ stdenv, lib
-, openblas
-, boost
-, cudaSupport ? true
-, cudnnSupport ? false
-, cudnn ? null
-, cudatoolkit
 , fetchFromGitHub
+, cmake
+, boost
 , google-gflags
 , glog
-, hdf5
+, hdf5-cpp
 , leveldb
 , lmdb
 , opencv
 , protobuf
 , snappy
+, atlas
+, doxygen
+, cudaSupport ? true, cudatoolkit
+, cudnnSupport ? false, cudnn ? null
+, pythonSupport ? false, python ? null, numpy ? null
 }:
 
+assert cudnnSupport -> cudaSupport;
+assert pythonSupport -> (python != null && numpy != null);
 
-let optional = stdenv.lib.optional;
+stdenv.mkDerivation rec {
-in stdenv.mkDerivation rec {
   name = "caffe-${version}";
   version = "1.0-rc5";
 
@@ -29,32 +31,34 @@ in stdenv.mkDerivation rec {
     sha256 = "0lfmmc0n6xvkpygvxclzrvd0zigb4yfc5612anv2ahlxpfi9031c";
   };
 
-  preConfigure = "mv Makefile.config.example Makefile.config";
-
-  makeFlags = [ "BLAS=open"
-                (if !cudaSupport then "CPU_ONLY=1" else "CUDA_DIR=${cudatoolkit}") ]
-              ++ optional cudnnSupport "USE_CUDNN=1";
-
-  # too many issues with tests to run them for now
-  doCheck = false;
-  checkTarget = "runtest";
-
   enableParallelBuilding = true;
 
-  buildInputs = [ openblas boost google-gflags glog hdf5 leveldb lmdb opencv
+  nativeBuildInputs = [ cmake doxygen ];
-                  protobuf snappy ]
-                ++ optional cudaSupport cudatoolkit
-                ++ optional cudnnSupport cudnn;
 
-  installPhase = ''
+  cmakeFlags = [ "-DCUDA_ARCH_NAME=All" ]
-    mkdir -p $out/{bin,share,lib}
+               ++ lib.optional (!cudaSupport) "-DCPU_ONLY=ON"
-    for bin in $(find build/tools -executable -type f -name '*.bin');
+               ++ lib.optional (!pythonSupport) "-DBUILD_python=OFF";
-    do
-      cp $bin $out/bin/$(basename $bin .bin)
-    done
 
-    cp -r build/examples $out/share
+  buildInputs = [ boost google-gflags glog protobuf hdf5-cpp lmdb leveldb snappy opencv atlas ]
-    cp -r build/lib $out
+                ++ lib.optional cudaSupport cudatoolkit
+                ++ lib.optional cudnnSupport cudnn
+                ++ lib.optionals pythonSupport [ python numpy ];
+
+  propagatedBuildInputs = lib.optional pythonSupport python.pkgs.protobuf;
+
+  outputs = [ "out" "bin" ];
+  # Don't propagate bin.
+  outputBin = "out";
+
+  postInstall = ''
+    # Internal static library.
+    rm $out/lib/libproto.a
+
+    moveToOutput "bin" "$bin"
+  '' + lib.optionalString pythonSupport ''
+    mkdir -p $out/${python.sitePackages}
+    mv $out/python/caffe $out/${python.sitePackages}
+    rm -rf $out/python
   '';
 
   meta = with stdenv.lib; {

pkgs/top-level/python-packages.nix

@@ -1565,6 +1565,13 @@ in {
     };
   };
 
+  caffe = pkgs.caffe.override {
+    python = self.python;
+    boost = self.boost;
+    numpy = self.numpy;
+    pythonSupport = true;
+  };
+
   capstone = buildPythonPackage rec {
     name = "capstone-3.0.4";
     src = pkgs.fetchurl {