131 lines
4.2 KiB
Nix
131 lines
4.2 KiB
Nix
{
|
|
description = "Robossembler Development Environments on Nix";
|
|
inputs = {
|
|
nixpkgs.url = "github:nixos/nixpkgs";
|
|
flake-utils.follows = "ros-overlay/flake-utils";
|
|
ros-overlay.url = "github:lopsided98/nix-ros-overlay?ref=develop";
|
|
ros-nixpkgs.follows = "ros-overlay/nixpkgs";
|
|
gazebo-overlay = {
|
|
url = "github:movefasta/gazebo-sim-overlay";
|
|
inputs.flake-utils.follows = "ros-overlay/flake-utils";
|
|
inputs.nixpkgs.follows = "ros-overlay/nixpkgs";
|
|
|
|
# Transitive inputs "not in registry" https://github.com/NixOS/nix/issues/5790
|
|
# inputs.nixgl.inputs.nixpkgs.follows = "ros-overlay/nixpkgs";
|
|
# inputs.nixgl.inputs.flake-utils.follows = "ros-overlay/flake-utils";
|
|
};
|
|
};
|
|
|
|
outputs = { self, nixpkgs, ros-overlay, ros-nixpkgs, gazebo-overlay, flake-utils }:
|
|
flake-utils.lib.eachDefaultSystem
|
|
(system:
|
|
let
|
|
pkgs = import nixpkgs { inherit system; overlays = [
|
|
(import ./overlay.nix)
|
|
gazebo-overlay.overlays.default
|
|
];
|
|
config.allowUnfree = true;
|
|
};
|
|
ros = import ros-nixpkgs { inherit system; overlays = [
|
|
ros-overlay.overlays.default
|
|
(import ./ros)
|
|
];};
|
|
in
|
|
{
|
|
|
|
rlib = import ./. {
|
|
inherit pkgs;
|
|
inherit (pkgs) lib rbs-workbench cad-gen cg-gen;
|
|
inherit (pkgs.stdenv) mkDerivation;
|
|
inherit (pkgs.libsForQt5) wrapQtAppsHook;
|
|
};
|
|
|
|
packages = {
|
|
inherit (pkgs) cad-gen cg-gen blender cad-gen-docker;
|
|
};
|
|
|
|
inherit ros pkgs;
|
|
|
|
devShells = {
|
|
default =
|
|
let
|
|
my-python-packages = p: with p; [
|
|
unified-planning
|
|
scipy
|
|
spatialmath
|
|
rcg-pipeline
|
|
rbs-workbench
|
|
];
|
|
in
|
|
pkgs.mkShell {
|
|
packages = (with pkgs; [ freecad ]) ++ [ (pkgs.python3.withPackages my-python-packages) ];
|
|
};
|
|
|
|
test =
|
|
let
|
|
my-python-packages = p: with p; [ scenario ];
|
|
in
|
|
pkgs.mkShell {
|
|
packages = [ (pkgs.python3.withPackages my-python-packages) ];
|
|
};
|
|
|
|
ros = pkgs.mkShell {
|
|
name = "ROS environment";
|
|
packages = [
|
|
pkgs.vcstool
|
|
ros.colcon
|
|
pkgs.scenario
|
|
(with ros.rosPackages.humble; buildEnv {
|
|
paths = [
|
|
ros-core
|
|
rmw-fastrtps-dynamic-cpp
|
|
rviz2
|
|
];
|
|
})
|
|
];
|
|
# ROS 2 middleware variants, for Cyclone DDS replace with "rmw_cyclonedds_cpp" and add rmw-cyclonedds-cpp to paths
|
|
RMW_IMPLEMENTATION = "rmw_fastrtps_dynamic_cpp";
|
|
};
|
|
|
|
all = pkgs.mkShell { packages = with pkgs; [ fstl cad-gen cg-gen ]; };
|
|
|
|
ci = pkgs.mkShell { buildInputs = [ pkgs.jq ]; };
|
|
|
|
gazebo = gazebo-overlay.devShells.${system}.default;
|
|
|
|
perception =
|
|
let
|
|
perception-packages = p: with p; [
|
|
pytorchWithCuda
|
|
torchvision
|
|
torchaudio
|
|
detectron2
|
|
scipy
|
|
pyrr
|
|
pillow
|
|
numpy
|
|
opencv4
|
|
boto3
|
|
tensorboardx
|
|
albumentations
|
|
simplejson
|
|
|
|
ultralytics
|
|
];
|
|
in
|
|
pkgs.mkShell {
|
|
packages = [ pkgs.s3cmd pkgs.libGL ] ++ [ (pkgs.python3.withPackages perception-packages) ];
|
|
};
|
|
};
|
|
}
|
|
);
|
|
|
|
nixConfig = {
|
|
extra-substituters = [ "https://ros.cachix.org" "https://robossembler.cachix.org" "https://cuda-maintainers.cachix.org" ];
|
|
extra-trusted-public-keys = [
|
|
"ros.cachix.org-1:dSyZxI8geDCJrwgvCOHDoAfOm5sV1wCPjBkKL+38Rvo="
|
|
"robossembler.cachix.org-1:56jBJHroRQSGpZFkW8XMquuzQTjAF/XTo6MogmBM7SQ="
|
|
"cuda-maintainers.cachix.org-1:0dq3bujKpuEPMCX6U4WylrUDZ9JyUG0VpVZa7CNfq5E="
|
|
];
|
|
};
|
|
}
|