Make GPU rendering toggleable to prevent crashes due to bad initialisation

Author: jameshball

CHANGELOG.md

@@ -1,3 +1,7 @@
+- 1.26.4
+  - Make GPU rendering toggleable to prevent crashes due to bad initialisation
+
+
 - 1.26.3
   - Fix potential bug with updating frequency spinner and slider from not-GUI thread
 

pom.xml

@@ -6,7 +6,7 @@
 
     <groupId>sh.ball</groupId>
     <artifactId>osci-render</artifactId>
-    <version>1.26.3</version>
+    <version>1.26.4</version>
 
     <name>osci-render</name>
 

src/main/java/sh/ball/engine/Camera.java

@@ -24,6 +24,8 @@ public class Camera {
 
   private double focalLength;
   private Vector3 pos;
+  private boolean hideEdges = false;
+  private boolean usingGpu = false;
 
   public Camera(double focalLength, Vector3 pos) {
     this.focalLength = focalLength;
@@ -105,4 +107,20 @@ public void setFocalLength(double focalLength) {
   public double getFocalLength() {
     return focalLength;
   }
+
+  public void hideEdges(boolean hideEdges) {
+    this.hideEdges = hideEdges;
+  }
+
+  public boolean edgesHidden() {
+    return hideEdges;
+  }
+
+  public void usingGpu(boolean usingGpu) {
+    this.usingGpu = usingGpu;
+  }
+
+  public boolean isUsingGpu() {
+    return usingGpu;
+  }
 }

src/main/java/sh/ball/engine/CameraDrawKernel.java

@@ -36,7 +36,9 @@ public class CameraDrawKernel extends Kernel {
   private float focalLength;
   private int hideEdges = 0;
   private int usingObjectSet = 0;
+  private boolean initialisedGpu = false;
   private boolean failedGpu = false;
+  private int maxGroupSize = 256;
 
   public CameraDrawKernel() {}
 
@@ -131,9 +133,13 @@ public List<Shape> draw(Camera camera, WorldObject object) {
     this.cameraPosZ = (float) cameraPos.z;
     this.focalLength = (float) camera.getFocalLength();
 
-    this.hideEdges = object.edgesHidden() ? 1 : 0;
+    this.hideEdges = camera.edgesHidden() ? 1 : 0;
 
-    return executeGpu();
+    if (!camera.isUsingGpu() || failedGpu || vertices.length / 3 < MIN_GPU_VERTICES) {
+      return executeCpu();
+    } else {
+      return executeGpu();
+    }
   }
 
   private List<Shape> postProcessVertices() {
@@ -164,16 +170,14 @@ private List<Shape> executeCpu() {
   }
 
   private List<Shape> executeGpu() {
-    if (failedGpu || vertices.length / 3 < MIN_GPU_VERTICES) {
-      return executeCpu();
-    }
-
     try {
-      int maxGroupSize = 256;
-      try {
-        maxGroupSize = getKernelMaxWorkGroupSize(getTargetDevice());
-      } catch (QueryFailedException e) {
-        logger.log(Level.WARNING, e.getMessage(), e);
+      if (!initialisedGpu) {
+        try {
+          maxGroupSize = getKernelMaxWorkGroupSize(getTargetDevice());
+          initialisedGpu = true;
+        } catch (QueryFailedException e) {
+          logger.log(Level.WARNING, e.getMessage(), e);
+        }
       }
 
       execute(Range.create(roundUp(vertices.length / 3, maxGroupSize), maxGroupSize));

src/main/java/sh/ball/engine/WorldObject.java

@@ -26,7 +26,6 @@ public class WorldObject {
   private List<List<Vector3>> vertexPath;
   private Vector3 position;
   private Vector3 rotation;
-  private boolean hideEdges = false;
 
   public WorldObject(Vector3[] vertices, int[] edgeIndices, int[][] faceIndices) {
     vertexPath = new ArrayList<>();
@@ -319,12 +318,4 @@ public WorldObject clone() {
     return new WorldObject(new ArrayList<>(objVertices), new ArrayList<>(vertexPath), position,
       rotation);
   }
-
-  public void hideEdges(boolean hideEdges) {
-    this.hideEdges = hideEdges;
-  }
-
-  public boolean edgesHidden() {
-    return hideEdges;
-  }
 }

src/main/java/sh/ball/gui/controller/MainController.java

@@ -148,6 +148,8 @@ public class MainController implements Initializable, FrequencyListener, MidiLis
   @FXML
   private CheckMenuItem hideHiddenMeshesCheckMenuItem;
   @FXML
+  private CheckMenuItem renderUsingGpuCheckMenuItem;
+  @FXML
   private Spinner<Integer> midiChannelSpinner;
   @FXML
   private Spinner<Double> attackSpinner;
@@ -408,6 +410,8 @@ public void initialize(URL url, ResourceBundle resourceBundle) {
 
     hideHiddenMeshesCheckMenuItem.selectedProperty().addListener((e, old, hidden) -> objController.hideHiddenMeshes(hidden));
 
+    renderUsingGpuCheckMenuItem.selectedProperty().addListener((e, old, usingGpu) -> objController.renderUsingGpu(usingGpu));
+
     NumberFormat format = NumberFormat.getIntegerInstance();
     UnaryOperator<TextFormatter.Change> filter = c -> {
       if (c.isContentChange()) {
@@ -691,6 +695,7 @@ private void changeFrameSource(int index) {
       }
       // FIXME: Is this safe since we are accessing GUI object in non-GUI thread?
       objController.hideHiddenMeshes(hideHiddenMeshesCheckMenuItem.isSelected());
+      objController.renderUsingGpu(renderUsingGpuCheckMenuItem.isSelected());
       executor.submit(producer);
     } else if (samples != null) {
       samples.enable();
@@ -1141,6 +1146,10 @@ private void saveProject(String projectFileName) {
       hiddenEdges.appendChild(document.createTextNode(String.valueOf(hideHiddenMeshesCheckMenuItem.isSelected())));
       root.appendChild(hiddenEdges);
 
+      Element usingGpu = document.createElement("usingGpu");
+      usingGpu.appendChild(document.createTextNode(String.valueOf(renderUsingGpuCheckMenuItem.isSelected())));
+      root.appendChild(usingGpu);
+
       Element translationIncrement = document.createElement("translationIncrement");
       translationIncrement.appendChild(document.createTextNode(translationIncrementSpinner.getValue().toString()));
       root.appendChild(translationIncrement);
@@ -1263,6 +1272,11 @@ private void openProject(String projectFileName) {
         hideHiddenMeshesCheckMenuItem.setSelected(Boolean.parseBoolean(hiddenEdges.getTextContent()));
       }
 
+      Element usingGpu = (Element) root.getElementsByTagName("usingGpu").item(0);
+      if (usingGpu != null) {
+        renderUsingGpuCheckMenuItem.setSelected(Boolean.parseBoolean(usingGpu.getTextContent()));
+      }
+
       Element translationIncrement = (Element) root.getElementsByTagName("translationIncrement").item(0);
       if (translationIncrement != null) {
         translationIncrementSpinner.getValueFactory().setValue(Double.parseDouble(translationIncrement.getTextContent()));

src/main/java/sh/ball/gui/controller/ObjController.java

@@ -182,4 +182,8 @@ public List<Element> save(Document document) {
 
   @Override
   public void load(Element root) {}
+
+  public void renderUsingGpu(boolean usingGpu) {
+    producer.setFrameSettings(ObjSettingsFactory.renderUsingGpu(usingGpu));
+  }
 }

src/main/java/sh/ball/parser/obj/ObjFrameSettings.java

@@ -11,15 +11,17 @@ public class ObjFrameSettings {
   public Double rotateSpeed;
   public boolean resetRotation = false;
   public Boolean hideEdges = null;
+  public Boolean usingGpu = null;
 
-  protected ObjFrameSettings(Double focalLength, Vector3 cameraPos, Vector3 baseRotation, Vector3 currentRotation, Double rotateSpeed, boolean resetRotation, Boolean hideEdges) {
+  protected ObjFrameSettings(Double focalLength, Vector3 cameraPos, Vector3 baseRotation, Vector3 currentRotation, Double rotateSpeed, boolean resetRotation, Boolean hideEdges, Boolean usingGpu) {
     this.focalLength = focalLength;
     this.cameraPos = cameraPos;
     this.baseRotation = baseRotation;
     this.currentRotation = currentRotation;
     this.rotateSpeed = rotateSpeed;
     this.resetRotation = resetRotation;
     this.hideEdges = hideEdges;
+    this.usingGpu = usingGpu;
   }
 
   protected ObjFrameSettings(double focalLength) {

src/main/java/sh/ball/parser/obj/ObjFrameSource.java

@@ -68,14 +68,17 @@ public void setFrameSettings(Object settings) {
         object.resetRotation();
       }
       if (obj.hideEdges != null) {
-        object.hideEdges(obj.hideEdges);
+        camera.hideEdges(obj.hideEdges);
+      }
+      if (obj.usingGpu != null) {
+        camera.usingGpu(obj.usingGpu);
       }
     }
   }
 
   // TODO: Refactor!
   @Override
   public Object getFrameSettings() {
-    return new ObjFrameSettings(null, null, baseRotation, currentRotation, null, false, object.edgesHidden());
+    return new ObjFrameSettings(null, null, baseRotation, currentRotation, null, false, camera.edgesHidden(), camera.isUsingGpu());
   }
 }

src/main/java/sh/ball/parser/obj/ObjSettingsFactory.java

@@ -17,7 +17,7 @@ public static ObjFrameSettings baseRotation(Vector3 baseRotation) {
   }
 
   public static ObjFrameSettings rotation(Vector3 baseRotation, Vector3 currentRotation) {
-    return new ObjFrameSettings(null, null, baseRotation, currentRotation, null, false, null);
+    return new ObjFrameSettings(null, null, baseRotation, currentRotation, null, false, null, null);
   }
 
   public static ObjFrameSettings rotateSpeed(double rotateSpeed) {
@@ -29,6 +29,10 @@ public static ObjFrameSettings resetRotation() {
   }
 
   public static ObjFrameSettings hideEdges(Boolean hideEdges) {
-    return new ObjFrameSettings(null, null, null, null, null, false, hideEdges);
+    return new ObjFrameSettings(null, null, null, null, null, false, hideEdges, null);
+  }
+
+  public static ObjFrameSettings renderUsingGpu(Boolean usingGpu) {
+    return new ObjFrameSettings(null, null, null, null, null, false, null, usingGpu);
   }
 }