Index: webrtc/modules/audio_processing/logging/apm_data_dumper.h |
diff --git a/webrtc/modules/audio_processing/logging/apm_data_dumper.h b/webrtc/modules/audio_processing/logging/apm_data_dumper.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..269b9b3ef360661b1ba9c169ed746d1fc97b3b91 |
--- /dev/null |
+++ b/webrtc/modules/audio_processing/logging/apm_data_dumper.h |
@@ -0,0 +1,106 @@ |
+/* |
+ * Copyright (c) 2016 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. |
+ */ |
+ |
+#ifndef WEBRTC_MODULES_AUDIO_PROCESSING_LOGGING_APM_DATA_DUMPER_H_ |
+#define WEBRTC_MODULES_AUDIO_PROCESSING_LOGGING_APM_DATA_DUMPER_H_ |
+ |
+#include <stdio.h> |
+ |
+#include <map> |
+#include <string> |
+ |
+#include "webrtc/base/array_view.h" |
+#include "webrtc/base/constructormagic.h" |
+#include "webrtc/common_audio/wav_file.h" |
+ |
+// Check to verify that the define is properly set. |
+#if !defined(WEBRTC_AEC_DEBUG_DUMP) || \ |
+ (WEBRTC_AEC_DEBUG_DUMP != 0 && WEBRTC_AEC_DEBUG_DUMP != 1) |
+#error "Set WEBRTC_AEC_DEBUG_DUMP to either 0 or 1" |
+#endif |
+ |
+namespace webrtc { |
+ |
+// Class that handles dumping of variables into files. |
+class ApmDataDumper { |
+ public: |
+// Constructor that takes an instance index that may |
+// be used to distinguish data dumped from different |
+// instances of the code. |
+#if WEBRTC_AEC_DEBUG_DUMP == 1 |
+ explicit ApmDataDumper(int instance_index) |
+ : instance_index_(instance_index) {} |
+#else |
+ explicit ApmDataDumper(int instance_index) {} |
+#endif |
+ |
+ ~ApmDataDumper(); |
+ |
+ // Reitializes the data dumping such that new versions |
+ // of all files being dumped to are created. |
+ void InitiateNewSetOfRecordings() { |
+#if WEBRTC_AEC_DEBUG_DUMP == 1 |
+ ++recording_set_index_; |
+#endif |
+ } |
+ |
+ // Methods for performing dumping of data of various types into |
+ // various formats. |
+ void DumpRaw(const std::string& name, int v_length, const float* v) { |
+#if WEBRTC_AEC_DEBUG_DUMP == 1 |
+ FILE* file = GetRawFile(name); |
+ fwrite(v, sizeof(v[0]), v_length, file); |
+#endif |
+ } |
+ |
+ void DumpRaw(const std::string& name, rtc::ArrayView<const float> v) { |
+#if WEBRTC_AEC_DEBUG_DUMP == 1 |
+ DumpRaw(name, v.size(), v.data()); |
+#endif |
+ } |
+ |
+ void DumpWav(const std::string& name, |
+ int v_length, |
+ const float* v, |
+ int sample_rate_hz, |
+ int num_channels) { |
+#if WEBRTC_AEC_DEBUG_DUMP == 1 |
+ WavWriter* file = GetWavFile(name, sample_rate_hz, num_channels); |
+ file->WriteSamples(v, v_length); |
+#endif |
+ } |
+ |
+ void DumpMonoWav(const std::string& name, |
+ rtc::ArrayView<const float> v, |
+ int sample_rate_hz, |
+ int num_channels) { |
+#if WEBRTC_AEC_DEBUG_DUMP == 1 |
+ DumpWav(name, v.size(), v.data(), sample_rate_hz, num_channels); |
+#endif |
+ } |
+ |
+ private: |
+#if WEBRTC_AEC_DEBUG_DUMP == 1 |
+ const int instance_index_; |
+ int recording_set_index_ = 0; |
+ std::map<std::string, FILE*> raw_files_; |
+ std::map<std::string, WavWriter*> wav_files_; |
+ |
+ FILE* GetRawFile(const std::string& name); |
+ WavWriter* GetWavFile(const std::string& name, |
+ int sample_rate_hz, |
+ int num_channels); |
+#endif |
+ RTC_DISALLOW_IMPLICIT_CONSTRUCTORS(ApmDataDumper); |
+}; |
+ |
+} // namespace webrtc |
+ |
+#endif // WEBRTC_MODULES_AUDIO_PROCESSING_LOGGING_APM_DATA_DUMPER_H_ |