]> git.saurik.com Git - wxWidgets.git/blob - src/generic/treectrl.cpp
patch from Janos Vegh for improved template handling
[wxWidgets.git] / src / generic / treectrl.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: treectrl.cpp
3 // Purpose: generic tree control implementation
4 // Author: Robert Roebling
5 // Created: 01/02/97
6 // Modified: 22/10/98 - almost total rewrite, simpler interface (VZ)
7 // Id: $Id$
8 // Copyright: (c) 1998 Robert Roebling, Julian Smart and Markus Holzem
9 // Licence: wxWindows licence
10 /////////////////////////////////////////////////////////////////////////////
11
12 // =============================================================================
13 // declarations
14 // =============================================================================
15
16 // -----------------------------------------------------------------------------
17 // headers
18 // -----------------------------------------------------------------------------
19
20 #ifdef __GNUG__
21 #pragma implementation "treectrl.h"
22 #endif
23
24 // For compilers that support precompilation, includes "wx.h".
25 #include "wx/wxprec.h"
26
27 #ifdef __BORLANDC__
28 #pragma hdrstop
29 #endif
30
31 #include "wx/generic/treectrl.h"
32 #include "wx/generic/imaglist.h"
33 #include "wx/settings.h"
34 #include "wx/log.h"
35 #include "wx/intl.h"
36 #include "wx/dynarray.h"
37 #include "wx/dcclient.h"
38 #include "wx/msgdlg.h"
39
40 // -----------------------------------------------------------------------------
41 // array types
42 // -----------------------------------------------------------------------------
43
44 class WXDLLEXPORT wxGenericTreeItem;
45
46 WX_DEFINE_ARRAY(wxGenericTreeItem *, wxArrayTreeItems);
47
48 // -----------------------------------------------------------------------------
49 // private classes
50 // -----------------------------------------------------------------------------
51
52 // a tree item
53 class WXDLLEXPORT wxGenericTreeItem
54 {
55 public:
56 // ctors & dtor
57 wxGenericTreeItem() { m_data = NULL; }
58 wxGenericTreeItem( wxGenericTreeItem *parent,
59 const wxString& text,
60 wxDC& dc,
61 int image, int selImage,
62 wxTreeItemData *data );
63
64 ~wxGenericTreeItem();
65
66 // trivial accessors
67 wxArrayTreeItems& GetChildren() { return m_children; }
68
69 const wxString& GetText() const { return m_text; }
70 int GetImage() const { return m_image; }
71 int GetSelectedImage() const { return m_selImage; }
72 wxTreeItemData *GetData() const { return m_data; }
73
74 void SetText( const wxString &text, wxDC& dc );
75 void SetImage(int image) { m_image = image; }
76 void SetSelectedImage(int image) { m_selImage = image; }
77 void SetData(wxTreeItemData *data) { m_data = data; }
78
79 void SetHasPlus(bool has = TRUE) { m_hasPlus = has; }
80
81 void SetBold(bool bold) { m_isBold = bold; }
82
83 int GetX() const { return m_x; }
84 int GetY() const { return m_y; }
85
86 void SetHeight(int h) { m_height = h; }
87
88 void SetX(int x) { m_x = x; }
89 void SetY(int y) { m_y = y; }
90
91 wxGenericTreeItem *GetParent() const { return m_parent; }
92
93 // operations
94 // deletes all children notifying the treectrl about it if !NULL pointer
95 // given
96 void DeleteChildren(wxTreeCtrl *tree = NULL);
97 // FIXME don't know what is it for
98 void Reset();
99
100 // get count of all children (and grand children if 'recursively')
101 size_t GetChildrenCount(bool recursively = TRUE) const;
102
103 void Insert(wxGenericTreeItem *child, size_t index)
104 { m_children.Insert(child, index); }
105
106 void SetCross( int x, int y );
107 void GetSize( int &x, int &y );
108
109 // return the item at given position (or NULL if no item), onButton is TRUE
110 // if the point belongs to the item's button, otherwise it lies on the
111 // button's label
112 wxGenericTreeItem *HitTest( const wxPoint& point, bool &onButton );
113
114 void Expand() { m_isCollapsed = FALSE; }
115 void Collapse() { m_isCollapsed = TRUE; }
116
117 void SetHilight( bool set = TRUE ) { m_hasHilight = set; }
118
119 // status inquiries
120 bool HasChildren() const { return !m_children.IsEmpty(); }
121 bool HasHilight() const { return m_hasHilight; }
122 bool IsExpanded() const { return !m_isCollapsed; }
123 bool HasPlus() const { return m_hasPlus || HasChildren(); }
124 bool IsBold() const { return m_isBold; }
125
126 private:
127 wxString m_text;
128
129 int m_image,
130 m_selImage;
131
132 wxTreeItemData *m_data;
133
134 // use bitfields to save size
135 int m_isCollapsed :1;
136 int m_hasHilight :1; // same as focused
137 int m_hasPlus :1; // used for item which doesn't have
138 // children but still has a [+] button
139 int m_isBold :1; // render the label in bold font
140
141 int m_x, m_y;
142 long m_height, m_width;
143 int m_xCross, m_yCross;
144 int m_level;
145 wxArrayTreeItems m_children;
146 wxGenericTreeItem *m_parent;
147 };
148
149 // =============================================================================
150 // implementation
151 // =============================================================================
152
153 // -----------------------------------------------------------------------------
154 // wxTreeEvent
155 // -----------------------------------------------------------------------------
156
157 IMPLEMENT_DYNAMIC_CLASS(wxTreeEvent, wxNotifyEvent)
158
159 wxTreeEvent::wxTreeEvent( wxEventType commandType, int id )
160 : wxNotifyEvent( commandType, id )
161 {
162 m_code = 0;
163 m_itemOld = (wxGenericTreeItem *)NULL;
164 }
165
166 // -----------------------------------------------------------------------------
167 // wxGenericTreeItem
168 // -----------------------------------------------------------------------------
169
170 wxGenericTreeItem::wxGenericTreeItem(wxGenericTreeItem *parent,
171 const wxString& text,
172 wxDC& dc,
173 int image, int selImage,
174 wxTreeItemData *data)
175 : m_text(text)
176 {
177 m_image = image;
178 m_selImage = selImage;
179 m_data = data;
180 m_x = m_y = 0;
181 m_xCross = m_yCross = 0;
182
183 m_level = 0;
184
185 m_isCollapsed = TRUE;
186 m_hasHilight = FALSE;
187 m_hasPlus = FALSE;
188 m_isBold = FALSE;
189
190 m_parent = parent;
191
192 dc.GetTextExtent( m_text, &m_width, &m_height );
193 }
194
195 wxGenericTreeItem::~wxGenericTreeItem()
196 {
197 delete m_data;
198
199 wxASSERT_MSG( m_children.IsEmpty(),
200 _T("please call DeleteChildren() before deleting the item") );
201 }
202
203 void wxGenericTreeItem::DeleteChildren(wxTreeCtrl *tree)
204 {
205 size_t count = m_children.Count();
206 for ( size_t n = 0; n < count; n++ )
207 {
208 wxGenericTreeItem *child = m_children[n];
209 if ( tree )
210 {
211 tree->SendDeleteEvent(child);
212 }
213
214 child->DeleteChildren(tree);
215 delete child;
216 }
217
218 m_children.Empty();
219 }
220
221 void wxGenericTreeItem::SetText( const wxString &text, wxDC& dc )
222 {
223 m_text = text;
224
225 dc.GetTextExtent( m_text, &m_width, &m_height );
226 }
227
228 void wxGenericTreeItem::Reset()
229 {
230 m_text.Empty();
231 m_image =
232 m_selImage = -1;
233 m_data = NULL;
234 m_x = m_y =
235 m_height = m_width = 0;
236 m_xCross =
237 m_yCross = 0;
238
239 m_level = 0;
240
241 DeleteChildren();
242 m_isCollapsed = TRUE;
243
244 m_parent = (wxGenericTreeItem *)NULL;
245 }
246
247 size_t wxGenericTreeItem::GetChildrenCount(bool recursively) const
248 {
249 size_t count = m_children.Count();
250 if ( !recursively )
251 return count;
252
253 size_t total = count;
254 for ( size_t n = 0; n < count; n++ )
255 {
256 total += m_children[n]->GetChildrenCount();
257 }
258
259 return total;
260 }
261
262 void wxGenericTreeItem::SetCross( int x, int y )
263 {
264 m_xCross = x;
265 m_yCross = y;
266 }
267
268 void wxGenericTreeItem::GetSize( int &x, int &y )
269 {
270 if ( y < m_y ) y = m_y;
271 int width = m_x + m_width;
272 if (width > x) x = width;
273
274 if (IsExpanded())
275 {
276 size_t count = m_children.Count();
277 for ( size_t n = 0; n < count; n++ )
278 {
279 m_children[n]->GetSize( x, y );
280 }
281 }
282 }
283
284 wxGenericTreeItem *wxGenericTreeItem::HitTest( const wxPoint& point,
285 bool &onButton )
286 {
287 if ((point.y > m_y) && (point.y < m_y + m_height))
288 {
289 // FIXME why +5?
290 // Because that is the size of the plus sign, RR
291 if ((point.x > m_xCross-5) && (point.x < m_xCross+5) &&
292 (point.y > m_yCross-5) && (point.y < m_yCross+5) &&
293 (IsExpanded() || HasPlus()))
294 {
295 onButton = TRUE;
296 return this;
297 }
298
299 /* TODO: we should do a query here like
300 m_imageListNormal->GetSize( item->GetImage(), image_w, image_h ); */
301 int w = m_width;
302 if (m_image != -1) w += 24;
303
304 if ((point.x > m_x) && (point.x < m_x+w))
305 {
306 onButton = FALSE;
307 return this;
308 }
309 }
310 else
311 {
312 if (!m_isCollapsed)
313 {
314 size_t count = m_children.Count();
315 for ( size_t n = 0; n < count; n++ )
316 {
317 wxGenericTreeItem *res = m_children[n]->HitTest( point, onButton );
318 if ( res != NULL )
319 return res;
320 }
321 }
322 }
323
324 return NULL;
325 }
326
327 // -----------------------------------------------------------------------------
328 // wxTreeCtrl implementation
329 // -----------------------------------------------------------------------------
330
331 IMPLEMENT_DYNAMIC_CLASS(wxTreeCtrl, wxScrolledWindow)
332
333 BEGIN_EVENT_TABLE(wxTreeCtrl,wxScrolledWindow)
334 EVT_PAINT (wxTreeCtrl::OnPaint)
335 EVT_MOUSE_EVENTS (wxTreeCtrl::OnMouse)
336 EVT_CHAR (wxTreeCtrl::OnChar)
337 EVT_SET_FOCUS (wxTreeCtrl::OnSetFocus)
338 EVT_KILL_FOCUS (wxTreeCtrl::OnKillFocus)
339 EVT_IDLE (wxTreeCtrl::OnIdle)
340 END_EVENT_TABLE()
341
342 // -----------------------------------------------------------------------------
343 // construction/destruction
344 // -----------------------------------------------------------------------------
345 void wxTreeCtrl::Init()
346 {
347 m_current =
348 m_anchor = (wxGenericTreeItem *) NULL;
349 m_hasFocus = FALSE;
350 m_dirty = FALSE;
351
352 m_xScroll = 0;
353 m_yScroll = 0;
354 m_lineHeight = 10;
355 m_indent = 15;
356 m_spacing = 18;
357
358 m_hilightBrush = new wxBrush
359 (
360 wxSystemSettings::GetSystemColour(wxSYS_COLOUR_HIGHLIGHT),
361 wxSOLID
362 );
363
364 m_imageListNormal =
365 m_imageListState = (wxImageList *) NULL;
366
367 m_dragCount = 0;
368 }
369
370 bool wxTreeCtrl::Create(wxWindow *parent, wxWindowID id,
371 const wxPoint& pos, const wxSize& size,
372 long style,
373 const wxValidator &validator,
374 const wxString& name )
375 {
376 Init();
377
378 wxScrolledWindow::Create( parent, id, pos, size, style|wxHSCROLL|wxVSCROLL, name );
379
380 SetValidator( validator );
381
382 SetBackgroundColour( *wxWHITE );
383 m_dottedPen = wxPen( *wxBLACK, 0, 0 );
384
385 return TRUE;
386 }
387
388 wxTreeCtrl::~wxTreeCtrl()
389 {
390 wxDELETE( m_hilightBrush );
391
392 DeleteAllItems();
393 }
394
395 // -----------------------------------------------------------------------------
396 // accessors
397 // -----------------------------------------------------------------------------
398
399 size_t wxTreeCtrl::GetCount() const
400 {
401 return m_anchor == NULL ? 0u : m_anchor->GetChildrenCount();
402 }
403
404 void wxTreeCtrl::SetIndent(unsigned int indent)
405 {
406 m_indent = indent;
407 m_dirty = TRUE;
408 Refresh();
409 }
410
411 void wxTreeCtrl::SetSpacing(unsigned int spacing)
412 {
413 m_spacing = spacing;
414 m_dirty = TRUE;
415 Refresh();
416 }
417
418 size_t wxTreeCtrl::GetChildrenCount(const wxTreeItemId& item, bool recursively)
419 {
420 wxCHECK_MSG( item.IsOk(), 0u, _T("invalid tree item") );
421
422 return item.m_pItem->GetChildrenCount(recursively);
423 }
424
425 // -----------------------------------------------------------------------------
426 // functions to work with tree items
427 // -----------------------------------------------------------------------------
428
429 wxString wxTreeCtrl::GetItemText(const wxTreeItemId& item) const
430 {
431 wxCHECK_MSG( item.IsOk(), _T(""), _T("invalid tree item") );
432
433 return item.m_pItem->GetText();
434 }
435
436 int wxTreeCtrl::GetItemImage(const wxTreeItemId& item) const
437 {
438 wxCHECK_MSG( item.IsOk(), -1, _T("invalid tree item") );
439
440 return item.m_pItem->GetImage();
441 }
442
443 int wxTreeCtrl::GetItemSelectedImage(const wxTreeItemId& item) const
444 {
445 wxCHECK_MSG( item.IsOk(), -1, _T("invalid tree item") );
446
447 return item.m_pItem->GetSelectedImage();
448 }
449
450 wxTreeItemData *wxTreeCtrl::GetItemData(const wxTreeItemId& item) const
451 {
452 wxCHECK_MSG( item.IsOk(), NULL, _T("invalid tree item") );
453
454 return item.m_pItem->GetData();
455 }
456
457 void wxTreeCtrl::SetItemText(const wxTreeItemId& item, const wxString& text)
458 {
459 wxCHECK_RET( item.IsOk(), _T("invalid tree item") );
460
461 wxClientDC dc(this);
462 wxGenericTreeItem *pItem = item.m_pItem;
463 pItem->SetText(text, dc);
464 RefreshLine(pItem);
465 }
466
467 void wxTreeCtrl::SetItemImage(const wxTreeItemId& item, int image)
468 {
469 wxCHECK_RET( item.IsOk(), _T("invalid tree item") );
470
471 wxGenericTreeItem *pItem = item.m_pItem;
472 pItem->SetImage(image);
473 RefreshLine(pItem);
474 }
475
476 void wxTreeCtrl::SetItemSelectedImage(const wxTreeItemId& item, int image)
477 {
478 wxCHECK_RET( item.IsOk(), _T("invalid tree item") );
479
480 wxGenericTreeItem *pItem = item.m_pItem;
481 pItem->SetSelectedImage(image);
482 RefreshLine(pItem);
483 }
484
485 void wxTreeCtrl::SetItemData(const wxTreeItemId& item, wxTreeItemData *data)
486 {
487 wxCHECK_RET( item.IsOk(), _T("invalid tree item") );
488
489 item.m_pItem->SetData(data);
490 }
491
492 void wxTreeCtrl::SetItemHasChildren(const wxTreeItemId& item, bool has)
493 {
494 wxCHECK_RET( item.IsOk(), _T("invalid tree item") );
495
496 wxGenericTreeItem *pItem = item.m_pItem;
497 pItem->SetHasPlus(has);
498 RefreshLine(pItem);
499 }
500
501 void wxTreeCtrl::SetItemBold(const wxTreeItemId& item, bool bold)
502 {
503 wxCHECK_RET( item.IsOk(), _T("invalid tree item") );
504
505 // avoid redrawing the tree if no real change
506 wxGenericTreeItem *pItem = item.m_pItem;
507 if ( pItem->IsBold() != bold )
508 {
509 pItem->SetBold(bold);
510 RefreshLine(pItem);
511 }
512 }
513
514 // -----------------------------------------------------------------------------
515 // item status inquiries
516 // -----------------------------------------------------------------------------
517
518 bool wxTreeCtrl::IsVisible(const wxTreeItemId& WXUNUSED(item)) const
519 {
520 wxFAIL_MSG(_T("not implemented"));
521
522 return TRUE;
523 }
524
525 bool wxTreeCtrl::ItemHasChildren(const wxTreeItemId& item) const
526 {
527 wxCHECK_MSG( item.IsOk(), FALSE, _T("invalid tree item") );
528
529 return !item.m_pItem->GetChildren().IsEmpty();
530 }
531
532 bool wxTreeCtrl::IsExpanded(const wxTreeItemId& item) const
533 {
534 wxCHECK_MSG( item.IsOk(), FALSE, _T("invalid tree item") );
535
536 return item.m_pItem->IsExpanded();
537 }
538
539 bool wxTreeCtrl::IsSelected(const wxTreeItemId& item) const
540 {
541 wxCHECK_MSG( item.IsOk(), FALSE, _T("invalid tree item") );
542
543 return item.m_pItem->HasHilight();
544 }
545
546 bool wxTreeCtrl::IsBold(const wxTreeItemId& item) const
547 {
548 wxCHECK_MSG( item.IsOk(), FALSE, _T("invalid tree item") );
549
550 return item.m_pItem->IsBold();
551 }
552
553 // -----------------------------------------------------------------------------
554 // navigation
555 // -----------------------------------------------------------------------------
556
557 wxTreeItemId wxTreeCtrl::GetParent(const wxTreeItemId& item) const
558 {
559 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
560
561 return item.m_pItem->GetParent();
562 }
563
564 wxTreeItemId wxTreeCtrl::GetFirstChild(const wxTreeItemId& item, long& cookie) const
565 {
566 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
567
568 cookie = 0;
569 return GetNextChild(item, cookie);
570 }
571
572 wxTreeItemId wxTreeCtrl::GetNextChild(const wxTreeItemId& item, long& cookie) const
573 {
574 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
575
576 wxArrayTreeItems& children = item.m_pItem->GetChildren();
577 if ( (size_t)cookie < children.Count() )
578 {
579 return children.Item(cookie++);
580 }
581 else
582 {
583 // there are no more of them
584 return wxTreeItemId();
585 }
586 }
587
588 wxTreeItemId wxTreeCtrl::GetLastChild(const wxTreeItemId& item) const
589 {
590 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
591
592 wxArrayTreeItems& children = item.m_pItem->GetChildren();
593 return (children.IsEmpty() ? wxTreeItemId() : wxTreeItemId(children.Last()));
594 }
595
596 wxTreeItemId wxTreeCtrl::GetNextSibling(const wxTreeItemId& item) const
597 {
598 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
599
600 wxGenericTreeItem *i = item.m_pItem;
601 wxGenericTreeItem *parent = i->GetParent();
602 if ( parent == NULL )
603 {
604 // root item doesn't have any siblings
605 return wxTreeItemId();
606 }
607
608 wxArrayTreeItems& siblings = parent->GetChildren();
609 int index = siblings.Index(i);
610 wxASSERT( index != wxNOT_FOUND ); // I'm not a child of my parent?
611
612 size_t n = (size_t)(index + 1);
613 return n == siblings.Count() ? wxTreeItemId() : wxTreeItemId(siblings[n]);
614 }
615
616 wxTreeItemId wxTreeCtrl::GetPrevSibling(const wxTreeItemId& item) const
617 {
618 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
619
620 wxGenericTreeItem *i = item.m_pItem;
621 wxGenericTreeItem *parent = i->GetParent();
622 if ( parent == NULL )
623 {
624 // root item doesn't have any siblings
625 return wxTreeItemId();
626 }
627
628 wxArrayTreeItems& siblings = parent->GetChildren();
629 int index = siblings.Index(i);
630 wxASSERT( index != wxNOT_FOUND ); // I'm not a child of my parent?
631
632 return index == 0 ? wxTreeItemId()
633 : wxTreeItemId(siblings[(size_t)(index - 1)]);
634 }
635
636 wxTreeItemId wxTreeCtrl::GetFirstVisibleItem() const
637 {
638 wxFAIL_MSG(_T("not implemented"));
639
640 return wxTreeItemId();
641 }
642
643 wxTreeItemId wxTreeCtrl::GetNextVisible(const wxTreeItemId& item) const
644 {
645 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
646
647 wxFAIL_MSG(_T("not implemented"));
648
649 return wxTreeItemId();
650 }
651
652 wxTreeItemId wxTreeCtrl::GetPrevVisible(const wxTreeItemId& item) const
653 {
654 wxCHECK_MSG( item.IsOk(), wxTreeItemId(), _T("invalid tree item") );
655
656 wxFAIL_MSG(_T("not implemented"));
657
658 return wxTreeItemId();
659 }
660
661 // -----------------------------------------------------------------------------
662 // operations
663 // -----------------------------------------------------------------------------
664
665 wxTreeItemId wxTreeCtrl::DoInsertItem(const wxTreeItemId& parentId,
666 size_t previous,
667 const wxString& text,
668 int image, int selImage,
669 wxTreeItemData *data)
670 {
671 wxGenericTreeItem *parent = parentId.m_pItem;
672 if ( !parent )
673 {
674 // should we give a warning here?
675 return AddRoot(text, image, selImage, data);
676 }
677
678 wxClientDC dc(this);
679 wxGenericTreeItem *item = new wxGenericTreeItem(parent,
680 text, dc,
681 image, selImage,
682 data);
683
684 if ( data != NULL )
685 {
686 data->m_pItem = item;
687 }
688
689 parent->Insert( item, previous );
690
691 m_dirty = TRUE;
692
693 return item;
694 }
695
696 wxTreeItemId wxTreeCtrl::AddRoot(const wxString& text,
697 int image, int selImage,
698 wxTreeItemData *data)
699 {
700 wxCHECK_MSG( !m_anchor, wxTreeItemId(), _T("tree can have only one root") );
701
702 wxClientDC dc(this);
703 m_anchor = new wxGenericTreeItem((wxGenericTreeItem *)NULL, text, dc,
704 image, selImage, data);
705 if ( data != NULL )
706 {
707 data->m_pItem = m_anchor;
708 }
709
710 AdjustMyScrollbars();
711 Refresh();
712
713 return m_anchor;
714 }
715
716 wxTreeItemId wxTreeCtrl::PrependItem(const wxTreeItemId& parent,
717 const wxString& text,
718 int image, int selImage,
719 wxTreeItemData *data)
720 {
721 return DoInsertItem(parent, 0u, text, image, selImage, data);
722 }
723
724 wxTreeItemId wxTreeCtrl::InsertItem(const wxTreeItemId& parentId,
725 const wxTreeItemId& idPrevious,
726 const wxString& text,
727 int image, int selImage,
728 wxTreeItemData *data)
729 {
730 wxGenericTreeItem *parent = parentId.m_pItem;
731 if ( !parent )
732 {
733 // should we give a warning here?
734 return AddRoot(text, image, selImage, data);
735 }
736
737 int index = parent->GetChildren().Index(idPrevious.m_pItem);
738 wxASSERT_MSG( index != wxNOT_FOUND,
739 _T("previous item in wxTreeCtrl::InsertItem() is not a sibling") );
740 return DoInsertItem(parentId, (size_t)index, text, image, selImage, data);
741 }
742
743 wxTreeItemId wxTreeCtrl::AppendItem(const wxTreeItemId& parentId,
744 const wxString& text,
745 int image, int selImage,
746 wxTreeItemData *data)
747 {
748 wxGenericTreeItem *parent = parentId.m_pItem;
749 if ( !parent )
750 {
751 // should we give a warning here?
752 return AddRoot(text, image, selImage, data);
753 }
754
755 return DoInsertItem(parent, parent->GetChildren().Count(), text,
756 image, selImage, data);
757 }
758
759 void wxTreeCtrl::SendDeleteEvent(wxGenericTreeItem *item)
760 {
761 wxTreeEvent event( wxEVT_COMMAND_TREE_DELETE_ITEM, GetId() );
762 event.m_item = item;
763 event.SetEventObject( this );
764 ProcessEvent( event );
765 }
766
767 void wxTreeCtrl::DeleteChildren(const wxTreeItemId& itemId)
768 {
769 wxGenericTreeItem *item = itemId.m_pItem;
770 item->DeleteChildren(this);
771
772 m_dirty = TRUE;
773 }
774
775 void wxTreeCtrl::Delete(const wxTreeItemId& itemId)
776 {
777 wxGenericTreeItem *item = itemId.m_pItem;
778 wxGenericTreeItem *parent = item->GetParent();
779
780 if ( parent )
781 {
782 parent->GetChildren().Remove(item);
783 }
784
785 item->DeleteChildren(this);
786 SendDeleteEvent(item);
787 delete item;
788
789 m_dirty = TRUE;
790 }
791
792 void wxTreeCtrl::DeleteAllItems()
793 {
794 if ( m_anchor )
795 {
796 m_anchor->DeleteChildren(this);
797 delete m_anchor;
798
799 m_anchor = NULL;
800
801 m_dirty = TRUE;
802 }
803 }
804
805 void wxTreeCtrl::Expand(const wxTreeItemId& itemId)
806 {
807 wxGenericTreeItem *item = itemId.m_pItem;
808
809 if ( !item->HasPlus() )
810 return;
811
812 if ( item->IsExpanded() )
813 return;
814
815 wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_EXPANDING, GetId() );
816 event.m_item = item;
817 event.SetEventObject( this );
818 if ( ProcessEvent( event ) && event.m_code )
819 {
820 // cancelled by program
821 return;
822 }
823
824 item->Expand();
825 CalculatePositions();
826
827 RefreshSubtree(item);
828
829 event.SetEventType(wxEVT_COMMAND_TREE_ITEM_EXPANDED);
830 ProcessEvent( event );
831 }
832
833 void wxTreeCtrl::Collapse(const wxTreeItemId& itemId)
834 {
835 wxGenericTreeItem *item = itemId.m_pItem;
836
837 if ( !item->IsExpanded() )
838 return;
839
840 wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_COLLAPSING, GetId() );
841 event.m_item = item;
842 event.SetEventObject( this );
843 if ( ProcessEvent( event ) && event.m_code )
844 {
845 // cancelled by program
846 return;
847 }
848
849 item->Collapse();
850
851 wxArrayTreeItems& children = item->GetChildren();
852 size_t count = children.Count();
853 for ( size_t n = 0; n < count; n++ )
854 {
855 Collapse(children[n]);
856 }
857
858 CalculatePositions();
859
860 RefreshSubtree(item);
861
862 event.SetEventType(wxEVT_COMMAND_TREE_ITEM_COLLAPSED);
863 ProcessEvent( event );
864 }
865
866 void wxTreeCtrl::CollapseAndReset(const wxTreeItemId& item)
867 {
868 Collapse(item);
869 DeleteChildren(item);
870 }
871
872 void wxTreeCtrl::Toggle(const wxTreeItemId& itemId)
873 {
874 wxGenericTreeItem *item = itemId.m_pItem;
875
876 if ( item->IsExpanded() )
877 Collapse(itemId);
878 else
879 Expand(itemId);
880 }
881
882 void wxTreeCtrl::Unselect()
883 {
884 if ( m_current )
885 {
886 m_current->SetHilight( FALSE );
887 RefreshLine( m_current );
888 }
889 }
890
891 void wxTreeCtrl::SelectItem(const wxTreeItemId& itemId)
892 {
893 wxGenericTreeItem *item = itemId.m_pItem;
894
895 if ( m_current != item )
896 {
897 wxTreeEvent event( wxEVT_COMMAND_TREE_SEL_CHANGING, GetId() );
898 event.m_item = item;
899 event.m_itemOld = m_current;
900 event.SetEventObject( this );
901 if ( GetEventHandler()->ProcessEvent( event ) && event.WasVetoed() )
902 return;
903
904 if ( m_current )
905 {
906 m_current->SetHilight( FALSE );
907 RefreshLine( m_current );
908 }
909
910 m_current = item;
911 m_current->SetHilight( TRUE );
912 RefreshLine( m_current );
913
914 event.SetEventType(wxEVT_COMMAND_TREE_SEL_CHANGED);
915 GetEventHandler()->ProcessEvent( event );
916 }
917 }
918
919 void wxTreeCtrl::EnsureVisible(const wxTreeItemId& item)
920 {
921 wxGenericTreeItem *gitem = item.m_pItem;
922
923 // first expand all parent branches
924 wxGenericTreeItem *parent = gitem->GetParent();
925 while ( parent && !parent->IsExpanded() )
926 {
927 Expand(parent);
928
929 parent = parent->GetParent();
930 }
931
932 // now scroll to the item
933 int item_y = gitem->GetY();
934
935 int start_x = 0;
936 int start_y = 0;
937 ViewStart( &start_x, &start_y );
938 start_y *= 10;
939
940 int client_h = 0;
941 int client_w = 0;
942 GetClientSize( &client_w, &client_h );
943
944 if (item_y < start_y+3)
945 {
946 int x = 0;
947 int y = 0;
948 m_anchor->GetSize( x, y );
949 y += 2*m_lineHeight;
950 int x_pos = GetScrollPos( wxHORIZONTAL );
951 SetScrollbars( 10, 10, x/10, y/10, x_pos, (item_y-client_h/2)/10 );
952 }
953 else if (item_y > start_y+client_h-16)
954 {
955 int x = 0;
956 int y = 0;
957 m_anchor->GetSize( x, y );
958 y += 2*m_lineHeight;
959 int x_pos = GetScrollPos( wxHORIZONTAL );
960 SetScrollbars( 10, 10, x/10, y/10, x_pos, (item_y-client_h/2)/10 );
961 }
962 }
963
964 void wxTreeCtrl::ScrollTo(const wxTreeItemId& WXUNUSED(item))
965 {
966 wxFAIL_MSG(_T("not implemented"));
967 }
968
969 wxTextCtrl *wxTreeCtrl::EditLabel( const wxTreeItemId& WXUNUSED(item),
970 wxClassInfo* WXUNUSED(textCtrlClass) )
971 {
972 wxFAIL_MSG(_T("not implemented"));
973
974 return (wxTextCtrl*)NULL;
975 }
976
977 wxTextCtrl *wxTreeCtrl::GetEditControl() const
978 {
979 wxFAIL_MSG(_T("not implemented"));
980
981 return (wxTextCtrl*)NULL;
982 }
983
984 void wxTreeCtrl::EndEditLabel(const wxTreeItemId& WXUNUSED(item), bool WXUNUSED(discardChanges))
985 {
986 wxFAIL_MSG(_T("not implemented"));
987 }
988
989 // FIXME: tree sorting functions are not reentrant and not MT-safe!
990 static wxTreeCtrl *s_treeBeingSorted = NULL;
991
992 static int tree_ctrl_compare_func(wxGenericTreeItem **item1,
993 wxGenericTreeItem **item2)
994 {
995 wxCHECK_MSG( s_treeBeingSorted, 0, _T("bug in wxTreeCtrl::SortChildren()") );
996
997 return s_treeBeingSorted->OnCompareItems(*item1, *item2);
998 }
999
1000 int wxTreeCtrl::OnCompareItems(const wxTreeItemId& item1,
1001 const wxTreeItemId& item2)
1002 {
1003 return wxStrcmp(GetItemText(item1), GetItemText(item2));
1004 }
1005
1006 void wxTreeCtrl::SortChildren(const wxTreeItemId& itemId)
1007 {
1008 wxCHECK_RET( itemId.IsOk(), _T("invalid tree item") );
1009
1010 wxGenericTreeItem *item = itemId.m_pItem;
1011
1012 wxCHECK_RET( !s_treeBeingSorted,
1013 _T("wxTreeCtrl::SortChildren is not reentrant") );
1014
1015 wxArrayTreeItems& children = item->GetChildren();
1016 if ( children.Count() > 1 )
1017 {
1018 s_treeBeingSorted = this;
1019 children.Sort(tree_ctrl_compare_func);
1020 s_treeBeingSorted = NULL;
1021
1022 m_dirty = TRUE;
1023 }
1024 //else: don't make the tree dirty as nothing changed
1025 }
1026
1027 wxImageList *wxTreeCtrl::GetImageList() const
1028 {
1029 return m_imageListNormal;
1030 }
1031
1032 wxImageList *wxTreeCtrl::GetStateImageList() const
1033 {
1034 return m_imageListState;
1035 }
1036
1037 void wxTreeCtrl::SetImageList(wxImageList *imageList)
1038 {
1039 m_imageListNormal = imageList;
1040 // calculate a m_lineHeight value from the image sizes
1041 wxPaintDC dc(this);
1042 PrepareDC( dc );
1043 m_lineHeight = (int)(dc.GetCharHeight() + 4);
1044 int
1045 width = 0,
1046 height = 0,
1047 n = m_imageListNormal->GetImageCount();
1048 for(int i = 0; i < n ; i++)
1049 {
1050 m_imageListNormal->GetSize(i, width, height);
1051 height += height/5; //20% extra spacing
1052 if(height > m_lineHeight) m_lineHeight = height;
1053 }
1054 }
1055
1056 void wxTreeCtrl::SetStateImageList(wxImageList *imageList)
1057 {
1058 m_imageListState = imageList;
1059 }
1060
1061 // -----------------------------------------------------------------------------
1062 // helpers
1063 // -----------------------------------------------------------------------------
1064
1065 void wxTreeCtrl::AdjustMyScrollbars()
1066 {
1067 if (m_anchor)
1068 {
1069 int x = 0;
1070 int y = 0;
1071 m_anchor->GetSize( x, y );
1072 y += 2*m_lineHeight;
1073 int x_pos = GetScrollPos( wxHORIZONTAL );
1074 int y_pos = GetScrollPos( wxVERTICAL );
1075 SetScrollbars( 10, 10, x/10, y/10, x_pos, y_pos );
1076 }
1077 else
1078 {
1079 SetScrollbars( 0, 0, 0, 0 );
1080 }
1081 }
1082
1083 void wxTreeCtrl::PaintItem(wxGenericTreeItem *item, wxDC& dc)
1084 {
1085 // render bold items in bold
1086 wxFont fontOld;
1087 wxFont fontNew;
1088
1089 if (item->IsBold())
1090 {
1091 fontOld = dc.GetFont();
1092 if (fontOld.Ok())
1093 {
1094 // VZ: is there any better way to make a bold variant of old font?
1095 fontNew = wxFont( fontOld.GetPointSize(),
1096 fontOld.GetFamily(),
1097 fontOld.GetStyle(),
1098 wxBOLD,
1099 fontOld.GetUnderlined());
1100 dc.SetFont(fontNew);
1101 }
1102 else
1103 {
1104 wxFAIL_MSG(_T("wxDC::GetFont() failed!"));
1105 }
1106 }
1107
1108 long text_w = 0;
1109 long text_h = 0;
1110 dc.GetTextExtent( item->GetText(), &text_w, &text_h );
1111
1112 int image_h = 0;
1113 int image_w = 0;
1114 if ((item->IsExpanded()) && (item->GetSelectedImage() != -1))
1115 {
1116 m_imageListNormal->GetSize( item->GetSelectedImage(), image_w, image_h );
1117 image_w += 4;
1118 }
1119 else if (item->GetImage() != -1)
1120 {
1121 m_imageListNormal->GetSize( item->GetImage(), image_w, image_h );
1122 image_w += 4;
1123 }
1124
1125 int total_h = (image_h > text_h) ? image_h : text_h;
1126 if(m_lineHeight > total_h) total_h = m_lineHeight;
1127
1128 dc.DrawRectangle( item->GetX()-2, item->GetY(), image_w+text_w+4, total_h );
1129
1130 if ((item->IsExpanded()) && (item->GetSelectedImage() != -1))
1131 {
1132 dc.SetClippingRegion( item->GetX(), item->GetY(), image_w-2, total_h );
1133 m_imageListNormal->Draw( item->GetSelectedImage(), dc,
1134 item->GetX(),
1135 item->GetY() +((total_h > image_h)?((total_h-image_h)/2):0),
1136 wxIMAGELIST_DRAW_TRANSPARENT );
1137 dc.DestroyClippingRegion();
1138 }
1139 else if (item->GetImage() != -1)
1140 {
1141 dc.SetClippingRegion( item->GetX(), item->GetY(), image_w-2, total_h );
1142 m_imageListNormal->Draw( item->GetImage(), dc,
1143 item->GetX(),
1144 item->GetY() +((total_h > image_h)?((total_h-image_h)/2):0),
1145 wxIMAGELIST_DRAW_TRANSPARENT );
1146 dc.DestroyClippingRegion();
1147 }
1148
1149 dc.SetBackgroundMode(wxTRANSPARENT);
1150 dc.DrawText( item->GetText(), image_w + item->GetX(), item->GetY()
1151 + ((total_h > text_h) ? (total_h - text_h)/2 : 0));
1152
1153 // restore normal font for bold items
1154 if (fontOld.Ok())
1155 {
1156 dc.SetFont( fontOld);
1157 }
1158 }
1159
1160 void wxTreeCtrl::PaintLevel( wxGenericTreeItem *item, wxDC &dc, int level, int &y )
1161 {
1162 int horizX = level*m_indent;
1163
1164 item->SetX( horizX+m_indent+m_spacing );
1165 item->SetY( y-m_lineHeight/2 );
1166 item->SetHeight( m_lineHeight );
1167
1168 item->SetCross( horizX+m_indent, y );
1169
1170 int oldY = y;
1171
1172 int exposed_x = dc.LogicalToDeviceX( 0 );
1173 int exposed_y = dc.LogicalToDeviceY( item->GetY()-2 );
1174
1175 if (IsExposed( exposed_x, exposed_y, 10000, m_lineHeight+4 )) // 10000 = very much
1176 {
1177 int startX = horizX;
1178 int endX = horizX + (m_indent-5);
1179
1180 // if (!item->HasChildren()) endX += (m_indent+5);
1181 if (!item->HasChildren()) endX += 20;
1182
1183 dc.DrawLine( startX, y, endX, y );
1184
1185 if (item->HasPlus())
1186 {
1187 dc.DrawLine( horizX+(m_indent+5), y, horizX+(m_indent+15), y );
1188 dc.SetPen( *wxGREY_PEN );
1189 dc.SetBrush( *wxWHITE_BRUSH );
1190 dc.DrawRectangle( horizX+(m_indent-5), y-4, 11, 9 );
1191 dc.SetPen( *wxBLACK_PEN );
1192 dc.DrawLine( horizX+(m_indent-2), y, horizX+(m_indent+3), y );
1193
1194 if (!item->IsExpanded())
1195 {
1196 dc.DrawLine( horizX+m_indent, y-2, horizX+m_indent, y+3 );
1197 }
1198 }
1199
1200 if (item->HasHilight())
1201 {
1202 dc.SetTextForeground( wxSystemSettings::GetSystemColour( wxSYS_COLOUR_HIGHLIGHTTEXT ) );
1203
1204 dc.SetBrush( *m_hilightBrush );
1205
1206 if (m_hasFocus)
1207 dc.SetPen( *wxBLACK_PEN );
1208 else
1209 dc.SetPen( *wxTRANSPARENT_PEN );
1210
1211 PaintItem(item, dc);
1212
1213 dc.SetPen( *wxBLACK_PEN );
1214 dc.SetTextForeground( *wxBLACK );
1215 dc.SetBrush( *wxWHITE_BRUSH );
1216 }
1217 else
1218 {
1219 dc.SetBrush( *wxWHITE_BRUSH );
1220 dc.SetPen( *wxTRANSPARENT_PEN );
1221
1222 PaintItem(item, dc);
1223
1224 dc.SetPen( *wxBLACK_PEN );
1225 }
1226 }
1227
1228 if (item->IsExpanded())
1229 {
1230 int semiOldY = y;
1231
1232 wxArrayTreeItems& children = item->GetChildren();
1233 size_t count = children.Count();
1234 for ( size_t n = 0; n < count; n++ )
1235 {
1236 y += m_lineHeight;
1237 semiOldY = y;
1238 PaintLevel( children[n], dc, level+1, y );
1239 }
1240
1241 // it may happen that the item is expanded but has no items (when you
1242 // delete all its children for example) - don't draw the vertical line
1243 // in this case
1244 if (count > 0)
1245 dc.DrawLine( horizX+m_indent, oldY+5, horizX+m_indent, semiOldY );
1246 }
1247 }
1248
1249 // -----------------------------------------------------------------------------
1250 // wxWindows callbacks
1251 // -----------------------------------------------------------------------------
1252
1253 void wxTreeCtrl::OnPaint( wxPaintEvent &WXUNUSED(event) )
1254 {
1255 if ( !m_anchor )
1256 return;
1257
1258 wxPaintDC dc(this);
1259 PrepareDC( dc );
1260
1261 dc.SetFont( wxSystemSettings::GetSystemFont( wxSYS_DEFAULT_GUI_FONT ) );
1262
1263 dc.SetPen( m_dottedPen );
1264 if(GetImageList() == NULL)
1265 m_lineHeight = (int)(dc.GetCharHeight() + 4);
1266
1267 int y = m_lineHeight / 2 + 2;
1268 PaintLevel( m_anchor, dc, 0, y );
1269 }
1270
1271 void wxTreeCtrl::OnSetFocus( wxFocusEvent &WXUNUSED(event) )
1272 {
1273 m_hasFocus = TRUE;
1274
1275 if (m_current) RefreshLine( m_current );
1276 }
1277
1278 void wxTreeCtrl::OnKillFocus( wxFocusEvent &WXUNUSED(event) )
1279 {
1280 m_hasFocus = FALSE;
1281
1282 if (m_current) RefreshLine( m_current );
1283 }
1284
1285 void wxTreeCtrl::OnChar( wxKeyEvent &event )
1286 {
1287 wxTreeEvent te( wxEVT_COMMAND_TREE_KEY_DOWN, GetId() );
1288 te.m_code = event.KeyCode();
1289 te.SetEventObject( this );
1290 GetEventHandler()->ProcessEvent( te );
1291
1292 if (m_current == 0)
1293 {
1294 event.Skip();
1295 return;
1296 }
1297
1298 switch (event.KeyCode())
1299 {
1300 case '+':
1301 case WXK_ADD:
1302 if (m_current->HasPlus() && !IsExpanded(m_current))
1303 {
1304 Expand(m_current);
1305 }
1306 break;
1307
1308 case '-':
1309 case WXK_SUBTRACT:
1310 if (IsExpanded(m_current))
1311 {
1312 Collapse(m_current);
1313 }
1314 break;
1315
1316 case '*':
1317 case WXK_MULTIPLY:
1318 Toggle(m_current);
1319 break;
1320
1321 case ' ':
1322 case WXK_RETURN:
1323 {
1324 wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_ACTIVATED, GetId() );
1325 event.m_item = m_current;
1326 event.m_code = 0;
1327 event.SetEventObject( this );
1328 GetEventHandler()->ProcessEvent( event );
1329 }
1330 break;
1331
1332 // up goes to the previous sibling or to the last of its children if
1333 // it's expanded
1334 case WXK_UP:
1335 {
1336 wxTreeItemId prev = GetPrevSibling( m_current );
1337 if (!prev)
1338 {
1339 prev = GetParent( m_current );
1340 if (prev)
1341 {
1342 long cockie = 0;
1343 wxTreeItemId current = m_current;
1344 if (current == GetFirstChild( prev, cockie ))
1345 {
1346 // otherwise we return to where we came from
1347 SelectItem( prev );
1348 EnsureVisible( prev );
1349 break;
1350 }
1351 }
1352 }
1353 if (prev)
1354 {
1355 while ( IsExpanded(prev) && HasChildren(prev) )
1356 {
1357 wxTreeItemId child = GetLastChild(prev);
1358 if ( child )
1359 {
1360 prev = child;
1361 }
1362 }
1363
1364 SelectItem( prev );
1365 EnsureVisible( prev );
1366 }
1367 }
1368 break;
1369
1370 // left arrow goes to the parent
1371 case WXK_LEFT:
1372 {
1373 wxTreeItemId prev = GetParent( m_current );
1374 if (prev)
1375 {
1376 EnsureVisible( prev );
1377 SelectItem( prev );
1378 }
1379 }
1380 break;
1381
1382 case WXK_RIGHT:
1383 // this works the same as the down arrow except that we also expand the
1384 // item if it wasn't expanded yet
1385 Expand(m_current);
1386 // fall through
1387
1388 case WXK_DOWN:
1389 {
1390 if (IsExpanded(m_current) && HasChildren(m_current))
1391 {
1392 long cookie = 0;
1393 wxTreeItemId child = GetFirstChild( m_current, cookie );
1394 SelectItem( child );
1395 EnsureVisible( child );
1396 }
1397 else
1398 {
1399 wxTreeItemId next = GetNextSibling( m_current );
1400 if (next == 0)
1401 {
1402 wxTreeItemId current = m_current;
1403 while (current && !next)
1404 {
1405 current = GetParent( current );
1406 if (current) next = GetNextSibling( current );
1407 }
1408 }
1409 if (next != 0)
1410 {
1411 SelectItem( next );
1412 EnsureVisible( next );
1413 }
1414 }
1415 }
1416 break;
1417
1418 // <End> selects the last visible tree item
1419 case WXK_END:
1420 {
1421 wxTreeItemId last = GetRootItem();
1422
1423 while ( last.IsOk() && IsExpanded(last) )
1424 {
1425 wxTreeItemId lastChild = GetLastChild(last);
1426
1427 // it may happen if the item was expanded but then all of
1428 // its children have been deleted - so IsExpanded() returned
1429 // TRUE, but GetLastChild() returned invalid item
1430 if ( !lastChild )
1431 break;
1432
1433 last = lastChild;
1434 }
1435
1436 if ( last.IsOk() )
1437 {
1438 EnsureVisible( last );
1439 SelectItem( last );
1440 }
1441 }
1442 break;
1443
1444 // <Home> selects the root item
1445 case WXK_HOME:
1446 {
1447 wxTreeItemId prev = GetRootItem();
1448 if (prev)
1449 {
1450 EnsureVisible( prev );
1451 SelectItem( prev );
1452 }
1453 }
1454 break;
1455
1456 default:
1457 event.Skip();
1458 }
1459 }
1460
1461 wxTreeItemId wxTreeCtrl::HitTest(const wxPoint& point, int& WXUNUSED(flags))
1462 {
1463 wxClientDC dc(this);
1464 PrepareDC(dc);
1465 long x = dc.DeviceToLogicalX( (long)point.x );
1466 long y = dc.DeviceToLogicalY( (long)point.y );
1467
1468 bool onButton = FALSE;
1469 return m_anchor->HitTest( wxPoint(x, y), onButton );
1470 }
1471
1472 void wxTreeCtrl::OnMouse( wxMouseEvent &event )
1473 {
1474 if (!event.LeftIsDown()) m_dragCount = 0;
1475
1476 if ( !(event.LeftDown() || event.LeftDClick() || event.Dragging()) ) return;
1477
1478 if ( !m_anchor ) return;
1479
1480 wxClientDC dc(this);
1481 PrepareDC(dc);
1482 long x = dc.DeviceToLogicalX( (long)event.GetX() );
1483 long y = dc.DeviceToLogicalY( (long)event.GetY() );
1484
1485 bool onButton = FALSE;
1486 wxGenericTreeItem *item = m_anchor->HitTest( wxPoint(x,y), onButton );
1487
1488 if (item == NULL) return; /* we hit the blank area */
1489
1490 if (event.Dragging())
1491 {
1492 if (m_dragCount == 2) /* small drag latency (3?) */
1493 {
1494 m_dragCount = 0;
1495
1496 wxTreeEvent nevent(wxEVT_COMMAND_TREE_BEGIN_DRAG, GetId());
1497 nevent.m_item = m_current;
1498 nevent.SetEventObject(this);
1499 GetEventHandler()->ProcessEvent(nevent);
1500 }
1501 else
1502 {
1503 m_dragCount++;
1504 }
1505 return;
1506 }
1507
1508 if (!IsSelected(item))
1509 SelectItem(item); /* we dont support multiple selections, BTW */
1510
1511 if (event.LeftDClick())
1512 {
1513 wxTreeEvent event( wxEVT_COMMAND_TREE_ITEM_ACTIVATED, GetId() );
1514 event.m_item = item;
1515 event.m_code = 0;
1516 event.SetEventObject( this );
1517 GetEventHandler()->ProcessEvent( event );
1518 }
1519
1520 if (onButton)
1521 {
1522 Toggle( item );
1523 }
1524 }
1525
1526 void wxTreeCtrl::OnIdle( wxIdleEvent &WXUNUSED(event) )
1527 {
1528 /* after all changes have been done to the tree control,
1529 * we actually redraw the tree when everything is over */
1530
1531 if (!m_dirty)
1532 return;
1533
1534 m_dirty = FALSE;
1535
1536 CalculatePositions();
1537
1538 AdjustMyScrollbars();
1539 }
1540
1541 // -----------------------------------------------------------------------------
1542
1543 void wxTreeCtrl::CalculateLevel( wxGenericTreeItem *item, wxDC &dc, int level, int &y )
1544 {
1545 int horizX = level*m_indent;
1546
1547 item->SetX( horizX+m_indent+m_spacing );
1548 item->SetY( y-m_lineHeight/2 );
1549 item->SetHeight( m_lineHeight );
1550
1551 if ( !item->IsExpanded() )
1552 {
1553 // we dont need to calculate collapsed branches
1554 return;
1555 }
1556
1557 wxArrayTreeItems& children = item->GetChildren();
1558 size_t count = children.Count();
1559 for ( size_t n = 0; n < count; n++ )
1560 {
1561 y += m_lineHeight;
1562 CalculateLevel( children[n], dc, level+1, y ); // recurse
1563 }
1564 }
1565
1566 void wxTreeCtrl::CalculatePositions()
1567 {
1568 if ( !m_anchor ) return;
1569
1570 wxClientDC dc(this);
1571 PrepareDC( dc );
1572
1573 dc.SetFont( wxSystemSettings::GetSystemFont( wxSYS_DEFAULT_GUI_FONT ) );
1574
1575 dc.SetPen( m_dottedPen );
1576 if(GetImageList() == NULL)
1577 m_lineHeight = (int)(dc.GetCharHeight() + 4);
1578
1579 int y = m_lineHeight / 2 + 2;
1580 CalculateLevel( m_anchor, dc, 0, y ); // start recursion
1581 }
1582
1583 void wxTreeCtrl::RefreshSubtree(wxGenericTreeItem *item)
1584 {
1585 wxClientDC dc(this);
1586 PrepareDC(dc);
1587
1588 int cw = 0;
1589 int ch = 0;
1590 GetClientSize( &cw, &ch );
1591
1592 wxRect rect;
1593 rect.x = dc.LogicalToDeviceX( 0 );
1594 rect.width = cw;
1595 rect.y = dc.LogicalToDeviceY( item->GetY() );
1596 rect.height = ch;
1597
1598 Refresh( TRUE, &rect );
1599
1600 AdjustMyScrollbars();
1601 }
1602
1603 void wxTreeCtrl::RefreshLine( wxGenericTreeItem *item )
1604 {
1605 wxClientDC dc(this);
1606 PrepareDC( dc );
1607
1608 wxRect rect;
1609 rect.x = dc.LogicalToDeviceX( item->GetX() - 2 );
1610 rect.y = dc.LogicalToDeviceY( item->GetY() - 2 );
1611 rect.width = 1000;
1612 rect.height = dc.GetCharHeight() + 6;
1613
1614 Refresh( TRUE, &rect );
1615 }
1616