flaredetector: util: add function to convert normal index to folded index/cycle pair

2024-06-13 14:48:36 +02:00
parent d66df44bea
commit b5479b5dc1
1 changed files with 12 additions and 1 deletions
@@ -68,4 +68,15 @@ def plotFlareFits(ax, fits):

 def markFlare(ax, fits, flux):
    for fit in fits:
-        ax.plot(fit["FlareFitTime"], flux[fit["StandardIndex"]], color="red")
+        ax.plot(fit["FlareFitTime"], flux[fit["StandardIndex"]], color="red")
+
+def convertStarndardIndexToFoldedIndex(foldedLc, standardIndex):
+    foldedIndex = standardIndex; cycle = 0
+    for i in range(max(foldedLc.cycle)+1):
+        if(foldedIndex - len(foldedLc.phase[foldedLc.cycle == i]) >= 0):
+            foldedIndex = foldedIndex - len(foldedLc.phase[foldedLc.cycle == i])
+        else:
+            cycle = i
+            break
+
+    return cycle, foldedIndex