fpdfsdk/src/fpdfeditpage.cpp

   1 // Copyright 2014 PDFium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
   6
   7 #include "../../public/fpdf_edit.h"
   8 #include "../../public/fpdf_formfill.h"
   9 #include "../include/fsdk_define.h"
  10
  11 #if _FX_OS_ == _FX_ANDROID_
  12 #include "time.h"
  13 #else
  14 #include <ctime>
  15 #endif
  16
  17 DLLEXPORT FPDF_DOCUMENT STDCALL FPDF_CreateNewDocument() {
  18   CPDF_Document* pDoc = new CPDF_Document;
  19   pDoc->CreateNewDoc();
  20   time_t currentTime;
  21
  22   CFX_ByteString DateStr;
  23
  24   if (FSDK_IsSandBoxPolicyEnabled(FPDF_POLICY_MACHINETIME_ACCESS)) {
  25     if (-1 != time(&currentTime)) {
  26       tm* pTM = localtime(&currentTime);
  27       if (pTM) {
  28         DateStr.Format("D:%04d%02d%02d%02d%02d%02d", pTM->tm_year + 1900,
  29                        pTM->tm_mon + 1, pTM->tm_mday, pTM->tm_hour, pTM->tm_min,
  30                        pTM->tm_sec);
  31       }
  32     }
  33   }
  34
  35   CPDF_Dictionary* pInfoDict = NULL;
  36   pInfoDict = pDoc->GetInfo();
  37   if (pInfoDict) {
  38     if (FSDK_IsSandBoxPolicyEnabled(FPDF_POLICY_MACHINETIME_ACCESS))
  39       pInfoDict->SetAt("CreationDate", new CPDF_String(DateStr));
  40     pInfoDict->SetAt("Creator", new CPDF_String(L"PDFium"));
  41   }
  42
  43   return pDoc;
  44 }
  45
  46 DLLEXPORT void STDCALL FPDFPage_Delete(FPDF_DOCUMENT document, int page_index) {
  47   CPDF_Document* pDoc = CPDF_Document::FromFPDFDocument(document);
  48   if (!pDoc || page_index < 0 || page_index >= pDoc->GetPageCount())
  49     return;
  50
  51   pDoc->DeletePage(page_index);
  52 }
  53
  54 DLLEXPORT FPDF_PAGE STDCALL FPDFPage_New(FPDF_DOCUMENT document,
  55                                          int page_index,
  56                                          double width,
  57                                          double height) {
  58   CPDF_Document* pDoc = CPDF_Document::FromFPDFDocument(document);
  59   if (!pDoc)
  60     return nullptr;
  61
  62   if (page_index < 0)
  63     page_index = 0;
  64   if (pDoc->GetPageCount() < page_index)
  65     page_index = pDoc->GetPageCount();
  66
  67   CPDF_Dictionary* pPageDict = pDoc->CreateNewPage(page_index);
  68   if (!pPageDict)
  69     return NULL;
  70   CPDF_Array* pMediaBoxArray = new CPDF_Array;
  71   pMediaBoxArray->Add(new CPDF_Number(0));
  72   pMediaBoxArray->Add(new CPDF_Number(0));
  73   pMediaBoxArray->Add(new CPDF_Number(FX_FLOAT(width)));
  74   pMediaBoxArray->Add(new CPDF_Number(FX_FLOAT(height)));
  75
  76   pPageDict->SetAt("MediaBox", pMediaBoxArray);
  77   pPageDict->SetAt("Rotate", new CPDF_Number(0));
  78   pPageDict->SetAt("Resources", new CPDF_Dictionary);
  79
  80   CPDF_Page* pPage = new CPDF_Page;
  81   pPage->Load(pDoc, pPageDict);
  82   pPage->ParseContent();
  83
  84   return pPage;
  85 }
  86
  87 DLLEXPORT int STDCALL FPDFPage_GetRotation(FPDF_PAGE page) {
  88   CPDF_Page* pPage = (CPDF_Page*)page;
  89   if (!pPage || !pPage->m_pFormDict || !pPage->m_pFormDict->KeyExist("Type") ||
  90       !pPage->m_pFormDict->GetElement("Type")->GetDirect() ||
  91       pPage->m_pFormDict->GetElement("Type")->GetDirect()->GetString().Compare(
  92           "Page")) {
  93     return -1;
  94   }
  95   CPDF_Dictionary* pDict = pPage->m_pFormDict;
  96
  97   int rotate = 0;
  98   if (pDict != NULL) {
  99     if (pDict->KeyExist("Rotate"))
 100       rotate = pDict->GetElement("Rotate")->GetDirect()
 101                    ? pDict->GetElement("Rotate")->GetDirect()->GetInteger() / 90
 102                    : 0;
 103     else {
 104       if (pDict->KeyExist("Parent")) {
 105         CPDF_Dictionary* pPages =
 106             (CPDF_Dictionary*)pDict->GetElement("Parent")->GetDirect();
 107         while (pPages) {
 108           if (pPages->KeyExist("Rotate")) {
 109             rotate =
 110                 pPages->GetElement("Rotate")->GetDirect()
 111                     ? pPages->GetElement("Rotate")->GetDirect()->GetInteger() /
 112                           90
 113                     : 0;
 114             break;
 115           } else if (pPages->KeyExist("Parent"))
 116             pPages =
 117                 (CPDF_Dictionary*)pPages->GetElement("Parent")->GetDirect();
 118           else
 119             break;
 120         }
 121       }
 122     }
 123   } else {
 124     return -1;
 125   }
 126
 127   return rotate;
 128 }
 129
 130 DLLEXPORT void STDCALL FPDFPage_InsertObject(FPDF_PAGE page,
 131                                              FPDF_PAGEOBJECT page_obj) {
 132   CPDF_Page* pPage = (CPDF_Page*)page;
 133   if (!pPage || !pPage->m_pFormDict || !pPage->m_pFormDict->KeyExist("Type") ||
 134       !pPage->m_pFormDict->GetElement("Type")->GetDirect() ||
 135       pPage->m_pFormDict->GetElement("Type")->GetDirect()->GetString().Compare(
 136           "Page")) {
 137     return;
 138   }
 139   CPDF_PageObject* pPageObj = (CPDF_PageObject*)page_obj;
 140   if (pPageObj == NULL)
 141     return;
 142   FX_POSITION LastPersition = pPage->GetLastObjectPosition();
 143
 144   pPage->InsertObject(LastPersition, pPageObj);
 145   switch (pPageObj->m_Type) {
 146     case FPDF_PAGEOBJ_PATH: {
 147       CPDF_PathObject* pPathObj = (CPDF_PathObject*)pPageObj;
 148       pPathObj->CalcBoundingBox();
 149       break;
 150     }
 151     case FPDF_PAGEOBJ_TEXT: {
 152       //        CPDF_PathObject* pPathObj = (CPDF_PathObject*)pPageObj;
 153       //        pPathObj->CalcBoundingBox();
 154       break;
 155     }
 156     case FPDF_PAGEOBJ_IMAGE: {
 157       CPDF_ImageObject* pImageObj = (CPDF_ImageObject*)pPageObj;
 158       pImageObj->CalcBoundingBox();
 159       break;
 160     }
 161     case FPDF_PAGEOBJ_SHADING: {
 162       CPDF_ShadingObject* pShadingObj = (CPDF_ShadingObject*)pPageObj;
 163       pShadingObj->CalcBoundingBox();
 164       break;
 165     }
 166     case FPDF_PAGEOBJ_FORM: {
 167       CPDF_FormObject* pFormObj = (CPDF_FormObject*)pPageObj;
 168       pFormObj->CalcBoundingBox();
 169       break;
 170     }
 171     default:
 172       break;
 173   }
 174
 175   //    pPage->ParseContent();
 176   // pPage->GenerateContent();
 177 }
 178
 179 DLLEXPORT int STDCALL FPDFPage_CountObject(FPDF_PAGE page) {
 180   CPDF_Page* pPage = (CPDF_Page*)page;
 181   if (!pPage || !pPage->m_pFormDict || !pPage->m_pFormDict->KeyExist("Type") ||
 182       !pPage->m_pFormDict->GetElement("Type")->GetDirect() ||
 183       pPage->m_pFormDict->GetElement("Type")->GetDirect()->GetString().Compare(
 184           "Page")) {
 185     return -1;
 186   }
 187   return pPage->CountObjects();
 188   //    return 0;
 189 }
 190
 191 DLLEXPORT FPDF_PAGEOBJECT STDCALL FPDFPage_GetObject(FPDF_PAGE page,
 192                                                      int index) {
 193   CPDF_Page* pPage = (CPDF_Page*)page;
 194   if (!pPage || !pPage->m_pFormDict || !pPage->m_pFormDict->KeyExist("Type") ||
 195       pPage->m_pFormDict->GetElement("Type")->GetDirect()->GetString().Compare(
 196           "Page")) {
 197     return NULL;
 198   }
 199   return pPage->GetObjectByIndex(index);
 200   //    return NULL;
 201 }
 202
 203 DLLEXPORT FPDF_BOOL STDCALL FPDFPage_HasTransparency(FPDF_PAGE page) {
 204   if (!page)
 205     return FALSE;
 206   CPDF_Page* pPage = (CPDF_Page*)page;
 207
 208   return pPage->BackgroundAlphaNeeded();
 209 }
 210
 211 DLLEXPORT FPDF_BOOL STDCALL
 212 FPDFPageObj_HasTransparency(FPDF_PAGEOBJECT pageObject) {
 213   if (!pageObject)
 214     return FALSE;
 215   CPDF_PageObject* pPageObj = (CPDF_PageObject*)pageObject;
 216
 217   const CPDF_GeneralStateData* pGeneralState = pPageObj->m_GeneralState;
 218   int blend_type =
 219       pGeneralState ? pGeneralState->m_BlendType : FXDIB_BLEND_NORMAL;
 220   if (blend_type != FXDIB_BLEND_NORMAL)
 221     return TRUE;
 222
 223   CPDF_Dictionary* pSMaskDict =
 224       pGeneralState ? (CPDF_Dictionary*)pGeneralState->m_pSoftMask : NULL;
 225   if (pSMaskDict)
 226     return TRUE;
 227
 228   if (pGeneralState && pGeneralState->m_FillAlpha != 1.0f)
 229     return TRUE;
 230
 231   if (pPageObj->m_Type == PDFPAGE_PATH) {
 232     if (pGeneralState && pGeneralState->m_StrokeAlpha != 1.0f)
 233       return TRUE;
 234   }
 235
 236   if (pPageObj->m_Type == PDFPAGE_FORM) {
 237     CPDF_FormObject* pFormObj = (CPDF_FormObject*)pPageObj;
 238     if (pFormObj->m_pForm &&
 239         (pFormObj->m_pForm->m_Transparency & PDFTRANS_ISOLATED))
 240       return TRUE;
 241     if (pFormObj->m_pForm &&
 242         (!(pFormObj->m_pForm->m_Transparency & PDFTRANS_ISOLATED) &&
 243          (pFormObj->m_pForm->m_Transparency & PDFTRANS_GROUP)))
 244       return TRUE;
 245   }
 246   return FALSE;
 247 }
 248
 249 DLLEXPORT FPDF_BOOL STDCALL FPDFPage_GenerateContent(FPDF_PAGE page) {
 250   CPDF_Page* pPage = (CPDF_Page*)page;
 251   if (!pPage || !pPage->m_pFormDict || !pPage->m_pFormDict->KeyExist("Type") ||
 252       !pPage->m_pFormDict->GetElement("Type")->GetDirect() ||
 253       pPage->m_pFormDict->GetElement("Type")->GetDirect()->GetString().Compare(
 254           "Page")) {
 255     return FALSE;
 256   }
 257   CPDF_PageContentGenerate CG(pPage);
 258   CG.GenerateContent();
 259
 260   return TRUE;
 261 }
 262
 263 DLLEXPORT void STDCALL FPDFPageObj_Transform(FPDF_PAGEOBJECT page_object,
 264                                              double a,
 265                                              double b,
 266                                              double c,
 267                                              double d,
 268                                              double e,
 269                                              double f) {
 270   CPDF_PageObject* pPageObj = (CPDF_PageObject*)page_object;
 271   if (pPageObj == NULL)
 272     return;
 273
 274   CFX_AffineMatrix matrix((FX_FLOAT)a, (FX_FLOAT)b, (FX_FLOAT)c, (FX_FLOAT)d,
 275                           (FX_FLOAT)e, (FX_FLOAT)f);
 276   pPageObj->Transform(matrix);
 277 }
 278 DLLEXPORT void STDCALL FPDFPage_TransformAnnots(FPDF_PAGE page,
 279                                                 double a,
 280                                                 double b,
 281                                                 double c,
 282                                                 double d,
 283                                                 double e,
 284                                                 double f) {
 285   if (page == NULL)
 286     return;
 287   CPDF_Page* pPage = (CPDF_Page*)page;
 288   CPDF_AnnotList AnnotList(pPage);
 289   for (int i = 0; i < AnnotList.Count(); i++) {
 290     CPDF_Annot* pAnnot = AnnotList.GetAt(i);
 291     // transformAnnots Rectangle
 292     CPDF_Rect rect;
 293     pAnnot->GetRect(rect);
 294     CFX_AffineMatrix matrix((FX_FLOAT)a, (FX_FLOAT)b, (FX_FLOAT)c, (FX_FLOAT)d,
 295                             (FX_FLOAT)e, (FX_FLOAT)f);
 296     rect.Transform(&matrix);
 297     CPDF_Array* pRectArray = NULL;
 298     pRectArray = pAnnot->GetAnnotDict()->GetArray("Rect");
 299     if (!pRectArray)
 300       pRectArray = CPDF_Array::Create();
 301     pRectArray->SetAt(0, new CPDF_Number(rect.left));
 302     pRectArray->SetAt(1, new CPDF_Number(rect.bottom));
 303     pRectArray->SetAt(2, new CPDF_Number(rect.right));
 304     pRectArray->SetAt(3, new CPDF_Number(rect.top));
 305     pAnnot->GetAnnotDict()->SetAt("Rect", pRectArray);
 306
 307     // Transform AP's rectangle
 308     // To Do
 309   }
 310 }
 311
 312 DLLEXPORT void STDCALL FPDFPage_SetRotation(FPDF_PAGE page, int rotate) {
 313   CPDF_Page* pPage = (CPDF_Page*)page;
 314   if (!pPage || !pPage->m_pFormDict || !pPage->m_pFormDict->KeyExist("Type") ||
 315       !pPage->m_pFormDict->GetElement("Type")->GetDirect() ||
 316       pPage->m_pFormDict->GetElement("Type")->GetDirect()->GetString().Compare(
 317           "Page")) {
 318     return;
 319   }
 320   CPDF_Dictionary* pDict = pPage->m_pFormDict;
 321   rotate %= 4;
 322
 323   pDict->SetAt("Rotate", new CPDF_Number(rotate * 90));
 324 }