Chromium Code Reviews| Index: core/fpdfapi/fpdf_parser/cpdf_indirect_object_holder.cpp |
| diff --git a/core/fpdfapi/fpdf_parser/cpdf_indirect_object_holder.cpp b/core/fpdfapi/fpdf_parser/cpdf_indirect_object_holder.cpp |
| index 33f45a53bf1003227fc9a93dfccef077697108b7..9fb3c16c7b39a86a2db6bfab63c190e2c81aafbe 100644 |
| --- a/core/fpdfapi/fpdf_parser/cpdf_indirect_object_holder.cpp |
| +++ b/core/fpdfapi/fpdf_parser/cpdf_indirect_object_holder.cpp |
| @@ -9,39 +9,29 @@ |
| #include "core/fpdfapi/fpdf_parser/include/cpdf_object.h" |
| #include "core/fpdfapi/fpdf_parser/include/cpdf_parser.h" |
| -CPDF_IndirectObjectHolder::CPDF_IndirectObjectHolder(CPDF_Parser* pParser) |
| - : m_pParser(pParser), m_LastObjNum(0) { |
| - if (pParser) |
| - m_LastObjNum = m_pParser->GetLastObjNum(); |
| -} |
| +CPDF_IndirectObjectHolder::CPDF_IndirectObjectHolder() : m_LastObjNum(0) {} |
| CPDF_IndirectObjectHolder::~CPDF_IndirectObjectHolder() { |
| for (const auto& pair : m_IndirectObjs) |
| pair.second->Destroy(); |
| } |
| -CPDF_Object* CPDF_IndirectObjectHolder::GetIndirectObject(uint32_t objnum) { |
| +CPDF_Object* CPDF_IndirectObjectHolder::GetIndirectObject( |
| + uint32_t objnum) const { |
| if (objnum == 0) |
| return nullptr; |
| auto it = m_IndirectObjs.find(objnum); |
| - if (it != m_IndirectObjs.end()) |
| - return it->second->GetObjNum() != CPDF_Object::kInvalidObjNum ? it->second |
| - : nullptr; |
| - |
| - if (!m_pParser) |
| + if (it == m_IndirectObjs.end()) |
|
Tom Sepez
2016/08/17 16:46:45
nit: maybe ?-operator here as in
return it != m_
dsinclair
2016/08/17 16:59:25
Done.
|
| return nullptr; |
| + return it->second; |
| +} |
| - CPDF_Object* pObj = m_pParser->ParseIndirectObject(this, objnum); |
| - if (!pObj) |
| +CPDF_Object* CPDF_IndirectObjectHolder::GetIndirectObjectIfValid( |
| + uint32_t objnum) { |
| + CPDF_Object* pObj = GetIndirectObject(objnum); |
| + if (pObj && pObj->GetObjNum() == CPDF_Object::kInvalidObjNum) |
| return nullptr; |
| - |
| - pObj->m_ObjNum = objnum; |
| - m_LastObjNum = std::max(m_LastObjNum, objnum); |
| - if (m_IndirectObjs[objnum]) |
| - m_IndirectObjs[objnum]->Destroy(); |
| - |
| - m_IndirectObjs[objnum] = pObj; |
| return pObj; |
| } |
| @@ -50,36 +40,43 @@ uint32_t CPDF_IndirectObjectHolder::AddIndirectObject(CPDF_Object* pObj) { |
| return pObj->m_ObjNum; |
| m_LastObjNum++; |
| - m_IndirectObjs[m_LastObjNum] = pObj; |
| pObj->m_ObjNum = m_LastObjNum; |
| + EmplaceIndirectObject(pObj); |
| return m_LastObjNum; |
| } |
| void CPDF_IndirectObjectHolder::ReleaseIndirectObject(uint32_t objnum) { |
| - auto it = m_IndirectObjs.find(objnum); |
| - if (it == m_IndirectObjs.end() || |
| - it->second->GetObjNum() == CPDF_Object::kInvalidObjNum) { |
| + CPDF_Object* pObj = GetIndirectObjectIfValid(objnum); |
| + if (!pObj) |
| return; |
| - } |
| - it->second->Destroy(); |
| - m_IndirectObjs.erase(it); |
| + |
| + pObj->Destroy(); |
| + m_IndirectObjs.erase(objnum); |
| } |
| -bool CPDF_IndirectObjectHolder::InsertIndirectObject(uint32_t objnum, |
| - CPDF_Object* pObj) { |
| +bool CPDF_IndirectObjectHolder::InsertIndirectObjectIfHigherGeneration( |
|
Tom Sepez
2016/08/17 16:46:45
nit: More like ReplaceIndirectObjectIfHigherGenera
dsinclair
2016/08/17 16:59:25
Done.
|
| + uint32_t objnum, |
| + CPDF_Object* pObj) { |
| if (!objnum || !pObj) |
| return false; |
| - auto it = m_IndirectObjs.find(objnum); |
| - if (it != m_IndirectObjs.end()) { |
| - if (pObj->GetGenNum() <= it->second->GetGenNum()) { |
| + CPDF_Object* pOldObj = GetIndirectObject(objnum); |
| + if (pOldObj) { |
| + if (pObj->GetGenNum() <= pOldObj->GetGenNum()) { |
|
Tom Sepez
2016/08/17 16:46:44
nit: combine with &&
dsinclair
2016/08/17 16:59:25
Done.
|
| pObj->Destroy(); |
| return false; |
| } |
| - it->second->Destroy(); |
| } |
| + |
| pObj->m_ObjNum = objnum; |
| - m_IndirectObjs[objnum] = pObj; |
| - m_LastObjNum = std::max(m_LastObjNum, objnum); |
| + EmplaceIndirectObject(pObj); |
| return true; |
| } |
| + |
| +void CPDF_IndirectObjectHolder::EmplaceIndirectObject(CPDF_Object* pObj) { |
|
Tom Sepez
2016/08/17 16:46:45
nit: more like replace rather than emplace, Emplac
dsinclair
2016/08/17 16:59:25
Done.
|
| + m_LastObjNum = std::max(m_LastObjNum, pObj->m_ObjNum); |
| + if (m_IndirectObjs[pObj->m_ObjNum]) |
| + m_IndirectObjs[pObj->m_ObjNum]->Destroy(); |
| + |
| + m_IndirectObjs[pObj->m_ObjNum] = pObj; |
| +} |