Cleanup, remove some unneeded code / classes / TODO

Signed-off-by: NathanDissoubray <[email protected]>

Author: NathanDissoubray

diagram-util/src/main/java/com/powsybl/diagram/util/forcelayout/ForceLayout.java

@@ -130,6 +130,7 @@ public Vector2D getStablePosition(V vertex) {
         return algorithmRunner.getStablePosition(vertex);
     }
 
+    //TODO we could remove this since there are no use for it in the code, but its public so someone else might be using it.
     public Set<Spring> getSprings() {
         return springs;
     }

diagram-util/src/main/java/com/powsybl/diagram/util/forcelayout/geometry/Edge.java

@@ -21,10 +21,7 @@ public class Point {
     private static final double DEFAULT_MASS = 1.0;
 
     private Vector2D position;
-    //TODO remove this, it is not needed
-    // or maybe it is to calculate the Barnes Hut Dynamic schedule ?
     private Vector2D velocity;
-    //TODO remove this, forces are not kept at the point level anymore
     private Vector2D forces;
     private final double mass;
 

diagram-util/src/main/java/com/powsybl/diagram/util/forcelayout/geometry/Point.java

@@ -71,8 +71,7 @@ public static <V, E> SpringContainer<DefaultEdge> initializeSprings(ForceGraph<V
 
     @Override
     public void calculateLayout(ForceGraph<V, E> forceGraph) {
-        // it would be better if this was created with all the other forces but we need the graph to init the springs
-        // TODO that could cause an issue if we launch the runner on multiple different graph, we would keep adding spring force
+        // it would be better if this was created with all the other forces, but we need the graph to init the springs
         this.forces.add(new SpringForce<>(initializeSprings(forceGraph)));
 
         // do the loop on the nodes and forces
@@ -93,6 +92,12 @@ public void calculateLayout(ForceGraph<V, E> forceGraph) {
             }
         }
         LOGGER.info("Calculating the layout took {} steps", i);
+        try {
+            this.forces.remove(2);
+        } catch (Exception e) {
+            LOGGER.error("Tried to remove the Spring force from the Graph but did not succeed", new Exception(e));
+        }
+
     }
 
     private void updateVelocity(ForceGraph<V, E> forceGraph) {