1 files changed, 106 insertions, 0 deletions
diff --git a/core/fpdfdoc/tagged_int.h b/core/fpdfdoc/tagged_int.h
new file mode 100644
index 0000000000..e930f613b1
--- /dev/null
+++ b/core/fpdfdoc/tagged_int.h
@@ -0,0 +1,106 @@
+// 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 "core/include/fpdfdoc/fpdf_tagged.h"
+
+class CPDF_StructElementImpl;
+
+class CPDF_StructTreeImpl : public CPDF_StructTree {
+ public:
+  explicit CPDF_StructTreeImpl(const CPDF_Document* pDoc);
+  ~CPDF_StructTreeImpl() override;
+
+  // CPDF_StructTree
+  int CountTopElements() const override { return m_Kids.GetSize(); }
+  CPDF_StructElement* GetTopElement(int i) const override {
+    return (CPDF_StructElement*)m_Kids.GetAt(i);
+  }
+
+  void LoadDocTree();
+  void LoadPageTree(const CPDF_Dictionary* pPageDict);
+  CPDF_StructElementImpl* AddPageNode(
+      CPDF_Dictionary* pElement,
+      std::map<CPDF_Dictionary*, CPDF_StructElementImpl*>& map,
+      int nLevel = 0);
+  FX_BOOL AddTopLevelNode(CPDF_Dictionary* pDict,
+                          CPDF_StructElementImpl* pElement);
+
+ protected:
+  const CPDF_Dictionary* m_pTreeRoot;
+  const CPDF_Dictionary* m_pRoleMap;
+  const CPDF_Dictionary* m_pPage;
+  CFX_ArrayTemplate<CPDF_StructElementImpl*> m_Kids;
+  friend class CPDF_StructElementImpl;
+};
+
+class CPDF_StructElementImpl final : public CPDF_StructElement {
+ public:
+  CPDF_StructElementImpl(CPDF_StructTreeImpl* pTree,
+                         CPDF_StructElementImpl* pParent,
+                         CPDF_Dictionary* pDict);
+
+  // CPDF_StructTreeImpl
+  CPDF_StructTree* GetTree() const override { return m_pTree; }
+  const CFX_ByteString& GetType() const override { return m_Type; }
+  CPDF_StructElement* GetParent() const override { return m_pParent; }
+  CPDF_Dictionary* GetDict() const override { return m_pDict; }
+  int CountKids() const override { return m_Kids.GetSize(); }
+  const CPDF_StructKid& GetKid(int index) const override {
+    return m_Kids.GetData()[index];
+  }
+  CPDF_Object* GetAttr(const CFX_ByteStringC& owner,
+                       const CFX_ByteStringC& name,
+                       FX_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,
+                         FX_BOOL bInheritable = FALSE,
+                         int subindex = -1) override;
+  FX_ARGB GetColor(const CFX_ByteStringC& owner,
+                   const CFX_ByteStringC& name,
+                   FX_ARGB default_value,
+                   FX_BOOL bInheritable = FALSE,
+                   int subindex = -1) override;
+  FX_FLOAT GetNumber(const CFX_ByteStringC& owner,
+                     const CFX_ByteStringC& name,
+                     FX_FLOAT default_value,
+                     FX_BOOL bInheritable = FALSE,
+                     int subindex = -1) override;
+  int GetInteger(const CFX_ByteStringC& owner,
+                 const CFX_ByteStringC& name,
+                 int default_value,
+                 FX_BOOL bInheritable = FALSE,
+                 int subindex = -1) override;
+
+  void LoadKids(CPDF_Dictionary* pDict);
+  void LoadKid(FX_DWORD PageObjNum, CPDF_Object* pObj, CPDF_StructKid* pKid);
+  CPDF_Object* GetAttr(const CFX_ByteStringC& owner,
+                       const CFX_ByteStringC& name,
+                       FX_BOOL bInheritable,
+                       int subindex);
+  CPDF_StructElementImpl* Retain();
+  void Release();
+
+ protected:
+  ~CPDF_StructElementImpl() override;
+
+  CPDF_StructTreeImpl* m_pTree;
+  CFX_ByteString m_Type;
+  CPDF_StructElementImpl* m_pParent;
+  CPDF_Dictionary* m_pDict;
+  CFX_ArrayTemplate<CPDF_StructKid> m_Kids;
+  int m_RefCount;
+
+  friend class CPDF_StructTreeImpl;
+};
+
+#endif  // CORE_FPDFDOC_TAGGED_INT_H_