demuxer.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

#ifndef MEDIA_BASE_DEMUXER_H_
#define MEDIA_BASE_DEMUXER_H_

#include <deque>
#include <memory>
#include <vector>

#include "packager/base/compiler_specific.h"
#include "packager/media/base/container_names.h"
#include "packager/media/base/status.h"

namespace shaka {
namespace media {

class Decryptor;
class File;
class KeySource;
class MediaParser;
class MediaSample;
class MediaStream;
class StreamInfo;

class Demuxer {
 public:
  explicit Demuxer(const std::string& file_name);
  ~Demuxer();

  void SetKeySource(std::unique_ptr<KeySource> key_source);

  Status Initialize();

  Status Run();

  Status Parse();

  void Cancel();

  const std::vector<std::unique_ptr<MediaStream>>& streams() {
    return streams_;
  }

  MediaContainerName container_name() { return container_name_; }

 private:
  Demuxer(const Demuxer&) = delete;
  Demuxer& operator=(const Demuxer&) = delete;

  struct QueuedSample {
    QueuedSample(uint32_t track_id, std::shared_ptr<MediaSample> sample);
    ~QueuedSample();

    uint32_t track_id;
    std::shared_ptr<MediaSample> sample;
  };

  // Parser init event.
  void ParserInitEvent(const std::vector<std::shared_ptr<StreamInfo>>& streams);
  // Parser new sample event handler. Queues the samples if init event has not
  // been received, otherwise calls PushSample() to push the sample to
  // corresponding stream.
  bool NewSampleEvent(uint32_t track_id,
                      const std::shared_ptr<MediaSample>& sample);
  // Helper function to push the sample to corresponding stream.
  bool PushSample(uint32_t track_id,
                  const std::shared_ptr<MediaSample>& sample);

  std::string file_name_;
  File* media_file_;
  bool init_event_received_;
  Status init_parsing_status_;
  // Queued samples received in NewSampleEvent() before ParserInitEvent().
  std::deque<QueuedSample> queued_samples_;
  std::unique_ptr<MediaParser> parser_;
  std::vector<std::unique_ptr<MediaStream>> streams_;
  MediaContainerName container_name_;
  std::unique_ptr<uint8_t[]> buffer_;
  std::unique_ptr<KeySource> key_source_;
  bool cancelled_;
};

}  // namespace media
}  // namespace shaka

#endif  // MEDIA_BASE_DEMUXER_H_