@react-three/rapier 0.1.0 → 0.2.0

package/dist/react-three-rapier.cjs.prod.js

@@ -5,8 +5,8 @@ Object.defineProperty(exports, '__esModule', { value: true });
 var React = require('react');
 var useAsset = require('use-asset');
 var fiber = require('@react-three/fiber');
-var rapier3dCompat = require('@dimforge/rapier3d-compat');
 var three = require('three');
+var rapier3dCompat = require('@dimforge/rapier3d-compat');
 
 function _interopDefault (e) { return e && e.__esModule ? e : { 'default': e }; }
 
@@ -63,7 +63,7 @@ const scaleColliderArgs = (shape, args, scale) => {
   const scaleArray = [scale.x, scale.y, scale.z];
   return newArgs.map((arg, index) => scaleArray[index] * arg);
 };
-const createColliderFromOptions = (options, world, body, scale = {
+const createColliderFromOptions = (options, world, rigidBodyHandle, scale = {
   x: 1,
   y: 1,
   z: 1
@@ -105,12 +105,13 @@ const createColliderFromOptions = (options, world, body, scale = {
     });
   }
 
-  const collider = world.createCollider(colliderDesc, body.handle);
+  const collider = world.createCollider(colliderDesc, rigidBodyHandle);
   return collider;
 };
 const createCollidersFromChildren = (object, rigidBody, type, world) => {
   const colliders = [];
   let desc;
+  let offset = new three.Vector3();
   object.traverse(child => {
     if ("isMesh" in child) {
       const {
@@ -137,6 +138,7 @@ const createCollidersFromChildren = (object, rigidBody, type, world) => {
               boundingBox
             } = geometry;
             const size = boundingBox.getSize(new three.Vector3());
+            boundingBox.getCenter(offset);
             desc = rapier3dCompat.ColliderDesc.cuboid(size.x / 2 * scale.x, size.y / 2 * scale.y, size.z / 2 * scale.z);
           }
           break;
@@ -148,6 +150,7 @@ const createCollidersFromChildren = (object, rigidBody, type, world) => {
               boundingSphere
             } = geometry;
             const radius = boundingSphere.radius * scale.x;
+            offset.copy(boundingSphere.center);
             desc = rapier3dCompat.ColliderDesc.ball(radius);
           }
           break;
@@ -172,7 +175,7 @@ const createCollidersFromChildren = (object, rigidBody, type, world) => {
 
 
       const parentWorldScale = child.parent.getWorldScale(new three.Vector3());
-      desc.setTranslation(x * parentWorldScale.x, y * parentWorldScale.y, z * parentWorldScale.z).setRotation({
+      desc.setTranslation((x + offset.x) * parentWorldScale.x, (y + offset.y) * parentWorldScale.y, (z + offset.z) * parentWorldScale.z).setRotation({
         x: rx,
         y: ry,
         z: rz,
@@ -196,6 +199,99 @@ const scaleVertices = (vertices, scale) => {
   return scaledVerts;
 };
 
+// TODO: Flesh this out
+const createRigidBodyApi = ref => {
+  return {
+    raw: () => ref.current(),
+
+    get handle() {
+      return ref.current().handle;
+    },
+
+    applyImpulse({
+      x,
+      y,
+      z
+    }) {
+      ref.current().applyImpulse({
+        x,
+        y,
+        z
+      }, true);
+    },
+
+    applyTorqueImpulse({
+      x,
+      y,
+      z
+    }) {
+      ref.current().applyTorqueImpulse({
+        x,
+        y,
+        z
+      }, true);
+    },
+
+    translation() {
+      const {
+        x,
+        y,
+        z
+      } = ref.current().translation();
+      return new three.Vector3(x, y, z);
+    },
+
+    rotation() {
+      const {
+        x,
+        y,
+        z,
+        w
+      } = ref.current().rotation();
+      return new three.Quaternion(x, y, z, w);
+    }
+
+  };
+}; // TODO: Flesh this out
+
+const createColliderApi = ref => {
+  return {
+    raw: () => ref.current(),
+
+    get handle() {
+      return ref.current().handle;
+    }
+
+  };
+};
+const createWorldApi = ref => {
+  return {
+    raw: () => ref.current(),
+    getCollider: handle => ref.current().getCollider(handle),
+    getRigidBody: handle => ref.current().getRigidBody(handle),
+    createRigidBody: desc => ref.current().createRigidBody(desc),
+    createCollider: (desc, rigidBodyHandle) => ref.current().createCollider(desc, rigidBodyHandle),
+    removeRigidBody: rigidBody => ref.current().removeRigidBody(rigidBody),
+    removeCollider: collider => ref.current().removeCollider(collider, true),
+    createImpulseJoint: (params, rigidBodyA, rigidBodyB) => ref.current().createImpulseJoint(params, rigidBodyA, rigidBodyB),
+    removeImpulseJoint: joint => ref.current().removeImpulseJoint(joint, true),
+    forEachCollider: callback => ref.current().forEachCollider(callback)
+  };
+}; // TODO: Broken currently, waiting for Rapier3D to fix
+
+const createJointApi = ref => {
+  return {
+    raw: () => ref.current(),
+
+    get handle() {
+      return ref.current().handle;
+    },
+
+    configureMotorPosition: (targetPos, stiffness, damping) => ref.current().configureMotorPosition(targetPos, stiffness, damping),
+    configureMotorVelocity: (targetVel, damping) => ref.current().configureMotorVelocity(targetVel, damping)
+  };
+};
+
 const RapierContext = /*#__PURE__*/React.createContext(undefined);
 
 const importRapier = async () => {
@@ -210,30 +306,81 @@ const Physics = ({
   children
 }) => {
   const rapier = useAsset.useAsset(importRapier);
-  const stepFuncs = React.useRef([]);
-  const world = React.useMemo(() => {
-    if (!rapier.World) return null;
-    let world = new rapier.World(vectorArrayToObject(_gravity));
-    return world;
-  }, [rapier]);
+  const worldRef = React.useRef();
+  const getWorldRef = React.useRef(() => {
+    if (!worldRef.current) {
+      const world = new rapier.World(vectorArrayToObject(_gravity));
+      worldRef.current = world;
+    }
+
+    return worldRef.current;
+  });
+  const [colliderMeshes] = React.useState(() => new Map());
+  const [rigidBodyMeshes] = React.useState(() => new Map()); // Init world
+
+  React.useLayoutEffect(() => {
+    const world = getWorldRef.current();
+    return () => {
+      if (world) {
+        world.free();
+        worldRef.current = undefined;
+      }
+    };
+  }, []);
   const time = React.useRef(performance.now());
   fiber.useFrame(context => {
-    // Set timestep to current delta, to allow for variable frame rates
+    const world = worldRef.current;
+    if (!world) return; // Set timestep to current delta, to allow for variable frame rates
     // We cap the delta at 100, so that the physics simulation doesn't get wild
+
     const now = performance.now();
     const delta = Math.min(100, now - time.current);
     world.timestep = delta / 1000;
-    world.step(); // Run all step funcs
+    world.step(); // Update meshes
 
-    stepFuncs.current.forEach(func => func());
+    rigidBodyMeshes.forEach((mesh, handle) => {
+      const rigidBody = world.getRigidBody(handle);
+
+      if (!rigidBody || rigidBody.isSleeping() || rigidBody.isFixed() || !mesh.parent) {
+        return;
+      }
+
+      const {
+        x,
+        y,
+        z
+      } = rigidBody.translation();
+      const {
+        x: rx,
+        y: ry,
+        z: rz,
+        w: rw
+      } = rigidBody.rotation();
+      const scale = mesh.getWorldScale(new three.Vector3()); // haha matrixes I have no idea what I'm doing :)
+
+      const o = new three.Object3D();
+      o.position.set(x, y, z);
+      o.rotation.setFromQuaternion(new three.Quaternion(rx, ry, rz, rw));
+      o.scale.set(scale.x, scale.y, scale.z);
+      o.updateMatrix();
+      o.applyMatrix4(mesh.parent.matrixWorld.clone().invert());
+      o.updateMatrix();
+      mesh.position.setFromMatrixPosition(o.matrix);
+      mesh.rotation.setFromRotationMatrix(o.matrix);
+    });
     time.current = now;
   });
+  const api = React.useMemo(() => createWorldApi(getWorldRef), []);
   const context = React.useMemo(() => ({
-    RAPIER: rapier,
-    world,
-    colliders: _colliders,
-    stepFuncs: stepFuncs.current
-  }), [rapier]);
+    rapier,
+    world: api,
+    physicsOptions: {
+      colliders: _colliders,
+      gravity: _gravity
+    },
+    colliderMeshes,
+    rigidBodyMeshes
+  }), []);
   return /*#__PURE__*/React__default["default"].createElement(RapierContext.Provider, {
     value: context
   }, children);
@@ -282,64 +429,44 @@ function _objectSpread2(target) {
 
 const useRapier = () => {
   return React.useContext(RapierContext);
-}; // Private hook for updating the simulations on objects
-
-const useRapierStep = callback => {
-  const {
-    stepFuncs
-  } = useRapier();
-  React.useEffect(() => {
-    stepFuncs.push(callback);
-    return () => {
-      const index = stepFuncs.indexOf(callback);
-      stepFuncs.splice(index, 1);
-    };
-  }, [callback]);
 };
-const useCollider = (body, options = {}) => {
+const useRigidBody = (options = {}) => {
   const {
-    RAPIER,
-    world
-  } = useRapier();
-  const collider = React.useMemo(() => {
-    return createColliderFromOptions(options, world, body);
-  }, []);
-  React.useEffect(() => {
-    return () => {
-      world.removeCollider(collider, false);
-    };
-  }, []);
-  return [collider];
-};
-const useRigidBody = options => {
-  const {
-    RAPIER,
-    world
+    rapier,
+    world,
+    rigidBodyMeshes,
+    physicsOptions
   } = useRapier();
   const ref = React.useRef(); // Create rigidbody
 
-  const rigidBody = React.useMemo(() => {
-    var _options$linearVeloci, _options$angularVeloc, _options$gravityScale, _options$canSleep, _options$ccd;
-
-    const [lvx, lvy, lvz] = (_options$linearVeloci = options === null || options === void 0 ? void 0 : options.linearVelocity) !== null && _options$linearVeloci !== void 0 ? _options$linearVeloci : [0, 0, 0];
-    const [avx, avy, avz] = (_options$angularVeloc = options === null || options === void 0 ? void 0 : options.angularVelocity) !== null && _options$angularVeloc !== void 0 ? _options$angularVeloc : [0, 0, 0];
-    const gravityScale = (_options$gravityScale = options === null || options === void 0 ? void 0 : options.gravityScale) !== null && _options$gravityScale !== void 0 ? _options$gravityScale : 1;
-    const canSleep = (_options$canSleep = options === null || options === void 0 ? void 0 : options.canSleep) !== null && _options$canSleep !== void 0 ? _options$canSleep : true;
-    const ccdEnabled = (_options$ccd = options === null || options === void 0 ? void 0 : options.ccd) !== null && _options$ccd !== void 0 ? _options$ccd : false;
-    const type = rigidBodyTypeFromString((options === null || options === void 0 ? void 0 : options.type) || "dynamic");
-    (options === null || options === void 0 ? void 0 : options.position) || [0, 0, 0];
-    (options === null || options === void 0 ? void 0 : options.rotation) || [0, 0, 0];
-    const rigidBodyDesc = new RAPIER.RigidBodyDesc(type).setLinvel(lvx, lvy, lvz).setAngvel({
-      x: avx,
-      y: avy,
-      z: avz
-    }).setGravityScale(gravityScale).setCanSleep(canSleep).setCcdEnabled(ccdEnabled).setTranslation(0, 0, 0);
-    const body = world.createRigidBody(rigidBodyDesc);
-    return body;
-  }, []); // Setup
+  const rigidBodyRef = React.useRef();
+  const getRigidBodyRef = React.useRef(() => {
+    if (!rigidBodyRef.current) {
+      var _options$linearVeloci, _options$angularVeloc, _options$gravityScale, _options$canSleep, _options$ccd;
+
+      const type = rigidBodyTypeFromString((options === null || options === void 0 ? void 0 : options.type) || "dynamic");
+      const [lvx, lvy, lvz] = (_options$linearVeloci = options === null || options === void 0 ? void 0 : options.linearVelocity) !== null && _options$linearVeloci !== void 0 ? _options$linearVeloci : [0, 0, 0];
+      const [avx, avy, avz] = (_options$angularVeloc = options === null || options === void 0 ? void 0 : options.angularVelocity) !== null && _options$angularVeloc !== void 0 ? _options$angularVeloc : [0, 0, 0];
+      const gravityScale = (_options$gravityScale = options === null || options === void 0 ? void 0 : options.gravityScale) !== null && _options$gravityScale !== void 0 ? _options$gravityScale : 1;
+      const canSleep = (_options$canSleep = options === null || options === void 0 ? void 0 : options.canSleep) !== null && _options$canSleep !== void 0 ? _options$canSleep : true;
+      const ccdEnabled = (_options$ccd = options === null || options === void 0 ? void 0 : options.ccd) !== null && _options$ccd !== void 0 ? _options$ccd : false;
+      const desc = new rapier.RigidBodyDesc(type).setLinvel(lvx, lvy, lvz).setAngvel({
+        x: avx,
+        y: avy,
+        z: avz
+      }).setGravityScale(gravityScale).setCanSleep(canSleep).setCcdEnabled(ccdEnabled);
+      const rigidBody = world.createRigidBody(desc);
+      rigidBodyRef.current = rigidBody;
+    }
+
+    return rigidBodyRef.current;
+  }); // Setup
 
   React.useEffect(() => {
-    var _ref$current$parent, _options$colliders;
+    var _ref$current$parent, _ref, _options$colliders;
+
+    const rigidBody = getRigidBodyRef.current();
+    rigidBodyRef.current = rigidBody;
 
     if (!ref.current) {
       ref.current = new three.Object3D();
@@ -372,43 +499,42 @@ const useRigidBody = options => {
     }, false);
     rigidBody.resetForces(false);
     rigidBody.resetTorques(false);
-    const colliderSetting = (_options$colliders = options === null || options === void 0 ? void 0 : options.colliders) !== null && _options$colliders !== void 0 ? _options$colliders : false;
+    const colliderSetting = (_ref = (_options$colliders = options === null || options === void 0 ? void 0 : options.colliders) !== null && _options$colliders !== void 0 ? _options$colliders : physicsOptions.colliders) !== null && _ref !== void 0 ? _ref : false;
     const autoColliders = colliderSetting !== false ? createCollidersFromChildren(ref.current, rigidBody, colliderSetting, world) : [];
+    rigidBody.wakeUp();
+    rigidBodyMeshes.set(rigidBody.handle, ref.current);
     return () => {
+      autoColliders.forEach(collider => world.removeCollider(collider));
       world.removeRigidBody(rigidBody);
-      autoColliders.forEach(collider => world.removeCollider(collider, false));
+      rigidBodyRef.current = undefined;
+      rigidBodyMeshes.delete(rigidBody.handle);
     };
   }, []);
-  useRapierStep(() => {
-    if (rigidBody && ref.current) {
-      const {
-        x,
-        y,
-        z
-      } = rigidBody.translation();
-      const {
-        x: rx,
-        y: ry,
-        z: rz,
-        w: rw
-      } = rigidBody.rotation();
-      const scale = ref.current.getWorldScale(new three.Vector3());
-
-      if (ref.current.parent) {
-        // haha matrixes I have no idea what I'm doing :)
-        const o = new three.Object3D();
-        o.position.set(x, y, z);
-        o.rotation.setFromQuaternion(new three.Quaternion(rx, ry, rz, rw));
-        o.scale.set(scale.x, scale.y, scale.z);
-        o.updateMatrix();
-        o.applyMatrix4(ref.current.parent.matrixWorld.clone().invert());
-        o.updateMatrix();
-        ref.current.position.setFromMatrixPosition(o.matrix);
-        ref.current.rotation.setFromRotationMatrix(o.matrix);
-      }
+  const api = React.useMemo(() => createRigidBodyApi(getRigidBodyRef), []);
+  return [ref, api];
+};
+const useCollider = (body, options = {}) => {
+  const {
+    world
+  } = useRapier();
+  const colliderRef = React.useRef();
+  const objectRef = React.useRef();
+  const getColliderRef = React.useRef(() => {
+    if (!colliderRef.current) {
+      colliderRef.current = createColliderFromOptions(options, world, body.handle);
     }
+
+    return colliderRef.current;
   });
-  return [ref, rigidBody];
+  React.useEffect(() => {
+    const collider = getColliderRef.current();
+    return () => {
+      if (collider) world.removeCollider(collider);
+      colliderRef.current = undefined;
+    };
+  }, []);
+  const api = React.useMemo(() => createColliderApi(getColliderRef), []);
+  return [objectRef, api];
 };
 const useRigidBodyWithCollider = (rigidBodyOptions, colliderOptions) => {
   const {
@@ -421,9 +547,9 @@ const useRigidBodyWithCollider = (rigidBodyOptions, colliderOptions) => {
     }
 
     const scale = ref.current.getWorldScale(new three.Vector3());
-    const collider = createColliderFromOptions(colliderOptions, world, rigidBody, scale);
+    const collider = createColliderFromOptions(colliderOptions, world, rigidBody.handle, scale);
     return () => {
-      world.removeCollider(collider, false);
+      world.removeCollider(collider);
     };
   }, []);
   return [ref, rigidBody];
@@ -549,20 +675,42 @@ const useRoundConvexMesh = (rigidBodyOptions = {}, colliderOptions = {}) => {
 
 const useImpulseJoint = (body1, body2, params) => {
   const {
-    world,
-    RAPIER
+    world
   } = useRapier();
-  React.useLayoutEffect(() => {
-    let joint;
+  const jointRef = React.useRef();
+  const getJointRef = React.useRef(() => {
+    if (!jointRef.current) {
+      let rb1;
+      let rb2;
+
+      if ('handle' in body1 && 'handle' in body2) {
+        rb1 = world.getRigidBody(body1.handle);
+        rb2 = world.getRigidBody(body2.handle);
+        jointRef.current = world.createImpulseJoint(params, rb1, rb2);
+      }
 
-    if (body1 && body2 && params) {
-      joint = world.createImpulseJoint(params, body1.current, body2.current);
+      if ('current' in body1 && body1.current && 'current' in body2 && body2.current) {
+        rb1 = world.getRigidBody(body1.current.handle);
+        rb2 = world.getRigidBody(body2.current.handle);
+        const newJoint = world.createImpulseJoint(params, rb1, rb2);
+        jointRef.current = newJoint;
+      }
     }
 
+    return jointRef.current;
+  });
+  React.useEffect(() => {
+    const joint = getJointRef.current();
     return () => {
-      if (joint) world.removeImpulseJoint(joint, true);
+      if (joint) {
+        console.log('remove joint', joint);
+        world.removeImpulseJoint(joint);
+        jointRef.current = undefined;
+      }
     };
-  }, [body1, body2]);
+  }, []);
+  const api = React.useMemo(() => createJointApi(getJointRef), []);
+  return api;
 };
 /**
  *
@@ -573,9 +721,9 @@ const useImpulseJoint = (body1, body2, params) => {
 
 const useFixedJoint = (body1, body2, [body1Anchor, body1LocalFrame, body2Anchor, body2LocalFrame]) => {
   const {
-    RAPIER
+    rapier
   } = useRapier();
-  return useImpulseJoint(body1, body2, RAPIER.JointData.fixed(vectorArrayToObject(body1Anchor), _objectSpread2(_objectSpread2({}, vectorArrayToObject(body1LocalFrame)), {}, {
+  return useImpulseJoint(body1, body2, rapier.JointData.fixed(vectorArrayToObject(body1Anchor), _objectSpread2(_objectSpread2({}, vectorArrayToObject(body1LocalFrame)), {}, {
     w: 1
   }), vectorArrayToObject(body2Anchor), _objectSpread2(_objectSpread2({}, vectorArrayToObject(body2LocalFrame)), {}, {
     w: 1
@@ -590,9 +738,9 @@ const useFixedJoint = (body1, body2, [body1Anchor, body1LocalFrame, body2Anchor,
 
 const useSphericalJoint = (body1, body2, [body1Anchor, body2Anchor]) => {
   const {
-    RAPIER
+    rapier
   } = useRapier();
-  return useImpulseJoint(body1, body2, RAPIER.JointData.spherical(vectorArrayToObject(body1Anchor), vectorArrayToObject(body2Anchor)));
+  return useImpulseJoint(body1, body2, rapier.JointData.spherical(vectorArrayToObject(body1Anchor), vectorArrayToObject(body2Anchor)));
 };
 /**
  * The revolute joint prevents any relative movement between two rigid-bodies, except for relative
@@ -602,9 +750,9 @@ const useSphericalJoint = (body1, body2, [body1Anchor, body2Anchor]) => {
 
 const useRevoluteJoint = (body1, body2, [body1Anchor, body2Anchor, axis]) => {
   const {
-    RAPIER
+    rapier
   } = useRapier();
-  return useImpulseJoint(body1, body2, RAPIER.JointData.revolute(vectorArrayToObject(body1Anchor), vectorArrayToObject(body2Anchor), vectorArrayToObject(axis)));
+  return useImpulseJoint(body1, body2, rapier.JointData.revolute(vectorArrayToObject(body1Anchor), vectorArrayToObject(body2Anchor), vectorArrayToObject(axis)));
 };
 /**
  * The prismatic joint prevents any relative movement between two rigid-bodies, except for relative translations along one axis.
@@ -614,9 +762,9 @@ const useRevoluteJoint = (body1, body2, [body1Anchor, body2Anchor, axis]) => {
 
 const usePrismaticJoint = (body1, body2, [body1Anchor, body2Anchor, axis]) => {
   const {
-    RAPIER
+    rapier
   } = useRapier();
-  return useImpulseJoint(body1, body2, RAPIER.JointData.prismatic(vectorArrayToObject(body1Anchor), vectorArrayToObject(body2Anchor), vectorArrayToObject(axis)));
+  return useImpulseJoint(body1, body2, rapier.JointData.prismatic(vectorArrayToObject(body1Anchor), vectorArrayToObject(body2Anchor), vectorArrayToObject(axis)));
 };
 
 function _extends() {
@@ -671,7 +819,8 @@ function _objectWithoutProperties(source, excluded) {
   return target;
 }
 
-const _excluded = ["children"];
+const _excluded = ["children"],
+      _excluded2 = ["children"];
 const RigidBodyContext = /*#__PURE__*/React.createContext(undefined);
 
 const useParentRigidBody = () => React.useContext(RigidBodyContext); // RigidBody
@@ -683,28 +832,36 @@ const RigidBody = /*#__PURE__*/React.forwardRef((_ref, ref) => {
   } = _ref,
       props = _objectWithoutProperties(_ref, _excluded);
 
-  const [group, rigidBody] = useRigidBody(props);
+  const [object, rigidBody] = useRigidBody(props);
   React.useImperativeHandle(ref, () => rigidBody);
   return /*#__PURE__*/React__default["default"].createElement(RigidBodyContext.Provider, {
-    value: [group, rigidBody]
-  }, /*#__PURE__*/React__default["default"].createElement("group", {
-    ref: group
+    value: [object, rigidBody]
+  }, /*#__PURE__*/React__default["default"].createElement("object3D", {
+    ref: object
   }, children));
 }); // Colliders
 
-const AnyCollider = props => {
+const AnyCollider = _ref2 => {
+  let {
+    children
+  } = _ref2,
+      props = _objectWithoutProperties(_ref2, _excluded2);
+
   const {
     world
   } = useRapier();
-  const [object, rigidBody] = useParentRigidBody();
+  const [, rigidBody] = useParentRigidBody();
+  const ref = React.useRef(null);
   React.useEffect(() => {
-    const scale = object.current.getWorldScale(new three.Vector3());
-    const collider = createColliderFromOptions(props, world, rigidBody, scale);
+    const scale = ref.current.getWorldScale(new three.Vector3());
+    const collider = createColliderFromOptions(props, world, rigidBody.handle, scale);
     return () => {
-      world.removeCollider(collider, false);
+      world.removeCollider(collider);
     };
   }, []);
-  return null;
+  return /*#__PURE__*/React__default["default"].createElement("object3D", {
+    ref: ref
+  }, children);
 };
 
 const CuboidCollider = props => {
@@ -853,7 +1010,7 @@ const Debug = () => {
 
   fiber.useFrame(() => {
     const newColliders = [];
-    world.colliders.forEachCollider(collider => {
+    world.forEachCollider(collider => {
       newColliders.push(collider.handle);
     });
     setColliders(newColliders);