muxer.h

// Copyright 2014 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 or at
// https://developers.google.com/open-source/licenses/bsd
//
// Defines the muxer interface.

#ifndef MEDIA_BASE_MUXER_H_
#define MEDIA_BASE_MUXER_H_

#include <memory>
#include <vector>

#include "packager/base/time/clock.h"
#include "packager/media/base/media_handler.h"
#include "packager/media/base/muxer_options.h"
#include "packager/media/event/muxer_listener.h"
#include "packager/media/event/progress_listener.h"
#include "packager/status.h"

namespace shaka {
namespace media {

class MediaSample;

class Muxer : public MediaHandler {
 public:
  explicit Muxer(const MuxerOptions& options);
  virtual ~Muxer();

  void Cancel();

  void SetMuxerListener(std::unique_ptr<MuxerListener> muxer_listener);

  void SetProgressListener(std::unique_ptr<ProgressListener> progress_listener);

  const std::vector<std::shared_ptr<StreamInfo>>& streams() const {
    return streams_;
  }

  void set_clock(base::Clock* clock) {
    clock_ = clock;
  }

 protected:
  Status InitializeInternal() override { return Status::OK; }
  Status Process(std::unique_ptr<StreamData> stream_data) override;
  Status OnFlushRequest(size_t input_stream_index) override { return Finalize(); }

  const MuxerOptions& options() const { return options_; }
  MuxerListener* muxer_listener() { return muxer_listener_.get(); }
  ProgressListener* progress_listener() { return progress_listener_.get(); }
  base::Clock* clock() { return clock_; }

 private:
  // Initialize the muxer.
  virtual Status InitializeMuxer() = 0;

  // Final clean up.
  virtual Status Finalize() = 0;

  // Add a new sample.
  virtual Status AddSample(size_t stream_id,
                           std::shared_ptr<MediaSample> sample) = 0;

  // Finalize the segment or subsegment.
  virtual Status FinalizeSegment(size_t stream_id,
                                 std::shared_ptr<SegmentInfo> segment_info) = 0;

  MuxerOptions options_;
  std::vector<std::shared_ptr<StreamInfo>> streams_;
  std::vector<uint8_t> current_key_id_;
  bool encryption_started_ = false;
  bool cancelled_;

  std::unique_ptr<MuxerListener> muxer_listener_;
  std::unique_ptr<ProgressListener> progress_listener_;
  // An external injected clock, can be NULL.
  base::Clock* clock_;

  DISALLOW_COPY_AND_ASSIGN(Muxer);
};

}  // namespace media
}  // namespace shaka

#endif  // MEDIA_BASE_MUXER_H_