webvtt_sample_converter.h

// Copyright 2015 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 PACKAGER_MEDIA_FORMATS_WEBVTT_WEBVTT_SAMPLE_CONVERTER_H_
#define PACKAGER_MEDIA_FORMATS_WEBVTT_WEBVTT_SAMPLE_CONVERTER_H_

#include <stdint.h>
#include <list>

#include "packager/media/base/status.h"
#include "packager/media/formats/mp4/box.h"
#include "packager/media/formats/mp4/box_definitions.h"
#include "packager/media/formats/webvtt/cue.h"

namespace shaka {
namespace media {

void AppendBoxToVector(mp4::Box* box, std::vector<uint8_t>* output_vector);

class WebVttSampleConverter {
 public:
  WebVttSampleConverter();
  ~WebVttSampleConverter();

  void PushSample(std::shared_ptr<MediaSample> sample);

  void Flush();

  size_t ReadySamplesSize();

  std::shared_ptr<MediaSample> PopSample();

 private:
  // Handle |cues_| except the last item, and create samples from them.
  // All cues that overlap with the latest cue are not processed.
  // Usually the last cue (and cues that overlap with it) should not be
  // processed right away because the following cues may overlap with the latest
  // cue or the existing cues.
  // If a cue has been proceessed, then this returns true.
  bool HandleAllCuesButLatest();

  // Same as HandleAllCuesButLatest() but it also includes the latest cue.
  // If a cue has been processed, then this returns true.
  bool HandleAllCues();

  // Sweep line algorithm that handles the cues in |cues_|.
  // This does not erase |cues_|.
  // If a cue has been processed, this returns true.
  // |sweep_line| is the start time and |sweep_stop_time| is when the sweep
  // should stop.
  bool SweepCues(uint64_t sweep_line, uint64_t sweep_stop_time);

  // This is going to be in 'mdat' box. Keep this around until a sample is
  // ready.
  std::list<Cue> cues_;

  // For comment samples.
  std::list<mp4::VTTAdditionalTextBox> additional_texts_;

  // Samples that are ready to be processed.
  std::list<std::shared_ptr<MediaSample>> ready_samples_;

  // This keeps track of the max end time of the processed cues which is the
  // start time of the next cue. Used to check if cue_current_time has to be set
  // or an empty cue (gap) has to be added.
  uint64_t next_cue_start_time_;

  DISALLOW_COPY_AND_ASSIGN(WebVttSampleConverter);
};

}  // namespace media
}  // namespace shaka

#endif  // PACKAGER_MEDIA_FORMATS_WEBVTT_WEBVTT_SAMPLE_CONVERTER_H_