| // Copyright 2017 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_CPDF_STRUCTTREE_H_ |
| #define CORE_FPDFDOC_CPDF_STRUCTTREE_H_ |
| |
| #include <map> |
| #include <memory> |
| #include <vector> |
| |
| #include "core/fxcrt/retain_ptr.h" |
| #include "core/fxcrt/unowned_ptr.h" |
| |
| class CPDF_Dictionary; |
| class CPDF_Document; |
| class CPDF_StructElement; |
| |
| class CPDF_StructTree { |
| public: |
| static std::unique_ptr<CPDF_StructTree> LoadPage( |
| const CPDF_Document* pDoc, |
| const CPDF_Dictionary* pPageDict); |
| |
| explicit CPDF_StructTree(const CPDF_Document* pDoc); |
| ~CPDF_StructTree(); |
| |
| size_t CountTopElements() const { return m_Kids.size(); } |
| CPDF_StructElement* GetTopElement(size_t i) const { return m_Kids[i].Get(); } |
| const CPDF_Dictionary* GetRoleMap() const { return m_pRoleMap.Get(); } |
| const CPDF_Dictionary* GetPage() const { return m_pPage.Get(); } |
| const CPDF_Dictionary* GetTreeRoot() const { return m_pTreeRoot.Get(); } |
| |
| private: |
| using StructElementMap = |
| std::map<const CPDF_Dictionary*, RetainPtr<CPDF_StructElement>>; |
| |
| void LoadPageTree(const CPDF_Dictionary* pPageDict); |
| RetainPtr<CPDF_StructElement> AddPageNode(const CPDF_Dictionary* pElement, |
| StructElementMap* map, |
| int nLevel); |
| bool AddTopLevelNode(const CPDF_Dictionary* pDict, |
| const RetainPtr<CPDF_StructElement>& pElement); |
| |
| UnownedPtr<const CPDF_Dictionary> const m_pTreeRoot; |
| UnownedPtr<const CPDF_Dictionary> const m_pRoleMap; |
| UnownedPtr<const CPDF_Dictionary> m_pPage; |
| std::vector<RetainPtr<CPDF_StructElement>> m_Kids; |
| }; |
| |
| #endif // CORE_FPDFDOC_CPDF_STRUCTTREE_H_ |