/*
    RawSpeed - RAW file decoder.

    Copyright (C) 2009-2014 Klaus Post
    Copyright (C) 2017 Axel Waggershauser
    Copyright (C) 2018 Roman Lebedev

    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.

    This library 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
    Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public
    License along with this library; if not, write to the Free Software
    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/

#pragma once

#include "rawspeedconfig.h"
#include "adt/NORangesSet.h"
#include "io/Buffer.h"
#include "io/ByteStream.h"
#include "io/Endianness.h"
#include "parsers/TiffParserException.h"
#include "tiff/TiffEntry.h"
#include "tiff/TiffTag.h"
#include <cstdint>
#include <map>
#include <memory>
#include <string>
#include <vector>

namespace rawspeed {

class TiffIFD;
class TiffRootIFD;
template <typename T> class NORangesSet;

using TiffIFDOwner = std::unique_ptr<TiffIFD>;
using TiffRootIFDOwner = std::unique_ptr<TiffRootIFD>;
using TiffEntryOwner = std::unique_ptr<TiffEntry>;

class TiffIFD {
  uint32_t nextIFD = 0;

  TiffIFD* const parent;

  std::vector<TiffIFDOwner> subIFDs;

  int subIFDCount = 0;
  int subIFDCountRecursive = 0;

  std::map<TiffTag, TiffEntryOwner> entries;

  virtual void anchor() const;

  friend class TiffEntry;
  friend class FiffParser;
  friend class TiffParser;

  void recursivelyIncrementSubIFDCount();
  void checkSubIFDs(int headroom) const;
  void recursivelyCheckSubIFDs(int headroom) const;

  void add(TiffIFDOwner subIFD);
  void add(TiffEntryOwner entry);
  TiffRootIFDOwner parseMakerNote(NORangesSet<Buffer>* ifds,
                                  const TiffEntry* t);
  void parseIFDEntry(NORangesSet<Buffer>* ifds, ByteStream& bs);

  // TIFF IFD are tree-like structure, with branches.
  // A branch (IFD) can have branches (IFDs) of it's own.
  // We must be careful to weed-out all the degenerative cases that
  // can be produced e.g. via fuzzing, or other means.
  struct Limits final {
    // How many layers of IFD's can there be?
    // All RPU samples (as of 2018-02-11) are ok with 4.
    // However, let's be on the safe side, and pad it by one.
    static constexpr int Depth = 4 + 1;

    // How many sub-IFD's can this IFD have?
    // NOTE: only for the given IFD, *NOT* recursively including all sub-IFD's!
    // All RPU samples (as of 2018-02-11) are ok with 5.
    // However, let's be on the safe side, and double it.
    static constexpr int SubIFDCount = 5 * 2;

    // How many sub-IFD's can this IFD have, recursively?
    // All RPU samples (as of 2018-02-11) are ok with 14.
    // However, let's be on the safe side, and double it.
    static constexpr int RecursiveSubIFDCount = 14 * 2;
  };

public:
  explicit TiffIFD(TiffIFD* parent);

  TiffIFD(TiffIFD* parent, NORangesSet<Buffer>* ifds, DataBuffer data,
          uint32_t offset);

  virtual ~TiffIFD() = default;

  // make sure we never copy-constuct/assign a TiffIFD to keep the owning
  // subcontainers contents save
  TiffIFD(const TiffIFD&) = delete;
  TiffIFD& operator=(const TiffIFD&) = delete;

  [[nodiscard]] uint32_t getNextIFD() const { return nextIFD; }
  [[nodiscard]] std::vector<const TiffIFD*> getIFDsWithTag(TiffTag tag) const;
  [[nodiscard]] const TiffIFD* getIFDWithTag(TiffTag tag,
                                             uint32_t index = 0) const;
  [[nodiscard]] TiffEntry* getEntry(TiffTag tag) const;
  [[nodiscard]] TiffEntry* RAWSPEED_READONLY
  getEntryRecursive(TiffTag tag) const;
  [[nodiscard]] bool RAWSPEED_READONLY hasEntry(TiffTag tag) const {
    return entries.contains(tag);
  }
  [[nodiscard]] bool hasEntryRecursive(TiffTag tag) const {
    return getEntryRecursive(tag) != nullptr;
  }

  [[nodiscard]] const std::vector<TiffIFDOwner>& getSubIFDs() const {
    return subIFDs;
  }
  //  const std::map<TiffTag, TiffEntry*>& getEntries() const { return entries;
  //  }
};

struct TiffID final {
  std::string make;
  std::string model;
};

class TiffRootIFD final : public TiffIFD {
  void anchor() const override;

public:
  const DataBuffer rootBuffer;

  TiffRootIFD(TiffIFD* parent_, NORangesSet<Buffer>* ifds, DataBuffer data,
              uint32_t offset)
      : TiffIFD(parent_, ifds, data, offset), rootBuffer(data) {}

  // find the MAKE and MODEL tags identifying the camera
  // note: the returned strings are trimmed automatically
  [[nodiscard]] TiffID getID() const;
};

inline Endianness getTiffByteOrder(ByteStream bs, uint32_t pos,
                                   const char* context = "") {
  if (bs.hasPatternAt("II", pos))
    return Endianness::little;
  if (bs.hasPatternAt("MM", pos))
    return Endianness::big;

  ThrowTPE("Failed to parse TIFF endianness information in %s.", context);
}

} // namespace rawspeed
