// 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.
|
|
#include "public/fpdf_structtree.h"
|
|
#include <memory>
|
|
#include "core/fpdfapi/page/cpdf_page.h"
|
#include "core/fpdfapi/parser/cpdf_dictionary.h"
|
#include "core/fpdfdoc/cpdf_structelement.h"
|
#include "core/fpdfdoc/cpdf_structtree.h"
|
#include "fpdfsdk/fsdk_define.h"
|
|
namespace {
|
|
CPDF_StructTree* ToStructTree(FPDF_STRUCTTREE struct_tree) {
|
return static_cast<CPDF_StructTree*>(struct_tree);
|
}
|
|
CPDF_StructElement* ToStructTreeElement(FPDF_STRUCTELEMENT struct_element) {
|
return static_cast<CPDF_StructElement*>(struct_element);
|
}
|
|
unsigned long WideStringToBuffer(const WideString& str,
|
void* buffer,
|
unsigned long buflen) {
|
if (str.IsEmpty())
|
return 0;
|
|
ByteString encodedStr = str.UTF16LE_Encode();
|
const unsigned long len = encodedStr.GetLength();
|
if (buffer && len <= buflen)
|
memcpy(buffer, encodedStr.c_str(), len);
|
return len;
|
}
|
|
} // namespace
|
|
FPDF_EXPORT FPDF_STRUCTTREE FPDF_CALLCONV
|
FPDF_StructTree_GetForPage(FPDF_PAGE page) {
|
CPDF_Page* pPage = CPDFPageFromFPDFPage(page);
|
if (!pPage)
|
return nullptr;
|
return CPDF_StructTree::LoadPage(pPage->m_pDocument.Get(),
|
pPage->m_pFormDict.Get())
|
.release();
|
}
|
|
FPDF_EXPORT void FPDF_CALLCONV
|
FPDF_StructTree_Close(FPDF_STRUCTTREE struct_tree) {
|
std::unique_ptr<CPDF_StructTree>(ToStructTree(struct_tree));
|
}
|
|
FPDF_EXPORT int FPDF_CALLCONV
|
FPDF_StructTree_CountChildren(FPDF_STRUCTTREE struct_tree) {
|
CPDF_StructTree* tree = ToStructTree(struct_tree);
|
if (!tree)
|
return -1;
|
|
pdfium::base::CheckedNumeric<int> tmp_size = tree->CountTopElements();
|
return tmp_size.ValueOrDefault(-1);
|
}
|
|
FPDF_EXPORT FPDF_STRUCTELEMENT FPDF_CALLCONV
|
FPDF_StructTree_GetChildAtIndex(FPDF_STRUCTTREE struct_tree, int index) {
|
CPDF_StructTree* tree = ToStructTree(struct_tree);
|
if (!tree || index < 0 ||
|
static_cast<size_t>(index) >= tree->CountTopElements()) {
|
return nullptr;
|
}
|
return tree->GetTopElement(static_cast<size_t>(index));
|
}
|
|
FPDF_EXPORT unsigned long FPDF_CALLCONV
|
FPDF_StructElement_GetAltText(FPDF_STRUCTELEMENT struct_element,
|
void* buffer,
|
unsigned long buflen) {
|
CPDF_StructElement* elem = ToStructTreeElement(struct_element);
|
return (elem && elem->GetDict())
|
? WideStringToBuffer(elem->GetDict()->GetUnicodeTextFor("Alt"),
|
buffer, buflen)
|
: 0;
|
}
|
|
FPDF_EXPORT int FPDF_CALLCONV
|
FPDF_StructElement_GetMarkedContentID(FPDF_STRUCTELEMENT struct_element) {
|
CPDF_StructElement* elem = ToStructTreeElement(struct_element);
|
CPDF_Object* p =
|
(elem && elem->GetDict()) ? elem->GetDict()->GetObjectFor("K") : nullptr;
|
return p && p->IsNumber() ? p->GetInteger() : -1;
|
}
|
|
FPDF_EXPORT unsigned long FPDF_CALLCONV
|
FPDF_StructElement_GetType(FPDF_STRUCTELEMENT struct_element,
|
void* buffer,
|
unsigned long buflen) {
|
CPDF_StructElement* elem = ToStructTreeElement(struct_element);
|
return elem ? WideStringToBuffer(elem->GetType().UTF8Decode(), buffer, buflen)
|
: 0;
|
}
|
|
FPDF_EXPORT unsigned long FPDF_CALLCONV
|
FPDF_StructElement_GetTitle(FPDF_STRUCTELEMENT struct_element,
|
void* buffer,
|
unsigned long buflen) {
|
CPDF_StructElement* elem = ToStructTreeElement(struct_element);
|
return elem
|
? WideStringToBuffer(elem->GetTitle().UTF8Decode(), buffer, buflen)
|
: 0;
|
}
|
|
FPDF_EXPORT int FPDF_CALLCONV
|
FPDF_StructElement_CountChildren(FPDF_STRUCTELEMENT struct_element) {
|
CPDF_StructElement* elem = ToStructTreeElement(struct_element);
|
if (!elem)
|
return -1;
|
|
pdfium::base::CheckedNumeric<int> tmp_size = elem->CountKids();
|
return tmp_size.ValueOrDefault(-1);
|
}
|
|
FPDF_EXPORT FPDF_STRUCTELEMENT FPDF_CALLCONV
|
FPDF_StructElement_GetChildAtIndex(FPDF_STRUCTELEMENT struct_element,
|
int index) {
|
CPDF_StructElement* elem = ToStructTreeElement(struct_element);
|
if (!elem || index < 0 || static_cast<size_t>(index) >= elem->CountKids())
|
return nullptr;
|
|
return elem->GetKidIfElement(static_cast<size_t>(index));
|
}
|
