#
Junjie
2024-04-03 ff2b7ebf99e8aedccb3859ed9f049e49a9a613f1
zy-asrs-flow/src/pages/map/drawer/shelf/shelfThree.js
@@ -2,8 +2,15 @@
import Stats from 'three/addons/libs/stats.module.js';
import { OrbitControls } from 'three/examples/jsm/controls/OrbitControls'
import TWEEN from '@tweenjs/tween.js';
import { EffectComposer } from 'three/examples/jsm/postprocessing/EffectComposer';
import { RenderPass } from 'three/examples/jsm/postprocessing/RenderPass';
import { OutlinePass } from 'three/examples/jsm/postprocessing/OutlinePass';
import { ShaderPass } from 'three/examples/jsm/postprocessing/ShaderPass';
import { FXAAShader } from 'three/examples/jsm/shaders/FXAAShader';
import { FBXLoader } from 'three/examples/jsm/loaders/FBXLoader'
import * as Utils from '../../utils'
const help = true;
const help = false;
export default class ShelfThree {
@@ -19,12 +26,12 @@
        this.renderer = this.initRenderer();
        this.controls = this.initControls();
        this.stats = this.initStats();
        this.composer = this.initComposer();
        this.windowResize();
        this.initRaycaster();
        initLight(this.scene);
        this.animate();
        this.build();
    }
    animate = () => {
@@ -34,8 +41,8 @@
    }
    render = () => {
        this.renderer.shadowMap.enabled = true;
        this.renderer.render(this.scene, this.camera);
        // this.renderer.render(this.scene, this.camera);
        this.composer.render(); // use special effects
        this.stats.update();
        // Rotate the scene
        this.controls.update()
@@ -58,7 +65,10 @@
    initScene = () => {
        const scene = new THREE.Scene();
        // scene.background = new THREE.Color(0xf0f0f0);
        scene.background = new THREE.Color(0x333333);
        // scene.background = new THREE.Color(0x333333);
        // scene.background = new THREE.Color(0xf8f8f8);
        scene.background = new THREE.Color(0x7a7a7a);
        if (help) {
            scene.add(new THREE.AxesHelper(1000));
        }
@@ -67,7 +77,7 @@
    initCamera = () => {
        const camera = new THREE.PerspectiveCamera(70, this.getFullWidth() / this.getFullHeight(), 1, 60000);
        camera.position.set(-300, 300, 300);
        camera.position.set(-400, 400, 400);
        this.scene.add(camera);
        return camera;
    }
@@ -86,6 +96,27 @@
        return renderer;
    }
    initComposer = () => {
        const composer = new EffectComposer(this.renderer);
        const renderPass = new RenderPass(this.scene, this.camera);
        composer.addPass(renderPass);
        // pixel ratio
        const pixelRatio = this.renderer.getPixelRatio();
        const newWidth = Math.floor(this.getFullWidth() * pixelRatio) || 1;
        const newHeight = Math.floor(this.getFullHeight() * pixelRatio) || 1;
        composer.setSize(newWidth, newHeight);
        // antialias
        const effectFXAA = new ShaderPass(FXAAShader);
        effectFXAA.uniforms['resolution'].value.set(1 / newWidth, 1 / newHeight);
        composer.addPass(effectFXAA);
        // outline
        this.outlinePass = new OutlinePass(new THREE.Vector2(this.getFullWidth(), this.getFullHeight()), this.scene, this.camera);
        this.outlinePass.visibleEdgeColor.set('#ffffff'); // outline color
        this.outlinePass.edgeStrength = 10; // outline weight
        composer.addPass(this.outlinePass);
        return composer;
    }
    initControls = () => {
        const controls = new OrbitControls(this.camera, this.renderer.domElement);
        // controls.enableRotate = false; // 禁用整体旋转功能
@@ -100,7 +131,7 @@
        controls.rotateSpeed = 0.6; // 视角移动速度减慢
        controls.autoRotate = true; // 自动旋转
        controls.target = new THREE.Vector3(0, 0, 0)  // 平替 camera的lookAt
        controls.target = new THREE.Vector3(0, 250, 0)  // 屏幕中心点 平替 camera的lookAt
        return controls;
    }
@@ -123,14 +154,9 @@
    }
    initRaycaster = () => {
        let that = this;
        const raycaster = new THREE.Raycaster();
        const mouse = new THREE.Vector2();
        this.dom.addEventListener("click", (event) => {
            this.handleClick("null");
        })
        this.dom.addEventListener("dblclick", (event) => {
        this.handleClickEvent = (event) => {
            let x, y;
            if (event.changedTouches) {
                x = event.changedTouches[0].pageX;
@@ -139,36 +165,58 @@
                x = event.clientX;
                y = event.clientY;
            }
            let rect = this.dom.getBoundingClientRect();
            mouse.x = ((x - rect.left) / rect.width) * 2 - 1;
            mouse.y = -((y - rect.top) / rect.height) * 2 + 1;
            event.preventDefault();
            mouse.x = (x / window.innerWidth) * 2 - 1;
            mouse.y = -(y / window.innerHeight) * 2 + 1;
            raycaster.setFromCamera(mouse, that.camera);
            let intersects = raycaster.intersectObjects(that.objects, true);
            raycaster.setFromCamera(mouse, this.camera);
            let intersects = raycaster.intersectObjects(this.objects, true);
            if (intersects.length === 0) {
                this.handleClick("null");
                return;
            }
            let objName = intersects[0].object.name;
            if (objName.startsWith("locNo")) {
                if (this.handleClick) {
                    this.handleClick(objName, x, y);
            this.objects.forEach(obj => {
                if (obj.name === objName) {
                    this.outlinePass.selectedObjects = [obj];
                }
            } else {
                this.handleClick("null");
            })
            if (objName && this.handleClick) {
                this.handleClick(objName, x, y);
            }
        }, false);
        }
        this.dom.addEventListener("click", this.handleClickEvent, false);
    }
    build = () => {
        buildShelf(this.scene);
        buildPallet(this.scene);
    generateMesh = (fn) => {
        // fn(this.addObject);
        const loader = new FBXLoader();
        const promise = new Promise((resolve) => {
            loader.load('model/001-5.fbx', (mesh) => {
                mesh.position.set(0, 10, 0);
                mesh.scale.set(5, 5, 5);
                mesh.name = '1-1-1';
                // mesh.rotation.x = Utils.rotationParseNum(270);
                mesh.traverse(function (child) {
                    if (child.isMesh) {
                        child.name = '1-1-1';
                        child.castShadow = true;
                        child.receiveShadow = true;
                    }
                });
                this.addObject(mesh);
                resolve();
            })
        })
        return promise;
    }
    handleClick = () => {
    setNewSelectedMesh = (objName) => {
        for (const mesh of this.objects) {
            if (mesh.name === objName) {
                this.outlinePass.selectedObjects = [mesh];
                break;
            }
        }
    }
    destroy = () => {
@@ -187,10 +235,19 @@
            this.renderer.domElement = null;
            this.renderer = null;
        }
        if (this.composer) {
            this.composer.passes.forEach(pass => {
                if (pass.dispose) {
                    pass.dispose();
                }
            });
            this.composer = null;
        }
        if (this.controls) {
            this.controls.dispose();
            this.controls = null;
        }
        this.dom.removeEventListener("click", this.handleClickEvent);
        this.camera = null;
        this.objects = [];
        while (this.dom?.firstChild) {
@@ -208,14 +265,14 @@
const initLight = (scene) => {
    const directionalLight = new THREE.DirectionalLight(0xffffff, 8);
    directionalLight.position.set(-300, 300, 300);
    directionalLight.position.set(1, 1, 1);
    scene.add(directionalLight);
    const ambientLight = new THREE.AmbientLight(0xf0f0f0, 8);
    const ambientLight = new THREE.AmbientLight(0xf0f0f0, 10);
    scene.add(ambientLight);
    const spotLight = new THREE.SpotLight(0xffffff, 8);
    spotLight.position.set(-300, 300, 0);
    spotLight.position.set(-300, 1000, 0);
    spotLight.angle = Math.PI / 4;  // 角度
    spotLight.distance = 800;  // 距离
    spotLight.decay = 0;    // 光衰
@@ -232,26 +289,4 @@
    if (help) {
        scene.add(new THREE.SpotLightHelper(spotLight));
    }
}
const buildShelf = (scene) => {
    const boxGeometry = new THREE.BoxGeometry(100, 40, 100);
    const boxMaterial = new THREE.MeshStandardMaterial({
        color: '#222f3e',
    });
    const boxMesh = new THREE.Mesh(boxGeometry, boxMaterial);
    boxMesh.position.set(0, 20, 0);
    scene.add(boxMesh)
    return boxMesh;
}
const buildPallet = (scene) => {
    const boxGeometry = new THREE.BoxGeometry(100, 60, 100);
    const boxMaterial = new THREE.MeshStandardMaterial({
        color: '#b33939',
    });
    const boxMesh = new THREE.Mesh(boxGeometry, boxMaterial);
    boxMesh.position.set(0, 70, 0);
    scene.add(boxMesh)
    return boxMesh;
}
}