100 lines
3.4 KiB
TypeScript
100 lines
3.4 KiB
TypeScript
/**
|
||
* @author MaHaiBing
|
||
* @email mlt131220@163.com
|
||
* @date 2024/8/3 22:49
|
||
* @description 包 骨骼处理
|
||
*/
|
||
import {Skeleton, Bone, Object3D } from "three";
|
||
|
||
export class PackageSkeleton{
|
||
// 场景内所有骨骼的map
|
||
private boneMap = new Map<string,Bone>();
|
||
// 场景内所有骨架的map <skeletonUuid:string,skeleton:Skeleton>
|
||
private skeletonsMap = new Map<string, Skeleton>();
|
||
// 骨架未对应上的骨骼 <boneUuid:string,skeleton:Skeleton[]>
|
||
private unMatchBoneMap = new Map<string, Skeleton[]>();
|
||
|
||
constructor() {
|
||
}
|
||
|
||
addBones(bones:Bone[]){
|
||
bones.forEach(bone => {
|
||
if(this.boneMap.has(bone.uuid)) return;
|
||
|
||
this.boneMap.set(bone.uuid,bone);
|
||
|
||
if(this.unMatchBoneMap.has(bone.uuid)){
|
||
const skeletons = this.unMatchBoneMap.get(bone.uuid) as Skeleton[];
|
||
|
||
skeletons.forEach(skeleton => {
|
||
skeleton.bones.push(bone);
|
||
|
||
// 骨骼找到一个替换一个loader.parse时对应骨骼（Bone）还未加载，生成的空骨骼
|
||
const d = skeleton.bones.findIndex(bone => bone.userData.waitDelete);
|
||
if(d === -1) return;
|
||
skeleton.bones.splice(d,1);
|
||
})
|
||
|
||
this.unMatchBoneMap.delete(bone.uuid);
|
||
}
|
||
})
|
||
}
|
||
|
||
handleSkeletons(skeletons,group:Object3D){
|
||
skeletons.forEach(skeleton => {
|
||
if(this.skeletonsMap.has(skeleton.uuid)) return;
|
||
|
||
let skinnedMesh;
|
||
group.traverse((m)=> {
|
||
if(!m.skeleton?.uuid) return;
|
||
|
||
if(m.skeleton.uuid === skeleton.uuid){
|
||
skinnedMesh = m;
|
||
}
|
||
})
|
||
if(!skinnedMesh) return;
|
||
|
||
// 此骨架的原骨骼（Bone）uuid数组
|
||
const bonesUuid:string[] = skeleton.bones;
|
||
|
||
// 对比skinnedMesh.skeleton.bones 和 skeleton.bones
|
||
for(let i = skinnedMesh.skeleton.bones.length;i > 0;i--){
|
||
const bone = skinnedMesh.skeleton.bones[i - 1];
|
||
const u = bonesUuid.indexOf(bone.uuid);
|
||
|
||
if(u === -1){
|
||
// 和原数组没匹配上，说明此骨骼不是原骨骼，是loader.parse时对应骨骼（Bone）还未加载，生成的新的空骨骼
|
||
// 但不能直接从skeleton中删除，需要之后找到对应骨骼一个一个替换
|
||
bone.userData.waitDelete = true;
|
||
}else{
|
||
bonesUuid.splice(u,1);
|
||
}
|
||
}
|
||
|
||
if(bonesUuid.length > 0){
|
||
// 在已存储的骨骼中寻找
|
||
for(let i = bonesUuid.length;i > 0;i--){
|
||
if(this.boneMap.has(bonesUuid[i])){
|
||
skinnedMesh.skeleton.bones.push(<Bone>this.boneMap.get(bonesUuid[i]));
|
||
|
||
bonesUuid.splice(i,1);
|
||
}
|
||
}
|
||
|
||
bonesUuid.forEach(uuid => {
|
||
const skeletons = this.unMatchBoneMap.get(uuid) || [];
|
||
skeletons.push(skinnedMesh.skeleton);
|
||
this.unMatchBoneMap.set(uuid,skeletons);
|
||
})
|
||
}
|
||
|
||
this.skeletonsMap.set(skeleton.uuid,skinnedMesh.skeleton);
|
||
})
|
||
}
|
||
|
||
clear(){
|
||
this.boneMap.clear();
|
||
this.skeletonsMap.clear();
|
||
this.unMatchBoneMap.clear();
|
||
}
|
||
} |