Babylon.js/src/Behaviors/Mesh/babylon.pointerDragBehavior.ts

module BABYLON {
    /**
     * A behavior that when attached to a mesh will allow the mesh to be dragged around the screen based on pointer events
     */
    export class PointerDragBehavior implements Behavior<Mesh> {
        private static _AnyMouseID = -2;
        private _attachedNode: Mesh; 
        private _dragPlane: Mesh;
        private _scene:Scene;
        private _pointerObserver:Nullable<Observer<PointerInfo>>;
        private _beforeRenderObserver:Nullable<Observer<Scene>>;
        private static _planeScene:Scene;
        /**
         * The maximum tolerated angle between the drag plane and dragging pointer rays to trigger pointer events. Set to 0 to allow any angle (default: 0)
         */
        public maxDragAngle = 0;
        /**
         * @hidden
         */
        public _useAlternatePickedPointAboveMaxDragAngle = false;
        /**
         * The id of the pointer that is currently interacting with the behavior (-1 when no pointer is active)
         */
        public currentDraggingPointerID = -1;
        /**
         * The last position where the pointer hit the drag plane in world space
         */
        public lastDragPosition:Vector3;
        /**
         * If the behavior is currently in a dragging state
         */
        public dragging = false;
        /**
         * The distance towards the target drag position to move each frame. This can be useful to avoid jitter. Set this to 1 for no delay. (Default: 0.2)
         */
        public dragDeltaRatio = 0.2;
        /**
         * If the drag plane orientation should be updated during the dragging (Default: true)
         */
        public updateDragPlane = true;
        // Debug mode will display drag planes to help visualize behavior
        private _debugMode = false;
        private _moving = false;
        /**
         *  Fires each time the attached mesh is dragged with the pointer
         *  * delta between last drag position and current drag position in world space
         *  * dragDistance along the drag axis
         *  * dragPlaneNormal normal of the current drag plane used during the drag
         *  * dragPlanePoint in world space where the drag intersects the drag plane
         */
        public onDragObservable = new Observable<{delta:Vector3, dragPlanePoint:Vector3, dragPlaneNormal:Vector3, dragDistance:number, pointerId:number}>()
        /**
         *  Fires each time a drag begins (eg. mouse down on mesh)
         */
        public onDragStartObservable = new Observable<{dragPlanePoint:Vector3, pointerId:number}>()
        /**
         *  Fires each time a drag ends (eg. mouse release after drag)
         */
        public onDragEndObservable = new Observable<{dragPlanePoint:Vector3, pointerId:number}>()
        /**
         *  If the attached mesh should be moved when dragged
         */
        public moveAttached = true;

        /**
         *  If the drag behavior will react to drag events (Default: true)
         */
        public enabled = true;
        /**
         * If camera controls should be detached during the drag
         */
        public detachCameraControls = true;
        
        /**
         * If set, the drag plane/axis will be rotated based on the attached mesh's world rotation (Default: true)
         */
        public useObjectOrienationForDragging = true;

        private _options:{dragAxis?:Vector3, dragPlaneNormal?:Vector3};
        /**
         * Creates a pointer drag behavior that can be attached to a mesh
         * @param options The drag axis or normal of the plane that will be dragged across. If no options are specified the drag plane will always face the ray's origin (eg. camera)
         */
        constructor(options?:{dragAxis?:Vector3, dragPlaneNormal?:Vector3}){
            this._options = options ? options : {};
            
            var optionCount = 0;
            if(this._options.dragAxis){
                optionCount++;
            }
            if(this._options.dragPlaneNormal){
                optionCount++;
            }
            if(optionCount > 1){
                throw "Multiple drag modes specified in dragBehavior options. Only one expected";
            }
        }

        /**
         *  The name of the behavior
         */
        public get name(): string {
            return "PointerDrag";
        }

        /**
         *  Initializes the behavior
         */
        public init() {}

