mkv_writer.h

// Copyright 2015 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 MEDIA_FORMATS_WEBM_MKV_WRITER_H_
#define MEDIA_FORMATS_WEBM_MKV_WRITER_H_

#include <memory>
#include <string>

#include "packager/media/file/file_closer.h"
#include "packager/status.h"
#include "packager/third_party/libwebm/src/mkvmuxer.hpp"

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  // MEDIA_FORMATS_WEBM_MKV_WRITER_H_