Update NoiseRemoval_Briggs.cs

Issue #370 Fix bug where window was longer than the matrix.
QutEcoacoustics · Oct 14, 2020 · 5f0ddbe · 5f0ddbe
1 parent bbdedfd
commit 5f0ddbe
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/src/AudioAnalysisTools/DSP/NoiseRemoval_Briggs.cs b/src/AudioAnalysisTools/DSP/NoiseRemoval_Briggs.cs
@@ -199,14 +199,24 @@ public static double[] CalculateLocalVariance1(double[] data, int window)
         /// </summary>
         public static double[] CalculateLocalVariance2(double[,] matrix, int colId, int rowWindow)
         {
+            int rowCount = matrix.GetLength(0);
+            int colCount = matrix.GetLength(1);
+
+            if (rowWindow < 3)
+            {
+                throw new Exception("Invalid value for length of rowWindow. Must be >= 3");
+            }
+
+            if (rowWindow > rowCount)
+            {
+                rowWindow = rowCount - 1;
+            }
+
             // the column window must be an odd number.
             int colWindow = 5;
             int halfColWindow = colWindow / 2;
             int halfRowWindow = rowWindow / 2;
 
-            int rowCount = matrix.GetLength(0);
-            int colCount = matrix.GetLength(1);
-
             double[] variances = new double[rowCount];
 
             // set up the column Ids.