1868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)// Copyright 2013 The Chromium Authors. All rights reserved.
25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be
35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file.
45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
57dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "content/renderer/media/audio_decoder.h"
65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <vector>
87dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/basictypes.h"
107d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#include "base/strings/string_util.h"
11eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch#include "base/time/time.h"
125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "media/base/audio_bus.h"
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "media/base/limits.h"
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "media/filters/audio_file_reader.h"
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "media/filters/in_memory_url_protocol.h"
16868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "third_party/WebKit/public/platform/WebAudioBus.h"
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using media::AudioBus;
195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using media::AudioFileReader;
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using media::InMemoryUrlProtocol;
215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using std::vector;
22f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)using blink::WebAudioBus;
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
247dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochnamespace content {
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Decode in-memory audio file data.
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool DecodeAudioFileData(
28f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)    blink::WebAudioBus* destination_bus,
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const char* data, size_t data_size, double sample_rate) {
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(destination_bus);
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!destination_bus)
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Uses the FFmpeg library for audio file reading.
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  InMemoryUrlProtocol url_protocol(reinterpret_cast<const uint8*>(data),
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                   data_size, false);
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  AudioFileReader reader(&url_protocol);
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!reader.Open())
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  size_t number_of_channels = reader.channels();
435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  double file_sample_rate = reader.sample_rate();
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  size_t number_of_frames = static_cast<size_t>(reader.number_of_frames());
455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Apply sanity checks to make sure crazy values aren't coming out of
475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // FFmpeg.
485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!number_of_channels ||
495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      number_of_channels > static_cast<size_t>(media::limits::kMaxChannels) ||
505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      file_sample_rate < media::limits::kMinSampleRate ||
515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      file_sample_rate > media::limits::kMaxSampleRate)
525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Allocate and configure the output audio channel data.
555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  destination_bus->initialize(number_of_channels,
565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                              number_of_frames,
575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                              file_sample_rate);
585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Wrap the channel pointers which will receive the decoded PCM audio.
605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  vector<float*> audio_data;
615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  audio_data.reserve(number_of_channels);
625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (size_t i = 0; i < number_of_channels; ++i) {
635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    audio_data.push_back(destination_bus->channelData(i));
645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  scoped_ptr<AudioBus> audio_bus = AudioBus::WrapVector(
675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      number_of_frames, audio_data);
685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Decode the audio file data.
705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // TODO(crogers): If our estimate was low, then we still may fail to read
715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // all of the data from the file.
725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  size_t actual_frames = reader.Read(audio_bus.get());
735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Adjust WebKit's bus to account for the actual file length
755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // and valid data read.
765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (actual_frames != number_of_frames) {
775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DCHECK_LE(actual_frames, number_of_frames);
785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    destination_bus->resizeSmaller(actual_frames);
795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  double duration = actual_frames / file_sample_rate;
825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DVLOG(1) << "Decoded file data -"
845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)           << " data: " << data
855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)           << " data size: " << data_size
865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)           << " duration: " << duration
875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)           << " number of frames: " << actual_frames
885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)           << " sample rate: " << file_sample_rate
895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)           << " number of channels: " << number_of_channels;
905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return actual_frames > 0;
925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
947dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch}  // namespace content
95