immersive2/src/controllers/base.ts

import {
    AbstractMesh,
    HavokPlugin,
    PhysicsMotionType,
    Scene,
    TransformNode,
    Vector3,
    WebXRControllerComponent,
    WebXRDefaultExperience,
    WebXRInputSource
} from "@babylonjs/core";
import {MeshConverter} from "../diagram/meshConverter";
import {DiagramManager} from "../diagram/diagramManager";
import {DiagramEvent, DiagramEventType} from "../diagram/diagramEntity";
import log from "loglevel";
import {Controllers} from "./controllers";


export class Base {
    static stickVector = Vector3.Zero();
    protected controller: WebXRInputSource;
    protected speedFactor = 4;
    protected readonly scene: Scene;
    protected grabbedMesh: AbstractMesh = null;
    protected grabbedMeshParentId: string = null;
    protected previousParentId: string = null;
    protected previousRotation: Vector3 = null;
    protected previousScaling: Vector3 = null;
    protected previousPosition: Vector3 = null;

    protected readonly xr: WebXRDefaultExperience;
    protected readonly diagramManager: DiagramManager;
    private logger: log.Logger;
    private lastPosition: Vector3 = null;

    constructor(controller: WebXRInputSource,
                scene: Scene,
                xr: WebXRDefaultExperience,
                diagramManager: DiagramManager) {
        this.logger = log.getLogger('Base');
        this.controller = controller;

        this.scene = scene;


        this.scene.onBeforeRenderObservable.add(() => {
            if (this?.grabbedMesh?.physicsBody) {
                const hk = (this.scene.getPhysicsEngine().getPhysicsPlugin() as HavokPlugin);
                this.lastPosition = this?.grabbedMesh?.physicsBody?.transformNode.absolutePosition.clone();
                if (this.grabbedMeshParentId) {
                    const parent = this.scene.getTransformNodeById(this.grabbedMeshParentId);
                    if (parent) {
                        hk.setPhysicsBodyTransformation(this.grabbedMesh.physicsBody, parent);
                        hk.sync(this.grabbedMesh.physicsBody);
                    } else {
                        this.logger.error("parent not found for " + this.grabbedMeshParentId);

                    }

                } else {
                    this.logger.warn("no parent id");
                }

            }
        }, -1, false, this);
        this.xr = xr;
        this.diagramManager = diagramManager;

        this.controller.onMotionControllerInitObservable.add((init) => {
            if (init.components['xr-standard-trigger']) {
                init.components['xr-standard-trigger']
                    .onButtonStateChangedObservable
                    .add(() => {

                    });
            }
            this.initGrip(init.components['xr-standard-squeeze']);
        });
        Controllers.controllerObserver.add((event) => {
            if (event.type == 'pulse') {
                this.logger.debug(event);
                if (event.gripId == this?.controller?.grip?.id) {
                    this.controller?.motionController?.pulse(.25, 30);
                }
            }
        });

    }

    public disable() {
        this.controller.motionController.rootMesh.setEnabled(false);
        this.controller.pointer.setEnabled(false);
    }

    public enable() {
        this.controller.motionController.rootMesh.setEnabled(true);
        this.controller.pointer.setEnabled(true);
    }

    private buildTransformNode() {

    }

    private setupTransformNode(mesh: TransformNode) {
        const transformNode = new TransformNode("grabAnchor, this.scene");
        transformNode.id = "grabAnchor";
        transformNode.position = mesh.position.clone();
        transformNode.rotationQuaternion = mesh.rotationQuaternion.clone();
        transformNode.setParent(this.controller.motionController.rootMesh);
        return transformNode;
    }

