]> git.saurik.com Git - wxWidgets.git/blobdiff - src/msw/ole/dataobj.cpp
tests for sorted wxChoice added
[wxWidgets.git] / src / msw / ole / dataobj.cpp
index dbc7f122e63ff8e34490cb28d5ee68eeba1da62b..30f46b1cf0bd4adb649ed6f650627c57b2337074 100644 (file)
@@ -43,7 +43,7 @@
 #endif
 #include <oleauto.h>
 
-#ifndef __WIN32__  
+#ifndef __WIN32__
   #include <ole2.h>
   #include <olestd.h>
 #endif
 // functions
 // ----------------------------------------------------------------------------
 
-static const char *GetTymedName(DWORD tymed);
+#ifdef __WXDEBUG__
+    static const wxChar *GetTymedName(DWORD tymed);
+#endif // Debug
+
+// to be moved into wx/msw/bitmap.h
+extern size_t wxConvertBitmapToDIB(BITMAPINFO *pbi, const wxBitmap& bitmap);
+extern wxBitmap wxConvertDIBToBitmap(const BITMAPINFO *bmi);
 
 // ----------------------------------------------------------------------------
 // wxIEnumFORMATETC interface implementation
 // ----------------------------------------------------------------------------
+
 class wxIEnumFORMATETC : public IEnumFORMATETC
 {
 public:
-  wxIEnumFORMATETC(CLIPFORMAT cf);
+    wxIEnumFORMATETC(const wxDataFormat* formats, ULONG nCount);
+    ~wxIEnumFORMATETC() { delete [] m_formats; }
 
-  DECLARE_IUNKNOWN_METHODS;
+    DECLARE_IUNKNOWN_METHODS;
 
-  // IEnumFORMATETC
-  STDMETHODIMP Next(ULONG celt, FORMATETC *rgelt, ULONG *pceltFetched);
-  STDMETHODIMP Skip(ULONG celt);
-  STDMETHODIMP Reset();
-  STDMETHODIMP Clone(IEnumFORMATETC **ppenum);
+    // IEnumFORMATETC
+    STDMETHODIMP Next(ULONG celt, FORMATETC *rgelt, ULONG *pceltFetched);
+    STDMETHODIMP Skip(ULONG celt);
+    STDMETHODIMP Reset();
+    STDMETHODIMP Clone(IEnumFORMATETC **ppenum);
 
 private:
-  FORMATETC   m_format;   // (unique @@@) format we can provide data in
-  ULONG       m_nCurrent; // current enum position (currently either 0 or 1)
+    CLIPFORMAT *m_formats;  // formats we can provide data in
+    ULONG       m_nCount,   // number of formats we support
+                m_nCurrent; // current enum position
 };
 
 // ----------------------------------------------------------------------------
 // wxIDataObject implementation of IDataObject interface
 // ----------------------------------------------------------------------------
+
 class wxIDataObject : public IDataObject
 {
 public:
-  wxIDataObject(wxDataObject *pDataObject);
+    wxIDataObject(wxDataObject *pDataObject);
+    ~wxIDataObject();
+
+    // normally, wxDataObject controls our lifetime (i.e. we're deleted when it
+    // is), but in some cases, the situation is inversed, that is we delete it
+    // when this object is deleted - setting this flag enables such logic
+    void SetDeleteFlag() { m_mustDelete = TRUE; }
+
+    DECLARE_IUNKNOWN_METHODS;
+
+    // IDataObject
+    STDMETHODIMP GetData(FORMATETC *pformatetcIn, STGMEDIUM *pmedium);
+    STDMETHODIMP GetDataHere(FORMATETC *pformatetc, STGMEDIUM *pmedium);
+    STDMETHODIMP QueryGetData(FORMATETC *pformatetc);
+    STDMETHODIMP GetCanonicalFormatEtc(FORMATETC *In, FORMATETC *pOut);
+    STDMETHODIMP SetData(FORMATETC *pfetc, STGMEDIUM *pmedium, BOOL fRelease);
+    STDMETHODIMP EnumFormatEtc(DWORD dwDirection, IEnumFORMATETC **ppenumFEtc);
+    STDMETHODIMP DAdvise(FORMATETC *pfetc, DWORD ad, IAdviseSink *p, DWORD *pdw);
+    STDMETHODIMP DUnadvise(DWORD dwConnection);
+    STDMETHODIMP EnumDAdvise(IEnumSTATDATA **ppenumAdvise);
+
+private:
+    wxDataObject *m_pDataObject;      // pointer to C++ class we belong to
+
+    bool m_mustDelete;
+};
 
-  DECLARE_IUNKNOWN_METHODS;
+// ----------------------------------------------------------------------------
+// small helper class for getting screen DC (we're working with bitmaps and
+// DIBs here)
+// ----------------------------------------------------------------------------
 
