Pull out the PostFilter to its own NonlinearBeamformer API

webrtc/modules/audio_processing/beamformer/nonlinear_beamformer_unittest.cc

 /*
  *  Copyright (c) 2015 The WebRTC project authors. All Rights Reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
@@ skipping 47 matching lines @@
 void ProcessOneFrame(int sample_rate_hz,
                      AudioBuffer* capture_audio_buffer,
                      Beamformer<float>* beamformer) {
   if (sample_rate_hz > AudioProcessing::kSampleRate16kHz) {
     capture_audio_buffer->SplitIntoFrequencyBands();
   }
 
   beamformer->ProcessChunk(*capture_audio_buffer->split_data_f(),
                            capture_audio_buffer->split_data_f());
   capture_audio_buffer->set_num_channels(1);
+  beamformer->PostFilter(*capture_audio_buffer->split_data_f(),
+                         capture_audio_buffer->split_data_f());
 
   if (sample_rate_hz > AudioProcessing::kSampleRate16kHz) {
     capture_audio_buffer->MergeFrequencyBands();
   }
 }
 
 int BeamformerSampleRate(int sample_rate_hz) {
   return (sample_rate_hz > AudioProcessing::kSampleRate16kHz
               ? AudioProcessing::kSampleRate16kHz
               : sample_rate_hz);
@@ skipping 177 matching lines @@
      DISABLED_Stereo8kHz_ArrayGeometry1_TargetDirection1) {
   const float kOutputReference[] = {0.001318f, -0.001091f, 0.000990f,
                                     0.001318f, -0.001091f, 0.000990f};
 
   RunBitExactnessTest(AudioProcessing::kSampleRate8kHz, CreateArrayGeometry(1),
                       TargetDirection1, kOutputReference);
 }
 
 TEST(BeamformerBitExactnessTest,
      Stereo16kHz_ArrayGeometry1_TargetDirection1) {
-  const float kOutputReference[] = {0.000064f, 0.000211f, 0.000075f,
-                                    0.000064f, 0.000211f, 0.000075f};
+  const float kOutputReference[] = {0.000046f, 0.000297f, 0.000056f,
+                                    0.000046f, 0.000297f, 0.000056f};
 
   RunBitExactnessTest(AudioProcessing::kSampleRate16kHz, CreateArrayGeometry(1),
                       TargetDirection1, kOutputReference);
 }
 
 TEST(BeamformerBitExactnessTest,
      Stereo32kHz_ArrayGeometry1_TargetDirection1) {
   const float kOutputReference[] = {0.000183f, 0.000183f, 0.000183f,
                                     0.000183f, 0.000183f, 0.000183f};
 
   RunBitExactnessTest(AudioProcessing::kSampleRate32kHz, CreateArrayGeometry(1),
                       TargetDirection1, kOutputReference);
 }
 
 TEST(BeamformerBitExactnessTest,
      Stereo48kHz_ArrayGeometry1_TargetDirection1) {
-  const float kOutputReference[] = {0.000155f, 0.000152f, 0.000159f,
-                                    0.000155f, 0.000152f, 0.000159f};
+  const float kOutputReference[] = {0.000228f, 0.000224f, 0.000231f,
+                                    0.000228f, 0.000224f, 0.000231f};
 
   RunBitExactnessTest(AudioProcessing::kSampleRate48kHz, CreateArrayGeometry(1),
                       TargetDirection1, kOutputReference);
 }
 
 // TODO(peah): Investigate why the nonlinear_beamformer.cc causes a DCHECK in
 // this setup.
 TEST(BeamformerBitExactnessTest,
      DISABLED_Stereo8kHz_ArrayGeometry1_TargetDirection2) {
   const float kOutputReference[] = {0.001144f,  -0.001026f, 0.001074f,
                                     -0.016205f, -0.007324f, -0.015656f};
 
   RunBitExactnessTest(AudioProcessing::kSampleRate8kHz, CreateArrayGeometry(1),
                       TargetDirection2, kOutputReference);
 }
 
 TEST(BeamformerBitExactnessTest,
      Stereo16kHz_ArrayGeometry1_TargetDirection2) {
-  const float kOutputReference[] = {0.001144f, -0.001026f, 0.001074f,
-                                    0.001144f, -0.001026f, 0.001074f};
+  const float kOutputReference[] = {0.001807f, -0.001761f, 0.001908f,
+                                    0.001807f, -0.001761f, 0.001908f};
 
   RunBitExactnessTest(AudioProcessing::kSampleRate16kHz, CreateArrayGeometry(1),
                       TargetDirection2, kOutputReference);
 }
 
 TEST(BeamformerBitExactnessTest,
      Stereo32kHz_ArrayGeometry1_TargetDirection2) {
-  const float kOutputReference[] = {0.000732f, -0.000397f, 0.000610f,
-                                    0.000732f, -0.000397f, 0.000610f};
+  const float kOutputReference[] = {0.000916f, -0.000641f, 0.000122f,
+                                    0.000916f, -0.000641f, 0.000122f};
 
   RunBitExactnessTest(AudioProcessing::kSampleRate32kHz, CreateArrayGeometry(1),
                       TargetDirection2, kOutputReference);
 }
 
 TEST(BeamformerBitExactnessTest,
      Stereo48kHz_ArrayGeometry1_TargetDirection2) {
-  const float kOutputReference[] = {0.000106f, -0.000464f, 0.000188f,
-                                    0.000106f, -0.000464f, 0.000188f};
+  const float kOutputReference[] = {0.000274f, -0.000307f, 0.000280f,
+                                    0.000274f, -0.000307f, 0.000280f};
 
   RunBitExactnessTest(AudioProcessing::kSampleRate48kHz, CreateArrayGeometry(1),
                       TargetDirection2, kOutputReference);
 }
 
 TEST(BeamformerBitExactnessTest,
      Stereo8kHz_ArrayGeometry2_TargetDirection2) {
-  const float kOutputReference[] = {-0.000649f, 0.000576f, -0.000148f,
-                                    -0.000649f, 0.000576f, -0.000148f};
+  const float kOutputReference[] = {-0.001365f, -0.000137f, -0.000297f,
+                                    -0.001365f, -0.000137f, -0.000297f};
 
   RunBitExactnessTest(AudioProcessing::kSampleRate8kHz, CreateArrayGeometry(2),
                       TargetDirection2, kOutputReference);
 }
 
 TEST(BeamformerBitExactnessTest,
      Stereo16kHz_ArrayGeometry2_TargetDirection2) {
-  const float kOutputReference[] = {0.000808f, -0.000695f, 0.000739f,
-                                    0.000808f, -0.000695f, 0.000739f};
+  const float kOutputReference[] = {0.001243f, -0.001196f, 0.001312f,
+                                    0.001243f, -0.001196f, 0.001312f};
 
   RunBitExactnessTest(AudioProcessing::kSampleRate16kHz, CreateArrayGeometry(2),
                       TargetDirection2, kOutputReference);
 }
 
 TEST(BeamformerBitExactnessTest,
      Stereo32kHz_ArrayGeometry2_TargetDirection2) {
-  const float kOutputReference[] = {0.000580f, -0.000183f, 0.000458f,
-                                    0.000580f, -0.000183f, 0.000458f};
+  const float kOutputReference[] = {0.000671f, -0.000366f, 0.000122f,
+                                    0.000671f, -0.000366f, 0.000122f};
 
   RunBitExactnessTest(AudioProcessing::kSampleRate32kHz, CreateArrayGeometry(2),
                       TargetDirection2, kOutputReference);
 }
 
 TEST(BeamformerBitExactnessTest,
      Stereo48kHz_ArrayGeometry2_TargetDirection2) {
-  const float kOutputReference[] = {0.000075f, -0.000288f, 0.000156f,
-                                    0.000075f, -0.000288f, 0.000156f};
+  const float kOutputReference[] = {0.000135f, -0.000236f, 0.000179f,
+                                    0.000135f, -0.000236f, 0.000179f};
 
   RunBitExactnessTest(AudioProcessing::kSampleRate48kHz, CreateArrayGeometry(2),
                       TargetDirection2, kOutputReference);
 }
 
 // TODO(peah): Investigate why the nonlinear_beamformer.cc causes a DCHECK in
 // this setup.
 TEST(BeamformerBitExactnessTest,
      DISABLED_Stereo16kHz_ArrayGeometry3_TargetDirection1) {
   const float kOutputReference[] = {-0.000161f, 0.000171f, -0.000096f,
                                     0.001007f,  0.000427f, 0.000977f};
 
   RunBitExactnessTest(AudioProcessing::kSampleRate16kHz, CreateArrayGeometry(3),
                       TargetDirection1, kOutputReference);
 }
 
 }  // namespace webrtc