blob: 68b301e283b76cd44e6c1f0bc6ebd869b7c8d865 [file] [log] [blame]
// Copyright 2016 PDFium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
#include "core/fpdfapi/parser/cpdf_array.h"
#include <set>
#include <utility>
#include "core/fpdfapi/parser/cpdf_name.h"
#include "core/fpdfapi/parser/cpdf_number.h"
#include "core/fpdfapi/parser/cpdf_reference.h"
#include "core/fpdfapi/parser/cpdf_stream.h"
#include "core/fpdfapi/parser/cpdf_string.h"
#include "core/fxcrt/fx_stream.h"
#include "third_party/base/logging.h"
#include "third_party/base/stl_util.h"
CPDF_Array::CPDF_Array() {}
CPDF_Array::CPDF_Array(const WeakPtr<ByteStringPool>& pPool) : m_pPool(pPool) {}
CPDF_Array::~CPDF_Array() {
// Break cycles for cyclic references.
m_ObjNum = kInvalidObjNum;
for (auto& it : m_Objects) {
if (it && it->GetObjNum() == kInvalidObjNum)
it.release();
}
}
CPDF_Object::Type CPDF_Array::GetType() const {
return ARRAY;
}
bool CPDF_Array::IsArray() const {
return true;
}
CPDF_Array* CPDF_Array::AsArray() {
return this;
}
const CPDF_Array* CPDF_Array::AsArray() const {
return this;
}
std::unique_ptr<CPDF_Object> CPDF_Array::Clone() const {
return CloneObjectNonCyclic(false);
}
std::unique_ptr<CPDF_Object> CPDF_Array::CloneNonCyclic(
bool bDirect,
std::set<const CPDF_Object*>* pVisited) const {
pVisited->insert(this);
auto pCopy = pdfium::MakeUnique<CPDF_Array>();
for (const auto& pValue : m_Objects) {
if (!pdfium::ContainsKey(*pVisited, pValue.get())) {
std::set<const CPDF_Object*> visited(*pVisited);
if (auto obj = pValue->CloneNonCyclic(bDirect, &visited))
pCopy->m_Objects.push_back(std::move(obj));
}
}
return std::move(pCopy);
}
CFX_FloatRect CPDF_Array::GetRect() const {
CFX_FloatRect rect;
if (!IsArray() || m_Objects.size() != 4)
return rect;
rect.left = GetNumberAt(0);
rect.bottom = GetNumberAt(1);
rect.right = GetNumberAt(2);
rect.top = GetNumberAt(3);
return rect;
}
CFX_Matrix CPDF_Array::GetMatrix() const {
CFX_Matrix matrix;
if (!IsArray() || m_Objects.size() != 6)
return CFX_Matrix();
return CFX_Matrix(GetNumberAt(0), GetNumberAt(1), GetNumberAt(2),
GetNumberAt(3), GetNumberAt(4), GetNumberAt(5));
}
CPDF_Object* CPDF_Array::GetObjectAt(size_t i) {
if (i >= m_Objects.size())
return nullptr;
return m_Objects[i].get();
}
const CPDF_Object* CPDF_Array::GetObjectAt(size_t i) const {
if (i >= m_Objects.size())
return nullptr;
return m_Objects[i].get();
}
CPDF_Object* CPDF_Array::GetDirectObjectAt(size_t i) {
if (i >= m_Objects.size())
return nullptr;
return m_Objects[i]->GetDirect();
}
const CPDF_Object* CPDF_Array::GetDirectObjectAt(size_t i) const {
if (i >= m_Objects.size())
return nullptr;
return m_Objects[i]->GetDirect();
}
ByteString CPDF_Array::GetStringAt(size_t i) const {
if (i >= m_Objects.size())
return ByteString();
return m_Objects[i]->GetString();
}
WideString CPDF_Array::GetUnicodeTextAt(size_t i) const {
if (i >= m_Objects.size())
return WideString();
return m_Objects[i]->GetUnicodeText();
}
int CPDF_Array::GetIntegerAt(size_t i) const {
if (i >= m_Objects.size())
return 0;
return m_Objects[i]->GetInteger();
}
float CPDF_Array::GetNumberAt(size_t i) const {
if (i >= m_Objects.size())
return 0;
return m_Objects[i]->GetNumber();
}
CPDF_Dictionary* CPDF_Array::GetDictAt(size_t i) {
CPDF_Object* p = GetDirectObjectAt(i);
if (!p)
return nullptr;
if (CPDF_Dictionary* pDict = p->AsDictionary())
return pDict;
if (CPDF_Stream* pStream = p->AsStream())
return pStream->GetDict();
return nullptr;
}
const CPDF_Dictionary* CPDF_Array::GetDictAt(size_t i) const {
const CPDF_Object* p = GetDirectObjectAt(i);
if (!p)
return nullptr;
if (const CPDF_Dictionary* pDict = p->AsDictionary())
return pDict;
if (const CPDF_Stream* pStream = p->AsStream())
return pStream->GetDict();
return nullptr;
}
CPDF_Stream* CPDF_Array::GetStreamAt(size_t i) {
return ToStream(GetDirectObjectAt(i));
}
const CPDF_Stream* CPDF_Array::GetStreamAt(size_t i) const {
return ToStream(GetDirectObjectAt(i));
}
CPDF_Array* CPDF_Array::GetArrayAt(size_t i) {
return ToArray(GetDirectObjectAt(i));
}
const CPDF_Array* CPDF_Array::GetArrayAt(size_t i) const {
return ToArray(GetDirectObjectAt(i));
}
void CPDF_Array::Clear() {
m_Objects.clear();
}
void CPDF_Array::RemoveAt(size_t i) {
if (i < m_Objects.size())
m_Objects.erase(m_Objects.begin() + i);
}
void CPDF_Array::ConvertToIndirectObjectAt(size_t i,
CPDF_IndirectObjectHolder* pHolder) {
if (i >= m_Objects.size())
return;
if (!m_Objects[i] || m_Objects[i]->IsReference())
return;
CPDF_Object* pNew = pHolder->AddIndirectObject(std::move(m_Objects[i]));
m_Objects[i] = pNew->MakeReference(pHolder);
}
CPDF_Object* CPDF_Array::SetAt(size_t i, std::unique_ptr<CPDF_Object> pObj) {
ASSERT(IsArray());
ASSERT(!pObj || pObj->IsInline());
if (i >= m_Objects.size()) {
NOTREACHED();
return nullptr;
}
CPDF_Object* pRet = pObj.get();
m_Objects[i] = std::move(pObj);
return pRet;
}
CPDF_Object* CPDF_Array::InsertAt(size_t index,
std::unique_ptr<CPDF_Object> pObj) {
ASSERT(IsArray());
CHECK(!pObj || pObj->IsInline());
CPDF_Object* pRet = pObj.get();
if (index >= m_Objects.size()) {
// Allocate space first.
m_Objects.resize(index + 1);
m_Objects[index] = std::move(pObj);
} else {
// Directly insert.
m_Objects.insert(m_Objects.begin() + index, std::move(pObj));
}
return pRet;
}
CPDF_Object* CPDF_Array::Add(std::unique_ptr<CPDF_Object> pObj) {
ASSERT(IsArray());
CHECK(!pObj || pObj->IsInline());
CPDF_Object* pRet = pObj.get();
m_Objects.push_back(std::move(pObj));
return pRet;
}
bool CPDF_Array::WriteTo(IFX_ArchiveStream* archive,
const CPDF_Encryptor* encryptor) const {
if (!archive->WriteString("["))
return false;
for (size_t i = 0; i < GetCount(); ++i) {
if (!GetObjectAt(i)->WriteTo(archive, encryptor))
return false;
}
return archive->WriteString("]");
}