-  // IDataObject
-  STDMETHODIMP GetData(FORMATETC *pformatetcIn, STGMEDIUM *pmedium);
-  STDMETHODIMP GetDataHere(FORMATETC *pformatetc, STGMEDIUM *pmedium);
-  STDMETHODIMP QueryGetData(FORMATETC *pformatetc);
-  STDMETHODIMP GetCanonicalFormatEtc(FORMATETC *In, FORMATETC *pOut);
-  STDMETHODIMP SetData(FORMATETC *pfetc, STGMEDIUM *pmedium, BOOL fRelease);
-  STDMETHODIMP EnumFormatEtc(DWORD dwDirection, IEnumFORMATETC **ppenumFEtc);
-  STDMETHODIMP DAdvise(FORMATETC *pfetc, DWORD ad, IAdviseSink *p, DWORD *pdw);
-  STDMETHODIMP DUnadvise(DWORD dwConnection);
-  STDMETHODIMP EnumDAdvise(IEnumSTATDATA **ppenumAdvise);
+class ScreenHDC
+{
+public:
+    ScreenHDC() { m_hdc = GetDC(NULL);    }
+   ~ScreenHDC() { ReleaseDC(NULL, m_hdc); }
+    operator HDC() const { return m_hdc;  }
 
 private:
-  wxDataObject *m_pDataObject;      // pointer to C++ class we belong to
+    HDC m_hdc;
 };
 
 // ============================================================================
@@ -144,70 +178,90 @@ wxString wxDataFormat::GetId() const
 // ----------------------------------------------------------------------------
 
 BEGIN_IID_TABLE(wxIEnumFORMATETC)
-  ADD_IID(Unknown)
-  ADD_IID(EnumFORMATETC)
+    ADD_IID(Unknown)
+    ADD_IID(EnumFORMATETC)
 END_IID_TABLE;
 
 IMPLEMENT_IUNKNOWN_METHODS(wxIEnumFORMATETC)
 
-wxIEnumFORMATETC::wxIEnumFORMATETC(CLIPFORMAT cf)
+wxIEnumFORMATETC::wxIEnumFORMATETC(const wxDataFormat *formats, ULONG nCount)
 {
-  m_format.cfFormat = cf;
-  m_format.ptd      = NULL;
-  m_format.dwAspect = DVASPECT_CONTENT;
-  m_format.lindex   = -1;
-  m_format.tymed    = TYMED_HGLOBAL;
-  m_cRef = 0;
-  m_nCurrent = 0;
+    m_cRef = 0;
+    m_nCurrent = 0;
+    m_nCount = nCount;
+    m_formats = new CLIPFORMAT[nCount];
+    for ( ULONG n = 0; n < nCount; n++ ) {
+        m_formats[n] = formats[n].GetFormatId();
+    }
 }
 
 STDMETHODIMP wxIEnumFORMATETC::Next(ULONG      celt,
                                     FORMATETC *rgelt,
                                     ULONG     *pceltFetched)
 {
-  wxLogTrace(wxT("wxIEnumFORMATETC::Next"));
+    wxLogTrace(wxTRACE_OleCalls, wxT("wxIEnumFORMATETC::Next"));
 
-  if ( celt > 1 )
-    return S_FALSE;
+    if ( celt > 1 ) {
+        // we only return 1 element at a time - mainly because I'm too lazy to
+        // implement something which you're never asked for anyhow
+        return S_FALSE;
+    }
 
-  if ( m_nCurrent == 0 ) {
-    *rgelt = m_format;
-    m_nCurrent++;
+    if ( m_nCurrent < m_nCount ) {
+        FORMATETC format;
+        format.cfFormat = m_formats[m_nCurrent++];
+        format.ptd      = NULL;
+        format.dwAspect = DVASPECT_CONTENT;
+        format.lindex   = -1;
+        format.tymed    = TYMED_HGLOBAL;
+        *rgelt = format;
 
-    return S_OK;
-  }
-  else
-    return S_FALSE;
+        return S_OK;
+    }
+    else {
+        // bad index
+        return S_FALSE;
+    }
 }
 
 STDMETHODIMP wxIEnumFORMATETC::Skip(ULONG celt)
 {
-  wxLogTrace(wxT("wxIEnumFORMATETC::Skip"));
+    wxLogTrace(wxTRACE_OleCalls, wxT("wxIEnumFORMATETC::Skip"));
 
-  if ( m_nCurrent == 0 )
-    m_nCurrent++;
+    m_nCurrent += celt;
+    if ( m_nCurrent < m_nCount )
+        return S_OK;
 
-  return S_FALSE;
+    // no, can't skip this many elements
+    m_nCurrent -= celt;
+
+    return S_FALSE;
 }
 
 STDMETHODIMP wxIEnumFORMATETC::Reset()
 {
-  wxLogTrace(wxT("wxIEnumFORMATETC::Reset"));
+    wxLogTrace(wxTRACE_OleCalls, wxT("wxIEnumFORMATETC::Reset"));
 
-  m_nCurrent = 0;
+    m_nCurrent = 0;
 
-  return S_OK;
+    return S_OK;
 }
 
 STDMETHODIMP wxIEnumFORMATETC::Clone(IEnumFORMATETC **ppenum)
 {
-  wxLogTrace(wxT("wxIEnumFORMATETC::Clone"));
-
-  wxIEnumFORMATETC *pNew = new wxIEnumFORMATETC(m_format.cfFormat);
-  pNew->AddRef();
-  *ppenum = pNew;
+    wxLogTrace(wxTRACE_OleCalls, wxT("wxIEnumFORMATETC::Clone"));
+
+    // unfortunately, we can't reuse the code in ctor - types are different
+    wxIEnumFORMATETC *pNew = new wxIEnumFORMATETC(NULL, 0);
+    pNew->m_nCount = m_nCount;
+    pNew->m_formats = new CLIPFORMAT[m_nCount];
+    for ( ULONG n = 0; n < m_nCount; n++ ) {
+        pNew->m_formats[n] = m_formats[n];
+    }
+    pNew->AddRef();
+    *ppenum = pNew;
 
-  return S_OK;
+    return S_OK;
 }
 
 // ----------------------------------------------------------------------------
