OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "printing/emf_win.h" | 5 #include "printing/emf_win.h" |
6 | 6 |
7 #include "base/file_path.h" | 7 #include "base/file_path.h" |
8 #include "base/logging.h" | 8 #include "base/logging.h" |
9 #include "base/metrics/histogram.h" | 9 #include "base/metrics/histogram.h" |
10 #include "base/scoped_ptr.h" | 10 #include "base/scoped_ptr.h" |
(...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
43 ExtEscape(dc, escape, size, reinterpret_cast<LPCSTR>(bits), | 43 ExtEscape(dc, escape, size, reinterpret_cast<LPCSTR>(bits), |
44 sizeof(supported), reinterpret_cast<LPSTR>(&supported)); | 44 sizeof(supported), reinterpret_cast<LPSTR>(&supported)); |
45 } | 45 } |
46 return !!supported; | 46 return !!supported; |
47 } | 47 } |
48 | 48 |
49 Emf::Emf() : emf_(NULL), hdc_(NULL) { | 49 Emf::Emf() : emf_(NULL), hdc_(NULL) { |
50 } | 50 } |
51 | 51 |
52 Emf::~Emf() { | 52 Emf::~Emf() { |
53 CloseEmf(); | 53 Close(); |
54 DCHECK(!emf_ && !hdc_); | 54 DCHECK(!emf_ && !hdc_); |
55 } | 55 } |
56 | 56 |
57 bool Emf::Init(const void* src_buffer, uint32 src_buffer_size) { | 57 bool Emf::Init(const void* src_buffer, uint32 src_buffer_size) { |
58 DCHECK(!emf_ && !hdc_); | 58 DCHECK(!emf_ && !hdc_); |
59 emf_ = SetEnhMetaFileBits(src_buffer_size, | 59 emf_ = SetEnhMetaFileBits(src_buffer_size, |
60 reinterpret_cast<const BYTE*>(src_buffer)); | 60 reinterpret_cast<const BYTE*>(src_buffer)); |
61 return emf_ != NULL; | 61 return emf_ != NULL; |
62 } | 62 } |
63 | 63 |
| 64 uint32 Emf::GetDataSize() const { |
| 65 DCHECK(emf_ && !hdc_); |
| 66 return GetEnhMetaFileBits(emf_, 0, NULL); |
| 67 } |
| 68 |
| 69 bool Emf::GetData(void* buffer, uint32 size) const { |
| 70 DCHECK(emf_ && !hdc_); |
| 71 DCHECK(buffer && size); |
| 72 uint32 size2 = |
| 73 GetEnhMetaFileBits(emf_, size, reinterpret_cast<BYTE*>(buffer)); |
| 74 DCHECK(size2 == size); |
| 75 return size2 == size && size2 != 0; |
| 76 } |
| 77 |
| 78 bool Emf::FinishPage() { |
| 79 DCHECK(hdc_); |
| 80 if (!hdc_) |
| 81 return false; |
| 82 PageBreakRecord record(PageBreakRecord::END_PAGE); |
| 83 return !!GdiComment(hdc_, sizeof(record), |
| 84 reinterpret_cast<const BYTE *>(&record)); |
| 85 } |
| 86 |
| 87 void Emf::Close() { |
| 88 DCHECK(!hdc_); |
| 89 if (emf_) { |
| 90 DeleteEnhMetaFile(emf_); |
| 91 emf_ = NULL; |
| 92 } |
| 93 } |
| 94 |
| 95 bool Emf::SaveTo(const FilePath& file_path) const { |
| 96 HANDLE file = CreateFile(file_path.value().c_str(), GENERIC_WRITE, |
| 97 FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, |
| 98 CREATE_ALWAYS, 0, NULL); |
| 99 if (file == INVALID_HANDLE_VALUE) |
| 100 return false; |
| 101 |
| 102 bool success = false; |
| 103 std::vector<uint8> buffer; |
| 104 if (GetData(&buffer)) { |
| 105 DWORD written = 0; |
| 106 if (WriteFile(file, &*buffer.begin(), static_cast<DWORD>(buffer.size()), |
| 107 &written, NULL) && |
| 108 written == buffer.size()) { |
| 109 success = true; |
| 110 } |
| 111 } |
| 112 CloseHandle(file); |
| 113 return success; |
| 114 } |
| 115 |
| 116 gfx::Rect Emf::GetPageBounds(unsigned int page_number) const { |
| 117 DCHECK(emf_ && !hdc_); |
| 118 ENHMETAHEADER header; |
| 119 if (GetEnhMetaFileHeader(emf_, sizeof(header), &header) != sizeof(header)) { |
| 120 NOTREACHED(); |
| 121 return gfx::Rect(); |
| 122 } |
| 123 if (header.rclBounds.left == 0 && |
| 124 header.rclBounds.top == 0 && |
| 125 header.rclBounds.right == -1 && |
| 126 header.rclBounds.bottom == -1) { |
| 127 // A freshly created EMF buffer that has no drawing operation has invalid |
| 128 // bounds. Instead of having an (0,0) size, it has a (-1,-1) size. Detect |
| 129 // this special case and returns an empty Rect instead of an invalid one. |
| 130 return gfx::Rect(); |
| 131 } |
| 132 return gfx::Rect(header.rclBounds.left, |
| 133 header.rclBounds.top, |
| 134 header.rclBounds.right - header.rclBounds.left, |
| 135 header.rclBounds.bottom - header.rclBounds.top); |
| 136 } |
| 137 |
| 138 bool Emf::StartPage() { |
| 139 DCHECK(hdc_); |
| 140 if (!hdc_) |
| 141 return false; |
| 142 PageBreakRecord record(PageBreakRecord::START_PAGE); |
| 143 return !!GdiComment(hdc_, sizeof(record), |
| 144 reinterpret_cast<const BYTE *>(&record)); |
| 145 } |
| 146 |
64 bool Emf::CreateDc(HDC sibling, const RECT* rect) { | 147 bool Emf::CreateDc(HDC sibling, const RECT* rect) { |
65 DCHECK(!emf_ && !hdc_); | 148 DCHECK(!emf_ && !hdc_); |
66 hdc_ = CreateEnhMetaFile(sibling, NULL, rect, NULL); | 149 hdc_ = CreateEnhMetaFile(sibling, NULL, rect, NULL); |
67 DCHECK(hdc_); | 150 DCHECK(hdc_); |
68 return hdc_ != NULL; | 151 return hdc_ != NULL; |
69 } | 152 } |
70 | 153 |
71 bool Emf::CreateFileBackedDc(HDC sibling, const RECT* rect, | 154 bool Emf::CreateFileBackedDc(HDC sibling, |
| 155 const RECT* rect, |
72 const FilePath& path) { | 156 const FilePath& path) { |
73 DCHECK(!emf_ && !hdc_); | 157 DCHECK(!emf_ && !hdc_); |
74 DCHECK(!path.empty()); | 158 DCHECK(!path.empty()); |
75 hdc_ = CreateEnhMetaFile(sibling, path.value().c_str(), rect, NULL); | 159 hdc_ = CreateEnhMetaFile(sibling, path.value().c_str(), rect, NULL); |
76 DCHECK(hdc_); | 160 DCHECK(hdc_); |
77 return hdc_ != NULL; | 161 return hdc_ != NULL; |
78 } | 162 } |
79 | 163 |
80 bool Emf::CreateFromFile(const FilePath& metafile_path) { | 164 bool Emf::CreateFromFile(const FilePath& metafile_path) { |
81 DCHECK(!emf_ && !hdc_); | 165 DCHECK(!emf_ && !hdc_); |
82 emf_ = GetEnhMetaFile(metafile_path.value().c_str()); | 166 emf_ = GetEnhMetaFile(metafile_path.value().c_str()); |
83 DCHECK(emf_); | 167 DCHECK(emf_); |
84 return emf_ != NULL; | 168 return emf_ != NULL; |
85 } | 169 } |
86 | 170 |
87 | 171 |
88 bool Emf::CloseDc() { | 172 bool Emf::CloseDc() { |
89 DCHECK(!emf_ && hdc_); | 173 DCHECK(!emf_ && hdc_); |
90 emf_ = CloseEnhMetaFile(hdc_); | 174 emf_ = CloseEnhMetaFile(hdc_); |
91 DCHECK(emf_); | 175 DCHECK(emf_); |
92 hdc_ = NULL; | 176 hdc_ = NULL; |
93 return emf_ != NULL; | 177 return emf_ != NULL; |
94 } | 178 } |
95 | 179 |
96 void Emf::CloseEmf() { | |
97 DCHECK(!hdc_); | |
98 if (emf_) { | |
99 DeleteEnhMetaFile(emf_); | |
100 emf_ = NULL; | |
101 } | |
102 } | |
103 | |
104 bool Emf::Playback(HDC hdc, const RECT* rect) const { | 180 bool Emf::Playback(HDC hdc, const RECT* rect) const { |
105 DCHECK(emf_ && !hdc_); | 181 DCHECK(emf_ && !hdc_); |
106 RECT bounds; | 182 RECT bounds; |
107 if (!rect) { | 183 if (!rect) { |
108 // Get the natural bounds of the EMF buffer. | 184 // Get the natural bounds of the EMF buffer. |
109 bounds = GetBounds().ToRECT(); | 185 bounds = GetPageBounds(1).ToRECT(); |
110 rect = &bounds; | 186 rect = &bounds; |
111 } | 187 } |
112 return PlayEnhMetaFile(hdc, emf_, rect) != 0; | 188 return PlayEnhMetaFile(hdc, emf_, rect) != 0; |
113 } | 189 } |
114 | 190 |
115 bool Emf::SafePlayback(HDC context) const { | 191 bool Emf::SafePlayback(HDC context) const { |
116 DCHECK(emf_ && !hdc_); | 192 DCHECK(emf_ && !hdc_); |
117 XFORM base_matrix; | 193 XFORM base_matrix; |
118 if (!GetWorldTransform(context, &base_matrix)) { | 194 if (!GetWorldTransform(context, &base_matrix)) { |
119 NOTREACHED(); | 195 NOTREACHED(); |
120 return false; | 196 return false; |
121 } | 197 } |
122 return EnumEnhMetaFile(context, | 198 return EnumEnhMetaFile(context, |
123 emf_, | 199 emf_, |
124 &Emf::SafePlaybackProc, | 200 &Emf::SafePlaybackProc, |
125 reinterpret_cast<void*>(&base_matrix), | 201 reinterpret_cast<void*>(&base_matrix), |
126 &GetBounds().ToRECT()) != 0; | 202 &GetPageBounds(1).ToRECT()) != 0; |
127 } | |
128 | |
129 gfx::Rect Emf::GetBounds() const { | |
130 DCHECK(emf_ && !hdc_); | |
131 ENHMETAHEADER header; | |
132 if (GetEnhMetaFileHeader(emf_, sizeof(header), &header) != sizeof(header)) { | |
133 NOTREACHED(); | |
134 return gfx::Rect(); | |
135 } | |
136 if (header.rclBounds.left == 0 && | |
137 header.rclBounds.top == 0 && | |
138 header.rclBounds.right == -1 && | |
139 header.rclBounds.bottom == -1) { | |
140 // A freshly created EMF buffer that has no drawing operation has invalid | |
141 // bounds. Instead of having an (0,0) size, it has a (-1,-1) size. Detect | |
142 // this special case and returns an empty Rect instead of an invalid one. | |
143 return gfx::Rect(); | |
144 } | |
145 return gfx::Rect(header.rclBounds.left, | |
146 header.rclBounds.top, | |
147 header.rclBounds.right - header.rclBounds.left, | |
148 header.rclBounds.bottom - header.rclBounds.top); | |
149 } | |
150 | |
151 uint32 Emf::GetDataSize() const { | |
152 DCHECK(emf_ && !hdc_); | |
153 return GetEnhMetaFileBits(emf_, 0, NULL); | |
154 } | |
155 | |
156 bool Emf::GetData(void* buffer, uint32 size) const { | |
157 DCHECK(emf_ && !hdc_); | |
158 DCHECK(buffer && size); | |
159 uint32 size2 = | |
160 GetEnhMetaFileBits(emf_, size, reinterpret_cast<BYTE*>(buffer)); | |
161 DCHECK(size2 == size); | |
162 return size2 == size && size2 != 0; | |
163 } | 203 } |
164 | 204 |
165 bool Emf::GetData(std::vector<uint8>* buffer) const { | 205 bool Emf::GetData(std::vector<uint8>* buffer) const { |
166 uint32 size = GetDataSize(); | 206 uint32 size = GetDataSize(); |
167 if (!size) | 207 if (!size) |
168 return false; | 208 return false; |
169 | 209 |
170 buffer->resize(size); | 210 buffer->resize(size); |
171 if (!GetData(&buffer->front(), size)) | 211 if (!GetData(&buffer->front(), size)) |
172 return false; | 212 return false; |
173 return true; | 213 return true; |
174 } | 214 } |
175 | 215 |
176 bool Emf::SaveTo(const std::wstring& filename) const { | |
177 HANDLE file = CreateFile(filename.c_str(), GENERIC_WRITE, | |
178 FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, | |
179 CREATE_ALWAYS, 0, NULL); | |
180 if (file == INVALID_HANDLE_VALUE) | |
181 return false; | |
182 | |
183 bool success = false; | |
184 std::vector<uint8> buffer; | |
185 if (GetData(&buffer)) { | |
186 DWORD written = 0; | |
187 if (WriteFile(file, &*buffer.begin(), static_cast<DWORD>(buffer.size()), | |
188 &written, NULL) && | |
189 written == buffer.size()) { | |
190 success = true; | |
191 } | |
192 } | |
193 CloseHandle(file); | |
194 return success; | |
195 } | |
196 | |
197 int CALLBACK Emf::SafePlaybackProc(HDC hdc, | 216 int CALLBACK Emf::SafePlaybackProc(HDC hdc, |
198 HANDLETABLE* handle_table, | 217 HANDLETABLE* handle_table, |
199 const ENHMETARECORD* record, | 218 const ENHMETARECORD* record, |
200 int objects_count, | 219 int objects_count, |
201 LPARAM param) { | 220 LPARAM param) { |
202 const XFORM* base_matrix = reinterpret_cast<const XFORM*>(param); | 221 const XFORM* base_matrix = reinterpret_cast<const XFORM*>(param); |
203 EnumerationContext context; | 222 EnumerationContext context; |
204 context.handle_table = handle_table; | 223 context.handle_table = handle_table; |
205 context.objects_count = objects_count; | 224 context.objects_count = objects_count; |
206 context.hdc = hdc; | 225 context.hdc = hdc; |
(...skipping 195 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
402 break; | 421 break; |
403 } | 422 } |
404 default: { | 423 default: { |
405 res = Play(); | 424 res = Play(); |
406 break; | 425 break; |
407 } | 426 } |
408 } | 427 } |
409 return res; | 428 return res; |
410 } | 429 } |
411 | 430 |
412 bool Emf::StartPage() { | |
413 DCHECK(hdc_); | |
414 if (!hdc_) | |
415 return false; | |
416 PageBreakRecord record(PageBreakRecord::START_PAGE); | |
417 return !!GdiComment(hdc_, sizeof(record), | |
418 reinterpret_cast<const BYTE *>(&record)); | |
419 } | |
420 | |
421 bool Emf::EndPage() { | |
422 DCHECK(hdc_); | |
423 if (!hdc_) | |
424 return false; | |
425 PageBreakRecord record(PageBreakRecord::END_PAGE); | |
426 return !!GdiComment(hdc_, sizeof(record), | |
427 reinterpret_cast<const BYTE *>(&record)); | |
428 } | |
429 | |
430 | |
431 Emf::Enumerator::Enumerator(const Emf& emf, HDC context, const RECT* rect) { | 431 Emf::Enumerator::Enumerator(const Emf& emf, HDC context, const RECT* rect) { |
432 context_.handle_table = NULL; | 432 context_.handle_table = NULL; |
433 context_.objects_count = 0; | 433 context_.objects_count = 0; |
434 context_.hdc = NULL; | 434 context_.hdc = NULL; |
435 items_.clear(); | 435 items_.clear(); |
436 if (!EnumEnhMetaFile(context, | 436 if (!EnumEnhMetaFile(context, |
437 emf.emf(), | 437 emf.emf(), |
438 &Emf::Enumerator::EnhMetaFileProc, | 438 &Emf::Enumerator::EnhMetaFileProc, |
439 reinterpret_cast<void*>(this), | 439 reinterpret_cast<void*>(this), |
440 rect)) { | 440 rect)) { |
(...skipping 26 matching lines...) Expand all Loading... |
467 } else { | 467 } else { |
468 DCHECK_EQ(emf.context_.handle_table, handle_table); | 468 DCHECK_EQ(emf.context_.handle_table, handle_table); |
469 DCHECK_EQ(emf.context_.objects_count, objects_count); | 469 DCHECK_EQ(emf.context_.objects_count, objects_count); |
470 DCHECK_EQ(emf.context_.hdc, hdc); | 470 DCHECK_EQ(emf.context_.hdc, hdc); |
471 } | 471 } |
472 emf.items_.push_back(Record(&emf.context_, record)); | 472 emf.items_.push_back(Record(&emf.context_, record)); |
473 return 1; | 473 return 1; |
474 } | 474 } |
475 | 475 |
476 } // namespace printing | 476 } // namespace printing |
OLD | NEW |