trick_play_handler.h

// Copyright 2017 Google LLC. 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_BASE_TRICK_PLAY_HANDLER_H_
#define PACKAGER_MEDIA_BASE_TRICK_PLAY_HANDLER_H_
 
#include <cstddef>
#include <cstdint>
#include <list>
#include <memory>
 
#include <packager/media/base/media_handler.h>
#include <packager/media/base/media_sample.h>
#include <packager/media/base/stream_info.h>
#include <packager/status.h>
 
namespace shaka {
namespace media {
 
class VideoStreamInfo;
 
// The stream data in trick play streams are not simple duplicates. Some
// information get changed (e.g. VideoStreamInfo.trick_play_factor).
class TrickPlayHandler : public MediaHandler {
 public:
  explicit TrickPlayHandler(uint32_t factor);
 
 private:
  TrickPlayHandler(const TrickPlayHandler&) = delete;
  TrickPlayHandler& operator=(const TrickPlayHandler&) = delete;
 
  Status InitializeInternal() override;
  Status Process(std::unique_ptr<StreamData> stream_data) override;
  Status OnFlushRequest(size_t input_stream_index) override;
 
  Status OnStreamInfo(const StreamInfo& info);
  Status OnSegmentInfo(std::shared_ptr<const SegmentInfo> info);
  Status OnMediaSample(const MediaSample& sample);
  Status OnTrickFrame(const MediaSample& sample);
 
  const uint32_t factor_;
 
  uint64_t total_frames_ = 0;
  uint64_t total_key_frames_ = 0;
  uint64_t total_trick_frames_ = 0;
 
  // We cannot just send video info through as we need to calculate the play
  // rate using the first two trick play frames. This reference should only be
  // used to update the play back rate before video info is sent downstream.
  // After getting sent downstream, this should never be used.
  std::shared_ptr<VideoStreamInfo> video_info_;
 
  // We need to track the segment that most recently finished so that we can
  // extend its duration if there are empty segments.
  std::shared_ptr<SegmentInfo> previous_segment_;
 
  // Since we are dropping frames, the time that those frames would have been
  // on screen need to be added to the frame before them. Keep a reference to
  // the most recent trick play frame so that we can grow its duration as we
  // drop other frames.
  std::shared_ptr<MediaSample> previous_trick_frame_;
 
  // Since we cannot send messages downstream right away, keep a queue of
  // messages that need to be sent down. At the start, we use this to queue
  // messages until we can send out |video_info_|. To ensure messages are
  // kept in order, messages are only dispatched through this queue and never
  // directly.
  std::list<std::unique_ptr<StreamData>> delayed_messages_;
};
 
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_BASE_TRICK_PLAY_HANDLER_H_