pengjunjie/core
1
0
Fork 1
mirror of https://github.com/ONLYOFFICE/core.git synced 2026-02-10 18:05:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Delete merged object

Browse Source
This commit is contained in:
Svetlana Kulikova
2025-03-25 17:24:33 +03:00
parent b6741a504f
commit eebe3f321a
2 changed files with 102 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

102
PdfFile/PdfEditor.cpp
View File

@ -666,14 +666,25 @@ PdfWriter::CObjectBase* CObjectsManager::GetObj(int nID)
return m_mUniqueRef[nID].pObj;
return NULL;
}
void CObjectsManager::IncRefCount(int nID)
bool CObjectsManager::IncRefCount(int nID)
{
if (m_mUniqueRef.find(nID) != m_mUniqueRef.end())
{
m_mUniqueRef[nID].nRefCount++;
return true;
}
return false;
}
void CObjectsManager::DecRefCount(int nID)
bool CObjectsManager::DecRefCount(int nID)
{
if (m_mUniqueRef.find(nID) != m_mUniqueRef.end())
{
if (m_mUniqueRef[nID].pObj->IsHidden())
return false;
m_mUniqueRef[nID].pObj->SetHidden();
return true;
}
return false;
}
int CObjectsManager::FindObj(PdfWriter::CObjectBase* pObj)
{
@ -682,6 +693,69 @@ int CObjectsManager::FindObj(PdfWriter::CObjectBase* pObj)
return it->first;
return -1;
}
void CObjectsManager::DeleteObjTree(Object* obj, XRef* xref, int nStartRefID)
{
Object oTemp;
switch (obj->getType())
{
case objBool:
case objInt:
case objReal:
case objString:
case objName:
case objNull:
case objNone:
case objCmd:
case objError:
case objEOF:
break;
case objArray:
{
for (int nIndex = 0; nIndex < obj->arrayGetLength(); ++nIndex)
{
obj->arrayGetNF(nIndex, &oTemp);
DeleteObjTree(&oTemp, xref, nStartRefID);
oTemp.free();
}
break;
}
case objDict:
{
if (SplitSkipDict(obj, this, nStartRefID))
return;
for (int nIndex = 0; nIndex < obj->dictGetLength(); ++nIndex)
{
obj->dictGetValNF(nIndex, &oTemp);
DeleteObjTree(&oTemp, xref, nStartRefID);
oTemp.free();
}
break;
}
case objRef:
{
int nObjNum = obj->getRefNum();
PdfWriter::CObjectBase* pObj = GetObj(nObjNum + nStartRefID);
if (pObj && DecRefCount(nObjNum + nStartRefID))
return;
obj->fetch(xref, &oTemp);
DeleteObjTree(&oTemp, xref, nStartRefID);
oTemp.free();
break;
}
case objStream:
{
Dict* pODict = obj->streamGetDict();
for (int nIndex = 0; nIndex < pODict->getLength(); ++nIndex)
{
pODict->getValNF(nIndex, &oTemp);
DeleteObjTree(&oTemp, xref, nStartRefID);
oTemp.free();
}
break;
}
}
}
CPdfEditor::CPdfEditor(const std::wstring& _wsSrcFile, const std::wstring& _wsPassword, const std::wstring& _wsDstFile, CPdfReader* _pReader, CPdfWriter* _pWriter)
{
@ -1708,7 +1782,19 @@ bool CPdfEditor::DeletePage(int nPageIndex)
PdfWriter::CPage* pPage = pDoc->GetPage(nPageIndex);
int nObjID = m_mObjManager.FindObj(pPage);
if (nObjID > 0)
{
PDFDoc* pPDFDocument = NULL;
int nStartRefID = 0;
int nRefID = m_pReader->FindRefNum(nObjID, &pPDFDocument, &nStartRefID);
if (nRefID > 0)
{
Object oRef;
// TODO узнать gen
oRef.initRef(nRefID, 0);
m_mObjManager.DeleteObjTree(&oRef, pPDFDocument->getXRef(), nStartRefID);
}
pPage->SetHidden();
}
return pDoc->DeletePage(nPageIndex);
}
@ -2009,6 +2095,16 @@ bool CPdfEditor::DeleteAnnot(int nID, Object* oAnnots)
PdfWriter::CObjectBase* pObj = m_mObjManager.GetObj(nID);
if (pObj)
{
PDFDoc* pPDFDocument = NULL;
int nStartRefID = 0;
int nRefID = m_pReader->FindRefNum(nID, &pPDFDocument, &nStartRefID);
if (nRefID > 0)
{
Object oRef;
// TODO узнать gen
oRef.initRef(nRefID, 0);
m_mObjManager.DeleteObjTree(&oRef, pPDFDocument->getXRef(), nStartRefID);
}
pObj->SetHidden();
return true;
}

5
PdfFile/PdfEditor.h
View File

@ -51,9 +51,10 @@ class CObjectsManager
public:
void AddObj(int nID, PdfWriter::CObjectBase* pObj);
PdfWriter::CObjectBase* GetObj(int nID);
void IncRefCount(int nID);
void DecRefCount(int nID);
bool IncRefCount(int nID);
bool DecRefCount(int nID);
int FindObj(PdfWriter::CObjectBase* pObj);
void DeleteObjTree(Object* obj, XRef* xref, int nStartRefID);
private:
std::map<int, CObjectInfo> m_mUniqueRef; // map уникальных объектов