64 lines
2.1 KiB
TypeScript
64 lines
2.1 KiB
TypeScript
import { Raycaster, Mesh, Group, Vector2, Scene, OrthographicCamera,Color } from "three";
|
|
|
|
export class PickHelper {
|
|
private raycaster: Raycaster;
|
|
private pickedObject: Mesh | Group | undefined;
|
|
private objects: Mesh[];
|
|
private camera: OrthographicCamera;
|
|
private scene: Scene;
|
|
private bloomLayer: number;
|
|
private pickedColor = new Color("#63E2B7");
|
|
private lastPickedColor = null;
|
|
|
|
constructor(scene: Scene, camera: OrthographicCamera, bloomLayer: number) {
|
|
this.camera = camera;
|
|
this.scene = scene;
|
|
|
|
this.bloomLayer = bloomLayer;
|
|
|
|
this.raycaster = new Raycaster();
|
|
|
|
this.objects = [];
|
|
|
|
this.scene.traverseVisible(child => {
|
|
if (child.children.length == 0) {
|
|
this.objects.push(<Mesh>child);
|
|
}
|
|
});
|
|
}
|
|
|
|
pick(normalizedPosition: Vector2,) {
|
|
// 透过截锥投射一道光线
|
|
this.raycaster.setFromCamera(normalizedPosition, this.camera);
|
|
// 获取射线相交的物体列表
|
|
const intersectedObjects = this.raycaster.intersectObjects(this.objects, false);
|
|
if (intersectedObjects.length) {
|
|
// 选择第一个对象。这是最近的一个
|
|
this.select(intersectedObjects[0].object as Mesh);
|
|
}
|
|
}
|
|
|
|
select(object: Mesh | Group) {
|
|
if (!object || object.uuid === this.pickedObject?.uuid) return;
|
|
|
|
// 如果有被选中的物体，则恢复
|
|
if (this.pickedObject) {
|
|
this.pickedObject.layers.toggle(this.bloomLayer);
|
|
|
|
if(this.pickedObject.material && this.lastPickedColor){
|
|
this.pickedObject.material.color = this.lastPickedColor;
|
|
this.lastPickedColor = null;
|
|
}
|
|
|
|
this.pickedObject = undefined;
|
|
}
|
|
|
|
this.pickedObject = object;
|
|
this.pickedObject.layers.toggle(this.bloomLayer);
|
|
|
|
if(this.pickedObject.material){
|
|
this.lastPickedColor = this.pickedObject.material.color;
|
|
this.pickedObject.material.color = this.pickedColor;
|
|
}
|
|
}
|
|
} |