summaryrefslogtreecommitdiff
path: root/samples
diff options
context:
space:
mode:
Diffstat (limited to 'samples')
-rw-r--r--samples/pdfium_test.cc175
1 files changed, 103 insertions, 72 deletions
diff --git a/samples/pdfium_test.cc b/samples/pdfium_test.cc
index 66b84dac60..9ae9e2759f 100644
--- a/samples/pdfium_test.cc
+++ b/samples/pdfium_test.cc
@@ -332,6 +332,70 @@ FPDF_BOOL Is_Data_Avail(FX_FILEAVAIL* pThis, size_t offset, size_t size) {
void Add_Segment(FX_DOWNLOADHINTS* pThis, size_t offset, size_t size) {
}
+FPDF_BOOL RenderPage(const std::string& name,
+ const FPDF_DOCUMENT& doc,
+ const FPDF_FORMHANDLE& form,
+ const int page_index,
+ const Options& options) {
+ FPDF_PAGE page = FPDF_LoadPage(doc, page_index);
+ if (!page) {
+ return FALSE;
+ }
+ FPDF_TEXTPAGE text_page = FPDFText_LoadPage(page);
+ FORM_OnAfterLoadPage(page, form);
+ FORM_DoPageAAction(page, form, FPDFPAGE_AACTION_OPEN);
+
+ double scale = 1.0;
+ if (!options.scale_factor_as_string.empty()) {
+ std::stringstream(options.scale_factor_as_string) >> scale;
+ }
+ int width = static_cast<int>(FPDF_GetPageWidth(page) * scale);
+ int height = static_cast<int>(FPDF_GetPageHeight(page) * scale);
+
+ FPDF_BITMAP bitmap = FPDFBitmap_Create(width, height, 0);
+ if (!bitmap) {
+ fprintf(stderr, "Page was too large to be rendered.\n");
+ return FALSE;
+ }
+
+ FPDFBitmap_FillRect(bitmap, 0, 0, width, height, 0xFFFFFFFF);
+ FPDF_RenderPageBitmap(bitmap, page, 0, 0, width, height, 0, 0);
+
+ FPDF_FFLDraw(form, bitmap, page, 0, 0, width, height, 0, 0);
+ int stride = FPDFBitmap_GetStride(bitmap);
+ const char* buffer =
+ reinterpret_cast<const char*>(FPDFBitmap_GetBuffer(bitmap));
+
+ switch (options.output_format) {
+#ifdef _WIN32
+ case OUTPUT_BMP:
+ WriteBmp(name.c_str(), page_index, buffer, stride, width, height);
+ break;
+
+ case OUTPUT_EMF:
+ WriteEmf(page, name.c_str(), page_index);
+ break;
+#endif
+ case OUTPUT_PNG:
+ WritePng(name.c_str(), page_index, buffer, stride, width, height);
+ break;
+
+ case OUTPUT_PPM:
+ WritePpm(name.c_str(), page_index, buffer, stride, width, height);
+ break;
+
+ default:
+ break;
+ }
+
+ FPDFBitmap_Destroy(bitmap);
+ FORM_DoPageAAction(page, form, FPDFPAGE_AACTION_CLOSE);
+ FORM_OnBeforeClosePage(page, form);
+ FPDFText_ClosePage(text_page);
+ FPDF_ClosePage(page);
+ return TRUE;
+}
+
void RenderPdf(const std::string& name, const char* pBuf, size_t len,
const Options& options) {
fprintf(stderr, "Rendering PDF file %s.\n", name.c_str());
@@ -365,14 +429,33 @@ void RenderPdf(const std::string& name, const char* pBuf, size_t len,
hints.AddSegment = Add_Segment;
FPDF_DOCUMENT doc;
+ int nRet = PDF_DATA_NOTAVAIL;
+ FPDF_BOOL bIsLinearized = FALSE;
FPDF_AVAIL pdf_avail = FPDFAvail_Create(&file_avail, &file_access);
-
- (void)FPDFAvail_IsDocAvail(pdf_avail, &hints);
-
- if (FPDFAvail_IsLinearized(pdf_avail))
+ if (FPDFAvail_IsLinearized(pdf_avail) == PDF_LINEARIZED) {
+ fprintf(stderr, "Linearized path...\n");
doc = FPDFAvail_GetDocument(pdf_avail, nullptr);
- else
+ if (doc) {
+ while (nRet == PDF_DATA_NOTAVAIL) {
+ nRet = FPDFAvail_IsDocAvail(pdf_avail, &hints);
+ }
+ if (nRet == PDF_DATA_ERROR) {
+ fprintf(stderr, "Unknown error in checking if doc was available.\n");
+ return;
+ }
+ nRet = FPDFAvail_IsFormAvail(pdf_avail, &hints);
+ if (nRet == PDF_FORM_ERROR || nRet == PDF_FORM_NOTAVAIL) {
+ fprintf(stderr,
+ "Error %d was returned in checking if form was available.\n",
+ nRet);
+ return;
+ }
+ bIsLinearized = TRUE;
+ }
+ } else {
+ fprintf(stderr, "Non-linearized path...\n");
doc = FPDF_LoadCustomDocument(&file_access, nullptr);
+ }
if (!doc) {
unsigned long err = FPDF_GetLastError();
@@ -409,7 +492,6 @@ void RenderPdf(const std::string& name, const char* pBuf, size_t len,
}
(void)FPDF_GetDocPermissions(doc);
- (void)FPDFAvail_IsFormAvail(pdf_avail, &hints);
FPDF_FORMHANDLE form = FPDFDOC_InitFormFillEnvironment(doc, &form_callbacks);
int docType = DOCTYPE_PDF;
@@ -420,80 +502,29 @@ void RenderPdf(const std::string& name, const char* pBuf, size_t len,
FPDF_SetFormFieldHighlightColor(form, 0, 0xFFE4DD);
FPDF_SetFormFieldHighlightAlpha(form, 100);
- int first_page = FPDFAvail_GetFirstPageNum(doc);
- (void)FPDFAvail_IsPageAvail(pdf_avail, first_page, &hints);
-
- int page_count = FPDF_GetPageCount(doc);
- for (int i = 0; i < page_count; ++i) {
- (void)FPDFAvail_IsPageAvail(pdf_avail, i, &hints);
- }
-
FORM_DoDocumentJSAction(form);
FORM_DoDocumentOpenAction(form);
+ int page_count = FPDF_GetPageCount(doc);
int rendered_pages = 0;
int bad_pages = 0;
for (int i = 0; i < page_count; ++i) {
- FPDF_PAGE page = FPDF_LoadPage(doc, i);
- if (!page) {
- ++bad_pages;
- continue;
- }
- FPDF_TEXTPAGE text_page = FPDFText_LoadPage(page);
- FORM_OnAfterLoadPage(page, form);
- FORM_DoPageAAction(page, form, FPDFPAGE_AACTION_OPEN);
-
- double scale = 1.0;
- if (!options.scale_factor_as_string.empty()) {
- std::stringstream(options.scale_factor_as_string) >> scale;
- }
- int width = static_cast<int>(FPDF_GetPageWidth(page) * scale);
- int height = static_cast<int>(FPDF_GetPageHeight(page) * scale);
-
- FPDF_BITMAP bitmap = FPDFBitmap_Create(width, height, 0);
- if (!bitmap) {
- fprintf(stderr, "Page was too large to be rendered.\n");
- bad_pages++;
- continue;
+ if (bIsLinearized) {
+ nRet = PDF_DATA_NOTAVAIL;
+ while (nRet == PDF_DATA_NOTAVAIL) {
+ nRet = FPDFAvail_IsPageAvail(pdf_avail, i, &hints);
+ }
+ if (nRet == PDF_DATA_ERROR) {
+ fprintf(stderr, "Unknown error in checking if page %d is available.\n",
+ i);
+ return;
+ }
}
-
- FPDFBitmap_FillRect(bitmap, 0, 0, width, height, 0xFFFFFFFF);
- FPDF_RenderPageBitmap(bitmap, page, 0, 0, width, height, 0, 0);
- ++rendered_pages;
-
- FPDF_FFLDraw(form, bitmap, page, 0, 0, width, height, 0, 0);
- int stride = FPDFBitmap_GetStride(bitmap);
- const char* buffer =
- reinterpret_cast<const char*>(FPDFBitmap_GetBuffer(bitmap));
-
- switch (options.output_format) {
-#ifdef _WIN32
- case OUTPUT_BMP:
- WriteBmp(name.c_str(), i, buffer, stride, width, height);
- break;
-
- case OUTPUT_EMF:
- WriteEmf(page, name.c_str(), i);
- break;
-#endif
- case OUTPUT_PPM:
- WritePpm(name.c_str(), i, buffer, stride, width, height);
- break;
-
- case OUTPUT_PNG:
- WritePng(name.c_str(), i, buffer, stride, width, height);
- break;
-
- default:
- break;
+ if (RenderPage(name, doc, form, i, options)) {
+ ++rendered_pages;
+ } else {
+ ++bad_pages;
}
-
- FPDFBitmap_Destroy(bitmap);
-
- FORM_DoPageAAction(page, form, FPDFPAGE_AACTION_CLOSE);
- FORM_OnBeforeClosePage(page, form);
- FPDFText_ClosePage(text_page);
- FPDF_ClosePage(page);
}
FORM_DoDocumentAAction(form, FPDFDOC_AACTION_WC);