Chromium Code Reviews| 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 |