src/third_party/libwebm/source/mkvwriter.cpp

   1 // Copyright (c) 2012 The WebM project authors. All Rights Reserved.
   2 //
   3 // Use of this source code is governed by a BSD-style license
   4 // that can be found in the LICENSE file in the root of the source
   5 // tree. An additional intellectual property rights grant can be found
   6 // in the file PATENTS.  All contributing project authors may
   7 // be found in the AUTHORS file in the root of the source tree.
   8
   9 #include "mkvwriter.hpp"
  10
  11 #ifdef _MSC_VER
  12 #include <share.h>  // for _SH_DENYWR
  13 #endif
  14
  15 #include <new>
  16
  17 namespace mkvmuxer {
  18
  19 MkvWriter::MkvWriter() : file_(NULL), writer_owns_file_(true) {
  20 }
  21
  22 MkvWriter::MkvWriter(FILE* fp): file_(fp), writer_owns_file_(false) {
  23 }
  24
  25 MkvWriter::~MkvWriter() {
  26   Close();
  27 }
  28
  29 int32 MkvWriter::Write(const void* buffer, uint32 length) {
  30   if (!file_)
  31     return -1;
  32
  33   if (length == 0)
  34     return 0;
  35
  36   if (buffer == NULL)
  37     return -1;
  38
  39   const size_t bytes_written = fwrite(buffer, 1, length, file_);
  40
  41   return (bytes_written == length) ? 0 : -1;
  42 }
  43
  44 bool MkvWriter::Open(const char* filename) {
  45   if (filename == NULL)
  46     return false;
  47
  48   if (file_)
  49     return false;
  50
  51 #ifdef _MSC_VER
  52   file_ = _fsopen(filename, "wb", _SH_DENYWR);
  53 #else
  54   file_ = fopen(filename, "wb");
  55 #endif
  56   if (file_ == NULL)
  57     return false;
  58   return true;
  59 }
  60
  61 void MkvWriter::Close() {
  62   if (file_ && writer_owns_file_) {
  63     fclose(file_);
  64   }
  65   file_ = NULL;
  66 }
  67
  68 int64 MkvWriter::Position() const {
  69   if (!file_)
  70     return 0;
  71
  72 #ifdef _MSC_VER
  73     return _ftelli64(file_);
  74 #else
  75     return ftell(file_);
  76 #endif
  77 }
  78
  79 int32 MkvWriter::Position(int64 position) {
  80   if (!file_)
  81     return -1;
  82
  83 #ifdef _MSC_VER
  84     return _fseeki64(file_, position, SEEK_SET);
  85 #else
  86     return fseek(file_, position, SEEK_SET);
  87 #endif
  88 }
  89
  90 bool MkvWriter::Seekable() const {
  91   return true;
  92 }
  93
  94 void MkvWriter::ElementStartNotify(uint64, int64) {
  95 }
  96
  97 }  // namespace mkvmuxer