From 9b0dc1d0881ac813106742aee6704241c394cebc Mon Sep 17 00:00:00 2001
From: kapoorlab <randomaccessiblekapoor@gmail.com>
Date: Mon, 17 Jun 2024 23:15:43 +0200
Subject: [PATCH] add null pointer exception try catch

---
 .../trackmate/action/oneat/TrackCorrectorRunner.java   | 10 ++++++----
 1 file changed, 6 insertions(+), 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
index fd141f9..fd8886e 100644
--- 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) {