1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83
   84
   85
   86
   87
   88


media / muxers / mp4_muxer_delegate_fragment.h [blame]

// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef MEDIA_MUXERS_MP4_MUXER_DELEGATE_FRAGMENT_H_
#define MEDIA_MUXERS_MP4_MUXER_DELEGATE_FRAGMENT_H_

#include <memory>
#include <optional>
#include <string>
#include <vector>

#include "base/time/time.h"
#include "media/formats/mp4/writable_box_definitions.h"

namespace media {

class Mp4MuxerContext;

template <typename T>
uint32_t BuildFlags(const std::vector<T>& build_flags) {
  uint32_t flags = 0;
  for (auto flag : build_flags) {
    flags |= static_cast<uint32_t>(flag);
  }

  return flags;
}

// It uses the default track index for audio and video regardless of the
// actual track index. Correction of the track index will be done in the
// `Finalize` function that the caller MUST call before writing
// the fragment.
inline constexpr int kDefaultAudioIndex = 0;
inline constexpr int kDefaultVideoIndex = 1;

// This class is responsible for creating and managing the fragment that holds
// audio and video data. It is also responsible for creating the moof and mdat
// boxes that will be written to the file.

class Mp4MuxerDelegateFragment {
 public:
  Mp4MuxerDelegateFragment(Mp4MuxerContext& context,
                           int video_track_id,
                           int audio_track_id,
                           uint32_t sequence_number);
  ~Mp4MuxerDelegateFragment() = default;

  size_t GetAudioSampleSize() const;
  size_t GetVideoSampleSize() const;

  void AddAudioData(scoped_refptr<DecoderBuffer> encoded_data,
                    base::TimeTicks timestamp);
  void AddVideoData(scoped_refptr<DecoderBuffer> encoded_data,
                    base::TimeTicks timestamp);
  void AddAudioLastTimestamp(base::TimeDelta timestamp);
  void AddVideoLastTimestamp(base::TimeDelta timestamp);

  base::TimeTicks GetVideoStartTimestamp() const;
  const mp4::writable_boxes::MovieFragment& GetMovieFragment() const;
  const mp4::writable_boxes::MediaData& GetMediaData() const;

  bool HasSamples() const;
  // The function will ensure that the fragment is complete and ready to be
  // written to the file.
  void Finalize(base::TimeTicks start_audio_time,
                base::TimeTicks start_video_time);

 private:
  void AddNewTrack(uint32_t track_index);
  void AddDataToRun(mp4::writable_boxes::TrackFragmentRun& trun,
                    const DecoderBuffer& encoded_data,
                    base::TimeTicks timestamp);
  void AddDataToMdat(std::vector<uint8_t>& track_data,
                     const DecoderBuffer& encoded_data);
  void AddLastTimestamp(mp4::writable_boxes::TrackFragmentRun& trun,
                        base::TimeDelta timestamp);
  void SetBaseDecodeTime(base::TimeTicks start_audio_time,
                         base::TimeTicks start_video_time);

  const raw_ref<Mp4MuxerContext> context_;
  mp4::writable_boxes::MovieFragment moof_;
  mp4::writable_boxes::MediaData mdat_;
};

}  // namespace media

#endif  // MEDIA_MUXERS_MP4_MUXER_DELEGATE_FRAGMENT_H_