1 // Copyright (c) 2012 The Chromium 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.
5 #include "printing/emf_win.h"
13 #include "base/basictypes.h"
14 #include "base/files/file_path.h"
15 #include "base/files/file_util.h"
16 #include "base/files/scoped_temp_dir.h"
17 #include "base/memory/scoped_ptr.h"
18 #include "base/path_service.h"
19 #include "base/win/scoped_hdc.h"
20 #include "printing/printing_context.h"
21 #include "testing/gtest/include/gtest/gtest.h"
22 #include "ui/gfx/point.h"
23 #include "ui/gfx/size.h"
29 // This test is automatically disabled if no printer named "UnitTest Printer" is
31 class EmfPrintingTest : public testing::Test, public PrintingContext::Delegate {
33 typedef testing::Test Parent;
34 static bool IsTestCaseDisabled() {
35 // It is assumed this printer is a HP Color LaserJet 4550 PCL or 4700.
36 HDC hdc = CreateDC(L"WINSPOOL", L"UnitTest Printer", NULL, NULL);
43 // PrintingContext::Delegate methods.
44 virtual gfx::NativeView GetParentView() OVERRIDE { return NULL; }
45 virtual std::string GetAppLocale() OVERRIDE { return std::string(); }
48 const uint32 EMF_HEADER_SIZE = 128;
55 std::vector<char> data;
58 EXPECT_TRUE(emf.Init());
59 EXPECT_TRUE(emf.context() != NULL);
60 // An empty EMF is invalid, so we put at least a rectangle in it.
61 ::Rectangle(emf.context(), 10, 10, 190, 190);
62 EXPECT_TRUE(emf.FinishDocument());
63 size = emf.GetDataSize();
64 EXPECT_GT(size, EMF_HEADER_SIZE);
65 EXPECT_TRUE(emf.GetDataAsVector(&data));
66 EXPECT_EQ(data.size(), size);
71 EXPECT_TRUE(emf.InitFromData(&data.front(), size));
72 HDC hdc = CreateCompatibleDC(NULL);
74 RECT output_rect = {0, 0, 10, 10};
75 EXPECT_TRUE(emf.Playback(hdc, &output_rect));
76 EXPECT_TRUE(DeleteDC(hdc));
79 // Disabled if no "UnitTest printer" exist. Useful to reproduce bug 1186598.
80 TEST_F(EmfPrintingTest, Enumerate) {
81 if (IsTestCaseDisabled())
84 PrintSettings settings;
86 // My test case is a HP Color LaserJet 4550 PCL.
87 settings.set_device_name(L"UnitTest Printer");
90 scoped_ptr<PrintingContext> context(PrintingContext::Create(this));
91 EXPECT_EQ(context->InitWithSettings(settings), PrintingContext::OK);
93 base::FilePath emf_file;
94 EXPECT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &emf_file));
95 emf_file = emf_file.Append(FILE_PATH_LITERAL("printing"))
96 .Append(FILE_PATH_LITERAL("test"))
97 .Append(FILE_PATH_LITERAL("data"))
98 .Append(FILE_PATH_LITERAL("test4.emf"));
99 // Load any EMF with an image.
101 std::string emf_data;
102 base::ReadFileToString(emf_file, &emf_data);
103 ASSERT_TRUE(emf_data.size());
104 EXPECT_TRUE(emf.InitFromData(&emf_data[0], emf_data.size()));
106 // This will print to file. The reason is that when running inside a
107 // unit_test, PrintingContext automatically dumps its files to the
108 // current directory.
109 // TODO(maruel): Clean the .PRN file generated in current directory.
110 context->NewDocument(L"EmfTest.Enumerate");
112 // Process one at a time.
113 RECT page_bounds = emf.GetPageBounds(1).ToRECT();
114 Emf::Enumerator emf_enum(emf, context->context(), &page_bounds);
115 for (Emf::Enumerator::const_iterator itr = emf_enum.begin();
116 itr != emf_enum.end();
118 // To help debugging.
119 ptrdiff_t index = itr - emf_enum.begin();
120 // If you get this assert, you need to lookup iType in wingdi.h. It starts
123 EXPECT_TRUE(itr->SafePlayback(&emf_enum.context_)) <<
124 " index: " << index << " type: " << itr->record()->iType;
127 context->DocumentDone();
130 // Disabled if no "UnitTest printer" exists.
131 TEST_F(EmfPrintingTest, PageBreak) {
132 base::win::ScopedCreateDC dc(
133 CreateDC(L"WINSPOOL", L"UnitTest Printer", NULL, NULL));
137 std::vector<char> data;
140 EXPECT_TRUE(emf.Init());
141 EXPECT_TRUE(emf.context() != NULL);
144 EXPECT_TRUE(emf.StartPage(gfx::Size(), gfx::Rect(), 1));
145 ::Rectangle(emf.context(), 10, 10, 190, 190);
146 EXPECT_TRUE(emf.FinishPage());
149 EXPECT_EQ(3U, emf.GetPageCount());
150 EXPECT_TRUE(emf.FinishDocument());
151 size = emf.GetDataSize();
152 EXPECT_TRUE(emf.GetDataAsVector(&data));
153 EXPECT_EQ(data.size(), size);
156 // Playback the data.
158 di.cbSize = sizeof(DOCINFO);
159 di.lpszDocName = L"Test Job";
160 int job_id = ::StartDoc(dc.Get(), &di);
162 EXPECT_TRUE(emf.InitFromData(&data.front(), size));
163 EXPECT_TRUE(emf.SafePlayback(dc.Get()));
165 // Since presumably the printer is not real, let us just delete the job from
167 HANDLE printer = NULL;
168 if (::OpenPrinter(const_cast<LPTSTR>(L"UnitTest Printer"), &printer, NULL)) {
169 ::SetJob(printer, job_id, 0, NULL, JOB_CONTROL_DELETE);
170 ClosePrinter(printer);
174 TEST(EmfTest, FileBackedEmf) {
175 // Simplest use case.
176 base::ScopedTempDir scratch_metafile_dir;
177 ASSERT_TRUE(scratch_metafile_dir.CreateUniqueTempDir());
178 base::FilePath metafile_path;
179 EXPECT_TRUE(base::CreateTemporaryFileInDir(scratch_metafile_dir.path(),
182 std::vector<char> data;
185 EXPECT_TRUE(emf.InitToFile(metafile_path));
186 EXPECT_TRUE(emf.context() != NULL);
187 // An empty EMF is invalid, so we put at least a rectangle in it.
188 ::Rectangle(emf.context(), 10, 10, 190, 190);
189 EXPECT_TRUE(emf.FinishDocument());
190 size = emf.GetDataSize();
191 EXPECT_GT(size, EMF_HEADER_SIZE);
192 EXPECT_TRUE(emf.GetDataAsVector(&data));
193 EXPECT_EQ(data.size(), size);
196 base::GetFileSize(metafile_path, &file_size);
197 EXPECT_EQ(size, file_size);
199 // Playback the data.
200 HDC hdc = CreateCompatibleDC(NULL);
203 EXPECT_TRUE(emf.InitFromFile(metafile_path));
204 RECT output_rect = {0, 0, 10, 10};
205 EXPECT_TRUE(emf.Playback(hdc, &output_rect));
206 EXPECT_TRUE(DeleteDC(hdc));
209 TEST(EmfTest, RasterizeMetafile) {
211 EXPECT_TRUE(emf.Init());
212 EXPECT_TRUE(emf.context() != NULL);
213 HBRUSH brush = static_cast<HBRUSH>(GetStockObject(BLACK_BRUSH));
214 for (int i = 0; i < 4; ++i) {
215 RECT rect = { 5 + i, 5 + i, 5 + i + 1, 5 + i + 2};
216 FillRect(emf.context(), &rect, brush);
218 EXPECT_TRUE(emf.FinishDocument());
220 scoped_ptr<Emf> raster(emf.RasterizeMetafile(1));
221 // Just 1px bitmap but should be stretched to the same bounds.
222 EXPECT_EQ(emf.GetPageBounds(1), raster->GetPageBounds(1));
224 raster = emf.RasterizeMetafile(20);
225 EXPECT_EQ(emf.GetPageBounds(1), raster->GetPageBounds(1));
227 raster = emf.RasterizeMetafile(16 * 1024 * 1024);
228 // Expected size about 64MB.
229 EXPECT_LE(abs(int(raster->GetDataSize()) - 64 * 1024 * 1024), 1024 * 1024);
230 // Bounds should still be the same.
231 EXPECT_EQ(emf.GetPageBounds(1), raster->GetPageBounds(1));
234 } // namespace printing