        private _tmpVector = new Vector3(0,0,0);
        private _alternatePickedPoint = new Vector3(0,0,0);
        private _worldDragAxis = new Vector3(0,0,0);
        private _targetPosition = new BABYLON.Vector3(0,0,0);
        private _attachedElement:Nullable<HTMLElement> = null;
        /**
         * Attaches the drag behavior the passed in mesh
         * @param ownerNode The mesh that will be dragged around once attached
         */
        public attach(ownerNode: Mesh): void {
            this._scene = ownerNode.getScene();
            this._attachedNode = ownerNode;

            // Initialize drag plane to not interfere with existing scene
            if(!PointerDragBehavior._planeScene){
                if(this._debugMode){
                    PointerDragBehavior._planeScene = this._scene;
                }else{
                    PointerDragBehavior._planeScene = new BABYLON.Scene(this._scene.getEngine());
                    PointerDragBehavior._planeScene.detachControl();
                    this._scene.getEngine().scenes.pop();
                    this._scene.onDisposeObservable.addOnce(()=>{
                        PointerDragBehavior._planeScene.dispose();
                        (<any>PointerDragBehavior._planeScene) = null;
                    })
                }
            }
            this._dragPlane = BABYLON.Mesh.CreatePlane("pointerDragPlane", this._debugMode ? 1 : 10000, PointerDragBehavior._planeScene, false, BABYLON.Mesh.DOUBLESIDE);

            // State of the drag
            this.lastDragPosition = new BABYLON.Vector3(0,0,0);

            var pickPredicate = (m:AbstractMesh)=>{
                return this._attachedNode == m || m.isDescendantOf(this._attachedNode)
            }

            this._pointerObserver = this._scene.onPointerObservable.add((pointerInfo, eventState)=>{
                if(!this.enabled){
                    return;
                }
                
                if (pointerInfo.type == BABYLON.PointerEventTypes.POINTERDOWN) {

                    if(!this.dragging && pointerInfo.pickInfo && pointerInfo.pickInfo.hit && pointerInfo.pickInfo.pickedMesh && pointerInfo.pickInfo.pickedPoint && pointerInfo.pickInfo.ray && pickPredicate(pointerInfo.pickInfo.pickedMesh)){
                        this._startDrag((<PointerEvent>pointerInfo.event).pointerId, pointerInfo.pickInfo.ray, pointerInfo.pickInfo.pickedPoint);
                    }
                }else if(pointerInfo.type == BABYLON.PointerEventTypes.POINTERUP){
                    if(this.currentDraggingPointerID == (<PointerEvent>pointerInfo.event).pointerId){
                        this.releaseDrag();
                    }
                }else if(pointerInfo.type == BABYLON.PointerEventTypes.POINTERMOVE){
                    var pointerId = (<PointerEvent>pointerInfo.event).pointerId;

                    // If drag was started with anyMouseID specified, set pointerID to the next mouse that moved
                    if(this.currentDraggingPointerID === PointerDragBehavior._AnyMouseID && pointerId !== PointerDragBehavior._AnyMouseID && (<PointerEvent>pointerInfo.event).pointerType == "mouse"){
                        if(this._lastPointerRay[this.currentDraggingPointerID]){
                            this._lastPointerRay[pointerId] = this._lastPointerRay[this.currentDraggingPointerID];
                            delete this._lastPointerRay[this.currentDraggingPointerID];
                        }
                        this.currentDraggingPointerID = pointerId;
                    }
                    
                    // Keep track of last pointer ray, this is used simulating the start of a drag in startDrag()
                    if(!this._lastPointerRay[pointerId]){
                        this._lastPointerRay[pointerId] = new BABYLON.Ray(new BABYLON.Vector3(), new BABYLON.Vector3());
                    }
                    if(pointerInfo.pickInfo && pointerInfo.pickInfo.ray){
                        this._lastPointerRay[pointerId].origin.copyFrom(pointerInfo.pickInfo.ray.origin);
                        this._lastPointerRay[pointerId].direction.copyFrom(pointerInfo.pickInfo.ray.direction);

                        if(this.currentDraggingPointerID == pointerId && this.dragging){
                            this._moveDrag(pointerInfo.pickInfo.ray);
                         }
                    }
                }
            });

            this._beforeRenderObserver = this._scene.onBeforeRenderObservable.add(()=>{
                if(this._moving && this.moveAttached){
                    BoundingBoxGizmo._RemoveAndStorePivotPoint(this._attachedNode);
                    // Slowly move mesh to avoid jitter
                    this._targetPosition.subtractToRef((this._attachedNode).absolutePosition, this._tmpVector);
                    this._tmpVector.scaleInPlace(this.dragDeltaRatio);
                    (this._attachedNode).getAbsolutePosition().addToRef(this._tmpVector, this._tmpVector);
                    (this._attachedNode).setAbsolutePosition(this._tmpVector);
                    BoundingBoxGizmo._RestorePivotPoint(this._attachedNode);
                }
            });
        }

        /**
         * Force relase the drag action by code.
         */
        public releaseDrag(){
            this.dragging = false;
            this.onDragEndObservable.notifyObservers({dragPlanePoint: this.lastDragPosition, pointerId: this.currentDraggingPointerID});
            this.currentDraggingPointerID = -1;
            this._moving = false;

            // Reattach camera controls
            if(this.detachCameraControls && this._attachedElement && this._scene.activeCamera && !this._scene.activeCamera.leftCamera){
                this._scene.activeCamera.attachControl(this._attachedElement, true);
            }
        }

