17 changed files with 108 additions and 480 deletions
@ -1,408 +0,0 @@ |
|||||
import * as THREE from 'three' |
|
||||
import type Viewport from '@/core/engine/Viewport.ts' |
|
||||
import { Vector3 } from 'three/src/math/Vector3' |
|
||||
import InstanceMeshBlock from '@/core/manager/InstanceMeshBlock.ts' |
|
||||
import { getMatrixFromTf } from '@/core/ModelUtils.ts' |
|
||||
|
|
||||
/** |
|
||||
* 点实例管理器 |
|
||||
* 内部使用 InstanceMesh 用于管理大量点实例 |
|
||||
*/ |
|
||||
export default class InstancePointManager { |
|
||||
public readonly name: string |
|
||||
public readonly viewport: Viewport |
|
||||
public readonly allowSelect: boolean |
|
||||
public readonly allowDrag: boolean |
|
||||
public readonly blockCapacity: number = 1000 // 每个 block 的容量
|
|
||||
|
|
||||
public readonly blocks: InstanceMeshBlock[] = [] |
|
||||
private readonly geometry: THREE.BufferGeometry |
|
||||
private readonly material: THREE.Material |
|
||||
private readonly dummy: THREE.Object3D = new THREE.Object3D() |
|
||||
// itemId -> data
|
|
||||
private __uuidMap = new Map<string, PointManageWrap>() |
|
||||
|
|
||||
createPointSimple(id: string): PointManageWrap { |
|
||||
//@ts-ignore
|
|
||||
const wrap = this.createPoint({ |
|
||||
id: id, |
|
||||
name: id, |
|
||||
v: true, |
|
||||
tf: [[0, 0, 0], [0, 0, 0], [0, 0, 0]], |
|
||||
dt: {} |
|
||||
}) |
|
||||
|
|
||||
return wrap |
|
||||
} |
|
||||
|
|
||||
/** |
|
||||
* 创建点实例 |
|
||||
* @param item 点数据 |
|
||||
* @returns 分配的实例ID (如果失败返回-1) |
|
||||
*/ |
|
||||
createPoint(item: ItemJson): PointManageWrap { |
|
||||
// Try existing blocks
|
|
||||
let meshIndex = -1 |
|
||||
let blockIndex = -1 |
|
||||
for (const block of this.blocks) { |
|
||||
meshIndex = block.getFreeMeshIndex() |
|
||||
if (meshIndex >= 0) { |
|
||||
blockIndex = block.blockIndex |
|
||||
break |
|
||||
} |
|
||||
} |
|
||||
// 所有 block 都没有空闲索引，创建新的 block
|
|
||||
if (meshIndex < 0) { |
|
||||
const block = this.createBlock() |
|
||||
meshIndex = block.getFreeMeshIndex() |
|
||||
blockIndex = block.blockIndex |
|
||||
} |
|
||||
if (meshIndex < 0) { |
|
||||
system.showErrorDialog('InstancePointManager: No free index available after creating new block') |
|
||||
return null |
|
||||
} |
|
||||
|
|
||||
return new PointManageWrap(this, blockIndex, meshIndex, { |
|
||||
uuid: item.id, |
|
||||
name: item.name, |
|
||||
blockIndex: blockIndex, |
|
||||
meshIndex: meshIndex, |
|
||||
visible: item.v !== false, |
|
||||
//@ts-ignore
|
|
||||
userData: { |
|
||||
t: item.t, |
|
||||
createType: 'point', |
|
||||
blockIndex: blockIndex, |
|
||||
meshIndex: meshIndex, |
|
||||
entityId: item.id |
|
||||
} |
|
||||
}) |
|
||||
} |
|
||||
|
|
||||
|
|
||||
/** |
|
||||
* 更新点实例 |
|
||||
* @param item 点数据 |
|
||||
* @param option 更新选项 |
|
||||
*/ |
|
||||
updatePoint(item: ItemJson, option: { position?: Vector3, rotation?: Vector3, scale?: Vector3 } = {}): void { |
|
||||
const wrap = this.__uuidMap.get(item.id) |
|
||||
if (wrap === undefined) return |
|
||||
|
|
||||
wrap.visible = item.v !== false |
|
||||
wrap.userData.t = item.t |
|
||||
wrap.userData.entityId = item.id |
|
||||
|
|
||||
wrap.setMatrix4(getMatrixFromTf(item.tf)) |
|
||||
|
|
||||
// let [position, rotation, scale] = item.tf
|
|
||||
// if (option.position) {
|
|
||||
// position = option.position.toArray()
|
|
||||
// }
|
|
||||
// if (option.rotation) {
|
|
||||
// rotation = option.rotation.toArray()
|
|
||||
// }
|
|
||||
// if (option.scale) {
|
|
||||
// scale = option.scale.toArray()
|
|
||||
// }
|
|
||||
//
|
|
||||
// wrap.position.set(position[0], position[1], position[2])
|
|
||||
// wrap.rotation.set(
|
|
||||
// THREE.MathUtils.degToRad(rotation[0]),
|
|
||||
// THREE.MathUtils.degToRad(rotation[1]),
|
|
||||
// THREE.MathUtils.degToRad(rotation[2])
|
|
||||
// )
|
|
||||
// wrap.scale.set(scale[0], scale[1], scale[2])
|
|
||||
// this.syncMeshObject3D(wrap)
|
|
||||
} |
|
||||
|
|
||||
/** |
|
||||
* 获取点实例数据 |
|
||||
*/ |
|
||||
getObject3DLike(id: string): PointManageWrap { |
|
||||
return this.__uuidMap.get(id) |
|
||||
} |
|
||||
|
|
||||
/** |
|
||||
* 获取点实例数据 |
|
||||
*/ |
|
||||
findByMeshInstanceId(blockIndex: number, instanceId: number): PointManageWrap { |
|
||||
if (!this.blocks[blockIndex]) { |
|
||||
console.error('InstancePointManager: Invalid blockIndex', blockIndex) |
|
||||
return null |
|
||||
} |
|
||||
const uuid = this.blocks[blockIndex].__indexIdMap.get(instanceId) |
|
||||
if (!uuid) return |
|
||||
return this.__uuidMap.get(uuid) |
|
||||
} |
|
||||
|
|
||||
setBlockMatrixAt(wrap: PointManageWrap, matrix: THREE.Matrix4) { |
|
||||
const block = this.blocks[wrap.blockIndex] |
|
||||
if (!block) { |
|
||||
console.warn(`InstanceMeshManager: Block ${wrap.blockIndex} not found!`) |
|
||||
return |
|
||||
} |
|
||||
|
|
||||
if (!block.__indexIdMap.has(wrap.meshIndex)) { |
|
||||
wrap.parent = block.instancedMesh |
|
||||
this.__uuidMap.set(wrap.uuid, wrap) |
|
||||
block.__indexIdMap.set(wrap.meshIndex, wrap.uuid) |
|
||||
} |
|
||||
block.instancedMesh.setMatrixAt(wrap.meshIndex, matrix) |
|
||||
block.instancedMesh.instanceMatrix.needsUpdate = true |
|
||||
} |
|
||||
|
|
||||
// syncMeshObject3D(wrap: PointManageWrap) {
|
|
||||
// if (wrap.meshIndex < 0) {
|
|
||||
// console.error('InstancePointManager: Invalid meshIndex for wrap', wrap)
|
|
||||
// return
|
|
||||
// }
|
|
||||
// if (!wrap.visible) {
|
|
||||
// this.dummy.scale.set(0, 0, 0)
|
|
||||
// } else {
|
|
||||
// this.dummy.position.copy(wrap.position)
|
|
||||
// this.dummy.rotation.copy(wrap.rotation)
|
|
||||
// this.dummy.scale.copy(wrap.scale)
|
|
||||
// }
|
|
||||
// this.dummy.updateMatrix()
|
|
||||
//
|
|
||||
// const block = this.blocks[wrap.blockIndex]
|
|
||||
// if (!wrap.parent) {
|
|
||||
// wrap.parent = block.instancedMesh
|
|
||||
// this.__uuidMap.set(wrap.uuid, wrap)
|
|
||||
// block.__indexIdMap.set(wrap.meshIndex, wrap.uuid)
|
|
||||
// }
|
|
||||
// block.instancedMesh.setMatrixAt(wrap.meshIndex, this.dummy.matrix)
|
|
||||
// block.instancedMesh.instanceMatrix.needsUpdate = true
|
|
||||
// }
|
|
||||
|
|
||||
/** |
|
||||
* 删除点实例 |
|
||||
* @param id 点ID |
|
||||
*/ |
|
||||
deletePoint(id: string): void { |
|
||||
const wrap = this.__uuidMap.get(id) |
|
||||
if (!wrap) { |
|
||||
console.warn(`InstanceMeshManager: Wrap with id ${id} not found`) |
|
||||
return |
|
||||
} |
|
||||
|
|
||||
const block = this.blocks[wrap.blockIndex] |
|
||||
if (!block) { |
|
||||
console.warn(`InstanceMeshManager: Block ${wrap.blockIndex} not found for wrap ${id}`) |
|
||||
return |
|
||||
} |
|
||||
|
|
||||
// 隐藏实例
|
|
||||
this.dummy.scale.set(0, 0, 0) |
|
||||
this.dummy.updateMatrix() |
|
||||
block.instancedMesh.setMatrixAt(wrap.meshIndex, this.dummy.matrix) |
|
||||
block.instancedMesh.instanceMatrix.needsUpdate = true |
|
||||
|
|
||||
// 回收索引
|
|
||||
block.freeIndices.push(wrap.meshIndex) |
|
||||
this.__uuidMap.delete(id) |
|
||||
block.__indexIdMap.delete(wrap.meshIndex) |
|
||||
|
|
||||
wrap.dispose() |
|
||||
} |
|
||||
|
|
||||
public static create( |
|
||||
name: string, |
|
||||
viewport: Viewport, |
|
||||
geometry: THREE.BufferGeometry, |
|
||||
material: THREE.Material, |
|
||||
allowSelect: boolean, allowDrag: boolean |
|
||||
): InstancePointManager { |
|
||||
return new InstancePointManager(name, allowSelect, allowDrag, viewport, geometry, material) |
|
||||
} |
|
||||
|
|
||||
private createBlock(): InstanceMeshBlock { |
|
||||
const block = new InstanceMeshBlock( |
|
||||
this.name, this.allowSelect, this.allowDrag, this.viewport, |
|
||||
this.geometry, |
|
||||
this.material, this.blocks.length, this.blockCapacity |
|
||||
) |
|
||||
this.blocks.push(block) |
|
||||
return block |
|
||||
} |
|
||||
|
|
||||
private constructor(name: string, allowSelect: boolean, allowDrag: boolean, viewport: Viewport, geometry: THREE.BufferGeometry, material: THREE.Material) { |
|
||||
this.name = name |
|
||||
this.allowSelect = allowSelect |
|
||||
this.allowDrag = allowDrag |
|
||||
this.viewport = viewport |
|
||||
this.geometry = geometry |
|
||||
this.material = material |
|
||||
|
|
||||
// 创建第一个 block
|
|
||||
this.createBlock() |
|
||||
|
|
||||
|
|
||||
// this.instancedMesh = new THREE.InstancedMesh(geometry, material, maxInstances)
|
|
||||
// this.instancedMesh.instanceMatrix.setUsage(THREE.DynamicDrawUsage)
|
|
||||
// this.instancedMesh.name = name
|
|
||||
// console.log('InstancePointManager: [' + name + '] created with maxInstances:', maxInstances)
|
|
||||
// viewport.scene.add(this.instancedMesh)
|
|
||||
// if (allowSelect) {
|
|
||||
// this.viewport.entityManager._selectableObjects.push(this.instancedMesh)
|
|
||||
// }
|
|
||||
// if (allowDrag) {
|
|
||||
// this.viewport.entityManager._draggableObjects.push(this.instancedMesh)
|
|
||||
// }
|
|
||||
// this.instancedMesh.userData.t = name
|
|
||||
// this.instancedMesh.userData.entityId = 'InstancePointManager'
|
|
||||
//
|
|
||||
// this.dummy.scale.set(0, 0, 0)
|
|
||||
// for (let i = 0; i < maxInstances; i++) {
|
|
||||
// this.dummy.updateMatrix()
|
|
||||
// this.instancedMesh.setMatrixAt(i, this.dummy.matrix)
|
|
||||
// this.freeIndices.push(i)
|
|
||||
// }
|
|
||||
//
|
|
||||
// this.instancedMesh.instanceMatrix.needsUpdate = true
|
|
||||
} |
|
||||
|
|
||||
dispose() { |
|
||||
for (const block of this.blocks) { |
|
||||
block.dispose() |
|
||||
} |
|
||||
this.__uuidMap.clear() |
|
||||
this.blocks.length = 0 |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
export class PointManageWrap { |
|
||||
readonly manager: InstancePointManager |
|
||||
readonly blockIndex: number = -1 |
|
||||
readonly meshIndex: number = -1 |
|
||||
parent: THREE.Object3D | null = null |
|
||||
|
|
||||
uuid: string |
|
||||
name: string |
|
||||
visible: boolean |
|
||||
|
|
||||
//@ts-ignore
|
|
||||
userData: UserData = {} |
|
||||
|
|
||||
updateWorldMatrix() { |
|
||||
|
|
||||
} |
|
||||
|
|
||||
get type() { |
|
||||
return 'PointManageWrap' |
|
||||
} |
|
||||
|
|
||||
get isObject3D() { |
|
||||
return false |
|
||||
} |
|
||||
|
|
||||
constructor(pointManager: InstancePointManager, blockIndex: number, meshIndex: number, data: any) { |
|
||||
this.manager = pointManager |
|
||||
this.blockIndex = blockIndex |
|
||||
this.meshIndex = meshIndex |
|
||||
_.extend(this, data) |
|
||||
} |
|
||||
|
|
||||
dispose() { |
|
||||
this.manager.deletePoint(this.uuid) |
|
||||
this.parent = null |
|
||||
} |
|
||||
|
|
||||
setMatrix4(matrix: THREE.Matrix4): PointManageWrap { |
|
||||
this.manager.setBlockMatrixAt(this, matrix) |
|
||||
return this |
|
||||
} |
|
||||
|
|
||||
set matrix(matrix: THREE.Matrix4) { |
|
||||
this.setMatrix4(matrix) |
|
||||
} |
|
||||
|
|
||||
get matrix(): THREE.Matrix4 { |
|
||||
const block = this.manager.blocks[this.blockIndex] |
|
||||
const matrix = new THREE.Matrix4() |
|
||||
block.instancedMesh.getMatrixAt(this.meshIndex, matrix) |
|
||||
return matrix |
|
||||
} |
|
||||
|
|
||||
applyMatrix4(targetMatrix: THREE.Matrix4): PointManageWrap { |
|
||||
const matrix = this.matrix |
|
||||
const newMatrix = matrix.multiplyMatrices(targetMatrix, matrix) |
|
||||
this.setMatrix4(newMatrix) |
|
||||
return this |
|
||||
} |
|
||||
|
|
||||
createBox3() { |
|
||||
const instancedMesh = this.manager.blocks[this.blockIndex]?.instancedMesh |
|
||||
const matrix = new THREE.Matrix4() |
|
||||
instancedMesh.getMatrixAt(this.meshIndex, matrix) |
|
||||
// 创建包围盒并应用矩阵
|
|
||||
const geometry = instancedMesh.geometry |
|
||||
//@ts-ignore
|
|
||||
const localBox = new THREE.Box3().setFromBufferAttribute(geometry.attributes.position) |
|
||||
return localBox.clone().applyMatrix4(matrix) |
|
||||
} |
|
||||
|
|
||||
createBox3Points() { |
|
||||
// 原始代码 =================================
|
|
||||
// const instancedMesh = this.manager.blocks[this.blockIndex]?.instancedMesh
|
|
||||
//
|
|
||||
// const matrix = new THREE.Matrix4()
|
|
||||
// instancedMesh.getMatrixAt(this.meshIndex, matrix)
|
|
||||
// // 创建包围盒并应用矩阵
|
|
||||
// const geometry = instancedMesh.geometry
|
|
||||
// //@ts-ignore
|
|
||||
// const localBox = new THREE.Box3().setFromBufferAttribute(geometry.attributes.position)
|
|
||||
// return localBox.clone().applyMatrix4(matrix)
|
|
||||
// =================================
|
|
||||
|
|
||||
// TF 代码 =================================
|
|
||||
const item = this.manager.viewport.entityManager.findItemById(this.uuid) |
|
||||
|
|
||||
const [pos, rot, scale] = item.tf |
|
||||
// // item.tf = [[8.5, 0.1, 2], [0, 0, 0], [1.5, 0.1, 2]]
|
|
||||
const matrix = new THREE.Matrix4() |
|
||||
matrix.compose( |
|
||||
new THREE.Vector3(pos[0], pos[1], pos[2]), |
|
||||
new THREE.Quaternion().setFromEuler(new THREE.Euler( |
|
||||
THREE.MathUtils.degToRad(rot[0]), |
|
||||
THREE.MathUtils.degToRad(rot[1]), |
|
||||
THREE.MathUtils.degToRad(rot[2]), |
|
||||
'XYZ' |
|
||||
)), |
|
||||
new THREE.Vector3(scale[0], scale[1], scale[2]) |
|
||||
) |
|
||||
// 获取 AABB 包围盒 ==========================
|
|
||||
// const aabb = new THREE.Box3()
|
|
||||
// .setFromCenterAndSize(new THREE.Vector3(0, 0, 0), new THREE.Vector3(1, 1, 1))
|
|
||||
// .applyMatrix4(matrix)
|
|
||||
// const min = aabb.min
|
|
||||
// const max = aabb.max
|
|
||||
// return [
|
|
||||
// new THREE.Vector3(min.x - Constract.YELLOW_EXPAND_AMOUNT, max.y + Constract.YELLOW_EXPAND_AMOUNT, min.z - Constract.YELLOW_EXPAND_AMOUNT),
|
|
||||
// new THREE.Vector3(max.x + Constract.YELLOW_EXPAND_AMOUNT, max.y + Constract.YELLOW_EXPAND_AMOUNT, min.z - Constract.YELLOW_EXPAND_AMOUNT),
|
|
||||
// new THREE.Vector3(max.x + Constract.YELLOW_EXPAND_AMOUNT, max.y + Constract.YELLOW_EXPAND_AMOUNT, max.z + Constract.YELLOW_EXPAND_AMOUNT),
|
|
||||
// new THREE.Vector3(min.x - Constract.YELLOW_EXPAND_AMOUNT, max.y + Constract.YELLOW_EXPAND_AMOUNT, max.z + Constract.YELLOW_EXPAND_AMOUNT)
|
|
||||
// ]
|
|
||||
|
|
||||
// 获取 OOBB 包围盒
|
|
||||
const halfSize = new THREE.Vector3().setFromMatrixScale(matrix).multiplyScalar(0.5) |
|
||||
const vertices = [ |
|
||||
new THREE.Vector3(-halfSize.x, -halfSize.y, -halfSize.z), |
|
||||
new THREE.Vector3(halfSize.x, -halfSize.y, -halfSize.z), |
|
||||
new THREE.Vector3(halfSize.x, halfSize.y, -halfSize.z), |
|
||||
new THREE.Vector3(-halfSize.x, halfSize.y, -halfSize.z), |
|
||||
new THREE.Vector3(-halfSize.x, -halfSize.y, halfSize.z), |
|
||||
new THREE.Vector3(halfSize.x, -halfSize.y, halfSize.z), |
|
||||
new THREE.Vector3(halfSize.x, halfSize.y, halfSize.z), |
|
||||
new THREE.Vector3(-halfSize.x, halfSize.y, halfSize.z) |
|
||||
] |
|
||||
|
|
||||
// 应用旋转 + 平移到每个顶点
|
|
||||
const verticesWorld = vertices.map(v => |
|
||||
v.clone().applyMatrix4(matrix) |
|
||||
) |
|
||||
|
|
||||
return verticesWorld |
|
||||
} |
|
||||
} |
|
||||
Loading…
Reference in new issue