Home | History | Annotate | Download | only in source
      1 /*
      2  *  Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
      3  *
      4  *  Use of this source code is governed by a BSD-style license
      5  *  that can be found in the LICENSE file in the root of the source
      6  *  tree. An additional intellectual property rights grant can be found
      7  *  in the file PATENTS.  All contributing project authors may
      8  *  be found in the AUTHORS file in the root of the source tree.
      9  */
     10 
     11 #include "webrtc/modules/include/module_common_types.h"
     12 #include "webrtc/modules/utility/include/audio_frame_operations.h"
     13 
     14 namespace webrtc {
     15 
     16 void AudioFrameOperations::MonoToStereo(const int16_t* src_audio,
     17                                         size_t samples_per_channel,
     18                                         int16_t* dst_audio) {
     19   for (size_t i = 0; i < samples_per_channel; i++) {
     20     dst_audio[2 * i] = src_audio[i];
     21     dst_audio[2 * i + 1] = src_audio[i];
     22   }
     23 }
     24 
     25 int AudioFrameOperations::MonoToStereo(AudioFrame* frame) {
     26   if (frame->num_channels_ != 1) {
     27     return -1;
     28   }
     29   if ((frame->samples_per_channel_ * 2) >= AudioFrame::kMaxDataSizeSamples) {
     30     // Not enough memory to expand from mono to stereo.
     31     return -1;
     32   }
     33 
     34   int16_t data_copy[AudioFrame::kMaxDataSizeSamples];
     35   memcpy(data_copy, frame->data_,
     36          sizeof(int16_t) * frame->samples_per_channel_);
     37   MonoToStereo(data_copy, frame->samples_per_channel_, frame->data_);
     38   frame->num_channels_ = 2;
     39 
     40   return 0;
     41 }
     42 
     43 void AudioFrameOperations::StereoToMono(const int16_t* src_audio,
     44                                         size_t samples_per_channel,
     45                                         int16_t* dst_audio) {
     46   for (size_t i = 0; i < samples_per_channel; i++) {
     47     dst_audio[i] = (src_audio[2 * i] + src_audio[2 * i + 1]) >> 1;
     48   }
     49 }
     50 
     51 int AudioFrameOperations::StereoToMono(AudioFrame* frame) {
     52   if (frame->num_channels_ != 2) {
     53     return -1;
     54   }
     55 
     56   StereoToMono(frame->data_, frame->samples_per_channel_, frame->data_);
     57   frame->num_channels_ = 1;
     58 
     59   return 0;
     60 }
     61 
     62 void AudioFrameOperations::SwapStereoChannels(AudioFrame* frame) {
     63   if (frame->num_channels_ != 2) return;
     64 
     65   for (size_t i = 0; i < frame->samples_per_channel_ * 2; i += 2) {
     66     int16_t temp_data = frame->data_[i];
     67     frame->data_[i] = frame->data_[i + 1];
     68     frame->data_[i + 1] = temp_data;
     69   }
     70 }
     71 
     72 void AudioFrameOperations::Mute(AudioFrame& frame) {
     73   memset(frame.data_, 0, sizeof(int16_t) *
     74       frame.samples_per_channel_ * frame.num_channels_);
     75 }
     76 
     77 int AudioFrameOperations::Scale(float left, float right, AudioFrame& frame) {
     78   if (frame.num_channels_ != 2) {
     79     return -1;
     80   }
     81 
     82   for (size_t i = 0; i < frame.samples_per_channel_; i++) {
     83     frame.data_[2 * i] =
     84         static_cast<int16_t>(left * frame.data_[2 * i]);
     85     frame.data_[2 * i + 1] =
     86         static_cast<int16_t>(right * frame.data_[2 * i + 1]);
     87   }
     88   return 0;
     89 }
     90 
     91 int AudioFrameOperations::ScaleWithSat(float scale, AudioFrame& frame) {
     92   int32_t temp_data = 0;
     93 
     94   // Ensure that the output result is saturated [-32768, +32767].
     95   for (size_t i = 0; i < frame.samples_per_channel_ * frame.num_channels_;
     96        i++) {
     97     temp_data = static_cast<int32_t>(scale * frame.data_[i]);
     98     if (temp_data < -32768) {
     99       frame.data_[i] = -32768;
    100     } else if (temp_data > 32767) {
    101       frame.data_[i] = 32767;
    102     } else {
    103       frame.data_[i] = static_cast<int16_t>(temp_data);
    104     }
    105   }
    106   return 0;
    107 }
    108 
    109 }  // namespace webrtc
    110