228 lines
9 KiB
Nix
228 lines
9 KiB
Nix
{
|
|
description = "Robossembler Development Environments on Nix";
|
|
inputs = {
|
|
nixpkgs.follows = "ros-flake/nixpkgs";
|
|
flake-utils.follows = "ros-flake/flake-utils";
|
|
ros-flake.url = "github:lopsided98/nix-ros-overlay?ref=develop";
|
|
};
|
|
|
|
outputs = { self, nixpkgs, ros-flake, flake-utils }:
|
|
flake-utils.lib.eachDefaultSystem
|
|
(system:
|
|
let
|
|
pkgs = nixpkgs.legacyPackages.${system}.extend overlay;
|
|
ros = ros-flake.legacyPackages.${system};
|
|
|
|
overlay = final: prev: {
|
|
# -- PLACEHOLDER for overlay
|
|
# freecad = prev.freecad.overrideAttrs
|
|
# (old: { cmakeFlags = old.cmakeFlags ++ [ "-DBUILD_GUI=OFF" ]; });
|
|
# cmakeFlag -BUILD_GUI=OFF for build without gui
|
|
pythonPackagesExtensions = prev.pythonPackagesExtensions ++ [
|
|
(
|
|
python-final: python-prev: {
|
|
iopath = python-prev.iopath.overridePythonAttrs (oldAttrs: {
|
|
version = "unstable";
|
|
src = pkgs.fetchFromGitHub {
|
|
owner = "facebookresearch";
|
|
repo = "iopath";
|
|
rev = "9dcb63a46447556460fdb65c21aa6d094a7a788c";
|
|
hash = "sha256-X1LTOfbIAMy6gToNYS0JpeJpHi0jHPsEjrnCq2c9U0E=";
|
|
};
|
|
});
|
|
}
|
|
)
|
|
];
|
|
};
|
|
|
|
# overrided packages and attributes for nix-ros-overlay
|
|
humble-overrided = ros.humble.overrideScope (final: prev: {
|
|
# remove after update github:lopsided98/nixpkgs/nix-ros for https://github.com/NixOS/nixpkgs/pull/224099
|
|
behaviortree-cpp-v3 = prev.behaviortree-cpp-v3.overrideAttrs (old: {
|
|
propagatedBuildInputs = old.propagatedBuildInputs ++ [ pkgs.zeromq ];
|
|
});
|
|
moveit-core = prev.moveit-core.overrideAttrs (old: {
|
|
propagatedBuildInputs = old.propagatedBuildInputs ++ [ prev.tf2-kdl ];
|
|
});
|
|
moveit-kinematics = prev.moveit-kinematics.overrideAttrs (old: {
|
|
propagatedBuildInputs = old.propagatedBuildInputs ++ [ final.moveit-ros-planning ];
|
|
});
|
|
});
|
|
|
|
# robossembler custom packages; defined here for share between devShells and packages
|
|
cg_preferences = pkgs.fetchgit {
|
|
url = "https://gitflic.ru/project/brothermechanic/cg_preferences.git";
|
|
sha256 = "sha256-ahEakvluX7vd+OY6zaDeZIFS8WbsTevRzuEXlBu2/fg=";
|
|
};
|
|
rs_logo = pkgs.fetchgit {
|
|
url = "https://gitlab.com/robossembler/logo.git";
|
|
rev = "0a0235cd154a59b711e87456206ccd7ba69a7e4e";
|
|
sha256 = "sha256-/QsgKdtDHNymsTEmhIzSrOKNUqqkXl8coy+5LwIEkas=";
|
|
fetchLFS = true;
|
|
};
|
|
blender' = pkgs.blender.override {
|
|
python310Packages = pkgs.python311Packages;
|
|
cudaSupport = false;
|
|
spaceNavSupport = false;
|
|
waylandSupport = false;
|
|
};
|
|
blender = blender'.overrideAttrs (old: {
|
|
optix = pkgs.fetchzip {
|
|
# url taken from the archlinux blender PKGBUILD
|
|
url = "https://developer.download.nvidia.com/redist/optix/v7.4/OptiX-7.4.0-Include.zip";
|
|
sha256 = "sha256-ca08XetwaUYC9foeP5bff9kcDfuFgEzopvjspn2s8RY=";
|
|
};
|
|
patches = old.patches ++ [ "${cg_preferences}/patches/cg-defaults.patch" ];
|
|
buildInputs = old.buildInputs ++ [ pkgs.imath pkgs.git ];
|
|
postPatch = old.postPatch + ''
|
|
cp ${cg_preferences}/share/cg_preferences_service.py scripts/startup/
|
|
cp ${cg_preferences}/share/startup.blend release/datafiles/
|
|
cp ${rs_logo}/raster/robossembler_1024x512_roboty-rossii.png release/datafiles/splash.png
|
|
'';
|
|
env.NIX_CFLAGS_COMPILE = old.env.NIX_CFLAGS_COMPILE + " -I${pkgs.imath}/include/Imath";
|
|
postInstall = old.postInstall + ''
|
|
cp -r ${cg_preferences}/cg_preferences $out/share/blender/${
|
|
pkgs.lib.versions.majorMinor pkgs.blender.version
|
|
}/scripts/addons/
|
|
'';
|
|
});
|
|
bpy = pkgs.python3Packages.callPackage ./pkgs/development/python-modules/bpy { };
|
|
pysdf = pkgs.python3Packages.callPackage ./pkgs/development/python-modules/pysdf { };
|
|
pytorch3d = pkgs.python3Packages.callPackage ./pkgs/development/python-modules/pytorch3d { };
|
|
mayo = pkgs.libsForQt5.callPackage ./pkgs/misc/mayo { };
|
|
fstl = pkgs.libsForQt5.callPackage ./pkgs/misc/fstl { };
|
|
open3d = pkgs.python3Packages.callPackage ./pkgs/development/python-modules/open3d { };
|
|
unified-planning = pkgs.python3Packages.callPackage ./pkgs/development/python-modules/unified-planning { };
|
|
blenderproc = pkgs.python3Packages.callPackage ./blenderproc.nix { };
|
|
test-script = pkgs.python3Packages.callPackage ./pkgs/development/python-modules/test-script { };
|
|
webdataset = pkgs.python3Packages.callPackage ./pkgs/development/python-modules/webdataset { };
|
|
ansitable = pkgs.python3Packages.callPackage ./pkgs/development/python-modules/ansitable { };
|
|
megapose6d = pkgs.python3Packages.callPackage ./pkgs/development/python-modules/megapose6d { inherit webdataset; };
|
|
spatialmath = pkgs.python3Packages.callPackage ./pkgs/development/python-modules/spatialmath { inherit ansitable; };
|
|
|
|
test-system = pkgs.callPackage ./pkgs/systems/test { };
|
|
in
|
|
{
|
|
inherit pkgs;
|
|
|
|
rlib = import ./. {
|
|
inherit (pkgs.stdenv) mkDerivation;
|
|
inherit (pkgs) lib;
|
|
inherit mayo;
|
|
inherit (pkgs.libsForQt5) wrapQtAppsHook;
|
|
};
|
|
|
|
packages = {
|
|
inherit mayo pysdf fstl unified-planning test-script blender;
|
|
};
|
|
|
|
devices = {
|
|
inherit test-system;
|
|
};
|
|
|
|
devShells.camera =
|
|
pkgs.mkShell {
|
|
buildInputs = (with humble-overrided; [
|
|
ros2run
|
|
ros2launch
|
|
realsense2-camera
|
|
realsense2-description
|
|
]);
|
|
};
|
|
|
|
devShells.ci = pkgs.mkShell {
|
|
buildInputs =
|
|
(with pkgs; [ jq ]) ++
|
|
(with humble-overrided; [
|
|
superflore
|
|
(python3.withPackages (p: with p; [ python3Packages.rosdep python3Packages.rosdistro ]))
|
|
]);
|
|
};
|
|
|
|
devShells.ros =
|
|
pkgs.mkShell {
|
|
buildInputs = (with humble-overrided; [
|
|
ros2run
|
|
rmw-fastrtps-dynamic-cpp
|
|
urdf
|
|
ament-pep257
|
|
ament-lint-auto
|
|
xacro
|
|
ament-cmake
|
|
rclcpp-action
|
|
tf2
|
|
action-msgs
|
|
ament-copyright
|
|
pythonPackages.pytest
|
|
rosidl-default-generators
|
|
behaviortree-cpp-v3
|
|
geometry-msgs
|
|
rosidl-default-runtime
|
|
rclpy
|
|
std-msgs
|
|
ament-lint-common
|
|
ament-flake8
|
|
tf2-ros
|
|
rclcpp
|
|
moveit-core
|
|
moveit-msgs
|
|
moveit-ros-planning
|
|
moveit-ros-planning-interface
|
|
gazebo-ros2-control
|
|
plansys2-bringup
|
|
image-transport
|
|
cv-bridge
|
|
example-interfaces
|
|
]) ++ (with pkgs; [ nlohmann_json glog lua5_3_compat ]);
|
|
RMW_IMPLEMENTATION = "rmw_fastrtps_dynamic_cpp";
|
|
};
|
|
|
|
devShells.all = pkgs.mkShell {
|
|
packages = [ mayo blender bpy pysdf fstl ];
|
|
};
|
|
|
|
devShells.perception =
|
|
let
|
|
rbs-packages = p: with p; [
|
|
pytorch3d
|
|
torch
|
|
torchvision
|
|
torchaudio
|
|
detectron2
|
|
];
|
|
in
|
|
pkgs.mkShell {
|
|
packages = [ (pkgs.python3.withPackages rbs-packages) ];
|
|
};
|
|
|
|
devShells.default =
|
|
let
|
|
my-python-packages = p: with p; [
|
|
pysdf
|
|
unified-planning
|
|
pybullet
|
|
scipy
|
|
spatialmath
|
|
bpy
|
|
];
|
|
in
|
|
pkgs.mkShell {
|
|
packages = [ (pkgs.python3.withPackages my-python-packages) pkgs.freecad mayo pkgs.bashInteractive ];
|
|
buildInputs = [ ];
|
|
|
|
# Run when the shell is started up
|
|
shellHook = ''
|
|
${pkgs.python3}/bin/python --version
|
|
'';
|
|
};
|
|
}
|
|
);
|
|
|
|
nixConfig = {
|
|
extra-substituters = [ "https://ros.cachix.org" "https://robossembler.cachix.org" ];
|
|
extra-trusted-public-keys = [
|
|
"ros.cachix.org-1:dSyZxI8geDCJrwgvCOHDoAfOm5sV1wCPjBkKL+38Rvo="
|
|
"robossembler.cachix.org-1:56jBJHroRQSGpZFkW8XMquuzQTjAF/XTo6MogmBM7SQ="
|
|
];
|
|
};
|
|
}
|