Index: fpdfsdk/cpdfsdk_document.cpp |
diff --git a/fpdfsdk/cpdfsdk_document.cpp b/fpdfsdk/cpdfsdk_document.cpp |
new file mode 100644 |
index 0000000000000000000000000000000000000000..232adbaba52381aa308a9a229b8474c03ea185a2 |
--- /dev/null |
+++ b/fpdfsdk/cpdfsdk_document.cpp |
@@ -0,0 +1,261 @@ |
+// Copyright 2016 PDFium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
+ |
+#include "fpdfsdk/include/cpdfsdk_document.h" |
+ |
+#include "core/fpdfapi/fpdf_parser/include/cpdf_array.h" |
+#include "core/fpdfapi/fpdf_parser/include/cpdf_dictionary.h" |
+#include "core/fpdfapi/fpdf_parser/include/cpdf_document.h" |
+#include "core/fpdfapi/fpdf_parser/include/cpdf_object.h" |
+#include "core/fpdfdoc/include/cpdf_action.h" |
+#include "core/fpdfdoc/include/cpdf_docjsactions.h" |
+#include "core/fpdfdoc/include/cpdf_occontext.h" |
+#include "fpdfsdk/include/cpdfdoc_environment.h" |
+#include "fpdfsdk/include/cpdfsdk_annot.h" |
+#include "fpdfsdk/include/cpdfsdk_annothandlermgr.h" |
+#include "fpdfsdk/include/cpdfsdk_interform.h" |
+#include "fpdfsdk/include/cpdfsdk_pageview.h" |
+#include "fpdfsdk/include/cpdfsdk_widget.h" |
+#include "fpdfsdk/include/fsdk_actionhandler.h" |
+ |
+// static |
+CPDFSDK_Document* CPDFSDK_Document::FromFPDFFormHandle( |
+ FPDF_FORMHANDLE hHandle) { |
+ CPDFDoc_Environment* pEnv = static_cast<CPDFDoc_Environment*>(hHandle); |
+ return pEnv ? pEnv->GetSDKDocument() : nullptr; |
+} |
+ |
+CPDFSDK_Document::CPDFSDK_Document(UnderlyingDocumentType* pDoc, |
+ CPDFDoc_Environment* pEnv) |
+ : m_pDoc(pDoc), |
+ m_pFocusAnnot(nullptr), |
+ m_pEnv(pEnv), |
+ m_bChangeMask(FALSE), |
+ m_bBeingDestroyed(FALSE) {} |
+ |
+CPDFSDK_Document::~CPDFSDK_Document() { |
+ m_bBeingDestroyed = TRUE; |
+ |
+ for (auto& it : m_pageMap) |
+ it.second->KillFocusAnnotIfNeeded(); |
+ |
+ for (auto& it : m_pageMap) |
+ delete it.second; |
+ m_pageMap.clear(); |
+} |
+ |
+CPDFSDK_PageView* CPDFSDK_Document::GetPageView( |
+ UnderlyingPageType* pUnderlyingPage, |
+ bool ReNew) { |
+ auto it = m_pageMap.find(pUnderlyingPage); |
+ if (it != m_pageMap.end()) |
+ return it->second; |
+ |
+ if (!ReNew) |
+ return nullptr; |
+ |
+ CPDFSDK_PageView* pPageView = new CPDFSDK_PageView(this, pUnderlyingPage); |
+ m_pageMap[pUnderlyingPage] = pPageView; |
+ // Delay to load all the annotations, to avoid endless loop. |
+ pPageView->LoadFXAnnots(); |
+ return pPageView; |
+} |
+ |
+CPDFSDK_PageView* CPDFSDK_Document::GetCurrentView() { |
+ UnderlyingPageType* pPage = |
+ UnderlyingFromFPDFPage(m_pEnv->FFI_GetCurrentPage(m_pDoc)); |
+ return pPage ? GetPageView(pPage, true) : nullptr; |
+} |
+ |
+CPDFSDK_PageView* CPDFSDK_Document::GetPageView(int nIndex) { |
+ UnderlyingPageType* pTempPage = |
+ UnderlyingFromFPDFPage(m_pEnv->FFI_GetPage(m_pDoc, nIndex)); |
+ if (!pTempPage) |
+ return nullptr; |
+ |
+ auto it = m_pageMap.find(pTempPage); |
+ return it != m_pageMap.end() ? it->second : nullptr; |
+} |
+ |
+void CPDFSDK_Document::ProcJavascriptFun() { |
+ CPDF_Document* pPDFDoc = GetPDFDocument(); |
+ CPDF_DocJSActions docJS(pPDFDoc); |
+ int iCount = docJS.CountJSActions(); |
+ if (iCount < 1) |
+ return; |
+ for (int i = 0; i < iCount; i++) { |
+ CFX_ByteString csJSName; |
+ CPDF_Action jsAction = docJS.GetJSAction(i, csJSName); |
+ if (m_pEnv->GetActionHander()) |
+ m_pEnv->GetActionHander()->DoAction_JavaScript( |
+ jsAction, CFX_WideString::FromLocal(csJSName.AsStringC()), this); |
+ } |
+} |
+ |
+FX_BOOL CPDFSDK_Document::ProcOpenAction() { |
+ if (!m_pDoc) |
+ return FALSE; |
+ |
+ CPDF_Dictionary* pRoot = GetPDFDocument()->GetRoot(); |
+ if (!pRoot) |
+ return FALSE; |
+ |
+ CPDF_Object* pOpenAction = pRoot->GetDictBy("OpenAction"); |
+ if (!pOpenAction) |
+ pOpenAction = pRoot->GetArrayBy("OpenAction"); |
+ |
+ if (!pOpenAction) |
+ return FALSE; |
+ |
+ if (pOpenAction->IsArray()) |
+ return TRUE; |
+ |
+ if (CPDF_Dictionary* pDict = pOpenAction->AsDictionary()) { |
+ CPDF_Action action(pDict); |
+ if (m_pEnv->GetActionHander()) |
+ m_pEnv->GetActionHander()->DoAction_DocOpen(action, this); |
+ return TRUE; |
+ } |
+ return FALSE; |
+} |
+ |
+CPDF_OCContext* CPDFSDK_Document::GetOCContext() { |
+ if (!m_pOccontent) { |
+ m_pOccontent.reset( |
+ new CPDF_OCContext(GetPDFDocument(), CPDF_OCContext::View)); |
+ } |
+ return m_pOccontent.get(); |
+} |
+ |
+void CPDFSDK_Document::RemovePageView(UnderlyingPageType* pUnderlyingPage) { |
+ auto it = m_pageMap.find(pUnderlyingPage); |
+ if (it == m_pageMap.end()) |
+ return; |
+ |
+ CPDFSDK_PageView* pPageView = it->second; |
+ if (pPageView->IsLocked()) |
+ return; |
+ |
+ // This must happen before we remove |pPageView| from the map because |
+ // |KillFocusAnnotIfNeeded| can call into the |GetPage| method which will |
+ // look for this page view in the map, if it doesn't find it a new one will |
+ // be created. We then have two page views pointing to the same page and |
+ // bad things happen. |
+ pPageView->KillFocusAnnotIfNeeded(); |
+ |
+ // Remove the page from the map to make sure we don't accidentally attempt |
+ // to use the |pPageView| while we're cleaning it up. |
+ m_pageMap.erase(it); |
+ |
+ delete pPageView; |
+} |
+ |
+UnderlyingPageType* CPDFSDK_Document::GetPage(int nIndex) { |
+ return UnderlyingFromFPDFPage(m_pEnv->FFI_GetPage(m_pDoc, nIndex)); |
+} |
+ |
+CPDFSDK_InterForm* CPDFSDK_Document::GetInterForm() { |
+ if (!m_pInterForm) |
+ m_pInterForm.reset(new CPDFSDK_InterForm(this)); |
+ return m_pInterForm.get(); |
+} |
+ |
+void CPDFSDK_Document::UpdateAllViews(CPDFSDK_PageView* pSender, |
+ CPDFSDK_Annot* pAnnot) { |
+ for (const auto& it : m_pageMap) { |
+ CPDFSDK_PageView* pPageView = it.second; |
+ if (pPageView != pSender) { |
+ pPageView->UpdateView(pAnnot); |
+ } |
+ } |
+} |
+ |
+CPDFSDK_Annot* CPDFSDK_Document::GetFocusAnnot() { |
+ return m_pFocusAnnot; |
+} |
+ |
+FX_BOOL CPDFSDK_Document::SetFocusAnnot(CPDFSDK_Annot* pAnnot, FX_UINT nFlag) { |
+ if (m_bBeingDestroyed) |
+ return FALSE; |
+ |
+ if (m_pFocusAnnot == pAnnot) |
+ return TRUE; |
+ |
+ if (m_pFocusAnnot) { |
+ if (!KillFocusAnnot(nFlag)) |
+ return FALSE; |
+ } |
+ |
+ if (!pAnnot) |
+ return FALSE; |
+ |
+#ifdef PDF_ENABLE_XFA |
+ CPDFSDK_Annot* pLastFocusAnnot = m_pFocusAnnot; |
+#endif // PDF_ENABLE_XFA |
+ CPDFSDK_PageView* pPageView = pAnnot->GetPageView(); |
+ if (pPageView && pPageView->IsValid()) { |
+ CPDFSDK_AnnotHandlerMgr* pAnnotHandler = m_pEnv->GetAnnotHandlerMgr(); |
+ if (!m_pFocusAnnot) { |
+#ifdef PDF_ENABLE_XFA |
+ if (!pAnnotHandler->Annot_OnChangeFocus(pAnnot, pLastFocusAnnot)) |
+ return FALSE; |
+#endif // PDF_ENABLE_XFA |
+ if (!pAnnotHandler->Annot_OnSetFocus(pAnnot, nFlag)) |
+ return FALSE; |
+ if (!m_pFocusAnnot) { |
+ m_pFocusAnnot = pAnnot; |
+ return TRUE; |
+ } |
+ } |
+ } |
+ return FALSE; |
+} |
+ |
+FX_BOOL CPDFSDK_Document::KillFocusAnnot(FX_UINT nFlag) { |
+ if (m_pFocusAnnot) { |
+ CPDFSDK_AnnotHandlerMgr* pAnnotHandler = m_pEnv->GetAnnotHandlerMgr(); |
+ CPDFSDK_Annot* pFocusAnnot = m_pFocusAnnot; |
+ m_pFocusAnnot = nullptr; |
+ |
+#ifdef PDF_ENABLE_XFA |
+ if (!pAnnotHandler->Annot_OnChangeFocus(nullptr, pFocusAnnot)) |
+ return FALSE; |
+#endif // PDF_ENABLE_XFA |
+ |
+ if (pAnnotHandler->Annot_OnKillFocus(pFocusAnnot, nFlag)) { |
+ if (pFocusAnnot->GetAnnotSubtype() == CPDF_Annot::Subtype::WIDGET) { |
+ CPDFSDK_Widget* pWidget = (CPDFSDK_Widget*)pFocusAnnot; |
+ int nFieldType = pWidget->GetFieldType(); |
+ if (FIELDTYPE_TEXTFIELD == nFieldType || |
+ FIELDTYPE_COMBOBOX == nFieldType) { |
+ m_pEnv->FFI_OnSetFieldInputFocus(nullptr, nullptr, 0, FALSE); |
+ } |
+ } |
+ |
+ if (!m_pFocusAnnot) |
+ return TRUE; |
+ } else { |
+ m_pFocusAnnot = pFocusAnnot; |
+ } |
+ } |
+ return FALSE; |
+} |
+ |
+void CPDFSDK_Document::OnCloseDocument() { |
+ KillFocusAnnot(); |
+} |
+ |
+FX_BOOL CPDFSDK_Document::GetPermissions(int nFlag) { |
+ return GetPDFDocument()->GetUserPermissions() & nFlag; |
+} |
+ |
+IJS_Runtime* CPDFSDK_Document::GetJsRuntime() { |
+ return m_pEnv->GetJSRuntime(); |
+} |
+ |
+CFX_WideString CPDFSDK_Document::GetPath() { |
+ return m_pEnv->JS_docGetFilePath(); |
+} |