src/third_party/webrtc/common_audio/resampler/push_resampler.cc

   1 /*
   2  *  Copyright (c) 2013 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/common_audio/resampler/include/push_resampler.h"
  12
  13 #include <string.h>
  14
  15 #include "webrtc/common_audio/include/audio_util.h"
  16 #include "webrtc/common_audio/resampler/include/resampler.h"
  17 #include "webrtc/common_audio/resampler/push_sinc_resampler.h"
  18
  19 namespace webrtc {
  20
  21 template <typename T>
  22 PushResampler<T>::PushResampler()
  23     : src_sample_rate_hz_(0),
  24       dst_sample_rate_hz_(0),
  25       num_channels_(0) {
  26 }
  27
  28 template <typename T>
  29 PushResampler<T>::~PushResampler() {
  30 }
  31
  32 template <typename T>
  33 int PushResampler<T>::InitializeIfNeeded(int src_sample_rate_hz,
  34                                          int dst_sample_rate_hz,
  35                                          int num_channels) {
  36   if (src_sample_rate_hz == src_sample_rate_hz_ &&
  37       dst_sample_rate_hz == dst_sample_rate_hz_ &&
  38       num_channels == num_channels_)
  39     // No-op if settings haven't changed.
  40     return 0;
  41
  42   if (src_sample_rate_hz <= 0 || dst_sample_rate_hz <= 0 ||
  43       num_channels <= 0 || num_channels > 2)
  44     return -1;
  45
  46   src_sample_rate_hz_ = src_sample_rate_hz;
  47   dst_sample_rate_hz_ = dst_sample_rate_hz;
  48   num_channels_ = num_channels;
  49
  50   const int src_size_10ms_mono = src_sample_rate_hz / 100;
  51   const int dst_size_10ms_mono = dst_sample_rate_hz / 100;
  52   sinc_resampler_.reset(new PushSincResampler(src_size_10ms_mono,
  53                                               dst_size_10ms_mono));
  54   if (num_channels_ == 2) {
  55     src_left_.reset(new T[src_size_10ms_mono]);
  56     src_right_.reset(new T[src_size_10ms_mono]);
  57     dst_left_.reset(new T[dst_size_10ms_mono]);
  58     dst_right_.reset(new T[dst_size_10ms_mono]);
  59     sinc_resampler_right_.reset(new PushSincResampler(src_size_10ms_mono,
  60                                                       dst_size_10ms_mono));
  61   }
  62
  63   return 0;
  64 }
  65
  66 template <typename T>
  67 int PushResampler<T>::Resample(const T* src, int src_length, T* dst,
  68                                int dst_capacity) {
  69   const int src_size_10ms = src_sample_rate_hz_ * num_channels_ / 100;
  70   const int dst_size_10ms = dst_sample_rate_hz_ * num_channels_ / 100;
  71   if (src_length != src_size_10ms || dst_capacity < dst_size_10ms)
  72     return -1;
  73
  74   if (src_sample_rate_hz_ == dst_sample_rate_hz_) {
  75     // The old resampler provides this memcpy facility in the case of matching
  76     // sample rates, so reproduce it here for the sinc resampler.
  77     memcpy(dst, src, src_length * sizeof(T));
  78     return src_length;
  79   }
  80   if (num_channels_ == 2) {
  81     const int src_length_mono = src_length / num_channels_;
  82     const int dst_capacity_mono = dst_capacity / num_channels_;
  83     T* deinterleaved[] = {src_left_.get(), src_right_.get()};
  84     Deinterleave(src, src_length_mono, num_channels_, deinterleaved);
  85
  86     int dst_length_mono =
  87         sinc_resampler_->Resample(src_left_.get(), src_length_mono,
  88                                   dst_left_.get(), dst_capacity_mono);
  89     sinc_resampler_right_->Resample(src_right_.get(), src_length_mono,
  90                                     dst_right_.get(), dst_capacity_mono);
  91
  92     deinterleaved[0] = dst_left_.get();
  93     deinterleaved[1] = dst_right_.get();
  94     Interleave(deinterleaved, dst_length_mono, num_channels_, dst);
  95     return dst_length_mono * num_channels_;
  96   } else {
  97     return sinc_resampler_->Resample(src, src_length, dst, dst_capacity);
  98   }
  99 }
 100
 101 // Explictly generate required instantiations.
 102 template class PushResampler<int16_t>;
 103 template class PushResampler<float>;
 104
 105 }  // namespace webrtc