14 changed files with 505 additions and 245 deletions
@ -0,0 +1,74 @@ |
|||||
|
import * as THREE from 'three' |
||||
|
import type Viewport from '@/core/engine/Viewport.ts' |
||||
|
|
||||
|
export default class InstanceMeshBlock { |
||||
|
public readonly name: string |
||||
|
public readonly blockIndex: number |
||||
|
public readonly viewport: Viewport |
||||
|
public readonly instancedMesh: THREE.InstancedMesh |
||||
|
public readonly freeIndices: number[] = [] |
||||
|
|
||||
|
// instanceId -> itemId
|
||||
|
public readonly __indexIdMap = new Map<number, string>() |
||||
|
|
||||
|
getFreeMeshIndex(): number { |
||||
|
if (this.freeIndices.length === 0) { |
||||
|
return -1 // No free index available
|
||||
|
} |
||||
|
return this.freeIndices.pop() // Return the last free index
|
||||
|
} |
||||
|
|
||||
|
constructor(itemTypeName: string, allowSelect: boolean, allowDrag: boolean, viewport: Viewport, |
||||
|
geometry: THREE.BufferGeometry, material: THREE.Material, |
||||
|
blockIndex: number, capacity: number) { |
||||
|
this.name = itemTypeName |
||||
|
this.blockIndex = blockIndex |
||||
|
this.viewport = viewport |
||||
|
this.instancedMesh = new THREE.InstancedMesh(geometry, material, capacity) |
||||
|
this.instancedMesh.instanceMatrix.setUsage(THREE.DynamicDrawUsage) |
||||
|
|
||||
|
console.log('createBlock: ' + itemTypeName + '[' + blockIndex + '] capacity:', capacity) |
||||
|
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 = itemTypeName |
||||
|
this.instancedMesh.userData.entityId = 'InstanceMeshBlock_' + itemTypeName + '_' + blockIndex |
||||
|
_.extend(this.instancedMesh.userData,{ |
||||
|
t: itemTypeName, |
||||
|
blockIndex: blockIndex |
||||
|
}) |
||||
|
|
||||
|
const dummy = new THREE.Object3D() |
||||
|
dummy.scale.set(0, 0, 0) |
||||
|
dummy.updateMatrix() |
||||
|
|
||||
|
for (let i = 0; i < capacity; i++) { |
||||
|
this.instancedMesh.setMatrixAt(i, dummy.matrix) |
||||
|
this.freeIndices.push(i) |
||||
|
} |
||||
|
|
||||
|
this.instancedMesh.instanceMatrix.needsUpdate = true |
||||
|
} |
||||
|
|
||||
|
dispose() { |
||||
|
this.viewport.scene.remove(this.instancedMesh) |
||||
|
this.instancedMesh.geometry.dispose() |
||||
|
|
||||
|
if (this.instancedMesh.material) { |
||||
|
if (Array.isArray(this.instancedMesh.material)) { |
||||
|
this.instancedMesh.material.forEach(mat => mat.dispose()) |
||||
|
} else { |
||||
|
this.instancedMesh.material.dispose() |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
this.instancedMesh.dispose() |
||||
|
this.__indexIdMap.clear() |
||||
|
this.freeIndices.length = 0 |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,151 @@ |
|||||
|
import * as THREE from 'three' |
||||
|
import type Viewport from '@/core/engine/Viewport.ts' |
||||
|
import InstanceMeshBlock from '@/core/manager/InstanceMeshBlock.ts' |
||||
|
import { PointManageWrap } from '@/core/manager/InstancePointManager.ts' |
||||
|
|
||||
|
export default class InstanceMeshManager { |
||||
|
private __uuidMap = new Map<string, InstanceMeshWrap>() |
||||
|
|
||||
|
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() |
||||
|
|
||||
|
constructor(name: string, viewport: Viewport, allowSelect: boolean, allowDrag: boolean, |
||||
|
geometry: THREE.BufferGeometry, material: THREE.Material) { |
||||
|
this.name = name |
||||
|
this.viewport = viewport |
||||
|
this.allowSelect = allowSelect |
||||
|
this.allowDrag = allowDrag |
||||
|
this.geometry = geometry |
||||
|
this.material = material |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 获取点实例数据 |
||||
|
*/ |
||||
|
findByMeshInstanceId(blockIndex: number, instanceId: number): InstanceMeshWrap { |
||||
|
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) |
||||
|
} |
||||
|
|
||||
|
create(entityId: string): InstanceMeshWrap { |
||||
|
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 InstanceMeshWrap(entityId, this, blockIndex, meshIndex) |
||||
|
} |
||||
|
|
||||
|
delete(wrap: InstanceMeshWrap) { |
||||
|
const block = this.blocks[wrap.blockIndex] |
||||
|
if (!block) { |
||||
|
console.warn(`InstanceMeshManager: Block ${wrap.blockIndex} not found for wrap ${wrap.uuid}`) |
||||
|
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(wrap.uuid) |
||||
|
block.__indexIdMap.delete(wrap.meshIndex) |
||||
|
|
||||
|
wrap.dispose() |
||||
|
} |
||||
|
|
||||
|
// 创建新的 InstanceMeshBlock
|
||||
|
createBlock(): InstanceMeshBlock { |
||||
|
const blockIndex = this.blocks.length |
||||
|
const block = new InstanceMeshBlock(this.name, this.allowSelect, this.allowDrag, |
||||
|
this.viewport, this.geometry, this.material, |
||||
|
blockIndex, this.blockCapacity) |
||||
|
this.blocks.push(block) |
||||
|
return block |
||||
|
} |
||||
|
|
||||
|
setBlockMatrixAt(wrap: InstanceMeshWrap, matrix: THREE.Matrix4) { |
||||
|
const block = this.blocks[wrap.blockIndex] |
||||
|
if (!block) { |
||||
|
console.warn(`InstanceMeshManager: Block ${wrap.blockIndex} not found!`) |
||||
|
return |
||||
|
} |
||||
|
this.__uuidMap.set(wrap.uuid, wrap) |
||||
|
block.instancedMesh.setMatrixAt(wrap.meshIndex, matrix) |
||||
|
wrap.parent = block.instancedMesh |
||||
|
wrap.visible = true // 默认可见
|
||||
|
block.instancedMesh.instanceMatrix.needsUpdate = true |
||||
|
} |
||||
|
|
||||
|
dispose() { |
||||
|
for (const block of this.blocks) { |
||||
|
block.dispose() |
||||
|
} |
||||
|
this.blocks.length = 0 // 清空 blocks 数组
|
||||
|
this.geometry.dispose() // 释放几何体资源
|
||||
|
this.material.dispose() // 释放材质资源
|
||||
|
console.log(`InstanceMeshManager ${this.name} disposed.`) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
export class InstanceMeshWrap { |
||||
|
readonly entityId: string |
||||
|
readonly manager: InstanceMeshManager |
||||
|
readonly blockIndex: number |
||||
|
readonly meshIndex: number |
||||
|
|
||||
|
uuid: string |
||||
|
parent: THREE.Object3D | null = null |
||||
|
name: string |
||||
|
visible: boolean |
||||
|
|
||||
|
constructor(entityId: string, manager: InstanceMeshManager, blockIndex: number, meshIndex: number) { |
||||
|
this.uuid = system.createUUID() |
||||
|
this.entityId = entityId |
||||
|
this.manager = manager |
||||
|
this.meshIndex = meshIndex |
||||
|
this.blockIndex = blockIndex |
||||
|
} |
||||
|
|
||||
|
syncMatrix(matrix: THREE.Matrix4): InstanceMeshWrap { |
||||
|
this.manager.setBlockMatrixAt(this, matrix) |
||||
|
return this |
||||
|
} |
||||
|
|
||||
|
dispose() { |
||||
|
this.manager.delete(this) |
||||
|
this.parent = null |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue