Index: webrtc/modules/audio_coding/main/acm2/acm_dump.cc |
diff --git a/webrtc/modules/audio_coding/main/acm2/acm_dump.cc b/webrtc/modules/audio_coding/main/acm2/acm_dump.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..9c624d97b67eb728c6f54729ee8a37e6359f2631 |
--- /dev/null |
+++ b/webrtc/modules/audio_coding/main/acm2/acm_dump.cc |
@@ -0,0 +1,240 @@ |
+/* |
+ * Copyright (c) 2015 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. |
+ */ |
+ |
+#include "webrtc/modules/audio_coding/main/acm2/acm_dump.h" |
+ |
+#include <deque> |
+ |
+#include "webrtc/base/checks.h" |
+#include "webrtc/base/thread_annotations.h" |
+#include "webrtc/system_wrappers/interface/clock.h" |
+#include "webrtc/system_wrappers/interface/critical_section_wrapper.h" |
+#include "webrtc/system_wrappers/interface/file_wrapper.h" |
+ |
+#ifdef RTC_AUDIOCODING_DEBUG_DUMP |
+// Files generated at build-time by the protobuf compiler. |
+#ifdef WEBRTC_ANDROID_PLATFORM_BUILD |
+#include "external/webrtc/webrtc/modules/audio_coding/dump.pb.h" |
+#else |
+#include "webrtc/audio_coding/dump.pb.h" |
+#endif |
+#endif |
+ |
+namespace webrtc { |
+ |
+// Noop implementation if flag is not set |
+#ifndef RTC_AUDIOCODING_DEBUG_DUMP |
+class AcmDumpImpl final : public AcmDump { |
+ public: |
+ void StartLogging(const std::string& file_name, int duration_ms) override{}; |
+ void LogRtpPacket(bool incoming, |
+ const uint8_t* packet, |
+ size_t length) override{}; |
+ void LogDebugEvent(DebugEvent event_type, |
+ const std::string& event_message) override{}; |
+ void LogDebugEvent(DebugEvent event_type) override{}; |
+}; |
+#else |
+ |
+class AcmDumpImpl final : public AcmDump { |
+ public: |
+ AcmDumpImpl(); |
+ |
+ void StartLogging(const std::string& file_name, int duration_ms) override; |
+ void LogRtpPacket(bool incoming, |
+ const uint8_t* packet, |
+ size_t length) override; |
+ void LogDebugEvent(DebugEvent event_type, |
+ const std::string& event_message) override; |
+ void LogDebugEvent(DebugEvent event_type) override; |
+ |
+ private: |
+ // This function is identical to LogDebugEvent, but requires holding the lock. |
+ void LogDebugEventLocked(DebugEvent event_type, |
+ const std::string& event_message) |
+ EXCLUSIVE_LOCKS_REQUIRED(crit_); |
+ // Stops logging and clears the stored data and buffers. |
+ void Clear() EXCLUSIVE_LOCKS_REQUIRED(crit_); |
+ // Adds a new event to the logfile if logging is active, or adds it to the |
+ // list of recent log events otherwise. |
+ void HandleEvent(ACMDumpEvent* event) EXCLUSIVE_LOCKS_REQUIRED(crit_); |
+ // Writes the event to the file. Note that this will destroy the state of the |
+ // input argument. |
+ void StoreToFile(ACMDumpEvent* event) EXCLUSIVE_LOCKS_REQUIRED(crit_); |
+ // Adds the event to the list of recent events, and removes any events that |
+ // are too old and no longer fall in the time window. |
+ void AddRecentEvent(const ACMDumpEvent& event) |
+ EXCLUSIVE_LOCKS_REQUIRED(crit_); |
+ |
+ // Amount of time in microseconds to record log events, before starting the |
+ // actual log. |
+ const int recent_log_duration_us = 10000000; |
+ |
+ rtc::scoped_ptr<webrtc::CriticalSectionWrapper> crit_; |
+ rtc::scoped_ptr<webrtc::FileWrapper> file_ GUARDED_BY(crit_); |
+ rtc::scoped_ptr<ACMDumpEventStream> stream_ GUARDED_BY(crit_); |
+ std::deque<ACMDumpEvent> recent_log_events_ GUARDED_BY(crit_); |
+ bool currently_logging_ GUARDED_BY(crit_); |
+ int64_t start_time_us_ GUARDED_BY(crit_); |
+ int64_t duration_us_ GUARDED_BY(crit_); |
+ const webrtc::Clock* const clock_; |
+}; |
+ |
+namespace { |
+ |
+// Convert from AcmDump's debug event enum (runtime format) to the corresponding |
+// protobuf enum (serialized format). |
+ACMDumpDebugEvent_EventType convertDebugEvent(AcmDump::DebugEvent event_type) { |
+ switch (event_type) { |
+ case AcmDump::DebugEvent::kLogStart: |
+ return ACMDumpDebugEvent::LOG_START; |
+ case AcmDump::DebugEvent::kLogEnd: |
+ return ACMDumpDebugEvent::LOG_END; |
+ case AcmDump::DebugEvent::kAudioPlayout: |
+ return ACMDumpDebugEvent::AUDIO_PLAYOUT; |
+ } |
+ return ACMDumpDebugEvent::UNKNOWN_EVENT; |
+} |
+ |
+} // Anonymous namespace. |
+ |
+// AcmDumpImpl member functions. |
+AcmDumpImpl::AcmDumpImpl() |
+ : crit_(webrtc::CriticalSectionWrapper::CreateCriticalSection()), |
+ file_(webrtc::FileWrapper::Create()), |
+ stream_(new webrtc::ACMDumpEventStream()), |
+ currently_logging_(false), |
+ start_time_us_(0), |
+ duration_us_(0), |
+ clock_(webrtc::Clock::GetRealTimeClock()) { |
+} |
+ |
+void AcmDumpImpl::StartLogging(const std::string& file_name, int duration_ms) { |
+ CriticalSectionScoped lock(crit_.get()); |
+ Clear(); |
+ if (file_->OpenFile(file_name.c_str(), false) != 0) { |
+ return; |
+ } |
+ // Add LOG_START event to the recent event list. This call will also remove |
+ // any events that are too old from the recent event list. |
+ LogDebugEventLocked(DebugEvent::kLogStart, ""); |
+ currently_logging_ = true; |
+ start_time_us_ = clock_->TimeInMicroseconds(); |
+ duration_us_ = static_cast<int64_t>(duration_ms) * 1000; |
+ // Write all the recent events to the log file. |
+ for (auto&& event : recent_log_events_) { |
+ StoreToFile(&event); |
+ } |
+ recent_log_events_.clear(); |
+} |
+ |
+void AcmDumpImpl::LogRtpPacket(bool incoming, |
+ const uint8_t* packet, |
+ size_t length) { |
+ CriticalSectionScoped lock(crit_.get()); |
+ ACMDumpEvent rtp_event; |
+ const int64_t timestamp = clock_->TimeInMicroseconds(); |
+ rtp_event.set_timestamp_us(timestamp); |
+ rtp_event.set_type(webrtc::ACMDumpEvent::RTP_EVENT); |
+ rtp_event.mutable_packet()->set_direction( |
+ incoming ? ACMDumpRTPPacket::INCOMING : ACMDumpRTPPacket::OUTGOING); |
+ rtp_event.mutable_packet()->set_rtp_data(packet, length); |
+ HandleEvent(&rtp_event); |
+} |
+ |
+void AcmDumpImpl::LogDebugEvent(DebugEvent event_type, |
+ const std::string& event_message) { |
+ CriticalSectionScoped lock(crit_.get()); |
+ LogDebugEventLocked(event_type, event_message); |
+} |
+ |
+void AcmDumpImpl::LogDebugEvent(DebugEvent event_type) { |
+ CriticalSectionScoped lock(crit_.get()); |
+ LogDebugEventLocked(event_type, ""); |
+} |
+ |
+void AcmDumpImpl::LogDebugEventLocked(DebugEvent event_type, |
+ const std::string& event_message) { |
+ ACMDumpEvent event; |
+ int64_t timestamp = clock_->TimeInMicroseconds(); |
+ event.set_timestamp_us(timestamp); |
+ event.set_type(webrtc::ACMDumpEvent::DEBUG_EVENT); |
+ auto debug_event = event.mutable_debug_event(); |
+ debug_event->set_type(convertDebugEvent(event_type)); |
+ debug_event->set_message(event_message); |
+ HandleEvent(&event); |
+} |
+ |
+void AcmDumpImpl::Clear() { |
+ if (file_->Open()) { |
+ file_->CloseFile(); |
+ } |
+ currently_logging_ = false; |
+ stream_->Clear(); |
+} |
+ |
+void AcmDumpImpl::HandleEvent(ACMDumpEvent* event) { |
+ if (currently_logging_) { |
+ if (clock_->TimeInMicroseconds() < start_time_us_ + duration_us_) { |
+ StoreToFile(event); |
+ } else { |
+ LogDebugEventLocked(DebugEvent::kLogEnd, ""); |
+ Clear(); |
+ AddRecentEvent(*event); |
+ } |
+ } else { |
+ AddRecentEvent(*event); |
+ } |
+} |
+ |
+void AcmDumpImpl::StoreToFile(ACMDumpEvent* event) { |
+ // Reuse the same object at every log event. |
+ if (stream_->stream_size() < 1) { |
+ stream_->add_stream(); |
+ } |
+ DCHECK_EQ(stream_->stream_size(), 1); |
+ stream_->mutable_stream(0)->Swap(event); |
+ |
+ std::string dump_buffer; |
+ stream_->SerializeToString(&dump_buffer); |
+ file_->Write(dump_buffer.data(), dump_buffer.size()); |
+} |
+ |
+void AcmDumpImpl::AddRecentEvent(const ACMDumpEvent& event) { |
+ recent_log_events_.push_back(event); |
+ while (recent_log_events_.front().timestamp_us() < |
+ event.timestamp_us() - recent_log_duration_us) { |
+ recent_log_events_.pop_front(); |
+ } |
+} |
+ |
+bool AcmDump::ParseAcmDump(const std::string& file_name, |
+ ACMDumpEventStream* result) { |
+ char tmp_buffer[1024]; |
+ int bytes_read = 0; |
+ rtc::scoped_ptr<FileWrapper> dump_file(FileWrapper::Create()); |
+ if (dump_file->OpenFile(file_name.c_str(), true) != 0) { |
+ return false; |
+ } |
+ std::string dump_buffer; |
+ while ((bytes_read = dump_file->Read(tmp_buffer, sizeof(tmp_buffer))) > 0) { |
+ dump_buffer.append(tmp_buffer, bytes_read); |
+ } |
+ dump_file->CloseFile(); |
+ return result->ParseFromString(dump_buffer); |
+} |
+ |
+#endif // RTC_AUDIOCODING_DEBUG_DUMP |
+ |
+// AcmDump member functions. |
+rtc::scoped_ptr<AcmDump> AcmDump::Create() { |
+ return rtc::scoped_ptr<AcmDump>(new AcmDumpImpl()); |
+} |
+} // namespace webrtc |