@wandelbots/wandelbots-js-react-components 1.3.1 → 1.3.2

package/src/components/robots/KUKA_KR210_R2700.tsx

@@ -0,0 +1,291 @@
+import { useGLTF } from "@react-three/drei"
+import type * as THREE from "three"
+import type { RobotProps } from "./SupportedRobot"
+import { animated } from "@react-spring/three"
+import RobotAnimator from "./RobotAnimator"
+
+export function KUKA_KR210_R2700({
+  modelURL,
+  connectedMotionGroup,
+  ...props
+}: RobotProps) {
+  const gltf = useGLTF(modelURL) as any
+
+  const nodes = gltf.nodes
+  const materials = gltf.materials
+
+  const rotationOffsets = [0, 0, -Math.PI / 2, 0, 0, 0]
+
+  function setRotation(jointObjects: THREE.Object3D[], jointValues: number[]) {
+    jointObjects.forEach(
+      (object, index) =>
+        (object.rotation.y = jointValues[index]! + rotationOffsets[index]!),
+    )
+  }
+
+  return (
+    <>
+      <RobotAnimator
+        connectedMotionGroup={connectedMotionGroup}
+        robotRootObjectName="Scene"
+        onRotationChanged={setRotation}
+      />
+      <group {...props} dispose={null}>
+        <group name="Scene">
+          <group name="kuka_kr210_2700">
+            <animated.group name="joint_01">
+              <animated.group
+                name="joint_02"
+                position={[0.33, 0.645, 0]}
+                rotation={[-Math.PI / 2, 0, 0]}
+              >
+                <animated.group
+                  name="joint_03"
+                  position={[1.15, 0, 0]}
+                  rotation={[0, -Math.PI / 2, 0]}
+                >
+                  <animated.group
+                    name="joint_04"
+                    position={[0.115, 0, 0]}
+                    rotation={[Math.PI / 2, 0, 0]}
+                  >
+                    <animated.group
+                      name="joint_05"
+                      position={[0, -1.22, 0]}
+                      rotation={[-Math.PI / 2, 0, 0]}
+                    >
+                      <animated.group
+                        name="joint_06"
+                        rotation={[Math.PI / 2, 0, 0]}
+                      >
+                        <group
+                          name="flange"
+                          position={[0, -0.215, 0]}
+                          rotation={[-Math.PI, 0, 0]}
+                        />
+                        <group
+                          name="link_6"
+                          position={[0, -0.17, 0]}
+                          rotation={[0, 0, -Math.PI / 2]}
+                        >
+                          <mesh
+                            name="visuals006"
+                            castShadow
+                            receiveShadow
+                            geometry={nodes.visuals006.geometry}
+                            material={materials.material_Material_Metall_black}
+                          />
+                        </group>
+                      </animated.group>
+                      <group
+                        name="link_5"
+                        position={[0, 0.059, 0]}
+                        rotation={[Math.PI / 2, 0, -Math.PI / 2]}
+                      >
+                        <group name="visuals005">
+                          <mesh
+                            name="visuals005_1"
+                            castShadow
+                            receiveShadow
+                            geometry={nodes.visuals005_1.geometry}
+                            material={materials.material_Material_metal_black}
+                          />
+                          <mesh
+                            name="visuals005_2"
+                            castShadow
+                            receiveShadow
+                            geometry={nodes.visuals005_2.geometry}
+                            material={materials.material_Material_kuka_orange}
+                          />
+                        </group>
+                      </group>
+                    </animated.group>
+                    <group
+                      name="link_4"
+                      position={[0, -0.869, 0]}
+                      rotation={[0, 0, -Math.PI / 2]}
+                    >
+                      <group name="visuals004">
+                        <mesh
+                          name="visuals004_1"
+                          castShadow
+                          receiveShadow
+                          geometry={nodes.visuals004_1.geometry}
+                          material={materials.material_Material_metal_black}
+                        />
+                        <mesh
+                          name="visuals004_2"
+                          castShadow
+                          receiveShadow
+                          geometry={nodes.visuals004_2.geometry}
+                          material={materials.material_Material_kuka_orange}
+                        />
+                      </group>
+                    </group>
+                  </animated.group>
+                  <group
+                    name="link_3"
+                    position={[0, -0.105, 0]}
+                    rotation={[Math.PI / 2, 0, -Math.PI / 2]}
+                  >
+                    <group name="visuals003">
+                      <mesh
+                        name="visuals003_1"
+                        castShadow
+                        receiveShadow
+                        geometry={nodes.visuals003_1.geometry}
+                        material={materials.material_Material_rubber}
+                      />
+                      <mesh
+                        name="visuals003_2"
+                        castShadow
+                        receiveShadow
+                        geometry={nodes.visuals003_2.geometry}
+                        material={materials.material_Material_aluminium}
+                      />
+                      <mesh
+                        name="visuals003_3"
+                        castShadow
+                        receiveShadow
+                        geometry={nodes.visuals003_3.geometry}
+                        material={materials.material_Material_metal_black}
+                      />
+                      <mesh
+                        name="visuals003_4"
+                        castShadow
+                        receiveShadow
+                        geometry={nodes.visuals003_4.geometry}
+                        material={materials.material_Material_kuka_black}
+                      />
+                      <mesh
+                        name="visuals003_5"
+                        castShadow
+                        receiveShadow
+                        geometry={nodes.visuals003_5.geometry}
+                        material={materials.material_Material_Rubber_black}
+                      />
+                      <mesh
+                        name="visuals003_6"
+                        castShadow
+                        receiveShadow
+                        geometry={nodes.visuals003_6.geometry}
+                        material={materials.material_Material_kuka_orange}
+                      />
+                      <mesh
+                        name="visuals003_7"
+                        castShadow
+                        receiveShadow
+                        geometry={nodes.visuals003_7.geometry}
+                        material={materials.material_Material_stainless_steel}
+                      />
+                    </group>
+                  </group>
+                </animated.group>
+                <group
+                  name="link_2"
+                  position={[0, 0.009, 0]}
+                  rotation={[Math.PI / 2, 0, 0]}
+                >
+                  <group name="visuals002">
+                    <mesh
+                      name="visuals002_1"
+                      castShadow
+                      receiveShadow
+                      geometry={nodes.visuals002_1.geometry}
+                      material={materials.material_Material_metal_black}
+                    />
+                    <mesh
+                      name="visuals002_2"
+                      castShadow
+                      receiveShadow
+                      geometry={nodes.visuals002_2.geometry}
+                      material={materials.material_Material_Rubber_black}
+                    />
+                    <mesh
+                      name="visuals002_3"
+                      castShadow
+                      receiveShadow
+                      geometry={nodes.visuals002_3.geometry}
+                      material={materials.material_Material_kuka_orange}
+                    />
+                  </group>
+                </group>
+              </animated.group>
+              <group name="link_1" position={[0, 0.236, 0]}>
+                <group name="visuals001">
+                  <mesh
+                    name="visuals001_1"
+                    castShadow
+                    receiveShadow
+                    geometry={nodes.visuals001_1.geometry}
+                    material={materials.material_Material_kuka_black}
+                  />
+                  <mesh
+                    name="visuals001_2"
+                    castShadow
+                    receiveShadow
+                    geometry={nodes.visuals001_2.geometry}
+                    material={materials.material_Material_kuka_orange}
+                  />
+                  <mesh
+                    name="visuals001_3"
+                    castShadow
+                    receiveShadow
+                    geometry={nodes.visuals001_3.geometry}
+                    material={materials.material_Material_rubber}
+                  />
+                  <mesh
+                    name="visuals001_4"
+                    castShadow
+                    receiveShadow
+                    geometry={nodes.visuals001_4.geometry}
+                    material={materials.material_Material_metal_black}
+                  />
+                  <mesh
+                    name="visuals001_5"
+                    castShadow
+                    receiveShadow
+                    geometry={nodes.visuals001_5.geometry}
+                    material={materials.material_Material_aluminium}
+                  />
+                  <mesh
+                    name="visuals001_6"
+                    castShadow
+                    receiveShadow
+                    geometry={nodes.visuals001_6.geometry}
+                    material={materials.material_Material_rubber_black}
+                  />
+                  <mesh
+                    name="visuals001_7"
+                    castShadow
+                    receiveShadow
+                    geometry={nodes.visuals001_7.geometry}
+                    material={materials.material_Material_stainless_steel}
+                  />
+                </group>
+              </group>
+            </animated.group>
+            <group name="link_0">
+              <group name="visuals">
+                <mesh
+                  name="visuals_1"
+                  castShadow
+                  receiveShadow
+                  geometry={nodes.visuals_1.geometry}
+                  material={materials.material_Material_kuka_black}
+                />
+                <mesh
+                  name="visuals_2"
+                  castShadow
+                  receiveShadow
+                  geometry={nodes.visuals_2.geometry}
+                  material={materials.material_Material_metal_black}
+                />
+              </group>
+            </group>
+          </group>
+        </group>
+      </group>
+    </>
+  )
+}

