Preparational work for an upcoming addition of a threadchecking scheme for APM

webrtc/modules/audio_processing/gain_control_impl.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.
  */
 
@@ skipping 21 matching lines @@
   }
   assert(false);
   return -1;
 }
 }  // namespace
 
 const size_t GainControlImpl::kAllowedValuesOfSamplesPerFrame1;
 const size_t GainControlImpl::kAllowedValuesOfSamplesPerFrame2;
 
 GainControlImpl::GainControlImpl(const AudioProcessing* apm,
-                                 CriticalSectionWrapper* crit)
+                                 CriticalSectionWrapper* crit,
+                                 rtc::ThreadChecker* render_thread,
+                                 rtc::ThreadChecker* capture_thread)
     : ProcessingComponent(),
       apm_(apm),
       crit_(crit),
+      render_thread_(render_thread),
+      capture_thread_(capture_thread),
       mode_(kAdaptiveAnalog),
       minimum_capture_level_(0),
       maximum_capture_level_(255),
       limiter_enabled_(true),
       target_level_dbfs_(3),
       compression_gain_db_(9),
       analog_capture_level_(0),
       was_analog_level_set_(false),
       stream_is_saturated_(false),
       render_queue_element_max_size_(0) {
   AllocateRenderQueue();
 }
 
 GainControlImpl::~GainControlImpl() {}
 
 int GainControlImpl::ProcessRenderAudio(AudioBuffer* audio) {
+  RTC_DCHECK(render_thread_->CalledOnValidThread());
   if (!is_component_enabled()) {
     return apm_->kNoError;
   }
 
   assert(audio->num_frames_per_band() <= 160);
 
   int buffer_index = 0;
   for (int i = 0; i < num_handles(); i++) {
     Handle* my_handle = static_cast<Handle*>(handle(i));
     int err =
@@ skipping 15 matching lines @@
 
   render_queue_buffer_.resize(buffer_index);
   render_signal_queue_->Insert(&render_queue_buffer_);
 
   return apm_->kNoError;
 }
 
 // Read chunks of data that were received and queued on the render side from
 // a queue. All the data chunks are buffered into the farend signal of the AGC.
 void GainControlImpl::ReadQueuedRenderData() {
+  RTC_DCHECK(capture_thread_->CalledOnValidThread());
   if (!is_component_enabled()) {
     return;
   }
 
   bool samples_read = render_signal_queue_->Remove(&capture_queue_buffer_);
   while (samples_read) {
     int buffer_index = 0;
     const int num_frames_per_band =
         capture_queue_buffer_.size() / num_handles();
     for (int i = 0; i < num_handles(); i++) {
       Handle* my_handle = static_cast<Handle*>(handle(i));
       (void)WebRtcAgc_AddFarend(my_handle, &capture_queue_buffer_[buffer_index],
                                 num_frames_per_band);
 
       buffer_index += num_frames_per_band;
     }
     samples_read = render_signal_queue_->Remove(&capture_queue_buffer_);
   }
 }
 
 int GainControlImpl::AnalyzeCaptureAudio(AudioBuffer* audio) {
+  RTC_DCHECK(capture_thread_->CalledOnValidThread());
   if (!is_component_enabled()) {
     return apm_->kNoError;
   }
 
   assert(audio->num_frames_per_band() <= 160);
   assert(audio->num_channels() == num_handles());
 
   int err = apm_->kNoError;
 
   if (mode_ == kAdaptiveAnalog) {
@@ skipping 30 matching lines @@
         return GetHandleError(my_handle);
       }
 
     }
   }
 
   return apm_->kNoError;
 }
 
 int GainControlImpl::ProcessCaptureAudio(AudioBuffer* audio) {
+  RTC_DCHECK(capture_thread_->CalledOnValidThread());
   if (!is_component_enabled()) {
     return apm_->kNoError;
   }
 
   if (mode_ == kAdaptiveAnalog && !was_analog_level_set_) {
     return apm_->kStreamParameterNotSetError;
   }
 
   assert(audio->num_frames_per_band() <= 160);
   assert(audio->num_channels() == num_handles());
@@ skipping 34 matching lines @@
 
     analog_capture_level_ /= num_handles();
   }
 
   was_analog_level_set_ = false;
   return apm_->kNoError;
 }
 
 // TODO(ajm): ensure this is called under kAdaptiveAnalog.
 int GainControlImpl::set_stream_analog_level(int level) {
+  RTC_DCHECK(capture_thread_->CalledOnValidThread());
   // TODO(peah): Verify that this is really needed to do the reading.
   // here as well as in ProcessStream. It works since these functions
   // are called from the same thread, but it is not nice to do it in two
   // places if not needed.
   ReadQueuedRenderData();
 
   CriticalSectionScoped crit_scoped(crit_);
   was_analog_level_set_ = true;
   if (level < minimum_capture_level_ || level > maximum_capture_level_) {
     return apm_->kBadParameterError;
   }
   analog_capture_level_ = level;
 
   return apm_->kNoError;
 }
 
 int GainControlImpl::stream_analog_level() {
+  RTC_DCHECK(capture_thread_->CalledOnValidThread());
   // TODO(ajm): enable this assertion?
   //assert(mode_ == kAdaptiveAnalog);
 
   return analog_capture_level_;
 }
 
 int GainControlImpl::Enable(bool enable) {
   CriticalSectionScoped crit_scoped(crit_);
   return EnableComponent(enable);
 }
@@ skipping 158 matching lines @@
   return apm_->num_output_channels();
 }
 
 int GainControlImpl::GetHandleError(void* handle) const {
   // The AGC has no get_error() function.
   // (Despite listing errors in its interface...)
   assert(handle != NULL);
   return apm_->kUnspecifiedError;
 }
 }  // namespace webrtc