| OLD | NEW |
| 1 // Copyright 2016 PDFium Authors. All rights reserved. | 1 // Copyright 2016 PDFium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com | 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
| 6 | 6 |
| 7 #include "core/fpdfdoc/cpdf_pagelabel.h" | 7 #include "core/fpdfdoc/cpdf_pagelabel.h" |
| 8 | 8 |
| 9 #include "core/fpdfapi/parser/cpdf_dictionary.h" | 9 #include "core/fpdfapi/parser/cpdf_dictionary.h" |
| 10 #include "core/fpdfapi/parser/cpdf_document.h" | 10 #include "core/fpdfapi/parser/cpdf_document.h" |
| (...skipping 57 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 68 wsNumPortion = MakeLetters(num); | 68 wsNumPortion = MakeLetters(num); |
| 69 } | 69 } |
| 70 return wsNumPortion; | 70 return wsNumPortion; |
| 71 } | 71 } |
| 72 | 72 |
| 73 } // namespace | 73 } // namespace |
| 74 | 74 |
| 75 CPDF_PageLabel::CPDF_PageLabel(CPDF_Document* pDocument) | 75 CPDF_PageLabel::CPDF_PageLabel(CPDF_Document* pDocument) |
| 76 : m_pDocument(pDocument) {} | 76 : m_pDocument(pDocument) {} |
| 77 | 77 |
| 78 CFX_WideString CPDF_PageLabel::GetLabel(int nPage) const { | 78 bool CPDF_PageLabel::GetLabel(int nPage, CFX_WideString* wsLabel) const { |
| 79 CFX_WideString wsLabel; | |
| 80 if (!m_pDocument) | 79 if (!m_pDocument) |
| 81 return wsLabel; | 80 return false; |
| 81 |
| 82 if (nPage < 0 || nPage >= m_pDocument->GetPageCount()) |
| 83 return false; |
| 82 | 84 |
| 83 CPDF_Dictionary* pPDFRoot = m_pDocument->GetRoot(); | 85 CPDF_Dictionary* pPDFRoot = m_pDocument->GetRoot(); |
| 84 if (!pPDFRoot) | 86 if (!pPDFRoot) |
| 85 return wsLabel; | 87 return false; |
| 86 | 88 |
| 87 CPDF_Dictionary* pLabels = pPDFRoot->GetDictFor("PageLabels"); | 89 CPDF_Dictionary* pLabels = pPDFRoot->GetDictFor("PageLabels"); |
| 90 if (!pLabels) |
| 91 return false; |
| 92 |
| 88 CPDF_NumberTree numberTree(pLabels); | 93 CPDF_NumberTree numberTree(pLabels); |
| 89 CPDF_Object* pValue = nullptr; | 94 CPDF_Object* pValue = nullptr; |
| 90 int n = nPage; | 95 int n = nPage; |
| 91 while (n >= 0) { | 96 while (n >= 0) { |
| 92 pValue = numberTree.LookupValue(n); | 97 pValue = numberTree.LookupValue(n); |
| 93 if (pValue) | 98 if (pValue) |
| 94 break; | 99 break; |
| 95 n--; | 100 n--; |
| 96 } | 101 } |
| 97 | 102 |
| 98 if (pValue) { | 103 if (pValue) { |
| 99 pValue = pValue->GetDirect(); | 104 pValue = pValue->GetDirect(); |
| 100 if (CPDF_Dictionary* pLabel = pValue->AsDictionary()) { | 105 if (CPDF_Dictionary* pLabel = pValue->AsDictionary()) { |
| 101 if (pLabel->KeyExist("P")) | 106 if (pLabel->KeyExist("P")) |
| 102 wsLabel += pLabel->GetUnicodeTextFor("P"); | 107 *wsLabel += pLabel->GetUnicodeTextFor("P"); |
| 103 | 108 |
| 104 CFX_ByteString bsNumberingStyle = pLabel->GetStringFor("S", ""); | 109 CFX_ByteString bsNumberingStyle = pLabel->GetStringFor("S", ""); |
| 105 int nLabelNum = nPage - n + pLabel->GetIntegerFor("St", 1); | 110 int nLabelNum = nPage - n + pLabel->GetIntegerFor("St", 1); |
| 106 CFX_WideString wsNumPortion = | 111 CFX_WideString wsNumPortion = |
| 107 GetLabelNumPortion(nLabelNum, bsNumberingStyle); | 112 GetLabelNumPortion(nLabelNum, bsNumberingStyle); |
| 108 wsLabel += wsNumPortion; | 113 *wsLabel += wsNumPortion; |
| 109 return wsLabel; | 114 return true; |
| 110 } | 115 } |
| 111 } | 116 } |
| 112 wsLabel.Format(L"%d", nPage + 1); | 117 wsLabel->Format(L"%d", nPage + 1); |
| 113 return wsLabel; | 118 return true; |
| 114 } | 119 } |
| 115 | 120 |
| 116 int32_t CPDF_PageLabel::GetPageByLabel(const CFX_ByteStringC& bsLabel) const { | 121 int32_t CPDF_PageLabel::GetPageByLabel(const CFX_ByteStringC& bsLabel) const { |
| 117 if (!m_pDocument) | 122 if (!m_pDocument) |
| 118 return -1; | 123 return -1; |
| 119 | 124 |
| 120 CPDF_Dictionary* pPDFRoot = m_pDocument->GetRoot(); | 125 CPDF_Dictionary* pPDFRoot = m_pDocument->GetRoot(); |
| 121 if (!pPDFRoot) | 126 if (!pPDFRoot) |
| 122 return -1; | 127 return -1; |
| 123 | 128 |
| 124 int nPages = m_pDocument->GetPageCount(); | 129 int nPages = m_pDocument->GetPageCount(); |
| 125 for (int i = 0; i < nPages; i++) { | 130 for (int i = 0; i < nPages; i++) { |
| 126 if (PDF_EncodeText(GetLabel(i)).Compare(bsLabel)) | 131 CFX_WideString str; |
| 132 if (!GetLabel(i, &str)) |
| 133 continue; |
| 134 if (PDF_EncodeText(str).Compare(bsLabel)) |
| 127 return i; | 135 return i; |
| 128 } | 136 } |
| 129 | 137 |
| 130 int nPage = FXSYS_atoi(CFX_ByteString(bsLabel).c_str()); // NUL terminate. | 138 int nPage = FXSYS_atoi(CFX_ByteString(bsLabel).c_str()); // NUL terminate. |
| 131 return nPage > 0 && nPage <= nPages ? nPage : -1; | 139 return nPage > 0 && nPage <= nPages ? nPage : -1; |
| 132 } | 140 } |
| 133 | 141 |
| 134 int32_t CPDF_PageLabel::GetPageByLabel(const CFX_WideStringC& wsLabel) const { | 142 int32_t CPDF_PageLabel::GetPageByLabel(const CFX_WideStringC& wsLabel) const { |
| 135 return GetPageByLabel(PDF_EncodeText(wsLabel.c_str()).AsStringC()); | 143 return GetPageByLabel(PDF_EncodeText(wsLabel.c_str()).AsStringC()); |
| 136 } | 144 } |
| OLD | NEW |