SGCMarkus/flaredetector
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

flaredetector: util: add option to return phase instead of phase index

Browse Source
This commit is contained in:
SGCMarkus
2024-08-13 10:58:51 +02:00
parent 7f014938a9
commit 979c9b30c5
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main/flaredetector/util.py
+3 -3
View File
@@ -159,7 +159,7 @@ def findNearestIndexOfValue(array, value):
idx = (np.abs(array - value)).argmin() idx = (np.abs(array - value)).argmin()
return idx return idx
def getPhaseRangesNearPeak(maxArgs, phase): def getPhaseRangesNearPeak(maxArgs, phase, returnPhaseValue=False):
minPhases = phase[maxArgs[0]] minPhases = phase[maxArgs[0]]
maxPhases = phase[maxArgs[1]] maxPhases = phase[maxArgs[1]]
totalPhase = abs(phase[0]) + abs(phase[-1]) totalPhase = abs(phase[0]) + abs(phase[-1])
@@ -195,14 +195,14 @@ def getPhaseRangesNearPeak(maxArgs, phase):
for minPhaseBounds in minPhasesBounds: for minPhaseBounds in minPhasesBounds:
minPhaseBoundsIndices = [] minPhaseBoundsIndices = []
for l in minPhaseBounds: for l in minPhaseBounds:
minPhaseBoundsIndices.append([findNearestIndexOfValue(phase, lv) for lv in l]) minPhaseBoundsIndices.append([phase[findNearestIndexOfValue(phase, lv)] if returnPhaseValue else findNearestIndexOfValue(phase, lv) for lv in l])
minPhasesBoundsIndices.append(minPhaseBoundsIndices) minPhasesBoundsIndices.append(minPhaseBoundsIndices)
maxPhasesBoundsIndices = [] maxPhasesBoundsIndices = []
for maxPhaseBounds in maxPhasesBounds: for maxPhaseBounds in maxPhasesBounds:
maxPhaseBoundsIndices = [] maxPhaseBoundsIndices = []
for l in maxPhaseBounds: for l in maxPhaseBounds:
maxPhaseBoundsIndices.append([findNearestIndexOfValue(phase, lv) for lv in l]) maxPhaseBoundsIndices.append([phase[findNearestIndexOfValue(phase, lv)] if returnPhaseValue else findNearestIndexOfValue(phase, lv) for lv in l])
maxPhasesBoundsIndices.append(maxPhaseBoundsIndices) maxPhasesBoundsIndices.append(maxPhaseBoundsIndices)
return minPhasesBoundsIndices, maxPhasesBoundsIndices return minPhasesBoundsIndices, maxPhasesBoundsIndices