Allow separate dump sets for the data dumper in APM
This CL allows separate dump sets to be used when dumping internal
APM data using audioproc_f, opening up for reducing the amount of
data to be dumped.
Bug: webrtc:5298
Change-Id: I8286933ceed10db074f2064414cc08e8b12653fa
Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/196089
Commit-Queue: Per Åhgren <peah@webrtc.org>
Reviewed-by: Gustaf Ullberg <gustaf@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#33047}
diff --git a/modules/audio_processing/logging/apm_data_dumper.h b/modules/audio_processing/logging/apm_data_dumper.h
index 1824fdd..6d32b32 100644
--- a/modules/audio_processing/logging/apm_data_dumper.h
+++ b/modules/audio_processing/logging/apm_data_dumper.h
@@ -21,6 +21,7 @@
#include <unordered_map>
#endif
+#include "absl/types/optional.h"
#include "api/array_view.h"
#if WEBRTC_APM_DEBUG_DUMP == 1
#include "common_audio/wav_file.h"
@@ -64,6 +65,18 @@
#endif
}
+ // Default dump set.
+ static constexpr size_t kDefaultDumpSet = 0;
+
+ // Specifies what dump set to use. All dump commands with a different dump set
+ // than the one specified will be discarded. If not specificed, all dump sets
+ // will be used.
+ static void SetDumpSetToUse(int dump_set_to_use) {
+#if WEBRTC_APM_DEBUG_DUMP == 1
+ dump_set_to_use_ = dump_set_to_use;
+#endif
+ }
+
// Set an optional output directory.
static void SetOutputDirectory(const std::string& output_dir) {
#if WEBRTC_APM_DEBUG_DUMP == 1
@@ -82,8 +95,11 @@
// Methods for performing dumping of data of various types into
// various formats.
- void DumpRaw(const char* name, double v) {
+ void DumpRaw(const char* name, double v, int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
FILE* file = GetRawFile(name);
fwrite(&v, sizeof(v), 1, file);
@@ -91,8 +107,14 @@
#endif
}
- void DumpRaw(const char* name, size_t v_length, const double* v) {
+ void DumpRaw(const char* name,
+ size_t v_length,
+ const double* v,
+ int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
FILE* file = GetRawFile(name);
fwrite(v, sizeof(v[0]), v_length, file);
@@ -100,16 +122,24 @@
#endif
}
- void DumpRaw(const char* name, rtc::ArrayView<const double> v) {
+ void DumpRaw(const char* name,
+ rtc::ArrayView<const double> v,
+ int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
DumpRaw(name, v.size(), v.data());
}
#endif
}
- void DumpRaw(const char* name, float v) {
+ void DumpRaw(const char* name, float v, int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
FILE* file = GetRawFile(name);
fwrite(&v, sizeof(v), 1, file);
@@ -117,8 +147,14 @@
#endif
}
- void DumpRaw(const char* name, size_t v_length, const float* v) {
+ void DumpRaw(const char* name,
+ size_t v_length,
+ const float* v,
+ int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
FILE* file = GetRawFile(name);
fwrite(v, sizeof(v[0]), v_length, file);
@@ -126,24 +162,38 @@
#endif
}
- void DumpRaw(const char* name, rtc::ArrayView<const float> v) {
+ void DumpRaw(const char* name,
+ rtc::ArrayView<const float> v,
+ int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
DumpRaw(name, v.size(), v.data());
}
#endif
}
- void DumpRaw(const char* name, bool v) {
+ void DumpRaw(const char* name, bool v, int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
DumpRaw(name, static_cast<int16_t>(v));
}
#endif
}
- void DumpRaw(const char* name, size_t v_length, const bool* v) {
+ void DumpRaw(const char* name,
+ size_t v_length,
+ const bool* v,
+ int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
FILE* file = GetRawFile(name);
for (size_t k = 0; k < v_length; ++k) {
@@ -154,16 +204,24 @@
#endif
}
- void DumpRaw(const char* name, rtc::ArrayView<const bool> v) {
+ void DumpRaw(const char* name,
+ rtc::ArrayView<const bool> v,
+ int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
DumpRaw(name, v.size(), v.data());
}
#endif
}
- void DumpRaw(const char* name, int16_t v) {
+ void DumpRaw(const char* name, int16_t v, int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
FILE* file = GetRawFile(name);
fwrite(&v, sizeof(v), 1, file);
@@ -171,8 +229,14 @@
#endif
}
- void DumpRaw(const char* name, size_t v_length, const int16_t* v) {
+ void DumpRaw(const char* name,
+ size_t v_length,
+ const int16_t* v,
+ int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
FILE* file = GetRawFile(name);
fwrite(v, sizeof(v[0]), v_length, file);
@@ -180,16 +244,24 @@
#endif
}
- void DumpRaw(const char* name, rtc::ArrayView<const int16_t> v) {
+ void DumpRaw(const char* name,
+ rtc::ArrayView<const int16_t> v,
+ int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
DumpRaw(name, v.size(), v.data());
}
#endif
}
- void DumpRaw(const char* name, int32_t v) {
+ void DumpRaw(const char* name, int32_t v, int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
FILE* file = GetRawFile(name);
fwrite(&v, sizeof(v), 1, file);
@@ -197,8 +269,14 @@
#endif
}
- void DumpRaw(const char* name, size_t v_length, const int32_t* v) {
+ void DumpRaw(const char* name,
+ size_t v_length,
+ const int32_t* v,
+ int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
FILE* file = GetRawFile(name);
fwrite(v, sizeof(v[0]), v_length, file);
@@ -206,8 +284,11 @@
#endif
}
- void DumpRaw(const char* name, size_t v) {
+ void DumpRaw(const char* name, size_t v, int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
FILE* file = GetRawFile(name);
fwrite(&v, sizeof(v), 1, file);
@@ -215,8 +296,14 @@
#endif
}
- void DumpRaw(const char* name, size_t v_length, const size_t* v) {
+ void DumpRaw(const char* name,
+ size_t v_length,
+ const size_t* v,
+ int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
FILE* file = GetRawFile(name);
fwrite(v, sizeof(v[0]), v_length, file);
@@ -224,16 +311,26 @@
#endif
}
- void DumpRaw(const char* name, rtc::ArrayView<const int32_t> v) {
+ void DumpRaw(const char* name,
+ rtc::ArrayView<const int32_t> v,
+ int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
DumpRaw(name, v.size(), v.data());
}
#endif
}
- void DumpRaw(const char* name, rtc::ArrayView<const size_t> v) {
+ void DumpRaw(const char* name,
+ rtc::ArrayView<const size_t> v,
+ int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
DumpRaw(name, v.size(), v.data());
#endif
}
@@ -242,8 +339,12 @@
size_t v_length,
const float* v,
int sample_rate_hz,
- int num_channels) {
+ int num_channels,
+ int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
WavWriter* file = GetWavFile(name, sample_rate_hz, num_channels,
WavFile::SampleFormat::kFloat);
@@ -255,8 +356,12 @@
void DumpWav(const char* name,
rtc::ArrayView<const float> v,
int sample_rate_hz,
- int num_channels) {
+ int num_channels,
+ int dump_set = kDefaultDumpSet) {
#if WEBRTC_APM_DEBUG_DUMP == 1
+ if (dump_set_to_use_ && *dump_set_to_use_ != dump_set)
+ return;
+
if (recording_activated_) {
DumpWav(name, v.size(), v.data(), sample_rate_hz, num_channels);
}
@@ -266,6 +371,7 @@
private:
#if WEBRTC_APM_DEBUG_DUMP == 1
static bool recording_activated_;
+ static absl::optional<int> dump_set_to_use_;
static constexpr size_t kOutputDirMaxLength = 1024;
static char output_dir_[kOutputDirMaxLength];
const int instance_index_;