| 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/parser/cpdf_reference.h" | 7 #include "core/fpdfapi/parser/cpdf_reference.h" |
| 8 | 8 |
| 9 #include "core/fpdfapi/parser/cpdf_indirect_object_holder.h" | 9 #include "core/fpdfapi/parser/cpdf_indirect_object_holder.h" |
| 10 #include "third_party/base/ptr_util.h" |
| 10 #include "third_party/base/stl_util.h" | 11 #include "third_party/base/stl_util.h" |
| 11 | 12 |
| 12 CPDF_Reference::CPDF_Reference(CPDF_IndirectObjectHolder* pDoc, int objnum) | 13 CPDF_Reference::CPDF_Reference(CPDF_IndirectObjectHolder* pDoc, int objnum) |
| 13 : m_pObjList(pDoc), m_RefObjNum(objnum) {} | 14 : m_pObjList(pDoc), m_RefObjNum(objnum) {} |
| 14 | 15 |
| 15 CPDF_Reference::~CPDF_Reference() {} | 16 CPDF_Reference::~CPDF_Reference() {} |
| 16 | 17 |
| 17 CPDF_Object::Type CPDF_Reference::GetType() const { | 18 CPDF_Object::Type CPDF_Reference::GetType() const { |
| 18 return REFERENCE; | 19 return REFERENCE; |
| 19 } | 20 } |
| (...skipping 23 matching lines...) Expand all Loading... |
| 43 } | 44 } |
| 44 | 45 |
| 45 CPDF_Reference* CPDF_Reference::AsReference() { | 46 CPDF_Reference* CPDF_Reference::AsReference() { |
| 46 return this; | 47 return this; |
| 47 } | 48 } |
| 48 | 49 |
| 49 const CPDF_Reference* CPDF_Reference::AsReference() const { | 50 const CPDF_Reference* CPDF_Reference::AsReference() const { |
| 50 return this; | 51 return this; |
| 51 } | 52 } |
| 52 | 53 |
| 53 CPDF_Object* CPDF_Reference::Clone() const { | 54 std::unique_ptr<CPDF_Object> CPDF_Reference::Clone() const { |
| 54 return CloneObjectNonCyclic(false); | 55 return CloneObjectNonCyclic(false); |
| 55 } | 56 } |
| 56 | 57 |
| 57 CPDF_Object* CPDF_Reference::CloneNonCyclic( | 58 std::unique_ptr<CPDF_Object> CPDF_Reference::CloneNonCyclic( |
| 58 bool bDirect, | 59 bool bDirect, |
| 59 std::set<const CPDF_Object*>* pVisited) const { | 60 std::set<const CPDF_Object*>* pVisited) const { |
| 60 pVisited->insert(this); | 61 pVisited->insert(this); |
| 61 if (bDirect) { | 62 if (bDirect) { |
| 62 auto* pDirect = GetDirect(); | 63 auto* pDirect = GetDirect(); |
| 63 return pDirect && !pdfium::ContainsKey(*pVisited, pDirect) | 64 return pDirect && !pdfium::ContainsKey(*pVisited, pDirect) |
| 64 ? pDirect->CloneNonCyclic(true, pVisited) | 65 ? pDirect->CloneNonCyclic(true, pVisited) |
| 65 : nullptr; | 66 : nullptr; |
| 66 } | 67 } |
| 67 return new CPDF_Reference(m_pObjList, m_RefObjNum); | 68 return pdfium::MakeUnique<CPDF_Reference>(m_pObjList, m_RefObjNum); |
| 68 } | 69 } |
| 69 | 70 |
| 70 CPDF_Object* CPDF_Reference::SafeGetDirect() const { | 71 CPDF_Object* CPDF_Reference::SafeGetDirect() const { |
| 71 CPDF_Object* obj = GetDirect(); | 72 CPDF_Object* obj = GetDirect(); |
| 72 return (obj && !obj->IsReference()) ? obj : nullptr; | 73 return (obj && !obj->IsReference()) ? obj : nullptr; |
| 73 } | 74 } |
| 74 | 75 |
| 75 void CPDF_Reference::SetRef(CPDF_IndirectObjectHolder* pDoc, uint32_t objnum) { | 76 void CPDF_Reference::SetRef(CPDF_IndirectObjectHolder* pDoc, uint32_t objnum) { |
| 76 m_pObjList = pDoc; | 77 m_pObjList = pDoc; |
| 77 m_RefObjNum = objnum; | 78 m_RefObjNum = objnum; |
| 78 } | 79 } |
| 79 | 80 |
| 80 CPDF_Object* CPDF_Reference::GetDirect() const { | 81 CPDF_Object* CPDF_Reference::GetDirect() const { |
| 81 return m_pObjList ? m_pObjList->GetOrParseIndirectObject(m_RefObjNum) | 82 return m_pObjList ? m_pObjList->GetOrParseIndirectObject(m_RefObjNum) |
| 82 : nullptr; | 83 : nullptr; |
| 83 } | 84 } |
| OLD | NEW |