track_run_iterator.h

// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
 
#ifndef PACKAGER_MEDIA_FORMATS_MP4_TRACK_RUN_ITERATOR_H_
#define PACKAGER_MEDIA_FORMATS_MP4_TRACK_RUN_ITERATOR_H_
 
#include <cstdint>
#include <map>
#include <memory>
#include <vector>
 
#include <packager/macros/classes.h>
#include <packager/media/formats/mp4/box_definitions.h>
 
namespace shaka {
namespace media {
 
class DecryptConfig;
 
namespace mp4 {
 
struct SampleInfo;
struct TrackRunInfo;
 
class TrackRunIterator {
 public:
  explicit TrackRunIterator(const Movie* moov);
  ~TrackRunIterator();
 
  bool Init();
 
  bool Init(const MovieFragment& moof);
 
  bool IsRunValid() const;
  bool IsSampleValid() const;
 
  void AdvanceRun();
  void AdvanceSample();
 
  bool AuxInfoNeedsToBeCached();
 
  bool CacheAuxInfo(const uint8_t* buf, int size);
 
  int64_t GetMaxClearOffset();
 
  uint32_t track_id() const;
  int64_t aux_info_offset() const;
  int aux_info_size() const;
  bool is_encrypted() const;
  bool is_audio() const;
  bool is_video() const;
 
  const AudioSampleEntry& audio_description() const;
  const VideoSampleEntry& video_description() const;
 
  int64_t sample_offset() const;
  int sample_size() const;
  int64_t dts() const;
  int64_t cts() const;
  int64_t duration() const;
  bool is_keyframe() const;
 
  std::unique_ptr<DecryptConfig> GetDecryptConfig();
 
 private:
  void ResetRun();
  const TrackEncryption& track_encryption() const;
  int64_t GetTimestampAdjustment(const Movie& movie,
                                 const Track& track,
                                 const TrackFragment* traf);
 
  const Movie* moov_;
 
  std::vector<TrackRunInfo> runs_;
  std::vector<TrackRunInfo>::const_iterator run_itr_;
  std::vector<SampleInfo>::const_iterator sample_itr_;
 
  // Track the start dts of the next segment, only useful if decode_time box is
  // absent.
  std::vector<int64_t> next_fragment_start_dts_;
 
  int64_t sample_dts_;
  int64_t sample_offset_;
 
  // TrackId => adjustment map.
  std::map<uint32_t, int64_t> timestamp_adjustment_map_;
 
  DISALLOW_COPY_AND_ASSIGN(TrackRunIterator);
};
 
}  // namespace mp4
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_FORMATS_MP4_TRACK_RUN_ITERATOR_H_