blob: a7e4f5b42866c2b32bac8c094774c6ffff6ca383 [file] [log] [blame]
// 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 "core/fpdfdoc/cpdf_pagelabel.h"
#include "core/fpdfapi/parser/cpdf_dictionary.h"
#include "core/fpdfapi/parser/cpdf_document.h"
#include "core/fpdfapi/parser/fpdf_parser_decode.h"
#include "core/fpdfdoc/cpdf_numbertree.h"
namespace {
WideString MakeRoman(int num) {
const int kArabic[] = {1000, 900, 500, 400, 100, 90, 50, 40, 10, 9, 5, 4, 1};
const WideString kRoman[] = {L"m", L"cm", L"d", L"cd", L"c", L"xc", L"l",
L"xl", L"x", L"ix", L"v", L"iv", L"i"};
const int kMaxNum = 1000000;
num %= kMaxNum;
int i = 0;
WideString wsRomanNumber;
while (num > 0) {
while (num >= kArabic[i]) {
num = num - kArabic[i];
wsRomanNumber += kRoman[i];
}
i = i + 1;
}
return wsRomanNumber;
}
WideString MakeLetters(int num) {
if (num == 0)
return WideString();
WideString wsLetters;
const int nMaxCount = 1000;
const int nLetterCount = 26;
--num;
int count = num / nLetterCount + 1;
count %= nMaxCount;
wchar_t ch = L'a' + num % nLetterCount;
for (int i = 0; i < count; i++)
wsLetters += ch;
return wsLetters;
}
WideString GetLabelNumPortion(int num, const ByteString& bsStyle) {
if (bsStyle.IsEmpty())
return WideString();
if (bsStyle == "D")
return WideString::Format(L"%d", num);
if (bsStyle == "R") {
WideString wsNumPortion = MakeRoman(num);
wsNumPortion.MakeUpper();
return wsNumPortion;
}
if (bsStyle == "r")
return MakeRoman(num);
if (bsStyle == "A") {
WideString wsNumPortion = MakeLetters(num);
wsNumPortion.MakeUpper();
return wsNumPortion;
}
if (bsStyle == "a")
return MakeLetters(num);
return WideString();
}
} // namespace
CPDF_PageLabel::CPDF_PageLabel(CPDF_Document* pDocument)
: m_pDocument(pDocument) {}
CPDF_PageLabel::~CPDF_PageLabel() = default;
Optional<WideString> CPDF_PageLabel::GetLabel(int nPage) const {
if (!m_pDocument)
return {};
if (nPage < 0 || nPage >= m_pDocument->GetPageCount())
return {};
const CPDF_Dictionary* pPDFRoot = m_pDocument->GetRoot();
if (!pPDFRoot)
return {};
const CPDF_Dictionary* pLabels = pPDFRoot->GetDictFor("PageLabels");
if (!pLabels)
return {};
CPDF_NumberTree numberTree(pLabels);
const CPDF_Object* pValue = nullptr;
int n = nPage;
while (n >= 0) {
pValue = numberTree.LookupValue(n);
if (pValue)
break;
n--;
}
WideString label;
if (pValue) {
pValue = pValue->GetDirect();
if (const CPDF_Dictionary* pLabel = pValue->AsDictionary()) {
if (pLabel->KeyExist("P"))
label += pLabel->GetUnicodeTextFor("P");
ByteString bsNumberingStyle = pLabel->GetStringFor("S", ByteString());
int nLabelNum = nPage - n + pLabel->GetIntegerFor("St", 1);
WideString wsNumPortion = GetLabelNumPortion(nLabelNum, bsNumberingStyle);
label += wsNumPortion;
return {label};
}
}
label = WideString::Format(L"%d", nPage + 1);
return {label};
}