Avoid core dumps when SetImageList is used.

[wxWidgets.git] / src / msw / dcprint.cpp

diff --git a/src/msw/dcprint.cpp b/src/msw/dcprint.cpp
index 6802ea97826cf3e0d5a51c65345b4fe95b896627..819514bb60cd22b6b5681589547883c12d7673a4 100644 (file)
--- a/src/msw/dcprint.cpp
+++ b/src/msw/dcprint.cpp
@@ -1,153 +1,246 @@
 /////////////////////////////////////////////////////////////////////////////
 /////////////////////////////////////////////////////////////////////////////

-// Name:        dcprint.cpp
+// Name:        src/msw/dcprint.cpp

 // Purpose:     wxPrinterDC class
 // Author:      Julian Smart
 // Modified by:
 // Created:     01/02/97
 // RCS-ID:      $Id$
 // Purpose:     wxPrinterDC class
 // Author:      Julian Smart
 // Modified by:
 // Created:     01/02/97
 // RCS-ID:      $Id$

-// Copyright:   (c) Julian Smart and Markus Holzem
-// Licence:    wxWindows licence
+// Copyright:   (c) Julian Smart
+// Licence:     wxWindows licence

 /////////////////////////////////////////////////////////////////////////////
 
 /////////////////////////////////////////////////////////////////////////////
 

-#ifdef __GNUG__
-#pragma implementation
-#pragma implementation "dcprint.h"
+// ============================================================================
+// declarations
+// ============================================================================
+
+// ----------------------------------------------------------------------------
+// headers
+// ----------------------------------------------------------------------------
+
+#if defined(__GNUG__) && !defined(NO_GCC_PRAGMA)
+    #pragma implementation "dcprint.h"

 #endif
 
 // For compilers that support precompilation, includes "wx.h".
 #include "wx/wxprec.h"
 
 #ifdef __BORLANDC__
 #endif
 
 // For compilers that support precompilation, includes "wx.h".
 #include "wx/wxprec.h"
 
 #ifdef __BORLANDC__

-#pragma hdrstop
+    #pragma hdrstop

 #endif
 
 #ifndef WX_PRECOMP
 #endif
 
 #ifndef WX_PRECOMP

+    #include "wx/string.h"
+    #include "wx/log.h"
+    #include "wx/window.h"
+    #include "wx/dcmemory.h"

 #endif
 
 #endif
 

-#include "wx/dcprint.h"
-#include "math.h"
-#include "fstream.h"
+#if wxUSE_PRINTING_ARCHITECTURE

 
 

-#include <windows.h>
+#include "wx/msw/private.h"

 
 

-#if USE_COMMON_DIALOGS
-#include <commdlg.h>
+#if wxUSE_WXDIB
+#include "wx/msw/dib.h"

 #endif
 
 #endif
 

-#ifndef __WIN32__
-#include <print.h>
-#endif
+#include "wx/dcprint.h"
+#include "math.h"

 
 

-#ifdef DrawText
-#undef DrawText
+#if wxUSE_COMMON_DIALOGS
+    #include <commdlg.h>

 #endif
 
 #endif
 

-#ifdef GetCharWidth
-#undef GetCharWidth
+#ifndef __WIN32__
+    #include <print.h>

 #endif
 
 #endif
 

-#ifdef StartDoc
-#undef StartDoc
+// mingw32 defines GDI_ERROR incorrectly
+#if defined(__GNUWIN32__) || !defined(GDI_ERROR)
+    #undef GDI_ERROR
+    #define GDI_ERROR ((int)-1)

 #endif
 
 #endif
 

-#if !USE_SHARED_LIBRARY
+// ----------------------------------------------------------------------------
+// wxWin macros
+// ----------------------------------------------------------------------------
+

 IMPLEMENT_CLASS(wxPrinterDC, wxDC)
 IMPLEMENT_CLASS(wxPrinterDC, wxDC)

-#endif

 
 

