aladin-lite/src/js/FiniteStateMachine/CircleSelect.js

// Copyright 2015 - UDS/CNRS
// The Aladin Lite program is distributed under the terms
// of the GNU General Public License version 3.
//
// This file is part of Aladin Lite.
//
//    Aladin Lite is free software: you can redistribute it and/or modify
//    it under the terms of the GNU General Public License as published by
//    the Free Software Foundation, version 3 of the License.
//
//    Aladin Lite is distributed in the hope that it will be useful,
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//    GNU General Public License for more details.
//
//    The GNU General Public License is available in COPYING file
//    along with Aladin Lite.
//

import { FSM } from "../FiniteStateMachine";
import { View } from "../View";
import { Selector } from "../Selector";
/******************************************************************************
 * Aladin Lite project
 * 
 * Class Selector
 * 
 * The circle selection finite state machine
 * 
 * Author: Matthieu Baumann[CDS]
 * 
 *****************************************************************************/

export class CircleSelect extends FSM {
    // constructor
    constructor(options, view) {
        let start = (params) => {
            const {callback} = params;
            this.callback = callback;
            view.setMode(View.SELECT)
        }

        let mousedown = (params) => {
            const {coo} = params;
            // start a new selection
            this.startCoo = coo;
        }

        let mousemove = (params) => {
            const {coo} = params;
            this.coo = coo;

            view.requestRedraw();
        };

        let draw = () => {
            let ctx = view.catalogCtx;

            // draw the selection
            let colorValue = (typeof options.color === 'function') ? options.color(this.startCoo, this.coo) : options.color;
            ctx.fillStyle = colorValue + '7f';
            ctx.strokeStyle = colorValue;
            ctx.lineWidth = options.lineWidth;

            var r2 = (this.coo.x - this.startCoo.x) * (this.coo.x - this.startCoo.x) + (this.coo.y - this.startCoo.y) * (this.coo.y - this.startCoo.y);
            var r = Math.sqrt(r2);

            ctx.beginPath();
            ctx.arc(this.startCoo.x, this.startCoo.y, r, 0, 2 * Math.PI);
            ctx.fill();
            ctx.stroke();
        }

        let mouseup = (params) => {
            var x, y;
            const {coo} = params;
            this.coo = coo;
            // finish the selection
            var r2 = (this.coo.x - this.startCoo.x) * (this.coo.x - this.startCoo.x) + (this.coo.y - this.startCoo.y) * (this.coo.y - this.startCoo.y);
            var r = Math.sqrt(r2);

            x = this.startCoo.x;
            y = this.startCoo.y;

            let s = {
                x, y, r,
                label: 'circle',
                contains(s) {
                    let dx = (s.x - x)
                    let dy = (s.y - y);

                    return dx*dx + dy*dy <= r2;
                },
                bbox() {
                    return {
                        x: x - r,
                        y: y - r,
                        w: 2*r,
                        h: 2*r
                    }
                }
            };

            // execute general callback
            if (view.aladin.callbacksByEventName) {
                var callback = view.aladin.callbacksByEventName['objectsSelected'] || view.aladin.callbacksByEventName['select'];

                if (typeof callback === "function") {
                    let objList = Selector.getObjects(s, view);
                    view.selectObjects(objList);
                    callback(objList);
                }
            }

            // execute selection callback only
            (typeof this.callback === 'function') && this.callback(s, Selector.getObjects(s, view));

            this.dispatch("off");
        };

        let mouseout = (params) => {
            if (this.startCoo) {
                mouseup(params);
            }
        };

        let off = () => {
            view.aladin.showReticle(true)
            view.setCursor('default');

            view.setMode(View.PAN)
            view.requestRedraw();

            view.aladin.removeStatusBarMessage('selector')
        };

        super({
            state: 'off',
            transitions: {
                off: {
                    start,
                },
                start: {
                    mousedown
                },
                mousedown: {
                    mousemove
                },
                mousemove: {
                    draw,
                    mouseup,
                    mouseout
                },
                draw: {
                    mousemove,
                    mouseup,
                    mouseout
                },
                mouseout: {
                    off,
                    mousedown
                },
                mouseup: {
                    off,
                }
            }
        })
    };
}