| 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_
|
|
|