| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| 11 #ifndef WEBRTC_MODULES_AUDIO_PROCESSING_VOICE_DETECTION_IMPL_H_ | 11 #ifndef WEBRTC_MODULES_AUDIO_PROCESSING_VOICE_DETECTION_IMPL_H_ |
| 12 #define WEBRTC_MODULES_AUDIO_PROCESSING_VOICE_DETECTION_IMPL_H_ | 12 #define WEBRTC_MODULES_AUDIO_PROCESSING_VOICE_DETECTION_IMPL_H_ |
| 13 | 13 |
| 14 #include "webrtc/base/criticalsection.h" |
| 14 #include "webrtc/modules/audio_processing/include/audio_processing.h" | 15 #include "webrtc/modules/audio_processing/include/audio_processing.h" |
| 15 #include "webrtc/modules/audio_processing/processing_component.h" | 16 #include "webrtc/modules/audio_processing/processing_component.h" |
| 16 | 17 |
| 17 namespace webrtc { | 18 namespace webrtc { |
| 18 | 19 |
| 19 class AudioBuffer; | 20 class AudioBuffer; |
| 20 class CriticalSectionWrapper; | |
| 21 | 21 |
| 22 class VoiceDetectionImpl : public VoiceDetection, | 22 class VoiceDetectionImpl : public VoiceDetection, |
| 23 public ProcessingComponent { | 23 public ProcessingComponent { |
| 24 public: | 24 public: |
| 25 VoiceDetectionImpl(const AudioProcessing* apm, CriticalSectionWrapper* crit); | 25 VoiceDetectionImpl(const AudioProcessing* apm, |
| 26 rtc::CriticalSection* crit); |
| 26 virtual ~VoiceDetectionImpl(); | 27 virtual ~VoiceDetectionImpl(); |
| 27 | 28 |
| 28 int ProcessCaptureAudio(AudioBuffer* audio); | 29 int ProcessCaptureAudio(AudioBuffer* audio); |
| 29 | 30 |
| 30 // VoiceDetection implementation. | 31 // VoiceDetection implementation. |
| 31 bool is_enabled() const override; | 32 bool is_enabled() const override; |
| 32 | 33 |
| 33 // ProcessingComponent implementation. | 34 // ProcessingComponent implementation. |
| 34 int Initialize() override; | 35 int Initialize() override; |
| 35 | 36 |
| 36 private: | 37 private: |
| 37 // VoiceDetection implementation. | 38 // VoiceDetection implementation. |
| 38 int Enable(bool enable) override; | 39 int Enable(bool enable) override; |
| 39 int set_stream_has_voice(bool has_voice) override; | 40 int set_stream_has_voice(bool has_voice) override; |
| 40 bool stream_has_voice() const override; | 41 bool stream_has_voice() const override; |
| 41 int set_likelihood(Likelihood likelihood) override; | 42 int set_likelihood(Likelihood likelihood) override; |
| 42 Likelihood likelihood() const override; | 43 Likelihood likelihood() const override; |
| 43 int set_frame_size_ms(int size) override; | 44 int set_frame_size_ms(int size) override; |
| 44 int frame_size_ms() const override; | 45 int frame_size_ms() const override; |
| 45 | 46 |
| 46 // ProcessingComponent implementation. | 47 // ProcessingComponent implementation. |
| 47 void* CreateHandle() const override; | 48 void* CreateHandle() const override; |
| 48 int InitializeHandle(void* handle) const override; | 49 int InitializeHandle(void* handle) const override; |
| 49 int ConfigureHandle(void* handle) const override; | 50 int ConfigureHandle(void* handle) const override; |
| 50 void DestroyHandle(void* handle) const override; | 51 void DestroyHandle(void* handle) const override; |
| 51 int num_handles_required() const override; | 52 int num_handles_required() const override; |
| 52 int GetHandleError(void* handle) const override; | 53 int GetHandleError(void* handle) const override; |
| 53 | 54 |
| 55 // Not guarded as its public API is thread safe. |
| 54 const AudioProcessing* apm_; | 56 const AudioProcessing* apm_; |
| 55 CriticalSectionWrapper* crit_; | 57 |
| 56 bool stream_has_voice_; | 58 rtc::CriticalSection* const crit_; |
| 57 bool using_external_vad_; | 59 |
| 58 Likelihood likelihood_; | 60 bool stream_has_voice_ GUARDED_BY(crit_); |
| 59 int frame_size_ms_; | 61 bool using_external_vad_ GUARDED_BY(crit_); |
| 60 size_t frame_size_samples_; | 62 Likelihood likelihood_ GUARDED_BY(crit_); |
| 63 int frame_size_ms_ GUARDED_BY(crit_); |
| 64 size_t frame_size_samples_ GUARDED_BY(crit_); |
| 61 }; | 65 }; |
| 62 } // namespace webrtc | 66 } // namespace webrtc |
| 63 | 67 |
| 64 #endif // WEBRTC_MODULES_AUDIO_PROCESSING_VOICE_DETECTION_IMPL_H_ | 68 #endif // WEBRTC_MODULES_AUDIO_PROCESSING_VOICE_DETECTION_IMPL_H_ |
| OLD | NEW |