From 5bfa2b1898d18d76f62781999b7fba35c987d223 Mon Sep 17 00:00:00 2001
From: SGCMarkus <markusornik@gmail.com>
Date: Mon, 15 Apr 2024 14:40:46 +0200
Subject: [PATCH] flaredetector: add boundaries for flare range

---
 main/flaredetector/flaredetector.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/main/flaredetector/flaredetector.py b/main/flaredetector/flaredetector.py
index c880960..6e8afc5 100644
--- a/main/flaredetector/flaredetector.py
+++ b/main/flaredetector/flaredetector.py
@@ -30,6 +30,12 @@ def getFlareRange(flux, peakIndex):
                 break
         prevFlux = flux[i]
 
+    if(minInd == -np.inf):
+        minInd = peakIndex - maxBacklog
+
+    if(minInd < 0):
+        minInd = 0
+
     deltaCount = 0
     delta = 100
     prevFlux = flux[peakIndex]
@@ -42,4 +48,10 @@ def getFlareRange(flux, peakIndex):
                 break
         prevFlux = flux[i]
 
+    if(maxInd == np.inf):
+        maxInd = peakIndex + maxForwardlog
+
+    if(maxInd > len(flux)-1):
+        maxInd = len(flux)-1
+
     return minInd, maxInd