Update EventExtentions.cs

Issue #370 Conditions inserted to provent potential out of range index exceptions.
QutEcoacoustics · Sep 8, 2020 · f2ea271 · f2ea271
1 parent 3018b4b
commit f2ea271
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/AudioAnalysisTools/Events/EventExtentions.cs b/src/AudioAnalysisTools/Events/EventExtentions.cs
@@ -111,11 +111,25 @@ public static List<EventCommon> FilterOnDuration(List<EventCommon> events, doubl
         /// <returns>The average decibel value.</returns>
         public static double GetAverageDecibelsInEvent(SpectralEvent ev, double[,] spectrogramData, UnitConverters converter)
         {
+            // Get the length of the spectrogram
+            var frameCount = spectrogramData.GetLength(0);
+            var binCount = spectrogramData.GetLength(1);
+
             // extract the event from the spectrogram
             var lowerBin = converter.GetFreqBinFromHertz(ev.LowFrequencyHertz);
             var upperBin = converter.GetFreqBinFromHertz(ev.HighFrequencyHertz);
+            if (upperBin >= binCount)
+            {
+                upperBin = binCount - 1;
+            }
+
             var frameStart = converter.FrameFromStartTime(ev.EventStartSeconds);
             var frameEnd = converter.FrameFromStartTime(ev.EventEndSeconds);
+            if (frameEnd >= frameCount)
+            {
+                frameEnd = frameCount - 1;
+            }
+
             var subMatrix = MatrixTools.Submatrix<double>(spectrogramData, frameStart, lowerBin, frameEnd, upperBin);
 
             // extract the decibel array.