        private _startDragRay = new BABYLON.Ray(new BABYLON.Vector3(), new BABYLON.Vector3());
        private _lastPointerRay:{[key: number]: Ray} = {};
        /**
         * Simulates the start of a pointer drag event on the behavior
         * @param pointerId pointerID of the pointer that should be simulated (Default: Any mouse pointer ID)
         * @param fromRay initial ray of the pointer to be simulated (Default: Ray from camera to attached mesh)
         * @param startPickedPoint picked point of the pointer to be simulated (Default: attached mesh position)
         */
        public startDrag(pointerId:number = PointerDragBehavior._AnyMouseID, fromRay?:Ray, startPickedPoint?:Vector3){
            this._startDrag(pointerId, fromRay, startPickedPoint);
            
            var lastRay =  this._lastPointerRay[pointerId];
            if(pointerId === PointerDragBehavior._AnyMouseID){
                lastRay = this._lastPointerRay[<any>Object.keys(this._lastPointerRay)[0]];
            }
            
            if(lastRay){
                // if there was a last pointer ray drag the object there
                this._moveDrag(lastRay);
            }
        }

        private _startDrag(pointerId:number, fromRay?:Ray, startPickedPoint?:Vector3){
            if(!this._scene.activeCamera || this.dragging || !this._attachedNode){
                return;
            }
            
            BoundingBoxGizmo._RemoveAndStorePivotPoint(this._attachedNode);
            // Create start ray from the camera to the object
            if(fromRay){
                this._startDragRay.direction.copyFrom(fromRay.direction)
                this._startDragRay.origin.copyFrom(fromRay.origin)
            }else{
                this._startDragRay.origin.copyFrom(this._scene.activeCamera.position);
                this._attachedNode.getWorldMatrix().getTranslationToRef(this._tmpVector);
                this._tmpVector.subtractToRef(this._scene.activeCamera.position, this._startDragRay.direction);
            }
            
            this._updateDragPlanePosition(this._startDragRay, startPickedPoint?startPickedPoint:this._tmpVector);
           
            var pickedPoint = this._pickWithRayOnDragPlane(this._startDragRay);
            if(pickedPoint){
                this.dragging = true;
                this.currentDraggingPointerID = pointerId;
                this.lastDragPosition.copyFrom(pickedPoint);
                this.onDragStartObservable.notifyObservers({dragPlanePoint: pickedPoint, pointerId: this.currentDraggingPointerID});
                this._targetPosition.copyFrom((this._attachedNode).absolutePosition)

                // Detatch camera controls
                if(this.detachCameraControls && this._scene.activeCamera && !this._scene.activeCamera.leftCamera){
                    if(this._scene.activeCamera.inputs.attachedElement){
                        this._attachedElement = this._scene.activeCamera.inputs.attachedElement;
                        this._scene.activeCamera.detachControl(this._scene.activeCamera.inputs.attachedElement);
                    }else{
                        this._attachedElement = null;
                    }
                }
            }
            BoundingBoxGizmo._RestorePivotPoint(this._attachedNode);
        }

        private _dragDelta = new BABYLON.Vector3();
        private _moveDrag(ray:Ray){
            this._moving = true;
            var pickedPoint = this._pickWithRayOnDragPlane(ray);
            
            if (pickedPoint) {
                if(this.updateDragPlane){
                    this._updateDragPlanePosition(ray, pickedPoint);
                }
                
                var dragLength = 0;
                // depending on the drag mode option drag accordingly
                if(this._options.dragAxis){
                    // Convert local drag axis to world
                    Vector3.TransformCoordinatesToRef(this._options.dragAxis, this._attachedNode.getWorldMatrix().getRotationMatrix(), this._worldDragAxis);

                    // Project delta drag from the drag plane onto the drag axis
                    pickedPoint.subtractToRef(this.lastDragPosition, this._tmpVector);
                    dragLength = BABYLON.Vector3.Dot(this._tmpVector, this._worldDragAxis)
                    this._worldDragAxis.scaleToRef(dragLength, this._dragDelta);
                }else{
                    dragLength = this._dragDelta.length();
                    pickedPoint.subtractToRef(this.lastDragPosition, this._dragDelta);
                }
                this._targetPosition.addInPlace(this._dragDelta);
                this.onDragObservable.notifyObservers({dragDistance: dragLength, delta: this._dragDelta, dragPlanePoint: pickedPoint, dragPlaneNormal: this._dragPlane.forward, pointerId: this.currentDraggingPointerID});
                this.lastDragPosition.copyFrom(pickedPoint);
            }
        }

