Add next_frame_size method

3 files changed, 12 insertions, 0 deletions

diff --git a/src/simpl/base.cpp b/src/simpl/base.cpp
index 50b4654..cc9dc2c 100644
--- a/src/simpl/base.cpp
+++ b/src/simpl/base.cpp
@@ -188,6 +188,11 @@ void PeakDetection::static_frame_size(bool new_static_frame_size)
     _static_frame_size = new_static_frame_size;
 }
 
+int PeakDetection::next_frame_size()
+{
+    return _frame_size;
+}
+
 int PeakDetection::hop_size()
 {
     return _hop_size;
diff --git a/src/simpl/base.h b/src/simpl/base.h
index 735094b..7649f86 100644
--- a/src/simpl/base.h
+++ b/src/simpl/base.h
@@ -136,6 +136,7 @@ public:
     void frame_size(int new_frame_size);
     bool static_frame_size();
     void static_frame_size(bool new_static_frame_size);
+    int next_frame_size();
     int hop_size();
     void hop_size(int new_hop_size);
     int max_peaks();
diff --git a/tests/testbase.cpp b/tests/testbase.cpp
index c79cf18..34cefde 100644
--- a/tests/testbase.cpp
+++ b/tests/testbase.cpp
@@ -183,6 +183,7 @@ class TestPeakDetection : public CPPUNIT_NS::TestCase
     CPPUNIT_TEST(test_constructor);
     CPPUNIT_TEST(test_frame_size);
     CPPUNIT_TEST(test_static_frame_size);
+    CPPUNIT_TEST(test_next_frame_size);
     CPPUNIT_TEST(test_hop_size);
     CPPUNIT_TEST(test_max_peaks);
     CPPUNIT_TEST(test_window_type);
@@ -228,6 +229,11 @@ protected:
         pd->static_frame_size(true);
     }
 
+    void test_next_frame_size()
+    {
+        CPPUNIT_ASSERT(pd->next_frame_size() == pd->frame_size());
+    }
+
     void test_hop_size()
     {
         pd->hop_size(128);