matroskavideo.hpp

// ***************************************************************** -*- C++ -*-
/*
 * Copyright (C) 2004-2021 Exiv2 authors
 * This program is part of the Exiv2 distribution.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, 5th Floor, Boston, MA 02110-1301 USA.
 */
#ifndef MATROSKAVIDEO_HPP_
#define MATROSKAVIDEO_HPP_
 
// *****************************************************************************
#include "exiv2lib_export.h"
 
// included header files
#include "image.hpp"
 
// *****************************************************************************
// namespace extensions
namespace Exiv2 {
 
// *****************************************************************************
// class definitions
namespace Internal {
 
enum matroskaTypeEnum : char {
  String = 's',
  Integer = 'i',
  UInteger = 'u',
  Date = 'd',
  InternalField = 'n',
  Boolean = 'o',
  Binary = 'b',
  Master = 'm',
  Float = 'f',
  Utf8 = '8',
  UndefinedType = 'z',
};
 
enum matroskaProcessEnum : char {
  Process = 'p',
  Skip = 's',
  Composite = 'c',
  Undefined = 'u',
};
 
struct MatroskaTag {
  uint64_t _id;
  std::string _label;
  matroskaTypeEnum _type;
  matroskaProcessEnum _process;
 
  MatroskaTag(uint64_t id, std::string label, matroskaTypeEnum type, matroskaProcessEnum process) :
      _id(id), _label(std::move(label)), _type(type), _process(process) {
  }
 
  MatroskaTag(uint64_t id, std::string label) :
      _id(id),
      _label(std::move(label)),
      _type(matroskaTypeEnum::UndefinedType),
      _process(matroskaProcessEnum::Undefined) {
  }
 
  bool operator==(uint64_t id) const {
    return id == _id;
  }
 
  [[nodiscard]] bool isSkipped() const {
    return _process == Skip;
  }
  [[nodiscard]] bool isComposite() const {
    return _process == Composite;
  }
  void dump(std::ostream& os) const {
    os << " MatroskaTag "
       << " id: [0x" << std::hex << _id << "] label:[" << _label << "] type:[" << _type << "] process :[" << _process
       << "]\n";
  }
};
}  // namespace Internal
 
class EXIV2API MatroskaVideo : public Image {
 public:
 
 
  explicit MatroskaVideo(BasicIo::UniquePtr io);
 
 
  void readMetadata() override;
  void writeMetadata() override;
 
 
  [[nodiscard]] std::string mimeType() const override;
 
 protected:
  [[nodiscard]] static uint32_t findBlockSize(byte b);
  void decodeBlock();
  void decodeInternalTags(const Internal::MatroskaTag* tag, const byte* buf);
  void decodeStringTags(const Internal::MatroskaTag* tag, const byte* buf);
  void decodeIntegerTags(const Internal::MatroskaTag* tag, const byte* buf);
  void decodeBooleanTags(const Internal::MatroskaTag* tag, const byte* buf);
  void decodeDateTags(const Internal::MatroskaTag* tag, const byte* buf, size_t size);
  void decodeFloatTags(const Internal::MatroskaTag* tag, const byte* buf);
 
 private:
  bool continueTraversing_{};
  uint64_t height_{};
  uint64_t width_{};
  uint32_t track_count_{};
  double time_code_scale_ = 1.0;
  uint64_t stream_{};
 
  static constexpr double bytesMB = 1048576;
 
};  // class MatroskaVideo
 
// *****************************************************************************
// template, inline and free functions
 
// These could be static private functions on Image subclasses but then
// ImageFactory needs to be made a friend.
EXIV2API Image::UniquePtr newMkvInstance(BasicIo::UniquePtr io, bool create);
 
EXIV2API bool isMkvType(BasicIo& iIo, bool advance);
 
}  // namespace Exiv2
 
#endif  // #ifndef MATROSKAVIDEO_HPP_