| // Copyright 2014 PDFium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
| |
| #ifndef CORE_FPDFDOC_TAGGED_INT_H_ |
| #define CORE_FPDFDOC_TAGGED_INT_H_ |
| |
| #include <map> |
| #include <memory> |
| #include <vector> |
| |
| #include "core/fpdfdoc/fpdf_tagged.h" |
| #include "core/fxcrt/cfx_retain_ptr.h" |
| #include "third_party/base/stl_util.h" |
| |
| class CPDF_StructElement; |
| |
| struct CPDF_StructKid { |
| CPDF_StructKid(); |
| CPDF_StructKid(const CPDF_StructKid& that); |
| ~CPDF_StructKid(); |
| |
| enum { Invalid, Element, PageContent, StreamContent, Object } m_Type; |
| |
| CFX_RetainPtr<CPDF_StructElement> m_pElement; // For Element. |
| CPDF_Dictionary* m_pDict; // For Element. |
| uint32_t m_PageObjNum; // For PageContent, StreamContent, Object. |
| uint32_t m_RefObjNum; // For StreamContent, Object. |
| uint32_t m_ContentId; // For PageContent, StreamContent. |
| }; |
| |
| class CPDF_StructTree final : public IPDF_StructTree { |
| public: |
| explicit CPDF_StructTree(const CPDF_Document* pDoc); |
| ~CPDF_StructTree() override; |
| |
| // IPDF_StructTree: |
| int CountTopElements() const override; |
| IPDF_StructElement* GetTopElement(int i) const override; |
| |
| void LoadPageTree(const CPDF_Dictionary* pPageDict); |
| CFX_RetainPtr<CPDF_StructElement> AddPageNode( |
| CPDF_Dictionary* pElement, |
| std::map<CPDF_Dictionary*, CFX_RetainPtr<CPDF_StructElement>>* map, |
| int nLevel = 0); |
| bool AddTopLevelNode(CPDF_Dictionary* pDict, |
| const CFX_RetainPtr<CPDF_StructElement>& pElement); |
| |
| protected: |
| const CPDF_Dictionary* const m_pTreeRoot; |
| const CPDF_Dictionary* const m_pRoleMap; |
| const CPDF_Dictionary* m_pPage; |
| std::vector<CFX_RetainPtr<CPDF_StructElement>> m_Kids; |
| |
| friend class CPDF_StructElement; |
| }; |
| |
| class CPDF_StructElement final : public CFX_Retainable, |
| public IPDF_StructElement { |
| public: |
| template <typename T, typename... Args> |
| friend CFX_RetainPtr<T> pdfium::MakeRetain(Args&&... args); |
| |
| // IPDF_StructElement |
| IPDF_StructTree* GetTree() const override; |
| const CFX_ByteString& GetType() const override; |
| IPDF_StructElement* GetParent() const override; |
| CPDF_Dictionary* GetDict() const override; |
| int CountKids() const override; |
| IPDF_StructElement* GetKidIfElement(int index) const override; |
| CPDF_Object* GetAttr(const CFX_ByteStringC& owner, |
| const CFX_ByteStringC& name, |
| bool bInheritable = false, |
| float fLevel = 0.0F) override; |
| CFX_ByteString GetName(const CFX_ByteStringC& owner, |
| const CFX_ByteStringC& name, |
| const CFX_ByteStringC& default_value, |
| bool bInheritable = false, |
| int subindex = -1) override; |
| FX_ARGB GetColor(const CFX_ByteStringC& owner, |
| const CFX_ByteStringC& name, |
| FX_ARGB default_value, |
| bool bInheritable = false, |
| int subindex = -1) override; |
| float GetNumber(const CFX_ByteStringC& owner, |
| const CFX_ByteStringC& name, |
| float default_value, |
| bool bInheritable = false, |
| int subindex = -1) override; |
| int GetInteger(const CFX_ByteStringC& owner, |
| const CFX_ByteStringC& name, |
| int default_value, |
| bool bInheritable = false, |
| int subindex = -1) override; |
| |
| std::vector<CPDF_StructKid>* GetKids() { return &m_Kids; } |
| void LoadKids(CPDF_Dictionary* pDict); |
| void LoadKid(uint32_t PageObjNum, CPDF_Object* pObj, CPDF_StructKid* pKid); |
| CPDF_Object* GetAttr(const CFX_ByteStringC& owner, |
| const CFX_ByteStringC& name, |
| bool bInheritable, |
| int subindex); |
| |
| private: |
| CPDF_StructElement(CPDF_StructTree* pTree, |
| CPDF_StructElement* pParent, |
| CPDF_Dictionary* pDict); |
| ~CPDF_StructElement() override; |
| |
| CPDF_StructTree* const m_pTree; |
| CPDF_StructElement* const m_pParent; |
| CPDF_Dictionary* const m_pDict; |
| CFX_ByteString m_Type; |
| std::vector<CPDF_StructKid> m_Kids; |
| }; |
| |
| #endif // CORE_FPDFDOC_TAGGED_INT_H_ |