src/third_party/webrtc/modules/audio_coding/neteq/test/neteq_isac_quality_test.cc

   1 /*
   2  *  Copyright (c) 2014 The WebRTC project authors. All Rights Reserved.
   3  *
   4  *  Use of this source code is governed by a BSD-style license
   5  *  that can be found in the LICENSE file in the root of the source
   6  *  tree. An additional intellectual property rights grant can be found
   7  *  in the file PATENTS.  All contributing project authors may
   8  *  be found in the AUTHORS file in the root of the source tree.
   9  */
  10
  11 #include "webrtc/modules/audio_coding/codecs/isac/fix/interface/isacfix.h"
  12 #include "webrtc/modules/audio_coding/neteq/tools/neteq_quality_test.h"
  13 #include "webrtc/test/testsupport/fileutils.h"
  14
  15 using google::RegisterFlagValidator;
  16 using google::ParseCommandLineFlags;
  17 using std::string;
  18 using testing::InitGoogleTest;
  19
  20 namespace webrtc {
  21 namespace test {
  22
  23 static const int kIsacBlockDurationMs = 30;
  24 static const int kIsacInputSamplingKhz = 16;
  25 static const int kIsacOutputSamplingKhz = 16;
  26
  27 // Define switch for input file name.
  28 static bool ValidateInFilename(const char* flagname, const string& value) {
  29   FILE* fid = fopen(value.c_str(), "rb");
  30   if (fid != NULL) {
  31     fclose(fid);
  32     return true;
  33   }
  34   printf("Invalid input filename.");
  35   return false;
  36 }
  37
  38 DEFINE_string(in_filename,
  39               ResourcePath("audio_coding/speech_mono_16kHz", "pcm"),
  40               "Filename for input audio (should be 16 kHz sampled mono).");
  41
  42 static const bool in_filename_dummy =
  43     RegisterFlagValidator(&FLAGS_in_filename, &ValidateInFilename);
  44
  45 // Define switch for output file name.
  46 static bool ValidateOutFilename(const char* flagname, const string& value) {
  47   FILE* fid = fopen(value.c_str(), "wb");
  48   if (fid != NULL) {
  49     fclose(fid);
  50     return true;
  51   }
  52   printf("Invalid output filename.");
  53   return false;
  54 }
  55
  56 DEFINE_string(out_filename, OutputPath() + "neteq4_isac_quality_test.pcm",
  57               "Name of output audio file.");
  58
  59 static const bool out_filename_dummy =
  60     RegisterFlagValidator(&FLAGS_out_filename, &ValidateOutFilename);
  61
  62 // Define switch for bir rate.
  63 static bool ValidateBitRate(const char* flagname, int32_t value) {
  64   if (value >= 10 && value <= 32)
  65     return true;
  66   printf("Invalid bit rate, should be between 10 and 32 kbps.");
  67   return false;
  68 }
  69
  70 DEFINE_int32(bit_rate_kbps, 32, "Target bit rate (kbps).");
  71
  72 static const bool bit_rate_dummy =
  73     RegisterFlagValidator(&FLAGS_bit_rate_kbps, &ValidateBitRate);
  74
  75 // Define switch for runtime.
  76 static bool ValidateRuntime(const char* flagname, int32_t value) {
  77   if (value > 0)
  78     return true;
  79   printf("Invalid runtime, should be greater than 0.");
  80   return false;
  81 }
  82
  83 DEFINE_int32(runtime_ms, 10000, "Simulated runtime (milliseconds).");
  84
  85 static const bool runtime_dummy =
  86     RegisterFlagValidator(&FLAGS_runtime_ms, &ValidateRuntime);
  87
  88 class NetEqIsacQualityTest : public NetEqQualityTest {
  89  protected:
  90   NetEqIsacQualityTest();
  91   virtual void SetUp() OVERRIDE;
  92   virtual void TearDown() OVERRIDE;
  93   virtual int EncodeBlock(int16_t* in_data, int block_size_samples,
  94                           uint8_t* payload, int max_bytes);
  95  private:
  96   ISACFIX_MainStruct* isac_encoder_;
  97   int bit_rate_kbps_;
  98 };
  99
 100 NetEqIsacQualityTest::NetEqIsacQualityTest()
 101     : NetEqQualityTest(kIsacBlockDurationMs, kIsacInputSamplingKhz,
 102                        kIsacOutputSamplingKhz,
 103                        kDecoderISAC,
 104                        1,
 105                        FLAGS_in_filename,
 106                        FLAGS_out_filename),
 107       isac_encoder_(NULL),
 108       bit_rate_kbps_(FLAGS_bit_rate_kbps) {
 109 }
 110
 111 void NetEqIsacQualityTest::SetUp() {
 112   // Create encoder memory.
 113   WebRtcIsacfix_Create(&isac_encoder_);
 114   ASSERT_TRUE(isac_encoder_ != NULL);
 115   EXPECT_EQ(0, WebRtcIsacfix_EncoderInit(isac_encoder_, 1));
 116   // Set bitrate and block length.
 117   EXPECT_EQ(0, WebRtcIsacfix_Control(isac_encoder_, bit_rate_kbps_ * 1000,
 118                                      kIsacBlockDurationMs));
 119   NetEqQualityTest::SetUp();
 120 }
 121
 122 void NetEqIsacQualityTest::TearDown() {
 123   // Free memory.
 124   EXPECT_EQ(0, WebRtcIsacfix_Free(isac_encoder_));
 125   NetEqQualityTest::TearDown();
 126 }
 127
 128 int NetEqIsacQualityTest::EncodeBlock(int16_t* in_data,
 129                                       int block_size_samples,
 130                                       uint8_t* payload, int max_bytes) {
 131   // ISAC takes 10 ms for every call.
 132   const int subblocks = kIsacBlockDurationMs / 10;
 133   const int subblock_length = 10 * kIsacInputSamplingKhz;
 134   int value = 0;
 135
 136   int pointer = 0;
 137   for (int idx = 0; idx < subblocks; idx++, pointer += subblock_length) {
 138     // The Isac encoder does not perform encoding (and returns 0) until it
 139     // receives a sequence of sub-blocks that amount to the frame duration.
 140     EXPECT_EQ(0, value);
 141     value = WebRtcIsacfix_Encode(isac_encoder_, &in_data[pointer], payload);
 142   }
 143   EXPECT_GT(value, 0);
 144   return value;
 145 }
 146
 147 TEST_F(NetEqIsacQualityTest, Test) {
 148   Simulate(FLAGS_runtime_ms);
 149 }
 150
 151 }  // namespace test
 152 }  // namespace webrtc