SelectionType.java

/*
 * Copyright 2017 Laszlo Balazs-Csiki
 *
 * This file is part of Pixelitor. Pixelitor is free software: you
 * can redistribute it and/or modify it under the terms of the GNU
 * General Public License, version 3 as published by the Free
 * Software Foundation.
 *
 * Pixelitor 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.
 *
 * You should have received a copy of the GNU General Public License
 * along with Pixelitor. If not, see <http://www.gnu.org/licenses/>.
 */

package pixelitor.selection;

import pixelitor.tools.PMouseEvent;
import pixelitor.tools.UserDrag;

import java.awt.Shape;
import java.awt.geom.Ellipse2D;
import java.awt.geom.GeneralPath;
import java.awt.geom.Rectangle2D;

/**
 * The type of a new selection created interactively by the user
 */
public enum SelectionType {
    RECTANGLE("Rectangle") {
        @Override
        public Shape createShape(Object mouseInfo, Shape oldShape) {
            UserDrag userDrag = (UserDrag) mouseInfo;
            return userDrag.createPositiveRect();
        }
    }, ELLIPSE("Ellipse") {
        @Override
        public Shape createShape(Object mouseInfo, Shape oldShape) {
            UserDrag userDrag = (UserDrag) mouseInfo;
            Rectangle2D dr = userDrag.createPositiveRect();
            return new Ellipse2D.Double(dr.getX(), dr.getY(), dr.getWidth(), dr.getHeight());
        }
    }, LASSO("Freehand") {
        @Override
        public Shape createShape(Object mouseInfo, Shape oldShape) {
            UserDrag userDrag = (UserDrag) mouseInfo;
            boolean createNew;
            if (oldShape == null) {
                createNew = true;
            } else if (oldShape instanceof GeneralPath) {
                createNew = false;
            } else { // it is an Area, meaning that a new shape has been started
                createNew = true;
            }

            if (createNew) {
                GeneralPath p = new GeneralPath();
                p.moveTo(userDrag.getStartX(), userDrag.getStartY());
                p.lineTo(userDrag.getEndX(), userDrag.getEndY());
                return p;
            } else {
                GeneralPath gp = (GeneralPath) oldShape;
                gp.lineTo(userDrag.getEndX(), userDrag.getEndY());

                return gp;
            }
        }
    }, POLYGONAL_LASSO("Polygonal") {
        @Override
        public Shape createShape(Object mouseInfo, Shape oldShape) {
            PMouseEvent pe = (PMouseEvent) mouseInfo;
            boolean createNew;
            if (oldShape == null) {
                createNew = true;
            } else if (oldShape instanceof GeneralPath) {
                createNew = false;
            } else { // it is an Area, meaning that a new shape has been started
                createNew = true;
            }

            if (createNew) {
                GeneralPath p = new GeneralPath();
                p.moveTo(pe.getX(), pe.getY());
                return p;
            } else {
                GeneralPath gp = (GeneralPath) oldShape;
                gp.lineTo(pe.getX(), pe.getY());

                return gp;
            }
        }
    };

    private final String guiName;

    SelectionType(String guiName) {
        this.guiName = guiName;
    }

    public abstract Shape createShape(Object mouseInfo, Shape oldShape);

    @Override
    public String toString() {
        return guiName;
    }
}