// SPDX-License-Identifier: GPL-2.0-or-later

#pragma once

// *****************************************************************************

#include "exiv2lib_export.h"

// included header files

#include "image.hpp"

#include <set>

// *****************************************************************************

// namespace extensions

namespace Exiv2 {

[[deprecated]] EXIV2API bool enableBMFF(bool enable = true);

}  // namespace Exiv2

#ifdef EXV_ENABLE_BMFF

namespace Exiv2 {

struct Iloc {

  explicit Iloc(uint32_t ID = 0, uint32_t start = 0, uint32_t length = 0) : ID_(ID), start_(start), length_(length) {

  }

  uint32_t ID_;

  uint32_t start_;

  uint32_t length_;

  [[nodiscard]] std::string toString() const;

};  // class Iloc

// *****************************************************************************

// class definitions

/*!

  @brief Class to access BMFF images.

 */

class EXIV2API BmffImage : public Image {

 public:

  //! @name Creators

  //@{

  /*!

    @brief Constructor to open a BMFF image. Since the

        constructor can not return a result, callers should check the

        good() method after object construction to determine success

        or failure.

    @param io An auto-pointer that owns a BasicIo instance used for

        reading and writing image metadata. \b Important: The constructor

        takes ownership of the passed in BasicIo instance through the

        auto-pointer. Callers should not continue to use the BasicIo

        instance after it is passed to this method.  Use the Image::io()

        method to get a temporary reference.

    @param create Specifies if an existing image should be read (false)

        or if a new file should be created (true).

   */

  BmffImage(BasicIo::UniquePtr io, bool create, size_t max_box_depth = 1000);

  //@}

  //@{

  void parseTiff(uint32_t root_tag, uint64_t length);

  /*!

    @brief parse embedded tiff file (Exif metadata)

    @param root_tag root of parse tree Tag::root, Tag::cmt2 etc.

    @param length tiff block length

    @param start offset in file (default, io_->tell())

   @

   */

  void parseTiff(uint32_t root_tag, uint64_t length, uint64_t start);

  //@}

  //@{

  /*!

    @brief parse embedded xmp/xml

    @param length xmp block length

    @param start offset in file

   @

   */

  void parseXmp(uint64_t length, uint64_t start);

  //@}

  //@{

  /*!

  @brief Parse a Canon PRVW or THMB box and add an entry to the set

      of native previews.

  @param data Buffer containing the box

  @param out Logging stream

  @param bTrace Controls logging

  @param width_offset Index of image width field in data

  @param height_offset Index of image height field in data

  @param size_offset Index of image size field in data

  @param relative_position Location of the start of image data in the file,

      relative to the current file position indicator.

  */

  void parseCr3Preview(const DataBuf& data, std::ostream& out, bool bTrace, uint8_t version, size_t width_offset,

                       size_t height_offset, size_t size_offset, size_t relative_position);

  //@}

  //! @name Manipulators

  //@{

  void readMetadata() override;

  void writeMetadata() override;

  void setExifData(const ExifData&) override;

  void setIptcData(const IptcData&) override;

  void setXmpData(const XmpData&) override;

  void setComment(const std::string& comment) override;

  void printStructure(std::ostream& out, Exiv2::PrintStructureOption option, size_t depth) override;

  //@}

  //! @name Accessors

  //@{

  [[nodiscard]] std::string mimeType() const override;

  [[nodiscard]] uint32_t pixelWidth() const override;

  [[nodiscard]] uint32_t pixelHeight() const override;

  //@}

  static constexpr Exiv2::ByteOrder endian_{Exiv2::bigEndian};

 private:

  void openOrThrow() const;

  /*!

    @brief recursiveBoxHandler

    @throw Error if we visit a box more than once

    @param pbox_end The end location of the parent box. Boxes are

        nested, so we must not read beyond this.

    @return address of next box

    @warning This function should only be called by readMetadata()

   */

  uint64_t boxHandler(std::ostream& out, Exiv2::PrintStructureOption option, uint64_t pbox_end, size_t depth);

  uint32_t fileType_{0};

  std::set<size_t> visits_;

  uint64_t visits_max_{0};

  uint16_t unknownID_{0xffff};

  uint16_t exifID_{0xffff};

  uint16_t xmpID_{0};

  std::map<uint32_t, Iloc> ilocs_;

  bool bReadMetadata_{false};

  const size_t max_box_depth_;

  //@}

  /*!

    @brief box utilities

   */

  static std::string toAscii(uint32_t n);

  std::string boxName(uint32_t box);

  static bool superBox(uint32_t box);

  static bool fullBox(uint32_t box);

  static std::string uuidName(const Exiv2::DataBuf& uuid);

  /*!

    @brief Wrapper around brotli to uncompress JXL brob content.

   */

#ifdef EXV_HAVE_BROTLI

  static void brotliUncompress(const byte* compressedBuf, size_t compressedBufSize, DataBuf& arr);

#endif

};  // class BmffImage

// *****************************************************************************

// template, inline and free functions

// These could be static private functions on Image subclasses but then

// ImageFactory needs to be made a friend.

/*!

  @brief Create a new BMFF instance and return an auto-pointer to it.

         Caller owns the returned object and the auto-pointer ensures that

         it will be deleted.

 */

EXIV2API Image::UniquePtr newBmffInstance(BasicIo::UniquePtr io, bool create);

//! Check if the file iIo is a BMFF image.

EXIV2API bool isBmffType(BasicIo& iIo, bool advance);

}  // namespace Exiv2

#endif  // EXV_ENABLE_BMFF