add null pointer exception try catch

trackmate-sc · Jun 17, 2024 · 9b0dc1d · 9b0dc1d
1 parent 0cb6998
commit 9b0dc1d
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/main/java/fiji/plugin/trackmate/action/oneat/TrackCorrectorRunner.java b/src/main/java/fiji/plugin/trackmate/action/oneat/TrackCorrectorRunner.java
@@ -676,10 +676,12 @@ public  static  <T extends NativeType<T>> SimpleWeightedGraph<Spot, DefaultWeigh
 		                    graph.addVertex(addsourcetarget.getB());
 
 		                    if (graph.degreeOf(addsourcetarget.getB()) < 2) {
-		                        final DefaultWeightedEdge edge = graph.addEdge(addsourcetarget.getA(),
-		                                addsourcetarget.getB());
-
-		                        graph.setEdgeWeight(edge, cost);
+		                        final DefaultWeightedEdge edge = graph.addEdge(addsourcetarget.getA(), addsourcetarget.getB());
+		                        if (edge != null && graph.containsEdge(edge)) {
+		                            graph.setEdgeWeight(edge, cost);
+		                        } else {
+		                            System.out.println("Failed to add edge or edge no longer exists between " + addsourcetarget.getA() + " and " + addsourcetarget.getB());
+		                        }
 		                    }
 		                }
 		            } catch (Exception e) {