nal_unit_to_byte_stream_converter.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_CODECS_NAL_UNIT_TO_BYTE_STREAM_CONVERTER_H_
#define PACKAGER_MEDIA_CODECS_NAL_UNIT_TO_BYTE_STREAM_CONVERTER_H_
 
#include <cstddef>
#include <cstdint>
#include <vector>
 
#include <packager/macros/classes.h>
#include <packager/media/base/decrypt_config.h>
#include <packager/media/codecs/avc_decoder_configuration_record.h>
 
namespace shaka {
namespace media {
 
class BufferWriter;
class VideoStreamInfo;
 
void EscapeNalByteSequence(const uint8_t* input,
                           size_t input_size,
                           BufferWriter* output);
 
// Methods are virtual for mocking.
class NalUnitToByteStreamConverter {
 public:
  NalUnitToByteStreamConverter();
  virtual ~NalUnitToByteStreamConverter();
 
  virtual bool Initialize(const uint8_t* decoder_configuration_data,
                          size_t decoder_configuration_data_size);
 
  virtual bool ConvertUnitToByteStream(const uint8_t* sample,
                                       size_t sample_size,
                                       bool is_key_frame,
                                       std::vector<uint8_t>* output);
 
  virtual bool ConvertUnitToByteStreamWithSubsamples(
      const uint8_t* sample,
      size_t sample_size,
      bool is_key_frame,
      bool escape_encrypted_nalu,
      std::vector<uint8_t>* output,
      std::vector<SubsampleEntry>* subsamples);
 
 private:
  friend class NalUnitToByteStreamConverterTest;
 
  int nalu_length_size_;
  AVCDecoderConfigurationRecord decoder_config_;
  std::vector<uint8_t> decoder_configuration_in_byte_stream_;
 
  DISALLOW_COPY_AND_ASSIGN(NalUnitToByteStreamConverter);
};
 
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_CODECS_NAL_UNIT_TO_BYTE_STREAM_CONVERTER_H_