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