blob: d11f10b23ac64edf93b5626cb4152b6189e5c394 [file] [log] [blame]
/*
* Copyright (c) 2017 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 "modules/audio_processing/aec_dump/write_to_file_task.h"
#include "rtc_base/protobuf_utils.h"
namespace webrtc {
WriteToFileTask::WriteToFileTask(webrtc::FileWrapper* debug_file,
int64_t* num_bytes_left_for_log)
: debug_file_(debug_file),
num_bytes_left_for_log_(num_bytes_left_for_log) {}
WriteToFileTask::~WriteToFileTask() = default;
audioproc::Event* WriteToFileTask::GetEvent() {
return &event_;
}
bool WriteToFileTask::IsRoomForNextEvent(size_t event_byte_size) const {
int64_t next_message_size = event_byte_size + sizeof(int32_t);
return (*num_bytes_left_for_log_ < 0) ||
(*num_bytes_left_for_log_ >= next_message_size);
}
void WriteToFileTask::UpdateBytesLeft(size_t event_byte_size) {
RTC_DCHECK(IsRoomForNextEvent(event_byte_size));
if (*num_bytes_left_for_log_ >= 0) {
*num_bytes_left_for_log_ -= (sizeof(int32_t) + event_byte_size);
}
}
bool WriteToFileTask::Run() {
ProtoString event_string;
event_.SerializeToString(&event_string);
const size_t event_byte_size = event_.ByteSizeLong();
if (!IsRoomForNextEvent(event_byte_size)) {
// Ensure that no further events are written, even if they're smaller than
// the current event.
*num_bytes_left_for_log_ = 0;
return true;
}
UpdateBytesLeft(event_byte_size);
// Write message preceded by its size.
if (!debug_file_->Write(&event_byte_size, sizeof(int32_t))) {
RTC_NOTREACHED();
}
if (!debug_file_->Write(event_string.data(), event_string.length())) {
RTC_NOTREACHED();
}
return true; // Delete task from queue at once.
}
} // namespace webrtc