X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/cfe780fb99dc42d454625b1ea88ae687c54acbf1..b56c22469a1fb07c7e9ccde9334ccd96d4ee6155:/src/msw/ole/dataobj.cpp diff --git a/src/msw/ole/dataobj.cpp b/src/msw/ole/dataobj.cpp index fd57f7de78..30f46b1cf0 100644 --- a/src/msw/ole/dataobj.cpp +++ b/src/msw/ole/dataobj.cpp @@ -2,7 +2,7 @@ // Name: msw/ole/dataobj.cpp // Purpose: implementation of wx[I]DataObject class // Author: Vadim Zeitlin -// Modified by: +// Modified by: // Created: 10.05.98 // RCS-ID: $Id$ // Copyright: (c) 1998 Vadim Zeitlin @@ -27,145 +27,241 @@ #if defined(__BORLANDC__) #pragma hdrstop #endif +#ifndef WX_PRECOMP +#include "wx/intl.h" +#endif +#include "wx/defs.h" -#include +#if defined(__WIN32__) && !defined(__GNUWIN32__) || defined(wxUSE_NORLANDER_HEADERS) -#if defined(__WIN32__) && !defined(__GNUWIN32__) +#include "wx/log.h" +#include "wx/dataobj.h" -#include -#include -#include +#include +#ifdef wxUSE_NORLANDER_HEADERS + #include +#endif +#include #ifndef __WIN32__ #include #include #endif +#include "wx/msw/ole/oleutils.h" + // ---------------------------------------------------------------------------- // 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); - DECLARE_IUNKNOWN_METHODS; +private: + wxDataObject *m_pDataObject; // pointer to C++ class we belong to - // 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); + bool m_mustDelete; +}; + +// ---------------------------------------------------------------------------- +// small helper class for getting screen DC (we're working with bitmaps and +// DIBs here) +// ---------------------------------------------------------------------------- + +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; }; // ============================================================================ // implementation // ============================================================================ +// ---------------------------------------------------------------------------- +// wxDataFormat +// ---------------------------------------------------------------------------- + +void wxDataFormat::SetId(const wxChar *format) +{ + m_format = ::RegisterClipboardFormat(format); + if ( !m_format ) + { + wxLogError(_("Couldn't register clipboard format '%s'."), format); + } +} + +wxString wxDataFormat::GetId() const +{ + static const int max = 256; + + wxString s; + + wxCHECK_MSG( !IsStandard(), s, + wxT("name of predefined format cannot be retrieved") ); + + int len = ::GetClipboardFormatName(m_format, s.GetWriteBuf(max), max); + s.UngetWriteBuf(); + + if ( !len ) + { + wxLogError(_("The clipboard format '%d' doesn't exist."), m_format); + } + + return s; +} + // ---------------------------------------------------------------------------- // wxIEnumFORMATETC // ---------------------------------------------------------------------------- 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("wxIEnumFORMATETC::Next"); - - if ( celt > 1 ) - return S_FALSE; - - if ( m_nCurrent == 0 ) { - *rgelt = m_format; - m_nCurrent++; - - return S_OK; - } - else - return S_FALSE; + wxLogTrace(wxTRACE_OleCalls, wxT("wxIEnumFORMATETC::Next")); + + 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 < 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 { + // bad index + return S_FALSE; + } } STDMETHODIMP wxIEnumFORMATETC::Skip(ULONG celt) { - wxLogTrace("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("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("wxIEnumFORMATETC::Clone"); + 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; - wxIEnumFORMATETC *pNew = new wxIEnumFORMATETC(m_format.cfFormat); - pNew->AddRef(); - *ppenum = pNew; - - return S_OK; + return S_OK; } // ---------------------------------------------------------------------------- @@ -173,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("wxIDataObject::GetData"); - - // 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; - } + wxLogTrace(wxTRACE_OleCalls, wxT("wxIDataObject::GetData")); + + // is data is in our format? + HRESULT hr = QueryGetData(pformatetcIn); + if ( FAILED(hr) ) + return hr; + + // for the bitmaps and metafiles we use the handles instead of global memory + // to pass the data + wxDataFormat format = (wxDataFormatId)pformatetcIn->cfFormat; + + 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; + } + + pmedium->pUnkForRelease = NULL; + + // 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; + } - // copy data - pmedium->tymed = TYMED_HGLOBAL; - pmedium->hGlobal = hGlobal; - pmedium->pUnkForRelease = NULL; - - hr = GetDataHere(pformatetcIn, pmedium); - if ( FAILED(hr) ) { - GlobalFree(hGlobal); - return hr; - } - - return S_OK; + return S_OK; } STDMETHODIMP wxIDataObject::GetDataHere(FORMATETC *pformatetc, STGMEDIUM *pmedium) { - wxLogTrace("wxIDataObject::GetDataHere"); - - // put data in caller provided medium - if ( pmedium->tymed != TYMED_HGLOBAL ) - return DV_E_TYMED; + wxLogTrace(wxTRACE_OleCalls, wxT("wxIDataObject::GetDataHere")); + + // 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; + } - // copy data - void *pBuf = GlobalLock(pmedium->hGlobal); - if ( pBuf == NULL ) { - wxLogLastError("GlobalLock"); - return E_OUTOFMEMORY; - } - - m_pDataObject->GetDataHere(pBuf); - - GlobalUnlock(pmedium->hGlobal); - - return S_OK; + return S_OK; } -// set data functions (not implemented) +// set data functions STDMETHODIMP wxIDataObject::SetData(FORMATETC *pformatetc, STGMEDIUM *pmedium, BOOL fRelease) { - wxLogTrace("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("wxIDataObject::QueryGetData: invalid ptr."); - return E_INVALIDARG; - } - - // the only one allowed by current COM implementation - if ( pformatetc->lindex != -1 ) { - wxLogTrace("wxIDataObject::QueryGetData: bad lindex %d", - pformatetc->lindex); - return DV_E_LINDEX; - } + // do we accept data in this format? + if ( pformatetc == NULL ) { + wxLogTrace(wxTRACE_OleCalls, + wxT("wxIDataObject::QueryGetData: invalid ptr.")); + + return E_INVALIDARG; + } + + // the only one allowed by current COM implementation + if ( pformatetc->lindex != -1 ) { + wxLogTrace(wxTRACE_OleCalls, + wxT("wxIDataObject::QueryGetData: bad lindex %d"), + pformatetc->lindex); + + 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; + } - // we don't support anything other (THUMBNAIL, ICON, DOCPRINT...) - if ( pformatetc->dwAspect != DVASPECT_CONTENT ) { - wxLogTrace("wxIDataObject::QueryGetData: bad dwAspect %d", - pformatetc->dwAspect); - return DV_E_DVASPECT; - } - - // @@ we only transfer data by global memory (bad for large amounts of it!) - if ( !(pformatetc->tymed & TYMED_HGLOBAL) ) { - wxLogTrace("wxIDataObject::QueryGetData: %s != TYMED_HGLOBAL.", - GetTymedName(pformatetc->tymed)); - return DV_E_TYMED; - } - - // and now check the type of data requested - if ( m_pDataObject->IsSupportedFormat(pformatetc->cfFormat) ) { - wxLogTrace("wxIDataObject::QueryGetData: %s ok", - wxDataObject::GetFormatName(pformatetc->cfFormat)); return S_OK; - } - else { - wxLogTrace("wxIDataObject::QueryGetData: %s unsupported", - wxDataObject::GetFormatName(pformatetc->cfFormat)); - return DV_E_FORMATETC; - } } STDMETHODIMP wxIDataObject::GetCanonicalFormatEtc(FORMATETC *pFormatetcIn, FORMATETC *pFormatetcOut) { - wxLogTrace("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("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, @@ -346,18 +627,54 @@ 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 __DEBUG__ + // case 'xxx' is not a valid value for switch of enum 'wxDataFormat' + #ifdef __VISUALC__ + #pragma warning(disable:4063) + #endif // VC++ + static char s_szBuf[128]; switch ( format ) { case CF_TEXT: return "CF_TEXT"; @@ -377,33 +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; } - #else - return ""; -#endif + + #ifdef __VISUALC__ + #pragma warning(default:4063) + #endif // VC++ +} + +// ---------------------------------------------------------------------------- +// 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; +} + +// ---------------------------------------------------------------------------- +// wxBitmapDataObject2 supports CF_BITMAP format +// ---------------------------------------------------------------------------- + +// 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 +{ + // we put a bitmap handle into pBuf + *(WXHBITMAP *)pBuf = GetBitmap().GetHBITMAP(); + + return TRUE; +} + +bool wxBitmapDataObject2::SetData(size_t len, const void *pBuf) +{ + 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; +} + +#if 0 + +size_t wxBitmapDataObject::GetDataSize(const wxDataFormat& format) const +{ + 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 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; + } +} + +bool wxBitmapDataObject::GetDataHere(const wxDataFormat& format, + void *pBuf) const +{ + 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; +} + +bool wxBitmapDataObject::SetData(const wxDataFormat& format, + size_t size, const void *pBuf) +{ + 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; +} + +#endif // 0 + +// ---------------------------------------------------------------------------- +// wxFileDataObject +// ---------------------------------------------------------------------------- + +bool wxFileDataObject::SetData(size_t WXUNUSED(size), const void *pData) +{ + 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; } -#endif +#ifdef __VISUALC__ + #pragma warning(default:4701) +#endif // __VISUALC__ + +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