Unmute Tab Audio For Desktop Share

media/audio/virtual_audio_sink.h

+// Copyright 2016 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_AUDIO_VIRTUAL_AUDIO_SINK_H_
+#define MEDIA_AUDIO_VIRTUAL_AUDIO_SINK_H_
+
+#include "base/callback.h"
+#include "base/macros.h"
+#include "base/memory/weak_ptr.h"
+#include "media/audio/audio_io.h"
+#include "media/audio/audio_source_diverter.h"
+#include "media/base/audio_converter.h"
+#include "media/base/audio_shifter.h"
+
+namespace base {
+class SingleThreadTaskRunner;
+}
+
+namespace media {
+
+class VirtualAudioInputStream;
+
+// This class is an adapter between push model and pull model for audio stream.
+// Audio provider calls OnData() to push audio data into this class.
+// Audio consumer calls ProvideInput() to pull audio data.
+class MEDIA_EXPORT VirtualAudioSink
+    : public AudioPushSink,
+      public AudioConverter::InputCallback,
+      public base::SupportsWeakPtr<VirtualAudioSink> {
+ public:
+  typedef base::Callback<void(VirtualAudioSink* sink)> AfterCloseCallback;
+
+  // Construct an audio loopback pathway to the given |target| (not owned).
+  // |target| must outlive this instance.
+  VirtualAudioSink(AudioParameters param,
+                   VirtualAudioInputStream* target,
+                   AfterCloseCallback callback);
+  ~VirtualAudioSink() override;
+
+  // AudioPushSink
+  void Close() override;
+  void OnData(const AudioBus& source, base::TimeTicks reference_time) override;
+
+  // AudioConverter::InputCallback
+  double ProvideInput(AudioBus* audio_bus,
+                      base::TimeDelta buffer_delay) override;
+
+ private:
+  void StoreData(AudioBus* source, base::TimeTicks reference_time);
+
+  const AudioParameters params_;
+  VirtualAudioInputStream* const target_;
+  AudioShifter shifter_;
+  scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
+  AfterCloseCallback after_close_callback_;
+};
+
+}  // namespace media
+
+#endif  // MEDIA_AUDIO_VIRTUAL_AUDIO_SINK_H_