summaryrefslogtreecommitdiffstats
path: root/dom/media/MediaStreamWindowCapturer.cpp
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--dom/media/MediaStreamWindowCapturer.cpp77
1 files changed, 77 insertions, 0 deletions
diff --git a/dom/media/MediaStreamWindowCapturer.cpp b/dom/media/MediaStreamWindowCapturer.cpp
new file mode 100644
index 0000000000..142242eff0
--- /dev/null
+++ b/dom/media/MediaStreamWindowCapturer.cpp
@@ -0,0 +1,77 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-*/
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this file,
+ * You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "MediaStreamWindowCapturer.h"
+
+#include "AudioStreamTrack.h"
+#include "DOMMediaStream.h"
+#include "MediaTrackGraph.h"
+
+namespace mozilla {
+using dom::AudioStreamTrack;
+using dom::MediaStreamTrack;
+
+MediaStreamWindowCapturer::CapturedTrack::CapturedTrack(
+ MediaStreamTrack* aTrack, uint64_t aWindowID)
+ : mTrack(aTrack),
+ mPort(aTrack->Graph()->ConnectToCaptureTrack(aWindowID,
+ aTrack->GetTrack())) {}
+
+MediaStreamWindowCapturer::CapturedTrack::~CapturedTrack() {
+ if (mPort) {
+ mPort->Destroy();
+ }
+}
+
+MediaStreamWindowCapturer::MediaStreamWindowCapturer(DOMMediaStream* aStream,
+ uint64_t aWindowId)
+ : mStream(aStream), mWindowId(aWindowId) {
+ mStream->RegisterTrackListener(this);
+ nsTArray<RefPtr<AudioStreamTrack>> tracks;
+ mStream->GetAudioTracks(tracks);
+ for (const auto& t : tracks) {
+ if (t->Ended()) {
+ continue;
+ }
+ AddTrack(t);
+ }
+}
+
+MediaStreamWindowCapturer::~MediaStreamWindowCapturer() {
+ if (mStream) {
+ mStream->UnregisterTrackListener(this);
+ }
+}
+
+void MediaStreamWindowCapturer::NotifyTrackAdded(
+ const RefPtr<MediaStreamTrack>& aTrack) {
+ if (AudioStreamTrack* at = aTrack->AsAudioStreamTrack()) {
+ AddTrack(at);
+ }
+}
+
+void MediaStreamWindowCapturer::NotifyTrackRemoved(
+ const RefPtr<MediaStreamTrack>& aTrack) {
+ if (AudioStreamTrack* at = aTrack->AsAudioStreamTrack()) {
+ RemoveTrack(at);
+ }
+}
+
+void MediaStreamWindowCapturer::AddTrack(AudioStreamTrack* aTrack) {
+ if (aTrack->Ended()) {
+ return;
+ }
+ mTracks.AppendElement(MakeUnique<CapturedTrack>(aTrack, mWindowId));
+}
+
+void MediaStreamWindowCapturer::RemoveTrack(AudioStreamTrack* aTrack) {
+ for (size_t i = mTracks.Length(); i > 0; --i) {
+ if (mTracks[i - 1]->mTrack == aTrack) {
+ mTracks.RemoveElementAt(i - 1);
+ break;
+ }
+ }
+}
+} // namespace mozilla