com.powsybl.diagram.util.forcelayout.Point Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of powsybl-diagram-util Show documentation

Force layout for PowSyBl Diagram

There is a newer version: 4.5.1

/**
 * Copyright (c) 2021, RTE (http://www.rte-france.com)
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */
package com.powsybl.diagram.util.forcelayout;

import java.io.PrintWriter;
import java.util.Locale;
import java.util.function.Function;

/**
 * @author Mathilde Grapin {@literal }
 */
public class Point {
    private static final double DEFAULT_MASS = 1.0;

    private Vector position;
    private Vector velocity;
    private Vector forces;
    private final double mass;

    public Point(double x, double y) {
        this(x, y, DEFAULT_MASS);
    }

    public Point(double x, double y, double mass) {
        this.position = new Vector(x, y);
        this.velocity = new Vector(0, 0);
        this.forces = new Vector(0, 0);
        this.mass = mass;
    }

    public void applyForce(Vector force) {
        forces = forces.add(force);
    }

    public Vector getPosition() {
        return position;
    }

    public Vector getVelocity() {
        return velocity;
    }

    public Vector getForces() {
        return forces;
    }

    public void setPosition(Vector position) {
        this.position = position;
    }

    public void setVelocity(Vector velocity) {
        this.velocity = velocity;
    }

    public double getEnergy() {
        return 0.5 * mass * velocity.magnitudeSquare();
    }

    public double getMass() {
        return this.mass;
    }

    public void resetForces() {
        this.forces = new Vector(0, 0);
    }

    public  void toSVG(PrintWriter printWriter, Canvas canvas, Function tooltip, V vertex) {
        printWriter.println("");

        printWriter.printf("%s%n", tooltip.apply(vertex));

        Vector screenPosition = canvas.toScreen(getPosition());
        printWriter.printf(Locale.US, "%n",
            screenPosition.getX(), screenPosition.getY()
        );

        printWriter.println("");
    }
}