UniEnv Genesis Lab

This repository contains the implementation of the adaptor that connects the Genesis simulation engine to the UniEnv's node system interface. It is designed as a full simulation framework in the spirit of mjlab and IsaacLab, bringing the same modular, composable scene-building philosophy to Genesis.

Node System

mjlab (MuJoCo's Python interface) and IsaacLab (NVIDIA's GPU-accelerated RL framework for Omniverse) both share a common idea: a simulation environment should be assembled from independent, reusable building blocks — robots, sensors, scene elements — each responsible for its own lifecycle. genesis_adaptor mirrors this philosophy for the Genesis engine, providing:

• A node-based scene description system that lets you compose robots, sensors, and scene geometry declaratively.
• A priority-ordered lifecycle that guarantees correct initialisation order without manual dependency management.
• Pluggable controllers that decouple the physics entity from the control law, just as IsaacLab separates actuator models from robot assets.
• Transparent batching so that the same node code runs identically on a single environment or thousands of parallelised GPU environments.

Lifecycle Phases & Priorities

Each lifecycle phase is executed by sorting all registered nodes by their declared priority for that phase and calling them in descending order. A higher priority number runs first.

Typical priority assignments:

This means a floor plane is always added to the scene before a robot, and a camera that is attached to a robot link is always attached after the robot entity exists.

Composing a Scene

world  = GenesisWorld()
plane  = PlaneBuilder(world, name="floor")
table  = TableBuilder(world, name="table", pos=(0, 0, 0.6))
franka = FrankaFR3WithHand(world, name="franka", pos=(-0.7, 0, 0.6))
camera = CameraSensorNode(world, name="camera", ...)

env = WorldEnv(world, [plane, table, franka, camera])
context, obs, info = env.reset(seed=0, reload=True)
# obs = {
#   "franka": {"joint_position": ..., "eef_position": ..., ...},
#   "camera": {"rgb": ..., "depth": ...},
# }

WorldEnv calls each node's lifecycle methods in the correct priority order automatically. You do not manage dependencies by hand.

Articulated Rooms

Rooms furnished with fixed articulated furniture (RoboCasa kitchen cabinets, drawers and appliances; ReplicaCAD fridges, cupboards and doors; ArtVIP USD appliances; MolmoSpaces Ai2-THOR furniture; Lightwheel SimReady assets; NVIDIA SimReady Warehouse rigid props) are built by RoomBuilder (baked layouts or procedural sampling), and a single named joint of one furniture instance can be driven by OpenArticulatedJointTask — dense fraction-tracking reward, per-env goals, masked resets:

room = RoomBuilder(world, name="room", layout_yaml="robocasa_kitchen/open_kitchen")
task = OpenArticulatedJointTask(
    world, room, furniture_key="cabinet_hinge", joint_name="rightdoorhinge",
    grasp_link_name="cabinet_hinge_right_door_handle",
)
env = WorldEnv(world, [room, franka, task])

Procedural rooms sample from a catalog-backed furniture pool, e.g. make_artvip_pool(tags="kitchen", n=6) for the ArtVIP USD appliances, make_molmospaces_pool(tags="kitchen", n=6) for the MolmoSpaces THOR furniture, make_lightwheel_pool(tags="kitchen", n=6) for the Lightwheel SimReady assets or make_simready_warehouse_pool(tags="prop", n=8) for the Warehouse rigid props (see docs/articulated_rooms.md).

See docs/articulated_rooms.md for the API, the packaged assets (RoboCasa CC BY 4.0; ArtVIP is Apache-2.0 per its card with no committed LICENSE file — caveat there; ReplicaCAD and Lightwheel are CC BY-NC 4.0 — non-commercial, see the license caveats there; MolmoSpaces is CC BY 4.0 with attribution; the SimReady Warehouse card tags CC BY 4.0 but the archive commits no LICENSE file — caveat there) and how to regenerate them, plus the runnable examples/room_open_drawer_fr3.py.

Installation

Install from PyPI:

pip install unienv_genesis

Or install the adaptor locally from source:

git clone https://github.com/UniEnvOrg/genesis_adaptor/
cd genesis_adaptor
pip install -e .