EStatusCode DCTDecodeFilterTest::ModifyImageObject(PDFWriter* inWriter,ObjectIDType inImageObject)
{
EStatusCode status = eSuccess;
PDFDocumentCopyingContext* modifiedFileContext = inWriter->CreatePDFCopyingContextForModifiedFile();
do
{
// get image source dictionary
PDFObjectCastPtr<PDFStreamInput> imageStream(inWriter->GetModifiedFileParser().ParseNewObject(inImageObject));
RefCountPtr<PDFDictionary> imageDictionary(imageStream->QueryStreamDictionary());
// strt object for modified image
inWriter->GetObjectsContext().StartModifiedIndirectObject(inImageObject);
DictionaryContext* newImageDictionary = inWriter->GetObjectsContext().StartDictionary();
MapIterator<PDFNameToPDFObjectMap> it = imageDictionary->GetIterator();
// copy all but "Filter" and "Length"
ObjectIDTypeList indirectObjects;
while (it.MoveNext())
{
if(it.GetKey()->GetValue() == "Filter" || it.GetKey()->GetValue() == "Length")
continue;
newImageDictionary->WriteKey(it.GetKey()->GetValue());
EStatusCodeAndObjectIDTypeList result = modifiedFileContext->CopyDirectObjectWithDeepCopy(it.GetValue());
if(result.first != eSuccess)
{
status = result.first;
break;
}
indirectObjects.insert(indirectObjects.end(),result.second.begin(),result.second.end());
}
if(status != eSuccess)
break;
// start image stream for this dictionary (make sure it's unfiltered)
PDFStream* newImageStream = inWriter->GetObjectsContext().StartUnfilteredPDFStream(newImageDictionary);
// copy source stream through read filter
IByteReader* sourceImage = modifiedFileContext->GetSourceDocumentParser()->StartReadingFromStream(imageStream.GetPtr());
if(!sourceImage)
{
cout<<"failed to read DCT stream\n";
status = eFailure;
break;
}
OutputStreamTraits traits(newImageStream->GetWriteStream());
status = traits.CopyToOutputStream(sourceImage);
// finalize stream
inWriter->GetObjectsContext().EndPDFStream(newImageStream);
delete newImageStream;
// late check for status so stream is deleted
if(status != eSuccess)
break;
// copy remaining indirect objects from image stream dictionary
status = modifiedFileContext->CopyNewObjectsForDirectObject(indirectObjects);
}
while (false);
delete modifiedFileContext;
return status;
}
EStatusCode UseCopyingContext()
{
PDFWriter pdfWriter;
EStatusCode status;
PDFDocumentCopyingContext* firstContext = NULL;
PDFDocumentCopyingContext* secondContext = NULL;
do
{
status = pdfWriter.StartPDF(scBasePath + "UseCopyingContext.PDF",ePDFVersion13);
if(status != eSuccess)
break;
firstContext = pdfWriter.CreatePDFCopyingContext(scBasePath + "XObjectContent.PDF");
if(!firstContext)
{
status = eFailure;
break;
}
secondContext = pdfWriter.CreatePDFCopyingContext(scBasePath + "BasicTIFFImagesTest.PDF");
if(!firstContext)
{
status = eFailure;
break;
}
// appending pages, first from XObjectContent, then from BasicTIFFImageTest, and then again from XObjectContent
status = firstContext->AppendPDFPageFromPDF(0).first;
if(status != eSuccess)
break;
status = secondContext->AppendPDFPageFromPDF(0).first;
if(status != eSuccess)
break;
status = firstContext->AppendPDFPageFromPDF(1).first;
if(status != eSuccess)
break;
// placing pages as xobjects, from multiple PDFs
EStatusCodeAndObjectIDType resultFirst = firstContext->CreateFormXObjectFromPDFPage(0,ePDFPageBoxMediaBox);
if(resultFirst.first != eSuccess)
{
status = eFailure;
break;
}
EStatusCodeAndObjectIDType resultSecond = secondContext->CreateFormXObjectFromPDFPage(0,ePDFPageBoxMediaBox);
if(resultSecond.first != eSuccess)
{
status = eFailure;
break;
}
PDFPage* page = new PDFPage();
page->SetMediaBox(PDFRectangle(0,0,595,842));
PageContentContext* contentContext = pdfWriter.StartPageContentContext(page);
// placing the pages in a result page
contentContext->q();
contentContext->cm(0.5,0,0,0.5,0,421);
contentContext->Do(page->GetResourcesDictionary().AddFormXObjectMapping(resultFirst.second));
contentContext->Q();
contentContext->q();
contentContext->cm(0.5,0,0,0.5,297.5,0);
contentContext->Do(page->GetResourcesDictionary().AddFormXObjectMapping(resultSecond.second));
contentContext->Q();
status = pdfWriter.EndPageContentContext(contentContext);
if(status != eSuccess)
break;
status = pdfWriter.WritePageAndRelease(page);
if(status != eSuccess)
break;
status = pdfWriter.EndPDF();
if(status != eSuccess)
break;
}while(false);
delete firstContext;
delete secondContext;
return status;
}
EStatusCode PDFCopyingContextTest::Run()
{
EStatusCode status;
PDFWriter pdfWriter;
PDFDocumentCopyingContext* copyingContext = NULL;
do
{
status = pdfWriter.StartPDF("C:\\PDFLibTests\\PDFCopyingContextTest.PDF",ePDFVersion13,LogConfiguration(true,true,"c:\\pdflibtests\\PDFCopyingContextTest.txt"));
if(status != PDFHummus::eSuccess)
{
cout<<"failed to start PDF\n";
break;
}
copyingContext = pdfWriter.CreatePDFCopyingContext("C:\\PDFLibTests\\TestMaterials\\BasicTIFFImagesTest.PDF");
if(!copyingContext)
{
cout<<"failed to initialize copying context from BasicTIFFImagesTest\n";
status = PDFHummus::eFailure;
break;
}
EStatusCodeAndObjectIDType result = copyingContext->AppendPDFPageFromPDF(1);
if(result.first != PDFHummus::eSuccess)
{
cout<<"failed to append page 1 from BasicTIFFImagesTest.PDF\n";
status = result.first;
break;
}
result = copyingContext->AppendPDFPageFromPDF(18);
if(result.first != PDFHummus::eSuccess)
{
cout<<"failed to append page 18 from BasicTIFFImagesTest.PDF\n";
status = result.first;
break;
}
result = copyingContext->AppendPDFPageFromPDF(4);
if(result.first != PDFHummus::eSuccess)
{
cout<<"failed to append page 4 from BasicTIFFImagesTest.PDF\n";
status = result.first;
break;
}
copyingContext->End(); // delete will call End() as well...so can avoid
delete copyingContext;
copyingContext = NULL;
status = pdfWriter.EndPDF();
if(status != PDFHummus::eSuccess)
{
cout<<"failed in end PDF\n";
break;
}
}while(false);
delete copyingContext;
return status;
}
EStatusCode MergeToPDFForm::Run(const TestConfiguration& inTestConfiguration)
{
PDFWriter pdfWriter;
EStatusCode status;
PDFDocumentCopyingContext* copyingContext = NULL;
do
{
status = pdfWriter.StartPDF(RelativeURLToLocalPath(inTestConfiguration.mSampleFileBase,"MergeToPDFForm.pdf"),
ePDFVersion13,
LogConfiguration(true,true,RelativeURLToLocalPath(inTestConfiguration.mSampleFileBase,"MergeToPDFForm.txt")));
if(status != PDFHummus::eSuccess)
break;
// in this test we will merge 2 pages into a PDF form, and place it twice, forming a 2X2 design. amazing.
PDFPage* page = new PDFPage();
page->SetMediaBox(PDFRectangle(0,0,595,842));
copyingContext = pdfWriter.CreatePDFCopyingContext(RelativeURLToLocalPath(inTestConfiguration.mSampleFileBase,"TestMaterials/Linearized.pdf"));
if(status != PDFHummus::eSuccess)
break;
// create form for two pages.
PDFFormXObject* newFormXObject = pdfWriter.StartFormXObject(PDFRectangle(0,0,297.5,842));
XObjectContentContext* xobjectContentContext = newFormXObject->GetContentContext();
xobjectContentContext->q();
xobjectContentContext->cm(0.5,0,0,0.5,0,0);
status = copyingContext->MergePDFPageToFormXObject(newFormXObject,1);
if(status != eSuccess)
break;
xobjectContentContext->Q();
xobjectContentContext->q();
xobjectContentContext->cm(0.5,0,0,0.5,0,421);
status = copyingContext->MergePDFPageToFormXObject(newFormXObject,0);
if(status != eSuccess)
break;
xobjectContentContext->Q();
ObjectIDType formID = newFormXObject->GetObjectID();
status = pdfWriter.EndFormXObjectAndRelease(newFormXObject);
if(status != eSuccess)
break;
// now place it in the page
PageContentContext* pageContentContext = pdfWriter.StartPageContentContext(page);
string formName = page->GetResourcesDictionary().AddFormXObjectMapping(formID);
pageContentContext->q();
pageContentContext->Do(formName);
pageContentContext->cm(1,0,0,1,297.5,0);
pageContentContext->Do(formName);
pageContentContext->Q();
status = pdfWriter.EndPageContentContext(pageContentContext);
if(status != eSuccess)
break;
status = pdfWriter.WritePageAndRelease(page);
if(status != PDFHummus::eSuccess)
break;
status = pdfWriter.EndPDF();
if(status != PDFHummus::eSuccess)
break;
}while(false);
delete copyingContext;
return status;
}
PDFHummus::EStatusCode PDFModifiedPage::WritePage()
{
EStatusCode status = EndContentContext(); // just in case someone forgot to close the latest content context
do {
if (status != eSuccess || !mIsDirty) {
break;
}
// allocate an object ID for the new contents stream (for placing the form)
// we first create the modified page object, so that we can define a name for the new form xobject
// that is unique
ObjectsContext& objectContext = mWriter->GetObjectsContext();
ObjectIDType newContentObjectID = objectContext.GetInDirectObjectsRegistry().AllocateNewObjectID();
ObjectIDType newEncapsulatingObjectID = 0;
// create a copying context, so we can copy the page dictionary, and modify its contents + resources dict
PDFDocumentCopyingContext* copyingContext = mWriter->CreatePDFCopyingContextForModifiedFile();
// get the page object
ObjectIDType pageObjectID = copyingContext->GetSourceDocumentParser()->GetPageObjectID(mPageIndex);
PDFObjectCastPtr<PDFDictionary> pageDictionaryObject = copyingContext->GetSourceDocumentParser()->ParsePage(mPageIndex);
MapIterator<PDFNameToPDFObjectMap> pageDictionaryObjectIt = pageDictionaryObject->GetIterator();
// create modified page object
objectContext.StartModifiedIndirectObject(pageObjectID);
DictionaryContext* modifiedPageObject = mWriter->GetObjectsContext().StartDictionary();
// copy all elements of the page to the new page object, but the "Contents", "Resources" and "Annots" elements
while (pageDictionaryObjectIt.MoveNext())
{
if (pageDictionaryObjectIt.GetKey()->GetValue() != "Resources" &&
pageDictionaryObjectIt.GetKey()->GetValue() != "Contents" &&
pageDictionaryObjectIt.GetKey()->GetValue() != "Annots")
{
modifiedPageObject->WriteKey(pageDictionaryObjectIt.GetKey()->GetValue());
copyingContext->CopyDirectObjectAsIs(pageDictionaryObjectIt.GetValue());
}
}
// Write new annotations entry, joining existing annotations, and new ones (from links attaching or what not)
if (pageDictionaryObject->Exists("Annots") || mWriter->GetDocumentContext().GetAnnotations().size() > 0)
{
modifiedPageObject->WriteKey("Annots");
objectContext.StartArray();
// write old annots, if any exist
if(pageDictionaryObject->Exists("Annots")) {
PDFObjectCastPtr<PDFArray> anArray(copyingContext->GetSourceDocumentParser()->QueryDictionaryObject(pageDictionaryObject.GetPtr(), "Annots"));
SingleValueContainerIterator<PDFObjectVector> refs = anArray->GetIterator();
while (refs.MoveNext())
copyingContext->CopyDirectObjectAsIs(refs.GetItem());
}
// write new annots from links
ObjectIDTypeSet& annotations = mWriter->GetDocumentContext().GetAnnotations();
if (annotations.size() > 0)
{
ObjectIDTypeSet::iterator it = annotations.begin();
for (; it != annotations.end(); ++it)
objectContext.WriteNewIndirectObjectReference(*it);
}
annotations.clear();
objectContext.EndArray(eTokenSeparatorEndLine);
}
// Write new contents entry, joining the existing contents with the new one. take care of various scenarios of the existing Contents
modifiedPageObject->WriteKey("Contents");
if (!pageDictionaryObject->Exists("Contents"))
{ // no contents
objectContext.WriteIndirectObjectReference(newContentObjectID);
}
else
{
objectContext.StartArray();
if (mEnsureContentEncapsulation)
{
newEncapsulatingObjectID = objectContext.GetInDirectObjectsRegistry().AllocateNewObjectID();
objectContext.WriteNewIndirectObjectReference(newEncapsulatingObjectID);
}
RefCountPtr<PDFObject> pageContent(copyingContext->GetSourceDocumentParser()->QueryDictionaryObject(pageDictionaryObject.GetPtr(), "Contents"));
if (pageContent->GetType() == PDFObject::ePDFObjectStream)
{
// single content stream. must be a refrence which points to it
PDFObjectCastPtr<PDFIndirectObjectReference> ref(pageDictionaryObject->QueryDirectObject("Contents"));
objectContext.WriteIndirectObjectReference(ref->mObjectID, ref->mVersion);
}
else if (pageContent->GetType() == PDFObject::ePDFObjectArray)
{
PDFArray* anArray = (PDFArray*)pageContent.GetPtr();
// multiple content streams
SingleValueContainerIterator<PDFObjectVector> refs = anArray->GetIterator();
PDFObjectCastPtr<PDFIndirectObjectReference> ref;
while (refs.MoveNext())
{
ref = refs.GetItem();
objectContext.WriteIndirectObjectReference(ref->mObjectID, ref->mVersion);
}
}
else {
// this basically means no content...or whatever. just ignore.
}
objectContext.WriteNewIndirectObjectReference(newContentObjectID);
objectContext.EndArray();
objectContext.EndLine();
}
// Write a new resource entry. copy all but the "XObject" entry, which needs to be modified. Just for kicks i'm keeping the original
// form (either direct dictionary, or indirect object)
ObjectIDType resourcesIndirect = 0;
ObjectIDType newResourcesIndirect = 0;
vector<string> formResourcesNames;
modifiedPageObject->WriteKey("Resources");
if (!pageDictionaryObject->Exists("Resources"))
{
// check if there's inherited dict. if so - write directly as a modified version
PDFObjectCastPtr<PDFDictionary> parentDict(
pageDictionaryObject->Exists("Parent") ?
copyingContext->GetSourceDocumentParser()->QueryDictionaryObject(pageDictionaryObject.GetPtr(), "Parent"):
NULL);
if(!parentDict) {
formResourcesNames = WriteNewResourcesDictionary(objectContext);
}
else {
PDFObjectCastPtr<PDFDictionary> inheritedResources = findInheritedResources(copyingContext->GetSourceDocumentParser(),parentDict.GetPtr());
if(!inheritedResources) {
formResourcesNames = WriteNewResourcesDictionary(objectContext);
} else {
formResourcesNames = WriteModifiedResourcesDict(copyingContext->GetSourceDocumentParser(), inheritedResources.GetPtr(), objectContext, copyingContext);
}
}
}
else
{
// resources may be direct, or indirect. if direct, write as is, adding the new form xobject, otherwise wait till page object ends and write then
PDFObjectCastPtr<PDFIndirectObjectReference> resourceDictRef(pageDictionaryObject->QueryDirectObject("Resources"));
if (!resourceDictRef)
{
PDFObjectCastPtr<PDFDictionary> resourceDict(pageDictionaryObject->QueryDirectObject("Resources"));
formResourcesNames = WriteModifiedResourcesDict(copyingContext->GetSourceDocumentParser(), resourceDict.GetPtr(), objectContext, copyingContext);
}
else
{
resourcesIndirect = resourceDictRef->mObjectID;
// later will write a modified version of the resources dictionary, with the new form.
// only modify the resources dict object if wasn't already modified (can happen when sharing resources dict between multiple pages).
// in the case where it was alrady modified, create a new resources dictionary that's a copy, and use it instead, to avoid overwriting
// the previous modification
GetObjectWriteInformationResult res = objectContext.GetInDirectObjectsRegistry().GetObjectWriteInformation(resourcesIndirect);
if (res.first && res.second.mIsDirty)
{
newResourcesIndirect = objectContext.GetInDirectObjectsRegistry().AllocateNewObjectID();
modifiedPageObject->WriteObjectReferenceValue(newResourcesIndirect);
}
else
modifiedPageObject->WriteObjectReferenceValue(resourcesIndirect);
}
}
objectContext.EndDictionary(modifiedPageObject);
objectContext.EndIndirectObject();
if (resourcesIndirect != 0)
{
if (newResourcesIndirect != 0)
objectContext.StartNewIndirectObject(newResourcesIndirect);
else
objectContext.StartModifiedIndirectObject(resourcesIndirect);
PDFObjectCastPtr<PDFDictionary> resourceDict(copyingContext->GetSourceDocumentParser()->ParseNewObject(resourcesIndirect));
formResourcesNames = WriteModifiedResourcesDict(copyingContext->GetSourceDocumentParser(), resourceDict.GetPtr(), objectContext, copyingContext);
objectContext.EndIndirectObject();
}
// if required write encapsulation code, so that new stream is independent of graphic context of original
PDFStream* newStream;
PrimitiveObjectsWriter primitivesWriter;
if (newEncapsulatingObjectID != 0)
{
objectContext.StartNewIndirectObject(newEncapsulatingObjectID);
newStream = objectContext.StartPDFStream();
primitivesWriter.SetStreamForWriting(newStream->GetWriteStream());
primitivesWriter.WriteKeyword("q");
objectContext.EndPDFStream(newStream);
}
// last but not least, create the actual content stream object, placing the form
objectContext.StartNewIndirectObject(newContentObjectID);
newStream = objectContext.StartPDFStream();
primitivesWriter.SetStreamForWriting(newStream->GetWriteStream());
if (newEncapsulatingObjectID != 0) {
primitivesWriter.WriteKeyword("Q");
}
vector<string>::iterator it = formResourcesNames.begin();
for (; it != formResourcesNames.end(); ++it)
{
primitivesWriter.WriteKeyword("q");
primitivesWriter.WriteInteger(1);
primitivesWriter.WriteInteger(0);
primitivesWriter.WriteInteger(0);
primitivesWriter.WriteInteger(1);
primitivesWriter.WriteInteger(0);
primitivesWriter.WriteInteger(0);
primitivesWriter.WriteKeyword("cm");
primitivesWriter.WriteName(*it);
primitivesWriter.WriteKeyword("Do");
primitivesWriter.WriteKeyword("Q");
}
objectContext.EndPDFStream(newStream);
} while (false);
return status;
}