-wxPrinterDC::wxPrinterDC(const wxString& driver_name, const wxString& device_name, const wxString& file, const bool interactive, const int orientation)
+// ============================================================================
+// implementation
+// ============================================================================
+
+// ----------------------------------------------------------------------------
+// wxPrinterDC construction
+// ----------------------------------------------------------------------------
+
+// This form is deprecated
+wxPrinterDC::wxPrinterDC(const wxString& driver_name,
+                         const wxString& device_name,
+                         const wxString& file,
+                         bool interactive,
+                         int orientation)
+{
+    m_isInteractive = interactive;
+
+    if ( !file.empty() )
+        m_printData.SetFilename(file);
+
+#if wxUSE_COMMON_DIALOGS
+    if ( interactive )
+    {
+        PRINTDLG pd;
+
+        pd.lStructSize = sizeof( PRINTDLG );
+        pd.hwndOwner = (HWND) NULL;
+        pd.hDevMode = (HANDLE)NULL;
+        pd.hDevNames = (HANDLE)NULL;
+        pd.Flags = PD_RETURNDC | PD_NOSELECTION | PD_NOPAGENUMS;
+        pd.nFromPage = 0;
+        pd.nToPage = 0;
+        pd.nMinPage = 0;
+        pd.nMaxPage = 0;
+        pd.nCopies = 1;
+        pd.hInstance = (HINSTANCE)NULL;
+
+        m_ok = PrintDlg( &pd ) != 0;
+        if ( m_ok )
+        {
+            m_hDC = (WXHDC) pd.hDC;
+        }
+    }
+    else
+#endif // wxUSE_COMMON_DIALOGS
+    {
+        if ( !driver_name.empty() && !device_name.empty() && !file.empty() )
+        {
+            m_hDC = (WXHDC) CreateDC(driver_name, device_name, file, NULL);
+        }
+        else // we don't have all parameters, ask the user
+        {
+            wxPrintData printData;
+            printData.SetOrientation(orientation);
+            m_hDC = wxGetPrinterDC(printData);
+        }
+
+        m_ok = m_hDC ? TRUE: FALSE;
+
+        // as we created it, we must delete it as well
+        m_bOwnsDC = TRUE;
+    }
+
+    Init();
+}
+
+wxPrinterDC::wxPrinterDC(const wxPrintData& printData)

 {
 {

-  m_isInteractive = interactive;
-
-  if (!file.IsNull() && file != "")
-    m_filename = file;
-
-#if USE_COMMON_DIALOGS
-   if (interactive)
-   {
-     PRINTDLG pd;
-       
-     pd.lStructSize = sizeof( PRINTDLG );
-     pd.hwndOwner=NULL;
-     pd.hDevMode=(HANDLE)NULL;
-     pd.hDevNames=(HANDLE)NULL;
-     pd.Flags=PD_RETURNDC | PD_NOSELECTION | PD_NOPAGENUMS;
-     pd.nFromPage=0;
-     pd.nToPage=0;
-     pd.nMinPage=0;
-     pd.nMaxPage=0;
-     pd.nCopies=1;
-     pd.hInstance=(HINSTANCE)NULL;
-
-     if ( PrintDlg( &pd ) != 0 )
-     {
-       m_hDC = (WXHDC) pd.hDC;
-       m_ok = TRUE;
-     }
-     else
-     {
-       m_ok = FALSE;
-       return;
-     }
-
-//     m_dontDelete = TRUE;
-   }
-   else
+    m_printData = printData;
+
+    m_isInteractive = FALSE;
+
+    m_hDC = wxGetPrinterDC(printData);
+    m_ok = m_hDC != 0;
+    m_bOwnsDC = TRUE;
+
+    Init();
+}
+
+
+wxPrinterDC::wxPrinterDC(WXHDC dc)
+{
+    m_isInteractive = FALSE;
+
+    m_hDC = dc;
+    m_bOwnsDC = TRUE;
+    m_ok = TRUE;
+}
+
+void wxPrinterDC::Init()
+{
+    if ( m_hDC )
+    {
+        //     int width = GetDeviceCaps(m_hDC, VERTRES);
+        //     int height = GetDeviceCaps(m_hDC, HORZRES);
+        SetMapMode(wxMM_TEXT);
+
+        SetBrush(*wxBLACK_BRUSH);
+        SetPen(*wxBLACK_PEN);
+    }
+}
+
+// ----------------------------------------------------------------------------
+// wxPrinterDC {Start/End}{Page/Doc} methods
+// ----------------------------------------------------------------------------
+
+bool wxPrinterDC::StartDoc(const wxString& message)
+{
+    DOCINFO docinfo;
+    docinfo.cbSize = sizeof(DOCINFO);
+    docinfo.lpszDocName = (const wxChar*)message;
+
+    wxString filename(m_printData.GetFilename());
+
+    if (filename.IsEmpty())
+        docinfo.lpszOutput = NULL;
+    else
+        docinfo.lpszOutput = (const wxChar *) filename;
+
+#if defined(__WIN95__)
+    docinfo.lpszDatatype = NULL;
+    docinfo.fwType = 0;

 #endif
 #endif

-   if ((!driver_name.IsNull() && driver_name != "") &&
-       (!device_name.IsNull() && device_name != "") &&
-       (!file.IsNull() && file != ""))
-   {
-     m_hDC = (WXHDC) CreateDC((char *) (const char *) driver_name, (char *) (const char *) device_name, (char *) (const char *) file, NULL);
-     m_ok = m_hDC ? TRUE: FALSE;
-   }
-   else
-   {
-     m_hDC = wxGetPrinterDC(orientation);
-     m_ok = m_hDC ? TRUE: FALSE;
-   }
-   
-   if (m_hDC)
-   {
-//     int width = GetDeviceCaps(m_hDC, VERTRES);
-//     int height = GetDeviceCaps(m_hDC, HORZRES);
-     SetMapMode(MM_TEXT);
-   }
-  SetBrush(*wxBLACK_BRUSH);
-  SetPen(*wxBLACK_PEN);
+
+    if (!m_hDC)
+        return FALSE;
+
+    int ret = ::StartDoc(GetHdc(), &docinfo);
+
+    if (ret <= 0)
+    {
+        DWORD lastError = GetLastError();
+        wxLogDebug(wxT("wxDC::StartDoc failed with error: %ld\n"), lastError);
+    }
+
+    return (ret > 0);
+}
+
+void wxPrinterDC::EndDoc()
+{
+    if (m_hDC) ::EndDoc((HDC) m_hDC);

 }
 
 }
 