        private _pickWithRayOnDragPlane(ray:Nullable<Ray>){
            if(!ray){
                return null;
            }

            // Calculate angle between plane normal and ray
            var angle = Math.acos(Vector3.Dot(this._dragPlane.forward, ray.direction));
            // Correct if ray is casted from oposite side
            if(angle > Math.PI/2){
                angle = Math.PI - angle;
            }

            // If the angle is too perpendicular to the plane pick another point on the plane where it is looking
            if(this.maxDragAngle > 0 && angle > this.maxDragAngle){
                if(this._useAlternatePickedPointAboveMaxDragAngle){
                    // Invert ray direction along the towards object axis
                    this._tmpVector.copyFrom(ray.direction);
                    (this._attachedNode).absolutePosition.subtractToRef(ray.origin, this._alternatePickedPoint);
                    this._alternatePickedPoint.normalize();
                    this._alternatePickedPoint.scaleInPlace(-2*Vector3.Dot(this._alternatePickedPoint, this._tmpVector));
                    this._tmpVector.addInPlace(this._alternatePickedPoint);
                    
                    // Project resulting vector onto the drag plane and add it to the attached nodes absolute position to get a picked point
                    var dot = Vector3.Dot(this._dragPlane.forward, this._tmpVector);
                    this._dragPlane.forward.scaleToRef(-dot, this._alternatePickedPoint);
                    this._alternatePickedPoint.addInPlace(this._tmpVector);
                    this._alternatePickedPoint.addInPlace((this._attachedNode).absolutePosition);
                    return this._alternatePickedPoint
                }else{
                    return null;
                }
            }

            var pickResult = PointerDragBehavior._planeScene.pickWithRay(ray, (m)=>{return m == this._dragPlane})
            if (pickResult && pickResult.hit && pickResult.pickedMesh && pickResult.pickedPoint) {
                return pickResult.pickedPoint;
            }else{
                return null;
            }
        }

        // Variables to avoid instantiation in the below method
        private _pointA = new Vector3(0,0,0);
        private _pointB = new Vector3(0,0,0);
        private _pointC = new Vector3(0,0,0);
        private _lineA = new Vector3(0,0,0);
        private _lineB = new Vector3(0,0,0);
        private _localAxis = new Vector3(0,0,0);
        private _lookAt = new Vector3(0,0,0);
        // Position the drag plane based on the attached mesh position, for single axis rotate the plane along the axis to face the camera
        private _updateDragPlanePosition(ray:Ray, dragPlanePosition:Vector3){
            this._pointA.copyFrom(dragPlanePosition);
            if(this._options.dragAxis){
                this.useObjectOrienationForDragging ? Vector3.TransformCoordinatesToRef(this._options.dragAxis, this._attachedNode.getWorldMatrix().getRotationMatrix(), this._localAxis) : this._localAxis.copyFrom(this._options.dragAxis);

                // Calculate plane normal in direction of camera but perpendicular to drag axis
                this._pointA.addToRef(this._localAxis, this._pointB); // towards drag axis
                ray.origin.subtractToRef(this._pointA, this._pointC)
                this._pointA.addToRef(this._pointC.normalize(), this._pointC); // towards camera
                // Get perpendicular line from direction to camera and drag axis
                this._pointB.subtractToRef(this._pointA, this._lineA);
                this._pointC.subtractToRef(this._pointA, this._lineB);
                BABYLON.Vector3.CrossToRef(this._lineA, this._lineB, this._lookAt);
                // Get perpendicular line from previous result and drag axis to adjust lineB to be perpendiculat to camera
                BABYLON.Vector3.CrossToRef(this._lineA, this._lookAt, this._lookAt);
                this._lookAt.normalize();

                this._dragPlane.position.copyFrom(this._pointA);
                this._pointA.subtractToRef(this._lookAt, this._lookAt);
                this._dragPlane.lookAt(this._lookAt);
            }else if(this._options.dragPlaneNormal){
                this.useObjectOrienationForDragging ? Vector3.TransformCoordinatesToRef(this._options.dragPlaneNormal, this._attachedNode.getWorldMatrix().getRotationMatrix(),this._localAxis) : this._localAxis.copyFrom(this._options.dragPlaneNormal);
                this._dragPlane.position.copyFrom(this._pointA);
                this._pointA.subtractToRef(this._localAxis, this._lookAt);
                this._dragPlane.lookAt(this._lookAt);
            }else{
                this._dragPlane.position.copyFrom(this._pointA);
                this._dragPlane.lookAt(ray.origin);
            }
            this._dragPlane.computeWorldMatrix(true);
        }

        /**
         *  Detaches the behavior from the mesh
         */
        public detach(): void {
            if(this._pointerObserver){
                this._scene.onPointerObservable.remove(this._pointerObserver);
            }
            if(this._beforeRenderObserver){
                this._scene.onBeforeRenderObservable.remove(this._beforeRenderObserver);
            }
        }
    }
}