Index: samples/pdfium_test.cc |
diff --git a/samples/pdfium_test.cc b/samples/pdfium_test.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..67c15e0ea5d224dfbba70f1dc9c2c21e1ba4c9c8 |
--- /dev/null |
+++ b/samples/pdfium_test.cc |
@@ -0,0 +1,309 @@ |
+// Copyright (c) 2010 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include <limits.h> |
+#include <stdio.h> |
+#include <stdlib.h> |
+#include <string.h> |
+ |
+#include <list> |
+#include <string> |
+#include <utility> |
+ |
+#include "fpdfsdk/include/fpdf_dataavail.h" |
+#include "fpdfsdk/include/fpdf_ext.h" |
+#include "fpdfsdk/include/fpdfformfill.h" |
+#include "fpdfsdk/include/fpdftext.h" |
+#include "fpdfsdk/include/fpdfview.h" |
+#include "v8/include/v8.h" |
+ |
+#ifdef _WIN32 |
+ #define snprintf _snprintf |
+ /* in Windows, rb for open and read binary file */ |
+ #define FOPEN_READ "rb" |
+#else |
+ #define FOPEN_READ "r" |
+#endif |
+ |
+static void WritePpm(const char* pdf_name, int num, |
+ const char* buffer, int stride, int width, int height) { |
+ if (stride < 0 || width < 0 || height < 0) |
+ return; |
+ if (height > 0 && width > INT_MAX / height) |
+ return; |
+ int out_len = width * height; |
+ if (out_len > INT_MAX / 3) |
+ return; |
+ out_len *= 3; |
+ |
+ char filename[256]; |
+ snprintf(filename, sizeof(filename), "%s.%d.ppm", pdf_name, num); |
+ FILE* fp = fopen(filename, "w"); |
+ if (!fp) |
+ return; |
+ fprintf(fp, "P6\n# PDF test render\n%d %d\n255\n", width, height); |
+ // Source data is B, G, R, unused. |
+ // Dest data is R, G, B. |
+ char* result = new char[out_len]; |
+ if (result) { |
+ for (int h = 0; h < height; ++h) { |
+ const char* src_line = buffer + (stride * h); |
+ char* dest_line = result + (width * h * 3); |
+ for (int w = 0; w < width; ++w) { |
+ // R |
+ dest_line[w * 3] = src_line[(w * 4) + 2]; |
+ // G |
+ dest_line[(w * 3) + 1] = src_line[(w * 4) + 1]; |
+ // B |
+ dest_line[(w * 3) + 2] = src_line[w * 4]; |
+ } |
+ } |
+ fwrite(result, out_len, 1, fp); |
+ delete [] result; |
+ } |
+ fclose(fp); |
+} |
+ |
+int Form_Alert(IPDF_JSPLATFORM*, FPDF_WIDESTRING, FPDF_WIDESTRING, int, int) { |
+ printf("Form_Alert called.\n"); |
+ return 0; |
+} |
+ |
+void Unsupported_Handler(UNSUPPORT_INFO*, int type) { |
+ std::string feature = "Unknown"; |
+ switch (type) { |
+ case FPDF_UNSP_DOC_XFAFORM: |
+ feature = "XFA"; |
+ break; |
+ case FPDF_UNSP_DOC_PORTABLECOLLECTION: |
+ feature = "Portfolios_Packages"; |
+ break; |
+ case FPDF_UNSP_DOC_ATTACHMENT: |
+ case FPDF_UNSP_ANNOT_ATTACHMENT: |
+ feature = "Attachment"; |
+ break; |
+ case FPDF_UNSP_DOC_SECURITY: |
+ feature = "Rights_Management"; |
+ break; |
+ case FPDF_UNSP_DOC_SHAREDREVIEW: |
+ feature = "Shared_Review"; |
+ break; |
+ case FPDF_UNSP_DOC_SHAREDFORM_ACROBAT: |
+ case FPDF_UNSP_DOC_SHAREDFORM_FILESYSTEM: |
+ case FPDF_UNSP_DOC_SHAREDFORM_EMAIL: |
+ feature = "Shared_Form"; |
+ break; |
+ case FPDF_UNSP_ANNOT_3DANNOT: |
+ feature = "3D"; |
+ break; |
+ case FPDF_UNSP_ANNOT_MOVIE: |
+ feature = "Movie"; |
+ break; |
+ case FPDF_UNSP_ANNOT_SOUND: |
+ feature = "Sound"; |
+ break; |
+ case FPDF_UNSP_ANNOT_SCREEN_MEDIA: |
+ case FPDF_UNSP_ANNOT_SCREEN_RICHMEDIA: |
+ feature = "Screen"; |
+ break; |
+ case FPDF_UNSP_ANNOT_SIG: |
+ feature = "Digital_Signature"; |
+ break; |
+ } |
+ printf("Unsupported feature: %s.\n", feature.c_str()); |
+} |
+ |
+bool ParseCommandLine(int argc, const char* argv[], bool* write_images, |
+ std::list<const char*>* files) { |
+ *write_images = false; |
+ files->clear(); |
+ |
+ int cur_arg = 1; |
+ if (cur_arg < argc && |
+ strcmp(argv[cur_arg], "--write_images") == 0) { |
+ *write_images = true; |
+ cur_arg++; |
+ } |
+ |
+ if (cur_arg >= argc) |
+ return false; |
+ |
+ for (int i = cur_arg; i < argc; i++) |
+ files->push_back(argv[i]); |
+ |
+ return true; |
+} |
+ |
+class TestLoader { |
+ public: |
+ TestLoader(const char* pBuf, size_t len); |
+ |
+ const char* m_pBuf; |
+ size_t m_Len; |
+}; |
+ |
+TestLoader::TestLoader(const char* pBuf, size_t len) |
+ : m_pBuf(pBuf), m_Len(len) { |
+} |
+ |
+int Get_Block(void* param, unsigned long pos, unsigned char* pBuf, |
+ unsigned long size) { |
+ TestLoader* pLoader = (TestLoader*) param; |
+ if (pos + size < pos || pos + size > pLoader->m_Len) return 0; |
+ memcpy(pBuf, pLoader->m_pBuf + pos, size); |
+ return 1; |
+} |
+ |
+bool Is_Data_Avail(FX_FILEAVAIL* pThis, size_t offset, size_t size) { |
+ return true; |
+} |
+ |
+void Add_Segment(FX_DOWNLOADHINTS* pThis, size_t offset, size_t size) { |
+} |
+ |
+void RenderPdf(const char* name, const char* pBuf, size_t len, |
+ bool write_images) { |
+ printf("Rendering PDF file %s.\n", name); |
+ |
+ IPDF_JSPLATFORM platform_callbacks; |
+ memset(&platform_callbacks, '\0', sizeof(platform_callbacks)); |
+ platform_callbacks.version = 1; |
+ platform_callbacks.app_alert = Form_Alert; |
+ |
+ FPDF_FORMFILLINFO form_callbacks; |
+ memset(&form_callbacks, '\0', sizeof(form_callbacks)); |
+ form_callbacks.version = 1; |
+ form_callbacks.m_pJsPlatform = &platform_callbacks; |
+ |
+ TestLoader loader(pBuf, len); |
+ |
+ FPDF_FILEACCESS file_access; |
+ memset(&file_access, '\0', sizeof(file_access)); |
+ file_access.m_FileLen = len; |
+ file_access.m_GetBlock = Get_Block; |
+ file_access.m_Param = &loader; |
+ |
+ FX_FILEAVAIL file_avail; |
+ memset(&file_avail, '\0', sizeof(file_avail)); |
+ file_avail.version = 1; |
+ file_avail.IsDataAvail = Is_Data_Avail; |
+ |
+ FX_DOWNLOADHINTS hints; |
+ memset(&hints, '\0', sizeof(hints)); |
+ hints.version = 1; |
+ hints.AddSegment = Add_Segment; |
+ |
+ FPDF_DOCUMENT doc; |
+ FPDF_AVAIL pdf_avail = FPDFAvail_Create(&file_avail, &file_access); |
+ |
+ (void) FPDFAvail_IsDocAvail(pdf_avail, &hints); |
+ |
+ if (!FPDFAvail_IsLinearized(pdf_avail)) { |
+ printf("Non-linearized path...\n"); |
+ doc = FPDF_LoadCustomDocument(&file_access, NULL); |
+ } else { |
+ printf("Linearized path...\n"); |
+ doc = FPDFAvail_GetDocument(pdf_avail, NULL); |
+ } |
+ |
+ (void) FPDF_GetDocPermissions(doc); |
+ (void) FPDFAvail_IsFormAvail(pdf_avail, &hints); |
+ |
+ FPDF_FORMHANDLE form = FPDFDOC_InitFormFillEnviroument(doc, &form_callbacks); |
+ 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); |
+ |
+ for (int i = 0; i < page_count; ++i) { |
+ FPDF_PAGE page = FPDF_LoadPage(doc, i); |
+ FPDF_TEXTPAGE text_page = FPDFText_LoadPage(page); |
+ FORM_OnAfterLoadPage(page, form); |
+ FORM_DoPageAAction(page, form, FPDFPAGE_AACTION_OPEN); |
+ |
+ int width = static_cast<int>(FPDF_GetPageWidth(page)); |
+ int height = static_cast<int>(FPDF_GetPageHeight(page)); |
+ FPDF_BITMAP bitmap = FPDFBitmap_Create(width, height, 0); |
+ FPDFBitmap_FillRect(bitmap, 0, 0, width, height, 255, 255, 255, 255); |
+ |
+ FPDF_RenderPageBitmap(bitmap, page, 0, 0, width, height, 0, 0); |
+ FPDF_FFLDraw(form, bitmap, page, 0, 0, width, height, 0, 0); |
+ if (write_images) { |
+ const char* buffer = reinterpret_cast<const char*>( |
+ FPDFBitmap_GetBuffer(bitmap)); |
+ int stride = FPDFBitmap_GetStride(bitmap); |
+ WritePpm(name, i, buffer, stride, width, height); |
+ } |
+ |
+ 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); |
+ FPDFDOC_ExitFormFillEnviroument(form); |
+ FPDF_CloseDocument(doc); |
+ FPDFAvail_Destroy(pdf_avail); |
+ |
+ printf("Loaded, parsed and rendered %d pages.\n", page_count); |
+} |
+ |
+int main(int argc, const char* argv[]) { |
+ v8::V8::InitializeICU(); |
+ bool write_images = false; |
+ std::list<const char*> files; |
+ if (!ParseCommandLine(argc, argv, &write_images, &files)) { |
+ printf("Usage is: test [--write_images] /path/to/pdf\n"); |
+ printf("--write_images - to write page images <pdf-name>.<page-number>.ppm\n"); |
+ return 1; |
+ } |
+ |
+ FPDF_InitLibrary(NULL); |
+ |
+ UNSUPPORT_INFO unsuppored_info; |
+ memset(&unsuppored_info, '\0', sizeof(unsuppored_info)); |
+ unsuppored_info.version = 1; |
+ unsuppored_info.FSDK_UnSupport_Handler = Unsupported_Handler; |
+ |
+ FSDK_SetUnSpObjProcessHandler(&unsuppored_info); |
+ |
+ while (!files.empty()) { |
+ const char* filename = files.front(); |
+ files.pop_front(); |
+ FILE* file = fopen(filename, FOPEN_READ); |
+ if (!file) { |
+ fprintf(stderr, "Failed to open: %s\n", filename); |
+ continue; |
+ } |
+ (void) fseek(file, 0, SEEK_END); |
+ size_t len = ftell(file); |
+ (void) fseek(file, 0, SEEK_SET); |
+ char* pBuf = (char*) malloc(len); |
+ size_t ret = fread(pBuf, 1, len, file); |
+ (void) fclose(file); |
+ if (ret != len) { |
+ fprintf(stderr, "Failed to read: %s\n", filename); |
+ } else { |
+ RenderPdf(filename, pBuf, len, write_images); |
+ } |
+ free(pBuf); |
+ } |
+ |
+ FPDF_DestroyLibrary(); |
+ |
+ return 0; |
+} |
+ |