@@ -215,155 +269,340 @@ STDMETHODIMP wxIEnumFORMATETC::Clone(IEnumFORMATETC **ppenum)
 // ----------------------------------------------------------------------------
 
 BEGIN_IID_TABLE(wxIDataObject)
-  ADD_IID(Unknown)
-  ADD_IID(DataObject)
+    ADD_IID(Unknown)
+    ADD_IID(DataObject)
 END_IID_TABLE;
 
 IMPLEMENT_IUNKNOWN_METHODS(wxIDataObject)
 
 wxIDataObject::wxIDataObject(wxDataObject *pDataObject)
 {
-  m_cRef = 0;
-  m_pDataObject = pDataObject;
+    m_cRef = 0;
+    m_pDataObject = pDataObject;
+    m_mustDelete = FALSE;
+}
+
+wxIDataObject::~wxIDataObject()
+{
+    if ( m_mustDelete )
+    {
+        delete m_pDataObject;
+    }
 }
 
 // get data functions
 STDMETHODIMP wxIDataObject::GetData(FORMATETC *pformatetcIn, STGMEDIUM *pmedium)
 {
-  wxLogTrace(wxT("wxIDataObject::GetData"));
+    wxLogTrace(wxTRACE_OleCalls, wxT("wxIDataObject::GetData"));
 
-  // is data is in our format?
-  HRESULT hr = QueryGetData(pformatetcIn);
-  if ( FAILED(hr) )
-    return hr;
+    // is data is in our format?
+    HRESULT hr = QueryGetData(pformatetcIn);
+    if ( FAILED(hr) )
+        return hr;
 
-  // alloc memory
-  HGLOBAL hGlobal = GlobalAlloc(GMEM_MOVEABLE | GMEM_SHARE,
-                                m_pDataObject->GetDataSize());
-  if ( hGlobal == NULL ) {
-    wxLogLastError("GlobalAlloc");
-    return E_OUTOFMEMORY;
-  }
+    // for the bitmaps and metafiles we use the handles instead of global memory
+    // to pass the data
+    wxDataFormat format = (wxDataFormatId)pformatetcIn->cfFormat;
 
-  // copy data
-  pmedium->tymed          = TYMED_HGLOBAL;
-  pmedium->hGlobal        = hGlobal;
-  pmedium->pUnkForRelease = NULL;
+    switch ( format )
+    {
+        case wxDF_BITMAP:
+            pmedium->tymed = TYMED_GDI;
+            break;
+
+        case wxDF_METAFILE:
+            pmedium->tymed = TYMED_MFPICT;
+            break;
+
+        default:
+            // alloc memory
+            size_t size = m_pDataObject->GetDataSize(format);
+            if ( !size ) {
+                // it probably means that the method is just not implemented
+                wxLogDebug(wxT("Invalid data size - can't be 0"));
+
+                return DV_E_FORMATETC;
+            }
+
+            if ( !format.IsStandard() ) {
+                // for custom formats, put the size with the data - alloc the
+                // space for it
+                size += sizeof(size_t);
+            }
+
+            HGLOBAL hGlobal = GlobalAlloc(GMEM_MOVEABLE | GMEM_SHARE, size);
+            if ( hGlobal == NULL ) {
+                wxLogLastError("GlobalAlloc");
+                return E_OUTOFMEMORY;
+            }
+
+            // copy data
+            pmedium->tymed   = TYMED_HGLOBAL;
+            pmedium->hGlobal = hGlobal;
+    }
 
-  hr = GetDataHere(pformatetcIn, pmedium);
-  if ( FAILED(hr) ) {
-    GlobalFree(hGlobal);
-    return hr;
-  }
+    pmedium->pUnkForRelease = NULL;
 
-  return S_OK;
+    // do copy the data
+    hr = GetDataHere(pformatetcIn, pmedium);
+    if ( FAILED(hr) ) {
+        // free resources we allocated
+        if ( pmedium->tymed == TYMED_HGLOBAL ) {
+            GlobalFree(pmedium->hGlobal);
+        }
+
+        return hr;
+    }
+
+    return S_OK;
 }
 
 STDMETHODIMP wxIDataObject::GetDataHere(FORMATETC *pformatetc,
                                         STGMEDIUM *pmedium)
 {
-  wxLogTrace(wxT("wxIDataObject::GetDataHere"));
-
-  // put data in caller provided medium
-  if ( pmedium->tymed != TYMED_HGLOBAL )
-    return DV_E_TYMED;
-
-  // copy data
-  void *pBuf = GlobalLock(pmedium->hGlobal);
-  if ( pBuf == NULL ) {
-    wxLogLastError(wxT("GlobalLock"));
-    return E_OUTOFMEMORY;
-  }
-
-  m_pDataObject->GetDataHere(pBuf);
+    wxLogTrace(wxTRACE_OleCalls, wxT("wxIDataObject::GetDataHere"));
 
-  GlobalUnlock(pmedium->hGlobal);
+    // put data in caller provided medium
+    switch ( pmedium->tymed )
+    {
+        case TYMED_GDI:
+            if ( !m_pDataObject->GetDataHere(wxDF_BITMAP, &pmedium->hBitmap) )
+                return E_UNEXPECTED;
+            break;
+
+        case TYMED_MFPICT:
+            // this should be copied on bitmaps - but I don't have time for
+            // this now
+            wxFAIL_MSG(wxT("TODO - no support for metafiles in wxDataObject"));
+            break;
+
+        case TYMED_HGLOBAL:
+            {
+                // copy data
+                HGLOBAL hGlobal = pmedium->hGlobal;
+                void *pBuf = GlobalLock(hGlobal);
+                if ( pBuf == NULL ) {
+                    wxLogLastError(wxT("GlobalLock"));
+                    return E_OUTOFMEMORY;
+                }
+
+                if ( !wxDataFormat(pformatetc->cfFormat).IsStandard() ) {
+                    // for custom formats, put the size with the data
+                    size_t *p = (size_t *)pBuf;
+                    *p++ = GlobalSize(hGlobal);
+                    pBuf = p;
+                }
+
+                wxDataFormat format = pformatetc->cfFormat;
+                if ( !m_pDataObject->GetDataHere(format, pBuf) )
+                    return E_UNEXPECTED;
+
+                GlobalUnlock(hGlobal);
+            }
+            break;
+
+        default:
+            return DV_E_TYMED;
+    }
 
-  return S_OK;
+    return S_OK;
 }
 
