org.jhotdraw8.draw.tool.PolyCreationTool Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of org.jhotdraw8.draw Show documentation
JHotDraw8 Drawing Framework
The newest version!
/*
 * @(#)PolyCreationTool.java
 * Copyright © 2023 The authors and contributors of JHotDraw. MIT License.
 */
package org.jhotdraw8.draw.tool;

import javafx.geometry.Point2D;
import javafx.scene.Cursor;
import javafx.scene.input.MouseEvent;
import org.jspecify.annotations.Nullable;
import org.jhotdraw8.application.resources.Resources;
import org.jhotdraw8.draw.DrawingEditor;
import org.jhotdraw8.draw.DrawingView;
import org.jhotdraw8.draw.css.value.CssPoint2D;
import org.jhotdraw8.draw.figure.Figure;
import org.jhotdraw8.draw.figure.Layer;
import org.jhotdraw8.draw.figure.LayerFigure;
import org.jhotdraw8.draw.handle.HandleType;
import org.jhotdraw8.draw.key.Point2DListStyleableKey;
import org.jhotdraw8.draw.model.DrawingModel;
import org.jhotdraw8.icollection.VectorList;

import java.util.ArrayList;
import java.util.Objects;
import java.util.function.Supplier;

/**
 * CreationTool for polyline figures.
 *
 * @author Werner Randelshofer
 */
public class PolyCreationTool extends AbstractCreationTool {

    /**
     * The rubber band.
     */
    private @Nullable ArrayList points;

    private final Point2DListStyleableKey key;

    public PolyCreationTool(String name, Resources rsrc, Point2DListStyleableKey key, Supplier factory) {
        this(name, rsrc, key, factory, LayerFigure::new);
    }

    public PolyCreationTool(String name, Resources rsrc, Point2DListStyleableKey key, Supplier figureFactory, Supplier layerFactory) {
        super(name, rsrc, figureFactory, layerFactory);
        this.key = key;
        node.setCursor(Cursor.CROSSHAIR);
    }

    @Override
    protected void stopEditing() {
        if (createdFigure != null) {
            createdFigure = null;
            points = null;
        }
    }

    @Override
    protected void onMousePressed(MouseEvent event, DrawingView view) {
        if (event.getClickCount() != 1) {
            return;
        }
        double x1 = event.getX();
        double y1 = event.getY();

        DrawingModel dm = view.getModel();
        if (createdFigure == null) {
            createdFigure = createFigure();
            Figure parent = createdFigure == null ? null : getOrCreateParent(view, createdFigure);
            if (parent == null) {
                createdFigure = null;
                points = null;
                event.consume();
                return;
            }
            dm.addChildTo(createdFigure, parent);
            points = new ArrayList<>();
            Point2D c = view.getConstrainer().constrainPoint(createdFigure,
                    new CssPoint2D(createdFigure.worldToParent(view.viewToWorld(new Point2D(x1, y1))))).getConvertedValue();
            points.add(c);
            points.add(c); // we will update the last point on drag
            view.setActiveParent(parent);

        } else {
            assert points != null;
            Point2D c = view.getConstrainer().constrainPoint(createdFigure,
                    new CssPoint2D(createdFigure.worldToParent(view.viewToWorld(new Point2D(x1, y1))))).getConvertedValue();
            points.add(c);
        }
        dm.set(createdFigure, key, VectorList.copyOf(points));

        event.consume();
    }

    @Override
    protected void onMouseMoved(MouseEvent event, DrawingView dv) {
        if (createdFigure != null) {
            onMouseDragged(event, dv);
        }
    }

    @Override
    protected void onMouseDragged(MouseEvent event, DrawingView dv) {
        if (createdFigure != null && points != null) {
            double x2 = event.getX();
            double y2 = event.getY();
            Point2D c2 = dv.getConstrainer().constrainPoint(createdFigure, new CssPoint2D(
                    createdFigure.worldToParent(dv.viewToWorld(x2, y2)))).getConvertedValue();
            DrawingModel dm = dv.getModel();
            points.set(points.size() - 1, c2);
            dm.set(createdFigure, key, VectorList.copyOf(points));
        }
        event.consume();
    }

    @Override
    protected void onMouseClicked(MouseEvent event, DrawingView dv) {
        if (event.getClickCount() > 1) {
            if (createdFigure != null && points != null) {
                for (int i = points.size() - 1; i > 0; i--) {
                    if (Objects.equals(points.get(i), points.get(i - 1))) {
                        points.remove(i);
                    }
                }
                DrawingModel dm = dv.getModel();
                if (points.size() < 2) {
                    dm.removeFromParent(createdFigure);
                } else {
                    dm.set(createdFigure, key, VectorList.copyOf(points));
                    dv.getSelectedFigures().clear();
                    dv.getEditor().setHandleType(HandleType.POINT);
                    dv.getSelectedFigures().add(createdFigure);
                }
                createdFigure = null;
                points = null;
                fireToolDone();
            }
        }
    }


    @Override
    public void activate(DrawingEditor editor) {
        requestFocus();
        super.activate(editor);
        createdFigure = null;
    }

    @Override
    public String getHelpText() {
        return """
               PolyCreationTool
                 Click on the drawing view. The tool will create a new polygon with a point at that location.
                 Continue clicking on the drawing view. The tool will add each clicked point to the created polygon.
                 Press enter or escape, when you are done.""";
    }

}