-wxPrinterDC::wxPrinterDC(WXHDC theDC)
+void wxPrinterDC::StartPage()

 {
 {

-  m_isInteractive = FALSE;
-
-  m_hDC = theDC;
-  m_ok = TRUE;
-  if (m_hDC)
- {
-//     int width = GetDeviceCaps(m_hDC, VERTRES);
-//     int height = GetDeviceCaps(m_hDC, HORZRES);
-    SetMapMode(MM_TEXT);
-  }
-  SetBrush(*wxBLACK_BRUSH);
-  SetPen(*wxBLACK_PEN);
+    if (m_hDC)
+        ::StartPage((HDC) m_hDC);

 }
 
 }
 

-wxPrinterDC::~wxPrinterDC(void)
+void wxPrinterDC::EndPage()

 {
 {

+    if (m_hDC)
+        ::EndPage((HDC) m_hDC);

 }
 
 }
 

-WXHDC wxGetPrinterDC(int orientation)
+// Returns default device and port names
+static bool wxGetDefaultDeviceName(wxString& deviceName, wxString& portName)

 {
 {

-    HDC         hDC;
-    LPDEVMODE   lpDevMode = NULL;
+    deviceName.clear();
+

     LPDEVNAMES  lpDevNames;
     LPDEVNAMES  lpDevNames;

-    LPSTR       lpszDriverName;
-    LPSTR       lpszDeviceName;
-    LPSTR       lpszPortName;
+    LPTSTR      lpszDeviceName;
+    LPTSTR      lpszPortName;

 
     PRINTDLG    pd;
 
     PRINTDLG    pd;

-    // __GNUWIN32__ has trouble believing PRINTDLG is 66 bytes - thinks it is 68
+
+    // Cygwin has trouble believing PRINTDLG is 66 bytes - thinks it is 68
+#ifdef __GNUWIN32__
+    memset(&pd, 0, 66);

     pd.lStructSize    = 66; // sizeof(PRINTDLG);
     pd.lStructSize    = 66; // sizeof(PRINTDLG);

+#else
+    memset(&pd, 0, sizeof(PRINTDLG));
+    pd.lStructSize    = sizeof(PRINTDLG);
+#endif
+

     pd.hwndOwner      = (HWND)NULL;
     pd.hDevMode       = NULL; // Will be created by PrintDlg
     pd.hDevNames      = NULL; // Ditto
     pd.hwndOwner      = (HWND)NULL;
     pd.hDevMode       = NULL; // Will be created by PrintDlg
     pd.hDevNames      = NULL; // Ditto


@@ -161,48 +254,222 @@ WXHDC wxGetPrinterDC(int orientation)
         if (pd.hDevNames)
             GlobalFree(pd.hDevNames);
 
         if (pd.hDevNames)
             GlobalFree(pd.hDevNames);
 

-        return(0);
+        return FALSE;

     }
 
     }
 

