97 lines
3.2 KiB
C++
97 lines
3.2 KiB
C++
/* -*- Mode: C++; tab-width: 8; 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/. */
|
|
|
|
#ifndef OutputStreamManager_h
|
|
#define OutputStreamManager_h
|
|
|
|
#include "mozilla/RefPtr.h"
|
|
#include "nsTArray.h"
|
|
#include "MediaSegment.h"
|
|
|
|
namespace mozilla {
|
|
|
|
class MediaInputPort;
|
|
class MediaStream;
|
|
class MediaStreamGraph;
|
|
class OutputStreamManager;
|
|
class ProcessedMediaStream;
|
|
|
|
class OutputStreamData {
|
|
public:
|
|
~OutputStreamData();
|
|
void Init(OutputStreamManager* aOwner,
|
|
ProcessedMediaStream* aStream,
|
|
TrackID aNextAvailableTrackID);
|
|
|
|
// Connect the given input stream's audio and video tracks to mStream.
|
|
// Return false is mStream is already destroyed, otherwise true.
|
|
bool Connect(MediaStream* aStream, TrackID aAudioTrackID, TrackID aVideoTrackID);
|
|
// Disconnect mStream from its input stream.
|
|
// Return false is mStream is already destroyed, otherwise true.
|
|
bool Disconnect();
|
|
// Return true if aStream points to the same object as mStream.
|
|
// Used by OutputStreamManager to remove an output stream.
|
|
bool Equals(MediaStream* aStream) const;
|
|
// Return the graph mStream belongs to.
|
|
MediaStreamGraph* Graph() const;
|
|
// The next TrackID that will not cause a collision in mStream.
|
|
TrackID NextAvailableTrackID() const;
|
|
|
|
private:
|
|
OutputStreamManager* mOwner;
|
|
RefPtr<ProcessedMediaStream> mStream;
|
|
// mPort connects an input stream to our mStream.
|
|
nsTArray<RefPtr<MediaInputPort>> mPorts;
|
|
// For guaranteeing TrackID uniqueness in our mStream.
|
|
TrackID mNextAvailableTrackID = TRACK_INVALID;
|
|
};
|
|
|
|
class OutputStreamManager {
|
|
NS_INLINE_DECL_THREADSAFE_REFCOUNTING(OutputStreamManager);
|
|
|
|
public:
|
|
// Add the output stream to the collection.
|
|
void Add(ProcessedMediaStream* aStream,
|
|
TrackID aNextAvailableTrackID,
|
|
bool aFinishWhenEnded);
|
|
// Remove the output stream from the collection.
|
|
void Remove(MediaStream* aStream);
|
|
// Clear all output streams from the collection.
|
|
void Clear();
|
|
// The next TrackID that will not cause a collision in aOutputStream.
|
|
TrackID NextAvailableTrackIDFor(MediaStream* aOutputStream) const;
|
|
// Return true if the collection empty.
|
|
bool IsEmpty() const
|
|
{
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
return mStreams.IsEmpty();
|
|
}
|
|
// Connect the given input stream's tracks to all output streams.
|
|
void Connect(MediaStream* aStream,
|
|
TrackID aAudioTrackID,
|
|
TrackID aVideoTrackID);
|
|
// Disconnect the input stream to all output streams.
|
|
void Disconnect();
|
|
// Return the graph these streams belong to or null if empty.
|
|
MediaStreamGraph* Graph() const
|
|
{
|
|
MOZ_ASSERT(NS_IsMainThread());
|
|
return !IsEmpty() ? mStreams[0].Graph() : nullptr;
|
|
}
|
|
|
|
private:
|
|
~OutputStreamManager() {}
|
|
// Keep the input stream so we can connect the output streams that
|
|
// are added after Connect().
|
|
RefPtr<MediaStream> mInputStream;
|
|
TrackID mInputAudioTrackID = TRACK_INVALID;
|
|
TrackID mInputVideoTrackID = TRACK_INVALID;
|
|
nsTArray<OutputStreamData> mStreams;
|
|
};
|
|
|
|
} // namespace mozilla
|
|
|
|
#endif // OutputStreamManager_h
|