PdfNameTrees.h

// SPDX-FileCopyrightText: 2006 Dominik Seichter <domseichter@web.de>
// SPDX-FileCopyrightText: 2024 Francesco Pretto <ceztko@gmail.com>
// SPDX-License-Identifier: LGPL-2.0-or-later OR MPL-2.0
 
#ifndef PDF_NAME_TREES_H
#define PDF_NAME_TREES_H
 
#include "PdfNameTree.h"
#include "PdfNameTreeOperations.h"
 
namespace PoDoFo {
 
class PODOFO_API PdfNameTrees final : public PdfDictionaryElement, public PdfNameTreeOperations
{
    friend class PdfDocument;
 
private:
    PdfNameTrees(PdfDocument& doc);
 
    PdfNameTrees(PdfObject& obj);
 
    PdfNameTrees(const PdfNameTrees&) = delete;
 
public:
    template <typename TNameTree>
    TNameTree* GetTree()
    {
        return static_cast<TNameTree*>(getNameTree(getType<TNameTree>()));
    }
 
    template <typename TNameTree>
    const TNameTree* GetTree() const
    {
        return static_cast<const TNameTree*>(getNameTree(getType<TNameTree>()));
    }
 
    template <typename TNameTree>
    TNameTree& MustGetTree()
    {
        return static_cast<TNameTree&>(mustGetNameTree(getType<TNameTree>()));
    }
 
    template <typename TNameTree>
    const TNameTree& MustGetTree() const
    {
        return static_cast<const TNameTree&>(mustGetNameTree(getType<TNameTree>()));
    }
 
    template <typename TNameTree>
    TNameTree& GetOrCreateTree()
    {
        return static_cast<TNameTree&>(getOrCreateNameTree(getType<TNameTree>()));
    }
 
public:
    const PdfObject* GetValue(PdfKnownNameTree tree, const std::string_view& key) const override;
    PdfObject* GetValue(PdfKnownNameTree tree, const std::string_view& key) override;
    bool HasKey(PdfKnownNameTree tree, const std::string_view& key) const override;
    void ToDictionary(PdfKnownNameTree tree, PdfStringMap<PdfObject>& dict, bool skipClear = false) const override;
 
private:
    void AddValue(PdfKnownNameTree tree, const PdfString& key, const PdfObject& value) override;
    void AddValue(const PdfName& treeName, const PdfString& key, const PdfObject& value) override;
    const PdfObject* GetValue(const std::string_view& treeName, const std::string_view& key) const override;
    PdfObject* GetValue(const std::string_view& treeName, const std::string_view& key) override;
    bool HasKey(const std::string_view& treeName, const std::string_view& key) const override;
    void ToDictionary(const std::string_view& treeName, PdfStringMap<PdfObject>& dict, bool skipClear = false) const override;
 
private:
    PdfObject* getRootNode(const std::string_view& treeName) const;
    PdfObject& getOrCreateRootNode(const PdfName& treeName);
 
    PdfObject* getValue(const std::string_view& name, const std::string_view& key) const;
 
    PdfNameTreeBase* getNameTree(PdfKnownNameTree tree) const;
 
    PdfNameTreeBase& mustGetNameTree(PdfKnownNameTree tree) const;
 
    PdfNameTreeBase& getOrCreateNameTree(PdfKnownNameTree tree);
 
    template <typename TNameTree>
    static constexpr PdfKnownNameTree getType()
    {
        if (std::is_same_v<TNameTree, PdfEmbeddedFiles>)
            return PdfKnownNameTree::EmbeddedFiles;
        else if (std::is_same_v<TNameTree, PdfDestinations>)
            return PdfKnownNameTree::Dests;
        else
            return PdfKnownNameTree::Unknown;
    }
 
private:
    nullable<std::unique_ptr<PdfNameTreeBase>> m_Trees[(unsigned)PdfKnownNameTree::Renditions];
};
 
};
 
#endif // PDF_NAME_TREES_H