-    if (!pd.hDevNames)
+    if (pd.hDevNames)

     {
     {

-        if ( pd.hDevMode )
-            GlobalFree(pd.hDevMode);
+        lpDevNames = (LPDEVNAMES)GlobalLock(pd.hDevNames);
+        lpszDeviceName = (LPTSTR)lpDevNames + lpDevNames->wDeviceOffset;
+        lpszPortName   = (LPTSTR)lpDevNames + lpDevNames->wOutputOffset;
+
+        deviceName = lpszDeviceName;
+        portName = lpszPortName;
+
+        GlobalUnlock(pd.hDevNames);
+        GlobalFree(pd.hDevNames);
+        pd.hDevNames=NULL;
+    }
+
+    if (pd.hDevMode)
+    {
+        GlobalFree(pd.hDevMode);
+        pd.hDevMode=NULL;

     }
     }

+    return ( deviceName != wxEmptyString );
+}
+
+// Gets an HDC for the specified printer configuration
+WXHDC WXDLLEXPORT wxGetPrinterDC(const wxPrintData& printDataConst)
+{
+    wxPrintData printData = printDataConst;
+    printData.ConvertToNative();
+
+    wxChar* driverName = (wxChar*) NULL;
+
+    wxString devNameStr = printData.GetPrinterName();
+    wxChar* portName = (wxChar*) NULL; // Obsolete in WIN32
+
+    const wxChar* deviceName;
+    if ( !devNameStr )
+        deviceName = (wxChar*) NULL;
+    else
+        deviceName = devNameStr.c_str();
+
+    LPDEVMODE lpDevMode = (LPDEVMODE) NULL;

 
 

-    lpDevNames = (LPDEVNAMES)GlobalLock(pd.hDevNames);
-    lpszDriverName = (LPSTR)lpDevNames + lpDevNames->wDriverOffset;
-    lpszDeviceName = (LPSTR)lpDevNames + lpDevNames->wDeviceOffset;
-    lpszPortName   = (LPSTR)lpDevNames + lpDevNames->wOutputOffset;
-    GlobalUnlock(pd.hDevNames);
+    HGLOBAL hDevMode = (HGLOBAL)(DWORD) printData.GetNativeData();

 
 

-    if ( pd.hDevMode )
+    if ( hDevMode )
+        lpDevMode = (DEVMODE*) GlobalLock(hDevMode);
+
+    if ( !devNameStr )

     {
     {

-        lpDevMode = (DEVMODE*) GlobalLock(pd.hDevMode);
-        lpDevMode->dmOrientation = orientation;
-        lpDevMode->dmFields |= DM_ORIENTATION;
+        // Retrieve the default device name
+        wxString portName;
+        if ( !wxGetDefaultDeviceName(devNameStr, portName) )
+        {
+            return 0; // Could not get default device name
+        }
+        deviceName = devNameStr.c_str();

     }
 
 #ifdef __WIN32__
     }
 
 #ifdef __WIN32__

-    hDC = CreateDC(lpszDriverName, lpszDeviceName, lpszPortName, (DEVMODE *)lpDevMode);
+    HDC hDC = CreateDC(driverName, deviceName, portName, (DEVMODE *) lpDevMode);

 #else
 #else

-    hDC = CreateDC(lpszDriverName, lpszDeviceName, lpszPortName, (LPSTR)lpDevMode);
+    HDC hDC = CreateDC(driverName, deviceName, portName, (LPSTR) lpDevMode);

 #endif
 
 #endif
 

-    if (pd.hDevMode && lpDevMode)
-        GlobalUnlock(pd.hDevMode);
+    if (hDevMode && lpDevMode)
+        GlobalUnlock(hDevMode);

 
 

-    if (pd.hDevNames)
+    return (WXHDC) hDC;
+}
+
+// ----------------------------------------------------------------------------
+// wxPrinterDC bit blitting/bitmap drawing
+// ----------------------------------------------------------------------------
+
+// helper of DoDrawBitmap() and DoBlit()
+static
+bool DrawBitmapUsingStretchDIBits(HDC hdc,
+                                  const wxBitmap& bmp,
+                                  wxCoord x, wxCoord y)
+{
+#if wxUSE_WXDIB
+    wxDIB dib(bmp);
+    bool ok = dib.IsOk();
+    if ( !ok )
+        return FALSE;
+
+    DIBSECTION ds;
+    if ( !::GetObject(dib.GetHandle(), sizeof(ds), &ds) )

     {
     {

-           GlobalFree(pd.hDevNames);
-           pd.hDevNames=NULL;
+        wxLogLastError(_T("GetObject(DIBSECTION)"));
+
+        return FALSE;

     }
     }

-    if (pd.hDevMode)
+
+    // ok, we've got all data we need, do blit it
+    if ( ::StretchDIBits
+            (
+                hdc,
+                x, y,
+                ds.dsBmih.biWidth, ds.dsBmih.biHeight,
+                0, 0,
+                ds.dsBmih.biWidth, ds.dsBmih.biHeight,
+                ds.dsBm.bmBits,
+                (LPBITMAPINFO)&ds.dsBmih,
+                DIB_RGB_COLORS,
+                SRCCOPY
+            ) == GDI_ERROR )

     {
     {

-        GlobalFree(pd.hDevMode);
-        pd.hDevMode=NULL;
+        wxLogLastError(wxT("StretchDIBits"));
+
+        return FALSE;

     }
     }

-    return (WXHDC) hDC;
+
+    return TRUE;
+#else
+    return FALSE;
+#endif

 }
 
 }
 

