nixpkgs/pkgs/development/libraries/ngt/default.nix

{ stdenv, fetchFromGitHub, cmake, llvmPackages, enableAVX ? false }:
stdenv.mkDerivation rec {
  pname = "NGT";
  version = "v1.12.3-alpha";
  nativeBuildInputs = [ cmake ];

  src = fetchFromGitHub {
    owner = "yahoojapan";
    repo = "NGT";
    rev = "29c88ff6cd5824d3196986d1f50b834565b6c9dd";
    sha256 = "sha256-nu0MJNpaenOB4+evoSVLKmPIuZXVj1Rm9x53+TfhezY=";
  };

  buildInputs = [ llvmPackages.openmp ];
  NIX_ENFORCE_NO_NATIVE=! enableAVX;
  __AVX2__ = if enableAVX then 1 else 0;

  enableParallelBuilding = true;

  meta = with stdenv.lib; {
    homepage = "https://github.com/yahoojapan/NGT";
    description = "Nearest Neighbor Search with Neighborhood Graph and Tree for High-dimensional Data";
    platforms = platforms.linux ++ platforms.darwin;
    license = licenses.asl20;
    maintainers = with maintainers; [ tomberek ];
  };
}