    private grab(mesh: AbstractMesh) {

        if (this.xr.pointerSelection.getMeshUnderPointer) {
            mesh = this.xr.pointerSelection.getMeshUnderPointer(this.controller.uniqueId);
        }
        if (!mesh) {
            return;
        }
        if (!mesh?.metadata?.template) {
            if (mesh?.id == "handle") {
                mesh && mesh.setParent(this.controller.motionController.rootMesh);
                this.grabbedMesh = mesh;
            } else {
                return;
            }

        }
        this.previousParentId = mesh?.parent?.id;
        this.previousRotation = mesh?.rotation.clone();
        this.previousScaling = mesh?.scaling.clone();
        this.previousPosition = mesh?.position.clone();

        if ("toolbox" != mesh?.parent?.parent?.id) {
            if (mesh.physicsBody) {
                const transformNode = this.setupTransformNode(mesh);
                mesh.physicsBody.setMotionType(PhysicsMotionType.ANIMATED);
                this.grabbedMeshParentId = transformNode.id;
            } else {
                mesh.setParent(this.controller.motionController.rootMesh);
            }
            this.grabbedMesh = mesh;
        } else {
            const newMesh = this.diagramManager.createCopy(mesh);
            const transformNode = new TransformNode("grabAnchor, this.scene");
            transformNode.id = "grabAnchor";
            transformNode.position = newMesh.position.clone();
            if (newMesh.rotationQuaternion) {
                transformNode.rotationQuaternion = newMesh.rotationQuaternion.clone();
            } else {
                transformNode.rotation = newMesh.rotation.clone();
            }
            transformNode.setParent(this.controller.motionController.rootMesh);
            this.grabbedMeshParentId = transformNode.id;
            MeshConverter
                .applyPhysics(newMesh, this.scene)
                .setMotionType(PhysicsMotionType.ANIMATED);


            //newMesh && newMesh.setParent(this.controller.motionController.rootMesh);
            this.grabbedMesh = newMesh;
            this.previousParentId = null;
            const event: DiagramEvent = {
                type: DiagramEventType.ADD,
                entity: MeshConverter.toDiagramEntity(newMesh)
            }
            this.diagramManager.onDiagramEventObservable.notifyObservers(event);

        }
    }

    private handleGrabbed(mesh: AbstractMesh): boolean {
        if (!mesh?.metadata?.template
            && mesh?.id == "handle") {
            //mesh && mesh.setParent(null);
            this.grabbedMesh = null;
            this.previousParentId = null;
            mesh.setParent(null);
            return true;
        } else {
            return false;
        }
    }

    private reparent(mesh: AbstractMesh) {
        if (this.previousParentId) {
            const parent = this.scene.getMeshById(this.previousParentId);
            if (parent) {
                //mesh && mesh.setParent(this.scene.getMeshById(this.previousParentId));
                log.getLogger("Base").warn("Base", "Have not implemented snapping to parent yet");
                //@note: this is not implemented yet
            } else {
                mesh.setParent(null);
            }
        } else {
            const parent = this.scene.getTransformNodeById(this.grabbedMeshParentId);
            if (parent) {
                this.grabbedMeshParentId = null;
                parent.dispose();
            }
        }
    }

    private drop() {
        const mesh = this.grabbedMesh;
        if (!mesh) {
            return;
        }
        if (this.handleGrabbed(mesh)) {
            return;
        }

        this.reparent(mesh);
        this.previousParentId = null;
        this.previousScaling = null;
        this.previousRotation = null;
        this.previousPosition = null;
        this.grabbedMesh = null;

        const entity = MeshConverter.toDiagramEntity(mesh);
        const event: DiagramEvent = {
            type: DiagramEventType.DROP,
            entity: entity
        }

        this.diagramManager.onDiagramEventObservable.notifyObservers(event);
        const body = mesh?.physicsBody;
        if (body) {
            body.setMotionType(PhysicsMotionType.DYNAMIC);
            this.logger.debug(body.transformNode.absolutePosition);
            this.logger.debug(this.lastPosition);
            if (this.lastPosition) {
                body.setLinearVelocity(body.transformNode.absolutePosition.subtract(this.lastPosition).scale(20));
                //body.setLinearVelocity(this.lastPosition.subtract(body.transformNode.absolutePosition).scale(20));
                this.logger.debug(this.lastPosition.subtract(body.transformNode.absolutePosition).scale(20));
            }
        }
    }

    private initGrip(grip: WebXRControllerComponent) {
        grip.onButtonStateChangedObservable.add(() => {
            if (grip.changes.pressed) {
                if (grip.pressed) {
                    this.grab(this.scene.meshUnderPointer);
                } else {
                    this.drop();
                }
            }
        });
    }

}