ts_writer.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_TS_WRITER_H_
#define PACKAGER_MEDIA_FORMATS_MP2T_TS_WRITER_H_
 
#include <memory>
 
#include <packager/media/base/buffer_writer.h>
#include <packager/media/formats/mp2t/continuity_counter.h>
 
namespace shaka {
namespace media {
namespace mp2t {
 
class PesPacket;
class ProgramMapTableWriter;
 
class TsWriter {
 public:
  explicit TsWriter(std::unique_ptr<ProgramMapTableWriter> pmt_writer);
  virtual ~TsWriter();
 
  virtual bool NewSegment(BufferWriter* buffer);
 
  virtual void SignalEncrypted();
 
  virtual bool AddPesPacket(std::unique_ptr<PesPacket> pes_packet,
                            BufferWriter* buffer);
 
 private:
  TsWriter(const TsWriter&) = delete;
  TsWriter& operator=(const TsWriter&) = delete;
 
  // True if further segments generated by this instance should be encrypted.
  bool encrypted_ = false;
 
  ContinuityCounter pat_continuity_counter_;
  ContinuityCounter elementary_stream_continuity_counter_;
 
  std::unique_ptr<ProgramMapTableWriter> pmt_writer_;
};
 
}  // namespace mp2t
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_FORMATS_MP2T_TS_WRITER_H_