| 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 #include "core/fpdfapi/parser/cpdf_document.h" | 5 #include "core/fpdfapi/parser/cpdf_document.h" |
| 6 | 6 |
| 7 #include <memory> | 7 #include <memory> |
| 8 | 8 |
| 9 #include "core/fpdfapi/cpdf_modulemgr.h" | 9 #include "core/fpdfapi/cpdf_modulemgr.h" |
| 10 #include "core/fpdfapi/parser/cpdf_array.h" | 10 #include "core/fpdfapi/parser/cpdf_array.h" |
| 11 #include "core/fpdfapi/parser/cpdf_dictionary.h" | 11 #include "core/fpdfapi/parser/cpdf_dictionary.h" |
| 12 #include "core/fpdfapi/parser/cpdf_linearized.h" | |
| 13 #include "core/fpdfapi/parser/cpdf_parser.h" | 12 #include "core/fpdfapi/parser/cpdf_parser.h" |
| 14 #include "core/fxcrt/fx_memory.h" | 13 #include "core/fxcrt/fx_memory.h" |
| 15 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
| 16 | 15 |
| 17 namespace { | 16 namespace { |
| 18 | 17 |
| 18 using ScopedDictionary = |
| 19 std::unique_ptr<CPDF_Dictionary, ReleaseDeleter<CPDF_Dictionary>>; |
| 20 |
| 19 CPDF_Dictionary* CreatePageTreeNode(CPDF_Array* kids, | 21 CPDF_Dictionary* CreatePageTreeNode(CPDF_Array* kids, |
| 20 CPDF_Document* pDoc, | 22 CPDF_Document* pDoc, |
| 21 int count) { | 23 int count) { |
| 22 CPDF_Dictionary* pageNode = new CPDF_Dictionary(); | 24 CPDF_Dictionary* pageNode = new CPDF_Dictionary(); |
| 23 pageNode->SetStringFor("Type", "Pages"); | 25 pageNode->SetStringFor("Type", "Pages"); |
| 24 pageNode->SetReferenceFor("Kids", pDoc, pDoc->AddIndirectObject(kids)); | 26 pageNode->SetReferenceFor("Kids", pDoc, pDoc->AddIndirectObject(kids)); |
| 25 pageNode->SetIntegerFor("Count", count); | 27 pageNode->SetIntegerFor("Count", count); |
| 26 uint32_t pageNodeRef = pDoc->AddIndirectObject(pageNode); | 28 uint32_t pageNodeRef = pDoc->AddIndirectObject(pageNode); |
| 27 for (size_t i = 0; i < kids->GetCount(); i++) | 29 for (size_t i = 0; i < kids->GetCount(); i++) |
| 28 kids->GetDictAt(i)->SetReferenceFor("Parent", pDoc, pageNodeRef); | 30 kids->GetDictAt(i)->SetReferenceFor("Parent", pDoc, pageNodeRef); |
| (...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 68 | 70 |
| 69 m_pOwnedRootDict.reset(new CPDF_Dictionary()); | 71 m_pOwnedRootDict.reset(new CPDF_Dictionary()); |
| 70 m_pOwnedRootDict->SetReferenceFor("Pages", this, pagesDict->GetObjNum()); | 72 m_pOwnedRootDict->SetReferenceFor("Pages", this, pagesDict->GetObjNum()); |
| 71 m_pRootDict = m_pOwnedRootDict.get(); | 73 m_pRootDict = m_pOwnedRootDict.get(); |
| 72 m_PageList.SetSize(7); | 74 m_PageList.SetSize(7); |
| 73 } | 75 } |
| 74 | 76 |
| 75 private: | 77 private: |
| 76 std::unique_ptr<CPDF_Dictionary> m_pOwnedRootDict; | 78 std::unique_ptr<CPDF_Dictionary> m_pOwnedRootDict; |
| 77 }; | 79 }; |
| 78 | |
| 79 class TestLinearized : public CPDF_Linearized { | |
| 80 public: | |
| 81 explicit TestLinearized(CPDF_Dictionary* dict) : CPDF_Linearized(dict) {} | |
| 82 }; | |
| 83 } // namespace | 80 } // namespace |
| 84 | 81 |
| 85 class cpdf_document_test : public testing::Test { | 82 class cpdf_document_test : public testing::Test { |
| 86 public: | 83 public: |
| 87 void SetUp() override { | 84 void SetUp() override { |
| 88 CPDF_ModuleMgr* module_mgr = CPDF_ModuleMgr::Get(); | 85 CPDF_ModuleMgr* module_mgr = CPDF_ModuleMgr::Get(); |
| 89 module_mgr->InitPageModule(); | 86 module_mgr->InitPageModule(); |
| 90 } | 87 } |
| 91 void TearDown() override {} | 88 void TearDown() override {} |
| 92 }; | 89 }; |
| (...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 138 ASSERT_TRUE(page); | 135 ASSERT_TRUE(page); |
| 139 ASSERT_TRUE(page->KeyExist("PageNumbering")); | 136 ASSERT_TRUE(page->KeyExist("PageNumbering")); |
| 140 EXPECT_EQ(6, page->GetIntegerFor("PageNumbering")); | 137 EXPECT_EQ(6, page->GetIntegerFor("PageNumbering")); |
| 141 } | 138 } |
| 142 | 139 |
| 143 TEST_F(cpdf_document_test, UseCachedPageObjNumIfHaveNotPagesDict) { | 140 TEST_F(cpdf_document_test, UseCachedPageObjNumIfHaveNotPagesDict) { |
| 144 // ObjNum can be added in CPDF_DataAvail::IsPageAvail, and PagesDict | 141 // ObjNum can be added in CPDF_DataAvail::IsPageAvail, and PagesDict |
| 145 // can be not exists in this case. | 142 // can be not exists in this case. |
| 146 // (case, when hint table is used to page check in CPDF_DataAvail). | 143 // (case, when hint table is used to page check in CPDF_DataAvail). |
| 147 CPDF_Document document(pdfium::MakeUnique<CPDF_Parser>()); | 144 CPDF_Document document(pdfium::MakeUnique<CPDF_Parser>()); |
| 148 auto dict = pdfium::MakeUnique<CPDF_Dictionary>(); | 145 std::unique_ptr<CPDF_Dictionary> dict(new CPDF_Dictionary()); |
| 149 dict->SetBooleanFor("Linearized", true); | |
| 150 const int page_count = 100; | 146 const int page_count = 100; |
| 151 dict->SetIntegerFor("N", page_count); | 147 dict->SetIntegerFor("N", page_count); |
| 152 TestLinearized linearized(dict.get()); | 148 document.LoadLinearizedDoc(dict.get()); |
| 153 document.LoadLinearizedDoc(&linearized); | |
| 154 ASSERT_EQ(page_count, document.GetPageCount()); | 149 ASSERT_EQ(page_count, document.GetPageCount()); |
| 155 CPDF_Object* page_stub = new CPDF_Dictionary(); | 150 CPDF_Object* page_stub = new CPDF_Dictionary(); |
| 156 const uint32_t obj_num = document.AddIndirectObject(page_stub); | 151 const uint32_t obj_num = document.AddIndirectObject(page_stub); |
| 157 const int test_page_num = 33; | 152 const int test_page_num = 33; |
| 158 | 153 |
| 159 EXPECT_FALSE(document.IsPageLoaded(test_page_num)); | 154 EXPECT_FALSE(document.IsPageLoaded(test_page_num)); |
| 160 EXPECT_EQ(nullptr, document.GetPage(test_page_num)); | 155 EXPECT_EQ(nullptr, document.GetPage(test_page_num)); |
| 161 | 156 |
| 162 document.SetPageObjNum(test_page_num, obj_num); | 157 document.SetPageObjNum(test_page_num, obj_num); |
| 163 | 158 |
| 164 EXPECT_TRUE(document.IsPageLoaded(test_page_num)); | 159 EXPECT_TRUE(document.IsPageLoaded(test_page_num)); |
| 165 EXPECT_EQ(page_stub, document.GetPage(test_page_num)); | 160 EXPECT_EQ(page_stub, document.GetPage(test_page_num)); |
| 166 } | 161 } |
| OLD | NEW |