mkv_writer.h

// Copyright 2015 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_WEBM_MKV_WRITER_H_
#define PACKAGER_MEDIA_FORMATS_WEBM_MKV_WRITER_H_
 
#include <cstdint>
#include <memory>
#include <string>
 
#include <mkvmuxer/mkvmuxer.h>
#include <mkvmuxer/mkvmuxertypes.h>
 
#include <packager/file.h>
#include <packager/file/file_closer.h>
#include <packager/macros/classes.h>
#include <packager/status.h>
 
namespace shaka {
namespace media {
 
class MkvWriter : public mkvmuxer::IMkvWriter {
 public:
  MkvWriter();
  ~MkvWriter() override;
 
  Status Open(const std::string& name);
  Status Close();
 
  mkvmuxer::int32 Write(const void* buf, mkvmuxer::uint32 len) override;
  mkvmuxer::int64 Position() const override;
  mkvmuxer::int32 Position(mkvmuxer::int64 position) override;
  bool Seekable() const override;
  void ElementStartNotify(mkvmuxer::uint64 element_id,
                          mkvmuxer::int64 position) override;
 
  int64_t WriteFromFile(File* source);
  int64_t WriteFromFile(File* source, int64_t max_copy);
 
  File* file() { return file_.get(); }
 
 private:
  std::unique_ptr<File, FileCloser> file_;
  // Keep track of the position and whether we can seek.
  mkvmuxer::int64 position_;
  bool seekable_;
 
  DISALLOW_COPY_AND_ASSIGN(MkvWriter);
};
 
}  // namespace media
}  // namespace shaka
 
#endif  // PACKAGER_MEDIA_FORMATS_WEBM_MKV_WRITER_H_