| // 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_StructElementImpl; |
| |
| class CPDF_StructTreeImpl final : public IPDF_StructTree { |
| public: |
| explicit CPDF_StructTreeImpl(const CPDF_Document* pDoc); |
| ~CPDF_StructTreeImpl() override; |
| |
| // IPDF_StructTree: |
| int CountTopElements() const override; |
| IPDF_StructElement* GetTopElement(int i) const override; |
| |
| void LoadDocTree(); |
| void LoadPageTree(const CPDF_Dictionary* pPageDict); |
| CPDF_StructElementImpl* AddPageNode( |
| CPDF_Dictionary* pElement, |
| std::map<CPDF_Dictionary*, CPDF_StructElementImpl*>& map, |
| int nLevel = 0); |
| bool AddTopLevelNode(CPDF_Dictionary* pDict, |
| CPDF_StructElementImpl* pElement); |
| |
| protected: |
| const CPDF_Dictionary* const m_pTreeRoot; |
| const CPDF_Dictionary* const m_pRoleMap; |
| const CPDF_Dictionary* m_pPage; |
| std::vector<CFX_RetainPtr<CPDF_StructElementImpl>> m_Kids; |
| |
| friend class CPDF_StructElementImpl; |
| }; |
| |
| class CPDF_StructElementImpl final : public IPDF_StructElement { |
| public: |
| CPDF_StructElementImpl(CPDF_StructTreeImpl* pTree, |
| CPDF_StructElementImpl* pParent, |
| CPDF_Dictionary* pDict); |
| |
| // 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; |
| const CPDF_StructKid& GetKid(int index) const override; |
| CPDF_Object* GetAttr(const CFX_ByteStringC& owner, |
| const CFX_ByteStringC& name, |
| bool bInheritable = false, |
| FX_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; |
| FX_FLOAT GetNumber(const CFX_ByteStringC& owner, |
| const CFX_ByteStringC& name, |
| FX_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; |
| |
| 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); |
| CPDF_StructElementImpl* Retain(); |
| void Release(); |
| |
| protected: |
| ~CPDF_StructElementImpl() override; |
| |
| int m_RefCount; |
| CPDF_StructTreeImpl* const m_pTree; |
| CPDF_StructElementImpl* const m_pParent; |
| CPDF_Dictionary* const m_pDict; |
| CFX_ByteString m_Type; |
| std::vector<CPDF_StructKid> m_Kids; |
| |
| friend class CPDF_StructTreeImpl; |
| }; |
| |
| #endif // CORE_FPDFDOC_TAGGED_INT_H_ |