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