Fixed some wxDragImage bugs
[wxWidgets.git] / src / msw / dragimag.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: dragimag.cpp
3 // Purpose: wxDragImage
4 // Author: Julian Smart
5 // Modified by:
6 // Created: 08/04/99
7 // RCS-ID: $Id$
8 // Copyright: (c) Julian Smart
9 // Licence: wxWindows licence
10 /////////////////////////////////////////////////////////////////////////////
11
12 // ============================================================================
13 // declarations
14 // ============================================================================
15
16 // ----------------------------------------------------------------------------
17 // headers
18 // ----------------------------------------------------------------------------
19
20 #ifdef __GNUG__
21 #pragma implementation "dragimag.h"
22 #endif
23
24 // For compilers that support precompilation, includes "wx.h".
25 #include "wx/wxprec.h"
26 #include "wx/msw/private.h"
27
28 #ifdef __BORLANDC__
29 #pragma hdrstop
30 #endif
31
32 #if defined(__WIN95__)
33
34 #ifndef WX_PRECOMP
35 #include <stdio.h>
36 #include "wx/setup.h"
37 #include "wx/window.h"
38 #include "wx/dcclient.h"
39 #include "wx/dcscreen.h"
40 #include "wx/dcmemory.h"
41 #include "wx/settings.h"
42 #endif
43
44 #include "wx/log.h"
45 #include "wx/intl.h"
46 #include "wx/frame.h"
47 #include "wx/image.h"
48
49 #include "wx/msw/dragimag.h"
50 #include "wx/msw/private.h"
51
52 #if defined(__WIN95__) && !(defined(__GNUWIN32_OLD__) || defined(__TWIN32__))
53 #include <commctrl.h>
54 #endif
55
56 // ----------------------------------------------------------------------------
57 // macros
58 // ----------------------------------------------------------------------------
59
60 IMPLEMENT_DYNAMIC_CLASS(wxDragImage, wxObject)
61
62 #define GetHimageList() ((HIMAGELIST) m_hImageList)
63
64 // ============================================================================
65 // implementation
66 // ============================================================================
67
68 // ----------------------------------------------------------------------------
69 // wxDragImage ctors/dtor
70 // ----------------------------------------------------------------------------
71
72 wxDragImage::wxDragImage()
73 {
74 Init();
75 }
76
77 wxDragImage::~wxDragImage()
78 {
79 if ( m_hImageList )
80 ImageList_Destroy(GetHimageList());
81 #if !wxUSE_SIMPLER_DRAGIMAGE
82 if ( m_hCursorImageList )
83 ImageList_Destroy((HIMAGELIST) m_hCursorImageList);
84 #endif
85 }
86
87 void wxDragImage::Init()
88 {
89 m_hImageList = 0;
90 #if !wxUSE_SIMPLER_DRAGIMAGE
91 m_hCursorImageList = 0;
92 #endif
93 m_window = (wxWindow*) NULL;
94 m_fullScreen = FALSE;
95 }
96
97 // Attributes
98 ////////////////////////////////////////////////////////////////////////////
99
100
101 // Operations
102 ////////////////////////////////////////////////////////////////////////////
103
104 // Create a drag image from a bitmap and optional cursor
105 bool wxDragImage::Create(const wxBitmap& image, const wxCursor& cursor)
106 {
107 if ( m_hImageList )
108 ImageList_Destroy(GetHimageList());
109 m_hImageList = 0;
110
111 UINT flags = 0 ;
112 if (image.GetDepth() <= 4)
113 flags = ILC_COLOR4;
114 else if (image.GetDepth() <= 8)
115 flags = ILC_COLOR8;
116 else if (image.GetDepth() <= 16)
117 flags = ILC_COLOR16;
118 else if (image.GetDepth() <= 24)
119 flags = ILC_COLOR24;
120 else
121 flags = ILC_COLOR32;
122
123 bool mask = (image.GetMask() != 0);
124 if ( mask )
125 flags |= ILC_MASK;
126
127 m_hImageList = (WXHIMAGELIST) ImageList_Create(image.GetWidth(), image.GetHeight(), flags, 1, 1);
128
129 int index;
130 if (!mask)
131 {
132 HBITMAP hBitmap1 = (HBITMAP) image.GetHBITMAP();
133 index = ImageList_Add(GetHimageList(), hBitmap1, 0);
134 }
135 else
136 {
137 HBITMAP hBitmap1 = (HBITMAP) image.GetHBITMAP();
138 HBITMAP hBitmap2 = (HBITMAP) image.GetMask()->GetMaskBitmap();
139 HBITMAP hbmpMask = wxInvertMask(hBitmap2);
140
141 index = ImageList_Add(GetHimageList(), hBitmap1, hbmpMask);
142 ::DeleteObject(hbmpMask);
143 }
144 if ( index == -1 )
145 {
146 wxLogError(_("Couldn't add an image to the image list."));
147 }
148 m_cursor = cursor; // Can only combine with drag image after calling BeginDrag.
149
150 return (index != -1) ;
151 }
152
153 // Create a drag image from an icon and optional cursor
154 bool wxDragImage::Create(const wxIcon& image, const wxCursor& cursor)
155 {
156 if ( m_hImageList )
157 ImageList_Destroy(GetHimageList());
158 m_hImageList = 0;
159
160 UINT flags = 0 ;
161 if (image.GetDepth() <= 4)
162 flags = ILC_COLOR4;
163 else if (image.GetDepth() <= 8)
164 flags = ILC_COLOR8;
165 else if (image.GetDepth() <= 16)
166 flags = ILC_COLOR16;
167 else if (image.GetDepth() <= 24)
168 flags = ILC_COLOR24;
169 else
170 flags = ILC_COLOR32;
171 bool mask = TRUE;
172 if ( mask )
173 flags |= ILC_MASK;
174
175 m_hImageList = (WXHIMAGELIST) ImageList_Create(image.GetWidth(), image.GetHeight(), flags, 1, 1);
176
177 HICON hIcon = (HICON) image.GetHICON();
178
179 int index = ImageList_AddIcon(GetHimageList(), hIcon);
180 if ( index == -1 )
181 {
182 wxLogError(_("Couldn't add an image to the image list."));
183 }
184
185 m_cursor = cursor; // Can only combine with drag image after calling BeginDrag.
186
187 return (index != -1) ;
188 }
189
190 // Create a drag image from a string and optional cursor
191 bool wxDragImage::Create(const wxString& str, const wxCursor& cursor)
192 {
193 wxFont font(wxSystemSettings::GetSystemFont(wxSYS_DEFAULT_GUI_FONT));
194
195 long w, h;
196 wxScreenDC dc;
197 dc.SetFont(font);
198 dc.GetTextExtent(str, & w, & h);
199 dc.SetFont(wxNullFont);
200
201 wxMemoryDC dc2;
202 dc2.SetFont(font);
203 wxBitmap bitmap((int) w+2, (int) h+2);
204 dc2.SelectObject(bitmap);
205
206 dc2.SetBackground(* wxWHITE_BRUSH);
207 dc2.Clear();
208 dc2.SetBackgroundMode(wxTRANSPARENT);
209 dc2.SetTextForeground(* wxLIGHT_GREY);
210 dc2.DrawText(str, 0, 0);
211 dc2.DrawText(str, 1, 0);
212 dc2.DrawText(str, 2, 0);
213 dc2.DrawText(str, 1, 1);
214 dc2.DrawText(str, 2, 1);
215 dc2.DrawText(str, 1, 2);
216 dc2.DrawText(str, 2, 2);
217 dc2.SetTextForeground(* wxBLACK);
218 dc2.DrawText(str, 1, 1);
219
220 dc2.SelectObject(wxNullBitmap);
221
222 // Make the bitmap masked
223 wxImage image(bitmap);
224 image.SetMaskColour(255, 255, 255);
225 bitmap = image.ConvertToBitmap();
226
227 return Create(bitmap, cursor);
228 }
229
230 // Create a drag image for the given tree control item
231 bool wxDragImage::Create(const wxTreeCtrl& treeCtrl, wxTreeItemId& id)
232 {
233 if ( m_hImageList )
234 ImageList_Destroy(GetHimageList());
235 m_hImageList = (WXHIMAGELIST) TreeView_CreateDragImage((HWND) treeCtrl.GetHWND(), (HTREEITEM) (WXHTREEITEM) id);
236 return TRUE;
237 }
238
239 // Create a drag image for the given list control item
240 bool wxDragImage::Create(const wxListCtrl& listCtrl, long id)
241 {
242 if ( m_hImageList )
243 ImageList_Destroy(GetHimageList());
244 POINT pt;
245 pt.x = 0; pt.y = 0;
246 m_hImageList = (WXHIMAGELIST) ListView_CreateDragImage((HWND) listCtrl.GetHWND(), id, & pt);
247 return TRUE;
248 }
249
250 // Begin drag
251 bool wxDragImage::BeginDrag(const wxPoint& hotspot, wxWindow* window, bool fullScreen, wxRect* rect)
252 {
253 wxASSERT_MSG( (m_hImageList != 0), wxT("Image list must not be null in BeginDrag."));
254 wxASSERT_MSG( (window != 0), wxT("Window must not be null in BeginDrag."));
255
256 m_fullScreen = fullScreen;
257 if (rect)
258 m_boundingRect = * rect;
259
260 bool ret = (ImageList_BeginDrag(GetHimageList(), 0, hotspot.x, hotspot.y) != 0);
261 //bool ret = (ImageList_BeginDrag(GetHimageList(), 0, 0, 0) != 0);
262
263 if (!ret)
264 {
265 wxFAIL_MSG( _T("BeginDrag failed.") );
266
267 return FALSE;
268 }
269
270 if (m_cursor.Ok())
271 {
272 #if wxUSE_SIMPLER_DRAGIMAGE
273 m_oldCursor = window->GetCursor();
274 window->SetCursor(m_cursor);
275 #else
276 if (!m_hCursorImageList)
277 {
278 int cxCursor = GetSystemMetrics(SM_CXCURSOR);
279 int cyCursor = GetSystemMetrics(SM_CYCURSOR);
280
281 m_hCursorImageList = (WXHIMAGELIST) ImageList_Create(cxCursor, cyCursor, ILC_MASK, 1, 1);
282 }
283
284 // See if we can find the cursor hotspot
285 wxPoint curHotSpot(hotspot);
286
287 #if 0
288 ICONINFO iconInfo;
289 if (::GetIconInfo((HICON) (HCURSOR) m_cursor.GetHCURSOR(), & iconInfo) != 0)
290 {
291 curHotSpot.x -= iconInfo.xHotspot;
292 curHotSpot.y -= iconInfo.yHotspot;
293 }
294 #endif
295 wxString msg;
296 msg.Printf("Hotspot = %d, %d", curHotSpot.x, curHotSpot.y);
297 wxLogDebug(msg);
298
299 // First add the cursor to the image list
300 HCURSOR hCursor = (HCURSOR) m_cursor.GetHCURSOR();
301 int cursorIndex = ImageList_AddIcon((HIMAGELIST) m_hCursorImageList, (HICON) hCursor);
302
303 wxASSERT_MSG( (cursorIndex != -1), wxT("ImageList_AddIcon failed in BeginDrag."));
304
305 if (cursorIndex != -1)
306 {
307 ImageList_SetDragCursorImage((HIMAGELIST) m_hCursorImageList, cursorIndex, curHotSpot.x, curHotSpot.y);
308 }
309 #endif
310 }
311
312 #if !wxUSE_SIMPLER_DRAGIMAGE
313 if (m_cursor.Ok())
314 ::ShowCursor(FALSE);
315 #endif
316
317 m_window = window;
318
319 ::SetCapture(GetHwndOf(window));
320
321 return TRUE;
322 }
323
324 // Begin drag. hotspot is the location of the drag position relative to the upper-left
325 // corner of the image. This is full screen only. fullScreenRect gives the
326 // position of the window on the screen, to restrict the drag to.
327 bool wxDragImage::BeginDrag(const wxPoint& hotspot, wxWindow* window, wxWindow* fullScreenRect)
328 {
329 wxRect rect;
330
331 int x = fullScreenRect->GetPosition().x;
332 int y = fullScreenRect->GetPosition().y;
333
334 wxSize sz = fullScreenRect->GetSize();
335
336 if (fullScreenRect->GetParent() && !fullScreenRect->IsKindOf(CLASSINFO(wxFrame)))
337 fullScreenRect->GetParent()->ClientToScreen(& x, & y);
338
339 rect.x = x; rect.y = y;
340 rect.width = sz.x; rect.height = sz.y;
341
342 return BeginDrag(hotspot, window, TRUE, & rect);
343 }
344
345 // End drag
346 bool wxDragImage::EndDrag()
347 {
348 wxASSERT_MSG( (m_hImageList != 0), wxT("Image list must not be null in EndDrag."));
349
350 ImageList_EndDrag();
351
352 if ( !::ReleaseCapture() )
353 {
354 wxLogLastError(wxT("ReleaseCapture"));
355 }
356
357 #if wxUSE_SIMPLER_DRAGIMAGE
358 if (m_cursor.Ok() && m_oldCursor.Ok())
359 m_window->SetCursor(m_oldCursor);
360 #else
361 ::ShowCursor(TRUE);
362 #endif
363
364 m_window = (wxWindow*) NULL;
365
366 return TRUE;
367 }
368
369 // Move the image: call from OnMouseMove. Pt is in window client coordinates if window
370 // is non-NULL, or in screen coordinates if NULL.
371 bool wxDragImage::Move(const wxPoint& pt)
372 {
373 wxASSERT_MSG( (m_hImageList != 0), wxT("Image list must not be null in Move."));
374
375 // These are in window, not client coordinates.
376 // So need to convert to client coordinates.
377 wxPoint pt2(pt);
378 if (m_window && !m_fullScreen)
379 {
380 RECT rect;
381 rect.left = 0; rect.top = 0;
382 rect.right = 0; rect.bottom = 0;
383 DWORD style = ::GetWindowLong((HWND) m_window->GetHWND(), GWL_STYLE);
384 #ifdef __WIN32__
385 DWORD exStyle = ::GetWindowLong((HWND) m_window->GetHWND(), GWL_EXSTYLE);
386 ::AdjustWindowRectEx(& rect, style, FALSE, exStyle);
387 #else
388 ::AdjustWindowRect(& rect, style, FALSE);
389 #endif
390 // Subtract the (negative) values, i.e. add a small increment
391 pt2.x -= rect.left; pt2.y -= rect.top;
392 }
393 else if (m_window && m_fullScreen)
394 {
395 pt2 = m_window->ClientToScreen(pt2);
396 }
397
398 bool ret = (ImageList_DragMove( pt2.x, pt2.y ) != 0);
399
400 m_position = pt2;
401
402 return ret;
403 }
404
405 bool wxDragImage::Show()
406 {
407 wxASSERT_MSG( (m_hImageList != 0), wxT("Image list must not be null in Show."));
408
409 HWND hWnd = 0;
410 if (m_window && !m_fullScreen)
411 hWnd = (HWND) m_window->GetHWND();
412
413 bool ret = (ImageList_DragEnter( hWnd, m_position.x, m_position.y ) != 0);
414
415 return ret;
416 }
417
418 bool wxDragImage::Hide()
419 {
420 wxASSERT_MSG( (m_hImageList != 0), wxT("Image list must not be null in Hide."));
421
422 HWND hWnd = 0;
423 if (m_window && !m_fullScreen)
424 hWnd = (HWND) m_window->GetHWND();
425
426 bool ret = (ImageList_DragLeave( hWnd ) != 0);
427
428 return ret;
429 }
430
431 #endif
432 // __WIN95__
433