// 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); | |
} |