Create a Factory for NativeMetafile.

printing/emf_win.h

-// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef PRINTING_EMF_WIN_H_
 #define PRINTING_EMF_WIN_H_
 
 #include <windows.h>
 #include <vector>
 
 #include "base/basictypes.h"
+#include "base/gtest_prod_util.h"
+#include "printing/native_metafile_win.h"
 
 class FilePath;
 
 namespace gfx {
 class Rect;
 }
 
 namespace printing {
 
 // Simple wrapper class that manage an EMF data stream and its virtual HDC.
-class Emf {
+class Emf : public NativeMetafile {
  public:
   class Record;
   class Enumerator;
   struct EnumerationContext;
 
-  Emf();
-  ~Emf();
+  virtual ~Emf();
 
   // Initializes the Emf with the data in |src_buffer|. Returns true on success.
-  bool Init(const void* src_buffer, uint32 src_buffer_size);
+  virtual bool Init(const void* src_buffer, uint32 src_buffer_size);
 
   // Generates a virtual HDC that will record every GDI commands and compile it
   // in a EMF data stream.
   // hdc is used to setup the default DPI and color settings. hdc is optional.
   // rect specifies the dimensions (in .01-millimeter units) of the EMF. rect is
   // optional.
-  bool CreateDc(HDC sibling, const RECT* rect);
+  virtual bool CreateDc(HDC sibling, const RECT* rect);
 
   // Similar to the above method but the metafile is backed by a file.
-  bool CreateFileBackedDc(HDC sibling, const RECT* rect, const FilePath& path);
+  virtual bool CreateFileBackedDc(HDC sibling,
+                                  const RECT* rect,
+                                  const FilePath& path);
 
     // Load an EMF file.
-  bool CreateFromFile(const FilePath& metafile_path);
+  virtual bool CreateFromFile(const FilePath& metafile_path);
 
   // TODO(maruel): CreateFromFile(). If ever used. Maybe users would like to
   // have the ability to save web pages to an EMF file? Afterward, it is easy to
   // convert to PDF or PS.
 
   // Closes the HDC created by CreateDc() and generates the compiled EMF
   // data.
-  bool CloseDc();
+  virtual bool CloseDc();
 
   // Closes the EMF data handle when it is not needed anymore.
-  void CloseEmf();
+  virtual void CloseEmf();
 
   // "Plays" the EMF buffer in a HDC. It is the same effect as calling the
   // original GDI function that were called when recording the EMF. |rect| is in
   // "logical units" and is optional. If |rect| is NULL, the natural EMF bounds
   // are used.
   // Note: Windows has been known to have stack buffer overflow in its GDI
   // functions, whether used directly or indirectly through precompiled EMF
   // data. We have to accept the risk here. Since it is used only for printing,
   // it requires user intervention.
-  bool Playback(HDC hdc, const RECT* rect) const;
+  virtual bool Playback(HDC hdc, const RECT* rect) const;
 
   // The slow version of Playback(). It enumerates all the records and play them
   // back in the HDC. The trick is that it skip over the records known to have
   // issue with some printers. See Emf::Record::SafePlayback implementation for
   // details.
-  bool SafePlayback(HDC hdc) const;
+  virtual bool SafePlayback(HDC hdc) const;
 
   // Retrieves the bounds of the painted area by this EMF buffer. This value
   // should be passed to Playback to keep the exact same size.
-  gfx::Rect GetBounds() const;
+  virtual gfx::Rect GetBounds() const;
 
   // Retrieves the EMF stream size.
-  uint32 GetDataSize() const;
+  virtual uint32 GetDataSize() const;
 
   // Retrieves the EMF stream.
-  bool GetData(void* buffer, uint32 size) const;
+  virtual bool GetData(void* buffer, uint32 size) const;
 
   // Retrieves the EMF stream. It is an helper function.
-  bool GetData(std::vector<uint8>* buffer) const;
+  virtual bool GetData(std::vector<uint8>* buffer) const;
 
-  HENHMETAFILE emf() const {
+  virtual HENHMETAFILE emf() const {
     return emf_;
   }
 
-  HDC hdc() const {
+  virtual HDC hdc() const {
     return hdc_;
   }
 
   // Inserts a custom GDICOMMENT records indicating StartPage/EndPage calls
   // (since StartPage and EndPage do not work in a metafile DC). Only valid
   // when hdc_ is non-NULL.
-  bool StartPage();
-  bool EndPage();
+  virtual bool StartPage();
+  virtual bool EndPage();
 
   // Saves the EMF data to a file as-is. It is recommended to use the .emf file
   // extension but it is not enforced. This function synchronously writes to the
   // file. For testing only.
-  bool SaveTo(const std::wstring& filename) const;
+  virtual bool SaveTo(const std::wstring& filename) const;
+
+ protected:
+  Emf();
 
  private:
+  friend class NativeMetafileFactory;
+  FRIEND_TEST_ALL_PREFIXES(EmfTest, DC);
+  FRIEND_TEST_ALL_PREFIXES(EmfTest, FileBackedDC);
+  FRIEND_TEST_ALL_PREFIXES(EmfPrintingTest, Enumerate);
+  FRIEND_TEST_ALL_PREFIXES(EmfPrintingTest, PageBreak);
+
   // Playbacks safely one EMF record.
   static int CALLBACK SafePlaybackProc(HDC hdc,
                                        HANDLETABLE* handle_table,
                                        const ENHMETARECORD* record,
                                        int objects_count,
                                        LPARAM param);
 
   // Compiled EMF data handle.
   HENHMETAFILE emf_;
 
@@ skipping 67 matching lines @@
   std::vector<Record> items_;
 
   EnumerationContext context_;
 
   DISALLOW_COPY_AND_ASSIGN(Enumerator);
 };
 
 }  // namespace printing
 
 #endif  // PRINTING_EMF_WIN_H_