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 AnalyzeCaptureAudio(AudioBuffer* audio);
30  int ProcessCaptureAudio(AudioBuffer* audio);
31
32  // NoiseSuppression implementation.
33  virtual bool is_enabled() const OVERRIDE;
34  virtual float speech_probability() const OVERRIDE;
35
36 private:
37  // NoiseSuppression implementation.
38  virtual int Enable(bool enable) OVERRIDE;
39  virtual int set_level(Level level) OVERRIDE;
40  virtual Level level() const OVERRIDE;
41
42  // ProcessingComponent implementation.
43  virtual void* CreateHandle() const OVERRIDE;
44  virtual int InitializeHandle(void* handle) const OVERRIDE;
45  virtual int ConfigureHandle(void* handle) const OVERRIDE;
46  virtual void DestroyHandle(void* handle) const OVERRIDE;
47  virtual int num_handles_required() const OVERRIDE;
48  virtual int GetHandleError(void* handle) const OVERRIDE;
49
50  const AudioProcessing* apm_;
51  CriticalSectionWrapper* crit_;
52  Level level_;
53};
54
55}  // namespace webrtc
56
57#endif  // WEBRTC_MODULES_AUDIO_PROCESSING_NOISE_SUPPRESSION_IMPL_H_
58