7cf368083c96cacc62cfea6b6edbf294b3c60ac9
[pdfium.git] / core / src / fpdfdoc / tagged_int.h
1 // Copyright 2014 PDFium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
7 #ifndef CORE_SRC_FPDFDOC_TAGGED_INT_H_
8 #define CORE_SRC_FPDFDOC_TAGGED_INT_H_
9
10 #include "../../include/fpdfdoc/fpdf_tagged.h"
11
12 class CPDF_StructElementImpl;
13 class CPDF_StructTreeImpl : public CPDF_StructTree {
14  public:
15   CPDF_StructTreeImpl(const CPDF_Document* pDoc);
16   ~CPDF_StructTreeImpl();
17   int CountTopElements() const { return m_Kids.GetSize(); }
18   CPDF_StructElement* GetTopElement(int i) const {
19     return (CPDF_StructElement*)m_Kids.GetAt(i);
20   }
21   void LoadDocTree();
22   void LoadPageTree(const CPDF_Dictionary* pPageDict);
23   CPDF_StructElementImpl* AddPageNode(CPDF_Dictionary* pElement,
24                                       CFX_MapPtrToPtr& map,
25                                       int nLevel = 0);
26   FX_BOOL AddTopLevelNode(CPDF_Dictionary* pDict,
27                           CPDF_StructElementImpl* pElement);
28
29  protected:
30   const CPDF_Dictionary* m_pTreeRoot;
31   const CPDF_Dictionary* m_pRoleMap;
32   const CPDF_Dictionary* m_pPage;
33   CFX_ArrayTemplate<CPDF_StructElementImpl*> m_Kids;
34   friend class CPDF_StructElementImpl;
35 };
36 class CPDF_StructElementImpl final : public CPDF_StructElement {
37  public:
38   CPDF_StructElementImpl(CPDF_StructTreeImpl* pTree,
39                          CPDF_StructElementImpl* pParent,
40                          CPDF_Dictionary* pDict);
41   CPDF_StructTree* GetTree() const { return m_pTree; }
42   const CFX_ByteString& GetType() const { return m_Type; }
43   CPDF_StructElement* GetParent() const { return m_pParent; }
44   CPDF_Dictionary* GetDict() const { return m_pDict; }
45   int CountKids() const { return m_Kids.GetSize(); }
46   const CPDF_StructKid& GetKid(int index) const {
47     return m_Kids.GetData()[index];
48   }
49   CFX_PtrArray* GetObjectArray() { return &m_ObjectArray; }
50
51   CPDF_Object* GetAttr(const CFX_ByteStringC& owner,
52                        const CFX_ByteStringC& name,
53                        FX_BOOL bInheritable = FALSE,
54                        FX_FLOAT fLevel = 0.0F);
55
56   CFX_ByteString GetName(const CFX_ByteStringC& owner,
57                          const CFX_ByteStringC& name,
58                          const CFX_ByteStringC& default_value,
59                          FX_BOOL bInheritable = FALSE,
60                          int subindex = -1);
61   FX_ARGB GetColor(const CFX_ByteStringC& owner,
62                    const CFX_ByteStringC& name,
63                    FX_ARGB default_value,
64                    FX_BOOL bInheritable = FALSE,
65                    int subindex = -1);
66   FX_FLOAT GetNumber(const CFX_ByteStringC& owner,
67                      const CFX_ByteStringC& name,
68                      FX_FLOAT default_value,
69                      FX_BOOL bInheritable = FALSE,
70                      int subindex = -1);
71   int GetInteger(const CFX_ByteStringC& owner,
72                  const CFX_ByteStringC& name,
73                  int default_value,
74                  FX_BOOL bInheritable = FALSE,
75                  int subindex = -1);
76   CFX_PtrArray m_ObjectArray;
77   void LoadKids(CPDF_Dictionary* pDict);
78   void LoadKid(FX_DWORD PageObjNum, CPDF_Object* pObj, CPDF_StructKid* pKid);
79   CPDF_Object* GetAttr(const CFX_ByteStringC& owner,
80                        const CFX_ByteStringC& name,
81                        FX_BOOL bInheritable,
82                        int subindex);
83   CPDF_StructElementImpl* Retain();
84   void Release();
85
86  protected:
87   ~CPDF_StructElementImpl();
88
89   CPDF_StructTreeImpl* m_pTree;
90   CFX_ByteString m_Type;
91   CPDF_StructElementImpl* m_pParent;
92   CPDF_Dictionary* m_pDict;
93   CFX_ArrayTemplate<CPDF_StructKid> m_Kids;
94   int m_RefCount;
95
96   friend class CPDF_StructTreeImpl;
97 };
98
99 #endif  // CORE_SRC_FPDFDOC_TAGGED_INT_H_