+void wxPrinterDC::DoDrawBitmap(const wxBitmap& bmp,
+                               wxCoord x, wxCoord y,
+                               bool useMask)
+{
+    wxCHECK_RET( bmp.Ok(), _T("invalid bitmap in wxPrinterDC::DrawBitmap") );
+
+    int width = bmp.GetWidth(),
+        height = bmp.GetHeight();
+
+    if ( !(::GetDeviceCaps(GetHdc(), RASTERCAPS) & RC_STRETCHDIB) ||
+            !DrawBitmapUsingStretchDIBits(GetHdc(), bmp, x, y) )
+    {
+        // no support for StretchDIBits() or an error occured if we got here
+        wxMemoryDC memDC;
+        memDC.SelectObject(bmp);

 
 

+        Blit(x, y, width, height, &memDC, 0, 0, wxCOPY, useMask);
+
+        memDC.SelectObject(wxNullBitmap);
+    }
+}
+
+bool wxPrinterDC::DoBlit(wxCoord xdest, wxCoord ydest,
+                         wxCoord width, wxCoord height,
+                         wxDC *source,
+                         wxCoord WXUNUSED(xsrc), wxCoord WXUNUSED(ysrc),
+                         int WXUNUSED(rop), bool useMask,
+                         wxCoord WXUNUSED(xsrcMask), wxCoord WXUNUSED(ysrcMask))
+{
+    wxBitmap& bmp = source->GetSelectedBitmap();
+    wxMask *mask = useMask ? bmp.GetMask() : NULL;
+    if ( mask )
+    {
+        // If we are printing source colours are screen colours not printer
+        // colours and so we need copy the bitmap pixel by pixel.
+        RECT rect;
+        HDC dcSrc = GetHdcOf(*source);
+        MemoryHDC dcMask(dcSrc);
+        SelectInHDC selectMask(dcMask, (HBITMAP)mask->GetMaskBitmap());
+
+        for (int x = 0; x < width; x++)
+        {
+            for (int y = 0; y < height; y++)
+            {
+                COLORREF cref = ::GetPixel(dcMask, x, y);
+                if (cref)
+                {
+                    HBRUSH brush = ::CreateSolidBrush(::GetPixel(dcSrc, x, y));
+                    rect.left = xdest + x;
+                    rect.right = rect.left + 1;
+                    rect.top = ydest + y;
+                    rect.bottom = rect.top + 1;
+                    ::FillRect(GetHdc(), &rect, brush);
+                    ::DeleteObject(brush);
+                }
+            }
+        }
+    }
+    else // no mask
+    {
+        if ( !(::GetDeviceCaps(GetHdc(), RASTERCAPS) & RC_STRETCHDIB) ||
+                !DrawBitmapUsingStretchDIBits(GetHdc(), bmp, xdest, ydest) )
+        {
+            // no support for StretchDIBits
+
+            // as we are printing, source colours are screen colours not
+            // printer colours and so we need copy the bitmap pixel by pixel.
+            HDC dcSrc = GetHdcOf(*source);
+            RECT rect;
+            for (int y = 0; y < height; y++)
+            {
+                // optimization: draw identical adjacent pixels together.
+                for (int x = 0; x < width; x++)
+                {
+                    COLORREF col = ::GetPixel(dcSrc, x, y);
+                    HBRUSH brush = ::CreateSolidBrush( col );
+
+                    rect.left = xdest + x;
+                    rect.top = ydest + y;
+                    while( (x + 1 < width) &&
+                                (::GetPixel(dcSrc, x + 1, y) == col ) )
+                    {
+                        ++x;
+                    }
+                    rect.right = xdest + x + 1;
+                    rect.bottom = rect.top + 1;
+                    ::FillRect((HDC) m_hDC, &rect, brush);
+                    ::DeleteObject(brush);
+                }
+            }
+        }
+    }
+
+    return TRUE;
+}
+
+#endif
+    // wxUSE_PRINTING_ARCHITECTURE