david/librubberband
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Introduce a histogram filter for filtering discrete values

Browse Source
This commit is contained in:
Chris Cannam
2022-06-10 12:35:15 +01:00
parent 10e2c13551
commit 6940ad29d8
6 changed files with 438 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

109
src/test/TestSignalBits.cpp
View File

@@ -25,6 +25,7 @@
#include <boost/test/unit_test.hpp>
#include "../common/MovingMedian.h"
#include "../common/HistogramFilter.h"
#include "../finer/Peak.h"
using namespace RubberBand;
@@ -94,6 +95,114 @@ BOOST_AUTO_TEST_CASE(moving_median_n_1)
BOOST_TEST(arr == expected, tt::per_element());
}
BOOST_AUTO_TEST_CASE(histogram_median_simple_3)
{
HistogramFilter hf(5, 3); // nValues, filterLength
vector<int> arr { 1, 2, 3 };
vector<int> expected { 1, 2, 2 };
HistogramFilter::medianFilter(hf, arr);
BOOST_TEST(arr == expected, tt::per_element());
}
BOOST_AUTO_TEST_CASE(histogram_median_simple_4)
{
HistogramFilter hf(5, 4); // nValues, filterLength
vector<int> arr { 1, 2, 3, 4 };
vector<int> expected { 2, 2, 3, 3 };
HistogramFilter::medianFilter(hf, arr);
BOOST_TEST(arr == expected, tt::per_element());
}
BOOST_AUTO_TEST_CASE(histogram_median_simple_3_4)
{
HistogramFilter hf(5, 3); // nValues, filterLength
vector<int> arr { 3, 1, 0, 2 };
vector<int> expected { 1, 1, 1, 0 };
HistogramFilter::medianFilter(hf, arr);
BOOST_TEST(arr == expected, tt::per_element());
}
BOOST_AUTO_TEST_CASE(histogram_median_simple_5_4)
{
HistogramFilter hf(5, 5);
vector<int> arr { 3, 1, 0, 2 };
vector<int> expected { 1, 1, 1, 1 };
HistogramFilter::medianFilter(hf, arr);
BOOST_TEST(arr == expected, tt::per_element());
}
BOOST_AUTO_TEST_CASE(histogram_median_order_1)
{
HistogramFilter hf(4, 1);
vector<int> arr { 3, 1, 0, 0 };
vector<int> expected = arr;
HistogramFilter::medianFilter(hf, arr);
BOOST_TEST(arr == expected, tt::per_element());
}
BOOST_AUTO_TEST_CASE(histogram_median_n_1)
{
HistogramFilter hf(3, 6);
vector<int> arr { 1 };
vector<int> expected { 1 };
HistogramFilter::medianFilter(hf, arr);
BOOST_TEST(arr == expected, tt::per_element());
}
BOOST_AUTO_TEST_CASE(histogram_mode_simple_3)
{
HistogramFilter hf(5, 3); // nValues, filterLength
vector<int> arr { 1, 2, 2 };
vector<int> expected { 1, 2, 2 };
HistogramFilter::modalFilter(hf, arr);
BOOST_TEST(arr == expected, tt::per_element());
}
BOOST_AUTO_TEST_CASE(histogram_mode_simple_4)
{
HistogramFilter hf(5, 4); // nValues, filterLength
vector<int> arr { 1, 2, 2, 4 };
vector<int> expected { 2, 2, 2, 2 };
HistogramFilter::modalFilter(hf, arr);
BOOST_TEST(arr == expected, tt::per_element());
}
BOOST_AUTO_TEST_CASE(histogram_mode_simple_3_4)
{
HistogramFilter hf(5, 3); // nValues, filterLength
vector<int> arr { 3, 1, 0, 0 };
vector<int> expected { 1, 0, 0, 0 };
HistogramFilter::modalFilter(hf, arr);
BOOST_TEST(arr == expected, tt::per_element());
}
BOOST_AUTO_TEST_CASE(histogram_mode_simple_5_4)
{
HistogramFilter hf(5, 5);
vector<int> arr { 3, 1, 0, 0 };
vector<int> expected { 0, 0, 0, 0 };
HistogramFilter::modalFilter(hf, arr);
BOOST_TEST(arr == expected, tt::per_element());
}
BOOST_AUTO_TEST_CASE(histogram_mode_order_1)
{
HistogramFilter hf(4, 1);
vector<int> arr { 3, 1, 0, 0 };
vector<int> expected = arr;
HistogramFilter::modalFilter(hf, arr);
BOOST_TEST(arr == expected, tt::per_element());
}
BOOST_AUTO_TEST_CASE(histogram_mode_n_1)
{
HistogramFilter hf(3, 6);
vector<int> arr { 1 };
vector<int> expected { 1 };
HistogramFilter::modalFilter(hf, arr);
BOOST_TEST(arr == expected, tt::per_element());
}
BOOST_AUTO_TEST_CASE(peakpick_nearest_2_1)
{
Peak<double> pp(1);