Encrypted Media: Support Audio Decrypt-Only.

media/filters/audio_decoder_selector.cc

+// Copyright (c) 2012 The Chromium Authors. All rights reserved.

[ddorwin, 2012/12/13 05:08:25]

Update CL description.

[xhwang, 2012/12/13 11:24:36]

Done.

+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "media/filters/audio_decoder_selector.h"

[ddorwin, 2012/12/13 05:08:25]

Does this really belong in filters/? I know we have a lot of non-filter stuff in
there already.

[xhwang, 2012/12/13 11:24:36]

Well, it's affliated with AudioRendererImpl, which is in filters..

+
+#include "base/bind.h"
+#include "base/callback_helpers.h"
+#include "base/logging.h"
+#include "base/message_loop_proxy.h"
+#include "media/base/audio_decoder_config.h"
+#include "media/base/bind_to_loop.h"
+#include "media/base/demuxer_stream.h"
+#include "media/base/pipeline.h"
+#include "media/filters/decrypting_audio_decoder.h"
+#include "media/filters/decrypting_demuxer_stream.h"
+
+namespace media {
+
+AudioDecoderSelector::AudioDecoderSelector(
+    const scoped_refptr<base::MessageLoopProxy>& message_loop,
+    const AudioDecoderList& clear_decoders,
+    const SetDecryptorReadyCB& set_decryptor_ready_cb)
+    : message_loop_(message_loop),
+      clear_decoders_(clear_decoders),
+      set_decryptor_ready_cb_(set_decryptor_ready_cb),
+      ALLOW_THIS_IN_INITIALIZER_LIST(weak_ptr_factory_(this)),
+      weak_this_(weak_ptr_factory_.GetWeakPtr()) {
+}
+
+AudioDecoderSelector::~AudioDecoderSelector() {}
+
+void AudioDecoderSelector::SelectAudioDecoder(
+    const scoped_refptr<DemuxerStream>& stream,
+    const StatisticsCB& statistics_cb,
+    const SelectDecoderCB& select_decoder_cb) {
+  DCHECK(message_loop_->BelongsToCurrentThread());
+  DVLOG(2) << "SelectAudioDecoder()";

[ddorwin, 2012/12/13 05:08:25]

I always log above checks, though it probably doesn't matter much since the
check prints the line number. Still nice to have a function name, though, I
guess.
Regardless, It shouldn't be between checks.

[xhwang, 2012/12/13 11:24:36]

Done.

+  DCHECK(stream);
+
+  select_decoder_cb_ = BindToCurrentLoop(select_decoder_cb);

[ddorwin, 2012/12/13 05:08:25]

Maybe explain why.

[xhwang, 2012/12/13 11:24:36]

Done.

+
+  const AudioDecoderConfig& config = stream->audio_decoder_config();
+  if (!config.IsValidConfig()) {
+    DLOG(ERROR) << "Invalid audio stream config.";
+    base::ResetAndReturn(&select_decoder_cb_).Run(NULL, NULL);
+    return;
+  }
+
+  input_stream_ = stream;
+  statistics_cb_ = statistics_cb;
+
+  if (!config.is_encrypted()) {
+    if (clear_decoders_.empty()) {
+      DLOG(ERROR) << "No audio decoder can be used to decode the input stream.";
+      base::ResetAndReturn(&select_decoder_cb_).Run(NULL, NULL);
+      return;
+    }
+
+    InitializeNextClearDecoder();
+    return;
+  }
+
+  // This could happen if Encrypted Media Extension (EME) is not enabled.
+  if (set_decryptor_ready_cb_.is_null()) {
+    base::ResetAndReturn(&select_decoder_cb_).Run(NULL, NULL);
+    return;
+  }
+
+  audio_decoder_ = new DecryptingAudioDecoder(
+      message_loop_, set_decryptor_ready_cb_);

[ddorwin, 2012/12/13 05:08:25]

better indentation. Can you fit with one above?

[xhwang, 2012/12/13 11:24:36]

Done.

+
+  audio_decoder_->Initialize(
+      input_stream_,
+      BindToCurrentLoop(base::Bind(
+          &AudioDecoderSelector::DecryptingAudioDecoderInitDone, weak_this_)),
+      statistics_cb_);
+}
+
+void AudioDecoderSelector::DecryptingAudioDecoderInitDone(
+    PipelineStatus status) {
+  DCHECK(message_loop_->BelongsToCurrentThread());
+
+  if (status == PIPELINE_OK) {
+    clear_decoders_.clear();
+    base::ResetAndReturn(&select_decoder_cb_).Run(audio_decoder_, NULL);
+    return;
+  }
+
+  audio_decoder_ = NULL;
+
+  if (clear_decoders_.empty()) {
+    DLOG(ERROR) << "No audio decoder can be used to decode the input stream.";
+    base::ResetAndReturn(&select_decoder_cb_).Run(NULL, NULL);
+    return;
+  }
+
+  decrypted_stream_ = new DecryptingDemuxerStream(
+      message_loop_, set_decryptor_ready_cb_);
+
+  decrypted_stream_->Initialize(
+      input_stream_,
+      BindToCurrentLoop(base::Bind(
+          &AudioDecoderSelector::DecryptingDemuxerStreamInitDone, weak_this_)));
+}
+
+void AudioDecoderSelector::DecryptingDemuxerStreamInitDone(
+    PipelineStatus status) {
+  DCHECK(message_loop_->BelongsToCurrentThread());
+
+  if (status != PIPELINE_OK) {
+    decrypted_stream_ = NULL;
+    base::ResetAndReturn(&select_decoder_cb_).Run(NULL, NULL);
+    return;
+  }
+
+  DCHECK(!decrypted_stream_->audio_decoder_config().is_encrypted());
+  input_stream_ = decrypted_stream_;
+  InitializeNextClearDecoder();
+}
+
+void AudioDecoderSelector::InitializeNextClearDecoder() {
+  DCHECK(message_loop_->BelongsToCurrentThread());
+  DCHECK(!clear_decoders_.empty());
+
+  audio_decoder_ = clear_decoders_.front();
+  clear_decoders_.pop_front();
+  DCHECK(audio_decoder_);
+  audio_decoder_->Initialize(
+      input_stream_,
+      BindToCurrentLoop(base::Bind(
+          &AudioDecoderSelector::ClearDecoderInitDone, weak_this_)),
+      statistics_cb_);
+}
+
+void AudioDecoderSelector::ClearDecoderInitDone(PipelineStatus status) {
+  DCHECK(message_loop_->BelongsToCurrentThread());
+
+  if (status != PIPELINE_OK) {
+    if (!clear_decoders_.empty())
+      InitializeNextClearDecoder();
+    else
+      base::ResetAndReturn(&select_decoder_cb_).Run(NULL, NULL);
+    return;
+  }
+
+  clear_decoders_.clear();
+  base::ResetAndReturn(&select_decoder_cb_).Run(audio_decoder_,
+                                                decrypted_stream_);
+}
+
+}  // namespace media