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/modules/audio_processing/include/audio_processing.h"
15#include "webrtc/modules/audio_processing/processing_component.h"
16
17namespace webrtc {
18
19class AudioBuffer;
20class CriticalSectionWrapper;
21
22class NoiseSuppressionImpl : public NoiseSuppression,
23                             public ProcessingComponent {
24 public:
25  NoiseSuppressionImpl(const AudioProcessing* apm,
26                       CriticalSectionWrapper* crit);
27  virtual ~NoiseSuppressionImpl();
28
29  int ProcessCaptureAudio(AudioBuffer* audio);
30
31  // NoiseSuppression implementation.
32  virtual bool is_enabled() const OVERRIDE;
33  virtual float speech_probability() const OVERRIDE;
34
35 private:
36  // NoiseSuppression implementation.
37  virtual int Enable(bool enable) OVERRIDE;
38  virtual int set_level(Level level) OVERRIDE;
39  virtual Level level() const OVERRIDE;
40
41  // ProcessingComponent implementation.
42  virtual void* CreateHandle() const OVERRIDE;
43  virtual int InitializeHandle(void* handle) const OVERRIDE;
44  virtual int ConfigureHandle(void* handle) const OVERRIDE;
45  virtual void DestroyHandle(void* handle) const OVERRIDE;
46  virtual int num_handles_required() const OVERRIDE;
47  virtual int GetHandleError(void* handle) const OVERRIDE;
48
49  const AudioProcessing* apm_;
50  CriticalSectionWrapper* crit_;
51  Level level_;
52};
53
54}  // namespace webrtc
55
56#endif  // WEBRTC_MODULES_AUDIO_PROCESSING_NOISE_SUPPRESSION_IMPL_H_
57