src/msw/ole/droptgt.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // Name:        ole/droptgt.cpp
   3 // Purpose:     wxDropTarget implementation
   4 // Author:      Vadim Zeitlin
   5 // Modified by:
   6 // Created:
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) 1998 Vadim Zeitlin <zeitlin@dptmaths.ens-cachan.fr>
   9 // Licence:     wxWindows licence
  10 ///////////////////////////////////////////////////////////////////////////////
  11
  12 // ============================================================================
  13 // Declarations
  14 // ============================================================================
  15
  16 // ----------------------------------------------------------------------------
  17 // headers
  18 // ----------------------------------------------------------------------------
  19
  20 #if defined(__GNUG__) && !defined(NO_GCC_PRAGMA)
  21 #pragma implementation "droptgt.h"
  22 #endif
  23
  24 // For compilers that support precompilation, includes "wx.h".
  25 #include "wx/wxprec.h"
  26
  27 #if defined(__BORLANDC__)
  28 #pragma hdrstop
  29 #endif
  30
  31 #include "wx/setup.h"
  32
  33 #if wxUSE_OLE && wxUSE_DRAG_AND_DROP
  34
  35 #include "wx/msw/private.h"
  36 #include "wx/log.h"
  37
  38 #ifdef __WXWINCE__
  39     #include <winreg.h>
  40     #include <ole2.h>
  41 #endif
  42
  43 #ifdef __WIN32__
  44     #if !defined(__GNUWIN32__) || wxUSE_NORLANDER_HEADERS
  45         #if wxCHECK_W32API_VERSION( 1, 0 )
  46             #include "wx/msw/wrapwin.h"
  47         #endif
  48         #include <shlobj.h>            // for DROPFILES structure
  49     #endif
  50 #else
  51     #include <shellapi.h>
  52 #endif
  53
  54 #include "wx/dnd.h"
  55
  56 #include "wx/msw/ole/oleutils.h"
  57
  58 // ----------------------------------------------------------------------------
  59 // IDropTarget interface: forward all interesting things to wxDropTarget
  60 // (the name is unfortunate, but wx_I_DropTarget is not at all the same thing
  61 //  as wxDropTarget which is 'public' class, while this one is private)
  62 // ----------------------------------------------------------------------------
  63
  64 class wxIDropTarget : public IDropTarget
  65 {
  66 public:
  67     wxIDropTarget(wxDropTarget *p);
  68     virtual ~wxIDropTarget();
  69
  70     // accessors for wxDropTarget
  71     void SetHwnd(HWND hwnd) { m_hwnd = hwnd; }
  72
  73     // IDropTarget methods
  74     STDMETHODIMP DragEnter(LPDATAOBJECT, DWORD, POINTL, LPDWORD);
  75     STDMETHODIMP DragOver(DWORD, POINTL, LPDWORD);
  76     STDMETHODIMP DragLeave();
  77     STDMETHODIMP Drop(LPDATAOBJECT, DWORD, POINTL, LPDWORD);
  78
  79     DECLARE_IUNKNOWN_METHODS;
  80
  81 protected:
  82     IDataObject  *m_pIDataObject; // !NULL between DragEnter and DragLeave/Drop
  83     wxDropTarget *m_pTarget;      // the real target (we're just a proxy)
  84
  85     HWND          m_hwnd;         // window we're associated with
  86
  87     // get default drop effect for given keyboard flags
  88     static inline DWORD GetDropEffect(DWORD flags);
  89
  90     DECLARE_NO_COPY_CLASS(wxIDropTarget)
  91 };
  92
  93 // ----------------------------------------------------------------------------
  94 // private functions
  95 // ----------------------------------------------------------------------------
  96
  97 static wxDragResult ConvertDragEffectToResult(DWORD dwEffect);
  98 static DWORD ConvertDragResultToEffect(wxDragResult result);
  99
 100 // ============================================================================
 101 // wxIDropTarget implementation
 102 // ============================================================================
 103
 104 // Name    : static wxIDropTarget::GetDropEffect
 105 // Purpose : determine the drop operation from keyboard/mouse state.
 106 // Returns : DWORD combined from DROPEFFECT_xxx constants
 107 // Params  : [in] DWORD flags       kbd & mouse flags as passed to
 108 //                                  IDropTarget methods
 109 // Notes   : We do "move" normally and "copy" if <Ctrl> is pressed,
 110 //           which is the standard behaviour (currently there is no
 111 //           way to redefine it)
 112 DWORD wxIDropTarget::GetDropEffect(DWORD flags)
 113 {
 114   return flags & MK_CONTROL ? DROPEFFECT_COPY : DROPEFFECT_MOVE;
 115 }
 116
 117 wxIDropTarget::wxIDropTarget(wxDropTarget *pTarget)
 118 {
 119   m_pTarget      = pTarget;
 120   m_pIDataObject = NULL;
 121 }
 122
 123 wxIDropTarget::~wxIDropTarget()
 124 {
 125 }
 126
 127 BEGIN_IID_TABLE(wxIDropTarget)
 128   ADD_IID(Unknown)
 129   ADD_IID(DropTarget)
 130 END_IID_TABLE;
 131
 132 IMPLEMENT_IUNKNOWN_METHODS(wxIDropTarget)
 133
 134 // Name    : wxIDropTarget::DragEnter
 135 // Purpose : Called when the mouse enters the window (dragging something)
 136 // Returns : S_OK
 137 // Params  : [in] IDataObject *pIDataSource : source data
 138 //           [in] DWORD        grfKeyState  : kbd & mouse state
 139 //           [in] POINTL       pt           : mouse coordinates
 140 //           [out]DWORD       *pdwEffect    : effect flag
 141 // Notes   :
 142 STDMETHODIMP wxIDropTarget::DragEnter(IDataObject *pIDataSource,
 143                                       DWORD        grfKeyState,
 144                                       POINTL       pt,
 145                                       DWORD       *pdwEffect)
 146 {
 147     wxLogTrace(wxTRACE_OleCalls, wxT("IDropTarget::DragEnter"));
 148
 149     wxASSERT_MSG( m_pIDataObject == NULL,
 150                   _T("drop target must have data object") );
 151
 152     // show the list of formats supported by the source data object for the
 153     // debugging purposes, this is quite useful sometimes - please don't remove
 154 #if 0
 155     IEnumFORMATETC *penumFmt;
 156     if ( SUCCEEDED(pIDataSource->EnumFormatEtc(DATADIR_GET, &penumFmt)) )
 157     {
 158         FORMATETC fmt;
 159         while ( penumFmt->Next(1, &fmt, NULL) == S_OK )
 160         {
 161             wxLogDebug(_T("Drop source supports format %s"),
 162                        wxDataObject::GetFormatName(fmt.cfFormat));
 163         }
 164
 165         penumFmt->Release();
 166     }
 167     else
 168     {
 169         wxLogLastError(_T("IDataObject::EnumFormatEtc"));
 170     }
 171 #endif // 0
 172
 173     if ( !m_pTarget->IsAcceptedData(pIDataSource) ) {
 174         // we don't accept this kind of data
 175         *pdwEffect = DROPEFFECT_NONE;
 176
 177         return S_OK;
 178     }
 179
 180     // get hold of the data object
 181     m_pIDataObject = pIDataSource;
 182     m_pIDataObject->AddRef();
 183
 184     // we need client coordinates to pass to wxWin functions
 185     if ( !ScreenToClient(m_hwnd, (POINT *)&pt) )
 186     {
 187         wxLogLastError(wxT("ScreenToClient"));
 188     }
 189
 190     // give some visual feedback
 191     *pdwEffect = ConvertDragResultToEffect(
 192                     m_pTarget->OnEnter(pt.x, pt.y,
 193                         ConvertDragEffectToResult(GetDropEffect(grfKeyState))
 194                     )
 195                  );
 196
 197     return S_OK;
 198 }
 199
 200 // Name    : wxIDropTarget::DragOver
 201 // Purpose : Indicates that the mouse was moved inside the window represented
 202 //           by this drop target.
 203 // Returns : S_OK
 204 // Params  : [in] DWORD   grfKeyState     kbd & mouse state
 205 //           [in] POINTL  pt              mouse coordinates
 206 //           [out]LPDWORD pdwEffect       effect flag
 207 // Notes   : We're called on every WM_MOUSEMOVE, so this function should be
 208 //           very efficient.
 209 STDMETHODIMP wxIDropTarget::DragOver(DWORD   grfKeyState,
 210                                      POINTL  pt,
 211                                      LPDWORD pdwEffect)
 212 {
 213     // there are too many of them... wxLogDebug("IDropTarget::DragOver");
 214
 215     wxDragResult result;
 216     if ( m_pIDataObject ) {
 217         result = ConvertDragEffectToResult(GetDropEffect(grfKeyState));
 218     }
 219     else {
 220         // can't accept data anyhow normally
 221         result = wxDragNone;
 222     }
 223
 224     // we need client coordinates to pass to wxWin functions
 225     if ( !ScreenToClient(m_hwnd, (POINT *)&pt) )
 226     {
 227         wxLogLastError(wxT("ScreenToClient"));
 228     }
 229
 230     *pdwEffect = ConvertDragResultToEffect(
 231                     m_pTarget->OnDragOver(pt.x, pt.y, result)
 232                  );
 233
 234     return S_OK;
 235 }
 236
 237 // Name    : wxIDropTarget::DragLeave
 238 // Purpose : Informs the drop target that the operation has left its window.
 239 // Returns : S_OK
 240 // Notes   : good place to do any clean-up
 241 STDMETHODIMP wxIDropTarget::DragLeave()
 242 {
 243   wxLogTrace(wxTRACE_OleCalls, wxT("IDropTarget::DragLeave"));
 244
 245   // remove the UI feedback
 246   m_pTarget->OnLeave();
 247
 248   // release the held object
 249   RELEASE_AND_NULL(m_pIDataObject);
 250
 251   return S_OK;
 252 }
 253
 254 // Name    : wxIDropTarget::Drop
 255 // Purpose : Instructs the drop target to paste data that was just now
 256 //           dropped on it.
 257 // Returns : S_OK
 258 // Params  : [in] IDataObject *pIDataSource     the data to paste
 259 //           [in] DWORD        grfKeyState      kbd & mouse state
 260 //           [in] POINTL       pt               where the drop occured?
 261 //           [ouy]DWORD       *pdwEffect        operation effect
 262 // Notes   :
 263 STDMETHODIMP wxIDropTarget::Drop(IDataObject *pIDataSource,
 264                                  DWORD        grfKeyState,
 265                                  POINTL       pt,
 266                                  DWORD       *pdwEffect)
 267 {
 268     wxLogTrace(wxTRACE_OleCalls, wxT("IDropTarget::Drop"));
 269
 270     // TODO I don't know why there is this parameter, but so far I assume
 271     //      that it's the same we've already got in DragEnter
 272     wxASSERT( m_pIDataObject == pIDataSource );
 273
 274     // by default, nothing happens
 275     *pdwEffect = DROPEFFECT_NONE;
 276
 277     // we need client coordinates to pass to wxWin functions
 278     if ( !ScreenToClient(m_hwnd, (POINT *)&pt) )
 279     {
 280         wxLogLastError(wxT("ScreenToClient"));
 281     }
 282
 283     // first ask the drop target if it wants data
 284     if ( m_pTarget->OnDrop(pt.x, pt.y) ) {
 285         // it does, so give it the data source
 286         m_pTarget->SetDataSource(pIDataSource);
 287
 288         // and now it has the data
 289         wxDragResult rc = ConvertDragEffectToResult(GetDropEffect(grfKeyState));
 290         rc = m_pTarget->OnData(pt.x, pt.y, rc);
 291         if ( wxIsDragResultOk(rc) ) {
 292             // operation succeeded
 293             *pdwEffect = ConvertDragResultToEffect(rc);
 294         }
 295         //else: *pdwEffect is already DROPEFFECT_NONE
 296     }
 297     //else: OnDrop() returned FALSE, no need to copy data
 298
 299     // release the held object
 300     RELEASE_AND_NULL(m_pIDataObject);
 301
 302     return S_OK;
 303 }
 304
 305 // ============================================================================
 306 // wxDropTarget implementation
 307 // ============================================================================
 308
 309 // ----------------------------------------------------------------------------
 310 // ctor/dtor
 311 // ----------------------------------------------------------------------------
 312
 313 wxDropTarget::wxDropTarget(wxDataObject *dataObj)
 314             : wxDropTargetBase(dataObj)
 315 {
 316     // create an IDropTarget implementation which will notify us about d&d
 317     // operations.
 318     m_pIDropTarget = new wxIDropTarget(this);
 319     m_pIDropTarget->AddRef();
 320 }
 321
 322 wxDropTarget::~wxDropTarget()
 323 {
 324     ReleaseInterface(m_pIDropTarget);
 325 }
 326
 327 // ----------------------------------------------------------------------------
 328 // [un]register drop handler
 329 // ----------------------------------------------------------------------------
 330
 331 bool wxDropTarget::Register(WXHWND hwnd)
 332 {
 333     HRESULT hr;
 334
 335     // May exist in later WinCE versions
 336 #ifndef __WXWINCE__
 337     hr = ::CoLockObjectExternal(m_pIDropTarget, TRUE, FALSE);
 338     if ( FAILED(hr) ) {
 339         wxLogApiError(wxT("CoLockObjectExternal"), hr);
 340         return FALSE;
 341     }
 342 #endif
 343
 344     hr = ::RegisterDragDrop((HWND) hwnd, m_pIDropTarget);
 345     if ( FAILED(hr) ) {
 346     // May exist in later WinCE versions
 347 #ifndef __WXWINCE__
 348         ::CoLockObjectExternal(m_pIDropTarget, FALSE, FALSE);
 349 #endif
 350         wxLogApiError(wxT("RegisterDragDrop"), hr);
 351         return FALSE;
 352     }
 353
 354     // we will need the window handle for coords transformation later
 355     m_pIDropTarget->SetHwnd((HWND)hwnd);
 356
 357     return TRUE;
 358 }
 359
 360 void wxDropTarget::Revoke(WXHWND hwnd)
 361 {
 362     HRESULT hr = ::RevokeDragDrop((HWND) hwnd);
 363
 364     if ( FAILED(hr) ) {
 365         wxLogApiError(wxT("RevokeDragDrop"), hr);
 366     }
 367
 368     // May exist in later WinCE versions
 369 #ifndef __WXWINCE__
 370     ::CoLockObjectExternal(m_pIDropTarget, FALSE, TRUE);
 371 #endif
 372
 373     m_pIDropTarget->SetHwnd(0);
 374 }
 375
 376 // ----------------------------------------------------------------------------
 377 // base class pure virtuals
 378 // ----------------------------------------------------------------------------
 379
 380 // OnDrop() is called only if we previously returned TRUE from
 381 // IsAcceptedData(), so no need to check anything here
 382 bool wxDropTarget::OnDrop(wxCoord WXUNUSED(x), wxCoord WXUNUSED(y))
 383 {
 384     return TRUE;
 385 }
 386
 387 // copy the data from the data source to the target data object
 388 bool wxDropTarget::GetData()
 389 {
 390     wxDataFormat format = GetSupportedFormat(m_pIDataSource);
 391     if ( format == wxDF_INVALID ) {
 392         // this is strange because IsAcceptedData() succeeded previously!
 393         wxFAIL_MSG(wxT("strange - did supported formats list change?"));
 394
 395         return FALSE;
 396     }
 397
 398     STGMEDIUM stm;
 399     FORMATETC fmtMemory;
 400     fmtMemory.cfFormat  = format;
 401     fmtMemory.ptd       = NULL;
 402     fmtMemory.dwAspect  = DVASPECT_CONTENT;
 403     fmtMemory.lindex    = -1;
 404     fmtMemory.tymed     = TYMED_HGLOBAL;  // TODO to add other media
 405
 406     bool rc = FALSE;
 407
 408     HRESULT hr = m_pIDataSource->GetData(&fmtMemory, &stm);
 409     if ( SUCCEEDED(hr) ) {
 410         IDataObject *dataObject = m_dataObject->GetInterface();
 411
 412         hr = dataObject->SetData(&fmtMemory, &stm, TRUE);
 413         if ( SUCCEEDED(hr) ) {
 414             rc = TRUE;
 415         }
 416         else {
 417             wxLogApiError(wxT("IDataObject::SetData()"), hr);
 418         }
 419     }
 420     else {
 421         wxLogApiError(wxT("IDataObject::GetData()"), hr);
 422     }
 423
 424     return rc;
 425 }
 426
 427 // ----------------------------------------------------------------------------
 428 // callbacks used by wxIDropTarget
 429 // ----------------------------------------------------------------------------
 430
 431 // we need a data source, so wxIDropTarget gives it to us using this function
 432 void wxDropTarget::SetDataSource(IDataObject *pIDataSource)
 433 {
 434     m_pIDataSource = pIDataSource;
 435 }
 436
 437 // determine if we accept data of this type
 438 bool wxDropTarget::IsAcceptedData(IDataObject *pIDataSource) const
 439 {
 440     return GetSupportedFormat(pIDataSource) != wxDF_INVALID;
 441 }
 442
 443 // ----------------------------------------------------------------------------
 444 // helper functions
 445 // ----------------------------------------------------------------------------
 446
 447 wxDataFormat wxDropTarget::GetSupportedFormat(IDataObject *pIDataSource) const
 448 {
 449     // this strucutre describes a data of any type (first field will be
 450     // changing) being passed through global memory block.
 451     static FORMATETC s_fmtMemory = {
 452         0,
 453         NULL,
 454         DVASPECT_CONTENT,
 455         -1,
 456         TYMED_HGLOBAL       // TODO is it worth supporting other tymeds here?
 457     };
 458
 459     // get the list of supported formats
 460     size_t nFormats = m_dataObject->GetFormatCount(wxDataObject::Set);
 461     wxDataFormat format;
 462     wxDataFormat *formats;
 463     formats = nFormats == 1 ? &format :  new wxDataFormat[nFormats];
 464
 465     m_dataObject->GetAllFormats(formats, wxDataObject::Set);
 466
 467     // cycle through all supported formats
 468     size_t n;
 469     for ( n = 0; n < nFormats; n++ ) {
 470         s_fmtMemory.cfFormat = formats[n];
 471
 472         // NB: don't use SUCCEEDED macro here: QueryGetData returns S_FALSE
 473         //     for file drag and drop (format == CF_HDROP)
 474         if ( pIDataSource->QueryGetData(&s_fmtMemory) == S_OK ) {
 475             format = formats[n];
 476
 477             break;
 478         }
 479     }
 480
 481     if ( formats != &format ) {
 482         // free memory if we allocated it
 483         delete [] formats;
 484     }
 485
 486     return n < nFormats ? format : wxFormatInvalid;
 487 }
 488
 489 // ----------------------------------------------------------------------------
 490 // private functions
 491 // ----------------------------------------------------------------------------
 492
 493 static wxDragResult ConvertDragEffectToResult(DWORD dwEffect)
 494 {
 495     switch ( dwEffect ) {
 496         case DROPEFFECT_COPY:
 497             return wxDragCopy;
 498
 499         case DROPEFFECT_LINK:
 500             return wxDragLink;
 501
 502         case DROPEFFECT_MOVE:
 503             return wxDragMove;
 504
 505         default:
 506             wxFAIL_MSG(wxT("invalid value in ConvertDragEffectToResult"));
 507             // fall through
 508
 509         case DROPEFFECT_NONE:
 510             return wxDragNone;
 511     }
 512 }
 513
 514 static DWORD ConvertDragResultToEffect(wxDragResult result)
 515 {
 516     switch ( result ) {
 517         case wxDragCopy:
 518             return DROPEFFECT_COPY;
 519
 520         case wxDragLink:
 521             return DROPEFFECT_LINK;
 522
 523         case wxDragMove:
 524             return DROPEFFECT_MOVE;
 525
 526         default:
 527             wxFAIL_MSG(wxT("invalid value in ConvertDragResultToEffect"));
 528             // fall through
 529
 530         case wxDragNone:
 531             return DROPEFFECT_NONE;
 532     }
 533 }
 534
 535 #endif
 536  // wxUSE_DRAG_AND_DROP