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. 4 5#include "printing/emf_win.h" 6 7// For quick access. 8#include <wingdi.h> 9#include <winspool.h> 10 11#include <string> 12 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" 24 25namespace printing { 26 27namespace { 28 29// This test is automatically disabled if no printer named "UnitTest Printer" is 30// available. 31class EmfPrintingTest : public testing::Test, public PrintingContext::Delegate { 32 public: 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); 37 if (!hdc) 38 return true; 39 DeleteDC(hdc); 40 return false; 41 } 42 43 // PrintingContext::Delegate methods. 44 virtual gfx::NativeView GetParentView() OVERRIDE { return NULL; } 45 virtual std::string GetAppLocale() OVERRIDE { return std::string(); } 46}; 47 48const uint32 EMF_HEADER_SIZE = 128; 49 50} // namespace 51 52TEST(EmfTest, DC) { 53 // Simplest use case. 54 uint32 size; 55 std::vector<char> data; 56 { 57 Emf emf; 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); 67 } 68 69 // Playback the data. 70 Emf emf; 71 EXPECT_TRUE(emf.InitFromData(&data.front(), size)); 72 HDC hdc = CreateCompatibleDC(NULL); 73 EXPECT_TRUE(hdc); 74 RECT output_rect = {0, 0, 10, 10}; 75 EXPECT_TRUE(emf.Playback(hdc, &output_rect)); 76 EXPECT_TRUE(DeleteDC(hdc)); 77} 78 79// Disabled if no "UnitTest printer" exist. Useful to reproduce bug 1186598. 80TEST_F(EmfPrintingTest, Enumerate) { 81 if (IsTestCaseDisabled()) 82 return; 83 84 PrintSettings settings; 85 86 // My test case is a HP Color LaserJet 4550 PCL. 87 settings.set_device_name(L"UnitTest Printer"); 88 89 // Initialize it. 90 scoped_ptr<PrintingContext> context(PrintingContext::Create(this)); 91 EXPECT_EQ(context->InitWithSettings(settings), PrintingContext::OK); 92 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. 100 Emf emf; 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())); 105 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"); 111 context->NewPage(); 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(); 117 ++itr) { 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 121 // with EMR_HEADER. 122 EMR_HEADER; 123 EXPECT_TRUE(itr->SafePlayback(&emf_enum.context_)) << 124 " index: " << index << " type: " << itr->record()->iType; 125 } 126 context->PageDone(); 127 context->DocumentDone(); 128} 129 130// Disabled if no "UnitTest printer" exists. 131TEST_F(EmfPrintingTest, PageBreak) { 132 base::win::ScopedCreateDC dc( 133 CreateDC(L"WINSPOOL", L"UnitTest Printer", NULL, NULL)); 134 if (!dc.Get()) 135 return; 136 uint32 size; 137 std::vector<char> data; 138 { 139 Emf emf; 140 EXPECT_TRUE(emf.Init()); 141 EXPECT_TRUE(emf.context() != NULL); 142 int pages = 3; 143 while (pages) { 144 EXPECT_TRUE(emf.StartPage(gfx::Size(), gfx::Rect(), 1)); 145 ::Rectangle(emf.context(), 10, 10, 190, 190); 146 EXPECT_TRUE(emf.FinishPage()); 147 --pages; 148 } 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); 154 } 155 156 // Playback the data. 157 DOCINFO di = {0}; 158 di.cbSize = sizeof(DOCINFO); 159 di.lpszDocName = L"Test Job"; 160 int job_id = ::StartDoc(dc.Get(), &di); 161 Emf emf; 162 EXPECT_TRUE(emf.InitFromData(&data.front(), size)); 163 EXPECT_TRUE(emf.SafePlayback(dc.Get())); 164 ::EndDoc(dc.Get()); 165 // Since presumably the printer is not real, let us just delete the job from 166 // the queue. 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); 171 } 172} 173 174TEST(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(), 180 &metafile_path)); 181 uint32 size; 182 std::vector<char> data; 183 { 184 Emf emf; 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); 194 } 195 int64 file_size = 0; 196 base::GetFileSize(metafile_path, &file_size); 197 EXPECT_EQ(size, file_size); 198 199 // Playback the data. 200 HDC hdc = CreateCompatibleDC(NULL); 201 EXPECT_TRUE(hdc); 202 Emf emf; 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)); 207} 208 209TEST(EmfTest, RasterizeMetafile) { 210 Emf emf; 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); 217 } 218 EXPECT_TRUE(emf.FinishDocument()); 219 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)); 223 224 raster = emf.RasterizeMetafile(20); 225 EXPECT_EQ(emf.GetPageBounds(1), raster->GetPageBounds(1)); 226 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)); 232} 233 234} // namespace printing 235