update plotting scripts to handle case when there is no fit

2 files changed, 8 insertions, 0 deletions

diff --git a/utils/plot b/utils/plot
index 8fd2fae..6aa55fa 100755
--- a/utils/plot
+++ b/utils/plot
@@ -95,6 +95,10 @@ if __name__ == '__main__':
             if 'ev' not in event:
                 continue
 
+            if 'fit' not in event['ev'][0]:
+                # if nhit < 100 we don't fit the event
+                continue
+
             if id not in event['ev'][0]['fit']:
                 continue
 
diff --git a/utils/plot-fit-results b/utils/plot-fit-results
index 7a9708e..e1f7987 100755
--- a/utils/plot-fit-results
+++ b/utils/plot-fit-results
@@ -103,6 +103,10 @@ if __name__ == '__main__':
 
             a[i]['id'] = id
 
+            if 'fit' not in event['ev'][0]:
+                # if nhit < 100 we don't fit the event
+                continue
+
             if id not in event['ev'][0]['fit']:
                 a[i] = np.ma.masked
                 continue