blob: 41521df73f50e0756589c42866ef17a3276ddcd7 [file] [log] [blame]
# Copyright (c) 2018 The WebRTC project authors. All Rights Reserved.
#
# Use of this source code is governed by a BSD-style license
# that can be found in the LICENSE file in the root of the source
# tree. An additional intellectual property rights grant can be found
# in the file PATENTS. All contributing project authors may
# be found in the AUTHORS file in the root of the source tree.
import("../../../../webrtc.gni")
group("rnn_vad") {
deps = [
":lib",
]
}
rtc_source_set("lib") {
sources = [
"common.h",
"features_extraction.cc",
"features_extraction.h",
"fft_util.cc",
"fft_util.h",
"lp_residual.cc",
"lp_residual.h",
"pitch_info.h",
"pitch_search.cc",
"pitch_search.h",
"pitch_search_internal.cc",
"pitch_search_internal.h",
"ring_buffer.h",
"rnn.cc",
"rnn.h",
"sequence_buffer.h",
"spectral_features.cc",
"spectral_features.h",
"spectral_features_internal.cc",
"spectral_features_internal.h",
"symmetric_matrix_buffer.h",
]
deps = [
"..:biquad_filter",
"../../../../api:array_view",
"../../../../common_audio/",
"../../../../rtc_base:checks",
"../../../../rtc_base:rtc_base_approved",
"//third_party/rnnoise:kiss_fft",
"//third_party/rnnoise:rnn_vad",
]
}
if (rtc_include_tests) {
rtc_source_set("lib_test") {
testonly = true
sources = [
"test_utils.cc",
"test_utils.h",
]
deps = [
"../../../../api:array_view",
"../../../../rtc_base:checks",
"../../../../rtc_base:ptr_util",
"../../../../test:fileutils",
"../../../../test:test_support",
"//third_party/abseil-cpp/absl/memory",
]
}
unittest_resources = [
"../../../../resources/audio_processing/agc2/rnn_vad/band_energies.dat",
"../../../../resources/audio_processing/agc2/rnn_vad/fft.dat",
"../../../../resources/audio_processing/agc2/rnn_vad/pitch_buf_24k.dat",
"../../../../resources/audio_processing/agc2/rnn_vad/pitch_search_int.dat",
"../../../../resources/audio_processing/agc2/rnn_vad/pitch_lp_res.dat",
"../../../../resources/audio_processing/agc2/rnn_vad/samples.pcm",
"../../../../resources/audio_processing/agc2/rnn_vad/sil_features.dat",
"../../../../resources/audio_processing/agc2/rnn_vad/vad_prob.dat",
]
if (is_ios) {
bundle_data("unittests_bundle_data") {
testonly = true
sources = unittest_resources
outputs = [
"{{bundle_resources_dir}}/{{source_file_part}}",
]
}
}
rtc_source_set("unittests") {
testonly = true
sources = [
"features_extraction_unittest.cc",
"fft_util_unittest.cc",
"lp_residual_unittest.cc",
"pitch_search_internal_unittest.cc",
"pitch_search_unittest.cc",
"ring_buffer_unittest.cc",
"rnn_unittest.cc",
"rnn_vad_unittest.cc",
"sequence_buffer_unittest.cc",
"spectral_features_internal_unittest.cc",
"spectral_features_unittest.cc",
"symmetric_matrix_buffer_unittest.cc",
]
deps = [
":lib",
":lib_test",
"../..:audioproc_test_utils",
"../../../../api:array_view",
"../../../../common_audio/",
"../../../../rtc_base:checks",
"../../../../rtc_base:logging",
"../../../../test:test_support",
"//third_party/rnnoise:rnn_vad",
]
data = unittest_resources
if (is_ios) {
deps += [ ":unittests_bundle_data" ]
}
}
rtc_executable("rnn_vad_tool") {
testonly = true
sources = [
"rnn_vad_tool.cc",
]
deps = [
":lib",
"../../../../api:array_view",
"../../../../common_audio:common_audio",
"../../../../rtc_base:rtc_base_approved",
"../../../../test:test_support",
]
}
}