-// set data functions (not implemented)
+// set data functions
 STDMETHODIMP wxIDataObject::SetData(FORMATETC *pformatetc,
                                     STGMEDIUM *pmedium,
                                     BOOL       fRelease)
 {
-  wxLogTrace(wxT("wxIDataObject::SetData"));
-  return E_NOTIMPL;
+    wxLogTrace(wxTRACE_OleCalls, wxT("wxIDataObject::SetData"));
+
+    switch ( pmedium->tymed )
+    {
+        case TYMED_GDI:
+            m_pDataObject->SetData(wxDF_BITMAP, 0, &pmedium->hBitmap);
+            break;
+
+        case TYMED_MFPICT:
+            // this should be copied on bitmaps - but I don't have time for
+            // this now
+            wxFAIL_MSG(wxT("TODO - no support for metafiles in wxDataObject"));
+            break;
+
+        case TYMED_HGLOBAL:
+            {
+                // copy data
+                void *pBuf = GlobalLock(pmedium->hGlobal);
+                if ( pBuf == NULL ) {
+                    wxLogLastError("GlobalLock");
+
+                    return E_OUTOFMEMORY;
+                }
+
+                // we've got a problem with SetData() here because the base
+                // class version requires the size parameter which we don't
+                // have anywhere in OLE data transfer - so we need to
+                // synthetise it for known formats and we suppose that all data
+                // in custom formats starts with a DWORD containing the size
+                size_t size;
+                switch ( pformatetc->cfFormat )
+                {
+                    case CF_TEXT:
+                    case CF_OEMTEXT:
+                        size = strlen((const char *)pBuf);
+                        break;
+
+                    case CF_UNICODETEXT:
+                        size = wcslen((const wchar_t *)pBuf);
+                        break;
+
+                    case CF_BITMAP:
+                    case CF_HDROP:
+                        // these formats don't use size at all, anyhow (but
+                        // pass data by handle, which is always a single DWORD)
+                        size = 0;
+                        break;
+
+                    default:
+                        {
+                            // we suppose that the size precedes the data
+                            size_t *p = (size_t *)pBuf;
+                            size = *p++;
+                            pBuf = p;
+                        }
+                }
+
+                wxDataFormat format = pformatetc->cfFormat;
+                bool ok = m_pDataObject->SetData(format, size, pBuf);
+
+                GlobalUnlock(pmedium->hGlobal);
+
+                if ( !ok ) {
+                    return E_UNEXPECTED;
+                }
+            }
+            break;
+
+        default:
+            return DV_E_TYMED;
+    }
+
+    if ( fRelease ) {
+        // we own the medium, so we must release it - but do *not* free the
+        // bitmap handle fi we have it because we have copied it elsewhere
+        if ( pmedium->tymed == TYMED_GDI )
+        {
+            pmedium->hBitmap = 0;
+        }
+
+        ReleaseStgMedium(pmedium);
+    }
+
+    return S_OK;
 }
 
 // information functions
 STDMETHODIMP wxIDataObject::QueryGetData(FORMATETC *pformatetc)
 {
-  // do we accept data in this format?
-  if ( pformatetc == NULL ) {
-    wxLogTrace(wxT("wxIDataObject::QueryGetData: invalid ptr."));
-    return E_INVALIDARG;
-  }
+    // do we accept data in this format?
+    if ( pformatetc == NULL ) {
+        wxLogTrace(wxTRACE_OleCalls,
+                   wxT("wxIDataObject::QueryGetData: invalid ptr."));
 
-  // the only one allowed by current COM implementation
-  if ( pformatetc->lindex != -1 ) {
-    wxLogTrace(wxT("wxIDataObject::QueryGetData: bad lindex %d"),
-               pformatetc->lindex);
-    return DV_E_LINDEX;
-  }
+        return E_INVALIDARG;
+    }
 
-  // we don't support anything other (THUMBNAIL, ICON, DOCPRINT...)
-  if ( pformatetc->dwAspect != DVASPECT_CONTENT ) {
-    wxLogTrace(wxT("wxIDataObject::QueryGetData: bad dwAspect %d"),
-               pformatetc->dwAspect);
-    return DV_E_DVASPECT;
-  }
+    // the only one allowed by current COM implementation
+    if ( pformatetc->lindex != -1 ) {
+        wxLogTrace(wxTRACE_OleCalls,
+                   wxT("wxIDataObject::QueryGetData: bad lindex %d"),
+                   pformatetc->lindex);
 
-  // @@ we only transfer data by global memory (bad for large amounts of it!)
-  if ( !(pformatetc->tymed & TYMED_HGLOBAL) ) {
-    wxLogTrace(wxT("wxIDataObject::QueryGetData: %s != TYMED_HGLOBAL."),
-               GetTymedName(pformatetc->tymed));
-    return DV_E_TYMED;
-  }
+        return DV_E_LINDEX;
+    }
+
+    // we don't support anything other (THUMBNAIL, ICON, DOCPRINT...)
+    if ( pformatetc->dwAspect != DVASPECT_CONTENT ) {
+        wxLogTrace(wxTRACE_OleCalls,
+                   wxT("wxIDataObject::QueryGetData: bad dwAspect %d"),
+                   pformatetc->dwAspect);
+
+        return DV_E_DVASPECT;
+    }
+
+    // and now check the type of data requested
+    wxDataFormat format = pformatetc->cfFormat;
+    if ( m_pDataObject->IsSupportedFormat(format) ) {
+#ifdef __WXDEBUG__
+        wxLogTrace(wxTRACE_OleCalls, wxT("wxIDataObject::QueryGetData: %s ok"),
+                   wxDataObject::GetFormatName(format));
+#endif // Debug
+    }
+    else {
+        wxLogTrace(wxTRACE_OleCalls,
+                   wxT("wxIDataObject::QueryGetData: %s unsupported"),
+                   wxDataObject::GetFormatName(format));
+
+        return DV_E_FORMATETC;
+    }
+
+    // we only transfer data by global memory, except for some particular cases
+    DWORD tymed = pformatetc->tymed;
+    if ( (format == wxDF_BITMAP && !(tymed & TYMED_GDI)) &&
+         !(tymed & TYMED_HGLOBAL) ) {
+        // it's not what we're waiting for
+#ifdef __WXDEBUG__
+        wxLogTrace(wxTRACE_OleCalls,
+                   wxT("wxIDataObject::QueryGetData: %s != %s"),
+                   GetTymedName(tymed),
+                   GetTymedName(format == wxDF_BITMAP ? TYMED_GDI
+                                                      : TYMED_HGLOBAL));
+#endif // Debug
+
+        return DV_E_TYMED;
+    }
 
-  // and now check the type of data requested
-  if ( m_pDataObject->IsSupportedFormat((wxDataFormatId)pformatetc->cfFormat) ) {
-    wxLogTrace(wxT("wxIDataObject::QueryGetData: %s ok"),
-               wxDataObject::GetFormatName((wxDataFormatId)pformatetc->cfFormat));
     return S_OK;
-  }
-  else {
-    wxLogTrace(wxT("wxIDataObject::QueryGetData: %s unsupported"),
-               wxDataObject::GetFormatName((wxDataFormatId)pformatetc->cfFormat));
-    return DV_E_FORMATETC;
-  }
 }
 
 STDMETHODIMP wxIDataObject::GetCanonicalFormatEtc(FORMATETC *pFormatetcIn,
                                                   FORMATETC *pFormatetcOut)
 {
-  wxLogTrace(wxT("wxIDataObject::GetCanonicalFormatEtc"));
+    wxLogTrace(wxTRACE_OleCalls, wxT("wxIDataObject::GetCanonicalFormatEtc"));
+
+    // TODO we might want something better than this trivial implementation here
+    if ( pFormatetcOut != NULL )
+        pFormatetcOut->ptd = NULL;
 
-  // @@ implementation is trivial, we might want something better here
-  if ( pFormatetcOut != NULL )
-    pFormatetcOut->ptd = NULL;
-  return DATA_S_SAMEFORMATETC;
+    return DATA_S_SAMEFORMATETC;
 }
 
