| 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_array.h" | 5 #include "core/fpdfapi/parser/cpdf_array.h" |
| 6 #include "core/fpdfapi/parser/cpdf_number.h" | |
| 7 #include "core/fpdfapi/parser/cpdf_reference.h" | |
| 8 | 6 |
| 9 #include <memory> | 7 #include <memory> |
| 10 | 8 |
| 9 #include "core/fpdfapi/parser/cpdf_number.h" |
| 10 #include "core/fpdfapi/parser/cpdf_reference.h" |
| 11 #include "testing/gtest/include/gtest/gtest.h" | 11 #include "testing/gtest/include/gtest/gtest.h" |
| 12 #include "third_party/base/ptr_util.h" |
| 12 | 13 |
| 13 TEST(cpdf_array, RemoveAt) { | 14 TEST(cpdf_array, RemoveAt) { |
| 14 { | 15 { |
| 15 int elems[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; | 16 int elems[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; |
| 16 std::unique_ptr<CPDF_Array> arr(new CPDF_Array); | 17 std::unique_ptr<CPDF_Array> arr(new CPDF_Array); |
| 17 for (size_t i = 0; i < FX_ArraySize(elems); ++i) | 18 for (size_t i = 0; i < FX_ArraySize(elems); ++i) |
| 18 arr->AddInteger(elems[i]); | 19 arr->AddInteger(elems[i]); |
| 19 arr->RemoveAt(3, 3); | 20 arr->RemoveAt(3, 3); |
| 20 int expected[] = {1, 2, 3, 7, 8, 9, 10}; | 21 int expected[] = {1, 2, 3, 7, 8, 9, 10}; |
| 21 EXPECT_EQ(FX_ArraySize(expected), arr->GetCount()); | 22 EXPECT_EQ(FX_ArraySize(expected), arr->GetCount()); |
| (...skipping 78 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 100 static const size_t kNumOfRowElems = 5; | 101 static const size_t kNumOfRowElems = 5; |
| 101 int elems[kNumOfRows][kNumOfRowElems] = { | 102 int elems[kNumOfRows][kNumOfRowElems] = { |
| 102 {1, 2, 3, 4, 5}, {10, 9, 8, 7, 6}, {11, 12, 13, 14, 15}}; | 103 {1, 2, 3, 4, 5}, {10, 9, 8, 7, 6}, {11, 12, 13, 14, 15}}; |
| 103 std::unique_ptr<CPDF_Array> arr(new CPDF_Array); | 104 std::unique_ptr<CPDF_Array> arr(new CPDF_Array); |
| 104 // Indirect references to indirect objects. | 105 // Indirect references to indirect objects. |
| 105 std::unique_ptr<CPDF_IndirectObjectHolder> obj_holder( | 106 std::unique_ptr<CPDF_IndirectObjectHolder> obj_holder( |
| 106 new CPDF_IndirectObjectHolder()); | 107 new CPDF_IndirectObjectHolder()); |
| 107 for (size_t i = 0; i < kNumOfRows; ++i) { | 108 for (size_t i = 0; i < kNumOfRows; ++i) { |
| 108 CPDF_Array* arr_elem = new CPDF_Array; | 109 CPDF_Array* arr_elem = new CPDF_Array; |
| 109 for (size_t j = 0; j < kNumOfRowElems; ++j) { | 110 for (size_t j = 0; j < kNumOfRowElems; ++j) { |
| 110 CPDF_Number* obj = new CPDF_Number(elems[i][j]); | 111 std::unique_ptr<CPDF_Number> obj(new CPDF_Number(elems[i][j])); |
| 111 // Starts object number from 1. | 112 // Starts object number from 1. |
| 112 int obj_num = i * kNumOfRowElems + j + 1; | 113 int obj_num = i * kNumOfRowElems + j + 1; |
| 113 obj_holder->ReplaceIndirectObjectIfHigherGeneration(obj_num, obj); | 114 obj_holder->ReplaceIndirectObjectIfHigherGeneration(obj_num, |
| 115 std::move(obj)); |
| 114 arr_elem->InsertAt(j, new CPDF_Reference(obj_holder.get(), obj_num)); | 116 arr_elem->InsertAt(j, new CPDF_Reference(obj_holder.get(), obj_num)); |
| 115 } | 117 } |
| 116 arr->InsertAt(i, arr_elem); | 118 arr->InsertAt(i, arr_elem); |
| 117 } | 119 } |
| 118 ASSERT_EQ(kNumOfRows, arr->GetCount()); | 120 ASSERT_EQ(kNumOfRows, arr->GetCount()); |
| 119 // Not dereferencing reference objects means just creating new references | 121 // Not dereferencing reference objects means just creating new references |
| 120 // instead of new copies of direct objects. | 122 // instead of new copies of direct objects. |
| 121 std::unique_ptr<CPDF_Array> arr1(arr->Clone()->AsArray()); | 123 std::unique_ptr<CPDF_Array> arr1(arr->Clone()->AsArray()); |
| 122 EXPECT_EQ(arr->GetCount(), arr1->GetCount()); | 124 EXPECT_EQ(arr->GetCount(), arr1->GetCount()); |
| 123 // Dereferencing reference objects creates new copies of direct objects. | 125 // Dereferencing reference objects creates new copies of direct objects. |
| (...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 165 TEST(cpdf_array, Iterator) { | 167 TEST(cpdf_array, Iterator) { |
| 166 int elems[] = {-23, -11, 3, 455, 2345877, | 168 int elems[] = {-23, -11, 3, 455, 2345877, |
| 167 0, 7895330, -12564334, 10000, -100000}; | 169 0, 7895330, -12564334, 10000, -100000}; |
| 168 std::unique_ptr<CPDF_Array> arr(new CPDF_Array); | 170 std::unique_ptr<CPDF_Array> arr(new CPDF_Array); |
| 169 for (size_t i = 0; i < FX_ArraySize(elems); ++i) | 171 for (size_t i = 0; i < FX_ArraySize(elems); ++i) |
| 170 arr->InsertAt(i, new CPDF_Number(elems[i])); | 172 arr->InsertAt(i, new CPDF_Number(elems[i])); |
| 171 size_t index = 0; | 173 size_t index = 0; |
| 172 for (const auto& it : *arr) | 174 for (const auto& it : *arr) |
| 173 EXPECT_EQ(elems[index++], it->AsNumber()->GetInteger()); | 175 EXPECT_EQ(elems[index++], it->AsNumber()->GetInteger()); |
| 174 } | 176 } |
| OLD | NEW |