| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2006-2008 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 "chrome/common/gfx/emf.h" | |
| 6 | |
| 7 #include "base/gfx/rect.h" | |
| 8 #include "base/logging.h" | |
| 9 | |
| 10 namespace gfx { | |
| 11 | |
| 12 Emf::Emf() : emf_(NULL), hdc_(NULL) { | |
| 13 } | |
| 14 | |
| 15 Emf::~Emf() { | |
| 16 CloseEmf(); | |
| 17 DCHECK(!emf_ && !hdc_); | |
| 18 } | |
| 19 | |
| 20 bool Emf::CreateDc(HDC sibling, const RECT* rect) { | |
| 21 DCHECK(!emf_ && !hdc_); | |
| 22 hdc_ = CreateEnhMetaFile(sibling, NULL, rect, NULL); | |
| 23 DCHECK(hdc_); | |
| 24 return hdc_ != NULL; | |
| 25 } | |
| 26 | |
| 27 bool Emf::CreateFromData(const void* buffer, size_t size) { | |
| 28 DCHECK(!emf_ && !hdc_); | |
| 29 emf_ = SetEnhMetaFileBits(static_cast<unsigned>(size), | |
| 30 reinterpret_cast<const BYTE*>(buffer)); | |
| 31 DCHECK(emf_); | |
| 32 return emf_ != NULL; | |
| 33 } | |
| 34 | |
| 35 bool Emf::CloseDc() { | |
| 36 DCHECK(!emf_ && hdc_); | |
| 37 emf_ = CloseEnhMetaFile(hdc_); | |
| 38 DCHECK(emf_); | |
| 39 hdc_ = NULL; | |
| 40 return emf_ != NULL; | |
| 41 } | |
| 42 | |
| 43 void Emf::CloseEmf() { | |
| 44 DCHECK(!hdc_); | |
| 45 if (emf_) { | |
| 46 DeleteEnhMetaFile(emf_); | |
| 47 emf_ = NULL; | |
| 48 } | |
| 49 } | |
| 50 | |
| 51 bool Emf::Playback(HDC hdc, const RECT* rect) const { | |
| 52 DCHECK(emf_ && !hdc_); | |
| 53 RECT bounds; | |
| 54 if (!rect) { | |
| 55 // Get the natural bounds of the EMF buffer. | |
| 56 bounds = GetBounds().ToRECT(); | |
| 57 rect = &bounds; | |
| 58 } | |
| 59 return PlayEnhMetaFile(hdc, emf_, rect) != 0; | |
| 60 } | |
| 61 | |
| 62 bool Emf::SafePlayback(HDC context) const { | |
| 63 DCHECK(emf_ && !hdc_); | |
| 64 XFORM base_matrix; | |
| 65 if (!GetWorldTransform(context, &base_matrix)) { | |
| 66 NOTREACHED(); | |
| 67 return false; | |
| 68 } | |
| 69 | |
| 70 return EnumEnhMetaFile(context, | |
| 71 emf_, | |
| 72 &Emf::SafePlaybackProc, | |
| 73 reinterpret_cast<void*>(&base_matrix), | |
| 74 &GetBounds().ToRECT()) != 0; | |
| 75 } | |
| 76 | |
| 77 gfx::Rect Emf::GetBounds() const { | |
| 78 DCHECK(emf_ && !hdc_); | |
| 79 ENHMETAHEADER header; | |
| 80 if (GetEnhMetaFileHeader(emf_, sizeof(header), &header) != sizeof(header)) { | |
| 81 NOTREACHED(); | |
| 82 return gfx::Rect(); | |
| 83 } | |
| 84 if (header.rclBounds.left == 0 && | |
| 85 header.rclBounds.top == 0 && | |
| 86 header.rclBounds.right == -1 && | |
| 87 header.rclBounds.bottom == -1) { | |
| 88 // A freshly created EMF buffer that has no drawing operation has invalid | |
| 89 // bounds. Instead of having an (0,0) size, it has a (-1,-1) size. Detect | |
| 90 // this special case and returns an empty Rect instead of an invalid one. | |
| 91 return gfx::Rect(); | |
| 92 } | |
| 93 return gfx::Rect(header.rclBounds.left, | |
| 94 header.rclBounds.top, | |
| 95 header.rclBounds.right - header.rclBounds.left, | |
| 96 header.rclBounds.bottom - header.rclBounds.top); | |
| 97 } | |
| 98 | |
| 99 unsigned Emf::GetDataSize() const { | |
| 100 DCHECK(emf_ && !hdc_); | |
| 101 return GetEnhMetaFileBits(emf_, 0, NULL); | |
| 102 } | |
| 103 | |
| 104 bool Emf::GetData(void* buffer, size_t size) const { | |
| 105 DCHECK(emf_ && !hdc_); | |
| 106 DCHECK(buffer && size); | |
| 107 unsigned size2 = GetEnhMetaFileBits(emf_, static_cast<unsigned>(size), | |
| 108 reinterpret_cast<BYTE*>(buffer)); | |
| 109 DCHECK(size2 == size); | |
| 110 return size2 == size && size2 != 0; | |
| 111 } | |
| 112 | |
| 113 bool Emf::GetData(std::vector<uint8>* buffer) const { | |
| 114 unsigned size = GetDataSize(); | |
| 115 if (!size) | |
| 116 return false; | |
| 117 | |
| 118 buffer->resize(size); | |
| 119 if (!GetData(&buffer->front(), size)) | |
| 120 return false; | |
| 121 return true; | |
| 122 } | |
| 123 | |
| 124 bool Emf::SaveTo(const std::wstring& filename) const { | |
| 125 HANDLE file = CreateFile(filename.c_str(), GENERIC_WRITE, | |
| 126 FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, | |
| 127 CREATE_ALWAYS, 0, NULL); | |
| 128 if (file == INVALID_HANDLE_VALUE) | |
| 129 return false; | |
| 130 | |
| 131 bool success = false; | |
| 132 std::vector<uint8> buffer; | |
| 133 if (GetData(&buffer)) { | |
| 134 DWORD written = 0; | |
| 135 if (WriteFile(file, &*buffer.begin(), static_cast<DWORD>(buffer.size()), | |
| 136 &written, NULL) && | |
| 137 written == buffer.size()) { | |
| 138 success = true; | |
| 139 } | |
| 140 } | |
| 141 CloseHandle(file); | |
| 142 return success; | |
| 143 } | |
| 144 | |
| 145 int CALLBACK Emf::SafePlaybackProc(HDC hdc, | |
| 146 HANDLETABLE* handle_table, | |
| 147 const ENHMETARECORD* record, | |
| 148 int objects_count, | |
| 149 LPARAM param) { | |
| 150 const XFORM* base_matrix = reinterpret_cast<const XFORM*>(param); | |
| 151 EnumerationContext context; | |
| 152 context.handle_table = handle_table; | |
| 153 context.objects_count = objects_count; | |
| 154 context.hdc = hdc; | |
| 155 Record record_instance(&context, record); | |
| 156 bool success = record_instance.SafePlayback(base_matrix); | |
| 157 DCHECK(success); | |
| 158 return 1; | |
| 159 } | |
| 160 | |
| 161 Emf::Record::Record() { | |
| 162 } | |
| 163 | |
| 164 Emf::Record::Record(const EnumerationContext* context, | |
| 165 const ENHMETARECORD* record) | |
| 166 : record_(record), | |
| 167 context_(context) { | |
| 168 DCHECK(record_); | |
| 169 } | |
| 170 | |
| 171 bool Emf::Record::Play() const { | |
| 172 return 0 != PlayEnhMetaFileRecord(context_->hdc, | |
| 173 context_->handle_table, | |
| 174 record_, | |
| 175 context_->objects_count); | |
| 176 } | |
| 177 | |
| 178 bool Emf::Record::SafePlayback(const XFORM* base_matrix) const { | |
| 179 // For EMF field description, see [MS-EMF] Enhanced Metafile Format | |
| 180 // Specification. | |
| 181 // | |
| 182 // This is the second major EMF breakage I get; the first one being | |
| 183 // SetDCBrushColor/SetDCPenColor/DC_PEN/DC_BRUSH being silently ignored. | |
| 184 // | |
| 185 // This function is the guts of the fix for bug 1186598. Some printer drivers | |
| 186 // somehow choke on certain EMF records, but calling the corresponding | |
| 187 // function directly on the printer HDC is fine. Still, playing the EMF record | |
| 188 // fails. Go figure. | |
| 189 // | |
| 190 // The main issue is that SetLayout is totally unsupported on these printers | |
| 191 // (HP 4500/4700). I used to call SetLayout and I stopped. I found out this is | |
| 192 // not sufficient because GDI32!PlayEnhMetaFile internally calls SetLayout(!) | |
| 193 // Damn. | |
| 194 // | |
| 195 // So I resorted to manually parse the EMF records and play them one by one. | |
| 196 // The issue with this method compared to using PlayEnhMetaFile to play back | |
| 197 // an EMF buffer is that the later silently fixes the matrix to take in | |
| 198 // account the matrix currently loaded at the time of the call. | |
| 199 // The matrix magic is done transparently when using PlayEnhMetaFile but since | |
| 200 // I'm processing one field at a time, I need to do the fixup myself. Note | |
| 201 // that PlayEnhMetaFileRecord doesn't fix the matrix correctly even when | |
| 202 // called inside an EnumEnhMetaFile loop. Go figure (bis). | |
| 203 // | |
| 204 // So when I see a EMR_SETWORLDTRANSFORM and EMR_MODIFYWORLDTRANSFORM, I need | |
| 205 // to fix the matrix according to the matrix previously loaded before playing | |
| 206 // back the buffer. Otherwise, the previously loaded matrix would be ignored | |
| 207 // and the EMF buffer would always be played back at its native resolution. | |
| 208 // Duh. | |
| 209 // | |
| 210 // I also use this opportunity to skip over eventual EMR_SETLAYOUT record that | |
| 211 // could remain. | |
| 212 // | |
| 213 // Note: I should probably care about view ports and clipping, eventually. | |
| 214 bool res; | |
| 215 switch (record()->iType) { | |
| 216 case EMR_SETWORLDTRANSFORM: { | |
| 217 DCHECK_EQ(record()->nSize, sizeof(DWORD) * 2 + sizeof(XFORM)); | |
| 218 const XFORM* xform = reinterpret_cast<const XFORM*>(record()->dParm); | |
| 219 HDC hdc = context_->hdc; | |
| 220 if (base_matrix) { | |
| 221 res = 0 != SetWorldTransform(hdc, base_matrix) && | |
| 222 ModifyWorldTransform(hdc, xform, MWT_LEFTMULTIPLY); | |
| 223 } else { | |
| 224 res = 0 != SetWorldTransform(hdc, xform); | |
| 225 } | |
| 226 break; | |
| 227 } | |
| 228 case EMR_MODIFYWORLDTRANSFORM: { | |
| 229 DCHECK_EQ(record()->nSize, | |
| 230 sizeof(DWORD) * 2 + sizeof(XFORM) + sizeof(DWORD)); | |
| 231 const XFORM* xform = reinterpret_cast<const XFORM*>(record()->dParm); | |
| 232 const DWORD* option = reinterpret_cast<const DWORD*>(xform + 1); | |
| 233 HDC hdc = context_->hdc; | |
| 234 switch (*option) { | |
| 235 case MWT_IDENTITY: | |
| 236 if (base_matrix) { | |
| 237 res = 0 != SetWorldTransform(hdc, base_matrix); | |
| 238 } else { | |
| 239 res = 0 != ModifyWorldTransform(hdc, xform, MWT_IDENTITY); | |
| 240 } | |
| 241 break; | |
| 242 case MWT_LEFTMULTIPLY: | |
| 243 case MWT_RIGHTMULTIPLY: | |
| 244 res = 0 != ModifyWorldTransform(hdc, xform, *option); | |
| 245 break; | |
| 246 case 4: // MWT_SET | |
| 247 if (base_matrix) { | |
| 248 res = 0 != SetWorldTransform(hdc, base_matrix) && | |
| 249 ModifyWorldTransform(hdc, xform, MWT_LEFTMULTIPLY); | |
| 250 } else { | |
| 251 res = 0 != SetWorldTransform(hdc, xform); | |
| 252 } | |
| 253 break; | |
| 254 default: | |
| 255 res = false; | |
| 256 break; | |
| 257 } | |
| 258 break; | |
| 259 } | |
| 260 case EMR_SETLAYOUT: | |
| 261 // Ignore it. | |
| 262 res = true; | |
| 263 break; | |
| 264 default: { | |
| 265 res = Play(); | |
| 266 break; | |
| 267 } | |
| 268 } | |
| 269 return res; | |
| 270 } | |
| 271 | |
| 272 Emf::Enumerator::Enumerator(const Emf& emf, HDC context, const RECT* rect) { | |
| 273 context_.handle_table = NULL; | |
| 274 context_.objects_count = 0; | |
| 275 context_.hdc = NULL; | |
| 276 items_.clear(); | |
| 277 if (!EnumEnhMetaFile(context, | |
| 278 emf.emf(), | |
| 279 &Emf::Enumerator::EnhMetaFileProc, | |
| 280 reinterpret_cast<void*>(this), | |
| 281 rect)) { | |
| 282 NOTREACHED(); | |
| 283 items_.clear(); | |
| 284 } | |
| 285 DCHECK_EQ(context_.hdc, context); | |
| 286 } | |
| 287 | |
| 288 Emf::Enumerator::const_iterator Emf::Enumerator::begin() const { | |
| 289 return items_.begin(); | |
| 290 } | |
| 291 | |
| 292 Emf::Enumerator::const_iterator Emf::Enumerator::end() const { | |
| 293 return items_.end(); | |
| 294 } | |
| 295 | |
| 296 int CALLBACK Emf::Enumerator::EnhMetaFileProc(HDC hdc, | |
| 297 HANDLETABLE* handle_table, | |
| 298 const ENHMETARECORD* record, | |
| 299 int objects_count, | |
| 300 LPARAM param) { | |
| 301 Enumerator& emf = *reinterpret_cast<Enumerator*>(param); | |
| 302 if (!emf.context_.handle_table) { | |
| 303 DCHECK(!emf.context_.handle_table); | |
| 304 DCHECK(!emf.context_.objects_count); | |
| 305 emf.context_.handle_table = handle_table; | |
| 306 emf.context_.objects_count = objects_count; | |
| 307 emf.context_.hdc = hdc; | |
| 308 } else { | |
| 309 DCHECK_EQ(emf.context_.handle_table, handle_table); | |
| 310 DCHECK_EQ(emf.context_.objects_count, objects_count); | |
| 311 DCHECK_EQ(emf.context_.hdc, hdc); | |
| 312 } | |
| 313 emf.items_.push_back(Record(&emf.context_, record)); | |
| 314 return 1; | |
| 315 } | |
| 316 | |
| 317 } // namespace gfx | |
| OLD | NEW |