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..18f9e5e18153282110c046a5cbf449a876107144 |
--- /dev/null |
+++ b/webrtc/modules/audio_processing/logging/apm_data_dumper.h |
@@ -0,0 +1,103 @@ |
+/* |
+ * 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 <memory> |
+#include <string> |
+#include <unordered_map> |
+ |
+#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 { |
+ |
+#if WEBRTC_AEC_DEBUG_DUMP == 1 |
+// Functor used to use as a custom deleter in the map of file pointers to raw |
+// files. |
+struct RawFileCloseFunctor { |
+ void operator()(FILE* f) const { fclose(f); } |
+}; |
+#endif |
+ |
+// 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 |
+ |
+ // Reinitializes 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 char* 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 char* name, rtc::ArrayView<const float> v) { |
+#if WEBRTC_AEC_DEBUG_DUMP == 1 |
+ DumpRaw(name, v.size(), v.data()); |
+#endif |
+ } |
+ |
+ void DumpWav(const char* 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 |
+ } |
+ |
+ private: |
+#if WEBRTC_AEC_DEBUG_DUMP == 1 |
+ const int instance_index_; |
+ int recording_set_index_ = 0; |
+ std::unordered_map<std::string, std::unique_ptr<FILE, RawFileCloseFunctor>> |
+ raw_files_; |
+ std::unordered_map<std::string, std::unique_ptr<WavWriter>> wav_files_; |
+ |
+ FILE* GetRawFile(const char* name); |
+ WavWriter* GetWavFile(const char* 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_ |