package/src/components/robots/KUKA_KR270_R2700.tsx

@@ -0,0 +1,244 @@
+import { useGLTF } from "@react-three/drei"
+import type * as THREE from "three"
+import type { RobotProps } from "./SupportedRobot"
+import { animated } from "@react-spring/three"
+import RobotAnimator from "./RobotAnimator"
+
+export function KUKA_KR270_R2700({
+  modelURL,
+  connectedMotionGroup,
+  ...props
+}: RobotProps) {
+  const gltf = useGLTF(modelURL) as any
+
+  const nodes = gltf.nodes
+  const materials = gltf.materials
+
+  const rotationOffsets = [0, 0, -Math.PI / 2, 0, 0, 0]
+
+  function setRotation(jointObjects: THREE.Object3D[], jointValues: number[]) {
+    jointObjects.forEach(
+      (object, index) =>
+        (object.rotation.y = jointValues[index]! + rotationOffsets[index]!),
+    )
+  }
+
+  return (
+    <>
+      <RobotAnimator
+        connectedMotionGroup={connectedMotionGroup}
+        robotRootObjectName="Scene"
+        onRotationChanged={setRotation}
+      />
+      <group {...props} dispose={null}>
+        <group name="Scene">
+          <group name="kuka_kr270_2700_ultra">
+            <animated.group name="joint_01">
+              <animated.group
+                name="joint_02"
+                position={[0.35, 0.675, 0]}
+                rotation={[-Math.PI / 2, 0, 0]}
+              >
+                <animated.group
+                  name="joint_03"
+                  position={[1.15, 0, 0]}
+                  rotation={[0, -Math.PI / 2, 0]}
+                >
+                  <animated.group
+                    name="joint_04"
+                    position={[-0.041, 0, 0]}
+                    rotation={[Math.PI / 2, 0, 0]}
+                  >
+                    <animated.group
+                      name="joint_05"
+                      position={[0, -1.2, 0]}
+                      rotation={[-Math.PI / 2, 0, 0]}
+                    >
+                      <animated.group
+                        name="joint_06"
+                        rotation={[Math.PI / 2, 0, 0]}
+                      >
+                        <group
+                          name="flange"
+                          position={[0, -0.24, 0]}
+                          rotation={[-Math.PI, 0, 0]}
+                        />
+                        <group
+                          name="link_6"
+                          position={[0, -0.213, 0]}
+                          rotation={[0, 0, -Math.PI / 2]}
+                        >
+                          <group name="visuals006">
+                            <mesh
+                              name="visuals006_1"
+                              castShadow
+                              receiveShadow
+                              geometry={nodes.visuals006_1.geometry}
+                              material={materials.material_Material_Farbe__1_}
+                            />
+                            <mesh
+                              name="visuals006_2"
+                              castShadow
+                              receiveShadow
+                              geometry={nodes.visuals006_2.geometry}
+                              material={materials.material_Material_Metall}
+                            />
+                          </group>
+                        </group>
+                      </animated.group>
+                      <group
+                        name="link_5"
+                        position={[0, 0.044, 0]}
+                        rotation={[Math.PI / 2, 0, -Math.PI / 2]}
+                      >
+                        <mesh
+                          name="visuals005"
+                          castShadow
+                          receiveShadow
+                          geometry={nodes.visuals005.geometry}
+                          material={materials.material_Material_Farbe}
+                        />
+                      </group>
+                    </animated.group>
+                    <group
+                      name="link_4"
+                      position={[0, -0.954, 0]}
+                      rotation={[0, 0, -Math.PI / 2]}
+                    >
+                      <mesh
+                        name="visuals004"
+                        castShadow
+                        receiveShadow
+                        geometry={nodes.visuals004.geometry}
+                        material={materials.material_Material_Farbe}
+                      />
+                    </group>
+                  </animated.group>
+                  <group
+                    name="link_3"
+                    position={[0, -0.178, 0]}
+                    rotation={[Math.PI / 2, 0, -Math.PI / 2]}
+                  >
+                    <group name="visuals003">
+                      <mesh
+                        name="visuals003_1"
+                        castShadow
+                        receiveShadow
+                        geometry={nodes.visuals003_1.geometry}
+                        material={materials.material_Material_Farbe__1_}
+                      />
+                      <mesh
+                        name="visuals003_2"
+                        castShadow
+                        receiveShadow
+                        geometry={nodes.visuals003_2.geometry}
+                        material={materials.material_Material_Farbe__2_}
+                      />
+                      <mesh
+                        name="visuals003_3"
+                        castShadow
+                        receiveShadow
+                        geometry={nodes.visuals003_3.geometry}
+                        material={materials.material_Material_Metall}
+                      />
+                      <mesh
+                        name="visuals003_4"
+                        castShadow
+                        receiveShadow
+                        geometry={nodes.visuals003_4.geometry}
+                        material={materials.material_Material_Farbe}
+                      />
+                    </group>
+                  </group>
+                </animated.group>
+                <group name="link_2" rotation={[Math.PI / 2, 0, 0]}>
+                  <group name="visuals002">
+                    <mesh
+                      name="visuals002_1"
+                      castShadow
+                      receiveShadow
+                      geometry={nodes.visuals002_1.geometry}
+                      material={materials.material_Material_Farbe}
+                    />
+                    <mesh
+                      name="visuals002_2"
+                      castShadow
+                      receiveShadow
+                      geometry={nodes.visuals002_2.geometry}
+                      material={materials.material_Material_Farbe__1_}
+                    />
+                    <mesh
+                      name="visuals002_3"
+                      castShadow
+                      receiveShadow
+                      geometry={nodes.visuals002_3.geometry}
+                      material={materials.material_Material_Metall}
+                    />
+                    <mesh
+                      name="visuals002_4"
+                      castShadow
+                      receiveShadow
+                      geometry={nodes.visuals002_4.geometry}
+                      material={materials.material_Material_Farbe__2_}
+                    />
+                  </group>
+                </group>
+              </animated.group>
+              <group name="link_1" position={[0, 0.226, 0]}>
+                <group name="visuals001">
+                  <mesh
+                    name="visuals001_1"
+                    castShadow
+                    receiveShadow
+                    geometry={nodes.visuals001_1.geometry}
+                    material={materials.material_Material_Metall}
+                  />
+                  <mesh
+                    name="visuals001_2"
+                    castShadow
+                    receiveShadow
+                    geometry={nodes.visuals001_2.geometry}
+                    material={materials.material_Material_Farbe__1_}
+                  />
+                  <mesh
+                    name="visuals001_3"
+                    castShadow
+                    receiveShadow
+                    geometry={nodes.visuals001_3.geometry}
+                    material={materials.material_Material_Farbe__2_}
+                  />
+                  <mesh
+                    name="visuals001_4"
+                    castShadow
+                    receiveShadow
+                    geometry={nodes.visuals001_4.geometry}
+                    material={materials.material_Material_Farbe}
+                  />
+                </group>
+              </group>
+            </animated.group>
+            <group name="link_0-base">
+              <group name="joint_1" />
+              <group name="visuals">
+                <mesh
+                  name="visuals_1"
+                  castShadow
+                  receiveShadow
+                  geometry={nodes.visuals_1.geometry}
+                  material={materials.material_Material_Metall}
+                />
+                <mesh
+                  name="visuals_2"
+                  castShadow
+                  receiveShadow
+                  geometry={nodes.visuals_2.geometry}
+                  material={materials.material_Material_Farbe__1_}
+                />
+              </group>
+            </group>
+          </group>
+        </group>
+      </group>
+    </>
+  )
+}

