From 5110c4743751145c4ae1934cd1d83bc6c55bb43f Mon Sep 17 00:00:00 2001
From: John Abd-El-Malek <jabdelmalek@google.com>
Date: Sat, 17 May 2014 22:33:34 -0700
Subject: Initial commit.

---
 core/src/fpdfdoc/doc_formcontrol.cpp | 461 +++++++++++++++++++++++++++++++++++
 1 file changed, 461 insertions(+)
 create mode 100644 core/src/fpdfdoc/doc_formcontrol.cpp

(limited to 'core/src/fpdfdoc/doc_formcontrol.cpp')

diff --git a/core/src/fpdfdoc/doc_formcontrol.cpp b/core/src/fpdfdoc/doc_formcontrol.cpp
new file mode 100644
index 0000000000..f92b87e5f6
--- /dev/null
+++ b/core/src/fpdfdoc/doc_formcontrol.cpp
@@ -0,0 +1,461 @@
+// Copyright 2014 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 "../../include/fpdfdoc/fpdf_doc.h"
+CPDF_FormControl::CPDF_FormControl(CPDF_FormField* pField, CPDF_Dictionary* pWidgetDict)
+{
+    m_pField = pField;
+    m_pWidgetDict = pWidgetDict;
+    m_pForm = m_pField->m_pForm;
+}
+CFX_FloatRect CPDF_FormControl::GetRect()
+{
+    return m_pWidgetDict->GetRect("Rect");
+}
+CFX_ByteString CPDF_FormControl::GetOnStateName()
+{
+    ASSERT(GetType() == CPDF_FormField::CheckBox || GetType() == CPDF_FormField::RadioButton);
+    CFX_ByteString csOn;
+    CPDF_Dictionary* pAP = m_pWidgetDict->GetDict("AP");
+    if (pAP == NULL) {
+        return csOn;
+    }
+    CPDF_Dictionary* pN = pAP->GetDict("N");
+    if (pN == NULL) {
+        return csOn;
+    }
+    FX_POSITION pos = pN->GetStartPos();
+    while (pos) {
+        pN->GetNextElement(pos, csOn);
+        if (csOn != "Off") {
+            return csOn;
+        }
+    }
+    return CFX_ByteString();
+}
+void CPDF_FormControl::SetOnStateName(const CFX_ByteString& csOn)
+{
+    ASSERT(GetType() == CPDF_FormField::CheckBox || GetType() == CPDF_FormField::RadioButton);
+    CFX_ByteString csValue = csOn;
+    if (csValue.IsEmpty()) {
+        csValue = "Yes";
+    }
+    if (csValue == "Off") {
+        csValue = "Yes";
+    }
+    CFX_ByteString csAS = m_pWidgetDict->GetString("AS", "Off");
+    if (csAS != "Off") {
+        m_pWidgetDict->SetAtName("AS", csValue);
+    }
+    CPDF_Dictionary* pAP = m_pWidgetDict->GetDict("AP");
+    if (pAP == NULL) {
+        return;
+    }
+    FX_POSITION pos1 = pAP->GetStartPos();
+    while (pos1) {
+        CFX_ByteString csKey1;
+        CPDF_Object* pObj1 = pAP->GetNextElement(pos1, csKey1);
+        if (pObj1 == NULL) {
+            continue;
+        }
+        CPDF_Object* pObjDirect1 = pObj1->GetDirect();
+        if (pObjDirect1->GetType() != PDFOBJ_DICTIONARY) {
+            continue;
+        }
+        CPDF_Dictionary* pSubDict = (CPDF_Dictionary*)pObjDirect1;
+        FX_POSITION pos2 = pSubDict->GetStartPos();
+        while (pos2) {
+            CFX_ByteString csKey2;
+            CPDF_Object* pObj2 = pSubDict->GetNextElement(pos2, csKey2);
+            if (pObj2 == NULL) {
+                continue;
+            }
+            if (csKey2 != "Off") {
+                pSubDict->ReplaceKey(csKey2, csValue);
+                break;
+            }
+        }
+    }
+}
+CFX_ByteString CPDF_FormControl::GetCheckedAPState()
+{
+    ASSERT(GetType() == CPDF_FormField::CheckBox || GetType() == CPDF_FormField::RadioButton);
+    CFX_ByteString csOn = GetOnStateName();
+    if (GetType() == CPDF_FormField::RadioButton || GetType() == CPDF_FormField::CheckBox) {
+        CPDF_Object* pOpt = FPDF_GetFieldAttr(m_pField->m_pDict, "Opt");
+        if (pOpt != NULL && pOpt->GetType() == PDFOBJ_ARRAY) {
+            int iIndex = m_pField->GetControlIndex(this);
+            csOn.Format("%d", iIndex);
+        }
+    }
+    if (csOn.IsEmpty()) {
+        csOn = "Yes";
+    }
+    return csOn;
+}
+CFX_WideString CPDF_FormControl::GetExportValue()
+{
+    ASSERT(GetType() == CPDF_FormField::CheckBox || GetType() == CPDF_FormField::RadioButton);
+    CFX_ByteString csOn = GetOnStateName();
+    if (GetType() == CPDF_FormField::RadioButton || GetType() == CPDF_FormField::CheckBox) {
+        CPDF_Object* pOpt = FPDF_GetFieldAttr(m_pField->m_pDict, "Opt");
+        if (pOpt != NULL && pOpt->GetType() == PDFOBJ_ARRAY) {
+            int iIndex = m_pField->GetControlIndex(this);
+            csOn = ((CPDF_Array*)pOpt)->GetString(iIndex);
+        }
+    }
+    if (csOn.IsEmpty()) {
+        csOn = "Yes";
+    }
+    CFX_WideString csWOn = PDF_DecodeText(csOn);
+    return csWOn;
+}
+FX_BOOL CPDF_FormControl::IsChecked()
+{
+    ASSERT(GetType() == CPDF_FormField::CheckBox || GetType() == CPDF_FormField::RadioButton);
+    CFX_ByteString csOn = GetOnStateName();
+    CFX_ByteString csAS = m_pWidgetDict->GetString("AS");
+    return csAS == csOn;
+}
+FX_BOOL CPDF_FormControl::IsDefaultChecked()
+{
+    ASSERT(GetType() == CPDF_FormField::CheckBox || GetType() == CPDF_FormField::RadioButton);
+    CPDF_Object* pDV = FPDF_GetFieldAttr(m_pField->m_pDict, "DV");
+    if (pDV == NULL) {
+        return FALSE;
+    }
+    CFX_ByteString csDV = pDV->GetString();
+    CFX_ByteString csOn = GetOnStateName();
+    return (csDV == csOn);
+}
+void CPDF_FormControl::CheckControl(FX_BOOL bChecked)
+{
+    ASSERT(GetType() == CPDF_FormField::CheckBox || GetType() == CPDF_FormField::RadioButton);
+    CFX_ByteString csOn = GetOnStateName();
+    CFX_ByteString csOldAS = m_pWidgetDict->GetString("AS", "Off");
+    CFX_ByteString csAS = "Off";
+    if (bChecked) {
+        csAS = csOn;
+    }
+    if (csOldAS == csAS) {
+        return;
+    }
+    m_pWidgetDict->SetAtName("AS", csAS);
+    m_pForm->m_bUpdated = TRUE;
+}
+CPDF_Stream* FPDFDOC_GetAnnotAP(CPDF_Dictionary* pAnnotDict, CPDF_Annot::AppearanceMode mode);
+void CPDF_FormControl::DrawControl(CFX_RenderDevice* pDevice, CFX_AffineMatrix* pMatrix, CPDF_Page* pPage,
+                                   CPDF_Annot::AppearanceMode mode, const CPDF_RenderOptions* pOptions)
+{
+    if (m_pWidgetDict->GetInteger("F") & ANNOTFLAG_HIDDEN) {
+        return;
+    }
+    CPDF_Stream* pStream = FPDFDOC_GetAnnotAP(m_pWidgetDict, mode);
+    if (pStream == NULL) {
+        return;
+    }
+    CFX_FloatRect form_bbox = pStream->GetDict()->GetRect("BBox");
+    CFX_AffineMatrix form_matrix = pStream->GetDict()->GetMatrix("Matrix");
+    form_matrix.TransformRect(form_bbox);
+    CFX_FloatRect arect = m_pWidgetDict->GetRect("Rect");
+    CFX_AffineMatrix matrix;
+    matrix.MatchRect(arect, form_bbox);
+    matrix.Concat(*pMatrix);
+    CPDF_Form form(m_pField->m_pForm->m_pDocument, m_pField->m_pForm->m_pFormDict->GetDict("DR"), pStream);
+    form.ParseContent(NULL, NULL, NULL, NULL);
+    CPDF_RenderContext context;
+    context.Create(pPage);
+    context.DrawObjectList(pDevice, &form, &matrix, pOptions);
+}
+const FX_CHAR* g_sHighlightingMode[] = {"N", "I", "O", "P", "T", ""};
+CPDF_FormControl::HighlightingMode CPDF_FormControl::GetHighlightingMode()
+{
+    if (m_pWidgetDict == NULL) {
+        return Invert;
+    }
+    CFX_ByteString csH = m_pWidgetDict->GetString("H", "I");
+    int i = 0;
+    while (g_sHighlightingMode[i][0] != '\0') {
+        if (csH.Equal(g_sHighlightingMode[i])) {
+            return (HighlightingMode)i;
+        }
+        i ++;
+    }
+    return Invert;
+}
+CPDF_ApSettings CPDF_FormControl::GetMK(FX_BOOL bCreate)
+{
+    if (!m_pWidgetDict) {
+        return NULL;
+    }
+    CPDF_ApSettings mk = m_pWidgetDict->GetDict(FX_BSTRC("MK"));
+    if (!mk && bCreate) {
+        mk = CPDF_Dictionary::Create();
+        if (mk == NULL) {
+            return NULL;
+        }
+        m_pWidgetDict->SetAt(FX_BSTRC("MK"), mk);
+    }
+    return mk;
+}
+FX_BOOL CPDF_FormControl::HasMKEntry(CFX_ByteString csEntry)
+{
+    CPDF_ApSettings mk = GetMK(FALSE);
+    return mk.HasMKEntry(csEntry);
+}
+int CPDF_FormControl::GetRotation()
+{
+    CPDF_ApSettings mk = GetMK(FALSE);
+    return mk.GetRotation();
+}
+FX_ARGB CPDF_FormControl::GetColor(int& iColorType, CFX_ByteString csEntry)
+{
+    CPDF_ApSettings mk = GetMK(FALSE);
+    return mk.GetColor(iColorType, csEntry);
+}
+FX_FLOAT CPDF_FormControl::GetOriginalColor(int index, CFX_ByteString csEntry)
+{
+    CPDF_ApSettings mk = GetMK(FALSE);
+    return mk.GetOriginalColor(index, csEntry);
+}
+void CPDF_FormControl::GetOriginalColor(int& iColorType, FX_FLOAT fc[4], CFX_ByteString csEntry)
+{
+    CPDF_ApSettings mk = GetMK(FALSE);
+    mk.GetOriginalColor(iColorType, fc, csEntry);
+}
+CFX_WideString CPDF_FormControl::GetCaption(CFX_ByteString csEntry)
+{
+    CPDF_ApSettings mk = GetMK(FALSE);
+    return mk.GetCaption(csEntry);
+}
+CPDF_Stream* CPDF_FormControl::GetIcon(CFX_ByteString csEntry)
+{
+    CPDF_ApSettings mk = GetMK(FALSE);
+    return mk.GetIcon(csEntry);
+}
+CPDF_IconFit CPDF_FormControl::GetIconFit()
+{
+    CPDF_ApSettings mk = GetMK(FALSE);
+    return mk.GetIconFit();
+}
+int CPDF_FormControl::GetTextPosition()
+{
+    CPDF_ApSettings mk = GetMK(FALSE);
+    return mk.GetTextPosition();
+}
+CPDF_Action CPDF_FormControl::GetAction()
+{
+    if (m_pWidgetDict == NULL) {
+        return NULL;
+    }
+    if (m_pWidgetDict->KeyExist("A")) {
+        return m_pWidgetDict->GetDict("A");
+    } else {
+        CPDF_Object* pObj = FPDF_GetFieldAttr(m_pField->m_pDict, "A");
+        if (pObj == NULL) {
+            return NULL;
+        }
+        return pObj->GetDict();
+    }
+}
+CPDF_AAction CPDF_FormControl::GetAdditionalAction()
+{
+    if (m_pWidgetDict == NULL) {
+        return NULL;
+    }
+    if (m_pWidgetDict->KeyExist("AA")) {
+        return m_pWidgetDict->GetDict("AA");
+    } else {
+        return m_pField->GetAdditionalAction();
+    }
+}
+CPDF_DefaultAppearance CPDF_FormControl::GetDefaultAppearance()
+{
+    if (m_pWidgetDict == NULL) {
+        return CFX_ByteString();
+    }
+    if (m_pWidgetDict->KeyExist("DA")) {
+        return m_pWidgetDict->GetString("DA");
+    } else {
+        CPDF_Object* pObj = FPDF_GetFieldAttr(m_pField->m_pDict, "DA");
+        if (pObj == NULL) {
+            return m_pField->m_pForm->GetDefaultAppearance();
+        }
+        return pObj->GetString();
+    }
+}
+CPDF_Font* CPDF_FormControl::GetDefaultControlFont()
+{
+    CPDF_DefaultAppearance cDA = GetDefaultAppearance();
+    CFX_ByteString csFontNameTag;
+    FX_FLOAT fFontSize;
+    cDA.GetFont(csFontNameTag, fFontSize);
+    if (csFontNameTag.IsEmpty()) {
+        return NULL;
+    }
+    CPDF_Object* pObj = FPDF_GetFieldAttr(m_pWidgetDict, "DR");
+    if (pObj != NULL && pObj->GetType() == PDFOBJ_DICTIONARY) {
+        CPDF_Dictionary* pFonts = ((CPDF_Dictionary*)pObj)->GetDict("Font");
+        if (pFonts != NULL) {
+            CPDF_Dictionary *pElement = pFonts->GetDict(csFontNameTag);
+            CPDF_Font *pFont = m_pField->m_pForm->m_pDocument->LoadFont(pElement);
+            if (pFont != NULL) {
+                return pFont;
+            }
+        }
+    }
+    CPDF_Font *pFont = m_pField->m_pForm->GetFormFont(csFontNameTag);
+    if (pFont != NULL) {
+        return pFont;
+    }
+    CPDF_Dictionary *pPageDict = m_pWidgetDict->GetDict("P");
+    pObj = FPDF_GetFieldAttr(pPageDict, "Resources");
+    if (pObj != NULL && pObj->GetType() == PDFOBJ_DICTIONARY) {
+        CPDF_Dictionary* pFonts = ((CPDF_Dictionary*)pObj)->GetDict("Font");
+        if (pFonts != NULL) {
+            CPDF_Dictionary *pElement = pFonts->GetDict(csFontNameTag);
+            CPDF_Font *pFont = m_pField->m_pForm->m_pDocument->LoadFont(pElement);
+            if (pFont != NULL) {
+                return pFont;
+            }
+        }
+    }
+    return NULL;
+}
+int CPDF_FormControl::GetControlAlignment()
+{
+    if (m_pWidgetDict == NULL) {
+        return 0;
+    }
+    if (m_pWidgetDict->KeyExist("Q")) {
+        return m_pWidgetDict->GetInteger("Q", 0);
+    } else {
+        CPDF_Object* pObj = FPDF_GetFieldAttr(m_pField->m_pDict, "Q");
+        if (pObj == NULL) {
+            return m_pField->m_pForm->GetFormAlignment();
+        }
+        return pObj->GetInteger();
+    }
+}
+FX_BOOL CPDF_ApSettings::HasMKEntry(FX_BSTR csEntry)
+{
+    if (m_pDict == NULL) {
+        return FALSE;
+    }
+    return m_pDict->KeyExist(csEntry);
+}
+int CPDF_ApSettings::GetRotation()
+{
+    if (m_pDict == NULL) {
+        return 0;
+    }
+    return m_pDict->GetInteger(FX_BSTRC("R"));
+}
+FX_ARGB CPDF_ApSettings::GetColor(int& iColorType, FX_BSTR csEntry)
+{
+    iColorType = COLORTYPE_TRANSPARENT;
+    if (m_pDict == NULL) {
+        return 0;
+    }
+    FX_ARGB color = 0;
+    CPDF_Array* pEntry = m_pDict->GetArray(csEntry);
+    if (pEntry == NULL) {
+        return color;
+    }
+    FX_DWORD dwCount = pEntry->GetCount();
+    if (dwCount == 1) {
+        iColorType = COLORTYPE_GRAY;
+        FX_FLOAT g = pEntry->GetNumber(0) * 255;
+        color = ArgbEncode(255, (int)g, (int)g, (int)g);
+    } else if (dwCount == 3) {
+        iColorType = COLORTYPE_RGB;
+        FX_FLOAT r = pEntry->GetNumber(0) * 255;
+        FX_FLOAT g = pEntry->GetNumber(1) * 255;
+        FX_FLOAT b = pEntry->GetNumber(2) * 255;
+        color = ArgbEncode(255, (int)r, (int)g, (int)b);
+    } else if (dwCount == 4) {
+        iColorType = COLORTYPE_CMYK;
+        FX_FLOAT c = pEntry->GetNumber(0);
+        FX_FLOAT m = pEntry->GetNumber(1);
+        FX_FLOAT y = pEntry->GetNumber(2);
+        FX_FLOAT k = pEntry->GetNumber(3);
+        FX_FLOAT r = 1.0f - FX_MIN(1.0f, c + k);
+        FX_FLOAT g = 1.0f - FX_MIN(1.0f, m + k);
+        FX_FLOAT b = 1.0f - FX_MIN(1.0f, y + k);
+        color = ArgbEncode(255, (int)(r * 255), (int)(g * 255), (int)(b * 255));
+    }
+    return color;
+}
+FX_FLOAT CPDF_ApSettings::GetOriginalColor(int index, FX_BSTR csEntry)
+{
+    if (m_pDict == NULL) {
+        return 0;
+    }
+    CPDF_Array* pEntry = m_pDict->GetArray(csEntry);
+    if (pEntry != NULL) {
+        return pEntry->GetNumber(index);
+    }
+    return 0;
+}
+void CPDF_ApSettings::GetOriginalColor(int& iColorType, FX_FLOAT fc[4], FX_BSTR csEntry)
+{
+    iColorType = COLORTYPE_TRANSPARENT;
+    for (int i = 0; i < 4; i ++) {
+        fc[i] = 0;
+    }
+    if (m_pDict == NULL) {
+        return;
+    }
+    CPDF_Array* pEntry = m_pDict->GetArray(csEntry);
+    if (pEntry == NULL) {
+        return;
+    }
+    FX_DWORD dwCount = pEntry->GetCount();
+    if (dwCount == 1) {
+        iColorType = COLORTYPE_GRAY;
+        fc[0] = pEntry->GetNumber(0);
+    } else if (dwCount == 3) {
+        iColorType = COLORTYPE_RGB;
+        fc[0] = pEntry->GetNumber(0);
+        fc[1] = pEntry->GetNumber(1);
+        fc[2] = pEntry->GetNumber(2);
+    } else if (dwCount == 4) {
+        iColorType = COLORTYPE_CMYK;
+        fc[0] = pEntry->GetNumber(0);
+        fc[1] = pEntry->GetNumber(1);
+        fc[2] = pEntry->GetNumber(2);
+        fc[3] = pEntry->GetNumber(3);
+    }
+}
+CFX_WideString CPDF_ApSettings::GetCaption(FX_BSTR csEntry)
+{
+    CFX_WideString csCaption;
+    if (m_pDict == NULL) {
+        return csCaption;
+    }
+    return m_pDict->GetUnicodeText(csEntry);
+}
+CPDF_Stream* CPDF_ApSettings::GetIcon(FX_BSTR csEntry)
+{
+    if (m_pDict == NULL) {
+        return NULL;
+    }
+    return m_pDict->GetStream(csEntry);
+}
+CPDF_IconFit CPDF_ApSettings::GetIconFit()
+{
+    if (m_pDict == NULL) {
+        return NULL;
+    }
+    return m_pDict->GetDict(FX_BSTRC("IF"));
+}
+int CPDF_ApSettings::GetTextPosition()
+{
+    if (m_pDict == NULL) {
+        return TEXTPOS_CAPTION;
+    }
+    return m_pDict->GetInteger(FX_BSTRC("TP"), TEXTPOS_CAPTION);
+}
-- 
cgit v1.2.3