1b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org/*
2b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org *  Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
3b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org *
4b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org *  Use of this source code is governed by a BSD-style license
5b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org *  that can be found in the LICENSE file in the root of the source
6b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org *  tree. An additional intellectual property rights grant can be found
7b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org *  in the file PATENTS.  All contributing project authors may
8b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org *  be found in the AUTHORS file in the root of the source tree.
9b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org */
10b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org
11b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org#ifndef WEBRTC_MODULES_AUDIO_PROCESSING_NOISE_SUPPRESSION_IMPL_H_
12b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org#define WEBRTC_MODULES_AUDIO_PROCESSING_NOISE_SUPPRESSION_IMPL_H_
13b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org
149fb16139d917ba32720e031d3c871987d418668fpbos@webrtc.org#include "webrtc/modules/audio_processing/include/audio_processing.h"
159fb16139d917ba32720e031d3c871987d418668fpbos@webrtc.org#include "webrtc/modules/audio_processing/processing_component.h"
16b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org
17b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.orgnamespace webrtc {
18a1a60018a1f1ec863451ad0ed4eae58239882920andrew@webrtc.org
19b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.orgclass AudioBuffer;
20a1a60018a1f1ec863451ad0ed4eae58239882920andrew@webrtc.orgclass CriticalSectionWrapper;
21b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org
22b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.orgclass NoiseSuppressionImpl : public NoiseSuppression,
23b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org                             public ProcessingComponent {
24b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org public:
25a1a60018a1f1ec863451ad0ed4eae58239882920andrew@webrtc.org  NoiseSuppressionImpl(const AudioProcessing* apm,
26a1a60018a1f1ec863451ad0ed4eae58239882920andrew@webrtc.org                       CriticalSectionWrapper* crit);
27b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org  virtual ~NoiseSuppressionImpl();
28b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org
29ba6a0c5315dc3af937e66cd3d3ec7f646895a04daluebs@webrtc.org  int AnalyzeCaptureAudio(AudioBuffer* audio);
30b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org  int ProcessCaptureAudio(AudioBuffer* audio);
31b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org
32b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org  // NoiseSuppression implementation.
3324add9260e60f34e85ed8d936da014128e19fd84pbos@webrtc.org  virtual bool is_enabled() const OVERRIDE;
3424add9260e60f34e85ed8d936da014128e19fd84pbos@webrtc.org  virtual float speech_probability() const OVERRIDE;
35b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org
36b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org private:
37b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org  // NoiseSuppression implementation.
3824add9260e60f34e85ed8d936da014128e19fd84pbos@webrtc.org  virtual int Enable(bool enable) OVERRIDE;
3924add9260e60f34e85ed8d936da014128e19fd84pbos@webrtc.org  virtual int set_level(Level level) OVERRIDE;
4024add9260e60f34e85ed8d936da014128e19fd84pbos@webrtc.org  virtual Level level() const OVERRIDE;
41b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org
42b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org  // ProcessingComponent implementation.
4324add9260e60f34e85ed8d936da014128e19fd84pbos@webrtc.org  virtual void* CreateHandle() const OVERRIDE;
4424add9260e60f34e85ed8d936da014128e19fd84pbos@webrtc.org  virtual int InitializeHandle(void* handle) const OVERRIDE;
4524add9260e60f34e85ed8d936da014128e19fd84pbos@webrtc.org  virtual int ConfigureHandle(void* handle) const OVERRIDE;
46ffc2de0133683fb103eef61f21134f469bc099dbbjornv@webrtc.org  virtual void DestroyHandle(void* handle) const OVERRIDE;
4724add9260e60f34e85ed8d936da014128e19fd84pbos@webrtc.org  virtual int num_handles_required() const OVERRIDE;
4824add9260e60f34e85ed8d936da014128e19fd84pbos@webrtc.org  virtual int GetHandleError(void* handle) const OVERRIDE;
49b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org
50a1a60018a1f1ec863451ad0ed4eae58239882920andrew@webrtc.org  const AudioProcessing* apm_;
51a1a60018a1f1ec863451ad0ed4eae58239882920andrew@webrtc.org  CriticalSectionWrapper* crit_;
52b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org  Level level_;
53b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org};
54a1a60018a1f1ec863451ad0ed4eae58239882920andrew@webrtc.org
55b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org}  // namespace webrtc
56b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org
57b015cbede88899f67a53fbbe581b02ce8e32794andrew@webrtc.org#endif  // WEBRTC_MODULES_AUDIO_PROCESSING_NOISE_SUPPRESSION_IMPL_H_
58