ts_writer.h

// Copyright 2016 Google Inc. 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 <list>
#include <map>
#include <vector>

#include "packager/base/memory/scoped_ptr.h"
#include "packager/media/base/media_stream.h"
#include "packager/media/file/file.h"
#include "packager/media/file/file_closer.h"
#include "packager/media/formats/mp2t/pes_packet.h"

namespace edash_packager {
namespace media {
namespace mp2t {

class ContinuityCounter {
 public:
  ContinuityCounter();
  ~ContinuityCounter();

  int GetNext();

 private:
  int counter_ = 0;
  DISALLOW_COPY_AND_ASSIGN(ContinuityCounter);
};

class TsWriter {
 public:
  TsWriter();
  virtual ~TsWriter();

  virtual bool Initialize(const StreamInfo& stream_info);

  virtual bool NewSegment(const std::string& file_name);

  virtual bool FinalizeSegment();

  virtual bool AddPesPacket(scoped_ptr<PesPacket> pes_packet);

 private:
  std::vector<uint8_t> psi_ts_packets_;

  uint32_t time_scale_ = 0u;

  ContinuityCounter pmt_continuity_counter_;
  ContinuityCounter pat_continuity_counter_;
  ContinuityCounter elementary_stream_continuity_counter_;

  scoped_ptr<File, FileCloser> current_file_;

  DISALLOW_COPY_AND_ASSIGN(TsWriter);
};

}  // namespace mp2t
}  // namespace media
}  // namespace edash_packager

#endif  // PACKAGER_MEDIA_FORMATS_MP2T_TS_WRITER_H_