]> git.saurik.com Git - wxWidgets.git/blame - src/msw/treectrl.cpp
drawing optimization fix
[wxWidgets.git] / src / msw / treectrl.cpp
CommitLineData
b823f5a1
JS
1/////////////////////////////////////////////////////////////////////////////
2// Name: treectrl.cpp
3// Purpose: wxTreeCtrl
4// Author: Julian Smart
08b7c251 5// Modified by: Vadim Zeitlin to be less MSW-specific on 10.10.98
b823f5a1
JS
6// Created: 1997
7// RCS-ID: $Id$
8// Copyright: (c) Julian Smart
08b7c251 9// Licence: wxWindows licence
b823f5a1 10/////////////////////////////////////////////////////////////////////////////
2bda0e17 11
08b7c251
VZ
12// ============================================================================
13// declarations
14// ============================================================================
15
16// ----------------------------------------------------------------------------
17// headers
18// ----------------------------------------------------------------------------
2bda0e17 19#ifdef __GNUG__
08b7c251 20 #pragma implementation "treectrl.h"
2bda0e17
KB
21#endif
22
23// For compilers that support precompilation, includes "wx.h".
24#include "wx/wxprec.h"
25
26#ifdef __BORLANDC__
08b7c251 27 #pragma hdrstop
2bda0e17
KB
28#endif
29
30#ifndef WX_PRECOMP
08b7c251 31 #include "wx.h"
2bda0e17
KB
32#endif
33
2bda0e17
KB
34#if defined(__WIN95__)
35
08b7c251
VZ
36#include "wx/log.h"
37#include "wx/imaglist.h"
38
2bda0e17
KB
39#include "wx/msw/private.h"
40
41#ifndef __GNUWIN32__
08b7c251 42 #include <commctrl.h>
2bda0e17
KB
43#endif
44
06e38c8e
JS
45#ifdef GetFirstChild
46#undef GetFirstChild
47#endif
48
49#ifdef GetNextChild
50#undef GetNextChild
51#endif
52
53#ifdef GetNextSibling
54#undef GetNextSibling
55#endif
56
add28c55 57#include "wx/msw/treectrl.h"
08b7c251 58
2bda0e17
KB
59// Bug in headers, sometimes
60#ifndef TVIS_FOCUSED
08b7c251 61 #define TVIS_FOCUSED 0x0001
2bda0e17
KB
62#endif
63
08b7c251
VZ
64// ----------------------------------------------------------------------------
65// private classes
66// ----------------------------------------------------------------------------
2bda0e17 67
08b7c251
VZ
68// a convenient wrapper around TV_ITEM struct which adds a ctor
69struct wxTreeViewItem : public TV_ITEM
70{
71 wxTreeViewItem(const wxTreeItemId& item,
72 UINT mask_, UINT stateMask_ = 0)
73 {
74 mask = mask_;
75 stateMask = stateMask_;
06e38c8e 76 hItem = (HTREEITEM) (WXHTREEITEM) item;
08b7c251
VZ
77 }
78};
2bda0e17 79
08b7c251
VZ
80// ----------------------------------------------------------------------------
81// macros
82// ----------------------------------------------------------------------------
83
84#if !USE_SHARED_LIBRARY
85 IMPLEMENT_DYNAMIC_CLASS(wxTreeCtrl, wxControl)
2bda0e17
KB
86#endif
87
08b7c251 88// hide the ugly cast (of course, the macro is _quite_ ugly too...)
06e38c8e 89#define wxhWnd ((HWND)m_hWnd)
08b7c251
VZ
90
91// ----------------------------------------------------------------------------
92// variables
93// ----------------------------------------------------------------------------
94
95// handy table for sending events
96static const wxEventType g_events[2][2] =
2bda0e17 97{
08b7c251
VZ
98 { wxEVT_COMMAND_TREE_ITEM_COLLAPSED, wxEVT_COMMAND_TREE_ITEM_COLLAPSING },
99 { wxEVT_COMMAND_TREE_ITEM_EXPANDED, wxEVT_COMMAND_TREE_ITEM_EXPANDING }
100};
101
102// ============================================================================
103// implementation
104// ============================================================================
105
106// ----------------------------------------------------------------------------
107// construction and destruction
108// ----------------------------------------------------------------------------
109
110void wxTreeCtrl::Init()
111{
112 m_imageListNormal = NULL;
113 m_imageListState = NULL;
114 m_textCtrl = NULL;
2bda0e17
KB
115}
116
08b7c251
VZ
117bool wxTreeCtrl::Create(wxWindow *parent, wxWindowID id,
118 const wxPoint& pos, const wxSize& size,
119 long style, const wxValidator& validator,
120 const wxString& name)
2bda0e17 121{
08b7c251 122 Init();
2bda0e17 123
08b7c251 124 wxSystemSettings settings;
2bda0e17 125
08b7c251
VZ
126 SetName(name);
127 SetValidator(validator);
2bda0e17 128
08b7c251 129 m_windowStyle = style;
2bda0e17 130
08b7c251 131 SetParent(parent);
2bda0e17 132
08b7c251 133 m_windowId = (id == -1) ? NewControlId() : id;
2bda0e17 134
08b7c251 135 DWORD wstyle = WS_VISIBLE | WS_CHILD | WS_TABSTOP | TVS_HASLINES;
2bda0e17 136
08b7c251
VZ
137 bool want3D;
138 WXDWORD exStyle = Determine3DEffects(WS_EX_CLIENTEDGE, &want3D) ;
2bda0e17 139
08b7c251
VZ
140 // Even with extended styles, need to combine with WS_BORDER
141 // for them to look right.
142 if ( want3D || wxStyleHasBorder(m_windowStyle) )
143 {
144 wstyle |= WS_BORDER;
145 }
2bda0e17 146
08b7c251
VZ
147 if ( m_windowStyle & wxTR_HAS_BUTTONS )
148 wstyle |= TVS_HASBUTTONS;
2bda0e17 149
08b7c251
VZ
150 if ( m_windowStyle & wxTR_EDIT_LABELS )
151 wstyle |= TVS_EDITLABELS;
2bda0e17 152
08b7c251
VZ
153 if ( m_windowStyle & wxTR_LINES_AT_ROOT )
154 wstyle |= TVS_LINESATROOT;
2bda0e17 155
08b7c251
VZ
156 // Create the tree control.
157 m_hWnd = (WXHWND)::CreateWindowEx
158 (
159 exStyle,
160 WC_TREEVIEW,
161 "",
162 wstyle,
163 pos.x, pos.y, size.x, size.y,
164 (HWND)parent->GetHWND(),
165 (HMENU)m_windowId,
166 wxGetInstance(),
167 NULL
168 );
7798a18e 169
08b7c251 170 wxCHECK_MSG( m_hWnd, FALSE, "Failed to create tree ctrl" );
2bda0e17 171
08b7c251
VZ
172 if ( parent )
173 parent->AddChild(this);
2bda0e17 174
08b7c251 175 SubclassWin(m_hWnd);
2bda0e17 176
08b7c251 177 return TRUE;
2bda0e17
KB
178}
179
08b7c251 180wxTreeCtrl::~wxTreeCtrl()
2bda0e17 181{
08b7c251 182 DeleteTextCtrl();
2bda0e17 183
08b7c251
VZ
184 // delete user data to prevent memory leaks
185 DeleteAllItems();
2bda0e17
KB
186}
187
08b7c251
VZ
188// ----------------------------------------------------------------------------
189// accessors
190// ----------------------------------------------------------------------------
2bda0e17 191
08b7c251 192// simple wrappers which add error checking in debug mode
2bda0e17 193
08b7c251 194bool wxTreeCtrl::DoGetItem(wxTreeViewItem* tvItem) const
2bda0e17 195{
06e38c8e 196 if ( !TreeView_GetItem(wxhWnd, tvItem) )
2bda0e17 197 {
08b7c251
VZ
198 wxLogLastError("TreeView_GetItem");
199
200 return FALSE;
201 }
202
203 return TRUE;
2bda0e17
KB
204}
205
08b7c251 206void wxTreeCtrl::DoSetItem(wxTreeViewItem* tvItem)
2bda0e17 207{
06e38c8e 208 if ( TreeView_SetItem(wxhWnd, tvItem) == -1 )
2bda0e17 209 {
08b7c251
VZ
210 wxLogLastError("TreeView_SetItem");
211 }
2bda0e17
KB
212}
213
08b7c251 214size_t wxTreeCtrl::GetCount() const
2bda0e17 215{
06e38c8e 216 return (size_t)TreeView_GetCount(wxhWnd);
2bda0e17
KB
217}
218
08b7c251 219unsigned int wxTreeCtrl::GetIndent() const
2bda0e17 220{
06e38c8e 221 return TreeView_GetIndent(wxhWnd);
2bda0e17
KB
222}
223
08b7c251 224void wxTreeCtrl::SetIndent(unsigned int indent)
2bda0e17 225{
06e38c8e 226 TreeView_SetIndent(wxhWnd, indent);
2bda0e17
KB
227}
228
08b7c251 229wxImageList *wxTreeCtrl::GetImageList() const
2bda0e17 230{
08b7c251 231 return m_imageListNormal;
2bda0e17
KB
232}
233
08b7c251 234wxImageList *wxTreeCtrl::GetStateImageList() const
2bda0e17 235{
08b7c251 236 return m_imageListNormal;
2bda0e17
KB
237}
238
08b7c251 239void wxTreeCtrl::SetAnyImageList(wxImageList *imageList, int which)
2bda0e17 240{
08b7c251 241 // no error return
06e38c8e 242 TreeView_SetImageList(wxhWnd,
08b7c251
VZ
243 imageList ? imageList->GetHIMAGELIST() : 0,
244 which);
2bda0e17
KB
245}
246
08b7c251 247void wxTreeCtrl::SetImageList(wxImageList *imageList)
2bda0e17 248{
08b7c251 249 SetAnyImageList(m_imageListNormal = imageList, TVSIL_NORMAL);
2bda0e17
KB
250}
251
08b7c251 252void wxTreeCtrl::SetStateImageList(wxImageList *imageList)
2bda0e17 253{
08b7c251 254 SetAnyImageList(m_imageListState = imageList, TVSIL_STATE);
2bda0e17
KB
255}
256
08b7c251
VZ
257// ----------------------------------------------------------------------------
258// Item access
259// ----------------------------------------------------------------------------
260
261wxString wxTreeCtrl::GetItemText(const wxTreeItemId& item) const
2bda0e17 262{
08b7c251 263 char buf[512]; // the size is arbitrary...
02ce7b72 264
08b7c251
VZ
265 wxTreeViewItem tvItem(item, TVIF_TEXT);
266 tvItem.pszText = buf;
267 tvItem.cchTextMax = WXSIZEOF(buf);
268 if ( !DoGetItem(&tvItem) )
269 {
270 // don't return some garbage which was on stack, but an empty string
271 buf[0] = '\0';
272 }
2bda0e17 273
08b7c251
VZ
274 return wxString(buf);
275}
2bda0e17 276
08b7c251
VZ
277void wxTreeCtrl::SetItemText(const wxTreeItemId& item, const wxString& text)
278{
279 wxTreeViewItem tvItem(item, TVIF_TEXT);
280 tvItem.pszText = (char *)text.c_str(); // conversion is ok
281 DoSetItem(&tvItem);
282}
2bda0e17 283
08b7c251
VZ
284int wxTreeCtrl::GetItemImage(const wxTreeItemId& item) const
285{
286 wxTreeViewItem tvItem(item, TVIF_IMAGE);
287 DoGetItem(&tvItem);
2bda0e17 288
08b7c251
VZ
289 return tvItem.iImage;
290}
2bda0e17 291
08b7c251
VZ
292void wxTreeCtrl::SetItemImage(const wxTreeItemId& item, int image)
293{
294 wxTreeViewItem tvItem(item, TVIF_IMAGE);
295 tvItem.iImage = image;
296 DoSetItem(&tvItem);
297}
2bda0e17 298
08b7c251
VZ
299int wxTreeCtrl::GetItemSelectedImage(const wxTreeItemId& item) const
300{
301 wxTreeViewItem tvItem(item, TVIF_SELECTEDIMAGE);
302 DoGetItem(&tvItem);
2bda0e17 303
08b7c251 304 return tvItem.iSelectedImage;
2bda0e17
KB
305}
306
08b7c251 307void wxTreeCtrl::SetItemSelectedImage(const wxTreeItemId& item, int image)
2bda0e17 308{
08b7c251
VZ
309 wxTreeViewItem tvItem(item, TVIF_SELECTEDIMAGE);
310 tvItem.iSelectedImage = image;
311 DoSetItem(&tvItem);
2bda0e17
KB
312}
313
08b7c251 314wxTreeItemData *wxTreeCtrl::GetItemData(const wxTreeItemId& item) const
2bda0e17 315{
08b7c251
VZ
316 wxTreeViewItem tvItem(item, TVIF_PARAM);
317 if ( !DoGetItem(&tvItem) )
318 {
319 return NULL;
320 }
2bda0e17 321
3a5a2f56 322 return (wxTreeItemData *)tvItem.lParam;
2bda0e17
KB
323}
324
08b7c251 325void wxTreeCtrl::SetItemData(const wxTreeItemId& item, wxTreeItemData *data)
2bda0e17 326{
08b7c251
VZ
327 wxTreeViewItem tvItem(item, TVIF_PARAM);
328 tvItem.lParam = (LPARAM)data;
329 DoSetItem(&tvItem);
330}
2bda0e17 331
3a5a2f56
VZ
332void wxTreeCtrl::SetItemHasChildren(const wxTreeItemId& item, bool has)
333{
334 wxTreeViewItem tvItem(item, TVIF_CHILDREN);
335 tvItem.cChildren = (int)has;
336 DoSetItem(&tvItem);
337}
338
add28c55
VZ
339void wxTreeCtrl::SetItemBold(const wxTreeItemId& item, bool bold)
340{
341 wxTreeViewItem tvItem(item, TVIF_STATE, TVIS_BOLD);
342 tvItem.state = bold ? TVIS_BOLD : 0;
343 DoSetItem(&tvItem);
344}
345
08b7c251
VZ
346// ----------------------------------------------------------------------------
347// Item status
348// ----------------------------------------------------------------------------
2bda0e17 349
08b7c251
VZ
350bool wxTreeCtrl::IsVisible(const wxTreeItemId& item) const
351{
add28c55 352 // Bug in Gnu-Win32 headers, so don't use the macro TreeView_GetItemRect
08b7c251 353 RECT rect;
add28c55 354 return SendMessage(wxhWnd, TVM_GETITEMRECT, FALSE, (LPARAM)&rect) != 0;
06e38c8e 355
2bda0e17
KB
356}
357
08b7c251 358bool wxTreeCtrl::ItemHasChildren(const wxTreeItemId& item) const
2bda0e17 359{
08b7c251
VZ
360 wxTreeViewItem tvItem(item, TVIF_CHILDREN);
361 DoGetItem(&tvItem);
2bda0e17 362
08b7c251 363 return tvItem.cChildren != 0;
2bda0e17
KB
364}
365
08b7c251 366bool wxTreeCtrl::IsExpanded(const wxTreeItemId& item) const
2bda0e17 367{
08b7c251
VZ
368 // probably not a good idea to put it here
369 //wxASSERT( ItemHasChildren(item) );
2bda0e17 370
08b7c251
VZ
371 wxTreeViewItem tvItem(item, TVIF_STATE, TVIS_EXPANDED);
372 DoGetItem(&tvItem);
2bda0e17 373
08b7c251 374 return (tvItem.state & TVIS_EXPANDED) != 0;
2bda0e17
KB
375}
376
08b7c251 377bool wxTreeCtrl::IsSelected(const wxTreeItemId& item) const
2bda0e17 378{
08b7c251
VZ
379 wxTreeViewItem tvItem(item, TVIF_STATE, TVIS_SELECTED);
380 DoGetItem(&tvItem);
2bda0e17 381
08b7c251 382 return (tvItem.state & TVIS_SELECTED) != 0;
2bda0e17
KB
383}
384
add28c55
VZ
385bool wxTreeCtrl::IsBold(const wxTreeItemId& item) const
386{
387 wxTreeViewItem tvItem(item, TVIF_STATE, TVIS_BOLD);
388 DoGetItem(&tvItem);
389
390 return (tvItem.state & TVIS_BOLD) != 0;
391}
392
08b7c251
VZ
393// ----------------------------------------------------------------------------
394// navigation
395// ----------------------------------------------------------------------------
2bda0e17 396
08b7c251
VZ
397wxTreeItemId wxTreeCtrl::GetRootItem() const
398{
a3168196 399 return wxTreeItemId((WXHTREEITEM) TreeView_GetRoot(wxhWnd));
08b7c251 400}
2bda0e17 401
08b7c251
VZ
402wxTreeItemId wxTreeCtrl::GetSelection() const
403{
a3168196 404 return wxTreeItemId((WXHTREEITEM) TreeView_GetSelection(wxhWnd));
2bda0e17
KB
405}
406
08b7c251 407wxTreeItemId wxTreeCtrl::GetParent(const wxTreeItemId& item) const
2bda0e17 408{
06e38c8e 409 return wxTreeItemId((WXHTREEITEM) TreeView_GetParent(wxhWnd, (HTREEITEM) (WXHTREEITEM) item));
08b7c251 410}
2bda0e17 411
08b7c251 412wxTreeItemId wxTreeCtrl::GetFirstChild(const wxTreeItemId& item,
06e38c8e 413 long& _cookie) const
08b7c251
VZ
414{
415 // remember the last child returned in 'cookie'
06e38c8e 416 _cookie = (long)TreeView_GetChild(wxhWnd, (HTREEITEM) (WXHTREEITEM)item);
2bda0e17 417
06e38c8e 418 return wxTreeItemId((WXHTREEITEM)_cookie);
2bda0e17
KB
419}
420
08b7c251 421wxTreeItemId wxTreeCtrl::GetNextChild(const wxTreeItemId& WXUNUSED(item),
06e38c8e 422 long& _cookie) const
2bda0e17 423{
2e5dddb0
JS
424 wxTreeItemId l=wxTreeItemId((WXHTREEITEM) TreeView_GetNextSibling(wxhWnd,
425 (HTREEITEM) (WXHTREEITEM)_cookie));
426 _cookie=(long)l;
427 return l;
08b7c251 428}
2bda0e17 429
08b7c251
VZ
430wxTreeItemId wxTreeCtrl::GetNextSibling(const wxTreeItemId& item) const
431{
a3168196 432 return wxTreeItemId((WXHTREEITEM) TreeView_GetNextSibling(wxhWnd, (HTREEITEM) (WXHTREEITEM) item));
2bda0e17
KB
433}
434
08b7c251 435wxTreeItemId wxTreeCtrl::GetPrevSibling(const wxTreeItemId& item) const
2bda0e17 436{
a3168196 437 return wxTreeItemId((WXHTREEITEM) TreeView_GetPrevSibling(wxhWnd, (HTREEITEM) (WXHTREEITEM) item));
2bda0e17
KB
438}
439
08b7c251 440wxTreeItemId wxTreeCtrl::GetFirstVisibleItem() const
2bda0e17 441{
a3168196 442 return wxTreeItemId((WXHTREEITEM) TreeView_GetFirstVisible(wxhWnd));
2bda0e17
KB
443}
444
08b7c251 445wxTreeItemId wxTreeCtrl::GetNextVisible(const wxTreeItemId& item) const
2bda0e17 446{
08b7c251
VZ
447 wxASSERT_MSG( IsVisible(item), "The item you call GetNextVisible() "
448 "for must be visible itself!");
02ce7b72 449
a3168196 450 return wxTreeItemId((WXHTREEITEM) TreeView_GetNextVisible(wxhWnd, (HTREEITEM) (WXHTREEITEM) item));
08b7c251 451}
02ce7b72 452
08b7c251
VZ
453wxTreeItemId wxTreeCtrl::GetPrevVisible(const wxTreeItemId& item) const
454{
455 wxASSERT_MSG( IsVisible(item), "The item you call GetPrevVisible() "
456 "for must be visible itself!");
02ce7b72 457
a3168196 458 return wxTreeItemId((WXHTREEITEM) TreeView_GetPrevVisible(wxhWnd, (HTREEITEM) (WXHTREEITEM) item));
08b7c251 459}
02ce7b72 460
08b7c251
VZ
461// ----------------------------------------------------------------------------
462// Usual operations
463// ----------------------------------------------------------------------------
02ce7b72 464
08b7c251
VZ
465wxTreeItemId wxTreeCtrl::DoInsertItem(const wxTreeItemId& parent,
466 wxTreeItemId hInsertAfter,
467 const wxString& text,
468 int image, int selectedImage,
469 wxTreeItemData *data)
470{
471 TV_INSERTSTRUCT tvIns;
06e38c8e
JS
472 tvIns.hParent = (HTREEITEM) (WXHTREEITEM)parent;
473 tvIns.hInsertAfter = (HTREEITEM) (WXHTREEITEM) hInsertAfter;
08b7c251
VZ
474 UINT mask = 0;
475 if ( !text.IsEmpty() )
476 {
477 mask |= TVIF_TEXT;
478 tvIns.item.pszText = (char *)text.c_str(); // cast is ok
479 }
02ce7b72 480
08b7c251
VZ
481 if ( image != -1 )
482 {
483 mask |= TVIF_IMAGE;
484 tvIns.item.iImage = image;
3a5a2f56 485
6b037754 486 if ( selectedImage == -1 )
3a5a2f56
VZ
487 {
488 // take the same image for selected icon if not specified
489 selectedImage = image;
490 }
08b7c251 491 }
02ce7b72 492
08b7c251
VZ
493 if ( selectedImage != -1 )
494 {
495 mask |= TVIF_SELECTEDIMAGE;
496 tvIns.item.iSelectedImage = selectedImage;
497 }
02ce7b72 498
08b7c251
VZ
499 if ( data != NULL )
500 {
501 mask |= TVIF_PARAM;
502 tvIns.item.lParam = (LPARAM)data;
503 }
02ce7b72 504
08b7c251 505 tvIns.item.mask = mask;
02ce7b72 506
4fabb575 507 HTREEITEM id = (HTREEITEM) TreeView_InsertItem(wxhWnd, &tvIns);
08b7c251
VZ
508 if ( id == 0 )
509 {
510 wxLogLastError("TreeView_InsertItem");
511 }
02ce7b72 512
fd3f686c
VZ
513 if ( data != NULL )
514 {
515 // associate the application tree item with Win32 tree item handle
516 data->SetId((WXHTREEITEM)id);
517 }
518
06e38c8e 519 return wxTreeItemId((WXHTREEITEM)id);
2bda0e17
KB
520}
521
08b7c251
VZ
522// for compatibility only
523wxTreeItemId wxTreeCtrl::InsertItem(const wxTreeItemId& parent,
524 const wxString& text,
525 int image, int selImage,
526 long insertAfter)
2bda0e17 527{
06e38c8e 528 return DoInsertItem(parent, (WXHTREEITEM)insertAfter, text,
08b7c251 529 image, selImage, NULL);
2bda0e17
KB
530}
531
08b7c251
VZ
532wxTreeItemId wxTreeCtrl::AddRoot(const wxString& text,
533 int image, int selectedImage,
534 wxTreeItemData *data)
2bda0e17 535{
06e38c8e 536 return DoInsertItem(wxTreeItemId((WXHTREEITEM) 0), (WXHTREEITEM) 0,
08b7c251 537 text, image, selectedImage, data);
2bda0e17
KB
538}
539
08b7c251
VZ
540wxTreeItemId wxTreeCtrl::PrependItem(const wxTreeItemId& parent,
541 const wxString& text,
542 int image, int selectedImage,
543 wxTreeItemData *data)
2bda0e17 544{
06e38c8e 545 return DoInsertItem(parent, (WXHTREEITEM) TVI_FIRST,
08b7c251 546 text, image, selectedImage, data);
2bda0e17
KB
547}
548
08b7c251
VZ
549wxTreeItemId wxTreeCtrl::InsertItem(const wxTreeItemId& parent,
550 const wxTreeItemId& idPrevious,
551 const wxString& text,
552 int image, int selectedImage,
553 wxTreeItemData *data)
2bda0e17 554{
08b7c251 555 return DoInsertItem(parent, idPrevious, text, image, selectedImage, data);
2bda0e17
KB
556}
557
08b7c251
VZ
558wxTreeItemId wxTreeCtrl::AppendItem(const wxTreeItemId& parent,
559 const wxString& text,
560 int image, int selectedImage,
561 wxTreeItemData *data)
2bda0e17 562{
06e38c8e 563 return DoInsertItem(parent, (WXHTREEITEM) TVI_LAST,
08b7c251 564 text, image, selectedImage, data);
2bda0e17
KB
565}
566
08b7c251 567void wxTreeCtrl::Delete(const wxTreeItemId& item)
2bda0e17 568{
06e38c8e 569 if ( !TreeView_DeleteItem(wxhWnd, (HTREEITEM)(WXHTREEITEM)item) )
bbcdf8bc 570 {
08b7c251 571 wxLogLastError("TreeView_DeleteItem");
bbcdf8bc 572 }
bbcdf8bc
JS
573}
574
08b7c251 575void wxTreeCtrl::DeleteAllItems()
bbcdf8bc 576{
06e38c8e 577 if ( !TreeView_DeleteAllItems(wxhWnd) )
bbcdf8bc 578 {
08b7c251 579 wxLogLastError("TreeView_DeleteAllItems");
bbcdf8bc 580 }
2bda0e17
KB
581}
582
08b7c251 583void wxTreeCtrl::DoExpand(const wxTreeItemId& item, int flag)
2bda0e17 584{
08b7c251
VZ
585 wxASSERT_MSG( flag == TVE_COLLAPSE || flag == TVE_COLLAPSERESET ||
586 flag == TVE_EXPAND || flag == TVE_TOGGLE,
587 "Unknown flag in wxTreeCtrl::DoExpand" );
588
589 // TreeView_Expand doesn't send TVN_ITEMEXPAND(ING) messages, so we must
590 // emulate them
06e38c8e 591 if ( TreeView_Expand(wxhWnd, (HTREEITEM) (WXHTREEITEM) item, flag) != 0 )
08b7c251
VZ
592 {
593 wxTreeEvent event(wxEVT_NULL, m_windowId);
594 event.m_item = item;
595
596 bool isExpanded = IsExpanded(item);
2bda0e17 597
08b7c251 598 event.SetEventObject(this);
2bda0e17 599
08b7c251
VZ
600 // @@@ return values of {EXPAND|COLLAPS}ING event handler is discarded
601 event.SetEventType(g_events[isExpanded][TRUE]);
602 GetEventHandler()->ProcessEvent(event);
2bda0e17 603
08b7c251
VZ
604 event.SetEventType(g_events[isExpanded][FALSE]);
605 GetEventHandler()->ProcessEvent(event);
606 }
607 else
608 {
609 // I wonder if it really ever happens...
610 wxLogDebug("TreeView_Expand: change didn't took place.");
611 }
2bda0e17
KB
612}
613
08b7c251 614void wxTreeCtrl::Expand(const wxTreeItemId& item)
2bda0e17 615{
08b7c251 616 DoExpand(item, TVE_EXPAND);
2bda0e17 617}
2bda0e17 618
08b7c251 619void wxTreeCtrl::Collapse(const wxTreeItemId& item)
2bda0e17 620{
08b7c251 621 DoExpand(item, TVE_COLLAPSE);
2bda0e17
KB
622}
623
08b7c251 624void wxTreeCtrl::CollapseAndReset(const wxTreeItemId& item)
2bda0e17 625{
08b7c251 626 DoExpand(item, TVE_COLLAPSERESET);
2bda0e17
KB
627}
628
08b7c251 629void wxTreeCtrl::Toggle(const wxTreeItemId& item)
2bda0e17 630{
08b7c251 631 DoExpand(item, TVE_TOGGLE);
2bda0e17
KB
632}
633
42c5812d
UU
634void wxTreeCtrl::ExpandItem(const wxTreeItemId& item, int action)
635{
636 DoExpand(item, action);
637}
638
08b7c251 639void wxTreeCtrl::Unselect()
2bda0e17 640{
06e38c8e 641 SelectItem(wxTreeItemId((WXHTREEITEM) 0));
08b7c251 642}
02ce7b72 643
08b7c251
VZ
644void wxTreeCtrl::SelectItem(const wxTreeItemId& item)
645{
06e38c8e 646 if ( !TreeView_SelectItem(wxhWnd, (HTREEITEM) (WXHTREEITEM) item) )
2bda0e17 647 {
08b7c251 648 wxLogLastError("TreeView_SelectItem");
2bda0e17 649 }
08b7c251 650}
2bda0e17 651
08b7c251
VZ
652void wxTreeCtrl::EnsureVisible(const wxTreeItemId& item)
653{
654 // no error return
06e38c8e 655 TreeView_EnsureVisible(wxhWnd, (HTREEITEM) (WXHTREEITEM) item);
08b7c251
VZ
656}
657
658void wxTreeCtrl::ScrollTo(const wxTreeItemId& item)
659{
06e38c8e 660 if ( !TreeView_SelectSetFirstVisible(wxhWnd, (HTREEITEM) (WXHTREEITEM) item) )
2bda0e17 661 {
08b7c251 662 wxLogLastError("TreeView_SelectSetFirstVisible");
2bda0e17 663 }
08b7c251
VZ
664}
665
666wxTextCtrl* wxTreeCtrl::GetEditControl() const
667{
668 return m_textCtrl;
669}
670
671void wxTreeCtrl::DeleteTextCtrl()
672{
673 if ( m_textCtrl )
2bda0e17 674 {
08b7c251
VZ
675 m_textCtrl->UnsubclassWin();
676 m_textCtrl->SetHWND(0);
677 delete m_textCtrl;
678 m_textCtrl = NULL;
2bda0e17 679 }
08b7c251 680}
2bda0e17 681
08b7c251
VZ
682wxTextCtrl* wxTreeCtrl::EditLabel(const wxTreeItemId& item,
683 wxClassInfo* textControlClass)
684{
685 wxASSERT( textControlClass->IsKindOf(CLASSINFO(wxTextCtrl)) );
686
06e38c8e 687 HWND hWnd = (HWND) TreeView_EditLabel(wxhWnd, (HTREEITEM) (WXHTREEITEM) item);
2bda0e17 688
08b7c251 689 wxCHECK_MSG( hWnd, NULL, "Can't edit tree ctrl label" );
2bda0e17 690
08b7c251 691 DeleteTextCtrl();
2bda0e17 692
08b7c251
VZ
693 m_textCtrl = (wxTextCtrl *)textControlClass->CreateObject();
694 m_textCtrl->SetHWND((WXHWND)hWnd);
695 m_textCtrl->SubclassWin((WXHWND)hWnd);
2bda0e17 696
08b7c251 697 return m_textCtrl;
2bda0e17
KB
698}
699
08b7c251
VZ
700// End label editing, optionally cancelling the edit
701void wxTreeCtrl::EndEditLabel(const wxTreeItemId& item, bool discardChanges)
2bda0e17 702{
06e38c8e 703 TreeView_EndEditLabelNow(wxhWnd, discardChanges);
08b7c251
VZ
704
705 DeleteTextCtrl();
2bda0e17
KB
706}
707
08b7c251 708wxTreeItemId wxTreeCtrl::HitTest(const wxPoint& point, int& flags)
2bda0e17 709{
08b7c251
VZ
710 TV_HITTESTINFO hitTestInfo;
711 hitTestInfo.pt.x = (int)point.x;
712 hitTestInfo.pt.y = (int)point.y;
2bda0e17 713
06e38c8e 714 TreeView_HitTest(wxhWnd, &hitTestInfo);
2bda0e17 715
08b7c251
VZ
716 flags = 0;
717
718 // avoid repetition
719 #define TRANSLATE_FLAG(flag) if ( hitTestInfo.flags & TVHT_##flag ) \
720 flags |= wxTREE_HITTEST_##flag
721
722 TRANSLATE_FLAG(ABOVE);
723 TRANSLATE_FLAG(BELOW);
724 TRANSLATE_FLAG(NOWHERE);
725 TRANSLATE_FLAG(ONITEMBUTTON);
726 TRANSLATE_FLAG(ONITEMICON);
727 TRANSLATE_FLAG(ONITEMINDENT);
728 TRANSLATE_FLAG(ONITEMLABEL);
729 TRANSLATE_FLAG(ONITEMRIGHT);
730 TRANSLATE_FLAG(ONITEMSTATEICON);
731 TRANSLATE_FLAG(TOLEFT);
732 TRANSLATE_FLAG(TORIGHT);
2bda0e17 733
08b7c251
VZ
734 #undef TRANSLATE_FLAG
735
06e38c8e 736 return wxTreeItemId((WXHTREEITEM) hitTestInfo.hItem);
08b7c251
VZ
737}
738
739void wxTreeCtrl::SortChildren(const wxTreeItemId& item,
740 wxTreeItemCmpFunc *cmpFunction)
741{
742 if ( cmpFunction == NULL )
2bda0e17 743 {
06e38c8e 744 TreeView_SortChildren(wxhWnd, (HTREEITEM) (WXHTREEITEM) item, 0);
2bda0e17 745 }
08b7c251 746 else
2bda0e17 747 {
08b7c251
VZ
748 // TODO: use TreeView_SortChildrenCB
749 wxFAIL_MSG("wxTreeCtrl::SortChildren not implemented");
2bda0e17 750 }
08b7c251 751}
2bda0e17 752
4fabb575
JS
753// TODO
754size_t wxTreeCtrl::GetChildrenCount(const wxTreeItemId& item, bool recursively)
755{
756 return 0;
757}
758
08b7c251
VZ
759// ----------------------------------------------------------------------------
760// implementation
761// ----------------------------------------------------------------------------
2bda0e17 762
08b7c251
VZ
763bool wxTreeCtrl::MSWCommand(WXUINT cmd, WXWORD id)
764{
765 if ( cmd == EN_UPDATE )
2bda0e17 766 {
08b7c251
VZ
767 wxCommandEvent event(wxEVT_COMMAND_TEXT_UPDATED, id);
768 event.SetEventObject( this );
769 ProcessCommand(event);
2bda0e17 770 }
08b7c251 771 else if ( cmd == EN_KILLFOCUS )
2bda0e17 772 {
08b7c251
VZ
773 wxCommandEvent event(wxEVT_KILL_FOCUS, id);
774 event.SetEventObject( this );
775 ProcessCommand(event);
2bda0e17 776 }
08b7c251 777 else
2bda0e17 778 {
08b7c251
VZ
779 // nothing done
780 return FALSE;
2bda0e17 781 }
08b7c251
VZ
782
783 // command processed
784 return TRUE;
785}
786
787// process WM_NOTIFY Windows message
fd3f686c 788bool wxTreeCtrl::MSWNotify(WXWPARAM wParam, WXLPARAM lParam, WXLPARAM *result)
08b7c251
VZ
789{
790 wxTreeEvent event(wxEVT_NULL, m_windowId);
791 wxEventType eventType = wxEVT_NULL;
792 NMHDR *hdr = (NMHDR *)lParam;
793
794 switch ( hdr->code )
2bda0e17 795 {
08b7c251
VZ
796 case TVN_BEGINDRAG:
797 eventType = wxEVT_COMMAND_TREE_BEGIN_DRAG;
798 // fall through
799
800 case TVN_BEGINRDRAG:
801 {
802 if ( eventType == wxEVT_NULL )
803 eventType = wxEVT_COMMAND_TREE_BEGIN_RDRAG;
804 //else: left drag, already set above
805
806 NM_TREEVIEW *tv = (NM_TREEVIEW *)lParam;
807
06e38c8e 808 event.m_item = (WXHTREEITEM) tv->itemNew.hItem;
08b7c251
VZ
809 event.m_pointDrag = wxPoint(tv->ptDrag.x, tv->ptDrag.y);
810 break;
811 }
812
813 case TVN_BEGINLABELEDIT:
814 {
815 eventType = wxEVT_COMMAND_TREE_BEGIN_LABEL_EDIT;
816 TV_DISPINFO *info = (TV_DISPINFO *)lParam;
817
06e38c8e 818 event.m_item = (WXHTREEITEM) info->item.hItem;
08b7c251
VZ
819 break;
820 }
821
822 case TVN_DELETEITEM:
823 {
824 eventType = wxEVT_COMMAND_TREE_DELETE_ITEM;
825 NM_TREEVIEW *tv = (NM_TREEVIEW *)lParam;
826
06e38c8e 827 event.m_item = (WXHTREEITEM) tv->itemOld.hItem;
08b7c251
VZ
828 break;
829 }
830
831 case TVN_ENDLABELEDIT:
832 {
833 eventType = wxEVT_COMMAND_TREE_END_LABEL_EDIT;
834 TV_DISPINFO *info = (TV_DISPINFO *)lParam;
835
06e38c8e 836 event.m_item = (WXHTREEITEM) info->item.hItem;
08b7c251
VZ
837 break;
838 }
839
840 case TVN_GETDISPINFO:
841 eventType = wxEVT_COMMAND_TREE_GET_INFO;
842 // fall through
843
844 case TVN_SETDISPINFO:
845 {
846 if ( eventType == wxEVT_NULL )
847 eventType = wxEVT_COMMAND_TREE_SET_INFO;
848 //else: get, already set above
849
850 TV_DISPINFO *info = (TV_DISPINFO *)lParam;
851
06e38c8e 852 event.m_item = (WXHTREEITEM) info->item.hItem;
08b7c251
VZ
853 break;
854 }
855
856 case TVN_ITEMEXPANDING:
857 event.m_code = FALSE;
858 // fall through
859
860 case TVN_ITEMEXPANDED:
861 {
862 NM_TREEVIEW* tv = (NM_TREEVIEW*)lParam;
863
864 bool expand = FALSE;
865 switch ( tv->action )
866 {
867 case TVE_EXPAND:
868 expand = TRUE;
869 break;
870
871 case TVE_COLLAPSE:
872 expand = FALSE;
873 break;
874
875 default:
876 wxLogDebug("unexpected code %d in TVN_ITEMEXPAND "
877 "message", tv->action);
878 }
879
06e38c8e 880 bool ing = (hdr->code == TVN_ITEMEXPANDING);
08b7c251
VZ
881 eventType = g_events[expand][ing];
882
06e38c8e 883 event.m_item = (WXHTREEITEM) tv->itemNew.hItem;
08b7c251
VZ
884 break;
885 }
886
887 case TVN_KEYDOWN:
888 {
889 eventType = wxEVT_COMMAND_TREE_KEY_DOWN;
890 TV_KEYDOWN *info = (TV_KEYDOWN *)lParam;
891
892 event.m_code = wxCharCodeMSWToWX(info->wVKey);
893 break;
894 }
895
896 case TVN_SELCHANGED:
897 eventType = wxEVT_COMMAND_TREE_SEL_CHANGED;
898 // fall through
899
900 case TVN_SELCHANGING:
901 {
902 if ( eventType == wxEVT_NULL )
903 eventType = wxEVT_COMMAND_TREE_SEL_CHANGING;
904 //else: already set above
905
906 NM_TREEVIEW* tv = (NM_TREEVIEW *)lParam;
907
06e38c8e
JS
908 event.m_item = (WXHTREEITEM) tv->itemNew.hItem;
909 event.m_itemOld = (WXHTREEITEM) tv->itemOld.hItem;
08b7c251
VZ
910 break;
911 }
912
913 default:
fd3f686c 914 return wxControl::MSWNotify(wParam, lParam, result);
2bda0e17 915 }
08b7c251
VZ
916
917 event.SetEventObject(this);
918 event.SetEventType(eventType);
919
fd3f686c 920 bool processed = GetEventHandler()->ProcessEvent(event);
08b7c251
VZ
921
922 // post processing
fd3f686c 923 if ( hdr->code == TVN_DELETEITEM )
2bda0e17 924 {
08b7c251
VZ
925 // NB: we might process this message using wxWindows event tables, but
926 // due to overhead of wxWin event system we prefer to do it here
927 // (otherwise deleting a tree with many items is just too slow)
fd3f686c
VZ
928 NM_TREEVIEW* tv = (NM_TREEVIEW *)lParam;
929 wxTreeItemData *data = (wxTreeItemData *)tv->itemOld.lParam;
930 delete data; // may be NULL, ok
2bda0e17 931 }
08b7c251 932
fd3f686c
VZ
933 *result = !event.IsAllowed();
934
935 return processed;
2bda0e17
KB
936}
937
08b7c251 938// ----------------------------------------------------------------------------
2bda0e17 939// Tree event
08b7c251
VZ
940// ----------------------------------------------------------------------------
941
2bda0e17
KB
942IMPLEMENT_DYNAMIC_CLASS(wxTreeEvent, wxCommandEvent)
943
08b7c251 944wxTreeEvent::wxTreeEvent(wxEventType commandType, int id)
fd3f686c 945 : wxNotifyEvent(commandType, id)
2bda0e17 946{
08b7c251
VZ
947 m_code = 0;
948 m_itemOld = 0;
2bda0e17
KB
949}
950
08b7c251 951#endif // __WIN95__
2bda0e17 952