Upgrade AudioBus to support wrapping, interleaving.

media/base/audio_bus.h

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef MEDIA_BASE_AUDIO_BUS_H_
 #define MEDIA_BASE_AUDIO_BUS_H_
 
 #include <vector>
 
 #include "base/memory/aligned_memory.h"
 #include "base/memory/scoped_ptr.h"
 #include "media/base/media_export.h"
 
 namespace media {
 class AudioParameters;
 
 // Scoped container for "busing" audio channel data around.  Each channel is
-// stored in planar format and guaranteed to have a 16-byte alignment.  AudioBus
-// objects can be created normally or via wrapping.  Normally, AudioBus will
-// dice up a contiguous memory block for channel data.  When wrapped, AudioBus
-// instead routes requests for channel data to the wrapped object.
+// stored in planar format and guaranteed to be aligned by kChannelAlignment.
+// AudioBus objects can be created normally or via wrapping.  Normally, AudioBus
+// will dice up a contiguous memory block for channel data.  When wrapped,
+// AudioBus instead routes requests for channel data to the wrapped object.
 class MEDIA_EXPORT AudioBus {
  public:
+  // Guaranteed alignment of each channel's data; use 16-byte alignment for easy
+  // SSE optimizations.
+  enum { kChannelAlignment = 16 };
+
   // Creates a new AudioBus and allocates |channels| of length |frames|.  Uses
-  // channels() and frames_per_buffer() if given an AudioParameters object.
+  // channels() and frames_per_buffer() from AudioParameters if given.
   static scoped_ptr<AudioBus> Create(int channels, int frames);
   static scoped_ptr<AudioBus> Create(const AudioParameters& params);
 
   // Creates a new AudioBus from an existing channel vector.  Does not transfer
   // ownership of |channel_data| to AudioBus; i.e., |channel_data| must outlive
-  // the returned AudioBus.  Each channel pointer must be 16-byte aligned.
+  // the returned AudioBus.  Each channel must be aligned by kChannelAlignment.
   static scoped_ptr<AudioBus> WrapVector(
       int frames, const std::vector<float*>& channel_data);
 
-  // Returns a raw pointer to internal channel data.  Useful for copying state
-  // between two AudioBus objects created with the same parameters.  data_size()
-  // is in bytes.  Can not be used with an AudioBus constructed via wrapping.
-  void* data();
-  int data_size() const;
+  // Creates a new AudioBus by wrapping an existing block of memory.  Block must
+  // be at least CalculateMemorySize() bytes in size.  |data| must outlive the
+  // returned AudioBus.  |data| must be aligned by kChannelAlignment.
+  static scoped_ptr<AudioBus> WrapMemory(const AudioParameters& params,
+                                         void* data);
+  // Returns the required memory size to use the WrapMemory() method.
+  static int CalculateMemorySize(const AudioParameters& params);
+
+  // Helper methods for converting an AudioBus from and to interleaved integer
+  // data.  Expects interleaving to be [ch0, ch1, ..., chN, ch0, ch1, ...] with
+  // |bytes_per_sample| per value.  Values are scaled and bias corrected during
+  // conversion.  ToInterleaved() will also clip values to format range.
+  // Handles uint8, int16, and int32 currently.
+  void FromInterleaved(const void* source, int frames, int bytes_per_sample);
+  void ToInterleaved(int frames, int bytes_per_sample, void* dest) const;
+
+  // Helper method for copying channel data from one AudioBus to another.  Both
+  // AudioBus object must have the same frames() and channels().
+  void CopyTo(AudioBus* dest) const;
 
   // Returns a raw pointer to the requested channel.  Pointer is guaranteed to
   // have a 16-byte alignment.
   float* channel(int channel) { return channel_data_[channel]; }
   const float* channel(int channel) const { return channel_data_[channel]; }
 
   int channels() const { return channel_data_.size(); }
   int frames() const { return frames_; }
 
   // Helper method for zeroing out all channels of audio data.
   void Zero();
   void ZeroFrames(int frames);
 
  private:
   friend class scoped_ptr<AudioBus>;
   ~AudioBus();
 
   AudioBus(int channels, int frames);
+  AudioBus(int channels, int frames, float* data);
   AudioBus(int frames, const std::vector<float*>& channel_data);
 
+  // Helper method for building |channel_data_| from a block of memory.  |data|
+  // must be at least BlockSize() bytes in size.
+  void BuildChannelData(int channels, int aligned_frame, float* data);
+
   // Contiguous block of channel memory.
   scoped_ptr_malloc<float, base::ScopedPtrAlignedFree> data_;
-  int data_size_;
 
   std::vector<float*> channel_data_;
   int frames_;
 
   DISALLOW_COPY_AND_ASSIGN(AudioBus);
 };
 
 }  // namespace media
 
 #endif  // MEDIA_BASE_AUDIO_BUS_H_