mujoco-react 8.11.0 → 9.1.0

package/README.md

@@ -154,7 +154,7 @@ import { ScenarioLighting, VisualScenarioEffects } from "mujoco-react";
 <MujocoCanvas config={sceneConfig}>
   <VisualScenarioEffects
     scenario={scenario}
-    materialFilter={(object) => object.name.startsWith("prop_")}
+    materialFilter={({ object }) => object.name.startsWith("prop_")}
   />
   <ScenarioLighting preset={scenario.lighting} />
 </MujocoCanvas>;
@@ -216,7 +216,7 @@ function Scene() {
 import { useBeforePhysicsStep } from "mujoco-react";
 
 function MyController() {
-  useBeforePhysicsStep((_model, data) => {
+  useBeforePhysicsStep(({ data }) => {
     data.ctrl[0] = Math.sin(data.time);
   });
 
@@ -284,7 +284,7 @@ import type { ControlGroupInfo } from "mujoco-react";
 function HoldTcpPose() {
   const armRef = useRef<ControlGroupInfo | null>(null);
 
-  useBeforePhysicsStep((model, data) => {
+  useBeforePhysicsStep(({ model, data }) => {
     armRef.current ??= resolveControlGroup(model, { siteName: RobotSites.franka.tcp });
     if (!armRef.current) return;
 
@@ -305,7 +305,7 @@ Build policy-ready observation vectors from common MuJoCo state without hard-cod
 import { buildObservation, useBeforePhysicsStep } from "mujoco-react";
 
 function PolicyDriver() {
-  useBeforePhysicsStep((model, data) => {
+  useBeforePhysicsStep(({ model, data }) => {
     const obs = buildObservation(model, data, {
       qpos: true,
       qvel: true,
@@ -361,7 +361,7 @@ function useWebSocketControls(url: string) {
   }, [url]);
 
   // Apply incoming actuator controls each physics step.
-  useBeforePhysicsStep((model, data) => {
+  useBeforePhysicsStep(({ model, data }) => {
     const ctrl = latestCtrlRef.current;
     if (!ctrl) return;
     for (let i = 0; i < Math.min(ctrl.length, model.nu); i++) {
@@ -370,7 +370,7 @@ function useWebSocketControls(url: string) {
   });
 
   // Send simulation feedback back after physics.
-  useAfterPhysicsStep((model, data) => {
+  useAfterPhysicsStep(({ data }) => {
     const ws = wsRef.current;
     if (!ws || ws.readyState !== WebSocket.OPEN) return;
 
@@ -433,8 +433,8 @@ The built-in `useIkController()` uses Damped Least-Squares. Pass `ikSolveFn` to
 import { RobotSites } from "mujoco-react";
 import type { IKSolveFn } from "mujoco-react";
 
-const myIK: IKSolveFn = (pos, quat, currentQ) => {
-  return myAnalyticalSolver(pos, currentQ); // return joint angles or null
+const myIK: IKSolveFn = ({ position, currentQ }) => {
+  return myAnalyticalSolver(position, currentQ); // return joint angles or null
 };
 
 const ik = useIkController({ siteName: RobotSites.franka.tcp, ikSolveFn: myIK });
@@ -526,7 +526,7 @@ interface SceneConfig {
   sceneObjects?: SceneObject[];     // Objects injected into scene XML at load time
   homeJoints?: number[];            // Initial joint positions
   xmlPatches?: XmlPatch[];          // Patches applied to XML files during loading
-  onReset?: (model, data) => void;  // Called during reset after mj_resetData
+  onReset?: ({ model, data }) => void;  // Called during reset after mj_resetData
 }
 ```
 
@@ -636,10 +636,10 @@ Thin wrapper around R3F `<Canvas>`. Accepts all R3F Canvas props plus:
 | Prop | Type | Description |
 |------|------|-------------|
 | `config` | `SceneConfig` | **Required.** Scene/robot configuration |
-| `onReady` | `(api: MujocoSimAPI) => void` | Fires when model is loaded |
+| `onReady` | `({ api }) => void` | Fires when model is loaded |
 | `onError` | `(error: Error) => void` | Fires on scene load failure |
-| `onStep` | `(time: number) => void` | Called each physics step |
-| `onSelection` | `(bodyId: number, name: string) => void` | Called on double-click |
+| `onStep` | `({ time, model, data }) => void` | Called each physics step |
+| `onSelection` | `({ bodyId, name }) => void` | Called on double-click |
 | `gravity` | `[number, number, number]` | Override model gravity |
 | `timestep` | `number` | Override model.opt.timestep |
 | `substeps` | `number` | mj_step calls per frame |
@@ -664,10 +664,10 @@ Physics provider for use inside your own R3F `<Canvas>`. Same physics props as `
 | Prop | Type | Description |
 |------|------|-------------|
 | `config` | `SceneConfig` | **Required.** Scene/robot configuration |
-| `onReady` | `(api: MujocoSimAPI) => void` | Fires when model is loaded |
+| `onReady` | `({ api }) => void` | Fires when model is loaded |
 | `onError` | `(error: Error) => void` | Fires on scene load failure |
-| `onStep` | `(time: number) => void` | Called each physics step |
-| `onSelection` | `(bodyId: number, name: string) => void` | Called on double-click |
+| `onStep` | `({ time, model, data }) => void` | Called each physics step |
+| `onSelection` | `({ bodyId, name }) => void` | Called on double-click |
 | `gravity` | `[number, number, number]` | Override model gravity |
 | `timestep` | `number` | Override model.opt.timestep |
 | `substeps` | `number` | mj_step calls per frame |
@@ -715,7 +715,7 @@ drei PivotControls gizmo that tracks a MuJoCo site and drives IK on drag. Requir
 | `controller` | `IkContextValue` | **required** | Controller from `useIkController()` |
 | `siteName` | `string?` | controller's site | MuJoCo site to track |
 | `scale` | `number?` | `0.18` | Gizmo handle scale |
-| `onDrag` | `(pos, quat) => void` | -- | Custom drag handler (disables auto-IK) |
+| `onDrag` | `({ position, quaternion }) => void` | -- | Custom drag handler (disables auto-IK) |
 
 ### `<DragInteraction />`
 
@@ -826,7 +826,7 @@ if (mujoco) {
 Run logic **before** `mj_step` each frame. Write to `data.ctrl`, apply forces, drive automation.
 
 ```tsx
-useBeforePhysicsStep((model, data) => {
+useBeforePhysicsStep(({ data }) => {
   data.ctrl[0] = Math.sin(data.time);
 });
 ```
@@ -983,6 +983,20 @@ const video = useVideoRecorder({ fps: 30, mimeType: "video/webm" });
 // video.start(), video.stop() -> returns Blob
 ```
 
+### Frame Capture
+
+Capture dataset/debug stills from the rendered MuJoCo canvas:
+
+```tsx
+const apiRef = useRef<MujocoSimAPI>(null);
+
+const frame = await apiRef.current?.captureFrame({ type: "image/png" });
+const blob = await apiRef.current?.captureFrameBlob({ type: "image/png" });
+```
+
+Use `useFrameCapture()` or the standalone `captureFrame()` helpers when you own
+the canvas or want to capture a custom container.
+
 ### `useCtrlNoise(config)`
 
 Apply Gaussian noise to controls for robustness testing:
@@ -1104,7 +1118,10 @@ The full API object available via `ref` or `useMujoco()` (when `isReady`):
 |--------|-------------|
 | `raycast(origin, direction, maxDist?)` | Physics raycast via `mj_ray` |
 | `project2DTo3D(x, y, camPos, lookAt)` | Screen-to-world raycast (returns bodyId + geomId) |
+| `getCanvas()` | Return the underlying R3F canvas element |
 | `getCanvasSnapshot(w?, h?, mime?)` | Base64 screenshot |
+| `captureFrame(options?)` | Capture the canvas as a data URL |
+| `captureFrameBlob(options?)` | Capture the canvas as a Blob |
 
 ### Scene Management
 

package/dist/{chunk-SEWQULWO.js → chunk-33CV6HSV.js}

@@ -341,7 +341,7 @@ function applyScenarioMaterials(scene, scenario, snapshots, materialFilter) {
     for (const material of normalizeMaterials(object.material)) {
       const mutable = getMutableScenarioMaterial(material);
       if (!mutable) continue;
-      if (materialFilter && !materialFilter(object, material)) continue;
+      if (materialFilter && !materialFilter({ object, material })) continue;
       if (!snapshots.has(material)) {
         snapshots.set(material, {
           color: mutable.color.clone(),
@@ -396,5 +396,5 @@ function clamp01(value) {
  */
 
 export { ScenarioLighting, SplatEnvironment, VisualScenarioEffects, createPairedSplatEnvironment, createSparkSplatViewerUrl, createSplatEnvironmentUserData, getScenarioBackground, getScenarioCameraPosition, useSplatEnvironment, useVisualScenarioEffects, withSplatEnvironment };
-//# sourceMappingURL=chunk-SEWQULWO.js.map
-//# sourceMappingURL=chunk-SEWQULWO.js.map
+//# sourceMappingURL=chunk-33CV6HSV.js.map
+//# sourceMappingURL=chunk-33CV6HSV.js.map

package/dist/chunk-33CV6HSV.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/components/VisualScenario.tsx"],"names":[],"mappings":";;;;;;AA2BA,IAAM,kBAAA,GAAqB,SAAA;AAEpB,SAAS,gBAAA,CAAiB;AAAA,EAC/B,MAAA,GAAS,QAAA;AAAA,EACT,UAAA,GAAa,IAAA;AAAA,EACb,SAAA,GAAY;AACd,CAAA,EAA0B;AACxB,EAAA,IAAI,WAAW,WAAA,EAAa;AAC1B,IAAA,uBACE,IAAA,CAAA,QAAA,EAAA,EACE,QAAA,EAAA;AAAA,sBAAA,GAAA,CAAC,cAAA,EAAA,EAAa,SAAA,EAAW,IAAA,GAAO,SAAA,EAAW,CAAA;AAAA,sBAC3C,GAAA;AAAA,QAAC,kBAAA;AAAA,QAAA;AAAA,UACC,QAAA,EAAU,CAAC,GAAA,EAAK,EAAA,EAAI,CAAC,CAAA;AAAA,UACrB,WAAW,GAAA,GAAM,SAAA;AAAA,UACjB;AAAA;AAAA,OACF;AAAA,sBACA,GAAA,CAAC,kBAAA,EAAA,EAAiB,QAAA,EAAU,CAAC,EAAA,EAAI,KAAK,GAAG,CAAA,EAAG,SAAA,EAAW,IAAA,GAAO,SAAA,EAAW;AAAA,KAAA,EAC3E,CAAA;AAAA,EAEJ;AAEA,EAAA,IAAI,WAAW,WAAA,EAAa;AAC1B,IAAA,uBACE,IAAA,CAAA,QAAA,EAAA,EACE,QAAA,EAAA;AAAA,sBAAA,GAAA,CAAC,cAAA,EAAA,EAAa,SAAA,EAAW,IAAA,GAAO,SAAA,EAAW,CAAA;AAAA,sBAC3C,GAAA;AAAA,QAAC,kBAAA;AAAA,QAAA;AAAA,UACC,QAAA,EAAU,CAAC,CAAA,EAAG,EAAA,EAAI,CAAC,CAAA;AAAA,UACnB,WAAW,IAAA,GAAO,SAAA;AAAA,UAClB;AAAA;AAAA,OACF;AAAA,sBACA,GAAA,CAAC,YAAA,EAAA,EAAW,QAAA,EAAU,CAAC,IAAA,EAAM,MAAM,GAAG,CAAA,EAAG,SAAA,EAAW,GAAA,GAAM,SAAA,EAAW;AAAA,KAAA,EACvE,CAAA;AAAA,EAEJ;AAEA,EAAA,IAAI,WAAW,OAAA,EAAS;AACtB,IAAA,uBACE,IAAA,CAAA,QAAA,EAAA,EACE,QAAA,EAAA;AAAA,sBAAA,GAAA,CAAC,cAAA,EAAA,EAAa,SAAA,EAAW,IAAA,GAAO,SAAA,EAAW,CAAA;AAAA,sBAC3C,GAAA;AAAA,QAAC,kBAAA;AAAA,QAAA;AAAA,UACC,QAAA,EAAU,CAAC,GAAA,EAAK,IAAA,EAAM,GAAG,CAAA;AAAA,UACzB,WAAW,GAAA,GAAM,SAAA;AAAA,UACjB;AAAA;AAAA,OACF;AAAA,sBACA,GAAA,CAAC,YAAA,EAAA,EAAW,QAAA,EAAU,CAAC,GAAA,EAAK,KAAK,GAAG,CAAA,EAAG,SAAA,EAAW,IAAA,GAAO,SAAA,EAAW;AAAA,KAAA,EACtE,CAAA;AAAA,EAEJ;AAEA,EAAA,uBACE,IAAA,CAAA,QAAA,EAAA,EACE,QAAA,EAAA;AAAA,oBAAA,GAAA,CAAC,cAAA,EAAA,EAAa,SAAA,EAAW,IAAA,GAAO,SAAA,EAAW,CAAA;AAAA,oBAC3C,GAAA;AAAA,MAAC,kBAAA;AAAA,MAAA;AAAA,QACC,QAAA,EAAU,CAAC,GAAA,EAAK,EAAA,EAAI,CAAC,CAAA;AAAA,QACrB,WAAW,GAAA,GAAM,SAAA;AAAA,QACjB;AAAA;AAAA;AACF,GAAA,EACF,CAAA;AAEJ;AAEO,SAAS,qBAAA,CACd,MAAA,EACA,QAAA,GAAW,kBAAA,EACX;AACA,EAAA,IAAI,MAAA,KAAW,aAAa,OAAO,SAAA;AACnC,EAAA,IAAI,MAAA,KAAW,aAAa,OAAO,SAAA;AACnC,EAAA,IAAI,MAAA,KAAW,SAAS,OAAO,SAAA;AAC/B,EAAA,OAAO,QAAA;AACT;AAEO,SAAS,yBAAA,CACd,cACA,QAAA,EAC0B;AAC1B,EAAA,MAAM,CAAC,CAAA,EAAG,CAAA,EAAG,CAAC,CAAA,GAAI,YAAA;AAClB,EAAA,MAAM,MAAA,GAAS,QAAA,EAAU,MAAA,EAAQ,MAAA,IAAU,CAAA;AAE3C,EAAA,OAAO;AAAA,IACL,QAAQ,CAAA,GAAI,MAAA,GAAS,GAAA,EAAK,OAAA,CAAQ,CAAC,CAAC,CAAA;AAAA,IACpC,QAAQ,CAAA,GAAI,MAAA,GAAS,GAAA,EAAK,OAAA,CAAQ,CAAC,CAAC,CAAA;AAAA,IACpC,QAAQ,CAAA,GAAI,MAAA,GAAS,IAAA,EAAM,OAAA,CAAQ,CAAC,CAAC;AAAA,GACvC;AACF;AAEO,SAAS,sBAAsB,KAAA,EAAmC;AACvE,EAAA,wBAAA,CAAyB,KAAK,CAAA;AAC9B,EAAA,OAAO,IAAA;AACT;AAEO,SAAS,wBAAA,CAAyB;AAAA,EACvC,QAAA;AAAA,EACA,OAAA,GAAU,IAAA;AAAA,EACV,eAAA,GAAkB,IAAA;AAAA,EAClB,QAAA,GAAW,IAAA;AAAA,EACX,aAAA,GAAgB,IAAA;AAAA,EAChB,cAAA,GAAiB,IAAA;AAAA,EACjB,UAAA;AAAA,EACA,OAAA;AAAA,EACA,MAAA;AAAA,EACA;AACF,CAAA,EAA+B;AAC7B,EAAA,MAAM,EAAE,EAAA,EAAI,KAAA,EAAO,UAAA,KAAe,QAAA,EAAS;AAE3C,EAAA,SAAA,CAAU,MAAM;AACd,IAAA,IAAI,CAAC,OAAA,IAAW,CAAC,QAAA,EAAU;AACzB,MAAA,OAAO,MAAA;AAAA,IACT;AAEA,IAAA,MAAM,mBAAmB,EAAA,CAAG,mBAAA;AAC5B,IAAA,MAAM,qBAAqB,KAAA,CAAM,UAAA;AACjC,IAAA,MAAM,cAAc,KAAA,CAAM,GAAA;AAC1B,IAAA,MAAM,iBAAA,uBAAwB,GAAA,EAO5B;AAEF,IAAA,IAAI,aAAA,EAAe;AACjB,MAAA,EAAA,CAAG,mBAAA,GAAsB,QAAA,CAAS,MAAA,EAAQ,QAAA,IAAY,CAAA;AAAA,IACxD;AAEA,IAAA,IAAI,eAAA,EAAiB;AACnB,MAAA,KAAA,CAAM,aAAa,IAAU,KAAA,CAAA,KAAA;AAAA,QAC3B,UAAA,IAAc,qBAAA,CAAsB,QAAA,CAAS,QAAQ;AAAA,OACvD;AAAA,IACF;AAEA,IAAA,IAAI,QAAA,EAAU;AACZ,MAAA,KAAA,CAAM,GAAA,GAAM,iBAAA,CAAkB,QAAA,EAAU,UAAA,EAAY,SAAS,MAAM,CAAA;AAAA,IACrE;AAEA,IAAA,IAAI,cAAA,IAAkB,SAAS,SAAA,EAAW;AACxC,MAAA,sBAAA,CAAuB,KAAA,EAAO,QAAA,EAAU,iBAAA,EAAmB,cAAc,CAAA;AAAA,IAC3E;AAEA,IAAA,UAAA,EAAW;AAEX,IAAA,OAAO,MAAM;AACX,MAAA,EAAA,CAAG,mBAAA,GAAsB,gBAAA;AACzB,MAAA,KAAA,CAAM,UAAA,GAAa,kBAAA;AACnB,MAAA,KAAA,CAAM,GAAA,GAAM,WAAA;AAEZ,MAAA,KAAA,MAAW,CAAC,QAAA,EAAU,QAAQ,CAAA,IAAK,iBAAA,EAAmB;AACpD,QAAA,MAAM,OAAA,GAAU,2BAA2B,QAAQ,CAAA;AACnD,QAAA,IAAI,CAAC,OAAA,EAAS;AACd,QAAA,IAAI,SAAS,KAAA,EAAO,OAAA,CAAQ,KAAA,CAAM,IAAA,CAAK,SAAS,KAAK,CAAA;AACrD,QAAA,IAAI,OAAO,QAAA,CAAS,SAAA,KAAc,QAAA,EAAU;AAC1C,UAAA,OAAA,CAAQ,YAAY,QAAA,CAAS,SAAA;AAAA,QAC/B;AACA,QAAA,IAAI,OAAO,QAAA,CAAS,SAAA,KAAc,QAAA,EAAU;AAC1C,UAAA,OAAA,CAAQ,YAAY,QAAA,CAAS,SAAA;AAAA,QAC/B;AACA,QAAA,OAAA,CAAQ,WAAA,GAAc,IAAA;AAAA,MACxB;AAEA,MAAA,UAAA,EAAW;AAAA,IACb,CAAA;AAAA,EACF,CAAA,EAAG;AAAA,IACD,eAAA;AAAA,IACA,QAAA;AAAA,IACA,cAAA;AAAA,IACA,aAAA;AAAA,IACA,UAAA;AAAA,IACA,OAAA;AAAA,IACA,MAAA;AAAA,IACA,OAAA;AAAA,IACA,EAAA;AAAA,IACA,UAAA;AAAA,IACA,cAAA;AAAA,IACA,QAAA;AAAA,IACA;AAAA,GACD,CAAA;AACH;AASO,SAAS,gBAAA,CAAiB;AAAA,EAC/B,WAAA;AAAA,EACA,QAAA;AAAA,EACA,QAAA;AAAA,EACA,GAAA;AAAA,EACA,MAAA;AAAA,EACA,cAAA;AAAA,EACA,sBAAA;AAAA,EACA,QAAA;AAAA,EACA,eAAA,GAAkB,IAAA;AAAA,EAClB,GAAG;AACL,CAAA,EAA0B;AACxB,EAAA,MAAM,WAAW,mBAAA,CAAoB;AAAA,IACnC,WAAA;AAAA,IACA,QAAA;AAAA,IACA,QAAA;AAAA,IACA,GAAA;AAAA,IACA,MAAA;AAAA,IACA,cAAA,EAAgB;AAAA,GACjB,CAAA;AACD,EAAA,MAAM,gBAAA,GACJ,OAAO,UAAA,CAAW,QAAA,KAAa,QAAA,IAAY,WAAW,QAAA,KAAa,IAAA,GAC/D,UAAA,CAAW,QAAA,GACX,EAAC;AAEP,EAAA,uBACE,IAAA;AAAA,IAAC,OAAA;AAAA,IAAA;AAAA,MACE,GAAG,UAAA;AAAA,MACJ,QAAA,EAAU;AAAA,QACR,GAAG,gBAAA;AAAA,QACH,GAAG,QAAA,CAAS;AAAA,OACd;AAAA,MAEC,QAAA,EAAA;AAAA,QAAA,QAAA;AAAA,QACA,QAAA,IAAY,CAAC,eAAA,GAAkB,IAAA,uBAAQ,gBAAA,EAAA,EAAiB,CAAA;AAAA,QACxD;AAAA;AAAA;AAAA,GACH;AAEJ;AAEO,SAAS,mBAAA,CAAoB;AAAA,EAClC,WAAA;AAAA,EACA,QAAA;AAAA,EACA,QAAA;AAAA,EACA,GAAA;AAAA,EACA,MAAA;AAAA,EACA;AACF,CAAA,EAA4D;AAC1D,EAAA,MAAM,mBAAA,GAAsB,OAAA;AAAA,IAC1B,MACE,gBACC,QAAA,GACG,4BAAA,CAA6B,UAAU,EAAE,QAAA,EAAU,CAAA,GACnD,MAAA,CAAA;AAAA,IACN,CAAC,WAAA,EAAa,QAAA,EAAU,QAAQ;AAAA,GAClC;AACA,EAAA,MAAM,cAAc,GAAA,IAAO,mBAAA,EAAqB,KAAA,CAAM,GAAA,IAAO,UAAU,KAAA,EAAO,GAAA;AAC9E,EAAA,MAAM,iBACJ,MAAA,IACA,mBAAA,EAAqB,MAAM,MAAA,IAC3B,QAAA,EAAU,OAAO,MAAA,IACjB,KAAA;AACF,EAAA,MAAM,yBACJ,cAAA,IACA,mBAAA,EAAqB,cAAA,IACrB,QAAA,EAAU,OAAO,cAAA,IACjB,MAAA;AAEF,EAAA,OAAO,OAAA;AAAA,IACL,OAAO;AAAA,MACL,GAAA,EAAK,WAAA;AAAA,MACL,MAAA,EAAQ,cAAA;AAAA,MACR,cAAA,EAAgB,sBAAA;AAAA,MAChB,UAAU,8BAAA,CAA+B;AAAA,QACvC,WAAA,EAAa,mBAAA;AAAA,QACb,GAAA,EAAK,WAAA;AAAA,QACL,MAAA,EAAQ,cAAA;AAAA,QACR,cAAA,EAAgB;AAAA,OACjB;AAAA,KACH,CAAA;AAAA,IACA,CAAC,mBAAA,EAAqB,WAAA,EAAa,cAAA,EAAgB,sBAAsB;AAAA,GAC3E;AACF;AAOO,SAAS,4BAAA,CACd,QAAA,EACA,OAAA,GAKI,EAAC,EACqC;AAC1C,EAAA,MAAM,QAAQ,QAAA,CAAS,KAAA;AACvB,EAAA,MAAM,iBAAiB,KAAA,EAAO,cAAA;AAE9B,EAAA,IAAI,CAAC,OAAO,OAAA,IAAW,CAAC,MAAM,GAAA,IAAO,CAAC,gBAAgB,OAAA,EAAS;AAC7D,IAAA,OAAO,MAAA;AAAA,EACT;AAEA,EAAA,OAAO;AAAA,IACL,EAAA,EAAI,OAAA,CAAQ,EAAA,IAAM,QAAA,CAAS,EAAA,IAAM,mBAAA;AAAA,IACjC,KAAA,EAAO,OAAA,CAAQ,KAAA,IAAS,QAAA,CAAS,KAAA,IAAS,4BAAA;AAAA,IAC1C,WAAA,EACE,QAAQ,WAAA,KACP,QAAA,CAAS,cACN,CAAA,OAAA,EAAU,QAAA,CAAS,WAAW,CAAA,wCAAA,CAAA,GAC9B,MAAA,CAAA;AAAA,IACN,KAAA,EAAO;AAAA,MACL,KAAK,KAAA,CAAM,GAAA;AAAA,MACX,MAAA,EAAQ,MAAM,MAAA,IAAU,KAAA;AAAA,MACxB,UAAU,OAAA,CAAQ;AAAA,KACpB;AAAA,IACA,cAAA,EAAgB;AAAA,MACd,GAAG,cAAA;AAAA,MACH,SAAS,cAAA,CAAe;AAAA;AAC1B,GACF;AACF;AAEA,SAAS,yBAAyB,KAAA,EAA+D;AAC/F,EAAA,OAAO,CAAC,CAAC,KAAA,IAAS,gBAAA,IAAoB,SAAS,OAAA,IAAW,KAAA;AAC5D;AAEA,SAAS,iBAAA,CAAkB,aAA0B,IAAA,EAAsB;AACzE,EAAA,MAAM,MAAM,WAAA,CAAY,GAAA;AACxB,EAAA,IAAI,CAAC,KAAK,OAAO,IAAA;AAEjB,EAAA,MAAM,OAAO,GAAA,CAAI,QAAA,CAAS,GAAG,CAAA,GAAI,MAAM,GAAA,GAAM,GAAA;AAC7C,EAAA,IAAI,IAAA,CAAK,WAAW,IAAI,CAAA,SAAU,IAAA,CAAK,KAAA,CAAM,KAAK,MAAM,CAAA;AACxD,EAAA,OAAO,IAAA;AACT;AAEA,SAAS,YAAY,KAAA,EAAoC;AACvD,EAAA,MAAM,IAAA,uBAAW,GAAA,EAAY;AAC7B,EAAA,MAAM,SAAmB,EAAC;AAC1B,EAAA,KAAA,MAAW,QAAQ,KAAA,EAAO;AACxB,IAAA,IAAI,IAAA,CAAK,GAAA,CAAI,IAAI,CAAA,EAAG;AACpB,IAAA,IAAA,CAAK,IAAI,IAAI,CAAA;AACb,IAAA,MAAA,CAAO,KAAK,IAAI,CAAA;AAAA,EAClB;AACA,EAAA,OAAO,MAAA;AACT;AASO,SAAS,oBAAA,CACd,WAAA,EACA,KAAA,EACA,OAAA,GAA4C,EAAC,EAChC;AACb,EAAA,MAAM,WAAA,GAAc,yBAAyB,KAAK,CAAA,GAC9C,QACA,KAAA,GACE,4BAAA,CAA6B,KAAA,EAAO,OAAO,CAAA,GAC3C,MAAA;AACN,EAAA,MAAM,OAAA,GAAU,aAAa,cAAA,CAAe,OAAA;AAC5C,EAAA,IAAI,CAAC,SAAS,OAAO,WAAA;AAErB,EAAA,OAAO;AAAA,IACL,GAAG,WAAA;AAAA,IACH,kBAAkB,WAAA,CAAY;AAAA,MAC5B,GAAI,WAAA,CAAY,gBAAA,IAAoB,EAAC;AAAA,MACrC,iBAAA,CAAkB,aAAa,OAAO;AAAA,KACvC;AAAA,GACH;AACF;AAEO,SAAS,8BAAA,CAA+B;AAAA,EAC7C,WAAA;AAAA,EACA,GAAA;AAAA,EACA,MAAA,GAAS,KAAA;AAAA,EACT;AACF,CAAA,EAKG;AACD,EAAA,OAAO;AAAA,IACL,IAAA,EAAM,mBAAA;AAAA,IACN,eAAe,WAAA,EAAa,EAAA;AAAA,IAC5B,kBAAkB,WAAA,EAAa,KAAA;AAAA,IAC/B,QAAA,EAAU,GAAA;AAAA,IACV,WAAA,EAAa,MAAA;AAAA,IACb,aAAA,EAAe,aAAa,KAAA,CAAM,QAAA;AAAA,IAClC,oBAAA,EAAsB,gBAAgB,MAAA,IAAU,SAAA;AAAA,IAChD,uBAAuB,cAAA,EAAgB,OAAA;AAAA,IACvC,wBAAA,EAA0B,cAAA,EAAgB,UAAA,IAAc;AAAC,GAC3D;AACF;AAEO,SAAS,yBAAA,CAA0B;AAAA,EACxC,SAAA;AAAA,EACA;AACF,CAAA,EAGG;AACD,EAAA,MAAM,GAAA,GAAM,IAAI,GAAA,CAAI,SAAA,EAAW,2BAA2B,CAAA;AAC1D,EAAA,GAAA,CAAI,YAAA,CAAa,GAAA,CAAI,OAAA,EAAS,QAAQ,CAAA;AACtC,EAAA,OAAO,SAAA,CAAU,UAAA,CAAW,MAAM,CAAA,GAAI,GAAA,CAAI,QAAA,EAAS,GAAI,CAAA,EAAG,GAAA,CAAI,QAAQ,CAAA,EAAG,GAAA,CAAI,MAAM,CAAA,CAAA;AACrF;AAEA,SAAS,gBAAA,GAAmB;AAC1B,EAAA,uBACE,GAAA,CAAC,WACC,QAAA,kBAAA,IAAA,CAAC,MAAA,EAAA,EAAK,UAAU,CAAC,CAAA,EAAG,CAAA,EAAG,GAAG,CAAA,EACxB,QAAA,EAAA;AAAA,oBAAA,GAAA,CAAC,iBAAY,IAAA,EAAM,CAAC,GAAA,EAAK,GAAA,EAAK,GAAG,CAAA,EAAG,CAAA;AAAA,oBACpC,GAAA;AAAA,MAAC,mBAAA;AAAA,MAAA;AAAA,QACC,KAAA,EAAM,SAAA;AAAA,QACN,WAAA,EAAW,IAAA;AAAA,QACX,OAAA,EAAS,IAAA;AAAA,QACT,SAAA,EAAS,IAAA;AAAA,QACT,IAAA,EAAY,KAAA,CAAA;AAAA;AAAA;AACd,GAAA,EACF,CAAA,EACF,CAAA;AAEJ;AAEA,SAAS,iBAAA,CACP,QAAA,EACA,UAAA,EACA,OAAA,EACA,MAAA,EACA;AACA,EAAA,IAAI,QAAA,CAAS,aAAa,WAAA,EAAa;AACrC,IAAA,OAAO,IAAU,KAAA,CAAA,GAAA;AAAA,MACf,UAAA,IAAc,qBAAA,CAAsB,QAAA,CAAS,QAAQ,CAAA;AAAA,MACrD,OAAA,IAAW,GAAA;AAAA,MACX,MAAA,IAAU;AAAA,KACZ;AAAA,EACF;AAEA,EAAA,IAAI,QAAA,CAAS,aAAa,WAAA,EAAa;AACrC,IAAA,OAAO,IAAU,KAAA,CAAA,GAAA;AAAA,MACf,UAAA,IAAc,qBAAA,CAAsB,QAAA,CAAS,QAAQ,CAAA;AAAA,MACrD,OAAA,IAAW,CAAA;AAAA,MACX,MAAA,IAAU;AAAA,KACZ;AAAA,EACF;AAEA,EAAA,OAAO,IAAA;AACT;AAEA,SAAS,sBAAA,CACP,KAAA,EACA,QAAA,EACA,SAAA,EAQA,cAAA,EACA;AACA,EAAA,MAAM,YAAY,QAAA,CAAS,SAAA;AAC3B,EAAA,IAAI,CAAC,SAAA,EAAW;AAEhB,EAAA,KAAA,CAAM,QAAA,CAAS,CAAC,MAAA,KAAW;AACzB,IAAA,IAAI,EAAE,kBAAwB,KAAA,CAAA,IAAA,CAAA,EAAO;AACnC,MAAA;AAAA,IACF;AAEA,IAAA,KAAA,MAAW,QAAA,IAAY,kBAAA,CAAmB,MAAA,CAAO,QAAQ,CAAA,EAAG;AAC1D,MAAA,MAAM,OAAA,GAAU,2BAA2B,QAAQ,CAAA;AACnD,MAAA,IAAI,CAAC,OAAA,EAAS;AACd,MAAA,IAAI,kBAAkB,CAAC,cAAA,CAAe,EAAE,MAAA,EAAQ,QAAA,EAAU,CAAA,EAAG;AAE7D,MAAA,IAAI,CAAC,SAAA,CAAU,GAAA,CAAI,QAAQ,CAAA,EAAG;AAC5B,QAAA,SAAA,CAAU,IAAI,QAAA,EAAU;AAAA,UACtB,KAAA,EAAO,OAAA,CAAQ,KAAA,CAAM,KAAA,EAAM;AAAA,UAC3B,WAAW,OAAA,CAAQ,SAAA;AAAA,UACnB,WAAW,OAAA,CAAQ;AAAA,SACpB,CAAA;AAAA,MACH;AAEA,MAAA,qBAAA,CAAsB,OAAA,EAAS,MAAA,EAAQ,QAAA,EAAU,SAAS,CAAA;AAAA,IAC5D;AAAA,EACF,CAAC,CAAA;AACH;AAEA,SAAS,qBAAA,CACP,QAAA,EACA,MAAA,EACA,QAAA,EACA,SAAA,EACA;AACA,EAAA,MAAM,IAAA,GAAO,SAAS,IAAA,IAAQ,CAAA;AAC9B,EAAA,MAAM,SAAA,GAAY,CAAA,EAAG,QAAA,CAAS,EAAA,IAAM,UAAU,CAAA,CAAA,EAAI,MAAA,CAAO,IAAI,CAAA,CAAA,EAAI,QAAA,CAAS,IAAI,CAAA,CAAA,EAAI,IAAI,CAAA,CAAA;AACtF,EAAA,MAAM,SAAA,GAAY,mBAAmB,SAAS,CAAA;AAE9C,EAAA,IAAI,UAAU,qBAAA,EAAuB;AACnC,IAAA,QAAA,CAAS,KAAA,CAAM,MAAA,CAAO,SAAA,EAAW,IAAA,EAAM,IAAI,CAAA;AAAA,EAC7C;AAEA,EAAA,IAAI,UAAU,sBAAA,EAAwB;AACpC,IAAA,QAAA,CAAS,SAAA,GAAY,OAAA;AAAA,MACnB,SAAA,CAAU,SAAA,IAAa,IAAA,GAAO,SAAA,GAAY;AAAA,KAC5C;AACA,IAAA,QAAA,CAAS,SAAA,GAAY,OAAA;AAAA,MACnB,SAAA,CAAU,aAAa,SAAA,GAAY;AAAA,KACrC;AAAA,EACF;AAEA,EAAA,QAAA,CAAS,WAAA,GAAc,IAAA;AACzB;AAEA,SAAS,mBACP,QAAA,EACkB;AAClB,EAAA,OAAO,MAAM,OAAA,CAAQ,QAAQ,CAAA,GAAI,QAAA,GAAW,CAAC,QAAQ,CAAA;AACvD;AAEA,SAAS,2BACP,QAAA,EACgE;AAChE,EAAA,IACE,QAAA,YAA0B,KAAA,CAAA,oBAAA,IAC1B,QAAA,YAA0B,KAAA,CAAA,oBAAA,EAC1B;AACA,IAAA,OAAO,QAAA;AAAA,EACT;AAEA,EAAA,OAAO,IAAA;AACT;AAEA,SAAS,mBAAmB,KAAA,EAAe;AACzC,EAAA,IAAI,IAAA,GAAO,UAAA;AACX,EAAA,KAAA,IAAS,QAAQ,CAAA,EAAG,KAAA,GAAQ,KAAA,CAAM,MAAA,EAAQ,SAAS,CAAA,EAAG;AACpD,IAAA,IAAA,IAAQ,KAAA,CAAM,WAAW,KAAK,CAAA;AAC9B,IAAA,IAAA,GAAO,IAAA,CAAK,IAAA,CAAK,IAAA,EAAM,QAAQ,CAAA;AAAA,EACjC;AACA,EAAA,OAAA,CAAQ,SAAS,CAAA,IAAK,UAAA;AACxB;AAEA,SAAS,QAAQ,KAAA,EAAe;AAC9B,EAAA,OAAO,KAAK,GAAA,CAAI,CAAA,EAAG,KAAK,GAAA,CAAI,CAAA,EAAG,KAAK,CAAC,CAAA;AACvC","file":"chunk-33CV6HSV.js","sourcesContent":["/**\n * @license\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { useThree } from '@react-three/fiber';\nimport type { ThreeElements } from '@react-three/fiber';\nimport type { ReactNode } from 'react';\nimport { useEffect, useMemo } from 'react';\nimport * as THREE from 'three';\nimport type {\n  PairedSplatEnvironmentConfig,\n  ScenarioMaterialConfig,\n  SceneConfig,\n  SplatCollisionProxyConfig,\n  SplatEnvironmentMetadata,\n  SplatEnvironmentMetadataInput,\n  SplatFormat,\n  SplatRendererKind,\n  SplatSceneInput,\n  ScenarioLightingPreset,\n  ScenarioLightingProps,\n  SplatEnvironmentProps,\n  VisualScenarioConfig,\n  VisualScenarioEffectsProps,\n} from '../types';\n\nconst DEFAULT_BACKGROUND = '#181a1f';\n\nexport function ScenarioLighting({\n  preset = 'studio',\n  castShadow = true,\n  intensity = 1,\n}: ScenarioLightingProps) {\n  if (preset === 'warehouse') {\n    return (\n      <>\n        <ambientLight intensity={0.18 * intensity} />\n        <directionalLight\n          position={[3.5, -2, 5]}\n          intensity={2.2 * intensity}\n          castShadow={castShadow}\n        />\n        <directionalLight position={[-2, 1.5, 2.5]} intensity={0.25 * intensity} />\n      </>\n    );\n  }\n\n  if (preset === 'low-light') {\n    return (\n      <>\n        <ambientLight intensity={0.08 * intensity} />\n        <directionalLight\n          position={[2, -2, 3]}\n          intensity={0.75 * intensity}\n          castShadow={castShadow}\n        />\n        <pointLight position={[-0.5, -0.8, 1.3]} intensity={0.6 * intensity} />\n      </>\n    );\n  }\n\n  if (preset === 'splat') {\n    return (\n      <>\n        <ambientLight intensity={0.42 * intensity} />\n        <directionalLight\n          position={[1.8, -2.4, 3.5]}\n          intensity={1.2 * intensity}\n          castShadow={castShadow}\n        />\n        <pointLight position={[0.4, 0.2, 1.4]} intensity={0.35 * intensity} />\n      </>\n    );\n  }\n\n  return (\n    <>\n      <ambientLight intensity={0.35 * intensity} />\n      <directionalLight\n        position={[2.5, -3, 4]}\n        intensity={1.6 * intensity}\n        castShadow={castShadow}\n      />\n    </>\n  );\n}\n\nexport function getScenarioBackground(\n  preset: ScenarioLightingPreset | undefined,\n  fallback = DEFAULT_BACKGROUND\n) {\n  if (preset === 'warehouse') return '#20242b';\n  if (preset === 'low-light') return '#0f1115';\n  if (preset === 'splat') return '#1b1f24';\n  return fallback;\n}\n\nexport function getScenarioCameraPosition(\n  basePosition: readonly [number, number, number],\n  scenario?: Pick<VisualScenarioConfig, 'camera'>\n): [number, number, number] {\n  const [x, y, z] = basePosition;\n  const jitter = scenario?.camera?.jitter ?? 0;\n\n  return [\n    Number((x + jitter * 0.6).toFixed(3)),\n    Number((y - jitter * 0.4).toFixed(3)),\n    Number((z + jitter * 0.25).toFixed(3)),\n  ];\n}\n\nexport function VisualScenarioEffects(props: VisualScenarioEffectsProps) {\n  useVisualScenarioEffects(props);\n  return null;\n}\n\nexport function useVisualScenarioEffects({\n  scenario,\n  enabled = true,\n  applyBackground = true,\n  applyFog = true,\n  applyRenderer = true,\n  applyMaterials = true,\n  background,\n  fogNear,\n  fogFar,\n  materialFilter,\n}: VisualScenarioEffectsProps) {\n  const { gl, scene, invalidate } = useThree();\n\n  useEffect(() => {\n    if (!enabled || !scenario) {\n      return undefined;\n    }\n\n    const previousExposure = gl.toneMappingExposure;\n    const previousBackground = scene.background;\n    const previousFog = scene.fog;\n    const materialSnapshots = new Map<\n      THREE.Material,\n      {\n        color?: THREE.Color;\n        roughness?: number;\n        metalness?: number;\n      }\n    >();\n\n    if (applyRenderer) {\n      gl.toneMappingExposure = scenario.camera?.exposure ?? 1;\n    }\n\n    if (applyBackground) {\n      scene.background = new THREE.Color(\n        background ?? getScenarioBackground(scenario.lighting)\n      );\n    }\n\n    if (applyFog) {\n      scene.fog = createScenarioFog(scenario, background, fogNear, fogFar);\n    }\n\n    if (applyMaterials && scenario.materials) {\n      applyScenarioMaterials(scene, scenario, materialSnapshots, materialFilter);\n    }\n\n    invalidate();\n\n    return () => {\n      gl.toneMappingExposure = previousExposure;\n      scene.background = previousBackground;\n      scene.fog = previousFog;\n\n      for (const [material, snapshot] of materialSnapshots) {\n        const mutable = getMutableScenarioMaterial(material);\n        if (!mutable) continue;\n        if (snapshot.color) mutable.color.copy(snapshot.color);\n        if (typeof snapshot.roughness === 'number') {\n          mutable.roughness = snapshot.roughness;\n        }\n        if (typeof snapshot.metalness === 'number') {\n          mutable.metalness = snapshot.metalness;\n        }\n        mutable.needsUpdate = true;\n      }\n\n      invalidate();\n    };\n  }, [\n    applyBackground,\n    applyFog,\n    applyMaterials,\n    applyRenderer,\n    background,\n    enabled,\n    fogFar,\n    fogNear,\n    gl,\n    invalidate,\n    materialFilter,\n    scenario,\n    scene,\n  ]);\n}\n\n/**\n * Renderer-agnostic Gaussian splat environment boundary.\n *\n * This component intentionally does not import a specific 3DGS renderer. Pass a\n * Spark/GaussianSplats3D object as `children` once the app chooses a renderer,\n * and pass MuJoCo/MJCF collision proxy visuals via `collisionProxy`.\n */\nexport function SplatEnvironment({\n  environment,\n  scenario,\n  renderer,\n  src,\n  format,\n  collisionProxy,\n  collisionProxyMetadata,\n  children,\n  showPlaceholder = true,\n  ...groupProps\n}: SplatEnvironmentProps) {\n  const metadata = useSplatEnvironment({\n    environment,\n    scenario,\n    renderer,\n    src,\n    format,\n    collisionProxy: collisionProxyMetadata,\n  });\n  const existingUserData =\n    typeof groupProps.userData === 'object' && groupProps.userData !== null\n      ? groupProps.userData\n      : {};\n\n  return (\n    <group\n      {...groupProps}\n      userData={{\n        ...existingUserData,\n        ...metadata.userData,\n      }}\n    >\n      {children}\n      {children || !showPlaceholder ? null : <SplatPlaceholder />}\n      {collisionProxy}\n    </group>\n  );\n}\n\nexport function useSplatEnvironment({\n  environment,\n  scenario,\n  renderer,\n  src,\n  format,\n  collisionProxy,\n}: SplatEnvironmentMetadataInput): SplatEnvironmentMetadata {\n  const scenarioEnvironment = useMemo(\n    () =>\n      environment ??\n      (scenario\n        ? createPairedSplatEnvironment(scenario, { renderer })\n        : undefined),\n    [environment, renderer, scenario]\n  );\n  const resolvedSrc = src ?? scenarioEnvironment?.splat.src ?? scenario?.splat?.src;\n  const resolvedFormat =\n    format ??\n    scenarioEnvironment?.splat.format ??\n    scenario?.splat?.format ??\n    'spz';\n  const resolvedCollisionProxy =\n    collisionProxy ??\n    scenarioEnvironment?.collisionProxy ??\n    scenario?.splat?.collisionProxy ??\n    undefined;\n\n  return useMemo(\n    () => ({\n      src: resolvedSrc,\n      format: resolvedFormat,\n      collisionProxy: resolvedCollisionProxy,\n      userData: createSplatEnvironmentUserData({\n        environment: scenarioEnvironment,\n        src: resolvedSrc,\n        format: resolvedFormat,\n        collisionProxy: resolvedCollisionProxy,\n      }),\n    }),\n    [scenarioEnvironment, resolvedSrc, resolvedFormat, resolvedCollisionProxy]\n  );\n}\n\n/**\n * Convert a generic visual scenario splat block into a paired visual/physics\n * environment config. Returns undefined until both the splat asset and MJCF\n * collision proxy are present.\n */\nexport function createPairedSplatEnvironment(\n  scenario: Pick<VisualScenarioConfig, 'id' | 'label' | 'environment' | 'splat'>,\n  options: {\n    id?: string;\n    label?: string;\n    description?: string;\n    renderer?: SplatRendererKind;\n  } = {}\n): PairedSplatEnvironmentConfig | undefined {\n  const splat = scenario.splat;\n  const collisionProxy = splat?.collisionProxy;\n\n  if (!splat?.enabled || !splat.src || !collisionProxy?.xmlPath) {\n    return undefined;\n  }\n\n  return {\n    id: options.id ?? scenario.id ?? 'splat-environment',\n    label: options.label ?? scenario.label ?? 'Gaussian splat environment',\n    description:\n      options.description ??\n      (scenario.environment\n        ? `Visual ${scenario.environment} splat paired with MJCF collision proxy.`\n        : undefined),\n    splat: {\n      src: splat.src,\n      format: splat.format ?? 'spz',\n      renderer: options.renderer,\n    },\n    collisionProxy: {\n      ...collisionProxy,\n      xmlPath: collisionProxy.xmlPath,\n    },\n  };\n}\n\nfunction isPairedSplatEnvironment(input: SplatSceneInput): input is PairedSplatEnvironmentConfig {\n  return !!input && 'collisionProxy' in input && 'splat' in input;\n}\n\nfunction sceneRelativePath(sceneConfig: SceneConfig, path: string): string {\n  const src = sceneConfig.src;\n  if (!src) return path;\n\n  const base = src.endsWith('/') ? src : src + '/';\n  if (path.startsWith(base)) return path.slice(base.length);\n  return path;\n}\n\nfunction uniquePaths(paths: readonly string[]): string[] {\n  const seen = new Set<string>();\n  const result: string[] = [];\n  for (const path of paths) {\n    if (seen.has(path)) continue;\n    seen.add(path);\n    result.push(path);\n  }\n  return result;\n}\n\n/**\n * Compose a MuJoCo scene config with a paired splat collision proxy.\n *\n * This keeps the common hybrid setup declarative:\n * robot XML remains `sceneFile`, the `.spz` remains a visual-only layer, and\n * the paired MJCF collision proxy is added to `environmentFiles`.\n */\nexport function withSplatEnvironment(\n  sceneConfig: SceneConfig,\n  input: SplatSceneInput,\n  options: { renderer?: SplatRendererKind } = {}\n): SceneConfig {\n  const environment = isPairedSplatEnvironment(input)\n    ? input\n    : input\n      ? createPairedSplatEnvironment(input, options)\n      : undefined;\n  const xmlPath = environment?.collisionProxy.xmlPath;\n  if (!xmlPath) return sceneConfig;\n\n  return {\n    ...sceneConfig,\n    environmentFiles: uniquePaths([\n      ...(sceneConfig.environmentFiles ?? []),\n      sceneRelativePath(sceneConfig, xmlPath),\n    ]),\n  };\n}\n\nexport function createSplatEnvironmentUserData({\n  environment,\n  src,\n  format = 'spz',\n  collisionProxy,\n}: {\n  environment?: PairedSplatEnvironmentConfig;\n  src?: string;\n  format?: SplatFormat;\n  collisionProxy?: SplatCollisionProxyConfig;\n}) {\n  return {\n    role: 'splat-environment',\n    environmentId: environment?.id,\n    environmentLabel: environment?.label,\n    splatSrc: src,\n    splatFormat: format,\n    splatRenderer: environment?.splat.renderer,\n    collisionProxyStatus: collisionProxy?.status ?? 'missing',\n    collisionProxyXmlPath: collisionProxy?.xmlPath,\n    collisionProxyPrimitives: collisionProxy?.primitives ?? [],\n  };\n}\n\nexport function createSparkSplatViewerUrl({\n  viewerUrl,\n  splatSrc,\n}: {\n  viewerUrl: string;\n  splatSrc: string;\n}) {\n  const url = new URL(viewerUrl, 'http://mujoco-react.local');\n  url.searchParams.set('splat', splatSrc);\n  return viewerUrl.startsWith('http') ? url.toString() : `${url.pathname}${url.search}`;\n}\n\nfunction SplatPlaceholder() {\n  return (\n    <group>\n      <mesh position={[0, 0, 1.2]}>\n        <boxGeometry args={[2.4, 2.4, 2.4]} />\n        <meshBasicMaterial\n          color=\"#8b8b8b\"\n          transparent\n          opacity={0.06}\n          wireframe\n          side={THREE.DoubleSide}\n        />\n      </mesh>\n    </group>\n  );\n}\n\nfunction createScenarioFog(\n  scenario: VisualScenarioConfig,\n  background: THREE.ColorRepresentation | undefined,\n  fogNear: number | undefined,\n  fogFar: number | undefined\n) {\n  if (scenario.lighting === 'low-light') {\n    return new THREE.Fog(\n      background ?? getScenarioBackground(scenario.lighting),\n      fogNear ?? 2.5,\n      fogFar ?? 9\n    );\n  }\n\n  if (scenario.lighting === 'warehouse') {\n    return new THREE.Fog(\n      background ?? getScenarioBackground(scenario.lighting),\n      fogNear ?? 5,\n      fogFar ?? 16\n    );\n  }\n\n  return null;\n}\n\nfunction applyScenarioMaterials(\n  scene: THREE.Scene,\n  scenario: VisualScenarioConfig,\n  snapshots: Map<\n    THREE.Material,\n    {\n      color?: THREE.Color;\n      roughness?: number;\n      metalness?: number;\n    }\n  >,\n  materialFilter: VisualScenarioEffectsProps['materialFilter']\n) {\n  const materials = scenario.materials;\n  if (!materials) return;\n\n  scene.traverse((object) => {\n    if (!(object instanceof THREE.Mesh)) {\n      return;\n    }\n\n    for (const material of normalizeMaterials(object.material)) {\n      const mutable = getMutableScenarioMaterial(material);\n      if (!mutable) continue;\n      if (materialFilter && !materialFilter({ object, material })) continue;\n\n      if (!snapshots.has(material)) {\n        snapshots.set(material, {\n          color: mutable.color.clone(),\n          roughness: mutable.roughness,\n          metalness: mutable.metalness,\n        });\n      }\n\n      applyScenarioMaterial(mutable, object, scenario, materials);\n    }\n  });\n}\n\nfunction applyScenarioMaterial(\n  material: THREE.MeshStandardMaterial | THREE.MeshPhysicalMaterial,\n  object: THREE.Object3D,\n  scenario: VisualScenarioConfig,\n  materials: ScenarioMaterialConfig\n) {\n  const seed = scenario.seed ?? 0;\n  const objectKey = `${scenario.id ?? 'scenario'}:${object.name}:${material.name}:${seed}`;\n  const variation = hashToUnitInterval(objectKey);\n\n  if (materials.randomizeObjectColors) {\n    material.color.setHSL(variation, 0.38, 0.42);\n  }\n\n  if (materials.randomizeTableMaterial) {\n    material.roughness = clamp01(\n      materials.roughness ?? 0.35 + variation * 0.45\n    );\n    material.metalness = clamp01(\n      materials.metalness ?? variation * 0.12\n    );\n  }\n\n  material.needsUpdate = true;\n}\n\nfunction normalizeMaterials(\n  material: THREE.Material | THREE.Material[]\n): THREE.Material[] {\n  return Array.isArray(material) ? material : [material];\n}\n\nfunction getMutableScenarioMaterial(\n  material: THREE.Material\n): THREE.MeshStandardMaterial | THREE.MeshPhysicalMaterial | null {\n  if (\n    material instanceof THREE.MeshStandardMaterial ||\n    material instanceof THREE.MeshPhysicalMaterial\n  ) {\n    return material;\n  }\n\n  return null;\n}\n\nfunction hashToUnitInterval(value: string) {\n  let hash = 2166136261;\n  for (let index = 0; index < value.length; index += 1) {\n    hash ^= value.charCodeAt(index);\n    hash = Math.imul(hash, 16777619);\n  }\n  return (hash >>> 0) / 4294967295;\n}\n\nfunction clamp01(value: number) {\n  return Math.max(0, Math.min(1, value));\n}\n\nexport type SplatCollisionProxy = ReactNode | ThreeElements['group'];\n"]}

package/dist/index.d.ts

@@ -1,8 +1,7 @@
 import * as react_jsx_runtime from 'react/jsx-runtime';
-import { M as MujocoContextValue, a as MujocoCanvasProps, b as MujocoSimAPI, S as SceneConfig, c as MujocoModule, P as PhysicsStepCallback, d as MujocoModel, e as MujocoData, C as ControlGroupInfo, A as ActuatedJointInfo, f as ControlGroupSelector, O as ObservationConfig, g as ObservationResult, I as IkConfig, h as IkContextValue, B as BodyProps, i as IkGizmoProps, D as DragInteractionProps, j as SceneLightsProps, k as ScenarioLightingProps, l as SplatEnvironmentProps, V as VisualScenarioEffectsProps, m as VisualScenarioConfig, n as SplatRendererKind, o as PairedSplatEnvironmentConfig, p as SplatFormat, q as SplatCollisionProxyConfig, r as SplatCollisionPrimitive, s as ScenarioLightingPreset, t as SplatEnvironmentMetadataInput, u as SplatEnvironmentMetadata, v as SplatSceneInput, w as DebugProps, G as GeomInfo, x as ContactListenerProps, T as TrajectoryPlayerProps, y as ActuatorInfo, z as Sites, E as SitePositionResult, F as Sensors, H as SensorHandle, J as SensorInfo, K as Joints, L as JointStateResult, N as Bodies, Q as BodyStateResult, R as Actuators, U as CtrlHandle, W as ContactInfo, X as KeyboardTeleopConfig, Y as PolicyConfig, Z as PolicyVector, _ as ObservationHandle, $ as TrajectoryInput, a0 as PlaybackState, a1 as TrajectoryFrame } from './types-BmneHLBM.js';
-export { a2 as BodyInfo, a3 as ControlJointInfo, a4 as Geoms, a5 as IKSolveFn, a6 as JointInfo, a7 as KeyBinding, a8 as Keyframes, a9 as ModelOptions, aa as MujocoContact, ab as MujocoContactArray, ac as ObservationLayoutItem, ad as ObservationOutput, ae as PhysicsConfig, af as PolicyActionInput, ag as PolicyInferenceInput, ah as PolicyObservationInput, ai as RayHit, aj as Register, ak as RegisteredRobotMap, al as ResourceSelector, am as RobotActuators, an as RobotBodies, ao as RobotGeoms, ap as RobotJoints, aq as RobotKeyframes, ar as RobotResource, as as RobotResources, at as RobotSensors, au as RobotSites, av as Robots, aw as ScenarioCameraConfig, ax as ScenarioMaterialConfig, ay as SceneMarker, az as SceneObject, aA as SensorResult, aB as SiteInfo, aC as SplatAssetConfig, aD as SplatScenarioConfig, aE as StateSnapshot, aF as TrajectoryData, aG as XmlPatch, aH as getContact, aI as registerRobotResources } from './types-BmneHLBM.js';
+import { M as MujocoContextValue, a as MujocoCanvasProps, b as MujocoSimAPI, S as SceneConfig, R as ReadyCallbackInput, c as StepCallbackInput, d as SelectionCallbackInput, e as MujocoModule, P as PhysicsStepCallback, f as MujocoModel, g as MujocoData, C as ControlGroupInfo, A as ActuatedJointInfo, h as ControlGroupSelector, O as ObservationConfig, i as ObservationResult, I as IkConfig, j as IkContextValue, B as BodyProps, k as IkGizmoProps, D as DragInteractionProps, l as SceneLightsProps, m as ScenarioLightingProps, n as SplatEnvironmentProps, V as VisualScenarioEffectsProps, o as VisualScenarioConfig, p as SplatRendererKind, q as PairedSplatEnvironmentConfig, r as SplatFormat, s as SplatCollisionProxyConfig, t as SplatCollisionPrimitive, u as ScenarioLightingPreset, v as SplatEnvironmentMetadataInput, w as SplatEnvironmentMetadata, x as SplatSceneInput, y as DebugProps, G as GeomInfo, z as ContactListenerProps, T as TrajectoryPlayerProps, E as ActuatorInfo, F as Sites, H as SitePositionResult, J as Sensors, K as SensorHandle, L as SensorInfo, N as Joints, Q as JointStateResult, U as Bodies, W as BodyStateResult, X as Actuators, Y as CtrlHandle, Z as ContactInfo, _ as KeyboardTeleopConfig, $ as PolicyConfig, a0 as PolicyVector, a1 as ObservationHandle, a2 as TrajectoryInput, a3 as TrajectoryStateChangeInput, a4 as PlaybackState, a5 as TrajectoryFrame, a6 as FrameCaptureOptions, a7 as FrameCaptureResult, a8 as FrameCaptureBlobResult, a9 as FrameCaptureAPI } from './types-C5gTvR7b.js';
+export { aa as BodyInfo, ab as ControlJointInfo, ac as FrameCaptureStatus, ad as FrameCaptureTarget, ae as FrameCaptureTargetRef, af as Geoms, ag as IKSolveFn, ah as IkGizmoDragInput, ai as IkSolveInput, aj as JointInfo, ak as KeyBinding, al as Keyframes, am as ModelOptions, an as MujocoContact, ao as MujocoContactArray, ap as MujocoFrameCaptureOptions, aq as ObservationLayoutItem, ar as ObservationOutput, as as PhysicsConfig, at as PhysicsStepInput, au as PolicyActionInput, av as PolicyInferenceInput, aw as PolicyObservationInput, ax as RayHit, ay as Register, az as RegisteredRobotMap, aA as ResetCallbackInput, aB as ResourceSelector, aC as RobotActuators, aD as RobotBodies, aE as RobotGeoms, aF as RobotJoints, aG as RobotKeyframes, aH as RobotResource, aI as RobotResources, aJ as RobotSensors, aK as RobotSites, aL as Robots, aM as ScenarioCameraConfig, aN as ScenarioMaterialConfig, aO as SceneMarker, aP as SceneObject, aQ as SensorResult, aR as SiteInfo, aS as SplatAssetConfig, aT as SplatScenarioConfig, aU as StateSnapshot, aV as TrajectoryData, aW as TrajectoryFrameCallbackInput, aX as VisualScenarioMaterialFilterInput, aY as XmlPatch, aZ as getContact, a_ as registerRobotResources } from './types-C5gTvR7b.js';
 import * as React$1 from 'react';
-import React__default from 'react';
 import { ThreeElements } from '@react-three/fiber';
 import * as THREE from 'three';
 
@@ -56,13 +55,13 @@ interface MujocoPhysicsProps {
     /** Scene/robot configuration. */
     config: SceneConfig;
     /** Fires when model is loaded and API is ready. */
-    onReady?: (api: MujocoSimAPI) => void;
+    onReady?: (input: ReadyCallbackInput) => void;
     /** Fires on scene load failure. */
     onError?: (error: Error) => void;
     /** Called each physics step. */
-    onStep?: (time: number) => void;
+    onStep?: (input: StepCallbackInput) => void;
     /** Called on body double-click selection. */
-    onSelection?: (bodyId: number, name: string) => void;
+    onSelection?: (input: SelectionCallbackInput) => void;
     /** Override model gravity. */
     gravity?: [number, number, number];
     /** Override model.opt.timestep. */
@@ -133,10 +132,10 @@ interface MujocoSimProviderProps {
     mujoco: MujocoModule;
     config: SceneConfig;
     apiRef?: React.ForwardedRef<MujocoSimAPI>;
-    onReady?: (api: MujocoSimAPI) => void;
+    onReady?: (input: ReadyCallbackInput) => void;
     onError?: (error: Error) => void;
-    onStep?: (time: number) => void;
-    onSelection?: (bodyId: number, name: string) => void;
+    onStep?: (input: StepCallbackInput) => void;
+    onSelection?: (input: SelectionCallbackInput) => void;
     gravity?: [number, number, number];
     timestep?: number;
     substeps?: number;
@@ -243,7 +242,7 @@ type ControllerComponent<TConfig> = React.FC<{
  * const MyController = createController<{ speed: number }>(
  *   { name: 'my-controller', defaultConfig: { speed: 1.0 } },
  *   function MyControllerImpl({ config }) {
- *     useBeforePhysicsStep((_model, data) => {
+ *     useBeforePhysicsStep(({ data }) => {
  *       data.ctrl[0] = config.speed;
  *     });
  *     return null;
@@ -270,7 +269,7 @@ declare function createController<TConfig>(options: ControllerOptions<TConfig>,
  *   { name: 'useMyController', defaultConfig: { gain: 1.0 } },
  *   function useMyControllerImpl(config) {
  *     // config is MyConfig | null — hooks must be called unconditionally
- *     useBeforePhysicsStep((_model, data) => {
+ *     useBeforePhysicsStep(({ data }) => {
  *       if (!config) return;
  *       data.ctrl[0] = config.gain * Math.sin(data.time);
  *     });
@@ -312,7 +311,7 @@ declare function Body({ name, type, size, position, rgba, mass, freejoint, frict
  * - `controller` — IkContextValue from `useIkController()`.
  * - `siteName` — MuJoCo site to track. Defaults to the controller's configured site.
  * - `scale` — Gizmo handle scale. Default: 0.18.
- * - `onDrag` — Custom drag callback `(pos, quat) => void`.
+ * - `onDrag` — Custom drag callback `({ position, quaternion }) => void`.
  *   When omitted, dragging enables IK and writes to the IK target.
  *   When provided, the consumer handles what happens during drag.
  */
@@ -655,7 +654,7 @@ interface TrajectoryPlayerOptions {
     loop?: boolean;
     mode?: 'kinematic' | 'physics';
     onComplete?: () => void;
-    onStateChange?: (state: PlaybackState) => void;
+    onStateChange?: (input: TrajectoryStateChangeInput) => void;
 }
 /**
  * Play back a trajectory, overriding simulation state.
@@ -758,33 +757,6 @@ declare function useVideoRecorder(options?: VideoRecorderOptions): {
  * useFrameCapture — still-frame capture for canvas-backed MuJoCo/R3F scenes.
  */
 
-type FrameCaptureStatus = 'idle' | 'capturing' | 'captured' | 'error';
-type FrameCaptureTarget = HTMLCanvasElement | HTMLElement | null | undefined;
-type FrameCaptureTargetRef = React__default.RefObject<HTMLCanvasElement | HTMLElement | null>;
-interface FrameCaptureOptions {
-    target?: FrameCaptureTarget | FrameCaptureTargetRef;
-    type?: string;
-    quality?: number;
-    waitForAnimationFrame?: boolean;
-}
-interface FrameCaptureResult {
-    canvas: HTMLCanvasElement;
-    dataUrl: string;
-    type: string;
-}
-interface FrameCaptureBlobResult {
-    canvas: HTMLCanvasElement;
-    blob: Blob;
-    type: string;
-}
-interface FrameCaptureAPI {
-    status: FrameCaptureStatus;
-    error: Error | null;
-    isCapturing: boolean;
-    capture: (options?: FrameCaptureOptions) => Promise<FrameCaptureResult>;
-    captureBlob: (options?: FrameCaptureOptions) => Promise<FrameCaptureBlobResult>;
-    reset: () => void;
-}
 /**
  * Capture the current canvas frame as a data URL.
  *
@@ -896,4 +868,4 @@ interface CameraAnimationAPI {
  */
 declare function useCameraAnimation(): CameraAnimationAPI;
 
-export { ActuatedJointInfo, ActuatorInfo, Actuators, Bodies, Body, BodyProps, BodyStateResult, type CameraAnimationAPI, ContactInfo, ContactListener, ContactListenerProps, ContactMarkers, ControlGroupInfo, ControlGroupSelector, type ControllerComponent, type ControllerOptions, CtrlHandle, Debug, DebugProps, DragInteraction, DragInteractionProps, FlexRenderer, type FrameCaptureAPI, type FrameCaptureBlobResult, type FrameCaptureOptions, type FrameCaptureResult, type FrameCaptureStatus, type FrameCaptureTarget, type FrameCaptureTargetRef, GeomInfo, IkConfig, IkContextValue, IkGizmo, IkGizmoProps, InstancedGeomRenderer, JointStateResult, Joints, KeyboardTeleopConfig, MujocoCanvas, MujocoCanvasProps, MujocoContextValue, MujocoData, type MujocoLoader, type MujocoLoaderOptions, MujocoModel, MujocoModule, MujocoPhysics, type MujocoPhysicsProps, MujocoProvider, type MujocoProviderProps, MujocoSimAPI, MujocoSimProvider, type MujocoWasmVariant, ObservationConfig, ObservationHandle, ObservationResult, PairedSplatEnvironmentConfig, PhysicsStepCallback, PlaybackState, PolicyConfig, PolicyVector, ScenarioLighting, ScenarioLightingPreset, ScenarioLightingProps, SceneConfig, SceneLights, SceneLightsProps, SensorHandle, SensorInfo, Sensors, SitePositionResult, Sites, SplatCollisionPrimitive, SplatCollisionProxyConfig, SplatEnvironment, SplatEnvironmentMetadata, SplatEnvironmentMetadataInput, SplatEnvironmentProps, SplatFormat, SplatRendererKind, SplatSceneInput, TendonRenderer, TrajectoryFrame, TrajectoryInput, TrajectoryPlayer, TrajectoryPlayerProps, VisualScenarioConfig, VisualScenarioEffects, VisualScenarioEffectsProps, buildObservation, captureFrame, captureFrameBlob, createContiguousControlGroup, createController, createControllerHook, createPairedSplatEnvironment, createSparkSplatViewerUrl, createSplatEnvironmentUserData, findActuatorByName, findBodyByName, findGeomByName, findJointByName, findKeyframeByName, findSensorByName, findSiteByName, findTendonByName, getActuatedJoints, getControlMap, getName, getScenarioBackground, getScenarioCameraPosition, loadScene, resolveControlGroup, useActuators, useAfterPhysicsStep, useBeforePhysicsStep, useBodyMeshes, useBodyState, useCameraAnimation, useContactEvents, useContacts, useCtrl, useCtrlNoise, useFrameCapture, useGamepad, useGravityCompensation, useIkController, useJointState, useKeyboardTeleop, useMujoco, useMujocoWasm, useObservation, usePolicy, useSceneLights, useSelectionHighlight, useSensor, useSensors, useSitePosition, useSplatEnvironment, useTrajectoryPlayer, useTrajectoryRecorder, useVideoRecorder, useVisualScenarioEffects, withSplatEnvironment };
+export { ActuatedJointInfo, ActuatorInfo, Actuators, Bodies, Body, BodyProps, BodyStateResult, type CameraAnimationAPI, ContactInfo, ContactListener, ContactListenerProps, ContactMarkers, ControlGroupInfo, ControlGroupSelector, type ControllerComponent, type ControllerOptions, CtrlHandle, Debug, DebugProps, DragInteraction, DragInteractionProps, FlexRenderer, FrameCaptureAPI, FrameCaptureBlobResult, FrameCaptureOptions, FrameCaptureResult, GeomInfo, IkConfig, IkContextValue, IkGizmo, IkGizmoProps, InstancedGeomRenderer, JointStateResult, Joints, KeyboardTeleopConfig, MujocoCanvas, MujocoCanvasProps, MujocoContextValue, MujocoData, type MujocoLoader, type MujocoLoaderOptions, MujocoModel, MujocoModule, MujocoPhysics, type MujocoPhysicsProps, MujocoProvider, type MujocoProviderProps, MujocoSimAPI, MujocoSimProvider, type MujocoWasmVariant, ObservationConfig, ObservationHandle, ObservationResult, PairedSplatEnvironmentConfig, PhysicsStepCallback, PlaybackState, PolicyConfig, PolicyVector, ReadyCallbackInput, ScenarioLighting, ScenarioLightingPreset, ScenarioLightingProps, SceneConfig, SceneLights, SceneLightsProps, SelectionCallbackInput, SensorHandle, SensorInfo, Sensors, SitePositionResult, Sites, SplatCollisionPrimitive, SplatCollisionProxyConfig, SplatEnvironment, SplatEnvironmentMetadata, SplatEnvironmentMetadataInput, SplatEnvironmentProps, SplatFormat, SplatRendererKind, SplatSceneInput, StepCallbackInput, TendonRenderer, TrajectoryFrame, TrajectoryInput, TrajectoryPlayer, TrajectoryPlayerProps, TrajectoryStateChangeInput, VisualScenarioConfig, VisualScenarioEffects, VisualScenarioEffectsProps, buildObservation, captureFrame, captureFrameBlob, createContiguousControlGroup, createController, createControllerHook, createPairedSplatEnvironment, createSparkSplatViewerUrl, createSplatEnvironmentUserData, findActuatorByName, findBodyByName, findGeomByName, findJointByName, findKeyframeByName, findSensorByName, findSiteByName, findTendonByName, getActuatedJoints, getControlMap, getName, getScenarioBackground, getScenarioCameraPosition, loadScene, resolveControlGroup, useActuators, useAfterPhysicsStep, useBeforePhysicsStep, useBodyMeshes, useBodyState, useCameraAnimation, useContactEvents, useContacts, useCtrl, useCtrlNoise, useFrameCapture, useGamepad, useGravityCompensation, useIkController, useJointState, useKeyboardTeleop, useMujoco, useMujocoWasm, useObservation, usePolicy, useSceneLights, useSelectionHighlight, useSensor, useSensors, useSitePosition, useSplatEnvironment, useTrajectoryPlayer, useTrajectoryRecorder, useVideoRecorder, useVisualScenarioEffects, withSplatEnvironment };