| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 #ifndef PRINTING_EMF_WIN_H_ | 5 #ifndef PRINTING_EMF_WIN_H_ |
| 6 #define PRINTING_EMF_WIN_H_ | 6 #define PRINTING_EMF_WIN_H_ |
| 7 | 7 |
| 8 #include <windows.h> | 8 #include <windows.h> |
| 9 #include <vector> | 9 #include <vector> |
| 10 | 10 |
| 11 #include "base/basictypes.h" | 11 #include "base/basictypes.h" |
| 12 #include "base/gtest_prod_util.h" | 12 #include "base/gtest_prod_util.h" |
| 13 #include "printing/native_metafile_win.h" | 13 #include "printing/native_metafile.h" |
| 14 | 14 |
| 15 class FilePath; | 15 class FilePath; |
| 16 | 16 |
| 17 namespace gfx { | |
| 18 class Rect; | |
| 19 } | |
| 20 | |
| 21 namespace printing { | 17 namespace printing { |
| 22 | 18 |
| 23 // Simple wrapper class that manage an EMF data stream and its virtual HDC. | 19 // Implementing NativeMetafile |
| 24 class Emf : public NativeMetafile { | 20 class Emf : public NativeMetafile { |
| 25 public: | 21 public: |
| 26 class Record; | 22 class Record; |
| 27 class Enumerator; | 23 class Enumerator; |
| 28 struct EnumerationContext; | 24 struct EnumerationContext; |
| 29 | 25 |
| 30 virtual ~Emf(); | 26 virtual ~Emf(); |
| 31 | 27 |
| 32 // Initializes the Emf with the data in |src_buffer|. Returns true on success. | 28 virtual bool Init() { return true; } |
| 33 virtual bool Init(const void* src_buffer, uint32 src_buffer_size); | 29 virtual bool Init(const void* src_buffer, uint32 src_buffer_size); |
| 34 | 30 |
| 35 // Generates a virtual HDC that will record every GDI commands and compile it | 31 virtual uint32 GetDataSize() const; |
| 36 // in a EMF data stream. | 32 virtual bool GetData(void* buffer, uint32 size) const; |
| 37 // hdc is used to setup the default DPI and color settings. hdc is optional. | 33 |
| 38 // rect specifies the dimensions (in .01-millimeter units) of the EMF. rect is | 34 virtual bool FinishPage(); |
| 39 // optional. | 35 virtual void Close(); |
| 36 |
| 37 // Saves the EMF data to a file as-is. It is recommended to use the .emf file |
| 38 // extension but it is not enforced. This function synchronously writes to the |
| 39 // file. For testing only. |
| 40 virtual bool SaveTo(const FilePath& file_path) const; |
| 41 |
| 42 // Should be passed to Playback to keep the exact same size. |
| 43 virtual gfx::Rect GetPageBounds(unsigned int page_number) const; |
| 44 |
| 45 virtual unsigned int GetPageCount() const { |
| 46 return 1; |
| 47 } |
| 48 |
| 40 virtual bool CreateDc(HDC sibling, const RECT* rect); | 49 virtual bool CreateDc(HDC sibling, const RECT* rect); |
| 41 | |
| 42 // Similar to the above method but the metafile is backed by a file. | |
| 43 virtual bool CreateFileBackedDc(HDC sibling, | 50 virtual bool CreateFileBackedDc(HDC sibling, |
| 44 const RECT* rect, | 51 const RECT* rect, |
| 45 const FilePath& path); | 52 const FilePath& path); |
| 53 virtual bool CreateFromFile(const FilePath& file_path); |
| 46 | 54 |
| 47 // Load an EMF file. | |
| 48 virtual bool CreateFromFile(const FilePath& metafile_path); | |
| 49 | |
| 50 // TODO(maruel): CreateFromFile(). If ever used. Maybe users would like to | |
| 51 // have the ability to save web pages to an EMF file? Afterward, it is easy to | |
| 52 // convert to PDF or PS. | |
| 53 | |
| 54 // Closes the HDC created by CreateDc() and generates the compiled EMF | |
| 55 // data. | |
| 56 virtual bool CloseDc(); | 55 virtual bool CloseDc(); |
| 57 | 56 |
| 58 // Closes the EMF data handle when it is not needed anymore. | |
| 59 virtual void CloseEmf(); | |
| 60 | |
| 61 // "Plays" the EMF buffer in a HDC. It is the same effect as calling the | |
| 62 // original GDI function that were called when recording the EMF. |rect| is in | |
| 63 // "logical units" and is optional. If |rect| is NULL, the natural EMF bounds | |
| 64 // are used. | |
| 65 // Note: Windows has been known to have stack buffer overflow in its GDI | |
| 66 // functions, whether used directly or indirectly through precompiled EMF | |
| 67 // data. We have to accept the risk here. Since it is used only for printing, | |
| 68 // it requires user intervention. | |
| 69 virtual bool Playback(HDC hdc, const RECT* rect) const; | 57 virtual bool Playback(HDC hdc, const RECT* rect) const; |
| 70 | |
| 71 // The slow version of Playback(). It enumerates all the records and play them | |
| 72 // back in the HDC. The trick is that it skip over the records known to have | |
| 73 // issue with some printers. See Emf::Record::SafePlayback implementation for | |
| 74 // details. | |
| 75 virtual bool SafePlayback(HDC hdc) const; | 58 virtual bool SafePlayback(HDC hdc) const; |
| 76 | 59 |
| 77 // Retrieves the bounds of the painted area by this EMF buffer. This value | |
| 78 // should be passed to Playback to keep the exact same size. | |
| 79 virtual gfx::Rect GetBounds() const; | |
| 80 | |
| 81 // Retrieves the EMF stream size. | |
| 82 virtual uint32 GetDataSize() const; | |
| 83 | |
| 84 // Retrieves the EMF stream. | |
| 85 virtual bool GetData(void* buffer, uint32 size) const; | |
| 86 | |
| 87 // Retrieves the EMF stream. It is an helper function. | |
| 88 virtual bool GetData(std::vector<uint8>* buffer) const; | 60 virtual bool GetData(std::vector<uint8>* buffer) const; |
| 61 virtual bool StartPage(); |
| 89 | 62 |
| 90 virtual HENHMETAFILE emf() const { | 63 virtual HENHMETAFILE emf() const { |
| 91 return emf_; | 64 return emf_; |
| 92 } | 65 } |
| 93 | 66 |
| 94 virtual HDC hdc() const { | 67 virtual HDC hdc() const { |
| 95 return hdc_; | 68 return hdc_; |
| 96 } | 69 } |
| 97 | 70 |
| 98 // Inserts a custom GDICOMMENT records indicating StartPage/EndPage calls | |
| 99 // (since StartPage and EndPage do not work in a metafile DC). Only valid | |
| 100 // when hdc_ is non-NULL. | |
| 101 virtual bool StartPage(); | |
| 102 virtual bool EndPage(); | |
| 103 | |
| 104 // Saves the EMF data to a file as-is. It is recommended to use the .emf file | |
| 105 // extension but it is not enforced. This function synchronously writes to the | |
| 106 // file. For testing only. | |
| 107 virtual bool SaveTo(const std::wstring& filename) const; | |
| 108 | |
| 109 protected: | 71 protected: |
| 110 Emf(); | 72 Emf(); |
| 111 | 73 |
| 112 private: | 74 private: |
| 113 friend class NativeMetafileFactory; | 75 friend class NativeMetafileFactory; |
| 114 FRIEND_TEST_ALL_PREFIXES(EmfTest, DC); | 76 FRIEND_TEST_ALL_PREFIXES(EmfTest, DC); |
| 115 FRIEND_TEST_ALL_PREFIXES(EmfTest, FileBackedDC); | 77 FRIEND_TEST_ALL_PREFIXES(EmfTest, FileBackedDC); |
| 116 FRIEND_TEST_ALL_PREFIXES(EmfPrintingTest, Enumerate); | 78 FRIEND_TEST_ALL_PREFIXES(EmfPrintingTest, Enumerate); |
| 117 FRIEND_TEST_ALL_PREFIXES(EmfPrintingTest, PageBreak); | 79 FRIEND_TEST_ALL_PREFIXES(EmfPrintingTest, PageBreak); |
| 118 | 80 |
| 81 gfx::Rect Emf::GetPageBounds() const; |
| 82 |
| 119 // Playbacks safely one EMF record. | 83 // Playbacks safely one EMF record. |
| 120 static int CALLBACK SafePlaybackProc(HDC hdc, | 84 static int CALLBACK SafePlaybackProc(HDC hdc, |
| 121 HANDLETABLE* handle_table, | 85 HANDLETABLE* handle_table, |
| 122 const ENHMETARECORD* record, | 86 const ENHMETARECORD* record, |
| 123 int objects_count, | 87 int objects_count, |
| 124 LPARAM param); | 88 LPARAM param); |
| 125 | 89 |
| 126 // Compiled EMF data handle. | 90 // Compiled EMF data handle. |
| 127 HENHMETAFILE emf_; | 91 HENHMETAFILE emf_; |
| 128 | 92 |
| 129 // Valid when generating EMF data through a virtual HDC. | 93 // Valid when generating EMF data through a virtual HDC. |
| 130 HDC hdc_; | 94 HDC hdc_; |
| 131 | 95 |
| 132 DISALLOW_COPY_AND_ASSIGN(Emf); | 96 DISALLOW_COPY_AND_ASSIGN(Emf); |
| 133 }; | 97 }; |
| 134 | 98 |
| 135 struct Emf::EnumerationContext { | 99 struct Emf::EnumerationContext { |
| 136 HANDLETABLE* handle_table; | 100 HANDLETABLE* handle_table; |
| 137 int objects_count; | 101 int objects_count; |
| 138 HDC hdc; | 102 HDC hdc; |
| 139 }; | 103 }; |
| 140 | 104 |
| 141 // One EMF record. It keeps pointers to the EMF buffer held by Emf::emf_. | 105 // One EMF record. It keeps pointers to the EMF buffer held by Emf::emf_. |
| 142 // The entries become invalid once Emf::CloseEmf() is called. | 106 // The entries become invalid once Emf::Close() is called. |
| 143 class Emf::Record { | 107 class Emf::Record { |
| 144 public: | 108 public: |
| 145 // Plays the record. | 109 // Plays the record. |
| 146 bool Play() const; | 110 bool Play() const; |
| 147 | 111 |
| 148 // Plays the record working around quirks with SetLayout, | 112 // Plays the record working around quirks with SetLayout, |
| 149 // SetWorldTransform and ModifyWorldTransform. See implementation for details. | 113 // SetWorldTransform and ModifyWorldTransform. See implementation for details. |
| 150 bool SafePlayback(const XFORM* base_matrix) const; | 114 bool SafePlayback(const XFORM* base_matrix) const; |
| 151 | 115 |
| 152 // Access the underlying EMF record. | 116 // Access the underlying EMF record. |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 185 private: | 149 private: |
| 186 // Processes one EMF record and saves it in the items_ array. | 150 // Processes one EMF record and saves it in the items_ array. |
| 187 static int CALLBACK EnhMetaFileProc(HDC hdc, | 151 static int CALLBACK EnhMetaFileProc(HDC hdc, |
| 188 HANDLETABLE* handle_table, | 152 HANDLETABLE* handle_table, |
| 189 const ENHMETARECORD* record, | 153 const ENHMETARECORD* record, |
| 190 int objects_count, | 154 int objects_count, |
| 191 LPARAM param); | 155 LPARAM param); |
| 192 | 156 |
| 193 // The collection of every EMF records in the currently loaded EMF buffer. | 157 // The collection of every EMF records in the currently loaded EMF buffer. |
| 194 // Initialized by Enumerate(). It keeps pointers to the EMF buffer held by | 158 // Initialized by Enumerate(). It keeps pointers to the EMF buffer held by |
| 195 // Emf::emf_. The entries become invalid once Emf::CloseEmf() is called. | 159 // Emf::emf_. The entries become invalid once Emf::Close() is called. |
| 196 std::vector<Record> items_; | 160 std::vector<Record> items_; |
| 197 | 161 |
| 198 EnumerationContext context_; | 162 EnumerationContext context_; |
| 199 | 163 |
| 200 DISALLOW_COPY_AND_ASSIGN(Enumerator); | 164 DISALLOW_COPY_AND_ASSIGN(Enumerator); |
| 201 }; | 165 }; |
| 202 | 166 |
| 203 } // namespace printing | 167 } // namespace printing |
| 204 | 168 |
| 205 #endif // PRINTING_EMF_WIN_H_ | 169 #endif // PRINTING_EMF_WIN_H_ |
| OLD | NEW |