OLD | NEW |
---|---|
1 // Copyright 2016 PDFium Authors. All rights reserved. | 1 // Copyright 2016 PDFium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com | 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
6 | 6 |
7 #include "core/fpdfapi/fpdf_parser/include/cpdf_array.h" | 7 #include "core/fpdfapi/fpdf_parser/include/cpdf_array.h" |
8 | 8 |
9 #include "core/fpdfapi/fpdf_parser/include/cpdf_name.h" | 9 #include "core/fpdfapi/fpdf_parser/include/cpdf_name.h" |
10 #include "core/fpdfapi/fpdf_parser/include/cpdf_number.h" | 10 #include "core/fpdfapi/fpdf_parser/include/cpdf_number.h" |
11 #include "core/fpdfapi/fpdf_parser/include/cpdf_reference.h" | 11 #include "core/fpdfapi/fpdf_parser/include/cpdf_reference.h" |
12 #include "core/fpdfapi/fpdf_parser/include/cpdf_stream.h" | 12 #include "core/fpdfapi/fpdf_parser/include/cpdf_stream.h" |
13 #include "core/fpdfapi/fpdf_parser/include/cpdf_string.h" | 13 #include "core/fpdfapi/fpdf_parser/include/cpdf_string.h" |
14 | 14 |
15 CPDF_Array::CPDF_Array() {} | 15 CPDF_Array::CPDF_Array() {} |
16 | 16 |
17 CPDF_Array::~CPDF_Array() { | 17 CPDF_Array::~CPDF_Array() { |
18 int size = m_Objects.GetSize(); | 18 for (auto& it : m_Objects) { |
Tom Sepez
2016/04/08 20:13:53
Now that this is a STL array, we can use
std::vec
| |
19 CPDF_Object** pList = m_Objects.GetData(); | 19 if (it) |
20 for (int i = 0; i < size; i++) { | 20 it->Release(); |
21 if (pList[i]) | |
22 pList[i]->Release(); | |
23 } | 21 } |
24 } | 22 } |
25 | 23 |
26 CPDF_Object::Type CPDF_Array::GetType() const { | 24 CPDF_Object::Type CPDF_Array::GetType() const { |
27 return ARRAY; | 25 return ARRAY; |
28 } | 26 } |
29 | 27 |
30 CPDF_Array* CPDF_Array::GetArray() const { | 28 CPDF_Array* CPDF_Array::GetArray() const { |
31 // The method should be made non-const if we want to not be const. | 29 // The method should be made non-const if we want to not be const. |
32 // See bug #234. | 30 // See bug #234. |
33 return const_cast<CPDF_Array*>(this); | 31 return const_cast<CPDF_Array*>(this); |
34 } | 32 } |
35 | 33 |
36 bool CPDF_Array::IsArray() const { | 34 bool CPDF_Array::IsArray() const { |
37 return true; | 35 return true; |
38 } | 36 } |
39 | 37 |
40 CPDF_Array* CPDF_Array::AsArray() { | 38 CPDF_Array* CPDF_Array::AsArray() { |
41 return this; | 39 return this; |
42 } | 40 } |
43 | 41 |
44 const CPDF_Array* CPDF_Array::AsArray() const { | 42 const CPDF_Array* CPDF_Array::AsArray() const { |
45 return this; | 43 return this; |
46 } | 44 } |
47 | 45 |
48 CPDF_Object* CPDF_Array::Clone(FX_BOOL bDirect) const { | 46 CPDF_Object* CPDF_Array::Clone(FX_BOOL bDirect) const { |
49 CPDF_Array* pCopy = new CPDF_Array(); | 47 CPDF_Array* pCopy = new CPDF_Array(); |
50 for (size_t i = 0; i < GetCount(); i++) { | 48 for (size_t i = 0; i < GetCount(); i++) { |
51 CPDF_Object* value = m_Objects.GetAt(i); | 49 CPDF_Object* value = m_Objects.at(i); |
52 pCopy->m_Objects.Add(value->Clone(bDirect)); | 50 pCopy->m_Objects.push_back(value->Clone(bDirect)); |
53 } | 51 } |
54 return pCopy; | 52 return pCopy; |
55 } | 53 } |
56 | 54 |
57 CFX_FloatRect CPDF_Array::GetRect() { | 55 CFX_FloatRect CPDF_Array::GetRect() { |
58 CFX_FloatRect rect; | 56 CFX_FloatRect rect; |
59 if (!IsArray() || m_Objects.GetSize() != 4) | 57 if (!IsArray() || m_Objects.size() != 4) |
60 return rect; | 58 return rect; |
61 | 59 |
62 rect.left = GetNumberAt(0); | 60 rect.left = GetNumberAt(0); |
63 rect.bottom = GetNumberAt(1); | 61 rect.bottom = GetNumberAt(1); |
64 rect.right = GetNumberAt(2); | 62 rect.right = GetNumberAt(2); |
65 rect.top = GetNumberAt(3); | 63 rect.top = GetNumberAt(3); |
66 return rect; | 64 return rect; |
67 } | 65 } |
68 | 66 |
69 CFX_Matrix CPDF_Array::GetMatrix() { | 67 CFX_Matrix CPDF_Array::GetMatrix() { |
70 CFX_Matrix matrix; | 68 CFX_Matrix matrix; |
71 if (!IsArray() || m_Objects.GetSize() != 6) | 69 if (!IsArray() || m_Objects.size() != 6) |
72 return matrix; | 70 return matrix; |
73 | 71 |
74 matrix.Set(GetNumberAt(0), GetNumberAt(1), GetNumberAt(2), GetNumberAt(3), | 72 matrix.Set(GetNumberAt(0), GetNumberAt(1), GetNumberAt(2), GetNumberAt(3), |
75 GetNumberAt(4), GetNumberAt(5)); | 73 GetNumberAt(4), GetNumberAt(5)); |
76 return matrix; | 74 return matrix; |
77 } | 75 } |
78 | 76 |
79 CPDF_Object* CPDF_Array::GetObjectAt(uint32_t i) const { | 77 CPDF_Object* CPDF_Array::GetObjectAt(size_t i) const { |
80 if (i >= (uint32_t)m_Objects.GetSize()) | 78 if (i >= m_Objects.size()) |
81 return nullptr; | 79 return nullptr; |
82 return m_Objects.GetAt(i); | 80 return m_Objects.at(i); |
83 } | 81 } |
84 | 82 |
85 CPDF_Object* CPDF_Array::GetDirectObjectAt(uint32_t i) const { | 83 CPDF_Object* CPDF_Array::GetDirectObjectAt(size_t i) const { |
86 if (i >= (uint32_t)m_Objects.GetSize()) | 84 if (i >= m_Objects.size()) |
87 return nullptr; | 85 return nullptr; |
88 return m_Objects.GetAt(i)->GetDirect(); | 86 return m_Objects.at(i)->GetDirect(); |
89 } | 87 } |
90 | 88 |
91 CFX_ByteString CPDF_Array::GetStringAt(uint32_t i) const { | 89 CFX_ByteString CPDF_Array::GetStringAt(size_t i) const { |
92 if (i >= (uint32_t)m_Objects.GetSize()) | 90 if (i >= m_Objects.size()) |
93 return CFX_ByteString(); | 91 return CFX_ByteString(); |
94 return m_Objects.GetAt(i)->GetString(); | 92 return m_Objects.at(i)->GetString(); |
95 } | 93 } |
96 | 94 |
97 CFX_ByteStringC CPDF_Array::GetConstStringAt(uint32_t i) const { | 95 CFX_ByteStringC CPDF_Array::GetConstStringAt(size_t i) const { |
98 if (i >= (uint32_t)m_Objects.GetSize()) | 96 if (i >= m_Objects.size()) |
99 return CFX_ByteStringC(); | 97 return CFX_ByteStringC(); |
100 return m_Objects.GetAt(i)->GetConstString(); | 98 return m_Objects.at(i)->GetConstString(); |
101 } | 99 } |
102 | 100 |
103 int CPDF_Array::GetIntegerAt(uint32_t i) const { | 101 int CPDF_Array::GetIntegerAt(size_t i) const { |
104 if (i >= (uint32_t)m_Objects.GetSize()) | 102 if (i >= m_Objects.size()) |
105 return 0; | 103 return 0; |
106 return m_Objects.GetAt(i)->GetInteger(); | 104 return m_Objects.at(i)->GetInteger(); |
107 } | 105 } |
108 | 106 |
109 FX_FLOAT CPDF_Array::GetNumberAt(uint32_t i) const { | 107 FX_FLOAT CPDF_Array::GetNumberAt(size_t i) const { |
110 if (i >= (uint32_t)m_Objects.GetSize()) | 108 if (i >= m_Objects.size()) |
111 return 0; | 109 return 0; |
112 return m_Objects.GetAt(i)->GetNumber(); | 110 return m_Objects.at(i)->GetNumber(); |
113 } | 111 } |
114 | 112 |
115 CPDF_Dictionary* CPDF_Array::GetDictAt(uint32_t i) const { | 113 CPDF_Dictionary* CPDF_Array::GetDictAt(size_t i) const { |
116 CPDF_Object* p = GetDirectObjectAt(i); | 114 CPDF_Object* p = GetDirectObjectAt(i); |
117 if (!p) | 115 if (!p) |
118 return NULL; | 116 return NULL; |
119 if (CPDF_Dictionary* pDict = p->AsDictionary()) | 117 if (CPDF_Dictionary* pDict = p->AsDictionary()) |
120 return pDict; | 118 return pDict; |
121 if (CPDF_Stream* pStream = p->AsStream()) | 119 if (CPDF_Stream* pStream = p->AsStream()) |
122 return pStream->GetDict(); | 120 return pStream->GetDict(); |
123 return NULL; | 121 return NULL; |
124 } | 122 } |
125 | 123 |
126 CPDF_Stream* CPDF_Array::GetStreamAt(uint32_t i) const { | 124 CPDF_Stream* CPDF_Array::GetStreamAt(size_t i) const { |
127 return ToStream(GetDirectObjectAt(i)); | 125 return ToStream(GetDirectObjectAt(i)); |
128 } | 126 } |
129 | 127 |
130 CPDF_Array* CPDF_Array::GetArrayAt(uint32_t i) const { | 128 CPDF_Array* CPDF_Array::GetArrayAt(size_t i) const { |
131 return ToArray(GetDirectObjectAt(i)); | 129 return ToArray(GetDirectObjectAt(i)); |
132 } | 130 } |
133 | 131 |
134 void CPDF_Array::RemoveAt(uint32_t i, uint32_t nCount) { | 132 void CPDF_Array::RemoveAt(size_t i, size_t nCount) { |
135 if (i >= (uint32_t)m_Objects.GetSize()) | 133 if (i >= m_Objects.size()) |
136 return; | 134 return; |
137 | 135 |
138 if (nCount <= 0 || nCount > m_Objects.GetSize() - i) | 136 if (nCount <= 0 || nCount > m_Objects.size() - i) |
139 return; | 137 return; |
140 | 138 |
141 for (uint32_t j = 0; j < nCount; ++j) { | 139 for (size_t j = 0; j < nCount; ++j) { |
142 if (CPDF_Object* p = m_Objects.GetAt(i + j)) | 140 if (CPDF_Object* p = m_Objects.at(i + j)) |
143 p->Release(); | 141 p->Release(); |
144 } | 142 } |
145 m_Objects.RemoveAt(i, nCount); | 143 m_Objects.erase(m_Objects.begin() + i, m_Objects.begin() + i + nCount); |
146 } | 144 } |
147 | 145 |
148 void CPDF_Array::SetAt(uint32_t i, | 146 void CPDF_Array::SetAt(size_t i, |
149 CPDF_Object* pObj, | 147 CPDF_Object* pObj, |
150 CPDF_IndirectObjectHolder* pObjs) { | 148 CPDF_IndirectObjectHolder* pObjs) { |
151 ASSERT(IsArray()); | 149 ASSERT(IsArray()); |
152 ASSERT(i < (uint32_t)m_Objects.GetSize()); | 150 ASSERT(i < m_Objects.size()); |
153 if (i >= (uint32_t)m_Objects.GetSize()) | 151 if (i >= m_Objects.size()) |
154 return; | 152 return; |
155 if (CPDF_Object* pOld = m_Objects.GetAt(i)) | 153 if (CPDF_Object* pOld = m_Objects.at(i)) |
156 pOld->Release(); | 154 pOld->Release(); |
157 if (pObj->GetObjNum()) { | 155 if (pObj->GetObjNum()) { |
158 ASSERT(pObjs); | 156 ASSERT(pObjs); |
159 pObj = new CPDF_Reference(pObjs, pObj->GetObjNum()); | 157 pObj = new CPDF_Reference(pObjs, pObj->GetObjNum()); |
160 } | 158 } |
161 m_Objects.SetAt(i, pObj); | 159 m_Objects[i] = pObj; |
162 } | 160 } |
163 | 161 |
164 void CPDF_Array::InsertAt(uint32_t index, | 162 void CPDF_Array::InsertAt(size_t index, |
165 CPDF_Object* pObj, | 163 CPDF_Object* pObj, |
166 CPDF_IndirectObjectHolder* pObjs) { | 164 CPDF_IndirectObjectHolder* pObjs) { |
167 if (pObj->GetObjNum()) { | 165 if (pObj->GetObjNum()) { |
168 ASSERT(pObjs); | 166 ASSERT(pObjs); |
169 pObj = new CPDF_Reference(pObjs, pObj->GetObjNum()); | 167 pObj = new CPDF_Reference(pObjs, pObj->GetObjNum()); |
170 } | 168 } |
171 m_Objects.InsertAt(index, pObj); | 169 if (index >= m_Objects.size()) { |
170 // Allocate space first. | |
171 m_Objects.resize(index + 1, nullptr); | |
172 m_Objects[index] = pObj; | |
173 } else { | |
174 // Directly insert. | |
175 m_Objects.insert(m_Objects.begin() + index, pObj); | |
176 } | |
172 } | 177 } |
173 | 178 |
174 void CPDF_Array::Add(CPDF_Object* pObj, CPDF_IndirectObjectHolder* pObjs) { | 179 void CPDF_Array::Add(CPDF_Object* pObj, CPDF_IndirectObjectHolder* pObjs) { |
175 if (pObj->GetObjNum()) { | 180 if (pObj->GetObjNum()) { |
176 ASSERT(pObjs); | 181 ASSERT(pObjs); |
177 pObj = new CPDF_Reference(pObjs, pObj->GetObjNum()); | 182 pObj = new CPDF_Reference(pObjs, pObj->GetObjNum()); |
178 } | 183 } |
179 m_Objects.Add(pObj); | 184 m_Objects.push_back(pObj); |
180 } | 185 } |
181 | 186 |
182 void CPDF_Array::AddName(const CFX_ByteString& str) { | 187 void CPDF_Array::AddName(const CFX_ByteString& str) { |
183 ASSERT(IsArray()); | 188 ASSERT(IsArray()); |
184 Add(new CPDF_Name(str)); | 189 Add(new CPDF_Name(str)); |
185 } | 190 } |
186 | 191 |
187 void CPDF_Array::AddString(const CFX_ByteString& str) { | 192 void CPDF_Array::AddString(const CFX_ByteString& str) { |
188 ASSERT(IsArray()); | 193 ASSERT(IsArray()); |
189 Add(new CPDF_String(str, FALSE)); | 194 Add(new CPDF_String(str, FALSE)); |
190 } | 195 } |
191 | 196 |
192 void CPDF_Array::AddInteger(int i) { | 197 void CPDF_Array::AddInteger(int i) { |
193 ASSERT(IsArray()); | 198 ASSERT(IsArray()); |
194 Add(new CPDF_Number(i)); | 199 Add(new CPDF_Number(i)); |
195 } | 200 } |
196 | 201 |
197 void CPDF_Array::AddNumber(FX_FLOAT f) { | 202 void CPDF_Array::AddNumber(FX_FLOAT f) { |
198 ASSERT(IsArray()); | 203 ASSERT(IsArray()); |
199 CPDF_Number* pNumber = new CPDF_Number(f); | 204 CPDF_Number* pNumber = new CPDF_Number(f); |
200 Add(pNumber); | 205 Add(pNumber); |
201 } | 206 } |
202 | 207 |
203 void CPDF_Array::AddReference(CPDF_IndirectObjectHolder* pDoc, | 208 void CPDF_Array::AddReference(CPDF_IndirectObjectHolder* pDoc, |
204 uint32_t objnum) { | 209 uint32_t objnum) { |
205 ASSERT(IsArray()); | 210 ASSERT(IsArray()); |
206 Add(new CPDF_Reference(pDoc, objnum)); | 211 Add(new CPDF_Reference(pDoc, objnum)); |
207 } | 212 } |
OLD | NEW |