shaka-packager/media/base/media_stream.h

// Copyright (c) 2013 Google Inc. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
//
// MediaStream connects Demuxer to Muxer.

#ifndef MEDIA_BASE_MEDIA_STREAM_H_
#define MEDIA_BASE_MEDIA_STREAM_H_

#include <deque>

#include "base/memory/ref_counted.h"
#include "base/memory/scoped_ptr.h"
#include "media/base/status.h"

namespace media {

class Demuxer;
class Muxer;
class MediaSample;
class StreamInfo;

class MediaStream {
 public:
  enum MediaStreamOperation {
    kPush,
    kPull,
  };
  // Create MediaStream from StreamInfo and Demuxer. MediaStream does not own
  // demuxer.
  MediaStream(scoped_refptr<StreamInfo> info, Demuxer* demuxer);
  ~MediaStream();

  // Connect the stream to Muxer. MediaStream does not own muxer.
  void Connect(Muxer* muxer);

  // Start the stream for pushing or pulling.
  Status Start(MediaStreamOperation operation);

  // Push sample to Muxer (triggered by Demuxer).
  Status PushSample(const scoped_refptr<MediaSample>& sample);

  // Pull sample from Demuxer (triggered by Muxer).
  Status PullSample(scoped_refptr<MediaSample>* sample);

  Demuxer* demuxer() { return demuxer_; }
  Muxer* muxer() { return muxer_; }
  const scoped_refptr<StreamInfo> info() const;

  // Returns a human-readable string describing |*this|.
  std::string ToString() const;

 private:
  // State transition diagram available @ http://goo.gl/ThJQbl.
  enum State {
    kIdle,
    kConnected,
    kDisconnected,
    kPushing,
    kPulling,
  };

  scoped_refptr<StreamInfo> info_;
  Demuxer* demuxer_;
  Muxer* muxer_;
  State state_;
  // An internal buffer to store samples temporarily.
  std::deque<scoped_refptr<MediaSample> > samples_;

  DISALLOW_COPY_AND_ASSIGN(MediaStream);
};

}  // namespace media

#endif  // MEDIA_BASE_MEDIA_STREAM_H_