Fix connections not positioning correctly when loading project

Author: ThomasJClark

ui/src/main/java/edu/wpi/grip/ui/pipeline/PipelineController.java

@@ -3,15 +3,14 @@
 import com.google.common.eventbus.EventBus;
 import com.google.common.eventbus.Subscribe;
 import com.google.inject.Singleton;
+import com.sun.javafx.application.PlatformImpl;
 import edu.wpi.grip.core.*;
 import edu.wpi.grip.core.events.*;
 import edu.wpi.grip.ui.annotations.ParametrizedController;
 import edu.wpi.grip.ui.pipeline.input.InputSocketController;
 import edu.wpi.grip.ui.pipeline.source.SourceController;
 import edu.wpi.grip.ui.pipeline.source.SourceControllerFactory;
-import edu.wpi.grip.ui.util.GRIPPlatform;
 import edu.wpi.grip.ui.util.ControllerMap;
-import javafx.application.Platform;
 import javafx.beans.InvalidationListener;
 import javafx.beans.property.ReadOnlyObjectProperty;
 import javafx.collections.ObservableList;
@@ -56,8 +55,6 @@ public final class PipelineController {
     private StepController.Factory stepControllerFactory;
     @Inject
     private AddSourceView addSourceView;
-    @Inject
-    private GRIPPlatform platform;
 
     private ControllerMap<StepController, Node> stepsMapManager;
     private ControllerMap<SourceController, Node> sourceMapManager;
@@ -182,7 +179,7 @@ private ConnectionView findConnectionView(Connection connection) {
      * details of adding the connection.
      */
     private void addConnectionView(Connection connection) {
-        platform.runAsSoonAsPossible(() -> {
+        PlatformImpl.runAndWait(() -> {
             // Before adding a connection control, we have to look up the controls for both sockets in the connection so
             // we know where to position it.
             final OutputSocketController outputSocketView = findOutputSocketView(connection.getOutputSocket());
@@ -207,35 +204,33 @@ private void addConnectionView(Connection connection) {
                     final double x2 = inputSocketBounds.getMinX() + inputSocketBounds.getWidth() / 2.0;
                     final double y2 = inputSocketBounds.getMinY() + inputSocketBounds.getHeight() / 2.0;
 
-                    // This can run whenever. Don't wait for it to complete.
-                    // This should be Platform.runLater
-                    Platform.runLater(() -> {
+                    PlatformImpl.runAndWait(() -> {
                         connectionView.inputHandleProperty().setValue(new Point2D(x1, y1));
                         connectionView.outputHandleProperty().setValue(new Point2D(x2, y2));
                         ((ReadOnlyObjectProperty) observable).get();
                     });
                 }
             };
 
-            inputSocketController.getRoot().localToSceneTransformProperty().addListener(handleListener);
-            outputSocketView.getRoot().localToSceneTransformProperty().addListener(handleListener);
-            handleListener.invalidated(inputSocketController.getRoot().localToSceneTransformProperty());
+            inputSocketController.getHandle().localToSceneTransformProperty().addListener(handleListener);
+            outputSocketView.getHandle().localToSceneTransformProperty().addListener(handleListener);
+            handleListener.invalidated(outputSocketView.getHandle().localToSceneTransformProperty());
 
             connections.getChildren().add(connectionView);
         });
     }
 
     @Subscribe
     public void onSourceAdded(SourceAddedEvent event) {
-        platform.runAsSoonAsPossible(() -> {
+        PlatformImpl.runAndWait(() -> {
             final SourceController sourceController = sourceControllerFactory.create(event.getSource());
             sourceMapManager.add(sourceController);
         });
     }
 
     @Subscribe
     public void onSourceRemoved(SourceRemovedEvent event) {
-        platform.runAsSoonAsPossible(() -> {
+        PlatformImpl.runAndWait(() -> {
             final SourceController sourceController = findSourceView(event.getSource());
             sourceMapManager.remove(sourceController);
             eventBus.unregister(sourceController);
@@ -245,7 +240,7 @@ public void onSourceRemoved(SourceRemovedEvent event) {
     @Subscribe
     public void onStepAdded(StepAddedEvent event) {
         // Add a new view to the pipeline for the step that was added
-        platform.runAsSoonAsPossible(() -> {
+        PlatformImpl.runAndWait(() -> {
             final int index = event.getIndex().or(stepBox.getChildren().size());
             final Step step = event.getStep();
 
@@ -259,7 +254,7 @@ public void onStepAdded(StepAddedEvent event) {
     @Subscribe
     public void onStepRemoved(StepRemovedEvent event) {
         // Remove the control that corresponds with the step that was removed
-        platform.runAsSoonAsPossible(() -> {
+        PlatformImpl.runAndWait(() -> {
             final StepController stepController = findStepController(event.getStep());
 
             stepsMapManager.remove(stepController);
@@ -269,7 +264,7 @@ public void onStepRemoved(StepRemovedEvent event) {
 
     @Subscribe
     public void onStepMoved(StepMovedEvent event) {
-        platform.runAsSoonAsPossible(() -> {
+        PlatformImpl.runAndWait(() -> {
             final StepController stepController = findStepController(event.getStep());
             stepsMapManager.moveByDistance(stepController, event.getDistance());
         });
@@ -278,13 +273,13 @@ public void onStepMoved(StepMovedEvent event) {
     @Subscribe
     public void onConnectionAdded(ConnectionAddedEvent event) {
         // Add the new connection view
-        platform.runAsSoonAsPossible(() -> addConnectionView(event.getConnection()));
+        addConnectionView(event.getConnection());
     }
 
     @Subscribe
     public void onConnectionRemoved(ConnectionRemovedEvent event) {
         // Remove the control that corresponds with the connection that was removed
-        platform.runAsSoonAsPossible(() -> {
+        PlatformImpl.runAndWait(() -> {
             final ConnectionView connectionView = findConnectionView(event.getConnection());
             connections.getChildren().remove(connectionView);
             eventBus.unregister(connectionView);