Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 1 | // Copyright 2016 PDFium Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
Dan Sinclair | 85c8e7f | 2016-11-21 13:50:32 -0500 | [diff] [blame] | 5 | #include <memory> |
| 6 | #include <string> |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 7 | #include <utility> |
Jane Liu | 548334e | 2017-08-03 16:33:40 -0400 | [diff] [blame] | 8 | #include <vector> |
Dan Sinclair | 85c8e7f | 2016-11-21 13:50:32 -0500 | [diff] [blame] | 9 | |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 10 | #include "build/build_config.h" |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 11 | #include "core/fpdfapi/font/cpdf_font.h" |
Miklos Vajna | 46b4373 | 2018-08-14 19:15:43 +0000 | [diff] [blame] | 12 | #include "core/fpdfapi/page/cpdf_formobject.h" |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 13 | #include "core/fpdfapi/page/cpdf_page.h" |
Henrique Nakashima | 6eb7939 | 2018-06-12 20:27:35 +0000 | [diff] [blame] | 14 | #include "core/fpdfapi/page/cpdf_pageobject.h" |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 15 | #include "core/fpdfapi/parser/cpdf_array.h" |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 16 | #include "core/fpdfapi/parser/cpdf_dictionary.h" |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 17 | #include "core/fpdfapi/parser/cpdf_number.h" |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 18 | #include "core/fpdfapi/parser/cpdf_stream.h" |
Artem Strygin | eababa1 | 2018-06-06 12:31:18 +0000 | [diff] [blame] | 19 | #include "core/fpdfapi/parser/cpdf_stream_acc.h" |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 20 | #include "core/fxcrt/fx_system.h" |
Lei Zhang | 0dcb8fd | 2019-02-07 00:46:04 +0000 | [diff] [blame] | 21 | #include "core/fxge/fx_font.h" |
Dan Sinclair | 00d47a6 | 2018-03-28 18:39:04 +0000 | [diff] [blame] | 22 | #include "fpdfsdk/cpdfsdk_helpers.h" |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 23 | #include "public/cpp/fpdf_scopers.h" |
Jane Liu | eda6525 | 2017-06-07 11:31:27 -0400 | [diff] [blame] | 24 | #include "public/fpdf_annot.h" |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 25 | #include "public/fpdf_edit.h" |
| 26 | #include "public/fpdfview.h" |
| 27 | #include "testing/embedder_test.h" |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 28 | #include "testing/embedder_test_constants.h" |
Lei Zhang | b6992dd | 2019-02-05 23:30:20 +0000 | [diff] [blame] | 29 | #include "testing/fx_string_testhelpers.h" |
Tom Sepez | 0aec19b | 2016-01-07 12:22:44 -0800 | [diff] [blame] | 30 | #include "testing/gmock/include/gmock/gmock-matchers.h" |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 31 | #include "testing/gtest/include/gtest/gtest.h" |
Lei Zhang | 4c64e96 | 2019-02-05 19:24:12 +0000 | [diff] [blame] | 32 | #include "testing/utils/hash.h" |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 33 | |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 34 | using pdfium::kHelloWorldChecksum; |
| 35 | |
| 36 | namespace { |
| 37 | |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 38 | const char kAllRemovedChecksum[] = "eee4600ac08b458ac7ac2320e225674c"; |
| 39 | |
| 40 | const wchar_t kBottomText[] = L"I'm at the bottom of the page"; |
| 41 | |
| 42 | #if defined(OS_WIN) |
| 43 | const char kBottomTextChecksum[] = "08d1ff3e5a42801bee6077fd366bef00"; |
| 44 | #elif defined(OS_MACOSX) |
| 45 | const char kBottomTextChecksum[] = "324e1db8164a040cf6104538baa95ba6"; |
| 46 | #else |
| 47 | const char kBottomTextChecksum[] = "eacaa24573b8ce997b3882595f096f00"; |
| 48 | #endif |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 49 | |
Hui Yingst | 08c4071 | 2020-04-29 01:37:35 +0000 | [diff] [blame] | 50 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 51 | #if defined(OS_WIN) |
| 52 | const char kFirstRemovedChecksum[] = "d468228cc55071a9e838f66afc0f58ff"; |
| 53 | #else |
| 54 | const char kFirstRemovedChecksum[] = "0c79dc1065b1d06991e3ac4aaa35d25c"; |
| 55 | #endif // defined(OS_WIN) |
| 56 | #else |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 57 | #if defined(OS_WIN) |
| 58 | const char kFirstRemovedChecksum[] = "aae6c5334721f90ec30d3d59f4ef7deb"; |
| 59 | #elif defined(OS_MACOSX) |
| 60 | const char kFirstRemovedChecksum[] = "17ca3778fd8bb395b46532f1fa17f702"; |
| 61 | #else |
| 62 | const char kFirstRemovedChecksum[] = "b76df015fe88009c3c342395df96abf1"; |
| 63 | #endif |
Hui Yingst | 08c4071 | 2020-04-29 01:37:35 +0000 | [diff] [blame] | 64 | #endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 65 | |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 66 | const wchar_t kLoadedFontText[] = L"I am testing my loaded font, WEE."; |
| 67 | |
| 68 | #if defined(OS_WIN) |
| 69 | const char kLoadedFontTextChecksum[] = "d60ba39f9698e32360d99e727dd93165"; |
| 70 | #elif defined(OS_MACOSX) |
| 71 | const char kLoadedFontTextChecksum[] = "fc921c0bbdde73986ac13c15a85db4c3"; |
| 72 | #else |
| 73 | const char kLoadedFontTextChecksum[] = "70592859010ffbf532a2237b8118bcc4"; |
| 74 | #endif |
| 75 | |
| 76 | const char kRedRectangleChecksum[] = "66d02eaa6181e2c069ce2ea99beda497"; |
| 77 | |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 78 | } // namespace |
| 79 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 80 | class FPDFEditEmbedderTest : public EmbedderTest { |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 81 | protected: |
| 82 | FPDF_DOCUMENT CreateNewDocument() { |
| 83 | document_ = FPDF_CreateNewDocument(); |
Tom Sepez | 41066c1 | 2017-05-18 09:28:49 -0700 | [diff] [blame] | 84 | cpdf_doc_ = CPDFDocumentFromFPDFDocument(document_); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 85 | return document_; |
| 86 | } |
| 87 | |
Lei Zhang | 710fa99 | 2018-05-25 16:24:48 +0000 | [diff] [blame] | 88 | void CheckFontDescriptor(const CPDF_Dictionary* font_dict, |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 89 | int font_type, |
| 90 | bool bold, |
| 91 | bool italic, |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 92 | pdfium::span<const uint8_t> span) { |
Lei Zhang | b1ec280 | 2018-05-25 21:55:24 +0000 | [diff] [blame] | 93 | const CPDF_Dictionary* font_desc = font_dict->GetDictFor("FontDescriptor"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 94 | ASSERT_TRUE(font_desc); |
Lei Zhang | a228ff3 | 2020-06-24 17:39:33 +0000 | [diff] [blame] | 95 | EXPECT_EQ("FontDescriptor", font_desc->GetNameFor("Type")); |
| 96 | ByteString font_name = font_desc->GetNameFor("FontName"); |
| 97 | EXPECT_FALSE(font_name.IsEmpty()); |
| 98 | EXPECT_EQ(font_dict->GetNameFor("BaseFont"), font_name); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 99 | |
| 100 | // Check that the font descriptor has the required keys according to spec |
| 101 | // 1.7 Table 5.19 |
| 102 | ASSERT_TRUE(font_desc->KeyExist("Flags")); |
Dan Sinclair | 10e1f05 | 2017-09-28 15:59:42 -0400 | [diff] [blame] | 103 | |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 104 | int font_flags = font_desc->GetIntegerFor("Flags"); |
Lei Zhang | 44d0383 | 2019-10-18 18:45:31 +0000 | [diff] [blame] | 105 | EXPECT_EQ(bold, FontStyleIsForceBold(font_flags)); |
Dan Sinclair | 10e1f05 | 2017-09-28 15:59:42 -0400 | [diff] [blame] | 106 | EXPECT_EQ(italic, FontStyleIsItalic(font_flags)); |
| 107 | EXPECT_TRUE(FontStyleIsNonSymbolic(font_flags)); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 108 | ASSERT_TRUE(font_desc->KeyExist("FontBBox")); |
Nicolás Peña | 5f95f36 | 2017-09-28 13:00:45 +0900 | [diff] [blame] | 109 | |
Lei Zhang | b1ec280 | 2018-05-25 21:55:24 +0000 | [diff] [blame] | 110 | const CPDF_Array* fontBBox = font_desc->GetArrayFor("FontBBox"); |
Nicolás Peña | 5f95f36 | 2017-09-28 13:00:45 +0900 | [diff] [blame] | 111 | ASSERT_TRUE(fontBBox); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 112 | EXPECT_EQ(4u, fontBBox->size()); |
Nicolás Peña | 5f95f36 | 2017-09-28 13:00:45 +0900 | [diff] [blame] | 113 | // Check that the coordinates are in the preferred order according to spec |
| 114 | // 1.7 Section 3.8.4 |
| 115 | EXPECT_TRUE(fontBBox->GetIntegerAt(0) < fontBBox->GetIntegerAt(2)); |
| 116 | EXPECT_TRUE(fontBBox->GetIntegerAt(1) < fontBBox->GetIntegerAt(3)); |
| 117 | |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 118 | EXPECT_TRUE(font_desc->KeyExist("ItalicAngle")); |
| 119 | EXPECT_TRUE(font_desc->KeyExist("Ascent")); |
| 120 | EXPECT_TRUE(font_desc->KeyExist("Descent")); |
| 121 | EXPECT_TRUE(font_desc->KeyExist("CapHeight")); |
| 122 | EXPECT_TRUE(font_desc->KeyExist("StemV")); |
Ryan Harrison | 275e260 | 2017-09-18 14:23:18 -0400 | [diff] [blame] | 123 | ByteString present("FontFile"); |
| 124 | ByteString absent("FontFile2"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 125 | if (font_type == FPDF_FONT_TRUETYPE) |
| 126 | std::swap(present, absent); |
| 127 | EXPECT_TRUE(font_desc->KeyExist(present)); |
| 128 | EXPECT_FALSE(font_desc->KeyExist(absent)); |
| 129 | |
Artem Strygin | eababa1 | 2018-06-06 12:31:18 +0000 | [diff] [blame] | 130 | auto streamAcc = |
| 131 | pdfium::MakeRetain<CPDF_StreamAcc>(font_desc->GetStreamFor(present)); |
| 132 | streamAcc->LoadAllDataRaw(); |
| 133 | |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 134 | // Check that the font stream is the one that was provided |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 135 | ASSERT_EQ(span.size(), streamAcc->GetSize()); |
Nicolás Peña | 79eab23 | 2017-09-28 13:29:05 +0900 | [diff] [blame] | 136 | if (font_type == FPDF_FONT_TRUETYPE) { |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 137 | ASSERT_EQ(static_cast<int>(span.size()), |
Artem Strygin | eababa1 | 2018-06-06 12:31:18 +0000 | [diff] [blame] | 138 | streamAcc->GetDict()->GetIntegerFor("Length1")); |
Nicolás Peña | 79eab23 | 2017-09-28 13:29:05 +0900 | [diff] [blame] | 139 | } |
Artem Strygin | eababa1 | 2018-06-06 12:31:18 +0000 | [diff] [blame] | 140 | |
| 141 | const uint8_t* stream_data = streamAcc->GetData(); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 142 | for (size_t j = 0; j < span.size(); j++) |
| 143 | EXPECT_EQ(span[j], stream_data[j]) << " at byte " << j; |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 144 | } |
| 145 | |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 146 | void CheckCompositeFontWidths(const CPDF_Array* widths_array, |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 147 | CPDF_Font* typed_font) { |
| 148 | // Check that W array is in a format that conforms to PDF spec 1.7 section |
| 149 | // "Glyph Metrics in CIDFonts" (these checks are not |
| 150 | // implementation-specific). |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 151 | EXPECT_GT(widths_array->size(), 1u); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 152 | int num_cids_checked = 0; |
| 153 | int cur_cid = 0; |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 154 | for (size_t idx = 0; idx < widths_array->size(); idx++) { |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 155 | int cid = widths_array->GetNumberAt(idx); |
| 156 | EXPECT_GE(cid, cur_cid); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 157 | ASSERT_FALSE(++idx == widths_array->size()); |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 158 | const CPDF_Object* next = widths_array->GetObjectAt(idx); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 159 | if (next->IsArray()) { |
| 160 | // We are in the c [w1 w2 ...] case |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 161 | const CPDF_Array* arr = next->AsArray(); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 162 | int cnt = static_cast<int>(arr->size()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 163 | size_t inner_idx = 0; |
| 164 | for (cur_cid = cid; cur_cid < cid + cnt; cur_cid++) { |
Nicolas Pena | 2334660 | 2018-01-30 21:42:41 +0000 | [diff] [blame] | 165 | uint32_t width = arr->GetNumberAt(inner_idx++); |
Dan Sinclair | 971a674 | 2018-03-28 19:23:25 +0000 | [diff] [blame] | 166 | EXPECT_EQ(width, typed_font->GetCharWidthF(cur_cid)) |
| 167 | << " at cid " << cur_cid; |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 168 | } |
| 169 | num_cids_checked += cnt; |
| 170 | continue; |
| 171 | } |
| 172 | // Otherwise, are in the c_first c_last w case. |
| 173 | ASSERT_TRUE(next->IsNumber()); |
| 174 | int last_cid = next->AsNumber()->GetInteger(); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 175 | ASSERT_FALSE(++idx == widths_array->size()); |
Nicolas Pena | 2334660 | 2018-01-30 21:42:41 +0000 | [diff] [blame] | 176 | uint32_t width = widths_array->GetNumberAt(idx); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 177 | for (cur_cid = cid; cur_cid <= last_cid; cur_cid++) { |
Dan Sinclair | 971a674 | 2018-03-28 19:23:25 +0000 | [diff] [blame] | 178 | EXPECT_EQ(width, typed_font->GetCharWidthF(cur_cid)) |
| 179 | << " at cid " << cur_cid; |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 180 | } |
| 181 | num_cids_checked += last_cid - cid + 1; |
| 182 | } |
| 183 | // Make sure we have a good amount of cids described |
| 184 | EXPECT_GT(num_cids_checked, 900); |
| 185 | } |
| 186 | CPDF_Document* cpdf_doc() { return cpdf_doc_; } |
| 187 | |
| 188 | private: |
| 189 | CPDF_Document* cpdf_doc_; |
| 190 | }; |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 191 | |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 192 | namespace { |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 193 | |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 194 | const char kExpectedPDF[] = |
| 195 | "%PDF-1.7\r\n" |
| 196 | "%\xA1\xB3\xC5\xD7\r\n" |
| 197 | "1 0 obj\r\n" |
| 198 | "<</Pages 2 0 R /Type/Catalog>>\r\n" |
| 199 | "endobj\r\n" |
| 200 | "2 0 obj\r\n" |
| 201 | "<</Count 1/Kids\\[ 4 0 R \\]/Type/Pages>>\r\n" |
| 202 | "endobj\r\n" |
| 203 | "3 0 obj\r\n" |
| 204 | "<</CreationDate\\(D:.*\\)/Creator\\(PDFium\\)>>\r\n" |
| 205 | "endobj\r\n" |
| 206 | "4 0 obj\r\n" |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 207 | "<</MediaBox\\[ 0 0 640 480\\]/Parent 2 0 R " |
| 208 | "/Resources<</ExtGState<</FXE1 5 0 R >>>>" |
Nicolas Pena | d9d6c29 | 2017-06-06 16:12:10 -0400 | [diff] [blame] | 209 | "/Rotate 0/Type/Page" |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 210 | ">>\r\n" |
| 211 | "endobj\r\n" |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 212 | "5 0 obj\r\n" |
| 213 | "<</BM/Normal/CA 1/ca 1>>\r\n" |
| 214 | "endobj\r\n" |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 215 | "xref\r\n" |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 216 | "0 6\r\n" |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 217 | "0000000000 65535 f\r\n" |
| 218 | "0000000017 00000 n\r\n" |
| 219 | "0000000066 00000 n\r\n" |
| 220 | "0000000122 00000 n\r\n" |
| 221 | "0000000192 00000 n\r\n" |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 222 | "0000000311 00000 n\r\n" |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 223 | "trailer\r\n" |
| 224 | "<<\r\n" |
| 225 | "/Root 1 0 R\r\n" |
| 226 | "/Info 3 0 R\r\n" |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 227 | "/Size 6/ID\\[<.*><.*>\\]>>\r\n" |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 228 | "startxref\r\n" |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 229 | "354\r\n" |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 230 | "%%EOF\r\n"; |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 231 | |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 232 | } // namespace |
| 233 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 234 | TEST_F(FPDFEditEmbedderTest, EmptyCreation) { |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 235 | EXPECT_TRUE(CreateEmptyDocument()); |
weili | 9b777de | 2016-08-19 16:19:46 -0700 | [diff] [blame] | 236 | FPDF_PAGE page = FPDFPage_New(document(), 0, 640.0, 480.0); |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 237 | EXPECT_NE(nullptr, page); |
Nicolas Pena | d9d6c29 | 2017-06-06 16:12:10 -0400 | [diff] [blame] | 238 | // The FPDFPage_GenerateContent call should do nothing. |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 239 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Tom Sepez | 0aec19b | 2016-01-07 12:22:44 -0800 | [diff] [blame] | 240 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
etienneb | 7712c26 | 2016-04-26 08:13:45 -0700 | [diff] [blame] | 241 | |
Nicolas Pena | d9d6c29 | 2017-06-06 16:12:10 -0400 | [diff] [blame] | 242 | EXPECT_THAT(GetString(), testing::MatchesRegex(std::string( |
| 243 | kExpectedPDF, sizeof(kExpectedPDF)))); |
weili | 9b777de | 2016-08-19 16:19:46 -0700 | [diff] [blame] | 244 | FPDF_ClosePage(page); |
Tom Sepez | d483eb4 | 2016-01-06 10:03:59 -0800 | [diff] [blame] | 245 | } |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 246 | |
| 247 | // Regression test for https://crbug.com/667012 |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 248 | TEST_F(FPDFEditEmbedderTest, RasterizePDF) { |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 249 | const char kAllBlackMd5sum[] = "5708fc5c4a8bd0abde99c8e8f0390615"; |
| 250 | |
Lei Zhang | f054289 | 2019-01-17 18:46:27 +0000 | [diff] [blame] | 251 | // Get the bitmap for the original document. |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 252 | ScopedFPDFBitmap orig_bitmap; |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 253 | { |
| 254 | EXPECT_TRUE(OpenDocument("black.pdf")); |
| 255 | FPDF_PAGE orig_page = LoadPage(0); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 256 | ASSERT_TRUE(orig_page); |
| 257 | orig_bitmap = RenderLoadedPage(orig_page); |
| 258 | CompareBitmap(orig_bitmap.get(), 612, 792, kAllBlackMd5sum); |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 259 | UnloadPage(orig_page); |
| 260 | } |
| 261 | |
| 262 | // Create a new document from |orig_bitmap| and save it. |
| 263 | { |
| 264 | FPDF_DOCUMENT temp_doc = FPDF_CreateNewDocument(); |
| 265 | FPDF_PAGE temp_page = FPDFPage_New(temp_doc, 0, 612, 792); |
| 266 | |
| 267 | // Add the bitmap to an image object and add the image object to the output |
| 268 | // page. |
Lei Zhang | cbd8957 | 2017-03-15 17:35:47 -0700 | [diff] [blame] | 269 | FPDF_PAGEOBJECT temp_img = FPDFPageObj_NewImageObj(temp_doc); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 270 | EXPECT_TRUE( |
| 271 | FPDFImageObj_SetBitmap(&temp_page, 1, temp_img, orig_bitmap.get())); |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 272 | EXPECT_TRUE(FPDFImageObj_SetMatrix(temp_img, 612, 0, 0, 792, 0, 0)); |
| 273 | FPDFPage_InsertObject(temp_page, temp_img); |
| 274 | EXPECT_TRUE(FPDFPage_GenerateContent(temp_page)); |
| 275 | EXPECT_TRUE(FPDF_SaveAsCopy(temp_doc, this, 0)); |
| 276 | FPDF_ClosePage(temp_page); |
| 277 | FPDF_CloseDocument(temp_doc); |
| 278 | } |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 279 | |
| 280 | // Get the generated content. Make sure it is at least as big as the original |
| 281 | // PDF. |
Lei Zhang | d72fd58 | 2018-07-27 19:37:27 +0000 | [diff] [blame] | 282 | EXPECT_GT(GetString().size(), 923u); |
Dan Sinclair | 04e4dc8 | 2017-10-18 12:17:14 -0400 | [diff] [blame] | 283 | VerifySavedDocument(612, 792, kAllBlackMd5sum); |
thestig | dc7ec03 | 2016-11-21 15:32:52 -0800 | [diff] [blame] | 284 | } |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 285 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 286 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 287 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 288 | #define MAYBE_AddPaths DISABLED_AddPaths |
| 289 | #else |
| 290 | #define MAYBE_AddPaths AddPaths |
| 291 | #endif |
| 292 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddPaths) { |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 293 | // Start with a blank page |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 294 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 295 | ASSERT_TRUE(page); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 296 | |
| 297 | // We will first add a red rectangle |
| 298 | FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 299 | ASSERT_TRUE(red_rect); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 300 | // Expect false when trying to set colors out of range |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 301 | EXPECT_FALSE(FPDFPageObj_SetStrokeColor(red_rect, 100, 100, 100, 300)); |
| 302 | EXPECT_FALSE(FPDFPageObj_SetFillColor(red_rect, 200, 256, 200, 0)); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 303 | |
| 304 | // Fill rectangle with red and insert to the page |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 305 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255)); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 306 | EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
Miklos Vajna | 491112b | 2018-05-30 13:30:10 +0000 | [diff] [blame] | 307 | |
| 308 | int fillmode = FPDF_FILLMODE_NONE; |
| 309 | FPDF_BOOL stroke = true; |
| 310 | EXPECT_TRUE(FPDFPath_GetDrawMode(red_rect, &fillmode, &stroke)); |
| 311 | EXPECT_EQ(FPDF_FILLMODE_ALTERNATE, fillmode); |
| 312 | EXPECT_FALSE(stroke); |
| 313 | |
Lei Zhang | 8da9823 | 2019-12-11 23:29:33 +0000 | [diff] [blame] | 314 | static const FS_MATRIX kMatrix = {1, 2, 3, 4, 5, 6}; |
| 315 | EXPECT_FALSE(FPDFPath_SetMatrix(nullptr, &kMatrix)); |
| 316 | EXPECT_TRUE(FPDFPath_SetMatrix(red_rect, &kMatrix)); |
| 317 | |
| 318 | FS_MATRIX matrix; |
| 319 | EXPECT_FALSE(FPDFPath_GetMatrix(nullptr, &matrix)); |
| 320 | EXPECT_TRUE(FPDFPath_GetMatrix(red_rect, &matrix)); |
| 321 | EXPECT_FLOAT_EQ(1.0f, matrix.a); |
| 322 | EXPECT_FLOAT_EQ(2.0f, matrix.b); |
| 323 | EXPECT_FLOAT_EQ(3.0f, matrix.c); |
| 324 | EXPECT_FLOAT_EQ(4.0f, matrix.d); |
| 325 | EXPECT_FLOAT_EQ(5.0f, matrix.e); |
| 326 | EXPECT_FLOAT_EQ(6.0f, matrix.f); |
| 327 | |
| 328 | // Set back the identity matrix. |
| 329 | matrix = {1.0f, 0.0f, 0.0f, 1.0f, 0.0f, 0.0f}; |
| 330 | EXPECT_TRUE(FPDFPath_SetMatrix(red_rect, &matrix)); |
Miklos Vajna | 97f4d67 | 2018-06-04 14:47:17 +0000 | [diff] [blame] | 331 | |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 332 | FPDFPage_InsertObject(page, red_rect); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 333 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 334 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 335 | CompareBitmap(page_bitmap.get(), 612, 792, kRedRectangleChecksum); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 336 | } |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 337 | |
| 338 | // Now add to that a green rectangle with some medium alpha |
| 339 | FPDF_PAGEOBJECT green_rect = FPDFPageObj_CreateNewRect(100, 100, 40, 40); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 340 | EXPECT_TRUE(FPDFPageObj_SetFillColor(green_rect, 0, 255, 0, 128)); |
Miklos Vajna | ed4705b | 2017-04-05 09:24:50 +0200 | [diff] [blame] | 341 | |
Miklos Vajna | 1ef04c9 | 2017-05-08 18:14:19 +0200 | [diff] [blame] | 342 | // Make sure the type of the rectangle is a path. |
| 343 | EXPECT_EQ(FPDF_PAGEOBJ_PATH, FPDFPageObj_GetType(green_rect)); |
| 344 | |
Miklos Vajna | ed4705b | 2017-04-05 09:24:50 +0200 | [diff] [blame] | 345 | // Make sure we get back the same color we set previously. |
| 346 | unsigned int R; |
| 347 | unsigned int G; |
| 348 | unsigned int B; |
| 349 | unsigned int A; |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 350 | EXPECT_TRUE(FPDFPageObj_GetFillColor(green_rect, &R, &G, &B, &A)); |
Lei Zhang | d72fd58 | 2018-07-27 19:37:27 +0000 | [diff] [blame] | 351 | EXPECT_EQ(0u, R); |
| 352 | EXPECT_EQ(255u, G); |
| 353 | EXPECT_EQ(0u, B); |
| 354 | EXPECT_EQ(128u, A); |
Miklos Vajna | ed4705b | 2017-04-05 09:24:50 +0200 | [diff] [blame] | 355 | |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 356 | // Make sure the path has 5 points (1 FXPT_TYPE::MoveTo and 4 |
| 357 | // FXPT_TYPE::LineTo). |
Miklos Vajna | 0150a54 | 2017-09-21 21:46:56 +0200 | [diff] [blame] | 358 | ASSERT_EQ(5, FPDFPath_CountSegments(green_rect)); |
Miklos Vajna | 36eed87 | 2017-09-20 22:52:43 +0200 | [diff] [blame] | 359 | // Verify actual coordinates. |
| 360 | FPDF_PATHSEGMENT segment = FPDFPath_GetPathSegment(green_rect, 0); |
| 361 | float x; |
| 362 | float y; |
| 363 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 364 | EXPECT_EQ(100, x); |
| 365 | EXPECT_EQ(100, y); |
| 366 | EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment)); |
| 367 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 368 | segment = FPDFPath_GetPathSegment(green_rect, 1); |
| 369 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 370 | EXPECT_EQ(100, x); |
| 371 | EXPECT_EQ(140, y); |
| 372 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 373 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 374 | segment = FPDFPath_GetPathSegment(green_rect, 2); |
| 375 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 376 | EXPECT_EQ(140, x); |
| 377 | EXPECT_EQ(140, y); |
| 378 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 379 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 380 | segment = FPDFPath_GetPathSegment(green_rect, 3); |
| 381 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 382 | EXPECT_EQ(140, x); |
| 383 | EXPECT_EQ(100, y); |
| 384 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 385 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 386 | segment = FPDFPath_GetPathSegment(green_rect, 4); |
| 387 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 388 | EXPECT_EQ(100, x); |
| 389 | EXPECT_EQ(100, y); |
| 390 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 391 | EXPECT_TRUE(FPDFPathSegment_GetClose(segment)); |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 392 | |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 393 | EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect, FPDF_FILLMODE_WINDING, 0)); |
| 394 | FPDFPage_InsertObject(page, green_rect); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 395 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 396 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 397 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 398 | "7b0b87604594e773add528fae567a558"); |
| 399 | } |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 400 | |
| 401 | // Add a black triangle. |
| 402 | FPDF_PAGEOBJECT black_path = FPDFPageObj_CreateNewPath(400, 100); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 403 | EXPECT_TRUE(FPDFPageObj_SetFillColor(black_path, 0, 0, 0, 200)); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 404 | EXPECT_TRUE(FPDFPath_SetDrawMode(black_path, FPDF_FILLMODE_ALTERNATE, 0)); |
| 405 | EXPECT_TRUE(FPDFPath_LineTo(black_path, 400, 200)); |
| 406 | EXPECT_TRUE(FPDFPath_LineTo(black_path, 300, 100)); |
| 407 | EXPECT_TRUE(FPDFPath_Close(black_path)); |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 408 | |
| 409 | // Make sure the path has 3 points (1 FXPT_TYPE::MoveTo and 2 |
| 410 | // FXPT_TYPE::LineTo). |
Miklos Vajna | 0150a54 | 2017-09-21 21:46:56 +0200 | [diff] [blame] | 411 | ASSERT_EQ(3, FPDFPath_CountSegments(black_path)); |
Miklos Vajna | 36eed87 | 2017-09-20 22:52:43 +0200 | [diff] [blame] | 412 | // Verify actual coordinates. |
| 413 | segment = FPDFPath_GetPathSegment(black_path, 0); |
| 414 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 415 | EXPECT_EQ(400, x); |
| 416 | EXPECT_EQ(100, y); |
| 417 | EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment)); |
| 418 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 419 | segment = FPDFPath_GetPathSegment(black_path, 1); |
| 420 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 421 | EXPECT_EQ(400, x); |
| 422 | EXPECT_EQ(200, y); |
| 423 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 424 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 425 | segment = FPDFPath_GetPathSegment(black_path, 2); |
| 426 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 427 | EXPECT_EQ(300, x); |
| 428 | EXPECT_EQ(100, y); |
| 429 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 430 | EXPECT_TRUE(FPDFPathSegment_GetClose(segment)); |
| 431 | // Make sure out of bounds index access fails properly. |
| 432 | EXPECT_EQ(nullptr, FPDFPath_GetPathSegment(black_path, 3)); |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 433 | |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 434 | FPDFPage_InsertObject(page, black_path); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 435 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 436 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 437 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 438 | "eadc8020a14dfcf091da2688733d8806"); |
| 439 | } |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 440 | |
| 441 | // Now add a more complex blue path. |
| 442 | FPDF_PAGEOBJECT blue_path = FPDFPageObj_CreateNewPath(200, 200); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 443 | EXPECT_TRUE(FPDFPageObj_SetFillColor(blue_path, 0, 0, 255, 255)); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 444 | EXPECT_TRUE(FPDFPath_SetDrawMode(blue_path, FPDF_FILLMODE_WINDING, 0)); |
| 445 | EXPECT_TRUE(FPDFPath_LineTo(blue_path, 230, 230)); |
| 446 | EXPECT_TRUE(FPDFPath_BezierTo(blue_path, 250, 250, 280, 280, 300, 300)); |
| 447 | EXPECT_TRUE(FPDFPath_LineTo(blue_path, 325, 325)); |
| 448 | EXPECT_TRUE(FPDFPath_LineTo(blue_path, 350, 325)); |
| 449 | EXPECT_TRUE(FPDFPath_BezierTo(blue_path, 375, 330, 390, 360, 400, 400)); |
| 450 | EXPECT_TRUE(FPDFPath_Close(blue_path)); |
| 451 | FPDFPage_InsertObject(page, blue_path); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 452 | const char kLastMD5[] = "9823e1a21bd9b72b6a442ba4f12af946"; |
| 453 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 454 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 455 | CompareBitmap(page_bitmap.get(), 612, 792, kLastMD5); |
| 456 | } |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 457 | |
| 458 | // Now save the result, closing the page and document |
Nicolas Pena | 207b727 | 2017-05-26 17:37:06 -0400 | [diff] [blame] | 459 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 460 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 461 | FPDF_ClosePage(page); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 462 | |
| 463 | // Render the saved result |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 464 | VerifySavedDocument(612, 792, kLastMD5); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 465 | } |
| 466 | |
Miklos Vajna | 45501f3 | 2019-07-30 07:41:02 +0000 | [diff] [blame] | 467 | TEST_F(FPDFEditEmbedderTest, ClipPath) { |
| 468 | // Load document with a clipped rectangle. |
| 469 | EXPECT_TRUE(OpenDocument("clip_path.pdf")); |
| 470 | FPDF_PAGE page = LoadPage(0); |
| 471 | ASSERT_TRUE(page); |
| 472 | |
| 473 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 474 | |
| 475 | FPDF_PAGEOBJECT triangle = FPDFPage_GetObject(page, 0); |
| 476 | ASSERT_TRUE(triangle); |
| 477 | |
| 478 | // Test that we got the expected triangle. |
| 479 | ASSERT_EQ(4, FPDFPath_CountSegments(triangle)); |
| 480 | |
| 481 | FPDF_PATHSEGMENT segment = FPDFPath_GetPathSegment(triangle, 0); |
| 482 | float x; |
| 483 | float y; |
| 484 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 485 | EXPECT_EQ(10, x); |
| 486 | EXPECT_EQ(10, y); |
| 487 | EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment)); |
| 488 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 489 | |
| 490 | segment = FPDFPath_GetPathSegment(triangle, 1); |
| 491 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 492 | EXPECT_EQ(25, x); |
| 493 | EXPECT_EQ(40, y); |
| 494 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 495 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 496 | |
| 497 | segment = FPDFPath_GetPathSegment(triangle, 2); |
| 498 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 499 | EXPECT_EQ(40, x); |
| 500 | EXPECT_EQ(10, y); |
| 501 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 502 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 503 | |
| 504 | segment = FPDFPath_GetPathSegment(triangle, 3); |
| 505 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 506 | EXPECT_TRUE(FPDFPathSegment_GetClose(segment)); |
| 507 | |
| 508 | // Test FPDFPageObj_GetClipPath(). |
| 509 | ASSERT_EQ(nullptr, FPDFPageObj_GetClipPath(nullptr)); |
| 510 | |
| 511 | FPDF_CLIPPATH clip_path = FPDFPageObj_GetClipPath(triangle); |
| 512 | ASSERT_TRUE(clip_path); |
| 513 | |
| 514 | // Test FPDFClipPath_CountPaths(). |
| 515 | ASSERT_EQ(-1, FPDFClipPath_CountPaths(nullptr)); |
| 516 | ASSERT_EQ(1, FPDFClipPath_CountPaths(clip_path)); |
| 517 | |
| 518 | // Test FPDFClipPath_CountPathSegments(). |
| 519 | ASSERT_EQ(-1, FPDFClipPath_CountPathSegments(nullptr, 0)); |
| 520 | ASSERT_EQ(-1, FPDFClipPath_CountPathSegments(clip_path, -1)); |
| 521 | ASSERT_EQ(-1, FPDFClipPath_CountPathSegments(clip_path, 1)); |
| 522 | ASSERT_EQ(4, FPDFClipPath_CountPathSegments(clip_path, 0)); |
| 523 | |
| 524 | // FPDFClipPath_GetPathSegment() negative testing. |
| 525 | ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(nullptr, 0, 0)); |
| 526 | ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, -1, 0)); |
| 527 | ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, 1, 0)); |
| 528 | ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, 0, -1)); |
| 529 | ASSERT_EQ(nullptr, FPDFClipPath_GetPathSegment(clip_path, 0, 4)); |
| 530 | |
| 531 | // FPDFClipPath_GetPathSegment() positive testing. |
| 532 | segment = FPDFClipPath_GetPathSegment(clip_path, 0, 0); |
| 533 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 534 | EXPECT_EQ(10, x); |
| 535 | EXPECT_EQ(15, y); |
| 536 | EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment)); |
| 537 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 538 | |
| 539 | segment = FPDFClipPath_GetPathSegment(clip_path, 0, 1); |
| 540 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 541 | EXPECT_EQ(40, x); |
| 542 | EXPECT_EQ(15, y); |
| 543 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 544 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 545 | |
| 546 | segment = FPDFClipPath_GetPathSegment(clip_path, 0, 2); |
| 547 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 548 | EXPECT_EQ(40, x); |
| 549 | EXPECT_EQ(35, y); |
| 550 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 551 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 552 | |
| 553 | segment = FPDFClipPath_GetPathSegment(clip_path, 0, 3); |
| 554 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 555 | EXPECT_EQ(10, x); |
| 556 | EXPECT_EQ(35, y); |
| 557 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 558 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 559 | |
| 560 | UnloadPage(page); |
| 561 | } |
| 562 | |
Lei Zhang | 571f932 | 2019-09-25 12:40:25 +0000 | [diff] [blame] | 563 | TEST_F(FPDFEditEmbedderTest, BUG_1399) { |
| 564 | // Load document with a clipped rectangle. |
| 565 | EXPECT_TRUE(OpenDocument("bug_1399.pdf")); |
| 566 | FPDF_PAGE page = LoadPage(0); |
| 567 | ASSERT_TRUE(page); |
| 568 | |
| 569 | ASSERT_EQ(7, FPDFPage_CountObjects(page)); |
| 570 | |
| 571 | FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 0); |
| 572 | ASSERT_TRUE(obj); |
| 573 | |
| 574 | ASSERT_EQ(2, FPDFPath_CountSegments(obj)); |
| 575 | |
| 576 | FPDF_PATHSEGMENT segment = FPDFPath_GetPathSegment(obj, 0); |
| 577 | float x; |
| 578 | float y; |
| 579 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 580 | EXPECT_FLOAT_EQ(107.718f, x); |
| 581 | EXPECT_FLOAT_EQ(719.922f, y); |
| 582 | EXPECT_EQ(FPDF_SEGMENT_MOVETO, FPDFPathSegment_GetType(segment)); |
| 583 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 584 | |
| 585 | segment = FPDFPath_GetPathSegment(obj, 1); |
| 586 | EXPECT_TRUE(FPDFPathSegment_GetPoint(segment, &x, &y)); |
| 587 | EXPECT_FLOAT_EQ(394.718f, x); |
| 588 | EXPECT_FLOAT_EQ(719.922f, y); |
| 589 | EXPECT_EQ(FPDF_SEGMENT_LINETO, FPDFPathSegment_GetType(segment)); |
| 590 | EXPECT_FALSE(FPDFPathSegment_GetClose(segment)); |
| 591 | |
| 592 | FPDF_CLIPPATH clip_path = FPDFPageObj_GetClipPath(obj); |
| 593 | ASSERT_TRUE(clip_path); |
| 594 | |
| 595 | EXPECT_EQ(-1, FPDFClipPath_CountPaths(clip_path)); |
| 596 | EXPECT_EQ(-1, FPDFClipPath_CountPathSegments(clip_path, 0)); |
| 597 | EXPECT_FALSE(FPDFClipPath_GetPathSegment(clip_path, 0, 0)); |
| 598 | |
| 599 | UnloadPage(page); |
| 600 | } |
| 601 | |
Hui Yingst | cb4203d | 2020-06-09 01:48:47 +0000 | [diff] [blame] | 602 | TEST_F(FPDFEditEmbedderTest, SetText) { |
Henrique Nakashima | 5ebfd64 | 2018-06-07 15:18:55 +0000 | [diff] [blame] | 603 | // Load document with some text. |
| 604 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 605 | FPDF_PAGE page = LoadPage(0); |
| 606 | ASSERT_TRUE(page); |
| 607 | |
| 608 | // Get the "Hello, world!" text object and change it. |
| 609 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 610 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 611 | ASSERT_TRUE(page_object); |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 612 | ScopedFPDFWideString text1 = GetFPDFWideString(L"Changed for SetText test"); |
Henrique Nakashima | 5ebfd64 | 2018-06-07 15:18:55 +0000 | [diff] [blame] | 613 | EXPECT_TRUE(FPDFText_SetText(page_object, text1.get())); |
| 614 | |
| 615 | // Verify the "Hello, world!" text is gone and "Changed for SetText test" is |
| 616 | // now displayed. |
| 617 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
Hui Yingst | cb4203d | 2020-06-09 01:48:47 +0000 | [diff] [blame] | 618 | |
| 619 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 620 | #if defined(OS_WIN) |
| 621 | const char kChangedChecksum[] = "119f828345f547c68e9454418fb7d80d"; |
Henrique Nakashima | 5ebfd64 | 2018-06-07 15:18:55 +0000 | [diff] [blame] | 622 | #else |
Hui Yingst | cb4203d | 2020-06-09 01:48:47 +0000 | [diff] [blame] | 623 | const char kChangedChecksum[] = "6bc5171f4eb329474989c6ccfa3d6303"; |
| 624 | #endif // defined(OS_WIN) |
| 625 | #else |
| 626 | #if defined(OS_WIN) |
| 627 | const char kChangedChecksum[] = "3137fdb27962671f5c3963a5e965eff5"; |
| 628 | #elif defined(OS_MACOSX) |
| 629 | const char kChangedChecksum[] = "904132275a1144ea06b0694537c80b4c"; |
| 630 | #else |
| 631 | const char kChangedChecksum[] = "a0c4ea6620772991f66bf7130379b08a"; |
Henrique Nakashima | 5ebfd64 | 2018-06-07 15:18:55 +0000 | [diff] [blame] | 632 | #endif |
Hui Yingst | cb4203d | 2020-06-09 01:48:47 +0000 | [diff] [blame] | 633 | #endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
Henrique Nakashima | 5ebfd64 | 2018-06-07 15:18:55 +0000 | [diff] [blame] | 634 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 635 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | cb4203d | 2020-06-09 01:48:47 +0000 | [diff] [blame] | 636 | CompareBitmap(page_bitmap.get(), 200, 200, kChangedChecksum); |
Henrique Nakashima | 5ebfd64 | 2018-06-07 15:18:55 +0000 | [diff] [blame] | 637 | } |
| 638 | |
| 639 | // Now save the result. |
| 640 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 641 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 642 | |
| 643 | UnloadPage(page); |
| 644 | |
| 645 | // Re-open the file and check the changes were kept in the saved .pdf. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 646 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 5ebfd64 | 2018-06-07 15:18:55 +0000 | [diff] [blame] | 647 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 648 | EXPECT_EQ(2, FPDFPage_CountObjects(saved_page)); |
| 649 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 650 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | cb4203d | 2020-06-09 01:48:47 +0000 | [diff] [blame] | 651 | CompareBitmap(page_bitmap.get(), 200, 200, kChangedChecksum); |
Henrique Nakashima | 5ebfd64 | 2018-06-07 15:18:55 +0000 | [diff] [blame] | 652 | } |
| 653 | |
| 654 | CloseSavedPage(saved_page); |
| 655 | CloseSavedDocument(); |
| 656 | } |
| 657 | |
Hui Yingst | 08c4071 | 2020-04-29 01:37:35 +0000 | [diff] [blame] | 658 | TEST_F(FPDFEditEmbedderTest, RemovePageObject) { |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 659 | // Load document with some text. |
| 660 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 661 | FPDF_PAGE page = LoadPage(0); |
| 662 | ASSERT_TRUE(page); |
| 663 | |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 664 | // Show what the original file looks like. |
| 665 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 666 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 667 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum); |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 668 | } |
| 669 | |
| 670 | // Get the "Hello, world!" text object and remove it. |
| 671 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 672 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 673 | ASSERT_TRUE(page_object); |
| 674 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 675 | |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 676 | // Verify the "Hello, world!" text is gone. |
| 677 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 678 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 679 | CompareBitmap(page_bitmap.get(), 200, 200, kFirstRemovedChecksum); |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 680 | } |
| 681 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 682 | |
| 683 | UnloadPage(page); |
| 684 | FPDFPageObj_Destroy(page_object); |
| 685 | } |
| 686 | |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 687 | void CheckMarkCounts(FPDF_PAGE page, |
| 688 | int start_from, |
| 689 | int expected_object_count, |
| 690 | size_t expected_prime_count, |
| 691 | size_t expected_square_count, |
| 692 | size_t expected_greater_than_ten_count, |
| 693 | size_t expected_bounds_count) { |
| 694 | int object_count = FPDFPage_CountObjects(page); |
| 695 | ASSERT_EQ(expected_object_count, object_count); |
| 696 | |
| 697 | size_t prime_count = 0; |
| 698 | size_t square_count = 0; |
| 699 | size_t greater_than_ten_count = 0; |
| 700 | size_t bounds_count = 0; |
| 701 | for (int i = 0; i < object_count; ++i) { |
| 702 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 703 | |
| 704 | int mark_count = FPDFPageObj_CountMarks(page_object); |
| 705 | for (int j = 0; j < mark_count; ++j) { |
| 706 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j); |
| 707 | |
| 708 | char buffer[256]; |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 709 | unsigned long name_len = 999u; |
| 710 | ASSERT_TRUE( |
| 711 | FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 712 | EXPECT_GT(name_len, 0u); |
| 713 | EXPECT_NE(999u, name_len); |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 714 | std::wstring name = |
| 715 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 716 | if (name == L"Prime") { |
| 717 | prime_count++; |
| 718 | } else if (name == L"Square") { |
| 719 | square_count++; |
| 720 | int expected_square = start_from + i; |
| 721 | EXPECT_EQ(1, FPDFPageObjMark_CountParams(mark)); |
| 722 | |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 723 | unsigned long get_param_key_return = 999u; |
| 724 | ASSERT_TRUE(FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer), |
| 725 | &get_param_key_return)); |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 726 | EXPECT_EQ((6u + 1u) * 2u, get_param_key_return); |
| 727 | std::wstring key = |
| 728 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 729 | EXPECT_EQ(L"Factor", key); |
| 730 | |
| 731 | EXPECT_EQ(FPDF_OBJECT_NUMBER, |
Henrique Nakashima | 94230e5 | 2018-07-11 22:02:02 +0000 | [diff] [blame] | 732 | FPDFPageObjMark_GetParamValueType(mark, "Factor")); |
Henrique Nakashima | 140dead | 2018-07-11 21:40:03 +0000 | [diff] [blame] | 733 | int square_root; |
Henrique Nakashima | 94230e5 | 2018-07-11 22:02:02 +0000 | [diff] [blame] | 734 | EXPECT_TRUE( |
| 735 | FPDFPageObjMark_GetParamIntValue(mark, "Factor", &square_root)); |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 736 | EXPECT_EQ(expected_square, square_root * square_root); |
| 737 | } else if (name == L"GreaterThanTen") { |
| 738 | greater_than_ten_count++; |
| 739 | } else if (name == L"Bounds") { |
| 740 | bounds_count++; |
| 741 | EXPECT_EQ(1, FPDFPageObjMark_CountParams(mark)); |
| 742 | |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 743 | unsigned long get_param_key_return = 999u; |
| 744 | ASSERT_TRUE(FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer), |
| 745 | &get_param_key_return)); |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 746 | EXPECT_EQ((8u + 1u) * 2u, get_param_key_return); |
| 747 | std::wstring key = |
| 748 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 749 | EXPECT_EQ(L"Position", key); |
| 750 | |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 751 | EXPECT_EQ(FPDF_OBJECT_STRING, |
Henrique Nakashima | 94230e5 | 2018-07-11 22:02:02 +0000 | [diff] [blame] | 752 | FPDFPageObjMark_GetParamValueType(mark, "Position")); |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 753 | unsigned long length; |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 754 | EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue( |
| 755 | mark, "Position", buffer, sizeof(buffer), &length)); |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 756 | ASSERT_GT(length, 0u); |
Henrique Nakashima | 140dead | 2018-07-11 21:40:03 +0000 | [diff] [blame] | 757 | std::wstring value = |
| 758 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 759 | |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 760 | // "Position" can be "First", "Last", or "End". |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 761 | if (i == 0) { |
| 762 | EXPECT_EQ((5u + 1u) * 2u, length); |
| 763 | EXPECT_EQ(L"First", value); |
| 764 | } else if (i == object_count - 1) { |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 765 | if (length == (4u + 1u) * 2u) { |
| 766 | EXPECT_EQ(L"Last", value); |
| 767 | } else if (length == (3u + 1u) * 2u) { |
| 768 | EXPECT_EQ(L"End", value); |
| 769 | } else { |
| 770 | FAIL(); |
| 771 | } |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 772 | } else { |
| 773 | FAIL(); |
| 774 | } |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 775 | } else { |
| 776 | FAIL(); |
| 777 | } |
| 778 | } |
| 779 | } |
| 780 | |
| 781 | // Expect certain number of tagged objects. The test file contains strings |
| 782 | // from 1 to 19. |
| 783 | EXPECT_EQ(expected_prime_count, prime_count); |
| 784 | EXPECT_EQ(expected_square_count, square_count); |
| 785 | EXPECT_EQ(expected_greater_than_ten_count, greater_than_ten_count); |
| 786 | EXPECT_EQ(expected_bounds_count, bounds_count); |
| 787 | } |
| 788 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 789 | TEST_F(FPDFEditEmbedderTest, ReadMarkedObjectsIndirectDict) { |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 790 | // Load document with some text marked with an indirect property. |
| 791 | EXPECT_TRUE(OpenDocument("text_in_page_marked_indirect.pdf")); |
| 792 | FPDF_PAGE page = LoadPage(0); |
| 793 | ASSERT_TRUE(page); |
| 794 | |
| 795 | CheckMarkCounts(page, 1, 19, 8, 4, 9, 1); |
| 796 | |
| 797 | UnloadPage(page); |
| 798 | } |
| 799 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 800 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 801 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 802 | #define MAYBE_RemoveMarkedObjectsPrime DISABLED_RemoveMarkedObjectsPrime |
| 803 | #else |
| 804 | #define MAYBE_RemoveMarkedObjectsPrime RemoveMarkedObjectsPrime |
| 805 | #endif |
| 806 | TEST_F(FPDFEditEmbedderTest, MAYBE_RemoveMarkedObjectsPrime) { |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 807 | // Load document with some text. |
| 808 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 809 | FPDF_PAGE page = LoadPage(0); |
| 810 | ASSERT_TRUE(page); |
| 811 | |
| 812 | // Show what the original file looks like. |
| 813 | { |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 814 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 815 | const char kOriginalMD5[] = "adf815e53c788a5272b4df07c610a1da"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 816 | #elif defined(OS_WIN) |
Lei Zhang | a2b7073 | 2019-06-25 08:34:22 +0000 | [diff] [blame] | 817 | const char kOriginalMD5[] = "00542ee435b37749c4453be63bf7bdb6"; |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 818 | #else |
Lei Zhang | a2b7073 | 2019-06-25 08:34:22 +0000 | [diff] [blame] | 819 | const char kOriginalMD5[] = "41647268d5911d049801803b15c2dfb0"; |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 820 | #endif |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 821 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 822 | CompareBitmap(page_bitmap.get(), 200, 200, kOriginalMD5); |
| 823 | } |
| 824 | |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 825 | constexpr int expected_object_count = 19; |
| 826 | CheckMarkCounts(page, 1, expected_object_count, 8, 4, 9, 1); |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 827 | |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 828 | // Get all objects marked with "Prime" |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 829 | std::vector<FPDF_PAGEOBJECT> primes; |
Henrique Nakashima | cbed949 | 2018-07-10 21:54:26 +0000 | [diff] [blame] | 830 | for (int i = 0; i < expected_object_count; ++i) { |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 831 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 832 | |
| 833 | int mark_count = FPDFPageObj_CountMarks(page_object); |
| 834 | for (int j = 0; j < mark_count; ++j) { |
| 835 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j); |
| 836 | |
| 837 | char buffer[256]; |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 838 | unsigned long name_len = 999u; |
| 839 | ASSERT_TRUE( |
| 840 | FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 841 | EXPECT_GT(name_len, 0u); |
| 842 | EXPECT_NE(999u, name_len); |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 843 | std::wstring name = |
| 844 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 845 | if (name == L"Prime") { |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 846 | primes.push_back(page_object); |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 847 | } |
| 848 | } |
| 849 | } |
| 850 | |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 851 | // Remove all objects marked with "Prime". |
| 852 | for (FPDF_PAGEOBJECT page_object : primes) { |
| 853 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 854 | FPDFPageObj_Destroy(page_object); |
| 855 | } |
| 856 | |
| 857 | EXPECT_EQ(11, FPDFPage_CountObjects(page)); |
| 858 | |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 859 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 860 | const char kNonPrimesMD5[] = "d29e2ddff56e0d12f340794d26796400"; |
| 861 | const char kNonPrimesAfterSaveMD5[] = "10eff2cd0037b661496981779601fa6f"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 862 | #elif defined(OS_WIN) |
Lei Zhang | a2b7073 | 2019-06-25 08:34:22 +0000 | [diff] [blame] | 863 | const char kNonPrimesMD5[] = "86e371fdae30c2471f476631f3f93413"; |
| 864 | const char kNonPrimesAfterSaveMD5[] = "86e371fdae30c2471f476631f3f93413"; |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 865 | #else |
Lei Zhang | a2b7073 | 2019-06-25 08:34:22 +0000 | [diff] [blame] | 866 | const char kNonPrimesMD5[] = "67ab13115d0cc34e99a1003c28047b40"; |
| 867 | const char kNonPrimesAfterSaveMD5[] = "67ab13115d0cc34e99a1003c28047b40"; |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 868 | #endif |
Henrique Nakashima | a5078b7 | 2018-09-10 17:09:42 +0000 | [diff] [blame] | 869 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 870 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 871 | CompareBitmap(page_bitmap.get(), 200, 200, kNonPrimesMD5); |
| 872 | } |
| 873 | |
Henrique Nakashima | a5078b7 | 2018-09-10 17:09:42 +0000 | [diff] [blame] | 874 | // Save the file. |
| 875 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 876 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 877 | UnloadPage(page); |
Henrique Nakashima | a5078b7 | 2018-09-10 17:09:42 +0000 | [diff] [blame] | 878 | |
| 879 | // Re-open the file and check the prime marks are not there anymore. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 880 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | a5078b7 | 2018-09-10 17:09:42 +0000 | [diff] [blame] | 881 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 882 | EXPECT_EQ(11, FPDFPage_CountObjects(saved_page)); |
| 883 | |
| 884 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 885 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Manoj Biswas | 7daf7a2 | 2019-03-22 23:02:34 +0000 | [diff] [blame] | 886 | CompareBitmap(page_bitmap.get(), 200, 200, kNonPrimesAfterSaveMD5); |
Henrique Nakashima | a5078b7 | 2018-09-10 17:09:42 +0000 | [diff] [blame] | 887 | } |
| 888 | |
| 889 | CloseSavedPage(saved_page); |
| 890 | CloseSavedDocument(); |
Henrique Nakashima | c90adc5 | 2018-03-27 16:26:44 +0000 | [diff] [blame] | 891 | } |
| 892 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 893 | TEST_F(FPDFEditEmbedderTest, RemoveMarks) { |
Henrique Nakashima | fed4adb | 2018-07-13 19:47:22 +0000 | [diff] [blame] | 894 | // Load document with some text. |
| 895 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 896 | FPDF_PAGE page = LoadPage(0); |
| 897 | ASSERT_TRUE(page); |
| 898 | |
| 899 | constexpr int kExpectedObjectCount = 19; |
| 900 | CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1); |
| 901 | |
| 902 | // Remove all "Prime" content marks. |
| 903 | for (int i = 0; i < kExpectedObjectCount; ++i) { |
| 904 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 905 | |
| 906 | int mark_count = FPDFPageObj_CountMarks(page_object); |
| 907 | for (int j = mark_count - 1; j >= 0; --j) { |
| 908 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j); |
| 909 | |
| 910 | char buffer[256]; |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 911 | unsigned long name_len = 999u; |
| 912 | ASSERT_TRUE( |
| 913 | FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 914 | EXPECT_GT(name_len, 0u); |
| 915 | EXPECT_NE(999u, name_len); |
Henrique Nakashima | fed4adb | 2018-07-13 19:47:22 +0000 | [diff] [blame] | 916 | std::wstring name = |
| 917 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 918 | if (name == L"Prime") { |
| 919 | // Remove mark. |
| 920 | EXPECT_TRUE(FPDFPageObj_RemoveMark(page_object, mark)); |
| 921 | |
| 922 | // Verify there is now one fewer mark in the page object. |
| 923 | EXPECT_EQ(mark_count - 1, FPDFPageObj_CountMarks(page_object)); |
| 924 | } |
| 925 | } |
| 926 | } |
| 927 | |
| 928 | // Verify there are 0 "Prime" content marks now. |
| 929 | CheckMarkCounts(page, 1, kExpectedObjectCount, 0, 4, 9, 1); |
| 930 | |
| 931 | // Save the file. |
| 932 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 933 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 934 | UnloadPage(page); |
| 935 | |
| 936 | // Re-open the file and check the prime marks are not there anymore. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 937 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | fed4adb | 2018-07-13 19:47:22 +0000 | [diff] [blame] | 938 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 939 | |
| 940 | CheckMarkCounts(saved_page, 1, kExpectedObjectCount, 0, 4, 9, 1); |
| 941 | |
| 942 | CloseSavedPage(saved_page); |
| 943 | CloseSavedDocument(); |
| 944 | } |
| 945 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 946 | TEST_F(FPDFEditEmbedderTest, RemoveMarkParam) { |
Henrique Nakashima | cf403ba | 2018-07-13 20:12:41 +0000 | [diff] [blame] | 947 | // Load document with some text. |
| 948 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 949 | FPDF_PAGE page = LoadPage(0); |
| 950 | ASSERT_TRUE(page); |
| 951 | |
| 952 | constexpr int kExpectedObjectCount = 19; |
| 953 | CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1); |
| 954 | |
| 955 | // Remove all "Square" content marks parameters. |
| 956 | for (int i = 0; i < kExpectedObjectCount; ++i) { |
| 957 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 958 | |
| 959 | int mark_count = FPDFPageObj_CountMarks(page_object); |
| 960 | for (int j = 0; j < mark_count; ++j) { |
| 961 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j); |
| 962 | |
| 963 | char buffer[256]; |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 964 | unsigned long name_len = 999u; |
| 965 | ASSERT_TRUE( |
| 966 | FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 967 | EXPECT_GT(name_len, 0u); |
| 968 | EXPECT_NE(999u, name_len); |
Henrique Nakashima | cf403ba | 2018-07-13 20:12:41 +0000 | [diff] [blame] | 969 | std::wstring name = |
| 970 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 971 | if (name == L"Square") { |
| 972 | // Show the mark has a "Factor" parameter. |
| 973 | int out_value; |
| 974 | EXPECT_TRUE( |
| 975 | FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value)); |
| 976 | |
| 977 | // Remove parameter. |
| 978 | EXPECT_TRUE(FPDFPageObjMark_RemoveParam(page_object, mark, "Factor")); |
| 979 | |
| 980 | // Verify the "Factor" parameter is gone. |
| 981 | EXPECT_FALSE( |
| 982 | FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value)); |
| 983 | } |
| 984 | } |
| 985 | } |
| 986 | |
| 987 | // Save the file. |
| 988 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 989 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 990 | UnloadPage(page); |
| 991 | |
| 992 | // Re-open the file and check the "Factor" parameters are still gone. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 993 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | cf403ba | 2018-07-13 20:12:41 +0000 | [diff] [blame] | 994 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 995 | |
| 996 | size_t square_count = 0; |
| 997 | for (int i = 0; i < kExpectedObjectCount; ++i) { |
| 998 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(saved_page, i); |
| 999 | |
| 1000 | int mark_count = FPDFPageObj_CountMarks(page_object); |
| 1001 | for (int j = 0; j < mark_count; ++j) { |
| 1002 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, j); |
| 1003 | |
| 1004 | char buffer[256]; |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 1005 | unsigned long name_len = 999u; |
| 1006 | ASSERT_TRUE( |
| 1007 | FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 1008 | EXPECT_GT(name_len, 0u); |
| 1009 | EXPECT_NE(999u, name_len); |
Henrique Nakashima | cf403ba | 2018-07-13 20:12:41 +0000 | [diff] [blame] | 1010 | std::wstring name = |
| 1011 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 1012 | if (name == L"Square") { |
| 1013 | // Verify the "Factor" parameter is still gone. |
| 1014 | int out_value; |
| 1015 | EXPECT_FALSE( |
| 1016 | FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value)); |
| 1017 | |
| 1018 | ++square_count; |
| 1019 | } |
| 1020 | } |
| 1021 | } |
| 1022 | |
| 1023 | // Verify the parameters are gone, but the marks are not. |
| 1024 | EXPECT_EQ(4u, square_count); |
| 1025 | |
| 1026 | CloseSavedPage(saved_page); |
| 1027 | CloseSavedDocument(); |
| 1028 | } |
| 1029 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1030 | TEST_F(FPDFEditEmbedderTest, MaintainMarkedObjects) { |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 1031 | // Load document with some text. |
| 1032 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 1033 | FPDF_PAGE page = LoadPage(0); |
| 1034 | ASSERT_TRUE(page); |
| 1035 | |
| 1036 | // Iterate over all objects, counting the number of times each content mark |
| 1037 | // name appears. |
| 1038 | CheckMarkCounts(page, 1, 19, 8, 4, 9, 1); |
| 1039 | |
| 1040 | // Remove first page object. |
| 1041 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 1042 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1043 | FPDFPageObj_Destroy(page_object); |
| 1044 | |
| 1045 | CheckMarkCounts(page, 2, 18, 8, 3, 9, 1); |
| 1046 | |
| 1047 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1048 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1049 | |
| 1050 | UnloadPage(page); |
| 1051 | |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1052 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 1053 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1054 | |
| 1055 | CheckMarkCounts(saved_page, 2, 18, 8, 3, 9, 1); |
| 1056 | |
| 1057 | CloseSavedPage(saved_page); |
| 1058 | CloseSavedDocument(); |
| 1059 | } |
| 1060 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1061 | TEST_F(FPDFEditEmbedderTest, MaintainIndirectMarkedObjects) { |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 1062 | // Load document with some text. |
| 1063 | EXPECT_TRUE(OpenDocument("text_in_page_marked_indirect.pdf")); |
| 1064 | FPDF_PAGE page = LoadPage(0); |
| 1065 | ASSERT_TRUE(page); |
| 1066 | |
| 1067 | // Iterate over all objects, counting the number of times each content mark |
| 1068 | // name appears. |
| 1069 | CheckMarkCounts(page, 1, 19, 8, 4, 9, 1); |
| 1070 | |
| 1071 | // Remove first page object. |
| 1072 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 1073 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1074 | FPDFPageObj_Destroy(page_object); |
| 1075 | |
| 1076 | CheckMarkCounts(page, 2, 18, 8, 3, 9, 1); |
| 1077 | |
| 1078 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1079 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1080 | |
| 1081 | UnloadPage(page); |
| 1082 | |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1083 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 1084 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1085 | |
| 1086 | CheckMarkCounts(saved_page, 2, 18, 8, 3, 9, 1); |
| 1087 | |
| 1088 | CloseSavedPage(saved_page); |
| 1089 | CloseSavedDocument(); |
| 1090 | } |
| 1091 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1092 | TEST_F(FPDFEditEmbedderTest, RemoveExistingPageObject) { |
Henrique Nakashima | c49e62e | 2018-04-16 20:58:47 +0000 | [diff] [blame] | 1093 | // Load document with some text. |
| 1094 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 1095 | FPDF_PAGE page = LoadPage(0); |
| 1096 | ASSERT_TRUE(page); |
| 1097 | |
| 1098 | // Get the "Hello, world!" text object and remove it. |
| 1099 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 1100 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 1101 | ASSERT_TRUE(page_object); |
| 1102 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1103 | |
| 1104 | // Verify the "Hello, world!" text is gone. |
| 1105 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 1106 | |
| 1107 | // Save the file |
| 1108 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1109 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1110 | UnloadPage(page); |
| 1111 | FPDFPageObj_Destroy(page_object); |
| 1112 | |
| 1113 | // Re-open the file and check the page object count is still 1. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1114 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | c49e62e | 2018-04-16 20:58:47 +0000 | [diff] [blame] | 1115 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1116 | EXPECT_EQ(1, FPDFPage_CountObjects(saved_page)); |
| 1117 | CloseSavedPage(saved_page); |
| 1118 | CloseSavedDocument(); |
| 1119 | } |
| 1120 | |
Hui Yingst | cb4203d | 2020-06-09 01:48:47 +0000 | [diff] [blame] | 1121 | TEST_F(FPDFEditEmbedderTest, RemoveExistingPageObjectSplitStreamsNotLonely) { |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1122 | // Load document with some text. |
| 1123 | EXPECT_TRUE(OpenDocument("hello_world_split_streams.pdf")); |
| 1124 | FPDF_PAGE page = LoadPage(0); |
| 1125 | ASSERT_TRUE(page); |
| 1126 | |
| 1127 | // Get the "Hello, world!" text object and remove it. There is another object |
| 1128 | // in the same stream that says "Goodbye, world!" |
| 1129 | ASSERT_EQ(3, FPDFPage_CountObjects(page)); |
| 1130 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 1131 | ASSERT_TRUE(page_object); |
| 1132 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1133 | |
| 1134 | // Verify the "Hello, world!" text is gone. |
| 1135 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
Hui Yingst | cb4203d | 2020-06-09 01:48:47 +0000 | [diff] [blame] | 1136 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 1137 | #if defined(OS_WIN) |
| 1138 | const char kHelloRemovedChecksum[] = "e05c7837a5cad61305d157720637f969"; |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1139 | #else |
Hui Yingst | cb4203d | 2020-06-09 01:48:47 +0000 | [diff] [blame] | 1140 | const char kHelloRemovedChecksum[] = "deed7dc2754dc80930f3b05e2ac86c94"; |
| 1141 | #endif // defined(OS_WIN) |
| 1142 | #else |
| 1143 | #if defined(OS_WIN) |
| 1144 | const char kHelloRemovedChecksum[] = "a97d4c72c969ba373c2dce675d277e65"; |
| 1145 | #elif defined(OS_MACOSX) |
| 1146 | const char kHelloRemovedChecksum[] = "3b3b27602a86dfe5996a33c42c59885b"; |
| 1147 | #else |
| 1148 | const char kHelloRemovedChecksum[] = "95b92950647a2190e1230911e7a1a0e9"; |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1149 | #endif |
Hui Yingst | cb4203d | 2020-06-09 01:48:47 +0000 | [diff] [blame] | 1150 | #endif // defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1151 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1152 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | cb4203d | 2020-06-09 01:48:47 +0000 | [diff] [blame] | 1153 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloRemovedChecksum); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1154 | } |
| 1155 | |
| 1156 | // Save the file |
| 1157 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1158 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1159 | UnloadPage(page); |
| 1160 | FPDFPageObj_Destroy(page_object); |
| 1161 | |
| 1162 | // Re-open the file and check the page object count is still 2. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1163 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1164 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1165 | |
| 1166 | EXPECT_EQ(2, FPDFPage_CountObjects(saved_page)); |
| 1167 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1168 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | cb4203d | 2020-06-09 01:48:47 +0000 | [diff] [blame] | 1169 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloRemovedChecksum); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1170 | } |
| 1171 | |
| 1172 | CloseSavedPage(saved_page); |
| 1173 | CloseSavedDocument(); |
| 1174 | } |
| 1175 | |
Hui Yingst | 08c4071 | 2020-04-29 01:37:35 +0000 | [diff] [blame] | 1176 | TEST_F(FPDFEditEmbedderTest, RemoveExistingPageObjectSplitStreamsLonely) { |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1177 | // Load document with some text. |
| 1178 | EXPECT_TRUE(OpenDocument("hello_world_split_streams.pdf")); |
| 1179 | FPDF_PAGE page = LoadPage(0); |
| 1180 | ASSERT_TRUE(page); |
| 1181 | |
| 1182 | // Get the "Greetings, world!" text object and remove it. This is the only |
| 1183 | // object in the stream. |
| 1184 | ASSERT_EQ(3, FPDFPage_CountObjects(page)); |
| 1185 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 2); |
| 1186 | ASSERT_TRUE(page_object); |
| 1187 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1188 | |
| 1189 | // Verify the "Greetings, world!" text is gone. |
| 1190 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1191 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1192 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1193 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1194 | } |
| 1195 | |
| 1196 | // Save the file |
| 1197 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1198 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1199 | UnloadPage(page); |
| 1200 | FPDFPageObj_Destroy(page_object); |
| 1201 | |
| 1202 | // Re-open the file and check the page object count is still 2. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1203 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1204 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1205 | |
| 1206 | EXPECT_EQ(2, FPDFPage_CountObjects(saved_page)); |
| 1207 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1208 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1209 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1210 | } |
| 1211 | |
| 1212 | CloseSavedPage(saved_page); |
| 1213 | CloseSavedDocument(); |
| 1214 | } |
| 1215 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1216 | TEST_F(FPDFEditEmbedderTest, GetContentStream) { |
Henrique Nakashima | 6eb7939 | 2018-06-12 20:27:35 +0000 | [diff] [blame] | 1217 | // Load document with some text split across streams. |
| 1218 | EXPECT_TRUE(OpenDocument("split_streams.pdf")); |
| 1219 | FPDF_PAGE page = LoadPage(0); |
| 1220 | ASSERT_TRUE(page); |
| 1221 | |
| 1222 | // Content stream 0: page objects 0-14. |
| 1223 | // Content stream 1: page objects 15-17. |
| 1224 | // Content stream 2: page object 18. |
| 1225 | ASSERT_EQ(19, FPDFPage_CountObjects(page)); |
| 1226 | for (int i = 0; i < 19; i++) { |
| 1227 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 1228 | ASSERT_TRUE(page_object); |
| 1229 | CPDF_PageObject* cpdf_page_object = |
| 1230 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1231 | if (i < 15) |
| 1232 | EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i; |
| 1233 | else if (i < 18) |
| 1234 | EXPECT_EQ(1, cpdf_page_object->GetContentStream()) << i; |
| 1235 | else |
| 1236 | EXPECT_EQ(2, cpdf_page_object->GetContentStream()) << i; |
| 1237 | } |
| 1238 | |
| 1239 | UnloadPage(page); |
| 1240 | } |
| 1241 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 1242 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 1243 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 1244 | #define MAYBE_RemoveAllFromStream DISABLED_RemoveAllFromStream |
| 1245 | #else |
| 1246 | #define MAYBE_RemoveAllFromStream RemoveAllFromStream |
| 1247 | #endif |
| 1248 | TEST_F(FPDFEditEmbedderTest, MAYBE_RemoveAllFromStream) { |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1249 | // Load document with some text split across streams. |
| 1250 | EXPECT_TRUE(OpenDocument("split_streams.pdf")); |
| 1251 | FPDF_PAGE page = LoadPage(0); |
| 1252 | ASSERT_TRUE(page); |
| 1253 | |
| 1254 | // Content stream 0: page objects 0-14. |
| 1255 | // Content stream 1: page objects 15-17. |
| 1256 | // Content stream 2: page object 18. |
| 1257 | ASSERT_EQ(19, FPDFPage_CountObjects(page)); |
| 1258 | |
| 1259 | // Loop backwards because objects will being removed, which shifts the indexes |
| 1260 | // after the removed position. |
| 1261 | for (int i = 18; i >= 0; i--) { |
| 1262 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 1263 | ASSERT_TRUE(page_object); |
| 1264 | CPDF_PageObject* cpdf_page_object = |
| 1265 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1266 | |
| 1267 | // Empty content stream 1. |
| 1268 | if (cpdf_page_object->GetContentStream() == 1) { |
| 1269 | EXPECT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1270 | FPDFPageObj_Destroy(page_object); |
| 1271 | } |
| 1272 | } |
| 1273 | |
| 1274 | // Content stream 0: page objects 0-14. |
| 1275 | // Content stream 2: page object 15. |
| 1276 | ASSERT_EQ(16, FPDFPage_CountObjects(page)); |
| 1277 | for (int i = 0; i < 16; i++) { |
| 1278 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 1279 | ASSERT_TRUE(page_object); |
| 1280 | CPDF_PageObject* cpdf_page_object = |
| 1281 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1282 | if (i < 15) |
| 1283 | EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i; |
| 1284 | else |
| 1285 | EXPECT_EQ(2, cpdf_page_object->GetContentStream()) << i; |
| 1286 | } |
| 1287 | |
| 1288 | // Generate contents should remove the empty stream and update the page |
| 1289 | // objects' contents stream indexes. |
| 1290 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1291 | |
| 1292 | // Content stream 0: page objects 0-14. |
| 1293 | // Content stream 1: page object 15. |
| 1294 | ASSERT_EQ(16, FPDFPage_CountObjects(page)); |
| 1295 | for (int i = 0; i < 16; i++) { |
| 1296 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 1297 | ASSERT_TRUE(page_object); |
| 1298 | CPDF_PageObject* cpdf_page_object = |
| 1299 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1300 | if (i < 15) |
| 1301 | EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i; |
| 1302 | else |
| 1303 | EXPECT_EQ(1, cpdf_page_object->GetContentStream()) << i; |
| 1304 | } |
| 1305 | |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 1306 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 1307 | const char kStream1RemovedMD5[] = "0e8856ca9abc7049412e64f9230c7c43"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 1308 | #elif defined(OS_WIN) |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1309 | const char kStream1RemovedMD5[] = "b4140f203523e38793283a5943d8075b"; |
| 1310 | #else |
| 1311 | const char kStream1RemovedMD5[] = "e86a3efc160ede6cfcb1f59bcacf1105"; |
| 1312 | #endif |
| 1313 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1314 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1315 | CompareBitmap(page_bitmap.get(), 200, 200, kStream1RemovedMD5); |
| 1316 | } |
| 1317 | |
| 1318 | // Save the file |
| 1319 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1320 | UnloadPage(page); |
| 1321 | |
| 1322 | // Re-open the file and check the page object count is still 16, and that |
| 1323 | // content stream 1 was removed. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1324 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1325 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1326 | |
| 1327 | // Content stream 0: page objects 0-14. |
| 1328 | // Content stream 1: page object 15. |
| 1329 | EXPECT_EQ(16, FPDFPage_CountObjects(saved_page)); |
| 1330 | for (int i = 0; i < 16; i++) { |
| 1331 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(saved_page, i); |
| 1332 | ASSERT_TRUE(page_object); |
| 1333 | CPDF_PageObject* cpdf_page_object = |
| 1334 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1335 | if (i < 15) |
| 1336 | EXPECT_EQ(0, cpdf_page_object->GetContentStream()) << i; |
| 1337 | else |
| 1338 | EXPECT_EQ(1, cpdf_page_object->GetContentStream()) << i; |
| 1339 | } |
| 1340 | |
| 1341 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1342 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1343 | CompareBitmap(page_bitmap.get(), 200, 200, kStream1RemovedMD5); |
| 1344 | } |
| 1345 | |
| 1346 | CloseSavedPage(saved_page); |
| 1347 | CloseSavedDocument(); |
| 1348 | } |
| 1349 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1350 | TEST_F(FPDFEditEmbedderTest, RemoveAllFromSingleStream) { |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1351 | // Load document with a single stream. |
| 1352 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 1353 | FPDF_PAGE page = LoadPage(0); |
| 1354 | ASSERT_TRUE(page); |
| 1355 | |
| 1356 | // Content stream 0: page objects 0-1. |
| 1357 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 1358 | |
| 1359 | // Loop backwards because objects will being removed, which shifts the indexes |
| 1360 | // after the removed position. |
| 1361 | for (int i = 1; i >= 0; i--) { |
| 1362 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 1363 | ASSERT_TRUE(page_object); |
| 1364 | CPDF_PageObject* cpdf_page_object = |
| 1365 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1366 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1367 | ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1368 | FPDFPageObj_Destroy(page_object); |
| 1369 | } |
| 1370 | |
| 1371 | // No more objects in the stream |
| 1372 | ASSERT_EQ(0, FPDFPage_CountObjects(page)); |
| 1373 | |
| 1374 | // Generate contents should remove the empty stream and update the page |
| 1375 | // objects' contents stream indexes. |
| 1376 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1377 | |
| 1378 | ASSERT_EQ(0, FPDFPage_CountObjects(page)); |
| 1379 | |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1380 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1381 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 1382 | CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedChecksum); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1383 | } |
| 1384 | |
| 1385 | // Save the file |
| 1386 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1387 | UnloadPage(page); |
| 1388 | |
| 1389 | // Re-open the file and check the page object count is still 0. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1390 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1391 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1392 | |
| 1393 | EXPECT_EQ(0, FPDFPage_CountObjects(saved_page)); |
| 1394 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1395 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 1396 | CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedChecksum); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1397 | } |
| 1398 | |
| 1399 | CloseSavedPage(saved_page); |
| 1400 | CloseSavedDocument(); |
| 1401 | } |
| 1402 | |
Hui Yingst | 08c4071 | 2020-04-29 01:37:35 +0000 | [diff] [blame] | 1403 | TEST_F(FPDFEditEmbedderTest, RemoveFirstFromSingleStream) { |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1404 | // Load document with a single stream. |
| 1405 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 1406 | FPDF_PAGE page = LoadPage(0); |
| 1407 | ASSERT_TRUE(page); |
| 1408 | |
| 1409 | // Content stream 0: page objects 0-1. |
| 1410 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 1411 | |
| 1412 | // Remove first object. |
| 1413 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 1414 | ASSERT_TRUE(page_object); |
| 1415 | CPDF_PageObject* cpdf_page_object = |
| 1416 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1417 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1418 | ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1419 | FPDFPageObj_Destroy(page_object); |
| 1420 | |
| 1421 | // One object left in the stream. |
| 1422 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 1423 | page_object = FPDFPage_GetObject(page, 0); |
| 1424 | ASSERT_TRUE(page_object); |
| 1425 | cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object); |
| 1426 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1427 | |
| 1428 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1429 | |
| 1430 | // Still one object left in the stream. |
| 1431 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 1432 | page_object = FPDFPage_GetObject(page, 0); |
| 1433 | ASSERT_TRUE(page_object); |
| 1434 | cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object); |
| 1435 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1436 | |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1437 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1438 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1439 | CompareBitmap(page_bitmap.get(), 200, 200, kFirstRemovedChecksum); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1440 | } |
| 1441 | |
| 1442 | // Save the file |
| 1443 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1444 | UnloadPage(page); |
| 1445 | |
| 1446 | // Re-open the file and check the page object count is still 0. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1447 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1448 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1449 | |
| 1450 | ASSERT_EQ(1, FPDFPage_CountObjects(saved_page)); |
| 1451 | page_object = FPDFPage_GetObject(saved_page, 0); |
| 1452 | ASSERT_TRUE(page_object); |
| 1453 | cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object); |
| 1454 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1455 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1456 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1457 | CompareBitmap(page_bitmap.get(), 200, 200, kFirstRemovedChecksum); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1458 | } |
| 1459 | |
| 1460 | CloseSavedPage(saved_page); |
| 1461 | CloseSavedDocument(); |
| 1462 | } |
| 1463 | |
Hui Yingst | 08c4071 | 2020-04-29 01:37:35 +0000 | [diff] [blame] | 1464 | TEST_F(FPDFEditEmbedderTest, RemoveLastFromSingleStream) { |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1465 | // Load document with a single stream. |
| 1466 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 1467 | FPDF_PAGE page = LoadPage(0); |
| 1468 | ASSERT_TRUE(page); |
| 1469 | |
| 1470 | // Content stream 0: page objects 0-1. |
| 1471 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 1472 | |
| 1473 | // Remove last object |
| 1474 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 1); |
| 1475 | ASSERT_TRUE(page_object); |
| 1476 | CPDF_PageObject* cpdf_page_object = |
| 1477 | CPDFPageObjectFromFPDFPageObject(page_object); |
| 1478 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1479 | ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1480 | FPDFPageObj_Destroy(page_object); |
| 1481 | |
| 1482 | // One object left in the stream. |
| 1483 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 1484 | page_object = FPDFPage_GetObject(page, 0); |
| 1485 | ASSERT_TRUE(page_object); |
| 1486 | cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object); |
| 1487 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1488 | |
| 1489 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1490 | |
| 1491 | // Still one object left in the stream. |
| 1492 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 1493 | page_object = FPDFPage_GetObject(page, 0); |
| 1494 | ASSERT_TRUE(page_object); |
| 1495 | cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object); |
| 1496 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1497 | |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1498 | using pdfium::kHelloWorldRemovedChecksum; |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1499 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1500 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1501 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldRemovedChecksum); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1502 | } |
| 1503 | |
| 1504 | // Save the file |
| 1505 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1506 | UnloadPage(page); |
| 1507 | |
| 1508 | // Re-open the file and check the page object count is still 0. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1509 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1510 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1511 | |
| 1512 | ASSERT_EQ(1, FPDFPage_CountObjects(saved_page)); |
| 1513 | page_object = FPDFPage_GetObject(saved_page, 0); |
| 1514 | ASSERT_TRUE(page_object); |
| 1515 | cpdf_page_object = CPDFPageObjectFromFPDFPageObject(page_object); |
| 1516 | ASSERT_EQ(0, cpdf_page_object->GetContentStream()); |
| 1517 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1518 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1519 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldRemovedChecksum); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1520 | } |
| 1521 | |
| 1522 | CloseSavedPage(saved_page); |
| 1523 | CloseSavedDocument(); |
| 1524 | } |
| 1525 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1526 | TEST_F(FPDFEditEmbedderTest, RemoveAllFromMultipleStreams) { |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1527 | // Load document with some text. |
| 1528 | EXPECT_TRUE(OpenDocument("hello_world_split_streams.pdf")); |
| 1529 | FPDF_PAGE page = LoadPage(0); |
| 1530 | ASSERT_TRUE(page); |
| 1531 | |
| 1532 | // Content stream 0: page objects 0-1. |
| 1533 | // Content stream 1: page object 2. |
| 1534 | ASSERT_EQ(3, FPDFPage_CountObjects(page)); |
| 1535 | |
| 1536 | // Loop backwards because objects will being removed, which shifts the indexes |
| 1537 | // after the removed position. |
| 1538 | for (int i = 2; i >= 0; i--) { |
| 1539 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, i); |
| 1540 | ASSERT_TRUE(page_object); |
| 1541 | ASSERT_TRUE(FPDFPage_RemoveObject(page, page_object)); |
| 1542 | FPDFPageObj_Destroy(page_object); |
| 1543 | } |
| 1544 | |
| 1545 | // No more objects in the page. |
| 1546 | ASSERT_EQ(0, FPDFPage_CountObjects(page)); |
| 1547 | |
| 1548 | // Generate contents should remove the empty streams and update the page |
| 1549 | // objects' contents stream indexes. |
| 1550 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1551 | |
| 1552 | ASSERT_EQ(0, FPDFPage_CountObjects(page)); |
| 1553 | |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1554 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1555 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 1556 | CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedChecksum); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1557 | } |
| 1558 | |
| 1559 | // Save the file |
| 1560 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1561 | UnloadPage(page); |
| 1562 | |
| 1563 | // Re-open the file and check the page object count is still 0. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1564 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1565 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1566 | |
| 1567 | EXPECT_EQ(0, FPDFPage_CountObjects(saved_page)); |
| 1568 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1569 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 1570 | CompareBitmap(page_bitmap.get(), 200, 200, kAllRemovedChecksum); |
Henrique Nakashima | 0dcf1f4 | 2018-06-21 18:51:15 +0000 | [diff] [blame] | 1571 | } |
| 1572 | |
| 1573 | CloseSavedPage(saved_page); |
| 1574 | CloseSavedDocument(); |
| 1575 | } |
| 1576 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1577 | TEST_F(FPDFEditEmbedderTest, InsertPageObjectAndSave) { |
Henrique Nakashima | c49e62e | 2018-04-16 20:58:47 +0000 | [diff] [blame] | 1578 | // Load document with some text. |
| 1579 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 1580 | FPDF_PAGE page = LoadPage(0); |
| 1581 | ASSERT_TRUE(page); |
| 1582 | |
| 1583 | // Add a red rectangle. |
| 1584 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 1585 | FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1586 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255)); |
Henrique Nakashima | c49e62e | 2018-04-16 20:58:47 +0000 | [diff] [blame] | 1587 | EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1588 | FPDFPage_InsertObject(page, red_rect); |
| 1589 | |
| 1590 | // Verify the red rectangle was added. |
| 1591 | ASSERT_EQ(3, FPDFPage_CountObjects(page)); |
| 1592 | |
| 1593 | // Save the file |
| 1594 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1595 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1596 | UnloadPage(page); |
| 1597 | |
| 1598 | // Re-open the file and check the page object count is still 3. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1599 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | c49e62e | 2018-04-16 20:58:47 +0000 | [diff] [blame] | 1600 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1601 | EXPECT_EQ(3, FPDFPage_CountObjects(saved_page)); |
| 1602 | CloseSavedPage(saved_page); |
| 1603 | CloseSavedDocument(); |
| 1604 | } |
| 1605 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1606 | TEST_F(FPDFEditEmbedderTest, InsertPageObjectEditAndSave) { |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1607 | // Load document with some text. |
| 1608 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 1609 | FPDF_PAGE page = LoadPage(0); |
| 1610 | ASSERT_TRUE(page); |
| 1611 | |
| 1612 | // Add a red rectangle. |
| 1613 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 1614 | FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1615 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 100, 100, 255)); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1616 | EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1617 | FPDFPage_InsertObject(page, red_rect); |
| 1618 | |
| 1619 | // Verify the red rectangle was added. |
| 1620 | ASSERT_EQ(3, FPDFPage_CountObjects(page)); |
| 1621 | |
| 1622 | // Generate content but change it again |
| 1623 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1624 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255)); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1625 | |
| 1626 | // Save the file |
| 1627 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1628 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1629 | UnloadPage(page); |
| 1630 | |
| 1631 | // Re-open the file and check the page object count is still 3. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1632 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 27cf78d | 2018-06-14 16:22:30 +0000 | [diff] [blame] | 1633 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1634 | EXPECT_EQ(3, FPDFPage_CountObjects(saved_page)); |
| 1635 | CloseSavedPage(saved_page); |
| 1636 | CloseSavedDocument(); |
| 1637 | } |
| 1638 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 1639 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 1640 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 1641 | #define MAYBE_InsertAndRemoveLargeFile DISABLED_InsertAndRemoveLargeFile |
| 1642 | #else |
| 1643 | #define MAYBE_InsertAndRemoveLargeFile InsertAndRemoveLargeFile |
| 1644 | #endif |
| 1645 | TEST_F(FPDFEditEmbedderTest, MAYBE_InsertAndRemoveLargeFile) { |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1646 | const int kOriginalObjectCount = 600; |
| 1647 | |
| 1648 | // Load document with many objects. |
| 1649 | EXPECT_TRUE(OpenDocument("many_rectangles.pdf")); |
| 1650 | FPDF_PAGE page = LoadPage(0); |
| 1651 | ASSERT_TRUE(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1652 | |
| 1653 | using pdfium::kManyRectanglesChecksum; |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1654 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1655 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1656 | CompareBitmap(page_bitmap.get(), 200, 300, kManyRectanglesChecksum); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1657 | } |
| 1658 | |
| 1659 | // Add a black rectangle. |
| 1660 | ASSERT_EQ(kOriginalObjectCount, FPDFPage_CountObjects(page)); |
| 1661 | FPDF_PAGEOBJECT black_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1662 | EXPECT_TRUE(FPDFPageObj_SetFillColor(black_rect, 0, 0, 0, 255)); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1663 | EXPECT_TRUE(FPDFPath_SetDrawMode(black_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1664 | FPDFPage_InsertObject(page, black_rect); |
| 1665 | |
| 1666 | // Verify the black rectangle was added. |
| 1667 | ASSERT_EQ(kOriginalObjectCount + 1, FPDFPage_CountObjects(page)); |
| 1668 | const char kPlusRectangleMD5[] = "6b9396ab570754b32b04ca629e902f77"; |
| 1669 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1670 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1671 | CompareBitmap(page_bitmap.get(), 200, 300, kPlusRectangleMD5); |
| 1672 | } |
| 1673 | |
| 1674 | // Save the file. |
| 1675 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1676 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 1677 | UnloadPage(page); |
| 1678 | |
| 1679 | // Re-open the file and check the rectangle added is still there. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1680 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1681 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1682 | EXPECT_EQ(kOriginalObjectCount + 1, FPDFPage_CountObjects(saved_page)); |
| 1683 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1684 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1685 | CompareBitmap(page_bitmap.get(), 200, 300, kPlusRectangleMD5); |
| 1686 | } |
| 1687 | |
| 1688 | // Remove the added rectangle. |
| 1689 | FPDF_PAGEOBJECT added_object = |
| 1690 | FPDFPage_GetObject(saved_page, kOriginalObjectCount); |
| 1691 | EXPECT_TRUE(FPDFPage_RemoveObject(saved_page, added_object)); |
| 1692 | FPDFPageObj_Destroy(added_object); |
| 1693 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1694 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1695 | CompareBitmap(page_bitmap.get(), 200, 300, kManyRectanglesChecksum); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1696 | } |
| 1697 | EXPECT_EQ(kOriginalObjectCount, FPDFPage_CountObjects(saved_page)); |
| 1698 | |
| 1699 | // Save the file again. |
| 1700 | EXPECT_TRUE(FPDFPage_GenerateContent(saved_page)); |
| 1701 | EXPECT_TRUE(FPDF_SaveAsCopy(saved_document_, this, 0)); |
| 1702 | |
| 1703 | CloseSavedPage(saved_page); |
| 1704 | CloseSavedDocument(); |
| 1705 | |
| 1706 | // Re-open the file (again) and check the black rectangle was removed and the |
| 1707 | // rest is intact. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1708 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1709 | saved_page = LoadSavedPage(0); |
| 1710 | EXPECT_EQ(kOriginalObjectCount, FPDFPage_CountObjects(saved_page)); |
| 1711 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1712 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1713 | CompareBitmap(page_bitmap.get(), 200, 300, kManyRectanglesChecksum); |
Henrique Nakashima | 657a1aa | 2018-09-12 16:19:22 +0000 | [diff] [blame] | 1714 | } |
| 1715 | |
| 1716 | CloseSavedPage(saved_page); |
| 1717 | CloseSavedDocument(); |
| 1718 | } |
| 1719 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1720 | TEST_F(FPDFEditEmbedderTest, AddAndRemovePaths) { |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 1721 | // Start with a blank page. |
| 1722 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
| 1723 | ASSERT_TRUE(page); |
| 1724 | |
| 1725 | // Render the blank page and verify it's a blank bitmap. |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1726 | using pdfium::kBlankPage612By792Checksum; |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 1727 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1728 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1729 | CompareBitmap(page_bitmap.get(), 612, 792, kBlankPage612By792Checksum); |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 1730 | } |
| 1731 | ASSERT_EQ(0, FPDFPage_CountObjects(page)); |
| 1732 | |
| 1733 | // Add a red rectangle. |
| 1734 | FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20); |
| 1735 | ASSERT_TRUE(red_rect); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1736 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255)); |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 1737 | EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1738 | FPDFPage_InsertObject(page, red_rect); |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 1739 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1740 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1741 | CompareBitmap(page_bitmap.get(), 612, 792, kRedRectangleChecksum); |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 1742 | } |
| 1743 | EXPECT_EQ(1, FPDFPage_CountObjects(page)); |
| 1744 | |
| 1745 | // Remove rectangle and verify it does not render anymore and the bitmap is |
| 1746 | // back to a blank one. |
| 1747 | EXPECT_TRUE(FPDFPage_RemoveObject(page, red_rect)); |
| 1748 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1749 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 1750 | CompareBitmap(page_bitmap.get(), 612, 792, kBlankPage612By792Checksum); |
Henrique Nakashima | 35841fa | 2018-03-15 15:25:16 +0000 | [diff] [blame] | 1751 | } |
| 1752 | EXPECT_EQ(0, FPDFPage_CountObjects(page)); |
| 1753 | |
| 1754 | // Trying to remove an object not in the page should return false. |
| 1755 | EXPECT_FALSE(FPDFPage_RemoveObject(page, red_rect)); |
| 1756 | |
| 1757 | FPDF_ClosePage(page); |
| 1758 | FPDFPageObj_Destroy(red_rect); |
| 1759 | } |
| 1760 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 1761 | TEST_F(FPDFEditEmbedderTest, PathsPoints) { |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 1762 | CreateNewDocument(); |
| 1763 | FPDF_PAGEOBJECT img = FPDFPageObj_NewImageObj(document_); |
| 1764 | // This should fail gracefully, even if img is not a path. |
Miklos Vajna | 0150a54 | 2017-09-21 21:46:56 +0200 | [diff] [blame] | 1765 | ASSERT_EQ(-1, FPDFPath_CountSegments(img)); |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 1766 | |
| 1767 | // This should fail gracefully, even if path is NULL. |
Miklos Vajna | 0150a54 | 2017-09-21 21:46:56 +0200 | [diff] [blame] | 1768 | ASSERT_EQ(-1, FPDFPath_CountSegments(nullptr)); |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 1769 | |
Miklos Vajna | 36eed87 | 2017-09-20 22:52:43 +0200 | [diff] [blame] | 1770 | // FPDFPath_GetPathSegment() with a non-path. |
| 1771 | ASSERT_EQ(nullptr, FPDFPath_GetPathSegment(img, 0)); |
| 1772 | // FPDFPath_GetPathSegment() with a NULL path. |
| 1773 | ASSERT_EQ(nullptr, FPDFPath_GetPathSegment(nullptr, 0)); |
| 1774 | float x; |
| 1775 | float y; |
| 1776 | // FPDFPathSegment_GetPoint() with a NULL segment. |
| 1777 | EXPECT_FALSE(FPDFPathSegment_GetPoint(nullptr, &x, &y)); |
| 1778 | |
| 1779 | // FPDFPathSegment_GetType() with a NULL segment. |
| 1780 | ASSERT_EQ(FPDF_SEGMENT_UNKNOWN, FPDFPathSegment_GetType(nullptr)); |
| 1781 | |
| 1782 | // FPDFPathSegment_GetClose() with a NULL segment. |
| 1783 | EXPECT_FALSE(FPDFPathSegment_GetClose(nullptr)); |
| 1784 | |
Miklos Vajna | 12abfd0 | 2017-09-15 07:49:03 +0200 | [diff] [blame] | 1785 | FPDFPageObj_Destroy(img); |
| 1786 | } |
| 1787 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 1788 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 1789 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 1790 | #define MAYBE_PathOnTopOfText DISABLED_PathOnTopOfText |
| 1791 | #else |
| 1792 | #define MAYBE_PathOnTopOfText PathOnTopOfText |
| 1793 | #endif |
| 1794 | TEST_F(FPDFEditEmbedderTest, MAYBE_PathOnTopOfText) { |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1795 | // Load document with some text |
| 1796 | EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
| 1797 | FPDF_PAGE page = LoadPage(0); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1798 | ASSERT_TRUE(page); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1799 | |
| 1800 | // Add an opaque rectangle on top of some of the text. |
| 1801 | FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(20, 100, 50, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1802 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255)); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1803 | EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1804 | FPDFPage_InsertObject(page, red_rect); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1805 | |
| 1806 | // Add a transparent triangle on top of other part of the text. |
| 1807 | FPDF_PAGEOBJECT black_path = FPDFPageObj_CreateNewPath(20, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1808 | EXPECT_TRUE(FPDFPageObj_SetFillColor(black_path, 0, 0, 0, 100)); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1809 | EXPECT_TRUE(FPDFPath_SetDrawMode(black_path, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1810 | EXPECT_TRUE(FPDFPath_LineTo(black_path, 30, 80)); |
| 1811 | EXPECT_TRUE(FPDFPath_LineTo(black_path, 40, 10)); |
| 1812 | EXPECT_TRUE(FPDFPath_Close(black_path)); |
| 1813 | FPDFPage_InsertObject(page, black_path); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1814 | |
| 1815 | // Render and check the result. Text is slightly different on Mac. |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 1816 | ScopedFPDFBitmap bitmap = RenderLoadedPage(page); |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 1817 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 1818 | const char md5[] = "e55bcd1facb7243dc6e16dd5f912265b"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 1819 | #elif defined(OS_WIN) |
Lei Zhang | 5e2c51c | 2018-07-27 22:33:34 +0000 | [diff] [blame] | 1820 | const char md5[] = "74dd9c393b8b2578d2b7feb032b7daad"; |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1821 | #else |
Henrique Nakashima | 09b4192 | 2017-10-27 20:39:29 +0000 | [diff] [blame] | 1822 | const char md5[] = "aa71b09b93b55f467f1290e5111babee"; |
Nicolas Pena | 5bcd9a3 | 2017-03-22 11:04:35 -0400 | [diff] [blame] | 1823 | #endif |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1824 | CompareBitmap(bitmap.get(), 200, 200, md5); |
Nicolas Pena | 0fc185e | 2017-02-08 12:13:20 -0500 | [diff] [blame] | 1825 | UnloadPage(page); |
| 1826 | } |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1827 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 1828 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 1829 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 1830 | #define MAYBE_EditOverExistingContent DISABLED_EditOverExistingContent |
| 1831 | #else |
| 1832 | #define MAYBE_EditOverExistingContent EditOverExistingContent |
| 1833 | #endif |
| 1834 | TEST_F(FPDFEditEmbedderTest, MAYBE_EditOverExistingContent) { |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1835 | // Load document with existing content |
| 1836 | EXPECT_TRUE(OpenDocument("bug_717.pdf")); |
| 1837 | FPDF_PAGE page = LoadPage(0); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1838 | ASSERT_TRUE(page); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1839 | |
| 1840 | // Add a transparent rectangle on top of the existing content |
| 1841 | FPDF_PAGEOBJECT red_rect2 = FPDFPageObj_CreateNewRect(90, 700, 25, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1842 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect2, 255, 0, 0, 100)); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1843 | EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect2, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1844 | FPDFPage_InsertObject(page, red_rect2); |
| 1845 | |
| 1846 | // Add an opaque rectangle on top of the existing content |
| 1847 | FPDF_PAGEOBJECT red_rect = FPDFPageObj_CreateNewRect(115, 700, 25, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1848 | EXPECT_TRUE(FPDFPageObj_SetFillColor(red_rect, 255, 0, 0, 255)); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1849 | EXPECT_TRUE(FPDFPath_SetDrawMode(red_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
| 1850 | FPDFPage_InsertObject(page, red_rect); |
| 1851 | |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 1852 | ScopedFPDFBitmap bitmap = RenderLoadedPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1853 | CompareBitmap(bitmap.get(), 612, 792, "ad04e5bd0f471a9a564fb034bd0fb073"); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1854 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 1855 | |
| 1856 | // Now save the result, closing the page and document |
| 1857 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
Nicolas Pena | 3ff5400 | 2017-07-05 11:55:35 -0400 | [diff] [blame] | 1858 | UnloadPage(page); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1859 | |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 1860 | ASSERT_TRUE(OpenSavedDocument()); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1861 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 1862 | VerifySavedRendering(saved_page, 612, 792, |
| 1863 | "ad04e5bd0f471a9a564fb034bd0fb073"); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1864 | |
| 1865 | ClearString(); |
| 1866 | // Add another opaque rectangle on top of the existing content |
| 1867 | FPDF_PAGEOBJECT green_rect = FPDFPageObj_CreateNewRect(150, 700, 25, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1868 | EXPECT_TRUE(FPDFPageObj_SetFillColor(green_rect, 0, 255, 0, 255)); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1869 | EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect, FPDF_FILLMODE_ALTERNATE, 0)); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1870 | FPDFPage_InsertObject(saved_page, green_rect); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1871 | |
| 1872 | // Add another transparent rectangle on top of existing content |
| 1873 | FPDF_PAGEOBJECT green_rect2 = FPDFPageObj_CreateNewRect(175, 700, 25, 50); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1874 | EXPECT_TRUE(FPDFPageObj_SetFillColor(green_rect2, 0, 255, 0, 100)); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1875 | EXPECT_TRUE(FPDFPath_SetDrawMode(green_rect2, FPDF_FILLMODE_ALTERNATE, 0)); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1876 | FPDFPage_InsertObject(saved_page, green_rect2); |
Lei Zhang | c113c7a | 2018-02-12 14:58:44 +0000 | [diff] [blame] | 1877 | const char kLastMD5[] = "4b5b00f824620f8c9b8801ebb98e1cdd"; |
| 1878 | { |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 1879 | ScopedFPDFBitmap new_bitmap = RenderSavedPage(saved_page); |
Lei Zhang | c113c7a | 2018-02-12 14:58:44 +0000 | [diff] [blame] | 1880 | CompareBitmap(new_bitmap.get(), 612, 792, kLastMD5); |
| 1881 | } |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1882 | EXPECT_TRUE(FPDFPage_GenerateContent(saved_page)); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1883 | |
| 1884 | // Now save the result, closing the page and document |
Lei Zhang | 0729be2 | 2018-02-05 21:13:51 +0000 | [diff] [blame] | 1885 | EXPECT_TRUE(FPDF_SaveAsCopy(saved_document_, this, 0)); |
Dan Sinclair | 04e4dc8 | 2017-10-18 12:17:14 -0400 | [diff] [blame] | 1886 | |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1887 | CloseSavedPage(saved_page); |
Dan Sinclair | 04e4dc8 | 2017-10-18 12:17:14 -0400 | [diff] [blame] | 1888 | CloseSavedDocument(); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1889 | |
| 1890 | // Render the saved result |
Lei Zhang | c113c7a | 2018-02-12 14:58:44 +0000 | [diff] [blame] | 1891 | VerifySavedDocument(612, 792, kLastMD5); |
wileyrya | e858aa4 | 2017-05-31 14:49:05 -0500 | [diff] [blame] | 1892 | } |
| 1893 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 1894 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 1895 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 1896 | #define MAYBE_AddStrokedPaths DISABLED_AddStrokedPaths |
| 1897 | #else |
| 1898 | #define MAYBE_AddStrokedPaths AddStrokedPaths |
| 1899 | #endif |
| 1900 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddStrokedPaths) { |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1901 | // Start with a blank page |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 1902 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1903 | |
| 1904 | // Add a large stroked rectangle (fill color should not affect it). |
| 1905 | FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(20, 20, 200, 400); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1906 | EXPECT_TRUE(FPDFPageObj_SetFillColor(rect, 255, 0, 0, 255)); |
| 1907 | EXPECT_TRUE(FPDFPageObj_SetStrokeColor(rect, 0, 255, 0, 255)); |
| 1908 | EXPECT_TRUE(FPDFPageObj_SetStrokeWidth(rect, 15.0f)); |
Miklos Vajna | 366df7f | 2018-05-22 14:27:29 +0000 | [diff] [blame] | 1909 | |
| 1910 | float width = 0; |
| 1911 | EXPECT_TRUE(FPDFPageObj_GetStrokeWidth(rect, &width)); |
| 1912 | EXPECT_EQ(15.0f, width); |
| 1913 | |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1914 | EXPECT_TRUE(FPDFPath_SetDrawMode(rect, 0, 1)); |
| 1915 | FPDFPage_InsertObject(page, rect); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1916 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1917 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1918 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 1919 | "64bd31f862a89e0a9e505a5af6efd506"); |
| 1920 | } |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1921 | |
| 1922 | // Add crossed-checkmark |
| 1923 | FPDF_PAGEOBJECT check = FPDFPageObj_CreateNewPath(300, 500); |
| 1924 | EXPECT_TRUE(FPDFPath_LineTo(check, 400, 400)); |
| 1925 | EXPECT_TRUE(FPDFPath_LineTo(check, 600, 600)); |
| 1926 | EXPECT_TRUE(FPDFPath_MoveTo(check, 400, 600)); |
| 1927 | EXPECT_TRUE(FPDFPath_LineTo(check, 600, 400)); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1928 | EXPECT_TRUE(FPDFPageObj_SetStrokeColor(check, 128, 128, 128, 180)); |
| 1929 | EXPECT_TRUE(FPDFPageObj_SetStrokeWidth(check, 8.35f)); |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1930 | EXPECT_TRUE(FPDFPath_SetDrawMode(check, 0, 1)); |
| 1931 | FPDFPage_InsertObject(page, check); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1932 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1933 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1934 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 1935 | "4b6f3b9d25c4e194821217d5016c3724"); |
| 1936 | } |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1937 | |
| 1938 | // Add stroked and filled oval-ish path. |
| 1939 | FPDF_PAGEOBJECT path = FPDFPageObj_CreateNewPath(250, 100); |
| 1940 | EXPECT_TRUE(FPDFPath_BezierTo(path, 180, 166, 180, 233, 250, 300)); |
| 1941 | EXPECT_TRUE(FPDFPath_LineTo(path, 255, 305)); |
| 1942 | EXPECT_TRUE(FPDFPath_BezierTo(path, 325, 233, 325, 166, 255, 105)); |
| 1943 | EXPECT_TRUE(FPDFPath_Close(path)); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 1944 | EXPECT_TRUE(FPDFPageObj_SetFillColor(path, 200, 128, 128, 100)); |
| 1945 | EXPECT_TRUE(FPDFPageObj_SetStrokeColor(path, 128, 200, 128, 150)); |
| 1946 | EXPECT_TRUE(FPDFPageObj_SetStrokeWidth(path, 10.5f)); |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1947 | EXPECT_TRUE(FPDFPath_SetDrawMode(path, FPDF_FILLMODE_ALTERNATE, 1)); |
| 1948 | FPDFPage_InsertObject(page, path); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1949 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1950 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1951 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 1952 | "ff3e6a22326754944cc6e56609acd73b"); |
| 1953 | } |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1954 | FPDF_ClosePage(page); |
Nicolas Pena | 2eb1a70 | 2017-02-09 18:17:33 -0500 | [diff] [blame] | 1955 | } |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 1956 | |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 1957 | // Tests adding text from standard font using FPDFPageObj_NewTextObj. |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 1958 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 1959 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 1960 | #define MAYBE_AddStandardFontText DISABLED_AddStandardFontText |
| 1961 | #else |
| 1962 | #define MAYBE_AddStandardFontText AddStandardFontText |
| 1963 | #endif |
| 1964 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddStandardFontText) { |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 1965 | // Start with a blank page |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 1966 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 1967 | |
| 1968 | // Add some text to the page |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 1969 | FPDF_PAGEOBJECT text_object1 = |
| 1970 | FPDFPageObj_NewTextObj(document(), "Arial", 12.0f); |
| 1971 | EXPECT_TRUE(text_object1); |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 1972 | ScopedFPDFWideString text1 = GetFPDFWideString(kBottomText); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 1973 | EXPECT_TRUE(FPDFText_SetText(text_object1, text1.get())); |
| 1974 | FPDFPageObj_Transform(text_object1, 1, 0, 0, 1, 20, 20); |
| 1975 | FPDFPage_InsertObject(page, text_object1); |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 1976 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1977 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1978 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 1979 | CompareBitmap(page_bitmap.get(), 612, 792, kBottomTextChecksum); |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 1980 | |
| 1981 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 1982 | VerifySavedDocument(612, 792, kBottomTextChecksum); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1983 | } |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 1984 | |
| 1985 | // Try another font |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 1986 | FPDF_PAGEOBJECT text_object2 = |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 1987 | FPDFPageObj_NewTextObj(document(), "TimesNewRomanBold", 15.0f); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 1988 | EXPECT_TRUE(text_object2); |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 1989 | ScopedFPDFWideString text2 = |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 1990 | GetFPDFWideString(L"Hi, I'm Bold. Times New Roman Bold."); |
| 1991 | EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get())); |
| 1992 | FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 100, 600); |
| 1993 | FPDFPage_InsertObject(page, text_object2); |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 1994 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 1995 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 1996 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 1997 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 1998 | const char md5[] = "26a516d923b0a18fbea0a24e3aca5562"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 1999 | #elif defined(OS_WIN) |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 2000 | const char md5[] = "3755dd35abd4c605755369401ee85b2d"; |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 2001 | #else |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 2002 | const char md5[] = "76fcc7d08aa15445efd2e2ceb7c6cc3b"; |
Nicolas Pena | 5bcd9a3 | 2017-03-22 11:04:35 -0400 | [diff] [blame] | 2003 | #endif |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 2004 | CompareBitmap(page_bitmap.get(), 612, 792, md5); |
| 2005 | |
| 2006 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2007 | VerifySavedDocument(612, 792, md5); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2008 | } |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 2009 | |
| 2010 | // And some randomly transformed text |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2011 | FPDF_PAGEOBJECT text_object3 = |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2012 | FPDFPageObj_NewTextObj(document(), "Courier-Bold", 20.0f); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2013 | EXPECT_TRUE(text_object3); |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 2014 | ScopedFPDFWideString text3 = GetFPDFWideString(L"Can you read me? <:)>"); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2015 | EXPECT_TRUE(FPDFText_SetText(text_object3, text3.get())); |
| 2016 | FPDFPageObj_Transform(text_object3, 1, 1.5, 2, 0.5, 200, 200); |
| 2017 | FPDFPage_InsertObject(page, text_object3); |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 2018 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2019 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2020 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 2021 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 2022 | const char md5[] = "532024c9ded47843313bb64a060118f3"; |
Lei Zhang | e67bcc7 | 2019-04-30 18:55:58 +0000 | [diff] [blame] | 2023 | #elif defined(OS_WIN) |
Lei Zhang | a2b7073 | 2019-06-25 08:34:22 +0000 | [diff] [blame] | 2024 | const char md5[] = "5ded49fe157f89627903553771431e3d"; |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 2025 | #else |
Lei Zhang | a2b7073 | 2019-06-25 08:34:22 +0000 | [diff] [blame] | 2026 | const char md5[] = "344534539aa7c5cc78404cfff4bde7fb"; |
Nicolas Pena | 5bcd9a3 | 2017-03-22 11:04:35 -0400 | [diff] [blame] | 2027 | #endif |
Lei Zhang | e039bab | 2019-03-18 19:57:56 +0000 | [diff] [blame] | 2028 | CompareBitmap(page_bitmap.get(), 612, 792, md5); |
| 2029 | |
| 2030 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2031 | VerifySavedDocument(612, 792, md5); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2032 | } |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 2033 | |
Lei Zhang | 8da9823 | 2019-12-11 23:29:33 +0000 | [diff] [blame] | 2034 | FS_MATRIX matrix; |
| 2035 | EXPECT_FALSE(FPDFTextObj_GetMatrix(nullptr, &matrix)); |
| 2036 | EXPECT_TRUE(FPDFTextObj_GetMatrix(text_object3, &matrix)); |
| 2037 | EXPECT_FLOAT_EQ(1.0f, matrix.a); |
| 2038 | EXPECT_FLOAT_EQ(1.5f, matrix.b); |
| 2039 | EXPECT_FLOAT_EQ(2.0f, matrix.c); |
| 2040 | EXPECT_FLOAT_EQ(0.5f, matrix.d); |
| 2041 | EXPECT_FLOAT_EQ(200.0f, matrix.e); |
| 2042 | EXPECT_FLOAT_EQ(200.0f, matrix.f); |
Miklos Vajna | c765d2a | 2018-06-19 15:45:42 +0000 | [diff] [blame] | 2043 | |
Miklos Vajna | 8625d3b | 2018-06-26 15:12:48 +0000 | [diff] [blame] | 2044 | EXPECT_EQ(0, FPDFTextObj_GetFontSize(nullptr)); |
| 2045 | EXPECT_EQ(20, FPDFTextObj_GetFontSize(text_object3)); |
| 2046 | |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 2047 | // TODO(npm): Why are there issues with text rotated by 90 degrees? |
| 2048 | // TODO(npm): FPDF_SaveAsCopy not giving the desired result after this. |
| 2049 | FPDF_ClosePage(page); |
Nicolas Pena | 4905840 | 2017-02-14 18:26:20 -0500 | [diff] [blame] | 2050 | } |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2051 | |
Daniel Hosseinian | 8cb6a65 | 2019-12-18 00:50:41 +0000 | [diff] [blame] | 2052 | TEST_F(FPDFEditEmbedderTest, GetTextRenderMode) { |
Miklos Vajna | 1448cc1 | 2018-07-03 13:52:33 +0000 | [diff] [blame] | 2053 | EXPECT_TRUE(OpenDocument("text_render_mode.pdf")); |
| 2054 | FPDF_PAGE page = LoadPage(0); |
| 2055 | ASSERT_TRUE(page); |
| 2056 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 2057 | |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2058 | EXPECT_EQ(FPDF_TEXTRENDERMODE_UNKNOWN, |
Daniel Hosseinian | 8cb6a65 | 2019-12-18 00:50:41 +0000 | [diff] [blame] | 2059 | FPDFTextObj_GetTextRenderMode(nullptr)); |
Miklos Vajna | 1448cc1 | 2018-07-03 13:52:33 +0000 | [diff] [blame] | 2060 | |
| 2061 | FPDF_PAGEOBJECT fill = FPDFPage_GetObject(page, 0); |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2062 | EXPECT_EQ(FPDF_TEXTRENDERMODE_FILL, FPDFTextObj_GetTextRenderMode(fill)); |
Miklos Vajna | 1448cc1 | 2018-07-03 13:52:33 +0000 | [diff] [blame] | 2063 | |
| 2064 | FPDF_PAGEOBJECT stroke = FPDFPage_GetObject(page, 1); |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2065 | EXPECT_EQ(FPDF_TEXTRENDERMODE_STROKE, FPDFTextObj_GetTextRenderMode(stroke)); |
| 2066 | |
| 2067 | UnloadPage(page); |
| 2068 | } |
| 2069 | |
Hui Yingst | 609f7d6 | 2020-04-23 23:14:13 +0000 | [diff] [blame] | 2070 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 2071 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 2072 | #define MAYBE_SetTextRenderMode DISABLED_SetTextRenderMode |
| 2073 | #else |
| 2074 | #define MAYBE_SetTextRenderMode SetTextRenderMode |
| 2075 | #endif |
| 2076 | TEST_F(FPDFEditEmbedderTest, MAYBE_SetTextRenderMode) { |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2077 | // Platform specific original bitmaps |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2078 | #if defined(OS_MACOSX) |
Lei Zhang | 0f6b434 | 2020-02-25 20:00:39 +0000 | [diff] [blame] | 2079 | const char md5[] = "2453a9a524ee3f5f525dd21345ec4d81"; |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2080 | #elif defined(OS_WIN) |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2081 | const char md5[] = "de6e86bad3e9fda753a8471a45cfbb58"; |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2082 | #else |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2083 | const char md5[] = "5a012d2920ac075c39ffa9437ea42faa"; |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2084 | #endif |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2085 | const char md5_stroke[] = "412e52e621b46bd77baf2162e1fb1a1d"; |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2086 | |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2087 | { |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2088 | EXPECT_TRUE(OpenDocument("text_render_mode.pdf")); |
| 2089 | FPDF_PAGE page = LoadPage(0); |
| 2090 | ASSERT_TRUE(page); |
| 2091 | ASSERT_EQ(2, FPDFPage_CountObjects(page)); |
| 2092 | |
| 2093 | // Check the bitmap |
| 2094 | { |
| 2095 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
| 2096 | CompareBitmap(page_bitmap.get(), 612, 446, md5); |
| 2097 | } |
| 2098 | |
| 2099 | // Cannot set on a null object. |
| 2100 | EXPECT_FALSE( |
| 2101 | FPDFTextObj_SetTextRenderMode(nullptr, FPDF_TEXTRENDERMODE_UNKNOWN)); |
| 2102 | EXPECT_FALSE( |
| 2103 | FPDFTextObj_SetTextRenderMode(nullptr, FPDF_TEXTRENDERMODE_INVISIBLE)); |
| 2104 | |
| 2105 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 2106 | ASSERT_TRUE(page_object); |
| 2107 | EXPECT_EQ(FPDF_TEXTRENDERMODE_FILL, |
| 2108 | FPDFTextObj_GetTextRenderMode(page_object)); |
| 2109 | |
| 2110 | // Cannot set UNKNOWN as a render mode. |
| 2111 | EXPECT_FALSE(FPDFTextObj_SetTextRenderMode(page_object, |
| 2112 | FPDF_TEXTRENDERMODE_UNKNOWN)); |
| 2113 | |
| 2114 | EXPECT_TRUE( |
| 2115 | FPDFTextObj_SetTextRenderMode(page_object, FPDF_TEXTRENDERMODE_STROKE)); |
| 2116 | EXPECT_EQ(FPDF_TEXTRENDERMODE_STROKE, |
| 2117 | FPDFTextObj_GetTextRenderMode(page_object)); |
| 2118 | |
| 2119 | // Check that bitmap displays changed content |
| 2120 | { |
| 2121 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
| 2122 | CompareBitmap(page_bitmap.get(), 612, 446, md5_stroke); |
| 2123 | } |
| 2124 | |
Daniel Hosseinian | afd2750 | 2020-07-01 04:59:03 +0000 | [diff] [blame] | 2125 | // Save a copy. |
| 2126 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 2127 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2128 | |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2129 | UnloadPage(page); |
Daniel Hosseinian | e83d8b4 | 2020-01-08 00:26:15 +0000 | [diff] [blame] | 2130 | } |
Miklos Vajna | 1448cc1 | 2018-07-03 13:52:33 +0000 | [diff] [blame] | 2131 | |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2132 | { |
Daniel Hosseinian | afd2750 | 2020-07-01 04:59:03 +0000 | [diff] [blame] | 2133 | // Open the saved copy and render it. Check that the changed text render |
| 2134 | // mode is kept in the saved copy. |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2135 | ASSERT_TRUE(OpenSavedDocument()); |
| 2136 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 2137 | ASSERT_TRUE(saved_page); |
| 2138 | |
| 2139 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(saved_page, 0); |
| 2140 | EXPECT_TRUE(page_object); |
Daniel Hosseinian | afd2750 | 2020-07-01 04:59:03 +0000 | [diff] [blame] | 2141 | EXPECT_EQ(FPDF_TEXTRENDERMODE_STROKE, |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2142 | FPDFTextObj_GetTextRenderMode(page_object)); |
| 2143 | |
| 2144 | ScopedFPDFBitmap bitmap = RenderSavedPage(saved_page); |
Daniel Hosseinian | afd2750 | 2020-07-01 04:59:03 +0000 | [diff] [blame] | 2145 | CompareBitmap(bitmap.get(), 612, 446, md5_stroke); |
Daniel Hosseinian | a933ebf | 2020-01-31 21:28:25 +0000 | [diff] [blame] | 2146 | |
| 2147 | CloseSavedPage(saved_page); |
| 2148 | CloseSavedDocument(); |
| 2149 | } |
Miklos Vajna | 1448cc1 | 2018-07-03 13:52:33 +0000 | [diff] [blame] | 2150 | } |
| 2151 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2152 | TEST_F(FPDFEditEmbedderTest, TestGetTextFontName) { |
Miklos Vajna | 53d4f0a | 2018-08-01 01:28:49 +0000 | [diff] [blame] | 2153 | EXPECT_TRUE(OpenDocument("text_font.pdf")); |
| 2154 | FPDF_PAGE page = LoadPage(0); |
| 2155 | ASSERT_TRUE(page); |
| 2156 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 2157 | |
| 2158 | // FPDFTextObj_GetFontName() positive testing. |
| 2159 | FPDF_PAGEOBJECT text = FPDFPage_GetObject(page, 0); |
| 2160 | unsigned long size = FPDFTextObj_GetFontName(text, nullptr, 0); |
| 2161 | const char kExpectedFontName[] = "Liberation Serif"; |
| 2162 | ASSERT_EQ(sizeof(kExpectedFontName), size); |
| 2163 | std::vector<char> font_name(size); |
| 2164 | ASSERT_EQ(size, FPDFTextObj_GetFontName(text, font_name.data(), size)); |
| 2165 | ASSERT_STREQ(kExpectedFontName, font_name.data()); |
| 2166 | |
| 2167 | // FPDFTextObj_GetFontName() negative testing. |
| 2168 | ASSERT_EQ(0U, FPDFTextObj_GetFontName(nullptr, nullptr, 0)); |
| 2169 | |
| 2170 | font_name.resize(2); |
| 2171 | font_name[0] = 'x'; |
| 2172 | font_name[1] = '\0'; |
| 2173 | size = FPDFTextObj_GetFontName(text, font_name.data(), font_name.size()); |
| 2174 | ASSERT_EQ(sizeof(kExpectedFontName), size); |
| 2175 | ASSERT_EQ(std::string("x"), std::string(font_name.data())); |
| 2176 | |
| 2177 | UnloadPage(page); |
| 2178 | } |
| 2179 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2180 | TEST_F(FPDFEditEmbedderTest, TestFormGetObjects) { |
Miklos Vajna | b66077d | 2018-07-11 13:25:02 +0000 | [diff] [blame] | 2181 | EXPECT_TRUE(OpenDocument("form_object.pdf")); |
| 2182 | FPDF_PAGE page = LoadPage(0); |
| 2183 | ASSERT_TRUE(page); |
| 2184 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 2185 | |
| 2186 | FPDF_PAGEOBJECT form = FPDFPage_GetObject(page, 0); |
| 2187 | EXPECT_EQ(FPDF_PAGEOBJ_FORM, FPDFPageObj_GetType(form)); |
| 2188 | ASSERT_EQ(-1, FPDFFormObj_CountObjects(nullptr)); |
| 2189 | ASSERT_EQ(2, FPDFFormObj_CountObjects(form)); |
| 2190 | |
Miklos Vajna | 1d273f1 | 2018-07-16 19:20:36 +0000 | [diff] [blame] | 2191 | // FPDFFormObj_GetObject() positive testing. |
| 2192 | FPDF_PAGEOBJECT text1 = FPDFFormObj_GetObject(form, 0); |
| 2193 | ASSERT_TRUE(text1); |
| 2194 | float left = 0; |
| 2195 | float bottom = 0; |
| 2196 | float right = 0; |
| 2197 | float top = 0; |
| 2198 | ASSERT_TRUE(FPDFPageObj_GetBounds(text1, &left, &bottom, &right, &top)); |
| 2199 | ASSERT_EQ(271, static_cast<int>(top)); |
| 2200 | |
| 2201 | FPDF_PAGEOBJECT text2 = FPDFFormObj_GetObject(form, 1); |
| 2202 | ASSERT_TRUE(text2); |
| 2203 | ASSERT_TRUE(FPDFPageObj_GetBounds(text2, &left, &bottom, &right, &top)); |
| 2204 | ASSERT_EQ(221, static_cast<int>(top)); |
| 2205 | |
| 2206 | // FPDFFormObj_GetObject() negative testing. |
| 2207 | ASSERT_EQ(nullptr, FPDFFormObj_GetObject(nullptr, 0)); |
| 2208 | ASSERT_EQ(nullptr, FPDFFormObj_GetObject(form, -1)); |
| 2209 | ASSERT_EQ(nullptr, FPDFFormObj_GetObject(form, 2)); |
| 2210 | |
Miklos Vajna | 46b4373 | 2018-08-14 19:15:43 +0000 | [diff] [blame] | 2211 | // Reset the form object matrix to identity. |
| 2212 | auto* pPageObj = CPDFPageObjectFromFPDFPageObject(form); |
| 2213 | CPDF_FormObject* pFormObj = pPageObj->AsForm(); |
| 2214 | pFormObj->Transform(pFormObj->form_matrix().GetInverse()); |
| 2215 | |
| 2216 | // FPDFFormObj_GetMatrix() positive testing. |
Lei Zhang | 8da9823 | 2019-12-11 23:29:33 +0000 | [diff] [blame] | 2217 | static constexpr FS_MATRIX kMatrix = {1.0f, 1.5f, 2.0f, 2.5f, 100.0f, 200.0f}; |
| 2218 | pFormObj->Transform(CFXMatrixFromFSMatrix(kMatrix)); |
Miklos Vajna | 46b4373 | 2018-08-14 19:15:43 +0000 | [diff] [blame] | 2219 | |
Lei Zhang | 8da9823 | 2019-12-11 23:29:33 +0000 | [diff] [blame] | 2220 | FS_MATRIX matrix; |
| 2221 | EXPECT_TRUE(FPDFFormObj_GetMatrix(form, &matrix)); |
| 2222 | EXPECT_FLOAT_EQ(kMatrix.a, matrix.a); |
| 2223 | EXPECT_FLOAT_EQ(kMatrix.b, matrix.b); |
| 2224 | EXPECT_FLOAT_EQ(kMatrix.c, matrix.c); |
| 2225 | EXPECT_FLOAT_EQ(kMatrix.d, matrix.d); |
| 2226 | EXPECT_FLOAT_EQ(kMatrix.e, matrix.e); |
| 2227 | EXPECT_FLOAT_EQ(kMatrix.f, matrix.f); |
Miklos Vajna | 46b4373 | 2018-08-14 19:15:43 +0000 | [diff] [blame] | 2228 | |
| 2229 | // FPDFFormObj_GetMatrix() negative testing. |
Lei Zhang | 8da9823 | 2019-12-11 23:29:33 +0000 | [diff] [blame] | 2230 | EXPECT_FALSE(FPDFFormObj_GetMatrix(nullptr, &matrix)); |
| 2231 | EXPECT_FALSE(FPDFFormObj_GetMatrix(form, nullptr)); |
| 2232 | EXPECT_FALSE(FPDFFormObj_GetMatrix(nullptr, nullptr)); |
Miklos Vajna | 46b4373 | 2018-08-14 19:15:43 +0000 | [diff] [blame] | 2233 | |
Miklos Vajna | b66077d | 2018-07-11 13:25:02 +0000 | [diff] [blame] | 2234 | UnloadPage(page); |
| 2235 | } |
| 2236 | |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2237 | // Tests adding text from standard font using FPDFText_LoadStandardFont. |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 2238 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 2239 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 2240 | #define MAYBE_AddStandardFontText2 DISABLED_AddStandardFontText2 |
| 2241 | #else |
| 2242 | #define MAYBE_AddStandardFontText2 AddStandardFontText2 |
| 2243 | #endif |
| 2244 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddStandardFontText2) { |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2245 | // Start with a blank page |
| 2246 | ScopedFPDFPage page(FPDFPage_New(CreateNewDocument(), 0, 612, 792)); |
| 2247 | |
| 2248 | // Load a standard font. |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2249 | ScopedFPDFFont font(FPDFText_LoadStandardFont(document(), "Helvetica")); |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2250 | ASSERT_TRUE(font); |
| 2251 | |
| 2252 | // Add some text to the page. |
| 2253 | FPDF_PAGEOBJECT text_object = |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2254 | FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f); |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2255 | EXPECT_TRUE(text_object); |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 2256 | ScopedFPDFWideString text = GetFPDFWideString(kBottomText); |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2257 | EXPECT_TRUE(FPDFText_SetText(text_object, text.get())); |
| 2258 | FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 20, 20); |
| 2259 | FPDFPage_InsertObject(page.get(), text_object); |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2260 | ScopedFPDFBitmap page_bitmap = RenderPage(page.get()); |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 2261 | CompareBitmap(page_bitmap.get(), 612, 792, kBottomTextChecksum); |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2262 | } |
| 2263 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2264 | TEST_F(FPDFEditEmbedderTest, LoadStandardFonts) { |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2265 | CreateNewDocument(); |
Lei Zhang | 590f424 | 2019-05-15 20:57:26 +0000 | [diff] [blame] | 2266 | static constexpr const char* kStandardFontNames[] = { |
Lei Zhang | d72fd58 | 2018-07-27 19:37:27 +0000 | [diff] [blame] | 2267 | "Arial", |
| 2268 | "Arial-Bold", |
| 2269 | "Arial-BoldItalic", |
| 2270 | "Arial-Italic", |
| 2271 | "Courier", |
| 2272 | "Courier-BoldOblique", |
| 2273 | "Courier-Oblique", |
| 2274 | "Courier-Bold", |
| 2275 | "CourierNew", |
| 2276 | "CourierNew-Bold", |
| 2277 | "CourierNew-BoldItalic", |
| 2278 | "CourierNew-Italic", |
| 2279 | "Helvetica", |
| 2280 | "Helvetica-Bold", |
| 2281 | "Helvetica-BoldOblique", |
| 2282 | "Helvetica-Oblique", |
| 2283 | "Symbol", |
| 2284 | "TimesNewRoman", |
| 2285 | "TimesNewRoman-Bold", |
| 2286 | "TimesNewRoman-BoldItalic", |
| 2287 | "TimesNewRoman-Italic", |
| 2288 | "ZapfDingbats"}; |
Lei Zhang | 590f424 | 2019-05-15 20:57:26 +0000 | [diff] [blame] | 2289 | for (const char* font_name : kStandardFontNames) { |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2290 | ScopedFPDFFont font(FPDFText_LoadStandardFont(document(), font_name)); |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2291 | EXPECT_TRUE(font) << font_name << " should be considered a standard font."; |
| 2292 | } |
Lei Zhang | 590f424 | 2019-05-15 20:57:26 +0000 | [diff] [blame] | 2293 | static constexpr const char* kNotStandardFontNames[] = { |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2294 | "Abcdefg", "ArialB", "Arial-Style", |
| 2295 | "Font Name", "FontArial", "NotAStandardFontName", |
| 2296 | "TestFontName", "Quack", "Symbol-Italic", |
| 2297 | "Zapf"}; |
Lei Zhang | 590f424 | 2019-05-15 20:57:26 +0000 | [diff] [blame] | 2298 | for (const char* font_name : kNotStandardFontNames) { |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2299 | ScopedFPDFFont font(FPDFText_LoadStandardFont(document(), font_name)); |
Nicolas Pena | 4c48b10 | 2018-06-13 18:23:46 +0000 | [diff] [blame] | 2300 | EXPECT_FALSE(font) << font_name |
| 2301 | << " should not be considered a standard font."; |
| 2302 | } |
| 2303 | } |
| 2304 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2305 | TEST_F(FPDFEditEmbedderTest, GraphicsData) { |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2306 | // New page |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 2307 | ScopedFPDFPage page(FPDFPage_New(CreateNewDocument(), 0, 612, 792)); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2308 | |
| 2309 | // Create a rect with nontrivial graphics |
| 2310 | FPDF_PAGEOBJECT rect1 = FPDFPageObj_CreateNewRect(10, 10, 100, 100); |
| 2311 | FPDFPageObj_SetBlendMode(rect1, "Color"); |
| 2312 | FPDFPage_InsertObject(page.get(), rect1); |
| 2313 | EXPECT_TRUE(FPDFPage_GenerateContent(page.get())); |
| 2314 | |
| 2315 | // Check that the ExtGState was created |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2316 | CPDF_Page* cpage = CPDFPageFromFPDFPage(page.get()); |
| 2317 | CPDF_Dictionary* graphics_dict = cpage->m_pResources->GetDictFor("ExtGState"); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2318 | ASSERT_TRUE(graphics_dict); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2319 | EXPECT_EQ(2u, graphics_dict->size()); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2320 | |
| 2321 | // Add a text object causing no change to the graphics dictionary |
| 2322 | FPDF_PAGEOBJECT text1 = FPDFPageObj_NewTextObj(document(), "Arial", 12.0f); |
| 2323 | // Only alpha, the last component, matters for the graphics dictionary. And |
| 2324 | // the default value is 255. |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 2325 | EXPECT_TRUE(FPDFPageObj_SetFillColor(text1, 100, 100, 100, 255)); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2326 | FPDFPage_InsertObject(page.get(), text1); |
| 2327 | EXPECT_TRUE(FPDFPage_GenerateContent(page.get())); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2328 | EXPECT_EQ(2u, graphics_dict->size()); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2329 | |
| 2330 | // Add a text object increasing the size of the graphics dictionary |
| 2331 | FPDF_PAGEOBJECT text2 = |
| 2332 | FPDFPageObj_NewTextObj(document(), "Times-Roman", 12.0f); |
| 2333 | FPDFPage_InsertObject(page.get(), text2); |
| 2334 | FPDFPageObj_SetBlendMode(text2, "Darken"); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 2335 | EXPECT_TRUE(FPDFPageObj_SetFillColor(text2, 0, 0, 255, 150)); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2336 | EXPECT_TRUE(FPDFPage_GenerateContent(page.get())); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2337 | EXPECT_EQ(3u, graphics_dict->size()); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2338 | |
| 2339 | // Add a path that should reuse graphics |
Nicolas Pena | ce67be4 | 2017-06-14 14:52:49 -0400 | [diff] [blame] | 2340 | FPDF_PAGEOBJECT path = FPDFPageObj_CreateNewPath(400, 100); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2341 | FPDFPageObj_SetBlendMode(path, "Darken"); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 2342 | EXPECT_TRUE(FPDFPageObj_SetFillColor(path, 200, 200, 100, 150)); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2343 | FPDFPage_InsertObject(page.get(), path); |
| 2344 | EXPECT_TRUE(FPDFPage_GenerateContent(page.get())); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2345 | EXPECT_EQ(3u, graphics_dict->size()); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2346 | |
| 2347 | // Add a rect increasing the size of the graphics dictionary |
| 2348 | FPDF_PAGEOBJECT rect2 = FPDFPageObj_CreateNewRect(10, 10, 100, 100); |
| 2349 | FPDFPageObj_SetBlendMode(rect2, "Darken"); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 2350 | EXPECT_TRUE(FPDFPageObj_SetFillColor(rect2, 0, 0, 255, 150)); |
| 2351 | EXPECT_TRUE(FPDFPageObj_SetStrokeColor(rect2, 0, 0, 0, 200)); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2352 | FPDFPage_InsertObject(page.get(), rect2); |
| 2353 | EXPECT_TRUE(FPDFPage_GenerateContent(page.get())); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2354 | EXPECT_EQ(4u, graphics_dict->size()); |
Nicolas Pena | 603a31d | 2017-06-14 11:41:18 -0400 | [diff] [blame] | 2355 | } |
| 2356 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2357 | TEST_F(FPDFEditEmbedderTest, DoubleGenerating) { |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2358 | // Start with a blank page |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2359 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2360 | |
| 2361 | // Add a red rectangle with some non-default alpha |
| 2362 | FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(10, 10, 100, 100); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 2363 | EXPECT_TRUE(FPDFPageObj_SetFillColor(rect, 255, 0, 0, 128)); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2364 | EXPECT_TRUE(FPDFPath_SetDrawMode(rect, FPDF_FILLMODE_WINDING, 0)); |
| 2365 | FPDFPage_InsertObject(page, rect); |
| 2366 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 2367 | |
| 2368 | // Check the ExtGState |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2369 | CPDF_Page* cpage = CPDFPageFromFPDFPage(page); |
| 2370 | CPDF_Dictionary* graphics_dict = cpage->m_pResources->GetDictFor("ExtGState"); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2371 | ASSERT_TRUE(graphics_dict); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2372 | EXPECT_EQ(2u, graphics_dict->size()); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2373 | |
| 2374 | // Check the bitmap |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2375 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2376 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2377 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 2378 | "5384da3406d62360ffb5cac4476fff1c"); |
| 2379 | } |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2380 | |
| 2381 | // Never mind, my new favorite color is blue, increase alpha |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 2382 | EXPECT_TRUE(FPDFPageObj_SetFillColor(rect, 0, 0, 255, 180)); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2383 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2384 | EXPECT_EQ(3u, graphics_dict->size()); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2385 | |
| 2386 | // Check that bitmap displays changed content |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2387 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2388 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2389 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 2390 | "2e51656f5073b0bee611d9cd086aa09c"); |
| 2391 | } |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2392 | |
| 2393 | // And now generate, without changes |
| 2394 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2395 | EXPECT_EQ(3u, graphics_dict->size()); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2396 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2397 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2398 | CompareBitmap(page_bitmap.get(), 612, 792, |
| 2399 | "2e51656f5073b0bee611d9cd086aa09c"); |
| 2400 | } |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2401 | |
| 2402 | // Add some text to the page |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2403 | FPDF_PAGEOBJECT text_object = |
| 2404 | FPDFPageObj_NewTextObj(document(), "Arial", 12.0f); |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 2405 | ScopedFPDFWideString text = |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2406 | GetFPDFWideString(L"Something something #text# something"); |
| 2407 | EXPECT_TRUE(FPDFText_SetText(text_object, text.get())); |
| 2408 | FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 300, 300); |
| 2409 | FPDFPage_InsertObject(page, text_object); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2410 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2411 | CPDF_Dictionary* font_dict = cpage->m_pResources->GetDictFor("Font"); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2412 | ASSERT_TRUE(font_dict); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2413 | EXPECT_EQ(1u, font_dict->size()); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2414 | |
| 2415 | // Generate yet again, check dicts are reasonably sized |
| 2416 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2417 | EXPECT_EQ(3u, graphics_dict->size()); |
| 2418 | EXPECT_EQ(1u, font_dict->size()); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2419 | FPDF_ClosePage(page); |
Nicolas Pena | a4ad01f | 2017-02-15 16:26:48 -0500 | [diff] [blame] | 2420 | } |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2421 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2422 | TEST_F(FPDFEditEmbedderTest, LoadSimpleType1Font) { |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2423 | CreateNewDocument(); |
| 2424 | // TODO(npm): use other fonts after disallowing loading any font as any type |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2425 | RetainPtr<CPDF_Font> stock_font = |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2426 | CPDF_Font::GetStockFont(cpdf_doc(), "Times-Bold"); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2427 | pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan(); |
| 2428 | ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(), |
| 2429 | FPDF_FONT_TYPE1, false)); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2430 | ASSERT_TRUE(font.get()); |
Tom Sepez | 525147a | 2018-05-03 17:19:53 +0000 | [diff] [blame] | 2431 | CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2432 | EXPECT_TRUE(typed_font->IsType1Font()); |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2433 | |
Lei Zhang | 710fa99 | 2018-05-25 16:24:48 +0000 | [diff] [blame] | 2434 | const CPDF_Dictionary* font_dict = typed_font->GetFontDict(); |
Lei Zhang | a228ff3 | 2020-06-24 17:39:33 +0000 | [diff] [blame] | 2435 | EXPECT_EQ("Font", font_dict->GetNameFor("Type")); |
| 2436 | EXPECT_EQ("Type1", font_dict->GetNameFor("Subtype")); |
| 2437 | EXPECT_EQ("TimesNewRomanPS-BoldMT", font_dict->GetNameFor("BaseFont")); |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2438 | ASSERT_TRUE(font_dict->KeyExist("FirstChar")); |
| 2439 | ASSERT_TRUE(font_dict->KeyExist("LastChar")); |
| 2440 | EXPECT_EQ(32, font_dict->GetIntegerFor("FirstChar")); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2441 | EXPECT_EQ(255, font_dict->GetIntegerFor("LastChar")); |
| 2442 | |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2443 | const CPDF_Array* widths_array = font_dict->GetArrayFor("Widths"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2444 | ASSERT_TRUE(widths_array); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2445 | ASSERT_EQ(224u, widths_array->size()); |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2446 | EXPECT_EQ(250, widths_array->GetNumberAt(0)); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2447 | EXPECT_EQ(569, widths_array->GetNumberAt(11)); |
| 2448 | EXPECT_EQ(500, widths_array->GetNumberAt(223)); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2449 | CheckFontDescriptor(font_dict, FPDF_FONT_TYPE1, true, false, span); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2450 | } |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2451 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2452 | TEST_F(FPDFEditEmbedderTest, LoadSimpleTrueTypeFont) { |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2453 | CreateNewDocument(); |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2454 | RetainPtr<CPDF_Font> stock_font = |
| 2455 | CPDF_Font::GetStockFont(cpdf_doc(), "Courier"); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2456 | pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan(); |
| 2457 | ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(), |
| 2458 | FPDF_FONT_TRUETYPE, false)); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2459 | ASSERT_TRUE(font.get()); |
Tom Sepez | 525147a | 2018-05-03 17:19:53 +0000 | [diff] [blame] | 2460 | CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2461 | EXPECT_TRUE(typed_font->IsTrueTypeFont()); |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2462 | |
Lei Zhang | 710fa99 | 2018-05-25 16:24:48 +0000 | [diff] [blame] | 2463 | const CPDF_Dictionary* font_dict = typed_font->GetFontDict(); |
Lei Zhang | a228ff3 | 2020-06-24 17:39:33 +0000 | [diff] [blame] | 2464 | EXPECT_EQ("Font", font_dict->GetNameFor("Type")); |
| 2465 | EXPECT_EQ("TrueType", font_dict->GetNameFor("Subtype")); |
| 2466 | EXPECT_EQ("CourierNewPSMT", font_dict->GetNameFor("BaseFont")); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2467 | ASSERT_TRUE(font_dict->KeyExist("FirstChar")); |
| 2468 | ASSERT_TRUE(font_dict->KeyExist("LastChar")); |
| 2469 | EXPECT_EQ(32, font_dict->GetIntegerFor("FirstChar")); |
| 2470 | EXPECT_EQ(255, font_dict->GetIntegerFor("LastChar")); |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2471 | |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2472 | const CPDF_Array* widths_array = font_dict->GetArrayFor("Widths"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2473 | ASSERT_TRUE(widths_array); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2474 | ASSERT_EQ(224u, widths_array->size()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2475 | EXPECT_EQ(600, widths_array->GetNumberAt(33)); |
| 2476 | EXPECT_EQ(600, widths_array->GetNumberAt(74)); |
| 2477 | EXPECT_EQ(600, widths_array->GetNumberAt(223)); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2478 | CheckFontDescriptor(font_dict, FPDF_FONT_TRUETYPE, false, false, span); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2479 | } |
| 2480 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2481 | TEST_F(FPDFEditEmbedderTest, LoadCIDType0Font) { |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2482 | CreateNewDocument(); |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2483 | RetainPtr<CPDF_Font> stock_font = |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2484 | CPDF_Font::GetStockFont(cpdf_doc(), "Times-Roman"); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2485 | pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan(); |
| 2486 | ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(), |
| 2487 | FPDF_FONT_TYPE1, 1)); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2488 | ASSERT_TRUE(font.get()); |
Tom Sepez | 525147a | 2018-05-03 17:19:53 +0000 | [diff] [blame] | 2489 | CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2490 | EXPECT_TRUE(typed_font->IsCIDFont()); |
| 2491 | |
| 2492 | // Check font dictionary entries |
Lei Zhang | 710fa99 | 2018-05-25 16:24:48 +0000 | [diff] [blame] | 2493 | const CPDF_Dictionary* font_dict = typed_font->GetFontDict(); |
Lei Zhang | a228ff3 | 2020-06-24 17:39:33 +0000 | [diff] [blame] | 2494 | EXPECT_EQ("Font", font_dict->GetNameFor("Type")); |
| 2495 | EXPECT_EQ("Type0", font_dict->GetNameFor("Subtype")); |
| 2496 | EXPECT_EQ("TimesNewRomanPSMT-Identity-H", font_dict->GetNameFor("BaseFont")); |
| 2497 | EXPECT_EQ("Identity-H", font_dict->GetNameFor("Encoding")); |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2498 | const CPDF_Array* descendant_array = |
| 2499 | font_dict->GetArrayFor("DescendantFonts"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2500 | ASSERT_TRUE(descendant_array); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2501 | EXPECT_EQ(1u, descendant_array->size()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2502 | |
| 2503 | // Check the CIDFontDict |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2504 | const CPDF_Dictionary* cidfont_dict = descendant_array->GetDictAt(0); |
Lei Zhang | a228ff3 | 2020-06-24 17:39:33 +0000 | [diff] [blame] | 2505 | EXPECT_EQ("Font", cidfont_dict->GetNameFor("Type")); |
| 2506 | EXPECT_EQ("CIDFontType0", cidfont_dict->GetNameFor("Subtype")); |
| 2507 | EXPECT_EQ("TimesNewRomanPSMT", cidfont_dict->GetNameFor("BaseFont")); |
Lei Zhang | b1ec280 | 2018-05-25 21:55:24 +0000 | [diff] [blame] | 2508 | const CPDF_Dictionary* cidinfo_dict = |
| 2509 | cidfont_dict->GetDictFor("CIDSystemInfo"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2510 | ASSERT_TRUE(cidinfo_dict); |
Lei Zhang | 9c950b1 | 2019-01-16 19:06:37 +0000 | [diff] [blame] | 2511 | const CPDF_Object* registry = cidinfo_dict->GetObjectFor("Registry"); |
| 2512 | ASSERT_TRUE(registry); |
KDr2 | 8da0e1b | 2019-01-17 03:44:29 +0000 | [diff] [blame] | 2513 | EXPECT_EQ(CPDF_Object::kString, registry->GetType()); |
Lei Zhang | 9c950b1 | 2019-01-16 19:06:37 +0000 | [diff] [blame] | 2514 | EXPECT_EQ("Adobe", registry->GetString()); |
| 2515 | const CPDF_Object* ordering = cidinfo_dict->GetObjectFor("Ordering"); |
| 2516 | ASSERT_TRUE(ordering); |
KDr2 | 8da0e1b | 2019-01-17 03:44:29 +0000 | [diff] [blame] | 2517 | EXPECT_EQ(CPDF_Object::kString, ordering->GetType()); |
Lei Zhang | 9c950b1 | 2019-01-16 19:06:37 +0000 | [diff] [blame] | 2518 | EXPECT_EQ("Identity", ordering->GetString()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2519 | EXPECT_EQ(0, cidinfo_dict->GetNumberFor("Supplement")); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2520 | CheckFontDescriptor(cidfont_dict, FPDF_FONT_TYPE1, false, false, span); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2521 | |
| 2522 | // Check widths |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2523 | const CPDF_Array* widths_array = cidfont_dict->GetArrayFor("W"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2524 | ASSERT_TRUE(widths_array); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2525 | EXPECT_GT(widths_array->size(), 1u); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2526 | CheckCompositeFontWidths(widths_array, typed_font); |
| 2527 | } |
| 2528 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2529 | TEST_F(FPDFEditEmbedderTest, LoadCIDType2Font) { |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2530 | CreateNewDocument(); |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2531 | RetainPtr<CPDF_Font> stock_font = |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2532 | CPDF_Font::GetStockFont(cpdf_doc(), "Helvetica-Oblique"); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2533 | pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan(); |
| 2534 | ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(), |
| 2535 | FPDF_FONT_TRUETYPE, 1)); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2536 | ASSERT_TRUE(font.get()); |
Tom Sepez | 525147a | 2018-05-03 17:19:53 +0000 | [diff] [blame] | 2537 | CPDF_Font* typed_font = CPDFFontFromFPDFFont(font.get()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2538 | EXPECT_TRUE(typed_font->IsCIDFont()); |
| 2539 | |
| 2540 | // Check font dictionary entries |
Lei Zhang | 710fa99 | 2018-05-25 16:24:48 +0000 | [diff] [blame] | 2541 | const CPDF_Dictionary* font_dict = typed_font->GetFontDict(); |
Lei Zhang | a228ff3 | 2020-06-24 17:39:33 +0000 | [diff] [blame] | 2542 | EXPECT_EQ("Font", font_dict->GetNameFor("Type")); |
| 2543 | EXPECT_EQ("Type0", font_dict->GetNameFor("Subtype")); |
| 2544 | EXPECT_EQ("Arial-ItalicMT", font_dict->GetNameFor("BaseFont")); |
| 2545 | EXPECT_EQ("Identity-H", font_dict->GetNameFor("Encoding")); |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2546 | const CPDF_Array* descendant_array = |
| 2547 | font_dict->GetArrayFor("DescendantFonts"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2548 | ASSERT_TRUE(descendant_array); |
Lei Zhang | f40380f | 2018-10-12 18:31:51 +0000 | [diff] [blame] | 2549 | EXPECT_EQ(1u, descendant_array->size()); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2550 | |
| 2551 | // Check the CIDFontDict |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2552 | const CPDF_Dictionary* cidfont_dict = descendant_array->GetDictAt(0); |
Lei Zhang | a228ff3 | 2020-06-24 17:39:33 +0000 | [diff] [blame] | 2553 | EXPECT_EQ("Font", cidfont_dict->GetNameFor("Type")); |
| 2554 | EXPECT_EQ("CIDFontType2", cidfont_dict->GetNameFor("Subtype")); |
| 2555 | EXPECT_EQ("Arial-ItalicMT", cidfont_dict->GetNameFor("BaseFont")); |
Lei Zhang | b1ec280 | 2018-05-25 21:55:24 +0000 | [diff] [blame] | 2556 | const CPDF_Dictionary* cidinfo_dict = |
| 2557 | cidfont_dict->GetDictFor("CIDSystemInfo"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2558 | ASSERT_TRUE(cidinfo_dict); |
| 2559 | EXPECT_EQ("Adobe", cidinfo_dict->GetStringFor("Registry")); |
| 2560 | EXPECT_EQ("Identity", cidinfo_dict->GetStringFor("Ordering")); |
| 2561 | EXPECT_EQ(0, cidinfo_dict->GetNumberFor("Supplement")); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2562 | CheckFontDescriptor(cidfont_dict, FPDF_FONT_TRUETYPE, false, true, span); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2563 | |
| 2564 | // Check widths |
Lei Zhang | de579ab | 2018-05-25 21:49:49 +0000 | [diff] [blame] | 2565 | const CPDF_Array* widths_array = cidfont_dict->GetArrayFor("W"); |
Nicolas Pena | d03ca42 | 2017-03-06 13:54:33 -0500 | [diff] [blame] | 2566 | ASSERT_TRUE(widths_array); |
| 2567 | CheckCompositeFontWidths(widths_array, typed_font); |
Nicolas Pena | be90aae | 2017-02-27 10:41:41 -0500 | [diff] [blame] | 2568 | } |
rbpotter | ce8e51e | 2017-04-28 12:42:47 -0700 | [diff] [blame] | 2569 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2570 | TEST_F(FPDFEditEmbedderTest, NormalizeNegativeRotation) { |
rbpotter | ce8e51e | 2017-04-28 12:42:47 -0700 | [diff] [blame] | 2571 | // Load document with a -90 degree rotation |
| 2572 | EXPECT_TRUE(OpenDocument("bug_713197.pdf")); |
| 2573 | FPDF_PAGE page = LoadPage(0); |
| 2574 | EXPECT_NE(nullptr, page); |
| 2575 | |
| 2576 | EXPECT_EQ(3, FPDFPage_GetRotation(page)); |
| 2577 | UnloadPage(page); |
| 2578 | } |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2579 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 2580 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 2581 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 2582 | #define MAYBE_AddTrueTypeFontText DISABLED_AddTrueTypeFontText |
| 2583 | #else |
| 2584 | #define MAYBE_AddTrueTypeFontText AddTrueTypeFontText |
| 2585 | #endif |
| 2586 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddTrueTypeFontText) { |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2587 | // Start with a blank page |
| 2588 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
| 2589 | { |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2590 | RetainPtr<CPDF_Font> stock_font = |
| 2591 | CPDF_Font::GetStockFont(cpdf_doc(), "Arial"); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2592 | pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan(); |
| 2593 | ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(), |
| 2594 | FPDF_FONT_TRUETYPE, 0)); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2595 | ASSERT_TRUE(font.get()); |
| 2596 | |
| 2597 | // Add some text to the page |
| 2598 | FPDF_PAGEOBJECT text_object = |
| 2599 | FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f); |
| 2600 | EXPECT_TRUE(text_object); |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 2601 | ScopedFPDFWideString text = GetFPDFWideString(kLoadedFontText); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2602 | EXPECT_TRUE(FPDFText_SetText(text_object, text.get())); |
| 2603 | FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 400, 400); |
| 2604 | FPDFPage_InsertObject(page, text_object); |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2605 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 2606 | CompareBitmap(page_bitmap.get(), 612, 792, kLoadedFontTextChecksum); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2607 | |
| 2608 | // Add some more text, same font |
| 2609 | FPDF_PAGEOBJECT text_object2 = |
| 2610 | FPDFPageObj_CreateTextObj(document(), font.get(), 15.0f); |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 2611 | ScopedFPDFWideString text2 = GetFPDFWideString(L"Bigger font size"); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2612 | EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get())); |
| 2613 | FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 200, 200); |
| 2614 | FPDFPage_InsertObject(page, text_object2); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2615 | } |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2616 | ScopedFPDFBitmap page_bitmap2 = RenderPage(page); |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 2617 | #if defined(OS_WIN) |
| 2618 | const char kInsertTrueTypeChecksum[] = "2199b579c49ab5f80c246a586a80ee90"; |
| 2619 | #elif defined(OS_MACOSX) |
| 2620 | const char kInsertTrueTypeChecksum[] = "9a1a7dfebe659513691aadd0d95b8d50"; |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2621 | #else |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 2622 | const char kInsertTrueTypeChecksum[] = "c1d10cce1761c4a998a16b2562030568"; |
Lei Zhang | e4cdac5 | 2019-04-30 16:45:57 +0000 | [diff] [blame] | 2623 | #endif |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 2624 | CompareBitmap(page_bitmap2.get(), 612, 792, kInsertTrueTypeChecksum); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2625 | |
Nicolas Pena | 207b727 | 2017-05-26 17:37:06 -0400 | [diff] [blame] | 2626 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2627 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2628 | FPDF_ClosePage(page); |
Dan Sinclair | 04e4dc8 | 2017-10-18 12:17:14 -0400 | [diff] [blame] | 2629 | |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 2630 | VerifySavedDocument(612, 792, kInsertTrueTypeChecksum); |
Nicolas Pena | b316185 | 2017-05-02 14:12:50 -0400 | [diff] [blame] | 2631 | } |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2632 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2633 | TEST_F(FPDFEditEmbedderTest, TransformAnnot) { |
Jane Liu | eda6525 | 2017-06-07 11:31:27 -0400 | [diff] [blame] | 2634 | // Open a file with one annotation and load its first page. |
| 2635 | ASSERT_TRUE(OpenDocument("annotation_highlight_long_content.pdf")); |
Lei Zhang | 75c8171 | 2018-02-08 17:22:39 +0000 | [diff] [blame] | 2636 | FPDF_PAGE page = LoadPage(0); |
Jane Liu | eda6525 | 2017-06-07 11:31:27 -0400 | [diff] [blame] | 2637 | ASSERT_TRUE(page); |
| 2638 | |
Lei Zhang | a21d593 | 2018-02-05 18:28:38 +0000 | [diff] [blame] | 2639 | { |
| 2640 | // Add an underline annotation to the page without specifying its rectangle. |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 2641 | ScopedFPDFAnnotation annot( |
Lei Zhang | a21d593 | 2018-02-05 18:28:38 +0000 | [diff] [blame] | 2642 | FPDFPage_CreateAnnot(page, FPDF_ANNOT_UNDERLINE)); |
| 2643 | ASSERT_TRUE(annot); |
Jane Liu | eda6525 | 2017-06-07 11:31:27 -0400 | [diff] [blame] | 2644 | |
Lei Zhang | a21d593 | 2018-02-05 18:28:38 +0000 | [diff] [blame] | 2645 | // FPDFPage_TransformAnnots() should run without errors when modifying |
| 2646 | // annotation rectangles. |
| 2647 | FPDFPage_TransformAnnots(page, 1, 2, 3, 4, 5, 6); |
| 2648 | } |
Jane Liu | eda6525 | 2017-06-07 11:31:27 -0400 | [diff] [blame] | 2649 | UnloadPage(page); |
| 2650 | } |
| 2651 | |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2652 | // TODO(npm): Add tests using Japanese fonts in other OS. |
Dan Sinclair | 698aed7 | 2017-09-26 16:24:49 -0400 | [diff] [blame] | 2653 | #if _FX_PLATFORM_ == _FX_PLATFORM_LINUX_ |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 2654 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 2655 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 2656 | #define MAYBE_AddCIDFontText DISABLED_AddCIDFontText |
| 2657 | #else |
| 2658 | #define MAYBE_AddCIDFontText AddCIDFontText |
| 2659 | #endif |
| 2660 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddCIDFontText) { |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2661 | // Start with a blank page |
| 2662 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
| 2663 | CFX_Font CIDfont; |
| 2664 | { |
| 2665 | // First, get the data from the font |
| 2666 | CIDfont.LoadSubst("IPAGothic", 1, 0, 400, 0, 932, 0); |
| 2667 | EXPECT_EQ("IPAGothic", CIDfont.GetFaceName()); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2668 | pdfium::span<const uint8_t> span = CIDfont.GetFontSpan(); |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2669 | |
| 2670 | // Load the data into a FPDF_Font. |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2671 | ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(), |
| 2672 | FPDF_FONT_TRUETYPE, 1)); |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2673 | ASSERT_TRUE(font.get()); |
| 2674 | |
| 2675 | // Add some text to the page |
| 2676 | FPDF_PAGEOBJECT text_object = |
| 2677 | FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f); |
| 2678 | ASSERT_TRUE(text_object); |
| 2679 | std::wstring wstr = L"ABCDEFGhijklmnop."; |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 2680 | ScopedFPDFWideString text = GetFPDFWideString(wstr); |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2681 | EXPECT_TRUE(FPDFText_SetText(text_object, text.get())); |
| 2682 | FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 200, 200); |
| 2683 | FPDFPage_InsertObject(page, text_object); |
| 2684 | |
| 2685 | // And add some Japanese characters |
| 2686 | FPDF_PAGEOBJECT text_object2 = |
| 2687 | FPDFPageObj_CreateTextObj(document(), font.get(), 18.0f); |
| 2688 | ASSERT_TRUE(text_object2); |
| 2689 | std::wstring wstr2 = |
| 2690 | L"\u3053\u3093\u306B\u3061\u306f\u4e16\u754C\u3002\u3053\u3053\u306B1" |
| 2691 | L"\u756A"; |
Lei Zhang | f0f6768 | 2019-04-08 17:03:21 +0000 | [diff] [blame] | 2692 | ScopedFPDFWideString text2 = GetFPDFWideString(wstr2); |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2693 | EXPECT_TRUE(FPDFText_SetText(text_object2, text2.get())); |
| 2694 | FPDFPageObj_Transform(text_object2, 1, 0, 0, 1, 100, 500); |
| 2695 | FPDFPage_InsertObject(page, text_object2); |
| 2696 | } |
| 2697 | |
Nicolas Pena | 207b727 | 2017-05-26 17:37:06 -0400 | [diff] [blame] | 2698 | // Check that the text renders properly. |
Lei Zhang | a2b7073 | 2019-06-25 08:34:22 +0000 | [diff] [blame] | 2699 | const char md5[] = "5159a72903fe57bf0cf645c894de8a74"; |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2700 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2701 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2702 | CompareBitmap(page_bitmap.get(), 612, 792, md5); |
| 2703 | } |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2704 | |
| 2705 | // Save the document, close the page. |
Nicolas Pena | 207b727 | 2017-05-26 17:37:06 -0400 | [diff] [blame] | 2706 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2707 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2708 | FPDF_ClosePage(page); |
Dan Sinclair | 04e4dc8 | 2017-10-18 12:17:14 -0400 | [diff] [blame] | 2709 | |
| 2710 | VerifySavedDocument(612, 792, md5); |
Nicolas Pena | f45ade3 | 2017-05-03 10:23:49 -0400 | [diff] [blame] | 2711 | } |
Dan Sinclair | 698aed7 | 2017-09-26 16:24:49 -0400 | [diff] [blame] | 2712 | #endif // _FX_PLATFORM_ == _FX_PLATFORM_LINUX_ |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 2713 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 2714 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 2715 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 2716 | #define MAYBE_SaveAndRender DISABLED_SaveAndRender |
| 2717 | #else |
| 2718 | #define MAYBE_SaveAndRender SaveAndRender |
| 2719 | #endif |
| 2720 | TEST_F(FPDFEditEmbedderTest, MAYBE_SaveAndRender) { |
Nicolas Pena | a0b48aa | 2017-06-29 11:01:46 -0400 | [diff] [blame] | 2721 | const char md5[] = "3c20472b0552c0c22b88ab1ed8c6202b"; |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 2722 | { |
| 2723 | EXPECT_TRUE(OpenDocument("bug_779.pdf")); |
| 2724 | FPDF_PAGE page = LoadPage(0); |
| 2725 | ASSERT_NE(nullptr, page); |
| 2726 | |
| 2727 | // Now add a more complex blue path. |
| 2728 | FPDF_PAGEOBJECT green_path = FPDFPageObj_CreateNewPath(20, 20); |
Lei Zhang | 3475b48 | 2019-05-13 18:30:57 +0000 | [diff] [blame] | 2729 | EXPECT_TRUE(FPDFPageObj_SetFillColor(green_path, 0, 255, 0, 200)); |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 2730 | // TODO(npm): stroking will cause the MD5s to differ. |
| 2731 | EXPECT_TRUE(FPDFPath_SetDrawMode(green_path, FPDF_FILLMODE_WINDING, 0)); |
| 2732 | EXPECT_TRUE(FPDFPath_LineTo(green_path, 20, 63)); |
| 2733 | EXPECT_TRUE(FPDFPath_BezierTo(green_path, 55, 55, 78, 78, 90, 90)); |
| 2734 | EXPECT_TRUE(FPDFPath_LineTo(green_path, 133, 133)); |
| 2735 | EXPECT_TRUE(FPDFPath_LineTo(green_path, 133, 33)); |
| 2736 | EXPECT_TRUE(FPDFPath_BezierTo(green_path, 38, 33, 39, 36, 40, 40)); |
| 2737 | EXPECT_TRUE(FPDFPath_Close(green_path)); |
| 2738 | FPDFPage_InsertObject(page, green_path); |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 2739 | ScopedFPDFBitmap page_bitmap = RenderLoadedPage(page); |
Lei Zhang | 107fa7b | 2018-02-09 21:48:15 +0000 | [diff] [blame] | 2740 | CompareBitmap(page_bitmap.get(), 612, 792, md5); |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 2741 | |
| 2742 | // Now save the result, closing the page and document |
| 2743 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 2744 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2745 | UnloadPage(page); |
| 2746 | } |
Dan Sinclair | 04e4dc8 | 2017-10-18 12:17:14 -0400 | [diff] [blame] | 2747 | |
| 2748 | VerifySavedDocument(612, 792, md5); |
Nicolas Pena | 9ba8fbc | 2017-06-28 15:31:56 -0400 | [diff] [blame] | 2749 | } |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 2750 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2751 | TEST_F(FPDFEditEmbedderTest, AddMark) { |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 2752 | // Load document with some text. |
| 2753 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 2754 | FPDF_PAGE page = LoadPage(0); |
| 2755 | ASSERT_TRUE(page); |
| 2756 | |
Lei Zhang | 2697cb1 | 2019-07-03 18:14:29 +0000 | [diff] [blame] | 2757 | CheckMarkCounts(page, 1, 19, 8, 4, 9, 1); |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 2758 | |
| 2759 | // Add to the first page object a "Bounds" mark with "Position": "First". |
| 2760 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 2761 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_AddMark(page_object, "Bounds"); |
| 2762 | EXPECT_TRUE(mark); |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 2763 | EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), page_object, mark, |
| 2764 | "Position", "First")); |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 2765 | |
Lei Zhang | 2697cb1 | 2019-07-03 18:14:29 +0000 | [diff] [blame] | 2766 | CheckMarkCounts(page, 1, 19, 8, 4, 9, 2); |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 2767 | |
| 2768 | // Save the file |
| 2769 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 2770 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2771 | UnloadPage(page); |
| 2772 | |
| 2773 | // Re-open the file and check the new mark is present. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 2774 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 2775 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 2776 | |
Lei Zhang | 2697cb1 | 2019-07-03 18:14:29 +0000 | [diff] [blame] | 2777 | CheckMarkCounts(saved_page, 1, 19, 8, 4, 9, 2); |
| 2778 | |
| 2779 | CloseSavedPage(saved_page); |
| 2780 | CloseSavedDocument(); |
| 2781 | } |
| 2782 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 2783 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 2784 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 2785 | #define MAYBE_AddMarkCompressedStream DISABLED_AddMarkCompressedStream |
| 2786 | #else |
| 2787 | #define MAYBE_AddMarkCompressedStream AddMarkCompressedStream |
| 2788 | #endif |
| 2789 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddMarkCompressedStream) { |
Lei Zhang | 2697cb1 | 2019-07-03 18:14:29 +0000 | [diff] [blame] | 2790 | // Load document with some text in a compressed stream. |
| 2791 | EXPECT_TRUE(OpenDocument("hello_world_compressed_stream.pdf")); |
| 2792 | FPDF_PAGE page = LoadPage(0); |
| 2793 | ASSERT_TRUE(page); |
| 2794 | |
| 2795 | // Render and check there are no marks. |
| 2796 | { |
| 2797 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 2798 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum); |
Lei Zhang | 2697cb1 | 2019-07-03 18:14:29 +0000 | [diff] [blame] | 2799 | } |
| 2800 | CheckMarkCounts(page, 0, 2, 0, 0, 0, 0); |
| 2801 | |
| 2802 | // Add to the first page object a "Bounds" mark with "Position": "First". |
| 2803 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 0); |
| 2804 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_AddMark(page_object, "Bounds"); |
| 2805 | EXPECT_TRUE(mark); |
| 2806 | EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), page_object, mark, |
| 2807 | "Position", "First")); |
| 2808 | |
| 2809 | // Render and check there is 1 mark. |
| 2810 | { |
| 2811 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 2812 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum); |
Lei Zhang | 2697cb1 | 2019-07-03 18:14:29 +0000 | [diff] [blame] | 2813 | } |
| 2814 | CheckMarkCounts(page, 0, 2, 0, 0, 0, 1); |
| 2815 | |
| 2816 | // Save the file. |
| 2817 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 2818 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2819 | UnloadPage(page); |
| 2820 | |
| 2821 | // Re-open the file and check the new mark is present. |
| 2822 | ASSERT_TRUE(OpenSavedDocument()); |
| 2823 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 2824 | |
| 2825 | { |
| 2826 | ScopedFPDFBitmap page_bitmap = RenderPage(saved_page); |
Hui Yingst | b4baceb | 2020-04-28 23:46:10 +0000 | [diff] [blame] | 2827 | CompareBitmap(page_bitmap.get(), 200, 200, kHelloWorldChecksum); |
Lei Zhang | 2697cb1 | 2019-07-03 18:14:29 +0000 | [diff] [blame] | 2828 | } |
| 2829 | CheckMarkCounts(saved_page, 0, 2, 0, 0, 0, 1); |
Henrique Nakashima | d8df8c3 | 2018-07-12 22:15:09 +0000 | [diff] [blame] | 2830 | |
| 2831 | CloseSavedPage(saved_page); |
| 2832 | CloseSavedDocument(); |
| 2833 | } |
| 2834 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 2835 | TEST_F(FPDFEditEmbedderTest, SetMarkParam) { |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 2836 | // Load document with some text. |
| 2837 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 2838 | FPDF_PAGE page = LoadPage(0); |
| 2839 | ASSERT_TRUE(page); |
| 2840 | |
| 2841 | constexpr int kExpectedObjectCount = 19; |
| 2842 | CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1); |
| 2843 | |
| 2844 | // Check the "Bounds" mark's "Position" param is "Last". |
| 2845 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18); |
| 2846 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1); |
| 2847 | ASSERT_TRUE(mark); |
| 2848 | char buffer[256]; |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 2849 | unsigned long name_len = 999u; |
| 2850 | ASSERT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 2851 | EXPECT_EQ((6u + 1u) * 2u, name_len); |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 2852 | ASSERT_EQ(L"Bounds", |
| 2853 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer))); |
| 2854 | unsigned long out_buffer_len; |
| 2855 | ASSERT_TRUE(FPDFPageObjMark_GetParamStringValue( |
| 2856 | mark, "Position", buffer, sizeof(buffer), &out_buffer_len)); |
| 2857 | ASSERT_EQ(L"Last", |
| 2858 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer))); |
| 2859 | |
| 2860 | // Set is to "End". |
| 2861 | EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), page_object, mark, |
| 2862 | "Position", "End")); |
| 2863 | |
| 2864 | // Verify the object passed must correspond to the mark passed. |
| 2865 | FPDF_PAGEOBJECT another_page_object = FPDFPage_GetObject(page, 17); |
| 2866 | EXPECT_FALSE(FPDFPageObjMark_SetStringParam(document(), another_page_object, |
| 2867 | mark, "Position", "End")); |
| 2868 | |
| 2869 | // Verify nothing else changed. |
| 2870 | CheckMarkCounts(page, 1, kExpectedObjectCount, 8, 4, 9, 1); |
| 2871 | |
| 2872 | // Verify "Position" now maps to "End". |
| 2873 | EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue( |
| 2874 | mark, "Position", buffer, sizeof(buffer), &out_buffer_len)); |
| 2875 | EXPECT_EQ(L"End", |
| 2876 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer))); |
| 2877 | |
| 2878 | // Save the file |
| 2879 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 2880 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2881 | UnloadPage(page); |
| 2882 | |
| 2883 | // Re-open the file and cerify "Position" still maps to "End". |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 2884 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 2885 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 2886 | |
| 2887 | CheckMarkCounts(saved_page, 1, kExpectedObjectCount, 8, 4, 9, 1); |
| 2888 | page_object = FPDFPage_GetObject(saved_page, 18); |
| 2889 | mark = FPDFPageObj_GetMark(page_object, 1); |
| 2890 | EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue( |
| 2891 | mark, "Position", buffer, sizeof(buffer), &out_buffer_len)); |
| 2892 | EXPECT_EQ(L"End", |
| 2893 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer))); |
| 2894 | |
| 2895 | CloseSavedPage(saved_page); |
| 2896 | CloseSavedDocument(); |
| 2897 | } |
| 2898 | |
Lei Zhang | 03e5e68 | 2019-09-16 19:45:55 +0000 | [diff] [blame] | 2899 | // TODO(crbug.com/pdfium/11): Fix this test and enable. |
| 2900 | #if defined(_SKIA_SUPPORT_) || defined(_SKIA_SUPPORT_PATHS_) |
| 2901 | #define MAYBE_AddMarkedText DISABLED_AddMarkedText |
| 2902 | #else |
| 2903 | #define MAYBE_AddMarkedText AddMarkedText |
| 2904 | #endif |
| 2905 | TEST_F(FPDFEditEmbedderTest, MAYBE_AddMarkedText) { |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2906 | // Start with a blank page. |
| 2907 | FPDF_PAGE page = FPDFPage_New(CreateNewDocument(), 0, 612, 792); |
| 2908 | |
Tom Sepez | ffff6c5 | 2019-07-30 21:56:19 +0000 | [diff] [blame] | 2909 | RetainPtr<CPDF_Font> stock_font = |
| 2910 | CPDF_Font::GetStockFont(cpdf_doc(), "Arial"); |
Tom Sepez | 20c41a5 | 2018-08-29 23:53:53 +0000 | [diff] [blame] | 2911 | pdfium::span<const uint8_t> span = stock_font->GetFont()->GetFontSpan(); |
| 2912 | ScopedFPDFFont font(FPDFText_LoadFont(document(), span.data(), span.size(), |
| 2913 | FPDF_FONT_TRUETYPE, 0)); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2914 | ASSERT_TRUE(font.get()); |
| 2915 | |
| 2916 | // Add some text to the page. |
| 2917 | FPDF_PAGEOBJECT text_object = |
| 2918 | FPDFPageObj_CreateTextObj(document(), font.get(), 12.0f); |
| 2919 | |
| 2920 | EXPECT_TRUE(text_object); |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 2921 | ScopedFPDFWideString text1 = GetFPDFWideString(kLoadedFontText); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2922 | EXPECT_TRUE(FPDFText_SetText(text_object, text1.get())); |
| 2923 | FPDFPageObj_Transform(text_object, 1, 0, 0, 1, 400, 400); |
| 2924 | FPDFPage_InsertObject(page, text_object); |
| 2925 | |
| 2926 | // Add a mark with the tag "TestMarkName" to that text. |
| 2927 | EXPECT_EQ(0, FPDFPageObj_CountMarks(text_object)); |
Henrique Nakashima | 6fc8d87 | 2018-09-18 16:48:31 +0000 | [diff] [blame] | 2928 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_AddMark(text_object, "Test Mark Name"); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2929 | EXPECT_TRUE(mark); |
| 2930 | EXPECT_EQ(1, FPDFPageObj_CountMarks(text_object)); |
| 2931 | EXPECT_EQ(mark, FPDFPageObj_GetMark(text_object, 0)); |
| 2932 | char buffer[256]; |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 2933 | unsigned long name_len = 999u; |
| 2934 | ASSERT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 2935 | EXPECT_EQ((14u + 1u) * 2, name_len); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2936 | std::wstring name = |
| 2937 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
Henrique Nakashima | 6fc8d87 | 2018-09-18 16:48:31 +0000 | [diff] [blame] | 2938 | EXPECT_EQ(L"Test Mark Name", name); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2939 | |
| 2940 | // Add parameters: |
| 2941 | // - int "IntKey" : 42 |
| 2942 | // - string "StringKey": "StringValue" |
Henrique Nakashima | 07520f6 | 2018-07-12 19:45:29 +0000 | [diff] [blame] | 2943 | // - blob "BlobKey": "\x01\x02\x03\0BlobValue1\0\0\0BlobValue2\0" |
Lei Zhang | 590f424 | 2019-05-15 20:57:26 +0000 | [diff] [blame] | 2944 | constexpr size_t kBlobLen = 28; |
Lei Zhang | d3b028b | 2018-11-30 22:22:00 +0000 | [diff] [blame] | 2945 | char block_value[kBlobLen]; |
| 2946 | memcpy(block_value, "\x01\x02\x03\0BlobValue1\0\0\0BlobValue2\0", kBlobLen); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2947 | EXPECT_EQ(0, FPDFPageObjMark_CountParams(mark)); |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 2948 | EXPECT_TRUE( |
| 2949 | FPDFPageObjMark_SetIntParam(document(), text_object, mark, "IntKey", 42)); |
| 2950 | EXPECT_TRUE(FPDFPageObjMark_SetStringParam(document(), text_object, mark, |
| 2951 | "StringKey", "StringValue")); |
| 2952 | EXPECT_TRUE(FPDFPageObjMark_SetBlobParam(document(), text_object, mark, |
Lei Zhang | d3b028b | 2018-11-30 22:22:00 +0000 | [diff] [blame] | 2953 | "BlobKey", block_value, kBlobLen)); |
Henrique Nakashima | 07520f6 | 2018-07-12 19:45:29 +0000 | [diff] [blame] | 2954 | EXPECT_EQ(3, FPDFPageObjMark_CountParams(mark)); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2955 | |
| 2956 | // Check the two parameters can be retrieved. |
| 2957 | EXPECT_EQ(FPDF_OBJECT_NUMBER, |
Henrique Nakashima | 94230e5 | 2018-07-11 22:02:02 +0000 | [diff] [blame] | 2958 | FPDFPageObjMark_GetParamValueType(mark, "IntKey")); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2959 | int int_value; |
Henrique Nakashima | 94230e5 | 2018-07-11 22:02:02 +0000 | [diff] [blame] | 2960 | EXPECT_TRUE(FPDFPageObjMark_GetParamIntValue(mark, "IntKey", &int_value)); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2961 | EXPECT_EQ(42, int_value); |
| 2962 | |
| 2963 | EXPECT_EQ(FPDF_OBJECT_STRING, |
Henrique Nakashima | 94230e5 | 2018-07-11 22:02:02 +0000 | [diff] [blame] | 2964 | FPDFPageObjMark_GetParamValueType(mark, "StringKey")); |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 2965 | unsigned long out_buffer_len = 999u; |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 2966 | EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue( |
| 2967 | mark, "StringKey", buffer, sizeof(buffer), &out_buffer_len)); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2968 | EXPECT_GT(out_buffer_len, 0u); |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 2969 | EXPECT_NE(999u, out_buffer_len); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2970 | name = GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
| 2971 | EXPECT_EQ(L"StringValue", name); |
| 2972 | |
Henrique Nakashima | 07520f6 | 2018-07-12 19:45:29 +0000 | [diff] [blame] | 2973 | EXPECT_EQ(FPDF_OBJECT_STRING, |
| 2974 | FPDFPageObjMark_GetParamValueType(mark, "BlobKey")); |
| 2975 | out_buffer_len = 0; |
Henrique Nakashima | a340677 | 2018-07-13 19:10:53 +0000 | [diff] [blame] | 2976 | EXPECT_TRUE(FPDFPageObjMark_GetParamBlobValue( |
| 2977 | mark, "BlobKey", buffer, sizeof(buffer), &out_buffer_len)); |
Henrique Nakashima | 07520f6 | 2018-07-12 19:45:29 +0000 | [diff] [blame] | 2978 | EXPECT_EQ(kBlobLen, out_buffer_len); |
Lei Zhang | d3b028b | 2018-11-30 22:22:00 +0000 | [diff] [blame] | 2979 | EXPECT_EQ(0, memcmp(block_value, buffer, kBlobLen)); |
Henrique Nakashima | 07520f6 | 2018-07-12 19:45:29 +0000 | [diff] [blame] | 2980 | |
Lei Zhang | f16376e | 2020-06-18 19:19:32 +0000 | [diff] [blame] | 2981 | // Render and check the bitmap is the expected one. |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2982 | { |
Lei Zhang | 30ff253 | 2019-01-31 21:37:55 +0000 | [diff] [blame] | 2983 | ScopedFPDFBitmap page_bitmap = RenderPage(page); |
Hui Yingst | 3b6136a | 2020-06-09 00:39:33 +0000 | [diff] [blame] | 2984 | CompareBitmap(page_bitmap.get(), 612, 792, kLoadedFontTextChecksum); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2985 | } |
| 2986 | |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 2987 | // Now save the result. |
| 2988 | EXPECT_EQ(1, FPDFPage_CountObjects(page)); |
| 2989 | EXPECT_TRUE(FPDFPage_GenerateContent(page)); |
| 2990 | EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
| 2991 | |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 2992 | FPDF_ClosePage(page); |
| 2993 | |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 2994 | // Re-open the file and check the changes were kept in the saved .pdf. |
Lei Zhang | 0b49405 | 2019-01-31 21:41:15 +0000 | [diff] [blame] | 2995 | ASSERT_TRUE(OpenSavedDocument()); |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 2996 | FPDF_PAGE saved_page = LoadSavedPage(0); |
| 2997 | EXPECT_EQ(1, FPDFPage_CountObjects(saved_page)); |
| 2998 | |
| 2999 | text_object = FPDFPage_GetObject(saved_page, 0); |
| 3000 | EXPECT_TRUE(text_object); |
| 3001 | EXPECT_EQ(1, FPDFPageObj_CountMarks(text_object)); |
| 3002 | mark = FPDFPageObj_GetMark(text_object, 0); |
| 3003 | EXPECT_TRUE(mark); |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 3004 | |
| 3005 | name_len = 999u; |
| 3006 | ASSERT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &name_len)); |
| 3007 | EXPECT_EQ((14u + 1u) * 2, name_len); |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 3008 | name = GetPlatformWString(reinterpret_cast<unsigned short*>(buffer)); |
Henrique Nakashima | 6fc8d87 | 2018-09-18 16:48:31 +0000 | [diff] [blame] | 3009 | EXPECT_EQ(L"Test Mark Name", name); |
Henrique Nakashima | b4bcf69 | 2018-07-11 21:19:22 +0000 | [diff] [blame] | 3010 | |
| 3011 | CloseSavedPage(saved_page); |
| 3012 | CloseSavedDocument(); |
Henrique Nakashima | 144107d | 2018-07-10 21:04:05 +0000 | [diff] [blame] | 3013 | } |
| 3014 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3015 | TEST_F(FPDFEditEmbedderTest, MarkGetName) { |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 3016 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 3017 | FPDF_PAGE page = LoadPage(0); |
| 3018 | ASSERT_TRUE(page); |
| 3019 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18); |
| 3020 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1); |
| 3021 | ASSERT_TRUE(mark); |
| 3022 | |
| 3023 | char buffer[256]; |
| 3024 | unsigned long out_len; |
| 3025 | |
| 3026 | // Show the positive cases of FPDFPageObjMark_GetName. |
| 3027 | out_len = 999u; |
| 3028 | EXPECT_TRUE(FPDFPageObjMark_GetName(mark, nullptr, 0, &out_len)); |
| 3029 | EXPECT_EQ((6u + 1u) * 2u, out_len); |
| 3030 | |
| 3031 | out_len = 999u; |
| 3032 | EXPECT_TRUE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), &out_len)); |
| 3033 | EXPECT_EQ(L"Bounds", |
| 3034 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer))); |
| 3035 | EXPECT_EQ((6u + 1u) * 2u, out_len); |
| 3036 | |
| 3037 | // Show the negative cases of FPDFPageObjMark_GetName. |
| 3038 | out_len = 999u; |
| 3039 | EXPECT_FALSE( |
| 3040 | FPDFPageObjMark_GetName(nullptr, buffer, sizeof(buffer), &out_len)); |
| 3041 | EXPECT_EQ(999u, out_len); |
| 3042 | |
| 3043 | EXPECT_FALSE(FPDFPageObjMark_GetName(mark, buffer, sizeof(buffer), nullptr)); |
| 3044 | |
| 3045 | UnloadPage(page); |
| 3046 | } |
| 3047 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3048 | TEST_F(FPDFEditEmbedderTest, MarkGetParamKey) { |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 3049 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 3050 | FPDF_PAGE page = LoadPage(0); |
| 3051 | ASSERT_TRUE(page); |
| 3052 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18); |
| 3053 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1); |
| 3054 | ASSERT_TRUE(mark); |
| 3055 | |
| 3056 | char buffer[256]; |
| 3057 | unsigned long out_len; |
| 3058 | |
| 3059 | // Show the positive cases of FPDFPageObjMark_GetParamKey. |
| 3060 | out_len = 999u; |
| 3061 | EXPECT_TRUE(FPDFPageObjMark_GetParamKey(mark, 0, nullptr, 0, &out_len)); |
| 3062 | EXPECT_EQ((8u + 1u) * 2u, out_len); |
| 3063 | |
| 3064 | out_len = 999u; |
| 3065 | EXPECT_TRUE( |
| 3066 | FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer), &out_len)); |
| 3067 | EXPECT_EQ(L"Position", |
| 3068 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer))); |
| 3069 | EXPECT_EQ((8u + 1u) * 2u, out_len); |
| 3070 | |
| 3071 | // Show the negative cases of FPDFPageObjMark_GetParamKey. |
| 3072 | out_len = 999u; |
| 3073 | EXPECT_FALSE(FPDFPageObjMark_GetParamKey(nullptr, 0, buffer, sizeof(buffer), |
| 3074 | &out_len)); |
| 3075 | EXPECT_EQ(999u, out_len); |
| 3076 | |
| 3077 | out_len = 999u; |
| 3078 | EXPECT_FALSE( |
| 3079 | FPDFPageObjMark_GetParamKey(mark, 1, buffer, sizeof(buffer), &out_len)); |
| 3080 | EXPECT_EQ(999u, out_len); |
| 3081 | |
| 3082 | EXPECT_FALSE( |
| 3083 | FPDFPageObjMark_GetParamKey(mark, 0, buffer, sizeof(buffer), nullptr)); |
| 3084 | |
| 3085 | UnloadPage(page); |
| 3086 | } |
| 3087 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3088 | TEST_F(FPDFEditEmbedderTest, MarkGetIntParam) { |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 3089 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 3090 | FPDF_PAGE page = LoadPage(0); |
| 3091 | ASSERT_TRUE(page); |
| 3092 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 8); |
| 3093 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 0); |
| 3094 | ASSERT_TRUE(mark); |
| 3095 | |
| 3096 | int out_value; |
| 3097 | |
| 3098 | // Show the positive cases of FPDFPageObjMark_GetParamIntValue. |
| 3099 | out_value = 999; |
| 3100 | EXPECT_TRUE(FPDFPageObjMark_GetParamIntValue(mark, "Factor", &out_value)); |
| 3101 | EXPECT_EQ(3, out_value); |
| 3102 | |
| 3103 | // Show the negative cases of FPDFPageObjMark_GetParamIntValue. |
| 3104 | out_value = 999; |
| 3105 | EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(nullptr, "Factor", &out_value)); |
| 3106 | EXPECT_EQ(999, out_value); |
| 3107 | |
| 3108 | out_value = 999; |
| 3109 | EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(mark, "ParamThatDoesNotExist", |
| 3110 | &out_value)); |
| 3111 | EXPECT_EQ(999, out_value); |
| 3112 | |
| 3113 | EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(mark, "Factor", nullptr)); |
| 3114 | |
| 3115 | page_object = FPDFPage_GetObject(page, 18); |
| 3116 | mark = FPDFPageObj_GetMark(page_object, 1); |
| 3117 | out_value = 999; |
| 3118 | EXPECT_FALSE(FPDFPageObjMark_GetParamIntValue(mark, "Position", &out_value)); |
| 3119 | EXPECT_EQ(999, out_value); |
| 3120 | |
| 3121 | UnloadPage(page); |
| 3122 | } |
| 3123 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3124 | TEST_F(FPDFEditEmbedderTest, MarkGetStringParam) { |
Henrique Nakashima | c3099d1 | 2018-09-18 18:08:15 +0000 | [diff] [blame] | 3125 | EXPECT_TRUE(OpenDocument("text_in_page_marked.pdf")); |
| 3126 | FPDF_PAGE page = LoadPage(0); |
| 3127 | ASSERT_TRUE(page); |
| 3128 | FPDF_PAGEOBJECT page_object = FPDFPage_GetObject(page, 18); |
| 3129 | FPDF_PAGEOBJECTMARK mark = FPDFPageObj_GetMark(page_object, 1); |
| 3130 | ASSERT_TRUE(mark); |
| 3131 | |
| 3132 | char buffer[256]; |
| 3133 | unsigned long out_len; |
| 3134 | |
| 3135 | // Show the positive cases of FPDFPageObjMark_GetParamStringValue. |
| 3136 | out_len = 999u; |
| 3137 | EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(mark, "Position", nullptr, 0, |
| 3138 | &out_len)); |
| 3139 | EXPECT_EQ((4u + 1u) * 2u, out_len); |
| 3140 | |
| 3141 | out_len = 999u; |
| 3142 | EXPECT_TRUE(FPDFPageObjMark_GetParamStringValue(mark, "Position", buffer, |
| 3143 | sizeof(buffer), &out_len)); |
| 3144 | EXPECT_EQ(L"Last", |
| 3145 | GetPlatformWString(reinterpret_cast<unsigned short*>(buffer))); |
| 3146 | EXPECT_EQ((4u + 1u) * 2u, out_len); |
| 3147 | |
| 3148 | // Show the negative cases of FPDFPageObjMark_GetParamStringValue. |
| 3149 | out_len = 999u; |
| 3150 | EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue(nullptr, "Position", buffer, |
| 3151 | sizeof(buffer), &out_len)); |
| 3152 | EXPECT_EQ(999u, out_len); |
| 3153 | |
| 3154 | out_len = 999u; |
| 3155 | EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue( |
| 3156 | mark, "ParamThatDoesNotExist", buffer, sizeof(buffer), &out_len)); |
| 3157 | EXPECT_EQ(999u, out_len); |
| 3158 | |
| 3159 | EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue(mark, "Position", buffer, |
| 3160 | sizeof(buffer), nullptr)); |
| 3161 | |
| 3162 | page_object = FPDFPage_GetObject(page, 8); |
| 3163 | mark = FPDFPageObj_GetMark(page_object, 0); |
| 3164 | out_len = 999u; |
| 3165 | EXPECT_FALSE(FPDFPageObjMark_GetParamStringValue(mark, "Factor", buffer, |
| 3166 | sizeof(buffer), &out_len)); |
| 3167 | EXPECT_EQ(999u, out_len); |
| 3168 | |
| 3169 | UnloadPage(page); |
| 3170 | } |
| 3171 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3172 | TEST_F(FPDFEditEmbedderTest, ExtractImageBitmap) { |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 3173 | ASSERT_TRUE(OpenDocument("embedded_images.pdf")); |
| 3174 | FPDF_PAGE page = LoadPage(0); |
| 3175 | ASSERT_TRUE(page); |
Miklos Vajna | 9262761 | 2017-09-25 12:59:29 +0200 | [diff] [blame] | 3176 | ASSERT_EQ(39, FPDFPage_CountObjects(page)); |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 3177 | |
| 3178 | FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 32); |
| 3179 | EXPECT_NE(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3180 | EXPECT_FALSE(FPDFImageObj_GetBitmap(obj)); |
| 3181 | |
| 3182 | obj = FPDFPage_GetObject(page, 33); |
| 3183 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3184 | FPDF_BITMAP bitmap = FPDFImageObj_GetBitmap(obj); |
| 3185 | EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap)); |
Lei Zhang | bf98014 | 2019-12-20 01:05:42 +0000 | [diff] [blame] | 3186 | CompareBitmap(bitmap, 109, 88, "cb3637934bb3b95a6e4ae1ea9eb9e56e"); |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 3187 | FPDFBitmap_Destroy(bitmap); |
| 3188 | |
| 3189 | obj = FPDFPage_GetObject(page, 34); |
| 3190 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3191 | bitmap = FPDFImageObj_GetBitmap(obj); |
| 3192 | EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap)); |
Lei Zhang | bf98014 | 2019-12-20 01:05:42 +0000 | [diff] [blame] | 3193 | CompareBitmap(bitmap, 103, 75, "c8d51fa6821ceb2a67f08446ff236c40"); |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 3194 | FPDFBitmap_Destroy(bitmap); |
| 3195 | |
| 3196 | obj = FPDFPage_GetObject(page, 35); |
| 3197 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3198 | bitmap = FPDFImageObj_GetBitmap(obj); |
| 3199 | EXPECT_EQ(FPDFBitmap_Gray, FPDFBitmap_GetFormat(bitmap)); |
| 3200 | CompareBitmap(bitmap, 92, 68, "9c6d76cb1e37ef8514f9455d759391f3"); |
| 3201 | FPDFBitmap_Destroy(bitmap); |
| 3202 | |
| 3203 | obj = FPDFPage_GetObject(page, 36); |
| 3204 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3205 | bitmap = FPDFImageObj_GetBitmap(obj); |
| 3206 | EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap)); |
Lei Zhang | bf98014 | 2019-12-20 01:05:42 +0000 | [diff] [blame] | 3207 | CompareBitmap(bitmap, 79, 60, "f4e72fb783a01c7b4614cdc25eaa98ac"); |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 3208 | FPDFBitmap_Destroy(bitmap); |
| 3209 | |
| 3210 | obj = FPDFPage_GetObject(page, 37); |
| 3211 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3212 | bitmap = FPDFImageObj_GetBitmap(obj); |
| 3213 | EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap)); |
Lei Zhang | bf98014 | 2019-12-20 01:05:42 +0000 | [diff] [blame] | 3214 | CompareBitmap(bitmap, 126, 106, "2cf9e66414c72461f4ccbf9cdebdfa1b"); |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 3215 | FPDFBitmap_Destroy(bitmap); |
| 3216 | |
| 3217 | obj = FPDFPage_GetObject(page, 38); |
| 3218 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3219 | bitmap = FPDFImageObj_GetBitmap(obj); |
| 3220 | EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap)); |
Lei Zhang | bf98014 | 2019-12-20 01:05:42 +0000 | [diff] [blame] | 3221 | CompareBitmap(bitmap, 194, 119, "a8f3a126cec274dab8242fd2ccdc1b8b"); |
Jane Liu | 28fb7ba | 2017-08-02 21:45:57 -0400 | [diff] [blame] | 3222 | FPDFBitmap_Destroy(bitmap); |
| 3223 | UnloadPage(page); |
| 3224 | } |
Jane Liu | 548334e | 2017-08-03 16:33:40 -0400 | [diff] [blame] | 3225 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3226 | TEST_F(FPDFEditEmbedderTest, ExtractJBigImageBitmap) { |
Lei Zhang | 53341dd | 2018-03-01 15:42:47 +0000 | [diff] [blame] | 3227 | ASSERT_TRUE(OpenDocument("bug_631912.pdf")); |
| 3228 | FPDF_PAGE page = LoadPage(0); |
| 3229 | ASSERT_TRUE(page); |
| 3230 | ASSERT_EQ(1, FPDFPage_CountObjects(page)); |
| 3231 | |
| 3232 | FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 0); |
| 3233 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3234 | { |
Tom Sepez | e08d2b1 | 2018-04-25 18:49:32 +0000 | [diff] [blame] | 3235 | ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj)); |
Lei Zhang | 1330ebb | 2018-03-05 15:16:37 +0000 | [diff] [blame] | 3236 | ASSERT_TRUE(bitmap); |
| 3237 | EXPECT_EQ(FPDFBitmap_Gray, FPDFBitmap_GetFormat(bitmap.get())); |
| 3238 | CompareBitmap(bitmap.get(), 1152, 720, "3f6a48e2b3e91b799bf34567f55cb4de"); |
Lei Zhang | 53341dd | 2018-03-01 15:42:47 +0000 | [diff] [blame] | 3239 | } |
| 3240 | |
| 3241 | UnloadPage(page); |
| 3242 | } |
| 3243 | |
Lei Zhang | 2e1b4f2 | 2020-07-08 23:40:18 +0000 | [diff] [blame] | 3244 | // TODO(crbug.com/pdfium/1554): Fix FPDFImageObj_GetBitmap() to take /SMask into |
| 3245 | // account, or provide a new API to do that. |
| 3246 | TEST_F(FPDFEditEmbedderTest, ExtractSMaskBitmap) { |
| 3247 | ASSERT_TRUE(OpenDocument("matte.pdf")); |
| 3248 | FPDF_PAGE page = LoadPage(0); |
| 3249 | ASSERT_TRUE(page); |
| 3250 | |
| 3251 | constexpr int kExpectedObjects = 4; |
| 3252 | ASSERT_EQ(kExpectedObjects, FPDFPage_CountObjects(page)); |
| 3253 | |
| 3254 | for (int i = 0; i < kExpectedObjects; ++i) { |
| 3255 | FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, i); |
| 3256 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3257 | ScopedFPDFBitmap bitmap(FPDFImageObj_GetBitmap(obj)); |
| 3258 | ASSERT_TRUE(bitmap); |
| 3259 | EXPECT_EQ(FPDFBitmap_BGR, FPDFBitmap_GetFormat(bitmap.get())); |
| 3260 | CompareBitmap(bitmap.get(), 50, 50, "46c9a1dbe0b44765ce46017ad629a2fe"); |
| 3261 | } |
| 3262 | |
| 3263 | UnloadPage(page); |
| 3264 | } |
| 3265 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3266 | TEST_F(FPDFEditEmbedderTest, GetImageData) { |
Jane Liu | 548334e | 2017-08-03 16:33:40 -0400 | [diff] [blame] | 3267 | EXPECT_TRUE(OpenDocument("embedded_images.pdf")); |
| 3268 | FPDF_PAGE page = LoadPage(0); |
| 3269 | ASSERT_TRUE(page); |
Miklos Vajna | 9262761 | 2017-09-25 12:59:29 +0200 | [diff] [blame] | 3270 | ASSERT_EQ(39, FPDFPage_CountObjects(page)); |
Jane Liu | 548334e | 2017-08-03 16:33:40 -0400 | [diff] [blame] | 3271 | |
| 3272 | // Retrieve an image object with flate-encoded data stream. |
| 3273 | FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 33); |
| 3274 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3275 | |
| 3276 | // Check that the raw image data has the correct length and hash value. |
| 3277 | unsigned long len = FPDFImageObj_GetImageDataRaw(obj, nullptr, 0); |
| 3278 | std::vector<char> buf(len); |
| 3279 | EXPECT_EQ(4091u, FPDFImageObj_GetImageDataRaw(obj, buf.data(), len)); |
| 3280 | EXPECT_EQ("f73802327d2e88e890f653961bcda81a", |
| 3281 | GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len)); |
| 3282 | |
| 3283 | // Check that the decoded image data has the correct length and hash value. |
| 3284 | len = FPDFImageObj_GetImageDataDecoded(obj, nullptr, 0); |
| 3285 | buf.clear(); |
| 3286 | buf.resize(len); |
| 3287 | EXPECT_EQ(28776u, FPDFImageObj_GetImageDataDecoded(obj, buf.data(), len)); |
| 3288 | EXPECT_EQ("cb3637934bb3b95a6e4ae1ea9eb9e56e", |
| 3289 | GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len)); |
| 3290 | |
Lei Zhang | 351e8b0 | 2018-12-20 01:10:06 +0000 | [diff] [blame] | 3291 | // Retrieve an image object with DCTDecode-encoded data stream. |
Jane Liu | 548334e | 2017-08-03 16:33:40 -0400 | [diff] [blame] | 3292 | obj = FPDFPage_GetObject(page, 37); |
| 3293 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3294 | |
| 3295 | // Check that the raw image data has the correct length and hash value. |
| 3296 | len = FPDFImageObj_GetImageDataRaw(obj, nullptr, 0); |
| 3297 | buf.clear(); |
| 3298 | buf.resize(len); |
| 3299 | EXPECT_EQ(4370u, FPDFImageObj_GetImageDataRaw(obj, buf.data(), len)); |
| 3300 | EXPECT_EQ("6aae1f3710335023a9e12191be66b64b", |
| 3301 | GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len)); |
| 3302 | |
| 3303 | // Check that the decoded image data has the correct length and hash value, |
| 3304 | // which should be the same as those of the raw data, since this image is |
| 3305 | // encoded by a single DCTDecode filter and decoding is a noop. |
| 3306 | len = FPDFImageObj_GetImageDataDecoded(obj, nullptr, 0); |
| 3307 | buf.clear(); |
| 3308 | buf.resize(len); |
| 3309 | EXPECT_EQ(4370u, FPDFImageObj_GetImageDataDecoded(obj, buf.data(), len)); |
| 3310 | EXPECT_EQ("6aae1f3710335023a9e12191be66b64b", |
| 3311 | GenerateMD5Base16(reinterpret_cast<uint8_t*>(buf.data()), len)); |
| 3312 | |
| 3313 | UnloadPage(page); |
| 3314 | } |
Jane Liu | 2e5f0ae | 2017-08-08 15:23:27 -0400 | [diff] [blame] | 3315 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3316 | TEST_F(FPDFEditEmbedderTest, GetImageMatrix) { |
Lei Zhang | 866d688 | 2018-10-24 17:31:01 +0000 | [diff] [blame] | 3317 | ASSERT_TRUE(OpenDocument("embedded_images.pdf")); |
| 3318 | FPDF_PAGE page = LoadPage(0); |
| 3319 | ASSERT_TRUE(page); |
| 3320 | ASSERT_EQ(39, FPDFPage_CountObjects(page)); |
| 3321 | |
| 3322 | FPDF_PAGEOBJECT obj; |
| 3323 | double a; |
| 3324 | double b; |
| 3325 | double c; |
| 3326 | double d; |
| 3327 | double e; |
| 3328 | double f; |
| 3329 | |
| 3330 | obj = FPDFPage_GetObject(page, 33); |
| 3331 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3332 | EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f)); |
| 3333 | EXPECT_DOUBLE_EQ(53.0, a); |
| 3334 | EXPECT_DOUBLE_EQ(0.0, b); |
| 3335 | EXPECT_DOUBLE_EQ(0.0, c); |
| 3336 | EXPECT_DOUBLE_EQ(43.0, d); |
| 3337 | EXPECT_DOUBLE_EQ(72.0, e); |
| 3338 | EXPECT_DOUBLE_EQ(646.510009765625, f); |
| 3339 | |
| 3340 | obj = FPDFPage_GetObject(page, 34); |
| 3341 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3342 | EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f)); |
| 3343 | EXPECT_DOUBLE_EQ(70.0, a); |
| 3344 | EXPECT_DOUBLE_EQ(0.0, b); |
| 3345 | EXPECT_DOUBLE_EQ(0.0, c); |
| 3346 | EXPECT_DOUBLE_EQ(51.0, d); |
| 3347 | EXPECT_DOUBLE_EQ(216.0, e); |
| 3348 | EXPECT_DOUBLE_EQ(646.510009765625, f); |
| 3349 | |
| 3350 | obj = FPDFPage_GetObject(page, 35); |
| 3351 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3352 | EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f)); |
| 3353 | EXPECT_DOUBLE_EQ(69.0, a); |
| 3354 | EXPECT_DOUBLE_EQ(0.0, b); |
| 3355 | EXPECT_DOUBLE_EQ(0.0, c); |
| 3356 | EXPECT_DOUBLE_EQ(51.0, d); |
| 3357 | EXPECT_DOUBLE_EQ(360.0, e); |
| 3358 | EXPECT_DOUBLE_EQ(646.510009765625, f); |
| 3359 | |
| 3360 | obj = FPDFPage_GetObject(page, 36); |
| 3361 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3362 | EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f)); |
| 3363 | EXPECT_DOUBLE_EQ(59.0, a); |
| 3364 | EXPECT_DOUBLE_EQ(0.0, b); |
| 3365 | EXPECT_DOUBLE_EQ(0.0, c); |
| 3366 | EXPECT_DOUBLE_EQ(45.0, d); |
| 3367 | EXPECT_DOUBLE_EQ(72.0, e); |
| 3368 | EXPECT_DOUBLE_EQ(553.510009765625, f); |
| 3369 | |
| 3370 | obj = FPDFPage_GetObject(page, 37); |
| 3371 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3372 | EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f)); |
| 3373 | EXPECT_DOUBLE_EQ(55.94000244140625, a); |
| 3374 | EXPECT_DOUBLE_EQ(0.0, b); |
| 3375 | EXPECT_DOUBLE_EQ(0.0, c); |
| 3376 | EXPECT_DOUBLE_EQ(46.950000762939453, d); |
| 3377 | EXPECT_DOUBLE_EQ(216.0, e); |
| 3378 | EXPECT_DOUBLE_EQ(552.510009765625, f); |
| 3379 | |
| 3380 | obj = FPDFPage_GetObject(page, 38); |
| 3381 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3382 | EXPECT_TRUE(FPDFImageObj_GetMatrix(obj, &a, &b, &c, &d, &e, &f)); |
| 3383 | EXPECT_DOUBLE_EQ(70.528999328613281, a); |
| 3384 | EXPECT_DOUBLE_EQ(0.0, b); |
| 3385 | EXPECT_DOUBLE_EQ(0.0, c); |
| 3386 | EXPECT_DOUBLE_EQ(43.149997711181641, d); |
| 3387 | EXPECT_DOUBLE_EQ(360.0, e); |
| 3388 | EXPECT_DOUBLE_EQ(553.3599853515625, f); |
| 3389 | |
| 3390 | UnloadPage(page); |
| 3391 | } |
| 3392 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3393 | TEST_F(FPDFEditEmbedderTest, DestroyPageObject) { |
Jane Liu | 2e5f0ae | 2017-08-08 15:23:27 -0400 | [diff] [blame] | 3394 | FPDF_PAGEOBJECT rect = FPDFPageObj_CreateNewRect(10, 10, 20, 20); |
| 3395 | ASSERT_TRUE(rect); |
| 3396 | |
| 3397 | // There should be no memory leaks with a call to FPDFPageObj_Destroy(). |
| 3398 | FPDFPageObj_Destroy(rect); |
| 3399 | } |
Jane Liu | be63ab9 | 2017-08-09 14:09:34 -0400 | [diff] [blame] | 3400 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3401 | TEST_F(FPDFEditEmbedderTest, GetImageFilters) { |
Jane Liu | be63ab9 | 2017-08-09 14:09:34 -0400 | [diff] [blame] | 3402 | EXPECT_TRUE(OpenDocument("embedded_images.pdf")); |
| 3403 | FPDF_PAGE page = LoadPage(0); |
| 3404 | ASSERT_TRUE(page); |
| 3405 | |
| 3406 | // Verify that retrieving the filter of a non-image object would fail. |
| 3407 | FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 32); |
| 3408 | ASSERT_NE(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3409 | ASSERT_EQ(0, FPDFImageObj_GetImageFilterCount(obj)); |
| 3410 | EXPECT_EQ(0u, FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0)); |
| 3411 | |
| 3412 | // Verify the returned filter string for an image object with a single filter. |
| 3413 | obj = FPDFPage_GetObject(page, 33); |
| 3414 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3415 | ASSERT_EQ(1, FPDFImageObj_GetImageFilterCount(obj)); |
| 3416 | unsigned long len = FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0); |
| 3417 | std::vector<char> buf(len); |
Lei Zhang | 0733a1b | 2017-08-31 12:36:31 -0700 | [diff] [blame] | 3418 | static constexpr char kFlateDecode[] = "FlateDecode"; |
| 3419 | EXPECT_EQ(sizeof(kFlateDecode), |
| 3420 | FPDFImageObj_GetImageFilter(obj, 0, buf.data(), len)); |
| 3421 | EXPECT_STREQ(kFlateDecode, buf.data()); |
Jane Liu | be63ab9 | 2017-08-09 14:09:34 -0400 | [diff] [blame] | 3422 | EXPECT_EQ(0u, FPDFImageObj_GetImageFilter(obj, 1, nullptr, 0)); |
| 3423 | |
| 3424 | // Verify all the filters for an image object with a list of filters. |
| 3425 | obj = FPDFPage_GetObject(page, 38); |
| 3426 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3427 | ASSERT_EQ(2, FPDFImageObj_GetImageFilterCount(obj)); |
| 3428 | len = FPDFImageObj_GetImageFilter(obj, 0, nullptr, 0); |
| 3429 | buf.clear(); |
| 3430 | buf.resize(len); |
Lei Zhang | 0733a1b | 2017-08-31 12:36:31 -0700 | [diff] [blame] | 3431 | static constexpr char kASCIIHexDecode[] = "ASCIIHexDecode"; |
| 3432 | EXPECT_EQ(sizeof(kASCIIHexDecode), |
| 3433 | FPDFImageObj_GetImageFilter(obj, 0, buf.data(), len)); |
| 3434 | EXPECT_STREQ(kASCIIHexDecode, buf.data()); |
Jane Liu | be63ab9 | 2017-08-09 14:09:34 -0400 | [diff] [blame] | 3435 | |
| 3436 | len = FPDFImageObj_GetImageFilter(obj, 1, nullptr, 0); |
| 3437 | buf.clear(); |
| 3438 | buf.resize(len); |
Lei Zhang | 0733a1b | 2017-08-31 12:36:31 -0700 | [diff] [blame] | 3439 | static constexpr char kDCTDecode[] = "DCTDecode"; |
| 3440 | EXPECT_EQ(sizeof(kDCTDecode), |
| 3441 | FPDFImageObj_GetImageFilter(obj, 1, buf.data(), len)); |
| 3442 | EXPECT_STREQ(kDCTDecode, buf.data()); |
Jane Liu | be63ab9 | 2017-08-09 14:09:34 -0400 | [diff] [blame] | 3443 | |
| 3444 | UnloadPage(page); |
| 3445 | } |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3446 | |
Lei Zhang | ab41f25 | 2018-12-23 03:10:50 +0000 | [diff] [blame] | 3447 | TEST_F(FPDFEditEmbedderTest, GetImageMetadata) { |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3448 | ASSERT_TRUE(OpenDocument("embedded_images.pdf")); |
| 3449 | FPDF_PAGE page = LoadPage(0); |
| 3450 | ASSERT_TRUE(page); |
| 3451 | |
| 3452 | // Check that getting the metadata of a null object would fail. |
| 3453 | FPDF_IMAGEOBJ_METADATA metadata; |
| 3454 | EXPECT_FALSE(FPDFImageObj_GetImageMetadata(nullptr, page, &metadata)); |
| 3455 | |
| 3456 | // Check that receiving the metadata with a null metadata object would fail. |
| 3457 | FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 35); |
| 3458 | EXPECT_FALSE(FPDFImageObj_GetImageMetadata(obj, page, nullptr)); |
| 3459 | |
| 3460 | // Check that when retrieving an image object's metadata without passing in |
| 3461 | // |page|, all values are correct, with the last two being default values. |
| 3462 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3463 | ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, nullptr, &metadata)); |
Julian Lunger | ecd063e | 2017-12-27 10:18:50 -0500 | [diff] [blame] | 3464 | EXPECT_EQ(7, metadata.marked_content_id); |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3465 | EXPECT_EQ(92u, metadata.width); |
| 3466 | EXPECT_EQ(68u, metadata.height); |
Lei Zhang | 351e8b0 | 2018-12-20 01:10:06 +0000 | [diff] [blame] | 3467 | EXPECT_FLOAT_EQ(96.0f, metadata.horizontal_dpi); |
| 3468 | EXPECT_FLOAT_EQ(96.0f, metadata.vertical_dpi); |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3469 | EXPECT_EQ(0u, metadata.bits_per_pixel); |
| 3470 | EXPECT_EQ(FPDF_COLORSPACE_UNKNOWN, metadata.colorspace); |
| 3471 | |
| 3472 | // Verify the metadata of a bitmap image with indexed colorspace. |
| 3473 | ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, page, &metadata)); |
Julian Lunger | ecd063e | 2017-12-27 10:18:50 -0500 | [diff] [blame] | 3474 | EXPECT_EQ(7, metadata.marked_content_id); |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3475 | EXPECT_EQ(92u, metadata.width); |
| 3476 | EXPECT_EQ(68u, metadata.height); |
Lei Zhang | 351e8b0 | 2018-12-20 01:10:06 +0000 | [diff] [blame] | 3477 | EXPECT_FLOAT_EQ(96.0f, metadata.horizontal_dpi); |
| 3478 | EXPECT_FLOAT_EQ(96.0f, metadata.vertical_dpi); |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3479 | EXPECT_EQ(1u, metadata.bits_per_pixel); |
| 3480 | EXPECT_EQ(FPDF_COLORSPACE_INDEXED, metadata.colorspace); |
| 3481 | |
| 3482 | // Verify the metadata of an image with RGB colorspace. |
| 3483 | obj = FPDFPage_GetObject(page, 37); |
| 3484 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3485 | ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, page, &metadata)); |
Julian Lunger | ecd063e | 2017-12-27 10:18:50 -0500 | [diff] [blame] | 3486 | EXPECT_EQ(9, metadata.marked_content_id); |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3487 | EXPECT_EQ(126u, metadata.width); |
| 3488 | EXPECT_EQ(106u, metadata.height); |
Lei Zhang | 351e8b0 | 2018-12-20 01:10:06 +0000 | [diff] [blame] | 3489 | EXPECT_FLOAT_EQ(162.173752f, metadata.horizontal_dpi); |
| 3490 | EXPECT_FLOAT_EQ(162.555878f, metadata.vertical_dpi); |
Jane Liu | ca89829 | 2017-08-16 11:25:35 -0400 | [diff] [blame] | 3491 | EXPECT_EQ(24u, metadata.bits_per_pixel); |
| 3492 | EXPECT_EQ(FPDF_COLORSPACE_DEVICERGB, metadata.colorspace); |
| 3493 | |
| 3494 | UnloadPage(page); |
| 3495 | } |
Lei Zhang | dc1b739 | 2020-05-14 21:15:53 +0000 | [diff] [blame] | 3496 | |
| 3497 | TEST_F(FPDFEditEmbedderTest, GetImageMetadataJpxLzw) { |
| 3498 | ASSERT_TRUE(OpenDocument("jpx_lzw.pdf")); |
| 3499 | FPDF_PAGE page = LoadPage(0); |
| 3500 | ASSERT_TRUE(page); |
| 3501 | |
| 3502 | FPDF_PAGEOBJECT obj = FPDFPage_GetObject(page, 0); |
| 3503 | ASSERT_EQ(FPDF_PAGEOBJ_IMAGE, FPDFPageObj_GetType(obj)); |
| 3504 | |
| 3505 | FPDF_IMAGEOBJ_METADATA metadata; |
| 3506 | ASSERT_TRUE(FPDFImageObj_GetImageMetadata(obj, page, &metadata)); |
| 3507 | EXPECT_EQ(-1, metadata.marked_content_id); |
| 3508 | EXPECT_EQ(612u, metadata.width); |
| 3509 | EXPECT_EQ(792u, metadata.height); |
| 3510 | EXPECT_FLOAT_EQ(72.0f, metadata.horizontal_dpi); |
| 3511 | EXPECT_FLOAT_EQ(72.0f, metadata.vertical_dpi); |
| 3512 | EXPECT_EQ(24u, metadata.bits_per_pixel); |
| 3513 | EXPECT_EQ(FPDF_COLORSPACE_UNKNOWN, metadata.colorspace); |
| 3514 | |
| 3515 | UnloadPage(page); |
| 3516 | } |