1/*
2 *  Copyright (c) 2011 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_MAIN_SOURCE_VOICE_DETECTION_IMPL_H_
12#define WEBRTC_MODULES_AUDIO_PROCESSING_MAIN_SOURCE_VOICE_DETECTION_IMPL_H_
13
14#include "audio_processing.h"
15#include "processing_component.h"
16
17namespace webrtc {
18class AudioProcessingImpl;
19class AudioBuffer;
20
21class VoiceDetectionImpl : public VoiceDetection,
22                           public ProcessingComponent {
23 public:
24  explicit VoiceDetectionImpl(const AudioProcessingImpl* apm);
25  virtual ~VoiceDetectionImpl();
26
27  int ProcessCaptureAudio(AudioBuffer* audio);
28
29  // VoiceDetection implementation.
30  virtual bool is_enabled() const;
31
32  // ProcessingComponent implementation.
33  virtual int Initialize();
34  virtual int get_version(char* version, int version_len_bytes) const;
35
36 private:
37  // VoiceDetection implementation.
38  virtual int Enable(bool enable);
39  virtual int set_stream_has_voice(bool has_voice);
40  virtual bool stream_has_voice() const;
41  virtual int set_likelihood(Likelihood likelihood);
42  virtual Likelihood likelihood() const;
43  virtual int set_frame_size_ms(int size);
44  virtual int frame_size_ms() const;
45
46  // ProcessingComponent implementation.
47  virtual void* CreateHandle() const;
48  virtual int InitializeHandle(void* handle) const;
49  virtual int ConfigureHandle(void* handle) const;
50  virtual int DestroyHandle(void* handle) const;
51  virtual int num_handles_required() const;
52  virtual int GetHandleError(void* handle) const;
53
54  const AudioProcessingImpl* apm_;
55  bool stream_has_voice_;
56  bool using_external_vad_;
57  Likelihood likelihood_;
58  int frame_size_ms_;
59  int frame_size_samples_;
60};
61}  // namespace webrtc
62
63#endif  // WEBRTC_MODULES_AUDIO_PROCESSING_MAIN_SOURCE_VOICE_DETECTION_IMPL_H_
64