Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(18)

Side by Side Diff: core/fpdfapi/fpdf_parser/cpdf_array.cpp

Issue 1867183002: Use std::vector as internal storage for CPDF_Array (Closed) Base URL: https://pdfium.googlesource.com/pdfium.git@master
Patch Set: Created 4 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698