1/*
2 *  Copyright (c) 2012 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#ifndef WEBRTC_MODULES_AUDIO_PROCESSING_NOISE_SUPPRESSION_IMPL_H_
12#define WEBRTC_MODULES_AUDIO_PROCESSING_NOISE_SUPPRESSION_IMPL_H_
13
14#include "webrtc/base/constructormagic.h"
15#include "webrtc/base/criticalsection.h"
16#include "webrtc/base/scoped_ptr.h"
17#include "webrtc/modules/audio_processing/include/audio_processing.h"
18
19namespace webrtc {
20
21class AudioBuffer;
22
23class NoiseSuppressionImpl : public NoiseSuppression {
24 public:
25  explicit NoiseSuppressionImpl(rtc::CriticalSection* crit);
26  ~NoiseSuppressionImpl() override;
27
28  // TODO(peah): Fold into ctor, once public API is removed.
29  void Initialize(size_t channels, int sample_rate_hz);
30  void AnalyzeCaptureAudio(AudioBuffer* audio);
31  void ProcessCaptureAudio(AudioBuffer* audio);
32
33  // NoiseSuppression implementation.
34  int Enable(bool enable) override;
35  bool is_enabled() const override;
36  int set_level(Level level) override;
37  Level level() const override;
38  float speech_probability() const override;
39
40 private:
41  class Suppressor;
42  rtc::CriticalSection* const crit_;
43  bool enabled_ GUARDED_BY(crit_) = false;
44  Level level_ GUARDED_BY(crit_) = kModerate;
45  size_t channels_ GUARDED_BY(crit_) = 0;
46  int sample_rate_hz_ GUARDED_BY(crit_) = 0;
47  std::vector<rtc::scoped_ptr<Suppressor>> suppressors_ GUARDED_BY(crit_);
48  RTC_DISALLOW_IMPLICIT_CONSTRUCTORS(NoiseSuppressionImpl);
49};
50}  // namespace webrtc
51
52#endif  // WEBRTC_MODULES_AUDIO_PROCESSING_NOISE_SUPPRESSION_IMPL_H_
53