package/src/components/robots/RobotAnimator.tsx

@@ -0,0 +1,83 @@
+import { useEffect, useRef } from "react"
+import { Globals, useSpring } from "@react-spring/three"
+import { useThree } from "@react-three/fiber"
+import type * as THREE from "three"
+import {
+  getAllJointsByName,
+  type RobotSceneJoint,
+} from "../utils/robotTreeQuery"
+import type { MotionStreamConnection } from "@wandelbots/wandelbots-js"
+import { useAutorun } from "../utils/hooks"
+
+type RobotAnimatorProps = {
+  connectedMotionGroup: MotionStreamConnection
+  robotRootObjectName: string
+  onRotationChanged: (joints: THREE.Object3D[], jointValues: number[]) => void
+  jointCollector?: (rootObject: THREE.Object3D) => RobotSceneJoint[]
+}
+
+export default function RobotAnimator({
+  connectedMotionGroup,
+  robotRootObjectName,
+  onRotationChanged,
+  jointCollector,
+}: RobotAnimatorProps) {
+  Globals.assign({ frameLoop: "always" })
+  const jointValues = useRef<number[]>([])
+  const jointObjects = useRef<THREE.Object3D[]>([])
+  const { scene, invalidate } = useThree()
+
+  useEffect(() => {
+    // All robots have a "Scene" object as root
+    // From this object the tree is traversed and all joints are extracted (ordered ASC)
+    const sceneObject = scene.getObjectByName(robotRootObjectName)
+    if (!sceneObject) {
+      return
+    }
+
+    jointObjects.current = jointCollector
+      ? jointCollector(sceneObject)
+      : getAllJointsByName(sceneObject)
+
+    // Set initial position
+    setRotation()
+    invalidate()
+  }, [])
+
+  function updateJoints(newJointValues: number[]) {
+    jointValues.current = newJointValues
+    setSpring.start(Object.assign({}, jointValues.current) as any)
+  }
+
+  function setRotation() {
+    const updatedJointValues = jointObjects.current.map((object, objectIndex) =>
+      (axisValues as any)[objectIndex].get(),
+    )
+    onRotationChanged(jointObjects.current, updatedJointValues)
+  }
+
+  useAutorun(() => {
+    const newJointValues =
+      connectedMotionGroup.rapidlyChangingMotionState.state.joint_position.joints.filter(
+        (item) => item !== undefined,
+      )
+
+    requestAnimationFrame(() => updateJoints(newJointValues))
+  })
+
+  const [axisValues, setSpring] = useSpring(() => ({
+    ...Object.assign(
+      {},
+      connectedMotionGroup.rapidlyChangingMotionState.state.joint_position
+        .joints,
+    ),
+    onChange: () => {
+      setRotation()
+      invalidate()
+    },
+    onResolve: () => {
+      setRotation()
+    },
+  }))
+  return null
+}