00b8f6b364
BUG= R=andrew@webrtc.org Review URL: https://webrtc-codereview.appspot.com/36229004 Cr-Commit-Position: refs/heads/master@{#8517} git-svn-id: http://webrtc.googlecode.com/svn/trunk@8517 4adac7df-926f-26a2-2b94-8c16560cd09d
91 lines
2.9 KiB
C++
91 lines
2.9 KiB
C++
/*
|
|
* Copyright (c) 2014 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_processing/beamformer/pcm_utils.h"
|
|
|
|
#include "webrtc/base/checks.h"
|
|
#include "webrtc/common_audio/include/audio_util.h"
|
|
#include "webrtc/common_audio/channel_buffer.h"
|
|
|
|
namespace webrtc {
|
|
|
|
size_t PcmRead(FILE* file,
|
|
size_t length,
|
|
int num_channels,
|
|
int16_t* const* buffer) {
|
|
CHECK_GE(num_channels, 1);
|
|
|
|
rtc::scoped_ptr<int16_t[]> interleaved_buffer(new int16_t[length]);
|
|
size_t elements_read = fread(interleaved_buffer.get(), sizeof(int16_t),
|
|
length, file);
|
|
if (elements_read != length) {
|
|
// This is only an error if we haven't reached the end of the file.
|
|
CHECK_NE(0, feof(file));
|
|
}
|
|
|
|
Deinterleave(interleaved_buffer.get(),
|
|
static_cast<int>(elements_read) / num_channels,
|
|
num_channels,
|
|
buffer);
|
|
return elements_read;
|
|
}
|
|
|
|
size_t PcmReadToFloat(FILE* file,
|
|
size_t length,
|
|
int num_channels,
|
|
float* const* buffer) {
|
|
CHECK_GE(num_channels, 1);
|
|
|
|
int num_frames = static_cast<int>(length) / num_channels;
|
|
rtc::scoped_ptr<ChannelBuffer<int16_t> > deinterleaved_buffer(
|
|
new ChannelBuffer<int16_t>(num_frames, num_channels));
|
|
|
|
size_t elements_read =
|
|
PcmRead(file, length, num_channels, deinterleaved_buffer->channels());
|
|
|
|
for (int i = 0; i < num_channels; ++i) {
|
|
S16ToFloat(deinterleaved_buffer->channels()[i], num_frames, buffer[i]);
|
|
}
|
|
return elements_read;
|
|
}
|
|
|
|
void PcmWrite(FILE* file,
|
|
size_t length,
|
|
int num_channels,
|
|
const int16_t* const* buffer) {
|
|
CHECK_GE(num_channels, 1);
|
|
|
|
rtc::scoped_ptr<int16_t[]> interleaved_buffer(new int16_t[length]);
|
|
Interleave(buffer,
|
|
static_cast<int>(length) / num_channels,
|
|
num_channels,
|
|
interleaved_buffer.get());
|
|
CHECK_EQ(length,
|
|
fwrite(interleaved_buffer.get(), sizeof(int16_t), length, file));
|
|
}
|
|
|
|
void PcmWriteFromFloat(FILE* file,
|
|
size_t length,
|
|
int num_channels,
|
|
const float* const* buffer) {
|
|
CHECK_GE(num_channels, 1);
|
|
|
|
int num_frames = static_cast<int>(length) / num_channels;
|
|
rtc::scoped_ptr<ChannelBuffer<int16_t> > deinterleaved_buffer(
|
|
new ChannelBuffer<int16_t>(num_frames, num_channels));
|
|
|
|
for (int i = 0; i < num_channels; ++i) {
|
|
FloatToS16(buffer[i], num_frames, deinterleaved_buffer->channels()[i]);
|
|
}
|
|
PcmWrite(file, length, num_channels, deinterleaved_buffer->channels());
|
|
}
|
|
|
|
} // namespace webrtc
|