-STDMETHODIMP wxIDataObject::EnumFormatEtc(DWORD dwDirection,
+STDMETHODIMP wxIDataObject::EnumFormatEtc(DWORD dwDir,
                                           IEnumFORMATETC **ppenumFormatEtc)
 {
-  wxLogTrace(wxT("wxIDataObject::EnumFormatEtc"));
+    wxLogTrace(wxTRACE_OleCalls, wxT("wxIDataObject::EnumFormatEtc"));
 
-  if ( dwDirection == DATADIR_SET ) {
-    // we don't allow setting of data anyhow
-    return E_NOTIMPL;
-  }
+    wxDataObject::Direction dir = dwDir == DATADIR_GET ? wxDataObject::Get
+                                                       : wxDataObject::Set;
+
+    size_t nFormatCount = m_pDataObject->GetFormatCount(dir);
+    wxDataFormat format, *formats;
+    formats = nFormatCount == 1 ? &format : new wxDataFormat[nFormatCount];
+    m_pDataObject->GetAllFormats(formats, dir);
+
+    wxIEnumFORMATETC *pEnum = new wxIEnumFORMATETC(formats, nFormatCount);
+    pEnum->AddRef();
+    *ppenumFormatEtc = pEnum;
 
-  wxIEnumFORMATETC *pEnum =
-    new wxIEnumFORMATETC(m_pDataObject->GetPreferredFormat());
-  pEnum->AddRef();
-  *ppenumFormatEtc = pEnum;
+    if ( formats != &format ) {
+        delete [] formats;
+    }
 
-  return S_OK;
+    return S_OK;
 }
 
+// ----------------------------------------------------------------------------
 // advise sink functions (not implemented)
+// ----------------------------------------------------------------------------
+
 STDMETHODIMP wxIDataObject::DAdvise(FORMATETC   *pformatetc,
                                     DWORD        advf,
                                     IAdviseSink *pAdvSink,
@@ -388,18 +627,49 @@ STDMETHODIMP wxIDataObject::EnumDAdvise(IEnumSTATDATA **ppenumAdvise)
 
 wxDataObject::wxDataObject()
 {
-  m_pIDataObject = new wxIDataObject(this);
-  m_pIDataObject->AddRef();
+    m_pIDataObject = new wxIDataObject(this);
+    m_pIDataObject->AddRef();
 }
 
 wxDataObject::~wxDataObject()
 {
-  m_pIDataObject->Release();
+    ReleaseInterface(m_pIDataObject);
+}
+
+void wxDataObject::SetAutoDelete()
+{
+    ((wxIDataObject *)m_pIDataObject)->SetDeleteFlag();
+    m_pIDataObject->Release();
+
+    // so that the dtor doesnt' crash
+    m_pIDataObject = NULL;
+}
+
+bool wxDataObject::IsSupportedFormat(const wxDataFormat& format) const
+{
+    size_t nFormatCount = GetFormatCount();
+    if ( nFormatCount == 1 ) {
+        return format == GetPreferredFormat();
+    }
+    else {
+        wxDataFormat *formats = new wxDataFormat[nFormatCount];
+        GetAllFormats(formats);
+
+        size_t n;
+        for ( n = 0; n < nFormatCount; n++ ) {
+            if ( formats[n] == format )
+                break;
+        }
+
+        delete [] formats;
+
+        // found?
+        return n < nFormatCount;
+    }
 }
 
 const char *wxDataObject::GetFormatName(wxDataFormat format)
 {
-#ifdef __WXDEBUG__
   // case 'xxx' is not a valid value for switch of enum 'wxDataFormat'
   #ifdef __VISUALC__
     #pragma warning(disable:4063)
@@ -424,81 +694,343 @@ const char *wxDataObject::GetFormatName(wxDataFormat format)
     case CF_HDROP:        return "CF_HDROP";
     case CF_LOCALE:       return "CF_LOCALE";
     default:
-      sprintf(s_szBuf, "clipboard format %d (unknown)", format);
+      sprintf(s_szBuf, "clipboard format 0x%x (unknown)", format);
       return s_szBuf;
   }
 
   #ifdef __VISUALC__
     #pragma warning(default:4063)
   #endif // VC++
+}
 
-#else // !Debug
-  return "";
-#endif // Debug
+// ----------------------------------------------------------------------------
+// wxBitmapDataObject supports CF_DIB format
+// ----------------------------------------------------------------------------
+
+size_t wxBitmapDataObject::GetDataSize() const
+{
+    return wxConvertBitmapToDIB(NULL, GetBitmap());
+}
+
+bool wxBitmapDataObject::GetDataHere(void *buf) const
+{
+    return wxConvertBitmapToDIB((BITMAPINFO *)buf, GetBitmap()) != 0;
+}
+
+bool wxBitmapDataObject::SetData(size_t len, const void *buf)
+{
+    wxBitmap bitmap(wxConvertDIBToBitmap((const BITMAPINFO *)buf));
+
+    if ( !bitmap.Ok() ) {
+        wxFAIL_MSG(wxT("pasting/dropping invalid bitmap"));
+
+        return FALSE;
+    }
+
+    SetBitmap(bitmap);
+
+    return TRUE;
 }
 
 // ----------------------------------------------------------------------------
-// wxPrivateDataObject
+// wxBitmapDataObject2 supports CF_BITMAP format
 // ----------------------------------------------------------------------------
 
-wxPrivateDataObject::wxPrivateDataObject()
+// the bitmaps aren't passed by value as other types of data (i.e. by copying
+// the data into a global memory chunk and passing it to the clipboard or
+// another application or whatever), but by handle, so these generic functions
+// don't make much sense to them.
+
+size_t wxBitmapDataObject2::GetDataSize() const
+{
+    return 0;
+}
+
+bool wxBitmapDataObject2::GetDataHere(void *pBuf) const
 {
-    m_size = 0;
-    m_data = NULL;
+    // we put a bitmap handle into pBuf
+    *(WXHBITMAP *)pBuf = GetBitmap().GetHBITMAP();
+
+    return TRUE;
 }
 
-void wxPrivateDataObject::Free()
+bool wxBitmapDataObject2::SetData(size_t len, const void *pBuf)
 {
-    if ( m_data )
-        free(m_data);
+    HBITMAP hbmp = *(HBITMAP *)pBuf;
+
+    BITMAP bmp;
+    if ( !GetObject(hbmp, sizeof(BITMAP), &bmp) )
+    {
+        wxLogLastError("GetObject(HBITMAP)");
+    }
+
+    wxBitmap bitmap(bmp.bmWidth, bmp.bmHeight, bmp.bmPlanes);
+    bitmap.SetHBITMAP((WXHBITMAP)hbmp);
+
+    if ( !bitmap.Ok() ) {
+        wxFAIL_MSG(wxT("pasting/dropping invalid bitmap"));
+
+        return FALSE;
+    }
+
+    SetBitmap(bitmap);
+
+    return TRUE;
 }
 
-void wxPrivateDataObject::SetData( const void *data, size_t size )
+#if 0
+
+size_t wxBitmapDataObject::GetDataSize(const wxDataFormat& format) const
 {
-    Free();
+    if ( format.GetFormatId() == CF_DIB )
+    {
+        // create the DIB
+        ScreenHDC hdc;
 
-    m_size = size;
-    m_data = malloc(size);
+        // shouldn't be selected into a DC or GetDIBits() would fail
+        wxASSERT_MSG( !m_bitmap.GetSelectedInto(),
+                      wxT("can't copy bitmap selected into wxMemoryDC") );
 
-    memcpy( m_data, data, size );
+        // first get the info
+        BITMAPINFO bi;
+        if ( !GetDIBits(hdc, (HBITMAP)m_bitmap.GetHBITMAP(), 0, 0,
+                        NULL, &bi, DIB_RGB_COLORS) )
+        {
+            wxLogLastError("GetDIBits(NULL)");
+
+            return 0;
+        }
+
+        return sizeof(BITMAPINFO) + bi.bmiHeader.biSizeImage;
+    }
+    else // CF_BITMAP
+    {
+        // no data to copy - we don't pass HBITMAP via global memory
+        return 0;
+    }
 }
 
-void wxPrivateDataObject::WriteData( void *dest ) const
+bool wxBitmapDataObject::GetDataHere(const wxDataFormat& format,
+                                     void *pBuf) const
 {
-    WriteData( m_data, dest );
+    wxASSERT_MSG( m_bitmap.Ok(), wxT("copying invalid bitmap") );
+
+    HBITMAP hbmp = (HBITMAP)m_bitmap.GetHBITMAP();
+    if ( format.GetFormatId() == CF_DIB )
+    {
+        // create the DIB
+        ScreenHDC hdc;
+
+        // shouldn't be selected into a DC or GetDIBits() would fail
+        wxASSERT_MSG( !m_bitmap.GetSelectedInto(),
+                      wxT("can't copy bitmap selected into wxMemoryDC") );
+
+        // first get the info
+        BITMAPINFO *pbi = (BITMAPINFO *)pBuf;
+        if ( !GetDIBits(hdc, hbmp, 0, 0, NULL, pbi, DIB_RGB_COLORS) )
+        {
+            wxLogLastError("GetDIBits(NULL)");
+
+            return 0;
+        }
+
+        // and now copy the bits
+        if ( !GetDIBits(hdc, hbmp, 0, pbi->bmiHeader.biHeight, pbi + 1,
+                        pbi, DIB_RGB_COLORS) )
+        {
+            wxLogLastError("GetDIBits");
+
+            return FALSE;
+        }
+    }
+    else // CF_BITMAP
+    {
+        // we put a bitmap handle into pBuf
+        *(HBITMAP *)pBuf = hbmp;
+    }
+
+    return TRUE;
 }
 
-size_t wxPrivateDataObject::GetSize() const
+bool wxBitmapDataObject::SetData(const wxDataFormat& format,
+                                 size_t size, const void *pBuf)
 {
-    return m_size;
+    HBITMAP hbmp;
+    if ( format.GetFormatId() == CF_DIB )
+    {
+        // here we get BITMAPINFO struct followed by the actual bitmap bits and
+        // BITMAPINFO starts with BITMAPINFOHEADER followed by colour info
+        ScreenHDC hdc;
+
+        BITMAPINFO *pbmi = (BITMAPINFO *)pBuf;
+        BITMAPINFOHEADER *pbmih = &pbmi->bmiHeader;
+        hbmp = CreateDIBitmap(hdc, pbmih, CBM_INIT,
+                              pbmi + 1, pbmi, DIB_RGB_COLORS);
+        if ( !hbmp )
+        {
+            wxLogLastError("CreateDIBitmap");
+        }
+
+        m_bitmap.SetWidth(pbmih->biWidth);
+        m_bitmap.SetHeight(pbmih->biHeight);
+    }
+    else // CF_BITMAP
+    {
+        // it's easy with bitmaps: we pass them by handle
+        hbmp = *(HBITMAP *)pBuf;
+
+        BITMAP bmp;
+        if ( !GetObject(hbmp, sizeof(BITMAP), &bmp) )
+        {
+            wxLogLastError("GetObject(HBITMAP)");
+        }
+
+        m_bitmap.SetWidth(bmp.bmWidth);
+        m_bitmap.SetHeight(bmp.bmHeight);
+        m_bitmap.SetDepth(bmp.bmPlanes);
+    }
+
+    m_bitmap.SetHBITMAP((WXHBITMAP)hbmp);
+
+    wxASSERT_MSG( m_bitmap.Ok(), wxT("pasting invalid bitmap") );
+
+    return TRUE;
 }
 
-void wxPrivateDataObject::WriteData( const void *data, void *dest ) const
+#endif // 0
+
+// ----------------------------------------------------------------------------
+// wxFileDataObject
+// ----------------------------------------------------------------------------
+
+bool wxFileDataObject::SetData(size_t WXUNUSED(size), const void *pData)
 {
-    memcpy( dest, data, GetSize() );
+    m_filenames.Empty();
+
+    // the documentation states that the first member of DROPFILES structure is
+    // a "DWORD offset of double NUL terminated file list". What they mean by
+    // this (I wonder if you see it immediately) is that the list starts at
+    // ((char *)&(pDropFiles.pFiles)) + pDropFiles.pFiles. We're also advised
+    // to use DragQueryFile to work with this structure, but not told where and
+    // how to get HDROP.
+    HDROP hdrop = (HDROP)pData;   // NB: it works, but I'm not sure about it
+
+    // get number of files (magic value -1)
+    UINT nFiles = ::DragQueryFile(hdrop, (unsigned)-1, NULL, 0u);
+
+    // for each file get the length, allocate memory and then get the name
+    wxString str;
+    UINT len, n;
+    for ( n = 0; n < nFiles; n++ ) {
+        // +1 for terminating NUL
+        len = ::DragQueryFile(hdrop, n, NULL, 0) + 1;
+
+        UINT len2 = ::DragQueryFile(hdrop, n, str.GetWriteBuf(len), len);
+        str.UngetWriteBuf();
+        m_filenames.Add(str);
+
+        if ( len2 != len - 1 ) {
+            wxLogDebug(wxT("In wxFileDropTarget::OnDrop DragQueryFile returned"
+                           " %d characters, %d expected."), len2, len - 1);
+        }
+    }
+
+    return TRUE;
 }
 
 // ----------------------------------------------------------------------------
 // private functions
 // ----------------------------------------------------------------------------
-static const char *GetTymedName(DWORD tymed)
+
+// otherwise VC++ would give here:
+//  "local variable 'bi' may be used without having been initialized"
+// even though in fact it may not
+#ifdef __VISUALC__
+    #pragma warning(disable:4701)
+#endif // __VISUALC__
+
+size_t wxConvertBitmapToDIB(BITMAPINFO *pbi, const wxBitmap& bitmap)
 {
-  static char s_szBuf[128];
-  switch ( tymed ) {
-    case TYMED_HGLOBAL:   return "TYMED_HGLOBAL";
-    case TYMED_FILE:      return "TYMED_FILE";
-    case TYMED_ISTREAM:   return "TYMED_ISTREAM";
-    case TYMED_ISTORAGE:  return "TYMED_ISTORAGE";
-    case TYMED_GDI:       return "TYMED_GDI";
-    case TYMED_MFPICT:    return "TYMED_MFPICT";
-    case TYMED_ENHMF:     return "TYMED_ENHMF";
-    default:
-      sprintf(s_szBuf, "type of media format %d (unknown)", tymed);
-      return s_szBuf;
-  }
+    // shouldn't be selected into a DC or GetDIBits() would fail
+    wxASSERT_MSG( !bitmap.GetSelectedInto(),
+                  wxT("can't copy bitmap selected into wxMemoryDC") );
+
+    HBITMAP hbmp = (HBITMAP)bitmap.GetHBITMAP();
+
+    BITMAPINFO bi;
+
+    // first get the info
+    ScreenHDC hdc;
+    if ( !GetDIBits(hdc, hbmp, 0, 0, NULL, pbi ? pbi : &bi, DIB_RGB_COLORS) )
+    {
+        wxLogLastError("GetDIBits(NULL)");
+
+        return 0;
+    }
+
+    if ( !pbi )
+    {
+        // we were only asked for size needed for the buffer, not to actually
+        // copy the data
+        return sizeof(BITMAPINFO) + bi.bmiHeader.biSizeImage;
+    }
+
+    // and now copy the bits
+    if ( !GetDIBits(hdc, hbmp, 0, pbi->bmiHeader.biHeight, pbi + 1,
+                    pbi, DIB_RGB_COLORS) )
+    {
+        wxLogLastError("GetDIBits");
+
+        return 0;
+    }
+
+    return sizeof(BITMAPINFO) + pbi->bmiHeader.biSizeImage;
 }
 
-// TODO: OLE parts of wxBitmap/File/MetafileDataObject
+#ifdef __VISUALC__
+    #pragma warning(default:4701)
+#endif // __VISUALC__
 
-#endif
+wxBitmap wxConvertDIBToBitmap(const BITMAPINFO *pbmi)
+{
+    // here we get BITMAPINFO struct followed by the actual bitmap bits and
+    // BITMAPINFO starts with BITMAPINFOHEADER followed by colour info
+    const BITMAPINFOHEADER *pbmih = &pbmi->bmiHeader;
+
+    ScreenHDC hdc;
+    HBITMAP hbmp = CreateDIBitmap(hdc, pbmih, CBM_INIT,
+                                  pbmi + 1, pbmi, DIB_RGB_COLORS);
+    if ( !hbmp )
+    {
+        wxLogLastError("CreateDIBitmap");
+    }
+
+    wxBitmap bitmap(pbmih->biWidth, pbmih->biHeight, pbmih->biBitCount);
+    bitmap.SetHBITMAP((WXHBITMAP)hbmp);
+
+    return bitmap;
+}
+
+#ifdef __WXDEBUG__
+
+static const wxChar *GetTymedName(DWORD tymed)
+{
+    static wxChar s_szBuf[128];
+    switch ( tymed ) {
+        case TYMED_HGLOBAL:   return wxT("TYMED_HGLOBAL");
+        case TYMED_FILE:      return wxT("TYMED_FILE");
+        case TYMED_ISTREAM:   return wxT("TYMED_ISTREAM");
+        case TYMED_ISTORAGE:  return wxT("TYMED_ISTORAGE");
+        case TYMED_GDI:       return wxT("TYMED_GDI");
+        case TYMED_MFPICT:    return wxT("TYMED_MFPICT");
+        case TYMED_ENHMF:     return wxT("TYMED_ENHMF");
+        default:
+            wxSprintf(s_szBuf, wxT("type of media format %d (unknown)"), tymed);
+            return s_szBuf;
+    }
+}
+
+#endif // Debug
+
+#endif // not using OLE at all