Reduce the runtime of some ACM tests in modules_tests

webrtc/modules/audio_coding/test/PCMFile.cc

 /*
  *  Copyright (c) 2012 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.
  */
 
-#include "PCMFile.h"
+#include "webrtc/modules/audio_coding/test/PCMFile.h"
 
 #include <ctype.h>
 #include <stdio.h>
 #include <string.h>
 
 #include "testing/gtest/include/gtest/gtest.h"
 #include "webrtc/modules/include/module_common_types.h"
 
 namespace webrtc {
 
 #define MAX_FILE_NAME_LENGTH_BYTE 500
 
 PCMFile::PCMFile()
     : pcm_file_(NULL),
       samples_10ms_(160),
       frequency_(16000),
       end_of_file_(false),
       auto_rewind_(false),
       rewinded_(false),
       read_stereo_(false),
-      save_stereo_(false) {
+      save_stereo_(false),
+      num_10ms_blocks_to_read_(rtc::Optional<int>()) {

[ivoc, 2015/12/10 12:23:57]

Can this be initialized in the .h file as well? Or is there a reason not to do
that?

[hlundin-webrtc, 2015/12/10 12:49:59]

In fact, it's not needed at all. The default ctor for rtc::Optional will give an
unset value.

   timestamp_ = (((uint32_t) rand() & 0x0000FFFF) << 16) |
       ((uint32_t) rand() & 0x0000FFFF);
 }
 
 PCMFile::PCMFile(uint32_t timestamp)
     : pcm_file_(NULL),
       samples_10ms_(160),
       frequency_(16000),
       end_of_file_(false),
       auto_rewind_(false),
       rewinded_(false),
       read_stereo_(false),
-      save_stereo_(false) {
+      save_stereo_(false),
+      num_10ms_blocks_to_read_(rtc::Optional<int>()) {
   timestamp_ = timestamp;
 }
 
 int16_t PCMFile::ChooseFile(std::string* file_name, int16_t max_len,
                             uint16_t* frequency_hz) {
   char tmp_name[MAX_FILE_NAME_LENGTH_BYTE];
 
   EXPECT_TRUE(fgets(tmp_name, MAX_FILE_NAME_LENGTH_BYTE, stdin) != NULL);
   tmp_name[MAX_FILE_NAME_LENGTH_BYTE - 1] = '\0';
   int16_t n = 0;
@@ skipping 74 matching lines @@
       rewinded_ = true;
     } else {
       end_of_file_ = true;
     }
   }
   audio_frame.samples_per_channel_ = samples_10ms_;
   audio_frame.sample_rate_hz_ = frequency_;
   audio_frame.num_channels_ = channels;
   audio_frame.timestamp_ = timestamp_;
   timestamp_ += samples_10ms_;
+  ++blocks_read_;
+  if (num_10ms_blocks_to_read_ && blocks_read_ >= *num_10ms_blocks_to_read_)
+    end_of_file_ = true;
   return samples_10ms_;
 }
 
 void PCMFile::Write10MsData(AudioFrame& audio_frame) {
   if (audio_frame.num_channels_ == 1) {
     if (!save_stereo_) {
       if (fwrite(audio_frame.data_, sizeof(uint16_t),
                  audio_frame.samples_per_channel_, pcm_file_) !=
           static_cast<size_t>(audio_frame.samples_per_channel_)) {
         return;
@@ skipping 25 matching lines @@
 void PCMFile::Write10MsData(int16_t* playout_buffer, size_t length_smpls) {
   if (fwrite(playout_buffer, sizeof(uint16_t), length_smpls, pcm_file_) !=
       length_smpls) {
     return;
   }
 }
 
 void PCMFile::Close() {
   fclose(pcm_file_);
   pcm_file_ = NULL;
+  blocks_read_ = 0;
+}
+
+void PCMFile::FastForward(int num_10ms_blocks) {
+  const int channels = read_stereo_ ? 2 : 1;
+  size_t num_bytes_to_move =
+      num_10ms_blocks * sizeof(int16_t) * samples_10ms_ * channels;
+  int error = fseek(pcm_file_, num_bytes_to_move, SEEK_CUR);
+  RTC_DCHECK_EQ(error, 0);
 }
 
 void PCMFile::Rewind() {
   rewind(pcm_file_);
   end_of_file_ = false;
+  blocks_read_ = 0;
 }
 
 bool PCMFile::Rewinded() {
   return rewinded_;
 }
 
 void PCMFile::SaveStereo(bool is_stereo) {
   save_stereo_ = is_stereo;
 }
 
 void PCMFile::ReadStereo(bool is_stereo) {
   read_stereo_ = is_stereo;
 }
 
+void PCMFile::SetNum10MsBlocksToRead(int value) {
+  num_10ms_blocks_to_read_ = rtc::Optional<int>(value);
+}
+
 }  // namespace webrtc