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 
17 namespace webrtc {
18 class AudioProcessingImpl;
19 class AudioBuffer;
20 
21 class 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