packed_audio_segmenter.h

// Copyright 2018 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_FORMATS_PACKED_AUDIO_PACKED_AUDIO_SEGMENTER_H_
#define PACKAGER_MEDIA_FORMATS_PACKED_AUDIO_PACKED_AUDIO_SEGMENTER_H_
 
#include <cstdint>
#include <memory>
#include <string>
#include <vector>
 
#include <packager/media/base/buffer_writer.h>
#include <packager/media/base/stream_info.h>
#include <packager/status.h>
 
namespace shaka {
namespace media {
 
class AACAudioSpecificConfig;
class Id3Tag;
class MediaSample;
 
constexpr double kPackedAudioTimescale = 90000;
 
constexpr char kTimestampOwnerIdentifier[] =
    "com.apple.streaming.transportStreamTimestamp";
 
constexpr char kAudioDescriptionOwnerIdentifier[] =
    "com.apple.streaming.audioDescription";
 
class PackedAudioSegmenter {
 public:
  explicit PackedAudioSegmenter(int32_t transport_stream_timestamp_offset);
  virtual ~PackedAudioSegmenter();
 
  // This function is made virtual for testing.
  virtual Status Initialize(const StreamInfo& stream_info);
 
  // This function is made virtual for testing.
  virtual Status AddSample(const MediaSample& sample);
 
  // This function is made virtual for testing.
  virtual Status FinalizeSegment();
 
  // This function is made virtual for testing.
  virtual double TimescaleScale() const;
 
  BufferWriter* segment_buffer() { return &segment_buffer_; }
 
 private:
  PackedAudioSegmenter(const PackedAudioSegmenter&) = delete;
  PackedAudioSegmenter& operator=(const PackedAudioSegmenter&) = delete;
 
  // These functions is made virtual for testing.
  virtual std::unique_ptr<AACAudioSpecificConfig> CreateAdtsConverter();
  virtual std::unique_ptr<Id3Tag> CreateId3Tag();
 
  Status EncryptionAudioSetup(const MediaSample& sample);
  Status StartNewSegment(const MediaSample& first_sample);
 
  const int32_t transport_stream_timestamp_offset_ = 0;
  // Codec for the stream.
  Codec codec_ = kUnknownCodec;
  std::vector<uint8_t> audio_codec_config_;
  // Calculated by output stream's timescale / input stream's timescale. This is
  // used to scale the timestamps.
  double timescale_scale_ = 0.0;
  // Whether it is the start of a new segment.
  bool start_of_new_segment_ = true;
 
  // Audio setup information for encrypted segment.
  std::string audio_setup_information_;
  // AAC is carried in ADTS.
  std::unique_ptr<AACAudioSpecificConfig> adts_converter_;
 
  BufferWriter segment_buffer_;
};
 
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_FORMATS_PACKED_AUDIO_PACKED_AUDIO_SEGMENTER_H_