pes_packet_generator.h

// Copyright 2016 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_MP2T_PES_PACKET_GENERATOR_H_
#define PACKAGER_MEDIA_FORMATS_MP2T_PES_PACKET_GENERATOR_H_
 
#include <list>
#include <memory>
 
#include <packager/macros/classes.h>
#include <packager/media/base/media_sample.h>
#include <packager/media/base/stream_info.h>
 
namespace shaka {
namespace media {
 
class AACAudioSpecificConfig;
class NalUnitToByteStreamConverter;
class StreamInfo;
 
namespace mp2t {
 
class PesPacket;
 
class PesPacketGenerator {
 public:
  explicit PesPacketGenerator(int32_t transport_stream_timestamp_offset);
  virtual ~PesPacketGenerator();
 
  virtual bool Initialize(const StreamInfo& stream);
 
  virtual bool PushSample(const MediaSample& sample);
 
  virtual size_t NumberOfReadyPesPackets();
 
  virtual std::unique_ptr<PesPacket> GetNextPesPacket();
 
  virtual bool Flush();
 
 private:
  friend class PesPacketGeneratorTest;
 
  StreamType stream_type_;
 
  const int32_t transport_stream_timestamp_offset_ = 0;
  // Calculated by 90000 / input stream's timescale. This is used to scale the
  // timestamps.
  double timescale_scale_ = 0.0;
 
  std::unique_ptr<NalUnitToByteStreamConverter> converter_;
  std::unique_ptr<AACAudioSpecificConfig> adts_converter_;
 
  // This is the PES packet that this object is currently working on.
  // This can be used to create a PES from multiple audio samples.
  std::unique_ptr<PesPacket> current_processing_pes_;
 
  // Audio stream id PES packet is codec dependent.
  uint8_t audio_stream_id_ = 0;
  std::list<std::unique_ptr<PesPacket>> pes_packets_;
 
  DISALLOW_COPY_AND_ASSIGN(PesPacketGenerator);
};
 
}  // namespace mp2t
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_FORMATS_MP2T_PES_PACKET_GENERATOR_H_