#

luxiaotao1123

2024-04-23 b73f7463490db854721446779cc11e3b910d8fa2

 src/components/agv.jsx

@@ -1,62 +1,73 @@
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 } from '@/config/setting'

const Agv = (props) => {
    const fbxRef = useRef();
    const mixerRef = useRef();
    const { position, theta } = 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 });

    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;
        }
        new TWEEN.Tween(groupPos)
            .to({
                x: position[0],
                y: position[1],
                z: position[2]
            }, INTERVAL_TIME)
            .easing(TWEEN.Easing.Linear.None)
            .onUpdate((e) => {
                setGroupPos({ ...e });
            })
            .start();
    }, [props]);

        return () => {
            mixerRef.current.stopAllAction();
        };
    }, []);
    useFrame((state, delta) => {
        TWEEN.update();
    })

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

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

    const forkModel = useMemo(() => {
        const fbx = useFBX('/models/agv/fork.fbx');
        if (!fbx.castShadow) {
            Common.setShadow(fbx);
        }
        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(theta)}
                scale={0.5}
                position={[groupPos.x, groupPos.y, groupPos.z]}
            >
                <primitive object={bodyModel} castShadow />
                <primitive object={loaderModel} castShadow position={[0, 100, 0]} />
                <primitive object={forkModel} castShadow position={[0, 120, 0]} />
                <Box position={[100, 100, 300]} />
            </group>
        </>
    )