// 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 <algorithm>
|
|
#include "core/fpdfapi/fpdf_page/include/cpdf_form.h"
|
#include "core/fpdfapi/fpdf_parser/include/cpdf_array.h"
|
#include "core/fpdfapi/fpdf_parser/include/cpdf_document.h"
|
#include "core/fpdfapi/fpdf_parser/include/cpdf_stream.h"
|
#include "core/fpdfapi/fpdf_render/include/cpdf_rendercontext.h"
|
#include "core/fpdfdoc/include/fpdf_doc.h"
|
|
namespace {
|
|
const FX_CHAR* const g_sHighlightingMode[] = {
|
// Must match order of HighlightingMode enum.
|
"N", "I", "O", "P", "T"};
|
|
} // namespace
|
|
CPDF_FormControl::CPDF_FormControl(CPDF_FormField* pField,
|
CPDF_Dictionary* pWidgetDict)
|
: m_pField(pField),
|
m_pWidgetDict(pWidgetDict),
|
m_pForm(m_pField->m_pForm) {}
|
|
CFX_ByteString CPDF_FormControl::GetOnStateName() const {
|
ASSERT(GetType() == CPDF_FormField::CheckBox ||
|
GetType() == CPDF_FormField::RadioButton);
|
CFX_ByteString csOn;
|
CPDF_Dictionary* pAP = m_pWidgetDict->GetDictBy("AP");
|
if (!pAP)
|
return csOn;
|
|
CPDF_Dictionary* pN = pAP->GetDictBy("N");
|
if (!pN)
|
return csOn;
|
|
for (const auto& it : *pN) {
|
if (it.first != "Off")
|
return it.first;
|
}
|
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->GetStringBy("AS", "Off");
|
if (csAS != "Off") {
|
m_pWidgetDict->SetAtName("AS", csValue);
|
}
|
CPDF_Dictionary* pAP = m_pWidgetDict->GetDictBy("AP");
|
if (!pAP) {
|
return;
|
}
|
for (const auto& it : *pAP) {
|
CPDF_Object* pObj1 = it.second;
|
if (!pObj1) {
|
continue;
|
}
|
CPDF_Object* pObjDirect1 = pObj1->GetDirect();
|
CPDF_Dictionary* pSubDict = pObjDirect1->AsDictionary();
|
if (!pSubDict)
|
continue;
|
|
auto subdict_it = pSubDict->begin();
|
while (subdict_it != pSubDict->end()) {
|
const CFX_ByteString& csKey2 = subdict_it->first;
|
CPDF_Object* pObj2 = subdict_it->second;
|
++subdict_it;
|
if (!pObj2) {
|
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) {
|
if (ToArray(FPDF_GetFieldAttr(m_pField->m_pDict, "Opt"))) {
|
int iIndex = m_pField->GetControlIndex(this);
|
csOn.Format("%d", iIndex);
|
}
|
}
|
if (csOn.IsEmpty())
|
csOn = "Yes";
|
return csOn;
|
}
|
|
CFX_WideString CPDF_FormControl::GetExportValue() const {
|
ASSERT(GetType() == CPDF_FormField::CheckBox ||
|
GetType() == CPDF_FormField::RadioButton);
|
CFX_ByteString csOn = GetOnStateName();
|
if (GetType() == CPDF_FormField::RadioButton ||
|
GetType() == CPDF_FormField::CheckBox) {
|
if (CPDF_Array* pArray =
|
ToArray(FPDF_GetFieldAttr(m_pField->m_pDict, "Opt"))) {
|
int iIndex = m_pField->GetControlIndex(this);
|
csOn = pArray->GetStringAt(iIndex);
|
}
|
}
|
if (csOn.IsEmpty())
|
csOn = "Yes";
|
return PDF_DecodeText(csOn);
|
}
|
|
bool CPDF_FormControl::IsChecked() const {
|
ASSERT(GetType() == CPDF_FormField::CheckBox ||
|
GetType() == CPDF_FormField::RadioButton);
|
CFX_ByteString csOn = GetOnStateName();
|
CFX_ByteString csAS = m_pWidgetDict->GetStringBy("AS");
|
return csAS == csOn;
|
}
|
|
bool CPDF_FormControl::IsDefaultChecked() const {
|
ASSERT(GetType() == CPDF_FormField::CheckBox ||
|
GetType() == CPDF_FormField::RadioButton);
|
CPDF_Object* pDV = FPDF_GetFieldAttr(m_pField->m_pDict, "DV");
|
if (!pDV) {
|
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->GetStringBy("AS", "Off");
|
CFX_ByteString csAS = "Off";
|
if (bChecked)
|
csAS = csOn;
|
if (csOldAS == csAS)
|
return;
|
m_pWidgetDict->SetAtName("AS", csAS);
|
}
|
|
void CPDF_FormControl::DrawControl(CFX_RenderDevice* pDevice,
|
CFX_Matrix* pMatrix,
|
CPDF_Page* pPage,
|
CPDF_Annot::AppearanceMode mode,
|
const CPDF_RenderOptions* pOptions) {
|
if (m_pWidgetDict->GetIntegerBy("F") & ANNOTFLAG_HIDDEN)
|
return;
|
|
CPDF_Stream* pStream = FPDFDOC_GetAnnotAP(m_pWidgetDict, mode);
|
if (!pStream)
|
return;
|
|
CFX_FloatRect form_bbox = pStream->GetDict()->GetRectBy("BBox");
|
CFX_Matrix form_matrix = pStream->GetDict()->GetMatrixBy("Matrix");
|
form_matrix.TransformRect(form_bbox);
|
CFX_FloatRect arect = m_pWidgetDict->GetRectBy("Rect");
|
CFX_Matrix matrix;
|
matrix.MatchRect(arect, form_bbox);
|
matrix.Concat(*pMatrix);
|
CPDF_Form form(m_pField->m_pForm->m_pDocument,
|
m_pField->m_pForm->m_pFormDict->GetDictBy("DR"), pStream);
|
form.ParseContent(nullptr, nullptr, nullptr);
|
CPDF_RenderContext context(pPage);
|
context.AppendLayer(&form, &matrix);
|
context.Render(pDevice, pOptions, nullptr);
|
}
|
|
CPDF_FormControl::HighlightingMode CPDF_FormControl::GetHighlightingMode() {
|
if (!m_pWidgetDict)
|
return Invert;
|
|
CFX_ByteString csH = m_pWidgetDict->GetStringBy("H", "I");
|
for (size_t i = 0; i < FX_ArraySize(g_sHighlightingMode); ++i) {
|
if (csH == g_sHighlightingMode[i])
|
return static_cast<HighlightingMode>(i);
|
}
|
return Invert;
|
}
|
|
CPDF_ApSettings CPDF_FormControl::GetMK() const {
|
return CPDF_ApSettings(m_pWidgetDict ? m_pWidgetDict->GetDictBy("MK")
|
: nullptr);
|
}
|
|
bool CPDF_FormControl::HasMKEntry(const CFX_ByteString& csEntry) const {
|
return GetMK().HasMKEntry(csEntry);
|
}
|
|
int CPDF_FormControl::GetRotation() {
|
return GetMK().GetRotation();
|
}
|
|
FX_ARGB CPDF_FormControl::GetColor(int& iColorType,
|
const CFX_ByteString& csEntry) {
|
return GetMK().GetColor(iColorType, csEntry);
|
}
|
|
FX_FLOAT CPDF_FormControl::GetOriginalColor(int index,
|
const CFX_ByteString& csEntry) {
|
return GetMK().GetOriginalColor(index, csEntry);
|
}
|
|
void CPDF_FormControl::GetOriginalColor(int& iColorType,
|
FX_FLOAT fc[4],
|
const CFX_ByteString& csEntry) {
|
GetMK().GetOriginalColor(iColorType, fc, csEntry);
|
}
|
CFX_WideString CPDF_FormControl::GetCaption(const CFX_ByteString& csEntry) {
|
return GetMK().GetCaption(csEntry);
|
}
|
|
CPDF_Stream* CPDF_FormControl::GetIcon(const CFX_ByteString& csEntry) {
|
return GetMK().GetIcon(csEntry);
|
}
|
|
CPDF_IconFit CPDF_FormControl::GetIconFit() {
|
return GetMK().GetIconFit();
|
}
|
|
int CPDF_FormControl::GetTextPosition() {
|
return GetMK().GetTextPosition();
|
}
|
|
CPDF_Action CPDF_FormControl::GetAction() {
|
if (!m_pWidgetDict)
|
return CPDF_Action();
|
|
if (m_pWidgetDict->KeyExist("A"))
|
return CPDF_Action(m_pWidgetDict->GetDictBy("A"));
|
|
CPDF_Object* pObj = FPDF_GetFieldAttr(m_pField->m_pDict, "A");
|
if (!pObj)
|
return CPDF_Action();
|
|
return CPDF_Action(pObj->GetDict());
|
}
|
|
CPDF_AAction CPDF_FormControl::GetAdditionalAction() {
|
if (!m_pWidgetDict)
|
return CPDF_AAction();
|
|
if (m_pWidgetDict->KeyExist("AA"))
|
return CPDF_AAction(m_pWidgetDict->GetDictBy("AA"));
|
return m_pField->GetAdditionalAction();
|
}
|
|
CPDF_DefaultAppearance CPDF_FormControl::GetDefaultAppearance() {
|
if (!m_pWidgetDict)
|
return CPDF_DefaultAppearance();
|
|
if (m_pWidgetDict->KeyExist("DA"))
|
return CPDF_DefaultAppearance(m_pWidgetDict->GetStringBy("DA"));
|
|
CPDF_Object* pObj = FPDF_GetFieldAttr(m_pField->m_pDict, "DA");
|
if (pObj)
|
return CPDF_DefaultAppearance(pObj->GetString());
|
return m_pField->m_pForm->GetDefaultAppearance();
|
}
|
|
CPDF_Font* CPDF_FormControl::GetDefaultControlFont() {
|
CPDF_DefaultAppearance cDA = GetDefaultAppearance();
|
CFX_ByteString csFontNameTag;
|
FX_FLOAT fFontSize;
|
cDA.GetFont(csFontNameTag, fFontSize);
|
if (csFontNameTag.IsEmpty())
|
return nullptr;
|
|
CPDF_Object* pObj = FPDF_GetFieldAttr(m_pWidgetDict, "DR");
|
if (CPDF_Dictionary* pDict = ToDictionary(pObj)) {
|
CPDF_Dictionary* pFonts = pDict->GetDictBy("Font");
|
if (pFonts) {
|
CPDF_Dictionary* pElement = pFonts->GetDictBy(csFontNameTag);
|
if (pElement) {
|
CPDF_Font* pFont = m_pField->m_pForm->m_pDocument->LoadFont(pElement);
|
if (pFont) {
|
return pFont;
|
}
|
}
|
}
|
}
|
if (CPDF_Font* pFormFont = m_pField->m_pForm->GetFormFont(csFontNameTag))
|
return pFormFont;
|
|
CPDF_Dictionary* pPageDict = m_pWidgetDict->GetDictBy("P");
|
pObj = FPDF_GetFieldAttr(pPageDict, "Resources");
|
if (CPDF_Dictionary* pDict = ToDictionary(pObj)) {
|
CPDF_Dictionary* pFonts = pDict->GetDictBy("Font");
|
if (pFonts) {
|
CPDF_Dictionary* pElement = pFonts->GetDictBy(csFontNameTag);
|
if (pElement) {
|
CPDF_Font* pFont = m_pField->m_pForm->m_pDocument->LoadFont(pElement);
|
if (pFont) {
|
return pFont;
|
}
|
}
|
}
|
}
|
return nullptr;
|
}
|
|
int CPDF_FormControl::GetControlAlignment() {
|
if (!m_pWidgetDict) {
|
return 0;
|
}
|
if (m_pWidgetDict->KeyExist("Q")) {
|
return m_pWidgetDict->GetIntegerBy("Q", 0);
|
}
|
CPDF_Object* pObj = FPDF_GetFieldAttr(m_pField->m_pDict, "Q");
|
if (pObj)
|
return pObj->GetInteger();
|
return m_pField->m_pForm->GetFormAlignment();
|
}
|
|
CPDF_ApSettings::CPDF_ApSettings(CPDF_Dictionary* pDict) : m_pDict(pDict) {}
|
|
bool CPDF_ApSettings::HasMKEntry(const CFX_ByteString& csEntry) const {
|
return m_pDict && m_pDict->KeyExist(csEntry);
|
}
|
|
int CPDF_ApSettings::GetRotation() const {
|
return m_pDict ? m_pDict->GetIntegerBy("R") : 0;
|
}
|
|
FX_ARGB CPDF_ApSettings::GetColor(int& iColorType,
|
const CFX_ByteString& csEntry) const {
|
iColorType = COLORTYPE_TRANSPARENT;
|
if (!m_pDict)
|
return 0;
|
|
CPDF_Array* pEntry = m_pDict->GetArrayBy(csEntry);
|
if (!pEntry)
|
return 0;
|
|
FX_ARGB color = 0;
|
size_t dwCount = pEntry->GetCount();
|
if (dwCount == 1) {
|
iColorType = COLORTYPE_GRAY;
|
FX_FLOAT g = pEntry->GetNumberAt(0) * 255;
|
color = ArgbEncode(255, (int)g, (int)g, (int)g);
|
} else if (dwCount == 3) {
|
iColorType = COLORTYPE_RGB;
|
FX_FLOAT r = pEntry->GetNumberAt(0) * 255;
|
FX_FLOAT g = pEntry->GetNumberAt(1) * 255;
|
FX_FLOAT b = pEntry->GetNumberAt(2) * 255;
|
color = ArgbEncode(255, (int)r, (int)g, (int)b);
|
} else if (dwCount == 4) {
|
iColorType = COLORTYPE_CMYK;
|
FX_FLOAT c = pEntry->GetNumberAt(0);
|
FX_FLOAT m = pEntry->GetNumberAt(1);
|
FX_FLOAT y = pEntry->GetNumberAt(2);
|
FX_FLOAT k = pEntry->GetNumberAt(3);
|
FX_FLOAT r = 1.0f - std::min(1.0f, c + k);
|
FX_FLOAT g = 1.0f - std::min(1.0f, m + k);
|
FX_FLOAT b = 1.0f - std::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,
|
const CFX_ByteString& csEntry) const {
|
if (!m_pDict)
|
return 0;
|
|
CPDF_Array* pEntry = m_pDict->GetArrayBy(csEntry);
|
return pEntry ? pEntry->GetNumberAt(index) : 0;
|
}
|
|
void CPDF_ApSettings::GetOriginalColor(int& iColorType,
|
FX_FLOAT fc[4],
|
const CFX_ByteString& csEntry) const {
|
iColorType = COLORTYPE_TRANSPARENT;
|
for (int i = 0; i < 4; i++) {
|
fc[i] = 0;
|
}
|
if (!m_pDict) {
|
return;
|
}
|
CPDF_Array* pEntry = m_pDict->GetArrayBy(csEntry);
|
if (!pEntry) {
|
return;
|
}
|
size_t dwCount = pEntry->GetCount();
|
if (dwCount == 1) {
|
iColorType = COLORTYPE_GRAY;
|
fc[0] = pEntry->GetNumberAt(0);
|
} else if (dwCount == 3) {
|
iColorType = COLORTYPE_RGB;
|
fc[0] = pEntry->GetNumberAt(0);
|
fc[1] = pEntry->GetNumberAt(1);
|
fc[2] = pEntry->GetNumberAt(2);
|
} else if (dwCount == 4) {
|
iColorType = COLORTYPE_CMYK;
|
fc[0] = pEntry->GetNumberAt(0);
|
fc[1] = pEntry->GetNumberAt(1);
|
fc[2] = pEntry->GetNumberAt(2);
|
fc[3] = pEntry->GetNumberAt(3);
|
}
|
}
|
|
CFX_WideString CPDF_ApSettings::GetCaption(
|
const CFX_ByteString& csEntry) const {
|
return m_pDict ? m_pDict->GetUnicodeTextBy(csEntry) : CFX_WideString();
|
}
|
|
CPDF_Stream* CPDF_ApSettings::GetIcon(const CFX_ByteString& csEntry) const {
|
return m_pDict ? m_pDict->GetStreamBy(csEntry) : nullptr;
|
}
|
|
CPDF_IconFit CPDF_ApSettings::GetIconFit() const {
|
return CPDF_IconFit(m_pDict ? m_pDict->GetDictBy("IF") : nullptr);
|
}
|
|
int CPDF_ApSettings::GetTextPosition() const {
|
return m_pDict ? m_pDict->GetIntegerBy("TP", TEXTPOS_CAPTION)
|
: TEXTPOS_CAPTION;
|
}
|
