#

luxiaotao1123

2024-04-23 b1a6ae4f052409d0c4c3a846bfad1a9b24a2c696

 src/components/agv.jsx

@@ -1,62 +1,130 @@
import { useRef, useEffect } from 'react';
import { useState, useMemo, useRef, useEffect } from 'react';
import { useFrame } from '@react-three/fiber';
import { useFBX, useAnimations } from '@react-three/drei';
import { AnimationMixer } from 'three';
import * as THREE from 'three';
import Box from './box';
import * as Common from '../utils/common';
import * as TWEEN from "@tweenjs/tween.js";
import { INTERVAL_TIME, MODEL_AGV_SCALE } from '@/config/setting'

const Agv = (props) => {
    const fbxRef = useRef();
    const mixerRef = useRef();
    const { position, theta, height, loaderTheta, forkLength, trayList, hasBox } = props;

    const fbx = useFBX('/models/agv/body.fbx');

    const setShadow = (obj) => {
        obj.castShadow = true;
        obj.receiveShadow = true;

        if (obj.children) {
            obj.children.forEach((child) => {
                setShadow(child);
            });
        }
    };

    const setColor = (obj) => {
        if (obj.material) {
            obj.material.color.set(0x4680BF);
        }
        if (obj.children) {
            obj.children.forEach((child) => {
                setColor(child);
            });
        }
    };

    // setColor(fbx)
    if (!fbx.castShadow) {
        setShadow(fbx);
    }
    const [groupPos, setGroupPos] = useState({ x: 0, y: 0, z: 0 });
    const [groupTheta, setGroupTheta] = useState(0);
    const [loaderHeight, setLoaderHeight] = useState(0);
    const [forkDis, setForkDis] = useState(0);

    useEffect(() => {
        mixerRef.current = new AnimationMixer(fbx);

        if (fbx.animations.length > 0) {
            useFrame((state, delta) => {
                mixerRef?.current?.update(delta);
            });

            const action = mixerRef.current.clipAction(fbx.animations[0]); // 假设只有一个动画，可以根据实际情况修改
            action.play();
        if (!props || Object.keys(props).length === 0) {
            return;
        }
        // group position
        const pos = {
            x: position?.[0],
            y: position?.[1],
            z: position?.[2],
        }
        if (!Common.deepEqual(groupPos, pos)) {
            new TWEEN.Tween(groupPos)
                .to(pos, INTERVAL_TIME)
                .easing(TWEEN.Easing.Linear.None)
                .onUpdate((e) => {
                    setGroupPos({ ...e });
                })
                .onComplete((e) => {
                    setGroupPos({ ...e });
                }).start();
        }

        return () => {
            mixerRef.current.stopAllAction();
        };
    }, []);
        // group theta
        if (groupTheta !== theta) {
            const minTheta = Common.minDiffTheta(groupTheta, theta);
            new TWEEN.Tween({ value: groupTheta })
                .to({ value: minTheta }, INTERVAL_TIME)
                .easing(TWEEN.Easing.Linear.None)
                .onUpdate((e) => {
                    setGroupTheta(e.value)
                })
                .onComplete((e) => {
                    setGroupTheta(e.value)
                }).start();
        }

        // loader height
        if (loaderHeight !== height) {
            new TWEEN.Tween({ value: loaderHeight })
                .to({ value: height }, INTERVAL_TIME)
                .easing(TWEEN.Easing.Linear.None)
                .onUpdate((e) => {
                    setLoaderHeight(e.value)
                })
                .onComplete((e) => {
                    setLoaderHeight(e.value)
                }).start();
        }

        // fork lenght
        if (forkDis !== forkLength) {
            new TWEEN.Tween({ value: forkDis })
                .to({ value: forkLength }, INTERVAL_TIME)
                .easing(TWEEN.Easing.Linear.None)
                .onUpdate((e) => {
                    setForkDis(e.value)
                })
                .onComplete((e) => {
                    setForkDis(e.value)
                    console.log(e.value);
                }).start();
        }

    }, [props]);

    useFrame((state, delta) => {
        TWEEN.update();
    })

    const bodyModel = useMemo(() => {
        const fbx = useFBX('/models/agv/body.fbx');
        if (!fbx.castShadow) {
            Common.setShadow(fbx);
        }
        fbx.scale.set(MODEL_AGV_SCALE, MODEL_AGV_SCALE, MODEL_AGV_SCALE);
        return fbx.clone();
    }, [])

    const loaderModel = useMemo(() => {
        const fbx = useFBX('/models/agv/loader.fbx');
        if (!fbx.castShadow) {
            Common.setShadow(fbx);
        }
        fbx.scale.set(MODEL_AGV_SCALE, MODEL_AGV_SCALE, MODEL_AGV_SCALE);
        return fbx.clone();
    }, [])

    const forkModel = useMemo(() => {
        const fbx = useFBX('/models/agv/fork.fbx');
        if (!fbx.castShadow) {
            Common.setShadow(fbx);
        }
        fbx.scale.set(MODEL_AGV_SCALE, MODEL_AGV_SCALE, MODEL_AGV_SCALE);
        return fbx.clone();
    }, [])

    return (
        <>
            <group rotation={[0, 0, 0]} scale={0.5} position={[0, 0, 0]}>
                <primitive object={fbx} castShadow />
            <group
                rotation-y={Common.rotationParseNum(groupTheta)}
                position={[groupPos.x, groupPos.y, groupPos.z]}
            >
                <primitive object={bodyModel} castShadow />
                <group position-y={loaderHeight}>
                    <primitive object={loaderModel} castShadow position={[0, 30, 0]} />
                    <group position-z={forkDis}>
                        <primitive object={forkModel} castShadow position={[0, 35, 0]} />
                        {hasBox && <Box position={[0, 34, 0]} />}
                    </group>
                </group>
            </group>
        </>
    )