| Index: webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer_unittest.cc
|
| diff --git a/webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer_unittest.cc b/webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer_unittest.cc
|
| index b0f94ec5e25d1ea6ed6c55639a6bc07524203072..b59ae36d8b2de94b9c0cafc33a253fd23c202cf8 100644
|
| --- a/webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer_unittest.cc
|
| +++ b/webrtc/modules/audio_processing/intelligibility/intelligibility_enhancer_unittest.cc
|
| @@ -26,54 +26,184 @@ namespace {
|
|
|
| // Target output for ERB create test. Generated with matlab.
|
| const float kTestCenterFreqs[] = {
|
| - 13.169f, 26.965f, 41.423f, 56.577f, 72.461f, 89.113f, 106.57f, 124.88f,
|
| - 144.08f, 164.21f, 185.34f, 207.5f, 230.75f, 255.16f, 280.77f, 307.66f,
|
| - 335.9f, 365.56f, 396.71f, 429.44f, 463.84f, 500.f};
|
| -const float kTestFilterBank[][9] = {
|
| - {0.2f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| - {0.2f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| - {0.2f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| - {0.2f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| - {0.2f, 0.25f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| - {0.f, 0.25f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| - {0.f, 0.25f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| - {0.f, 0.25f, 0.25f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| - {0.f, 0.f, 0.25f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| - {0.f, 0.f, 0.25f, 0.142857f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| - {0.f, 0.f, 0.25f, 0.285714f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| - {0.f, 0.f, 0.f, 0.285714f, 0.142857f, 0.f, 0.f, 0.f, 0.f},
|
| - {0.f, 0.f, 0.f, 0.285714f, 0.285714f, 0.f, 0.f, 0.f, 0.f},
|
| - {0.f, 0.f, 0.f, 0.f, 0.285714f, 0.142857f, 0.f, 0.f, 0.f},
|
| - {0.f, 0.f, 0.f, 0.f, 0.285714f, 0.285714f, 0.f, 0.f, 0.f},
|
| - {0.f, 0.f, 0.f, 0.f, 0.f, 0.285714f, 0.142857f, 0.f, 0.f},
|
| - {0.f, 0.f, 0.f, 0.f, 0.f, 0.285714f, 0.285714f, 0.f, 0.f},
|
| - {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.285714f, 0.142857f, 0.f},
|
| - {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.285714f, 0.285714f, 0.f},
|
| - {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.285714f, 0.f},
|
| - {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.285714f, 0.5f},
|
| - {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.5f}};
|
| + 14.5213f, 29.735f, 45.6781f, 62.3884f, 79.9058f, 98.2691f, 117.521f,
|
| + 137.708f, 158.879f, 181.084f, 204.378f, 228.816f, 254.459f, 281.371f,
|
| + 309.618f, 339.273f, 370.411f, 403.115f, 437.469f, 473.564f, 511.497f,
|
| + 551.371f, 593.293f, 637.386f, 683.77f, 732.581f, 783.96f, 838.06f,
|
| + 895.046f, 955.09f, 1018.38f, 1085.13f, 1155.54f, 1229.85f, 1308.32f,
|
| + 1391.22f, 1478.83f, 1571.5f, 1669.55f, 1773.37f, 1883.37f, 2000.f};
|
| +const float kTestFilterBank[][33] = {
|
| + {0.2f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.2f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.2f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.2f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.2f, 0.25f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.25f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.25f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.25f, 0.25f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.25f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.25f, 0.142857f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.25f, 0.285714f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.285714f, 0.142857f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.285714f, 0.285714f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.285714f, 0.142857f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.285714f, 0.285714f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.285714f, 0.142857f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.285714f, 0.285714f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.285714f, 0.142857f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.285714f, 0.285714f, 0.157895f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.285714f, 0.210526f, 0.117647f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.285714f, 0.315789f, 0.176471f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.315789f, 0.352941f, 0.142857f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.352941f, 0.285714f,
|
| + 0.157895f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.285714f,
|
| + 0.210526f, 0.111111f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.285714f, 0.315789f, 0.222222f, 0.111111f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.315789f, 0.333333f, 0.222222f, 0.111111f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.333333f, 0.333333f, 0.222222f, 0.111111f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.333333f, 0.333333f, 0.222222f, 0.111111f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.333333f, 0.333333f, 0.222222f, 0.111111f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.333333f, 0.333333f, 0.222222f,
|
| + 0.108108f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.333333f, 0.333333f,
|
| + 0.243243f, 0.153846f, 0.0833333f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.333333f,
|
| + 0.324324f, 0.230769f, 0.166667f, 0.0909091f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.324324f, 0.307692f, 0.25f, 0.181818f, 0.0833333f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.307692f, 0.333333f,
|
| + 0.363636f, 0.25f, 0.151515f, 0.0793651f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.166667f, 0.363636f, 0.333333f, 0.242424f,
|
| + 0.190476f, 0.133333f, 0.0689655f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.333333f, 0.30303f, 0.253968f, 0.2f, 0.137931f,
|
| + 0.0714286f, 0.f, 0.f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.30303f, 0.31746f, 0.333333f, 0.275862f, 0.214286f,
|
| + 0.125f, 0.0655738f, 0.f, 0.f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.15873f, 0.333333f, 0.344828f, 0.357143f,
|
| + 0.25f, 0.196721f, 0.137931f, 0.0816327f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.172414f, 0.357143f,
|
| + 0.3125f, 0.245902f, 0.172414f, 0.102041f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.3125f, 0.327869f, 0.344828f, 0.204082f, 0.f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.163934f, 0.344828f, 0.408163f, 0.5f},
|
| + {0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.204082f, 0.5f}};
|
| static_assert(arraysize(kTestCenterFreqs) == arraysize(kTestFilterBank),
|
| "Test filterbank badly initialized.");
|
|
|
| // Target output for gain solving test. Generated with matlab.
|
| const size_t kTestStartFreq = 12; // Lowest integral frequency for ERBs.
|
| -const float kTestZeroVar[] = {1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 1.f,
|
| - 1.f, 1.f, 1.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| - 0.f, 0.f, 0.f, 0.f, 0.f, 0.f};
|
| +const float kTestZeroVar[] = {
|
| + 1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0};
|
| static_assert(arraysize(kTestCenterFreqs) == arraysize(kTestZeroVar),
|
| "Power test data badly initialized.");
|
| const float kTestNonZeroVarLambdaTop[] = {
|
| - 1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 1.f,
|
| - 1.f, 1.f, 1.f, 0.f, 0.f, 0.0351f, 0.0636f, 0.0863f,
|
| - 0.1037f, 0.1162f, 0.1236f, 0.1251f, 0.1189f, 0.0993f};
|
| + 1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 1.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f,
|
| + 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0.f, 0};
|
| static_assert(arraysize(kTestCenterFreqs) ==
|
| arraysize(kTestNonZeroVarLambdaTop),
|
| "Power test data badly initialized.");
|
| const float kMaxTestError = 0.005f;
|
|
|
| // Enhancer initialization parameters.
|
| -const int kSamples = 2000;
|
| -const int kSampleRate = 1000;
|
| +const int kSamples = 1000;
|
| +const int kSampleRate = 4000;
|
| const int kNumChannels = 1;
|
| const int kFragmentSize = kSampleRate / 100;
|
|
|
| @@ -83,13 +213,11 @@ class IntelligibilityEnhancerTest : public ::testing::Test {
|
| protected:
|
| IntelligibilityEnhancerTest()
|
| : clear_data_(kSamples), noise_data_(kSamples), orig_data_(kSamples) {
|
| - config_.sample_rate_hz = kSampleRate;
|
| - enh_.reset(new IntelligibilityEnhancer(config_));
|
| + enh_.reset(new IntelligibilityEnhancer(kSampleRate, kNumChannels));
|
| }
|
|
|
| bool CheckUpdate() {
|
| - config_.sample_rate_hz = kSampleRate;
|
| - enh_.reset(new IntelligibilityEnhancer(config_));
|
| + enh_.reset(new IntelligibilityEnhancer(kSampleRate, kNumChannels));
|
| float* clear_cursor = &clear_data_[0];
|
| float* noise_cursor = &noise_data_[0];
|
| for (int i = 0; i < kSamples; i += kFragmentSize) {
|
| @@ -105,7 +233,6 @@ class IntelligibilityEnhancerTest : public ::testing::Test {
|
| return false;
|
| }
|
|
|
| - IntelligibilityEnhancer::Config config_;
|
| std::unique_ptr<IntelligibilityEnhancer> enh_;
|
| std::vector<float> clear_data_;
|
| std::vector<float> noise_data_;
|
| @@ -115,9 +242,9 @@ class IntelligibilityEnhancerTest : public ::testing::Test {
|
| // For each class of generated data, tests that render stream is updated when
|
| // it should be.
|
| TEST_F(IntelligibilityEnhancerTest, TestRenderUpdate) {
|
| - std::fill(noise_data_.begin(), noise_data_.end(), 0.0f);
|
| - std::fill(orig_data_.begin(), orig_data_.end(), 0.0f);
|
| - std::fill(clear_data_.begin(), clear_data_.end(), 0.0f);
|
| + std::fill(noise_data_.begin(), noise_data_.end(), 0.f);
|
| + std::fill(orig_data_.begin(), orig_data_.end(), 0.f);
|
| + std::fill(clear_data_.begin(), clear_data_.end(), 0.f);
|
| EXPECT_FALSE(CheckUpdate());
|
| std::srand(1);
|
| auto float_rand = []() { return std::rand() * 2.f / RAND_MAX - 1; };
|
| @@ -148,9 +275,8 @@ TEST_F(IntelligibilityEnhancerTest, TestSolveForGains) {
|
| std::vector<float> sols(enh_->bank_size_);
|
| float lambda = -0.001f;
|
| for (size_t i = 0; i < enh_->bank_size_; i++) {
|
| - enh_->filtered_clear_pow_[i] = 0.0f;
|
| - enh_->filtered_noise_pow_[i] = 0.0f;
|
| - enh_->rho_[i] = 0.02f;
|
| + enh_->filtered_clear_pow_[i] = 0.f;
|
| + enh_->filtered_noise_pow_[i] = 0.f;
|
| }
|
| enh_->SolveForGainsGivenLambda(lambda, enh_->start_freq_, &sols[0]);
|
| for (size_t i = 0; i < enh_->bank_size_; i++) {
|
| @@ -164,7 +290,7 @@ TEST_F(IntelligibilityEnhancerTest, TestSolveForGains) {
|
| for (size_t i = 0; i < enh_->bank_size_; i++) {
|
| EXPECT_NEAR(kTestNonZeroVarLambdaTop[i], sols[i], kMaxTestError);
|
| }
|
| - lambda = -1.0;
|
| + lambda = -1.f;
|
| enh_->SolveForGainsGivenLambda(lambda, enh_->start_freq_, &sols[0]);
|
| for (size_t i = 0; i < enh_->bank_size_; i++) {
|
| EXPECT_NEAR(kTestZeroVar[i], sols[i], kMaxTestError);
|
|
|
Chromium Code Reviews